Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
|
@@ -87,16 +87,16 @@ for source, rel_path in SOURCES.items():
|
|
| 87 |
|
| 88 |
# Recurso dinámico para búsqueda
|
| 89 |
@mcp.resource(
|
| 90 |
-
uri="retriever://documentos
|
| 91 |
name="DocumentRetriever",
|
| 92 |
description="Retrieve documents from different regulations using semantic search.",
|
| 93 |
mime_type="application/json"
|
| 94 |
)
|
| 95 |
-
def retrieve_docs(query: str, fuente: str = 'oms',
|
| 96 |
"""
|
| 97 |
Parameters:
|
| 98 |
query: Search query (required).
|
| 99 |
-
fuente: Source name (default: oms).
|
| 100 |
top_k: Number of results to return (default: 3).
|
| 101 |
"""
|
| 102 |
if fuente not in indices:
|
|
|
|
| 87 |
|
| 88 |
# Recurso dinámico para búsqueda
|
| 89 |
@mcp.resource(
|
| 90 |
+
uri="retriever://documentos",
|
| 91 |
name="DocumentRetriever",
|
| 92 |
description="Retrieve documents from different regulations using semantic search.",
|
| 93 |
mime_type="application/json"
|
| 94 |
)
|
| 95 |
+
def retrieve_docs(query: str, fuente: str = 'oms', top_k: int = 3) -> dict:
|
| 96 |
"""
|
| 97 |
Parameters:
|
| 98 |
query: Search query (required).
|
| 99 |
+
fuente: Source name (default: 'oms').
|
| 100 |
top_k: Number of results to return (default: 3).
|
| 101 |
"""
|
| 102 |
if fuente not in indices:
|