From 8ab675b119099b20423d9a3e5e38cb566f5f7cbf Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Sat, 6 Dec 2025 15:37:36 +0530
Subject: [PATCH] feat(action-registry): use tool types, add tool methods

---
 .../search/researcher/actions/registry.ts     | 20 +++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

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<ActionOutput[]> {
     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);