Add tools

app.py

@@ -13,7 +13,8 @@ async def make_request(url: str, data: Dict[str, Any]):
             return response.json()
         except:
             return None
-        
+
+# arXiv
 @server.tool()
 async def search_academic_papers_arxiv(keyword: str, limit: int = 5) -> str:
     """
@@ -37,6 +38,7 @@ async def get_arxiv_pub_text(arxiv_id: str) -> str:
         return "Unable to extract PDF | arXiv PDF not found"
     return response["message"]["text"]
 
+# DocFinder
 @server.tool()
 async def get_document_url(doc_id: str) -> str:
     """
@@ -68,8 +70,9 @@ async def search_specifications_with_keywords(keywords: str, threshold: int = 60
     results = response["results"]
     return "\n---\n".join([f"Specification ID: {spec['id']}\nTitle: {spec['title']}\nType: {'Technical Specification' if spec['spec_type'] == 'TS' else 'Technical Report'}\nVersion: {spec.get('version', 'unavailable')}\nScope: {spec.get('scope', 'unavailable')}\nWorking Group: {spec.get('working_group', 'not defined')}\nURL: {spec.get('url', 'unavailable')}" for spec in results])
 
+# SpecSplitter
 @server.tool()
-async def get_spec_text(spec_id: str) -> Union[Dict[str, str], str]:
+async def get_spec_text(spec_id: str) -> str:
     """
     Extract specification from 3GPP or ETSI
     Returns a dictionary k:v where k is the section (1., 2.2.1, ...) and v, the content of k, or a string if failed
@@ -78,6 +81,20 @@ async def get_spec_text(spec_id: str) -> Union[Dict[str, str], str]:
     response = await make_request('https://organizedprogrammers-specsplitter.hf.space/extract_text/structured', {"spec_id": spec_id})
     if not response:
         return "Unable to extract specification text"
-    return response
+    return "\n".join([f"{k}: {v}" for k, v in response.keys()])
+
+# SERPent
 
+@server.tool()
+async def search_google_patents(queries: List[str], n_results: int) -> str:
+    """
+    Search patents from Google Patents
+    You can generate multiple queries (at least 1)
+    Returns a list of patents from queries, for each query, {n_results} patents will be retrieved
+    Args: queries -> list of string, n_results -> integer [by default: 10]
+    """
+    response = await make_request("https://organizedprogrammers-serpent.hf.space/serp/search_patents", {"queries": queries, "n_results": n_results})
+    if not response:
+        return "Unable to fetch patents"
+    return "\n".join(f"[Patent ID: {patent['id']} | Title: {patent['title']} | Body: {patent['body']}]" for patent in response.results)
 app = server.streamable_http_app