From efe76d21459a29a901b387d8ebae823cd4a741ba Mon Sep 17 00:00:00 2001
From: marginal23326 <58261815+marginal23326@users.noreply.github.com>
Date: Tue, 18 Mar 2025 05:18:21 +0600
Subject: [PATCH] chore: update parameter names for consistency

---
 src/utils/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/utils/utils.py b/src/utils/utils.py
index b604812..3c43f1c 100644
--- a/src/utils/utils.py
+++ b/src/utils/utils.py
@@ -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", ""):