Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
|
@@ -94,10 +94,10 @@ mcp = FastMCP("OnBase", port=port)
|
|
| 94 |
@mcp.resource(
|
| 95 |
uri="info://available_retrievers",
|
| 96 |
name="AvailableRetrievers",
|
| 97 |
-
description="Lista
|
| 98 |
mime_type="application/json"
|
| 99 |
)
|
| 100 |
-
def
|
| 101 |
available = []
|
| 102 |
|
| 103 |
for source, indexes in retrievers_metadata.items():
|
|
@@ -117,6 +117,34 @@ def get_available_retrievers() -> dict:
|
|
| 117 |
"count": len(available)
|
| 118 |
}
|
| 119 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 120 |
# Funci贸n principal de b煤squeda
|
| 121 |
@mcp.tool()
|
| 122 |
def retrieve_docs(
|
|
|
|
| 94 |
@mcp.resource(
|
| 95 |
uri="info://available_retrievers",
|
| 96 |
name="AvailableRetrievers",
|
| 97 |
+
description="Lista de retrievers con metadatos (para uso interno)",
|
| 98 |
mime_type="application/json"
|
| 99 |
)
|
| 100 |
+
def _get_available_retrievers_internal() -> dict:
|
| 101 |
available = []
|
| 102 |
|
| 103 |
for source, indexes in retrievers_metadata.items():
|
|
|
|
| 117 |
"count": len(available)
|
| 118 |
}
|
| 119 |
|
| 120 |
+
|
| 121 |
+
|
| 122 |
+
|
| 123 |
+
@mcp.tool()
|
| 124 |
+
async def list_retrievers(ctx: Context) -> dict:
|
| 125 |
+
"""
|
| 126 |
+
Devuelve la lista de retrievers disponibles incluyendo nombre, t铆tulo y descripci贸n.
|
| 127 |
+
|
| 128 |
+
Returns:
|
| 129 |
+
dict: {
|
| 130 |
+
"retrievers": [{
|
| 131 |
+
"name": "vec_who_1",
|
| 132 |
+
"title": "WHO Technical Report.pdf",
|
| 133 |
+
"description": "..."
|
| 134 |
+
}],
|
| 135 |
+
"count": int
|
| 136 |
+
}
|
| 137 |
+
"""
|
| 138 |
+
# Accede al resource internamente
|
| 139 |
+
resource = await ctx.read_resource("info://available_retrievers")
|
| 140 |
+
return {
|
| 141 |
+
"retrievers": resource["retrievers"],
|
| 142 |
+
"count": resource["count"]
|
| 143 |
+
}
|
| 144 |
+
|
| 145 |
+
|
| 146 |
+
|
| 147 |
+
|
| 148 |
# Funci贸n principal de b煤squeda
|
| 149 |
@mcp.tool()
|
| 150 |
def retrieve_docs(
|