diff --git a/src/lib/agents/search/researcher/actions/registry.ts b/src/lib/agents/search/researcher/actions/registry.ts index 4172e8b..5a05ac1 100644 --- a/src/lib/agents/search/researcher/actions/registry.ts +++ b/src/lib/agents/search/researcher/actions/registry.ts @@ -1,5 +1,5 @@ +import { Tool, ToolCall } from '@/lib/models/types'; import { - ActionConfig, ActionOutput, AdditionalConfig, ClassifierOutput, @@ -25,6 +25,18 @@ class ActionRegistry { ); } + static getAvailableActionTools(config: { + classification: ClassifierOutput; + }): Tool[] { + const availableActions = this.getAvailableActions(config); + + return availableActions.map((action) => ({ + name: action.name, + description: action.description, + schema: action.schema, + })); + } + static getAvailableActionsDescriptions(config: { classification: ClassifierOutput; }): string { @@ -50,7 +62,7 @@ class ActionRegistry { } static async executeAll( - actions: ActionConfig[], + actions: ToolCall[], additionalConfig: AdditionalConfig, ): Promise { const results: ActionOutput[] = []; @@ -58,8 +70,8 @@ class ActionRegistry { await Promise.all( actions.map(async (actionConfig) => { const output = await this.execute( - actionConfig.type, - actionConfig.params, + actionConfig.name, + actionConfig.arguments, additionalConfig, ); results.push(output);