model changed

app/utils/model_loader.py

@@ -43,14 +43,14 @@ class ModelLoader(BaseModel):
         elif self.model_provider =="gemini":
             print("Loading model from gemini:")
             load_dotenv()
-            # gemini_api_key = os.getenv("GEMINI_API_KEY")
-            # model_name = self.config["llm"]["gemini"]["model_name"]
-            # llm = ChatGoogleGenerativeAI(
-            #     model=model_name,
-            #     google_api_key= gemini_api_key
-            # )
-            openai_api_key = os.getenv("OPENAI_API_KEY")
-            llm = ChatOpenAI(model="gpt-4o-mini",api_key=openai_api_key )
+            gemini_api_key = os.getenv("GEMINI_API_KEY")
+            model_name = self.config["llm"]["gemini"]["model_name"]
+            llm = ChatGoogleGenerativeAI(
+                model=model_name,
+                google_api_key= gemini_api_key
+            )
+            # openai_api_key = os.getenv("OPENAI_API_KEY")
+            # llm = ChatOpenAI(model="gpt-4o-mini",api_key=openai_api_key )
 
 
         elif self.model_provider =="openai":