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>;
 }