chore: update parameter names for consistency

This commit is contained in:
marginal23326
2025-03-18 05:18:21 +06:00
parent 33ca5e0bbb
commit efe76d2145

View File

@@ -45,7 +45,7 @@ def get_llm_model(provider: str, **kwargs):
base_url = kwargs.get("base_url")
return ChatAnthropic(
model_name=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
model=kwargs.get("model_name", "claude-3-5-sonnet-20241022"),
temperature=kwargs.get("temperature", 0.0),
base_url=base_url,
api_key=api_key,
@@ -102,7 +102,7 @@ def get_llm_model(provider: str, **kwargs):
return ChatGoogleGenerativeAI(
model=kwargs.get("model_name", "gemini-2.0-flash-exp"),
temperature=kwargs.get("temperature", 0.0),
google_api_key=api_key,
api_key=api_key,
)
elif provider == "ollama":
if not kwargs.get("base_url", ""):