From fd01b73b71b515206cfdf51c1d52136293494389 Mon Sep 17 00:00:00 2001 From: Marco Perini Date: Mon, 6 May 2024 13:17:47 +0200 Subject: [PATCH] fix(llm): fixed gemini api_key --- scrapegraphai/models/gemini.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scrapegraphai/models/gemini.py b/scrapegraphai/models/gemini.py index b8c79f4f..1c939c6c 100644 --- a/scrapegraphai/models/gemini.py +++ b/scrapegraphai/models/gemini.py @@ -15,4 +15,6 @@ class Gemini(ChatGoogleGenerativeAI): """ def __init__(self, llm_config: dict): + # replace "api_key" to "google_api_key" + llm_config["google_api_key"] = llm_config.pop("api_key", None) super().__init__(**llm_config)