| from typing import List, Optional |
| import requests |
| import json |
| from mcp.server.fastmcp import FastMCP |
|
|
| mcp = FastMCP("organizedprogrammers-mcp-server") |
|
|
| @mcp.tool() |
| def search_arxiv_papers(keyword: str, limit: int = 5) -> str: |
| """ |
| Search papers from arXiv database with specified keywords [optional: a limit of papers the user wants] |
| Args: keyword: string, [optional: limit: integer, set limit to 5 if not specified] |
| """ |
| response = requests.post("https://om4r932-arxiv.hf.space/search", headers={ |
| "Content-Type": "application/json" |
| }, data=json.dumps({ |
| "keyword": keyword, |
| "limit": limit |
| }), verify=False) |
|
|
| if response.status_code != 200: |
| return "Unable to find papers: error on post()" |
| |
| responseJson = response.json() |
|
|
| if responseJson.get("error") or not isinstance(responseJson['message'], dict): |
| return f"Unable to find papers: error on API -> {responseJson['message']}" |
|
|
| if len(responseJson["message"].keys()) == 0: |
| return "No papers has been found" |
| |
| return "\n".join([f"arXiv n°{paper_id} - {paper_meta['title']} by {paper_meta['authors']} : {paper_meta['abstract']}" for paper_id, paper_meta in responseJson['message'].items()]) |
|
|
| @mcp.tool() |
| def locate_3gpp_document(doc_id: str) -> str: |
| """ |
| Find 3GPP document location with the document's ID |
| Args: doc_id: string |
| """ |
| response = requests.post("https://organizedprogrammers-3gppdocfinder.hf.space/find", headers={ |
| "Content-Type": "application/json" |
| }, data=json.dumps({ |
| "doc_id": doc_id |
| }), verify=False) |
|
|
| if response.status_code != 200: |
| return f"Unable to find document: {response.status_code} - {response.content}" |
| |
| responseJson = response.json() |
|
|
| if responseJson.get("detail"): |
| return responseJson['detail'] |
| |
| return f"Document ID {responseJson['doc_id']} version {responseJson['version']} is downloadable via this link: {responseJson['url']}.\n{responseJson['scope']}" |
|
|
| @mcp.tool() |
| def locate_multiple_3gpp_documents(doc_ids: List[str]) -> str: |
| """ |
| Find 3GPP document location with the document's ID |
| Args: doc_id: string |
| """ |
| response = requests.post("https://organizedprogrammers-3gppdocfinder.hf.space/batch", headers={ |
| "Content-Type": "application/json" |
| }, data=json.dumps({ |
| "doc_ids": doc_ids |
| }), verify=False) |
|
|
| if response.status_code != 200: |
| return f"Unable to find document: {response.status_code} - {response.content}" |
| |
| responseJson = response.json() |
|
|
| if responseJson.get("detail"): |
| return responseJson['detail'] |
| |
| return "\n".join([f"The document {doc_id} is downloadable via this link: {url}" for doc_id, url in responseJson['results']] + [f"We can't find document {doc_id}" for doc_id in responseJson['missing']]) |
|
|
| @mcp.tool() |
| def locate_etsi_document(doc_id: str) -> str: |
| """ |
| Find ETSI document location with the document's ID (starts with SET or SCP) |
| Args: doc_id: string |
| """ |
| response = requests.post("https://organizedprogrammers-etsidocfinder.hf.space/find", headers={ |
| "Content-Type": "application/json" |
| }, data=json.dumps({ |
| "doc_id": doc_id |
| }), verify=False) |
|
|
| if response.status_code != 200: |
| return f"Unable to find document: {response.status_code} - {response.content}" |
| |
| responseJson = response.json() |
|
|
| if responseJson.get("detail"): |
| return responseJson['detail'] |
| |
| return f"Document ID {responseJson['doc_id']} is downloadable via this link: {responseJson['url']}" |
|
|
| @mcp.tool() |
| def search_3gpp_specifications(keywords: str, threshold: int, release: Optional[str] = "", working_group: Optional[str] = "", spec_type: Optional[str] = "") -> str: |
| """ |
| Search 3GPP specifications with specified keywords and filters using BM25 |
| Args: keywords: string, threshold: integer 0-100 [default 60], release: optional filter, string [only the number Rel-19 -> '19'], working_group: optional filter, string [options: C1,C2,...,C6,CP or S1,S2,...,S6,SP], spec_type: optional filter, string [either TS (Technical Specification) or TR (Technical Report)] |
| For each non-used optional filters, leave a empty string |
| """ |
| body = {"keywords": keywords, "threshold": threshold} |
| if release: |
| body['release'] = release |
| if working_group: |
| body['working_group'] = working_group |
| if spec_type: |
| body['spec_type'] = spec_type |
|
|
| response = requests.post("https://organizedprogrammers-3gppdocfinder.hf.space/search-spec/experimental", headers={ |
| "Content-Type": "application/json" |
| }, data=json.dumps(body), verify=False) |
|
|
| if response.status_code != 200: |
| return f"Unable to find document: {response.status_code} - {response.content}" |
| |
| responseJson = response.json() |
|
|
| if responseJson.get("detail"): |
| return responseJson['detail'] |
| |
| return "\n--\n".join([f"3GPP {spec['type']} {spec['id']} version {spec['version']} - {spec['title']} is downloadable via this link: {spec['url']}\n{spec['scope']}" for spec in responseJson['results']]) |
|
|
| @mcp.tool() |
| def ask_questions_to_3gpp_database(question: str, threshold: int = 65, release: Optional[str] = "", working_group: Optional[str] = "", spec_type: Optional[str] = "") -> str: |
| """ |
| Retrieve technical documents sections to help AI answer the user's technical question, if same topic already called, re-use the downloaded documents |
| 3GPP specifications are used as source documents, using BM25 to filter the documents |
| Args: question: string, threshold: integer 0-100 [default 60], release: optional filter, string [only the number Rel-19 -> '19'], working_group: optional filter, string [options: C1,C2,...,C6,CP or S1,S2,...,S6,SP], spec_type: optional filter, string [either TS (Technical Specification) or TR (Technical Report)] |
| For each non-used optional filters, leave a empty string |
| After extracting the documents, answer to the question with a complete and detailed paragraph with the sources cited |
| """ |
| body = {"question": question, "threshold": threshold} |
| if release: |
| body['release'] = release |
| if working_group: |
| body['working_group'] = working_group |
| if spec_type: |
| body['spec_type'] = spec_type |
|
|
| response = requests.post("https://organizedprogrammers-3gppdocfinder.hf.space/list-rag-docs", headers={ |
| "Content-Type": "application/json" |
| }, data=json.dumps(body), verify=False) |
|
|
| if response.status_code != 200: |
| return f"Unable to extract documents: {response.status_code}" |
|
|
| docs = response.json()['output'] |
| return docs |
|
|
| if __name__ == "__main__": |
| mcp.run(transport="stdio") |
