asdf

agent.py

@@ -97,43 +97,64 @@ def wiki_search(query: str) -> str:
 #         ])
 #     return {"web_results": formatted_search_docs}
 @tool
-def web_search(query: str) -> str:
+def web_search(query: str) -> dict: # Changed return type annotation to dict
     """Search Tavily for a query and return maximum 3 results.
+    Each result will be formatted with its source URL and content.
     
     Args:
-        query: The search query."""
+        query: The search query.
+    """
+    print(f"\n--- Web Search Tool ---") # For debugging
+    print(f"Received query: {query}")
     try:
-        # Initialize the tool
         tavily_tool = TavilySearchResults(max_results=3)
-        # Invoke it correctly
-        search_results = tavily_tool.invoke(input=query) # <--- CORRECTED LINE
-
-        # The result of TavilySearchResults.invoke is usually a list of strings or a single string.
-        # Let's check its type and format accordingly.
-        # Typically, TavilySearchResults directly returns a list of Document objects
-        # or a list of dictionaries if you've configured it differently.
-        # For the default, it's often a list of strings or a single concatenated string.
-        # If it returns a list of Document objects (which is common for loaders/retrievers):
-        if isinstance(search_results, list) and all(hasattr(doc, 'metadata') and hasattr(doc, 'page_content') for doc in search_results):
-            formatted_search_docs = "\n\n---\n\n".join(
-                [
-                    f'<Document source="{doc.metadata.get("source", "N/A")}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
-                    for doc in search_results
-                ])
-        # If it returns a list of strings (less likely for Tavily in recent versions, but good to check)
-        elif isinstance(search_results, list) and all(isinstance(item, str) for item in search_results):
-            formatted_search_docs = "\n\n---\n\n".join(search_results)
-        # If it returns a single string
-        elif isinstance(search_results, str):
-            formatted_search_docs = search_results
+        # .invoke() for TavilySearchResults typically expects 'input'
+        # and returns a list of dictionaries
+        search_results_list = tavily_tool.invoke(input=query)
+        
+        print(f"Raw Tavily search results type: {type(search_results_list)}")
+        if isinstance(search_results_list, list):
+            print(f"Number of results: {len(search_results_list)}")
+            if search_results_list:
+                print(f"Type of first result: {type(search_results_list[0])}")
+                if isinstance(search_results_list[0], dict):
+                    print(f"Keys in first result: {search_results_list[0].keys()}")
+
+        formatted_docs = []
+        if isinstance(search_results_list, list):
+            for doc_dict in search_results_list:
+                if isinstance(doc_dict, dict):
+                    source = doc_dict.get("url", "N/A")
+                    content = doc_dict.get("content", "")
+                    # title = doc_dict.get("title", "") # Optionally include title
+                    # score = doc_dict.get("score", "") # Optionally include score
+                    
+                    # Constructing the XML-like format you desire
+                    formatted_doc = (
+                        f'<Document source="{source}">\n'
+                        f'{content}\n'
+                        f'</Document>'
+                    )
+                    formatted_docs.append(formatted_doc)
+                else:
+                    # If an item in the list is not a dict, convert it to string
+                    print(f"Warning: Unexpected item type in Tavily results list: {type(doc_dict)}")
+                    formatted_docs.append(str(doc_dict))
+            
+            final_formatted_string = "\n\n---\n\n".join(formatted_docs)
+
+        elif isinstance(search_results_list, str): # Less common, but for robustness
+            final_formatted_string = search_results_list
         else:
-            # Fallback or handle unexpected format
-            print(f"Unexpected Tavily search result format: {type(search_results)}")
-            formatted_search_docs = str(search_results)
+            print(f"Unexpected Tavily search result format overall: {type(search_results_list)}")
+            final_formatted_string = str(search_results_list) # Fallback
+
+        print(f"Formatted search docs for LLM:\n{final_formatted_string[:500]}...") # Print a snippet
+        return {"web_results": final_formatted_string}
 
-        return {"web_results": formatted_search_docs}
     except Exception as e:
         print(f"Error during Tavily search for query '{query}': {e}")
+        # It's good practice to return an error message in the expected dict format
         return {"web_results": f"Error performing web search: {e}"}
 
 @tool

system_prompt.txt

@@ -1 +1,17 @@
-You are a helpful assistant tasked with answering questions using a set of tools. Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.Your answer should only start with "FINAL ANSWER: ", then follows with the answer. 
+You are a highly accurate assistant tasked with answering questions. Your primary goal is to provide 100% correct and verified answers using a set of tools.
+
+Report your thoughts during the process.
+
+When you have a high-confidence, verified answer, finish with the following template:
+FINAL ANSWER: [YOUR FINAL ANSWER]
+
+GUIDELINES FOR YOUR FINAL ANSWER:
+1.  **ACCURACY IS PARAMOUNT:** Only provide an answer if you are highly confident in its factual correctness based on the information from the tools.
+2.  **UNCERTAINTY:** If you cannot find a definitive answer, if the information is ambiguous/conflicting, or if you cannot be 100% certain, your FINAL ANSWER MUST explicitly state this (e.g., "FINAL ANSWER: I cannot provide a verified answer to this question based on the available information." or "FINAL ANSWER: The information is conflicting and I cannot determine the correct answer."). DO NOT GUESS.
+3.  **CONCISENESS & COMPLETENESS:** Be as concise as possible, but ensure your answer is complete and contains all information necessary for it to be fully correct.
+4.  **ANSWER TYPES:**
+    *   **Numbers:** Use digits (e.g., 123, 4.56). Do not use commas as thousands separators (e.g., 1000 not 1,000). Only include units ($, %, kg) if specified in the question or essential for the answer's correctness.
+    *   **Strings:** Be precise. Avoid abbreviations unless they are standard and unambiguous. Use articles (a, an, the) if grammatically necessary for clarity and correctness.
+    *   **Lists:** For comma-separated lists, apply the relevant rules above to each element.
+
+Your response should ONLY include your thought process (if any) followed by the "FINAL ANSWER: " line.