duration

app.py

@@ -18,7 +18,16 @@ You are a general AI assistant. I will ask you a question. Report your thoughts,
 
 IMPORTANT: When using tools that require file access (such as audio_transcriber_tool, excel_tool, analyze_code_tool, or image_tool), ALWAYS use the task_id parameter only. Do NOT use any file names mentioned by the user - ignore them completely and only pass the task_id.
 
-You MUST always provide a FINAL ANSWER. If you cannot find complete information, provide your best estimate based on available information.
+SEARCH STRATEGY: 
+- If wikipedia_search_tool fails or returns insufficient/irrelevant results, try these fallback strategies:
+  1. Try wikipedia_search_tool again with a broader, more general query (remove specific terms, use synonyms)
+  2. If Wikipedia still doesn't help, try arxiv_search_tool for academic/research topics
+  3. You can use multiple search attempts with different keywords to find better information
+- Always evaluate if the search results are relevant and sufficient before proceeding to your final answer
+- IMPORTANT: When you see [END_OF_SEARCH] in tool results, this means the search is complete and you have all available information
+- Do NOT perform additional searches after seeing [END_OF_SEARCH] - immediately proceed to analyze the provided information and give your final answer
+
+You MUST always provide a FINAL ANSWER. If you reach the recursion limit or tools fail, provide the best answer possible with available information.
 """
 
 
@@ -31,38 +40,49 @@ class BasicAgent:
         """Run the agent and return whatever FINAL_ANSWER the graph produces."""
         print(f"Agent received question: {question}")
         
-        # Create system prompt with task_id included
-        system_prompt_with_task = SYSTEM_PROMPT
-        if task_id:
-            system_prompt_with_task += f"\n\nIMPORTANT: Your current task_id is: {task_id}. When using any tools that require a task_id parameter (audio_transcriber_tool, excel_tool, analyze_code_tool, image_tool), use this exact task_id: {task_id}"
-        
-        # Initialize the state properly with all required fields
-        init_state = {
-            "messages": [
-                SystemMessage(content=system_prompt_with_task),
-                HumanMessage(content=question)
-            ]
-        }
-        
-        # Run the agent with increased steps
-        out_state = self.graph.invoke(init_state, {"recursion_limit": 15})
-        
-        # Extract the final answer from the last message
-        if out_state and "messages" in out_state:
-            last_message = out_state["messages"][-1]
-            if hasattr(last_message, 'content'):
-                content = last_message.content
-                print("content: ", content)
-                print("\n\n\n\n")
-                
-                # Look for FINAL ANSWER: pattern
-                if "FINAL ANSWER:" in content:
-                    final_answer = content.split("FINAL ANSWER:")[-1].strip()
-                    return final_answer
-                else:
+        try:
+            # Create system prompt with task_id included
+            system_prompt_with_task = SYSTEM_PROMPT
+            if task_id:
+                system_prompt_with_task += f"\n\nIMPORTANT: Your current task_id is: {task_id}. When using any tools that require a task_id parameter (audio_transcriber_tool, excel_tool, analyze_code_tool, image_tool), use this exact task_id: {task_id}"
+            
+            # Initialize the state properly with all required fields
+            init_state = {
+                "messages": [
+                    SystemMessage(content=system_prompt_with_task),
+                    HumanMessage(content=question)
+                ]
+            }
+            
+            # Run the agent with increased steps
+            out_state = self.graph.invoke(init_state, {"recursion_limit": 15})
+            
+            # Extract the final answer from the last message
+            if out_state and "messages" in out_state:
+                last_message = out_state["messages"][-1]
+                if hasattr(last_message, 'content') and last_message.content:
+                    content = last_message.content
+                    print("content: ", content)
+                    print("\n\n\n\n")
+                    
+                    # Look for FINAL ANSWER: pattern (case insensitive)
+                    if "FINAL ANSWER:" in content.upper():
+                        # Find the last occurrence of FINAL ANSWER
+                        parts = content.upper().split("FINAL ANSWER:")
+                        if len(parts) > 1:
+                            # Get the original case version
+                            answer_start = content.upper().rfind("FINAL ANSWER:") + len("FINAL ANSWER:")
+                            final_answer = content[answer_start:].strip()
+                            return final_answer
+                    
+                    # If no FINAL ANSWER found, return the whole content
                     return content
-        
-        return "No answer generated."
+                    
+            return "No valid response generated."
+            
+        except Exception as e:
+            print(f"Agent execution error: {e}")
+            return f"Unable to process the question due to an error. Please try again."
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):

tools.py

@@ -316,7 +316,7 @@ def arxiv_search_tool(arxiv_query: str) -> str:
 
 from langchain_openai import ChatOpenAI
 from langchain.schema import SystemMessage, HumanMessage
-LLM = ChatOpenAI(model_name="gpt-4.1-mini", temperature=0.2)
+LLM = ChatOpenAI(model_name="gpt-4o-mini", temperature=0.2)
 
 @tool
 def analyze_code_tool(task_id: str) -> str: