Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-21 10:50:14 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: tool classification and calling logic

Browse source
This commit is contained in:
Danny Avila 2025-12-08 14:57:14 -05:00
parent 3588929c70
commit 984607d339
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
6 changed files with 25 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -282,8 +282,10 @@ export interface BuildToolClassificationResult {
/**
* Checks if an agent is allowed to have classification features based on TOOL_CLASSIFICATION_AGENT_IDS.
* If TOOL_CLASSIFICATION_AGENT_IDS is not set, all agents are allowed (including when no agentId).
* If set, requires agentId to be in the list.
* @param agentId - The agent ID to check
* @returns Whether the agent is allowed (true if no restriction set, or agent is in the list)
* @returns Whether the agent is allowed
*/
export function isAgentAllowedForClassification(agentId?: string): boolean {
const allowedAgentIds = parseToolList(process.env.TOOL_CLASSIFICATION_AGENT_IDS);
@ -344,6 +346,14 @@ export async function buildToolClassification(
const { loadedTools, userId, agentId, loadAuthValues } = params;
const additionalTools: GenericTool[] = [];
/** Check if this agent is allowed to have classification features (requires agentId) */
if (!isAgentAllowedForClassification(agentId)) {
logger.debug(
`[buildToolClassification] Agent ${agentId ?? 'undefined'} not allowed for classification, skipping`,
);
return { toolRegistry: undefined, additionalTools };
}
const mcpTools = loadedTools.filter(isMCPTool);
if (mcpTools.length === 0) {
return { toolRegistry: undefined, additionalTools };
@ -355,14 +365,6 @@ export async function buildToolClassification(
/** Clean up temporary mcpJsonSchema property from tools now that registry is populated */
cleanupMCPToolSchemas(mcpTools);
/** Check if this agent is allowed to have classification features */
if (!isAgentAllowedForClassification(agentId)) {
logger.debug(
`[buildToolClassification] Agent ${agentId} not in TOOL_CLASSIFICATION_AGENT_IDS, skipping PTC/ToolSearch`,
);
return { toolRegistry, additionalTools };
}
/**
* Check if this agent actually has tools that match the patterns.
* Only enable PTC if the agent has programmatic tools.