Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-01-09 12:08:50 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: streamline tool classification logic for local and programmatic tools

Browse source
This commit is contained in:
Danny Avila 2026-01-07 19:00:25 -05:00
parent 6f70aac542
commit fff9cecad2
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
1 changed files with 25 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

51
packages/api/src/tools/classification.ts
View file

@ -376,36 +376,35 @@ export async function buildToolClassification(
return { toolRegistry, additionalTools };
}
try {
const authValues = await loadAuthValues({
userId,
authFields: [EnvVar.CODE_API_KEY],
/** Tool search uses local mode (no API key needed) */
if (hasDeferredTools) {
const toolSearchTool = createToolSearch({
mode: 'local',
toolRegistry,
});
const codeApiKey = authValues[EnvVar.CODE_API_KEY];
additionalTools.push(toolSearchTool);
logger.debug(`[buildToolClassification] Tool Search enabled for agent ${agentId}`);
}
if (!codeApiKey) {
logger.warn(
'[buildToolClassification] PTC/ToolSearch configured but CODE_API_KEY not available',
);
return { toolRegistry, additionalTools };
}
if (hasProgrammaticTools) {
const ptcTool = createProgrammaticToolCallingTool({ apiKey: codeApiKey });
additionalTools.push(ptcTool);
logger.debug(`[buildToolClassification] PTC tool enabled for agent ${agentId}`);
}
if (hasDeferredTools) {
const toolSearchTool = createToolSearch({
apiKey: codeApiKey,
toolRegistry,
/** PTC requires CODE_API_KEY for sandbox execution */
if (hasProgrammaticTools) {
try {
const authValues = await loadAuthValues({
userId,
authFields: [EnvVar.CODE_API_KEY],
});
additionalTools.push(toolSearchTool);
logger.debug(`[buildToolClassification] Tool Search enabled for agent ${agentId}`);
const codeApiKey = authValues[EnvVar.CODE_API_KEY];
if (!codeApiKey) {
logger.warn('[buildToolClassification] PTC configured but CODE_API_KEY not available');
} else {
const ptcTool = createProgrammaticToolCallingTool({ apiKey: codeApiKey });
additionalTools.push(ptcTool);
logger.debug(`[buildToolClassification] PTC tool enabled for agent ${agentId}`);
}
} catch (error) {
logger.error('[buildToolClassification] Error creating PTC tool:', error);
}
} catch (error) {
logger.error('[buildToolClassification] Error creating PTC/ToolSearch tools:', error);
}
return { toolRegistry, additionalTools };