Skip to content

Commit 0c29506

Browse files
authored
Merge pull request #410 from marginal23326/chore/update-params
chore: update parameter names for consistency
2 parents 94bba83 + efe76d2 commit 0c29506

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/utils/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def get_llm_model(provider: str, **kwargs):
4646
base_url = kwargs.get("base_url")
4747

4848
return ChatAnthropic(
49-
model_name=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
49+
model=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
5050
temperature=kwargs.get("temperature", 0.0),
5151
base_url=base_url,
5252
api_key=api_key,
@@ -103,7 +103,7 @@ def get_llm_model(provider: str, **kwargs):
103103
return ChatGoogleGenerativeAI(
104104
model=kwargs.get("model_name", "gemini-2.0-flash-exp"),
105105
temperature=kwargs.get("temperature", 0.0),
106-
google_api_key=api_key,
106+
api_key=api_key,
107107
)
108108
elif provider == "ollama":
109109
if not kwargs.get("base_url", ""):

0 commit comments

Comments
 (0)