Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, GradioUI, InferenceClientModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, GradioUI, InferenceClientModel, ManagedAgent
 
 # 1. THE BRAIN
 model = InferenceClientModel(
@@ -8,29 +8,38 @@ model = InferenceClientModel(
 )
 
 # 2. THE RESEARCHER (Specialist)
-# We define it as a standard CodeAgent. 
-# Providing a 'name' and 'description' is what makes it "manageable."
-researcher = CodeAgent(
+# We wrap the researcher in ManagedAgent so the Boss can "call" it.
+search_agent = CodeAgent(
     tools=[DuckDuckGoSearchTool()],
     model=model,
     name="research_specialist",
-    description="Expert at web search and retrieving technical data. Use this agent for any web-based questions."
+    description="Finds technical data and web information. Use for all research tasks."
 )
 
-# 3. THE BOSS (Manager)
-# We pass the researcher agent directly into the managed_agents list.
+managed_researcher = ManagedAgent(
+    agent=search_agent,
+    name="research_specialist",
+    description="A web-search specialist. Call this agent when you need current data from the internet."
+)
+
+# 3. THE BOSS LOGIC (The Template Fix)
+# This dictionary overrides the read-only system_prompt property.
+custom_templates = {
+    "system_prompt": """You are the BOSS OPERATOR (Elite PhD Level). 
+    1. Delegate sub-tasks to 'research_specialist' for web-based data.
+    2. Use your internal Python tools for math, simulation, and data analysis.
+    3. Always verify data before presenting a final answer.
+    4. Format all math in LaTeX ($E=mc^2$) and all data in Markdown tables."""
+}
+
+# 4. THE BOSS (Manager)
 manager = CodeAgent(
     tools=[], 
     model=model,
-    managed_agents=[researcher],
-    add_base_tools=True
+    managed_agents=[managed_researcher],
+    add_base_tools=True,
+    prompt_templates=custom_templates # <--- THIS IS THE CRITICAL FIX
 )
 
-# Set the Mission for the Boss
-manager.system_prompt = """You are the BOSS OPERATOR (Elite PhD Level). 
-1. Delegate sub-tasks to 'research_specialist' for web data.
-2. Use Python for math ($E=mc^2$) and data analysis.
-3. Provide final answers in professional Markdown tables."""
-
 if __name__ == "__main__":
     GradioUI(manager).launch()