Update agent.py

agent.py

@@ -17,24 +17,24 @@ dotenv.load_dotenv()
 class Agent:
     def __init__(self):
         # self.hf_token = os.getenv("HF_API_KEY")
-        self.openai_key = os.getenv("OPEN_AI_KEY")
-        # self.gemini_key = os.getenv("GEMINI_API_KEY")
+        # self.openai_key = os.getenv("OPEN_AI_KEY")
+        self.gemini_key = os.getenv("GEMINI_API_KEY")
         
         # self.model = HfApiModel(
         #    model_id="Qwen/Qwen3-235B-A22B",
         #    token=self.hf_token       
         # )
         
-        self.model = OpenAIServerModel(
-            model_id="gpt-4o-mini-2024-07-18",
-            api_key=self.openai_key
-        )
-        
-        # self.model_local = LiteLLMModel(
-        #    model_id="gemini/gemini-2.0-flash-exp",  
-        #    api_key=self.gemini_key 
+        # self.model = OpenAIServerModel(
+        #    model_id="gpt-4o-mini-2024-07-18",
+        #    api_key=self.openai_key
         # )
         
+        self.model = LiteLLMModel(
+            model_id="gemini/gemini-2.0-flash-exp",  
+            api_key=self.gemini_key 
+        )
+        
 
         
         self.agent = CodeAgent(