From 8aed9518a239b118a7031029206de55135a8a4dc Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Mon, 8 Dec 2025 13:09:52 +0530
Subject: [PATCH] feat(researcher): pass research block id

---
 src/lib/agents/search/researcher/actions/registry.ts | 4 ++--
 src/lib/agents/search/types.ts                       | 5 ++++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/lib/agents/search/researcher/actions/registry.ts b/src/lib/agents/search/researcher/actions/registry.ts
index 5a05ac1..7454763 100644
--- a/src/lib/agents/search/researcher/actions/registry.ts
+++ b/src/lib/agents/search/researcher/actions/registry.ts
@@ -50,7 +50,7 @@ class ActionRegistry {
   static async execute(
     name: string,
     params: any,
-    additionalConfig: AdditionalConfig,
+    additionalConfig: AdditionalConfig & { researchBlockId: string },
   ) {
     const action = this.actions.get(name);
 
@@ -63,7 +63,7 @@ class ActionRegistry {
 
   static async executeAll(
     actions: ToolCall[],
-    additionalConfig: AdditionalConfig,
+    additionalConfig: AdditionalConfig & { researchBlockId: string },
   ): Promise<ActionOutput[]> {
     const results: ActionOutput[] = [];
 
diff --git a/src/lib/agents/search/types.ts b/src/lib/agents/search/types.ts
index e83ac7d..6a6ac69 100644
--- a/src/lib/agents/search/types.ts
+++ b/src/lib/agents/search/types.ts
@@ -73,6 +73,7 @@ export type ResearcherInput = {
 
 export type ResearcherOutput = {
   findings: ActionOutput[];
+  searchFindings: Chunk[];
 };
 
 export type SearchActionOutput = {
@@ -103,6 +104,8 @@ export interface ResearchAction<
   enabled: (config: { classification: ClassifierOutput }) => boolean;
   execute: (
     params: z.infer<TSchema>,
-    additionalConfig: AdditionalConfig,
+    additionalConfig: AdditionalConfig & {
+      researchBlockId: string;
+    },
   ) => Promise<ActionOutput>;
 }