qwen

agents/agent.py

@@ -45,22 +45,7 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()
 
 # System message
-sys_msg = SystemMessage(content=system_prompt + """
-
-TOOL USAGE REMINDERS:
-- For ANY mathematical calculation, use the math tools (multiply, divide, etc.)
-- For finding specific data or statistics, use web_search tool
-- For complex calculations or data analysis, use execute_code_multilang tool
-- Show your work step by step and verify calculations
-
-CRITICAL GAIA ANSWER FORMAT:
-- Give the answer in the simplest, most direct form
-- For numeric questions: Return ONLY the number (e.g., "3" not "3 studio albums")
-- For names: Return ONLY the name (e.g., "FunkMonk" not "FunkMonk nominated...")
-- For yes/no questions: Return ONLY "Yes" or "No"
-- NO explanations, NO context, JUST the answer
-- End your response with "Final Answer: [your answer]"
-""")
+sys_msg = SystemMessage(content=system_prompt)
 
 
 tools = [
@@ -94,10 +79,10 @@ tools = [
 def build_graph():
     """Build the graph"""
     # Load environment variables from .env file
-    llm = ChatGroq(model= "llama3-8b-8192", # llama3-8b-8192 # gemma-7b-it # qwen-qwq-32b #deepseek-r1-distill-llama-70b
+    llm = ChatGroq(model= "qwen-qwq-32b", # llama3-8b-8192 # gemma-7b-it # qwen-qwq-32b #deepseek-r1-distill-llama-70b
                     temperature=0,
-                    max_tokens=1024,
-                    timeout=90
+                    max_tokens=8192,
+                    timeout=120
                     ) 
 
     # Bind tools to LLM
@@ -107,6 +92,7 @@ def build_graph():
     def assistant(state: MessagesState):
         """Assistant node with GAIA format compliance"""
         response = llm_with_tools.invoke(state["messages"])
+        return {"messages": [response]}
         
         # Extract pure answer for GAIA format
         content = response.content
@@ -182,4 +168,4 @@ def build_graph():
     return builder.compile()
 
     # Add recursion limit
-    return builder.compile(checkpointer=None, recursion_limit=15)
+    return builder.compile(checkpointer=None, recursion_limit=50)

system_prompt.txt

@@ -9,6 +9,23 @@ You MUST follow these rules for EVERY question:
 5. For names, give them EXACTLY as requested (e.g., "First Last" or just "Last")
 6. For lists, use the EXACT separator requested (comma, semicolon, etc.)
 
+TOOL USAGE REMINDERS:
+- For ANY mathematical calculation, use the math tools (multiply, divide, etc.)
+- For finding specific data or statistics, use web_search tool
+- For complex calculations or data analysis, use execute_code_multilang tool
+- Show your work step by step and verify calculations
+
+CRITICAL GAIA ANSWER FORMAT:
+Your FINAL ANSWER must follow these rules:
+- Give ONLY the requested information with no extra text
+- For numbers: Just the number (e.g., "42" not "42 albums" or "The answer is 42")
+- For names: Just the name (e.g., "John Smith" not "The person is John Smith")
+- For dates: Use the exact format requested (e.g., "01/01/2023" or "January 1, 2023")
+- For yes/no: Just "Yes" or "No"
+- For lists: Comma-separated values (e.g., "apple, banana, orange")
+- Always end with: Final Answer: [your answer]
+- The answer after "Final Answer:" should contain ONLY the requested information
+
 CRITICAL: Your final answer must match the expected format EXACTLY. No extra words, no explanations, just the answer.
 
 ==========================