Add tool for RAG questions

server.py

@@ -126,5 +126,32 @@ def search_3gpp_specifications(keywords: str, threshold: int, release: Optional[
     
     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")