Update base_agent.py

base_agent.py

@@ -27,6 +27,28 @@ class BaseAgent:
     def __call__(self, question: str, file_name: str = None, **kwargs) -> str:
         """Process the question and return an answer."""
         raise NotImplementedError("Subclasses must implement this method")
+
+    def extract_search_query(self, question: str) -> str:
+        """
+        Extract the key search terms from the question.
+        
+        Args:   
+            question: The original question
+            
+        Returns:
+            A reformulated search query
+        """
+        system_prompt = """
+        You are an expert at extracting search terms from questions. 
+        Given a question, identify the key terms that would be most effective for a web search.
+        Remove any unnecessary words and focus on specific names, dates, or key concepts.
+        Return only the search terms, with no additional text.
+        """
+        
+        user_prompt = f"Extract the most effective search terms from this question: {question}"
+        
+        response = self.llm_call(system_prompt, user_prompt)
+        return response.strip()
     
     def llm_call(self, system_prompt: str, user_prompt: str, 
                  temperature: float = 0.1, model: Optional[str] = None) -> str: