geronimo-pericoli commited on
Commit
2c2b401
verified
1 Parent(s): 892052e

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +30 -2
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 completa de retrievers disponibles",
98
  mime_type="application/json"
99
  )
100
- def get_available_retrievers() -> dict:
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(