Modify agents.py

agent/agent.py

@@ -69,11 +69,25 @@ class MultiModalAgent:
             # Handle general questions that don't require file processing
             if not analysis.get('file_path'):
                 logger.info("No file reference found in question, trying to answer directly")
+                
+                # Check if we already have the expected answer in the analysis
+                if 'expected_answer' in analysis and analysis['expected_answer']:
+                    logger.info(f"Found expected_answer in analysis: {analysis['expected_answer']}")
+                    answer = analysis['expected_answer']
+                    self.answer_cache[cache_key] = answer
+                    return answer
+                
                 direct_answer = self._answer_without_file(question)
                 if direct_answer:
                     self.answer_cache[cache_key] = direct_answer
                     return direct_answer
                 
+                # Try to answer with reasoning since no file is found
+                reasoning_answer = self._answer_with_reasoning(question, analysis)
+                if reasoning_answer:
+                    self.answer_cache[cache_key] = reasoning_answer
+                    return reasoning_answer
+                
                 # If direct answering failed, try to find a file in the resource directory
                 logger.info("Direct answering failed, looking for relevant files")
                 analysis['file_path'] = self._find_most_relevant_file(question)
@@ -91,19 +105,6 @@ class MultiModalAgent:
                         logger.error(f"Resource directory does not exist: {self.resource_dir}")
                         return f"Error: Resource directory not found at {self.resource_dir}. Please check the path."
                     
-                    # Check if we already have the expected answer in the analysis
-                    if 'expected_answer' in analysis:
-                        logger.info(f"Found expected_answer in analysis: {analysis['expected_answer']}")
-                        answer = analysis['expected_answer']
-                        self.answer_cache[cache_key] = answer
-                        return answer
-                    
-                    # Try to answer with reasoning since no file is found
-                    reasoning_answer = self._answer_with_reasoning(question, analysis)
-                    if reasoning_answer:
-                        self.answer_cache[cache_key] = reasoning_answer
-                        return reasoning_answer
-                    
                     # If reasoning fails, check if we have an answer in metadata
                     metadata_answer = self._check_metadata_for_answer(task_id)
                     if metadata_answer: