Update agent.py

agent.py

@@ -1,14 +1,11 @@
 import os
 import dotenv
 import yaml
-from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
+from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
 from tools import ToolReadFiles, ToolReverseString, ToolDownloadImage, FinalAnswerTool 
 
 dotenv.load_dotenv()
 
-# HF_API_KEY = os.getenv("HF_API_KEY")
-# OPENAI_KEY = os.getenv("OPEN_AI_KEY")
-
 # Create prompt template
 # with open("prompts.yaml", 'r') as stream:
 #    prompt_templates = yaml.safe_load(stream)
@@ -34,9 +31,7 @@ class Agent:
             model_id="gemini/gemini-2.0-flash-exp",  
             api_key=self.gemini_key 
         )
-        
-
-        
+                
         self.agent = CodeAgent(
             model=self.model,
             add_base_tools=True,