Update app.py

app.py

@@ -16,58 +16,32 @@ class BasicAgent:
         print("BasicAgent initialized.")
         
         # Create model using Hugging Face Inference API
-        # This will use the HF_TOKEN environment variable if available
-        try:
-            # Using a good open model available on HF Inference API
-            self.model = LiteLLMModel(
-                model_id="huggingface/meta-llama/Llama-3.2-3B-Instruct",
-                api_base="https://api-inference.huggingface.co/models/",
-                # HF_TOKEN will be automatically used if set in environment
-                max_tokens=1024
-            )
-            print("Initialized Hugging Face Inference API model")
-        except Exception as e:
-            print(f"Error initializing HF model: {e}")
-            # Fallback to a simpler approach if model fails
-            self.model = None
+        self.model = LiteLLMModel(
+            model_id="huggingface/meta-llama/Llama-3.2-3B-Instruct",
+            api_base="https://api-inference.huggingface.co/models/",
+            max_tokens=1024
+        )
+        print("Initialized Hugging Face Inference API model")
         
-        # Create the CodeAgent with tools if model is available
-        if self.model:
-            try:
-                self.agent = CodeAgent(
-                    tools=[
-                        DuckDuckGoSearchTool(),
-                        VisitWebpageTool(),
-                        FinalAnswerTool()
-                    ],
-                    model=self.model,
-                    additional_authorized_imports=["wikipedia", "requests", "json", "re", "datetime", "os"]
-                )
-                print("CodeAgent with tools initialized successfully")
-            except Exception as e:
-                print(f"Error initializing CodeAgent: {e}")
-                self.agent = None
-        else:
-            self.agent = None
+        # Create the CodeAgent with tools
+        self.agent = CodeAgent(
+            tools=[
+                DuckDuckGoSearchTool(),
+                VisitWebpageTool(),
+                FinalAnswerTool()
+            ],
+            model=self.model,
+            additional_authorized_imports=["wikipedia", "requests", "json", "re", "datetime", "os"]
+        )
+        print("CodeAgent with tools initialized successfully")
     
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         
-        if self.agent:
-            try:
-                # Use smolagents CodeAgent to process the question
-                answer = self.agent.run(question)
-                print(f"Agent returning answer: {answer[:100]}...")
-                return answer
-            except Exception as e:
-                print(f"Error in agent processing: {e}")
-                # Return a fallback answer in case of errors
-                return f"Error processing question: {str(e)}"
-        else:
-            # Fallback if agent initialization failed
-            fallback_answer = "Agent initialization failed. Please check model configuration."
-            print(f"Agent returning fallback answer: {fallback_answer}")
-            return fallback_answer
+        # Use smolagents CodeAgent to process the question
+        answer = self.agent.run(question)
+        print(f"Agent returning answer: {answer[:100]}...")
+        return answer
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """