CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-06-25 21:11:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(llm): fixed gemini api_key

Browse Source
This commit is contained in:
Marco Perini 2024-05-06 13:17:47 +02:00
parent 8c0b46eb40
commit fd01b73b71
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scrapegraphai/models/gemini.py
View File

@ -15,4 +15,6 @@ class Gemini(ChatGoogleGenerativeAI):
""" """
def __init__(self, llm_config: dict): 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) super().__init__(**llm_config)