Update app.py

multiagents with websearch and final answer.

app.py

@@ -4,6 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, tool
+from smolagents import OpenAIServerModel
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -12,22 +13,66 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     agent = None
-    def __init__(self):
+    web_agent = None
+    def check_reasoning(final_answer,agent_memory):
+        model = OpenAIServerModel("gpt-4o", max_tokens=8096)
+        prompt = (
+            f"Here is a user-given task and the agent steps: {agent_memory.get_succinct_steps()}."
+            "Please check that the reasoning process is correct: do it correctly answer the given task?"
+            "First list reasons why yes/no, then write your final decision: PASS in caps lock if it is satisfactory, FAIL if it is not."
+            
+        )
+        messages = [
+        {
+            "role": "user",
+            "content": [
+                {
+                    "type": "text",
+                    "text": prompt,
+                }
+            ],
+            }
+        ]
+        output = model(messages).content
+        print("Feedback: ", output)
+        if "FAIL" in output:
+            raise Exception(output)
+        return True
+    def __init__(self):      
         prompt = "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
         model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together")
+        
+        web_agent = CodeAgent(
+                model=model,
+                tools=[
+                GoogleSearchTool(provider="serper"),
+                VisitWebpageTool()
+                ],
+                name="web_agent",
+                description="Browses the web to find information",
+                verbosity_level=0,
+                max_steps=10,
+            )
+        
         agent = CodeAgent(
-        model=model,
+        model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),
         tools=[GoogleSearchTool(), VisitWebpageTool(), calculate_cargo_travel_time],
-        additional_authorized_imports=["pandas"],
+        managed_agents=[web_agent],
+        additional_authorized_imports=["pandas","json","numpy"],
+        planning_interval=5,
         max_steps=20,
+        final_answer_checks=[check_reasoning],
         )
         print("BasicAgent initialized.")
+    
     def __call__(self, question: str) -> str:
         answer = agent.run(question)
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         #fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {answer}")
-        return answer
+        final_answer = answer.replace("FINAL ANSWER:","").strip()
+        
+        print(f"Agent returning fixed answer: {final_answer}")
+        return final_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """