Update agent.py

agent.py

@@ -1,5 +1,5 @@
 import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, OpenAIModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIModel
 from dotenv import load_dotenv
 
 # Load environment variables (API keys)
@@ -14,13 +14,20 @@ class CustomAgent:
             # Using OpenAI gpt-4o-mini as requested/seen in logs
             self.model = OpenAIModel(model_id="gpt-4o-mini", api_key=api_key)
         else:
-            # Fallback to Hugging Face Model if no OpenAI key is set
-            self.model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+            # If no OpenAI key, we can use a Hugging Face model
+            # Note: HfApiModel has been renamed to LiteLLMModel or InferenceClientModel in some versions.
+            # However, since you are using OpenAI gpt-4o-mini, let's stick to that.
+            # If we need a HF fallback, we import InferenceClientModel inside
+            try:
+                from smolagents import LiteLLMModel
+                self.model = LiteLLMModel(model_id="huggingface/Qwen/Qwen2.5-Coder-32B-Instruct")
+            except ImportError:
+                # If LiteLLMModel is also not found, use whatever is available
+                raise Exception("OPENAI_API_KEY not found and fallback model could not be initialized.")
             
         self.search_tool = DuckDuckGoSearchTool()
         
         # Initialize the agent
-        # CodeAgent is the best choice for the final project
         self.agent = CodeAgent(
             tools=[self.search_tool],
             model=self.model,
@@ -42,7 +49,6 @@ class CustomAgent:
             result = self.agent.run(question)
             
             # CRITICAL: Always return a string
-            # If the result is a list (which matches your error), this converts it
             if isinstance(result, list):
                 return " ".join(map(str, result))