s

tools.py

@@ -310,22 +310,47 @@ web_search_tool = Tool(
 )
 
 def arvix_search(query: str) -> str:
-    """Search Arxiv for a query and return maximum 3 result.
+    """Search Arxiv for a query and return maximum 3 results.
     
     Args:
-        query: The search query."""
-    search_docs = ArxivLoader(query=query, load_max_docs=3).load()
-    formatted_search_docs = "\n\n---\n\n".join(
-        [
-            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
-            for doc in search_docs
-        ])
-    return {"arvix_results": formatted_search_docs}
-
+        query: The search query.
+    
+    Returns:
+        Formatted string containing search results from Arxiv papers.
+    """
+    try:
+        # Load documents from Arxiv
+        search_docs = ArxivLoader(query=query, load_max_docs=3).load()
+        
+        # Check if we got any results
+        if not search_docs:
+            return "No results found for the query: " + query
+        
+        # Process each document with safe metadata access
+        formatted_docs = []
+        for doc in search_docs:
+            # Use .get() method for safe dictionary access with fallback values
+            source = doc.metadata.get("source", "Unknown source")
+            page = doc.metadata.get("page", "N/A")
+            
+            # Create the formatted document string
+            formatted_doc = f'<Document source="{source}" page="{page}"/>\n{doc.page_content[:1000]}\n</Document>'
+            formatted_docs.append(formatted_doc)
+        
+        # Join all documents with separators
+        formatted_search_docs = "\n\n---\n\n".join(formatted_docs)
+        
+        # Return the string directly (not wrapped in a dictionary)
+        return formatted_search_docs
+        
+    except Exception as e:
+        # Handle any unexpected errors gracefully
+        return f"Error searching Arxiv: {str(e)}"
+        
 arvix_search_tool = Tool(
     name="arvix_search",
     func=arvix_search,
-    description="Searches for and retrieves information online using an inputted query on the website arvix, that is a database storing many factual scientific papers."
+    description="Searches for and retrieves information from Arxiv, a database storing scientific papers. Returns formatted text containing paper abstracts and metadata."
 )
 
 def extract_image_text(img_path: str) -> str: