File tree Expand file tree Collapse file tree
src/agentpool_toolsets/mcp_discovery Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -190,9 +190,9 @@ async def main(output_path: Path) -> None:
190190 test_table = db .create_table ("servers" , table )
191191 query_vec = next (iter (model .embed (["github repository issues" ]))).tolist ()
192192 results = test_table .search (query_vec ).limit (3 ).to_arrow ()
193- for i in range ( len ( results ) ):
194- name = results ["name" ][ i ]. as_py ()
195- desc = results ["description" ][ i ]. as_py () [:60 ]
193+ for row in results . to_pylist ( ):
194+ name = row ["name" ]
195+ desc = row ["description" ][:60 ]
196196 print (f" - { name } : { desc } ..." )
197197
198198
Original file line number Diff line number Diff line change @@ -279,17 +279,17 @@ async def search_mcp_servers( # noqa: D417
279279 # Format results, filtering by allowed/blocked
280280 lines = [f"Found MCP servers matching '{ query } ':\n " ]
281281 count = 0
282- for i in range ( len ( results ) ):
283- name = results ["name" ][ i ]. as_py ()
282+ for row in results . to_pylist ( ):
283+ name = row ["name" ]
284284
285285 # Filter by allowed/blocked
286286 if not self ._is_server_allowed (name ):
287287 continue
288288
289- desc = results ["description" ][ i ]. as_py ()
290- version = results ["version" ][ i ]. as_py ()
291- has_remote = results ["has_remote" ][ i ]. as_py ()
292- remote_types = results ["remote_types" ][ i ]. as_py ()
289+ desc = row ["description" ]
290+ version = row ["version" ]
291+ has_remote = row ["has_remote" ]
292+ remote_types = row ["remote_types" ]
293293
294294 lines .append (f"**{ name } ** (v{ version } )" )
295295 lines .append (f" { desc } " )
You can’t perform that action at this time.
0 commit comments