File size: 5,191 Bytes
27054c5 10b8ff0 27054c5 10b8ff0 27054c5 10b8ff0 27054c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 |
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']])
if __name__ == "__main__":
mcp.run(transport="stdio") |