Update agent.py

agent.py

@@ -1,7 +1,7 @@
 import os
 import dotenv
 import yaml
-from smolagents import CodeAgent, HfApiModel, LiteLLMModel, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
+from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
 from tools import ToolReadFiles, ToolReverseString, ToolDownloadImage, FinalAnswerTool 
 
 dotenv.load_dotenv()
@@ -14,23 +14,23 @@ 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 = 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-lite",  
+            api_key=self.gemini_key 
+        )
                 
         self.agent = CodeAgent(
             model=self.model,