Update MCP server

server.py

@@ -13,8 +13,7 @@ DOC3GPPFINDER_BASE = "https://organizedprogrammers-3gppdocfinder.hf.space"
 
 # Request function
 
-async def post_data_to_api(url, **kwargs):
-    data = dict(kwargs)
+async def post_data_to_api(url, data = None):
     if data is None or data == {}:
         return (None, "")
     headers = {"Accept": "application/json"}
@@ -28,8 +27,7 @@ async def post_data_to_api(url, **kwargs):
             traceback.print_exception(e)
             return (None, e)
         
-async def fake_post_data_to_api(url, **kwargs):
-    params = dict(kwargs)
+async def fake_post_data_to_api(url, params = None):
     if params is None or params == {}:
         return (None, "")
     headers = {"Accept": "application/json"}
@@ -66,7 +64,7 @@ async def get_arxiv_publications(keywords: str, limit: int):
     Arguments available: keywords: string [mandatory], limit: integer [mandatory, default = 5]
     """
     endpoint = ARXIV_BASE + "/search"
-    data = await post_data_to_api(endpoint, keyword=keywords, limit=limit)
+    data = await post_data_to_api(endpoint, {"keyword": keywords, "limit": limit})
     if isinstance(data, tuple) and data[0] is None:
         return f"An error has occured while getting publications: {data[1]}"
     if data["error"]:
@@ -90,29 +88,38 @@ async def get_document_url(doc_id: str, release: int = None):
     Arguments available: doc_id: string [mandatory], release: integer [optional for every case]
     """
     endpoint = DOC3GPPFINDER_BASE + "/find"
-    data = await post_data_to_api(endpoint, doc_id=doc_id, release=release)
+    data = await post_data_to_api(endpoint, {"doc_id": doc_id, "release": release})
     if isinstance(data, tuple) and data[0] is None:
         return f"An error while searching publications: {data[1]}"
-    output = f'Document ID: {doc_id}\nURL: {data.get("url", "Not found !")}'
-    output += f'\nScope: {data["scope"]}' if data.get("scope", None) is not None else ""
+    output = f'The document {doc_id} is available via this URL : {data.get("url", "No URL found !")}. '
+    output += f'\nThe scope of the document: {data["scope"]}' if data.get("scope", None) is not None or data.get("scope", None) == "" else ""
     return output
     
 @mcp.tool()
-async def search_specs(keywords: str, limit: int):
+async def search_specs(keywords: str, limit: int, release: str = None, wg: str = None, spec_type: str = None, mode: str = "and"):
     """
     Search 3GPP specifications only by their keywords [note that it will only work with keywords](and some filters [see kwargs field])
-    Arguments available: keywords: string [mandatory, separated by space], limit [mandatory, default = 5]
-    Kwargs available [optional]: (release: integer as string or 'Rel-xxx', wg: string = working group (S1, C4, SP, ...), spec_type: string (either TS or TR), mode: string (either 'and' or 'or') = search mode)
+    Arguments available: keywords: string [mandatory, separated by space], limit [mandatory, default = 5], release: string [optional, filter] (Rel-x where x is the release version, generally the first number of the full version), wg: string [optional, filter] = working group (S1, C4, SP, ...), spec_type: string [optional, filter] (either TS or TR), mode [mandatory, default = 'and'] = search mode (and = all keywords must be in the search, or = at least one keyword in the search)
     """
     endpoint = DOC3GPPFINDER_BASE + "/search-spec"
-    data = await post_data_to_api(endpoint, keywords=keywords)
+    data = await post_data_to_api(endpoint, {
+        "keywords": keywords,
+        "release": release,
+        "wg": wg,
+        "spec_type": spec_type,
+        "mode": mode
+    })
     if isinstance(data, tuple) and data[0] is None:
         return f"An error has occured while searching specifications"
     results = data['results'][:min(len(data['results'])-1, limit)]
     output = []
     for spec in results:
-        output.append(f"Specification ID: {spec['id']}\nTitle: {spec['title']}\nType: {spec['type']}\nRelease: {spec['release']}\nVersion: {spec['version']}\nWorking Group: {spec['working_group']}\nURL of spec: {spec['url']}\n")
-    
+        x = f"Found specification number {spec['id']} version {spec['release']}"
+        if spec['scope'] != "":
+            x += f" where {spec['scope'].lower()}\n"
+        else:
+            x += "\n"
+        output.append(x)
     return "-\n".join(output)
 
 @mcp.tool()
@@ -128,7 +135,7 @@ async def get_multiple_documents_url(doc_ids: List[str], release: int = None):
     results = data["results"]
     output = []
     for doc_id, url in results.items():
-        output.append(f'Document ID: {doc_id}\nURL: {url}\n')
+        output.append(f'The document {doc_id} is available via this URL: {url}\n')
     return "-\n".join(output)
 
 # PTT Endpoints