Update app.py

app.py

@@ -39,16 +39,9 @@ def add(a: int, b: int) -> int:
     return a + b
 
 @tool
-def arvix_search(query: str) -> str:
-    """Search Arxiv for a query and return maximum 3 results."""
-    try:
-        search_docs = ArxivLoader(query=query, load_max_docs=3).load()
-        formatted_search_docs = "\n\n---\n\n".join(
-            [f'<Document source="{doc.metadata["source"]}"/>\n{doc.page_content[:1000]}\n</Document>' 
-             for doc in search_docs])
-        return {"arvix_results": formatted_search_docs}
-    except Exception as e:
-        return {"arvix_results": f"Error: {str(e)}"}
+def modulus(a: int, b: int) -> int:
+    """Get the modulus of two numbers."""
+    return a % b
 
 @tool
 def subtract(a: int, b: int) -> int:
@@ -62,6 +55,18 @@ def divide(a: int, b: int) -> int:
         raise ValueError("Cannot divide by zero.")
     return a / b
 
+@tool
+def arvix_search(query: str) -> str:
+    """Search Arxiv for a query and return maximum 3 results."""
+    try:
+        search_docs = ArxivLoader(query=query, load_max_docs=3).load()
+        formatted_search_docs = "\n\n---\n\n".join(
+            [f'<Document source="{doc.metadata["source"]}"/>\n{doc.page_content[:1000]}\n</Document>' 
+             for doc in search_docs])
+        return {"arvix_results": formatted_search_docs}
+    except Exception as e:
+        return {"arvix_results": f"Error: {str(e)}"}
+
 @tool
 def wiki_search(query: str) -> str:
     """Search Wikipedia for a query and return maximum 2 results."""