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.
2 parents 94bba83 + efe76d2 commit 0c29506Copy full SHA for 0c29506
src/utils/utils.py
@@ -46,7 +46,7 @@ def get_llm_model(provider: str, **kwargs):
46
base_url = kwargs.get("base_url")
47
48
return ChatAnthropic(
49
- model_name=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
+ model=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
50
temperature=kwargs.get("temperature", 0.0),
51
base_url=base_url,
52
api_key=api_key,
@@ -103,7 +103,7 @@ def get_llm_model(provider: str, **kwargs):
103
return ChatGoogleGenerativeAI(
104
model=kwargs.get("model_name", "gemini-2.0-flash-exp"),
105
106
- google_api_key=api_key,
+ api_key=api_key,
107
)
108
elif provider == "ollama":
109
if not kwargs.get("base_url", ""):
0 commit comments