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)