Skip to content

Commit fd01b73

Browse files
committed
fix(llm): fixed gemini api_key
1 parent 8c0b46e commit fd01b73

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

scrapegraphai/models/gemini.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,6 @@ class Gemini(ChatGoogleGenerativeAI):
1515
"""
1616

1717
def __init__(self, llm_config: dict):
18+
# replace "api_key" to "google_api_key"
19+
llm_config["google_api_key"] = llm_config.pop("api_key", None)
1820
super().__init__(**llm_config)

0 commit comments

Comments
 (0)