We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 33ca5e0 commit efe76d2Copy full SHA for efe76d2
src/utils/utils.py
@@ -45,7 +45,7 @@ def get_llm_model(provider: str, **kwargs):
45
base_url = kwargs.get("base_url")
46
47
return ChatAnthropic(
48
- model_name=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
+ model=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
49
temperature=kwargs.get("temperature", 0.0),
50
base_url=base_url,
51
api_key=api_key,
@@ -102,7 +102,7 @@ def get_llm_model(provider: str, **kwargs):
102
return ChatGoogleGenerativeAI(
103
model=kwargs.get("model_name", "gemini-2.0-flash-exp"),
104
105
- google_api_key=api_key,
+ api_key=api_key,
106
)
107
elif provider == "ollama":
108
if not kwargs.get("base_url", ""):
0 commit comments