New tests - search tool parsing

manage_agents.py

@@ -1406,7 +1406,7 @@ class MainAgent:
             response = self.llm.invoke(answer_prompt)
             answer = response.content if hasattr(response, 'content') else str(response)
             # Remove any prefix like "Final Answer:" or "Answer:" and strip whitespace
-            answer = re.sub(r'^(Final Answer:|Answer:)', '', answer, flags=re.IGNORECASE).strip()
+            answer = re.sub(r'^(Final Answer:|Answer:|NASA award number:)', '', answer, flags=re.IGNORECASE).strip()
             state.messages.append(AIMessage(content=answer))
             return state
         builder.add_node("synthesize_answer", _synthesize_answer)

tools.py

@@ -95,7 +95,6 @@ class WikipediaTool(BaseTool):
     wikipedia_tool: WikipediaQueryRun = Field(default_factory=lambda: WikipediaQueryRun(
         api_wrapper=WikipediaAPIWrapper(top_k_results=5)
     ))
-    print("WikipediaTool initialized")
     def _run(self, question: str) -> str:
         """Search Wikipedia and return the result as a string"""
         try:
@@ -611,8 +610,6 @@ class ExcelTool(BaseContentTool):
         """Convert DataFrame to a readable text format optimized for LLM analysis."""
         # Use to_string for a clean, tabular format
         table_str = df.to_string(index=False)
-        print("table_str")
-        print(table_str)
         return f"Table:\n{table_str}"
 
     def _run(self, task_id: str, question: str = "") -> List[Document]:
@@ -694,7 +691,6 @@ class WebSearchTool(BaseTool):
     #search_tool: DuckDuckGoSearchResults = Field(default_factory=DuckDuckGoSearchResults)
     search_tool: TavilySearchResults = Field(default_factory=TavilySearchResults)
 
-    print("WebSearchTool initialized")
     def _extract_links_from_results(self, search_result) -> list:
         """Extract links from search results, robust to type."""
         links = []
@@ -783,8 +779,6 @@ class WebSearchTool(BaseTool):
     def _run(self, query: str) -> str:
         try:
             search_result = self.search_tool.run(query, max_results=5)
-            print("query")
-            print(query)
             links = self._extract_links_from_results(search_result)
             seen = set()
             results = []