♻️ LLM change

app.py

@@ -26,7 +26,6 @@ from tools import (
     normalize_place_name,
     is_drink,
     download_youtube_comments,
-    get_youtube_transcript,
 )
 
 # (Keep Constants as is)
@@ -38,16 +37,26 @@ final_answer = FinalAnswerTool()
 web_search = DuckDuckGoSearchTool()
 visit_webpage = VisitWebpageTool()
 wiki_search = WikipediaSearchTool()
-authorized_imports = ["pandas", "random", "time", "builtins.open"]
+authorized_imports = ["pandas", "random", "time", "builtins.open", "bs4"]
 py_run = PythonInterpreterTool(authorized_imports=authorized_imports)
 
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    model_id="Qwen/Qwen3-14b",
+    # model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     custom_role_conversions=None,
 )
 
+gaia_prompt = """
+---
+You are given tools to explore Wikipedia pages.
+- Use `list_wikipedia_sections(page_title)` whenever you need to know which sections exist.
+- If a section you want (e.g., "Discography") isn’t listed, you *must* call this tool *first*, inspect its output, and choose the closest alternative section name (e.g., "Studio albums" or "Albums").
+- After finding a valid section title, call the appropriate extraction tool (`extract_nth_table_…`, `count_tables_…`, etc.).
+- Do not attempt to parse the entire text manually without verifying section names first.
+"""
+
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
@@ -93,11 +102,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
                 normalize_place_name,
                 is_drink,
                 download_youtube_comments,
-                get_youtube_transcript,
             ],
             verbosity_level=2,
+            planning_interval=4,
             additional_authorized_imports=authorized_imports,
         )
+        agent.prompt_templates["system_prompt"] = (
+            agent.prompt_templates["system_prompt"] + gaia_prompt
+        )
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None