Update app.py

app.py

@@ -12,7 +12,8 @@ import regex
 import pandas as pd
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
-from smolagents import CodeAgent, PromptTemplates, VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool
+from smolagents import CodeAgent, VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool
+from smolagents import PromptTemplates, PlanningPromptTemplate, ManagedAgentPromptTemplate, FinalAnswerPromptTemplate
 from smolagents.models import ChatMessage
 from custom_tools import WebpageStructureAnalyzerTool
 
@@ -253,14 +254,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         combined_system_prompt = default_system_prompt + "\n\n" + my_added_prompts
         # (4) Create the prompt templates to plug in CodeAgent
         # default_prompt_templates = PromptTemplates()
-        my_prompt_templates = {"system_prompt": combined_system_prompt}
+        # my_prompt_templates = {"system_prompt": combined_system_prompt}
         #                       "planning": default_prompt_templates["planning"],
         #                       "managed_agent": default_prompt_templates["managed_agent"],
         #                       "final_answer": default_prompt_templates["final_answer"]}
-        # my_prompt_templates = PromptTemplates(system_prompt=combined_system_prompt,
-        #                                     planning=default_prompt_templates["planning"],
-        #                                     managed_agent=default_prompt_templates["managed_agent"],
-        #                                     final_answer=default_prompt_templates["final_answer"])
+        my_prompt_templates = PromptTemplates(system_prompt=combined_system_prompt,
+                                            planning=PlanningPromptTemplate(initial_plan="", update_plan_pre_messages="", update_plan_post_messages=""),
+                                            managed_agent=ManagedAgentPromptTemplate(task="", report=""),
+                                            final_answer=FinalAnswerPromptTemplate(pre_messages="", post_messages=""))
         # (5) Create the CodeAgent, passing the LLM wrapper and tools
         agent = CodeAgent(model=llm_model, 
                           tools=[web_structure_analyzer_tool, wiki_tool, search_tool, python_tool, html_parse_tool],