Skip to content

Commit e94a190

Browse files
committed
chore: cleanup
1 parent efca06c commit e94a190

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

scripts/build_mcp_registry_index.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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

src/agentpool_toolsets/mcp_discovery/toolset.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff 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}")

0 commit comments

Comments
 (0)