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

Merge pull request #159 from VinciGit00/fix-gemini-apikey

Browse Source 
fix(llm): fixed gemini api_key
This commit is contained in:
Marco Perini 2024-05-06 13:21:25 +02:00 committed by GitHub
commit b0539530ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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):
# replace "api_key" to "google_api_key"
llm_config["google_api_key"] = llm_config.pop("api_key", None)
super().__init__(**llm_config)