Skip to content

Commit efe76d2

Browse files
committed
chore: update parameter names for consistency
1 parent 33ca5e0 commit efe76d2

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
@@ -45,7 +45,7 @@ def get_llm_model(provider: str, **kwargs):
4545
base_url = kwargs.get("base_url")
4646

4747
return ChatAnthropic(
48-
model_name=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
48+
model=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
4949
temperature=kwargs.get("temperature", 0.0),
5050
base_url=base_url,
5151
api_key=api_key,
@@ -102,7 +102,7 @@ def get_llm_model(provider: str, **kwargs):
102102
return ChatGoogleGenerativeAI(
103103
model=kwargs.get("model_name", "gemini-2.0-flash-exp"),
104104
temperature=kwargs.get("temperature", 0.0),
105-
google_api_key=api_key,
105+
api_key=api_key,
106106
)
107107
elif provider == "ollama":
108108
if not kwargs.get("base_url", ""):

0 commit comments

Comments
 (0)