Fimeg/letta-code
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: Fix linkToolsToAgent to use parallel API calls for tool lookup (#443)

Browse Source 
Co-authored-by: Letta <noreply@letta.com>
This commit is contained in:
Charles Packer
2025-12-31 19:44:38 -08:00
committed by GitHub
parent 725a2a7cad
commit 4bf102fb27
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/agent/modify.ts
View File

@@ -202,17 +202,16 @@ export async function linkToolsToAgent(agentId: string): Promise<LinkResult> {
};
}
// Look up tool IDs from global tool list
// Use server names when querying, since that's how tools are registered on the server
const toolsToAddIds: string[] = [];
for (const toolName of toolsToAdd) {
const serverName = getServerToolName(toolName);
const toolsResponse = await client.tools.list({ name: serverName });
const tool = toolsResponse.items[0];
if (tool?.id) {
toolsToAddIds.push(tool.id);
}
}
// Look up tool IDs in parallel (instead of sequential calls)
const toolsToAddIds = (
await Promise.all(
toolsToAdd.map(async (toolName) => {
const serverName = getServerToolName(toolName);
const toolsResponse = await client.tools.list({ name: serverName });
return toolsResponse.items[0]?.id;
}),
)
).filter((id): id is string => !!id);
// Combine current tools with new tools
const newToolIds = [...currentToolIds, ...toolsToAddIds];