From 65b2c3f234571f1987d92f897ba063701c78c0a4 Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Wed, 8 Apr 2026 23:21:02 +0530
Subject: [PATCH] feat(additional-config): add focus mode

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

diff --git a/src/lib/agents/search/researcher/actions/registry.ts b/src/lib/agents/search/researcher/actions/registry.ts
index a8de5132..b96d6438 100644
--- a/src/lib/agents/search/researcher/actions/registry.ts
+++ b/src/lib/agents/search/researcher/actions/registry.ts
@@ -67,6 +67,7 @@ class ActionRegistry {
     additionalConfig: AdditionalConfig & {
       researchBlockId: string;
       fileIds: string[];
+      mode: SearchAgentConfig['mode'];
     },
   ) {
     const action = this.actions.get(name);
@@ -83,6 +84,7 @@ class ActionRegistry {
     additionalConfig: AdditionalConfig & {
       researchBlockId: string;
       fileIds: string[];
+      mode: SearchAgentConfig['mode'];
     },
   ): Promise<ActionOutput[]> {
     const results: ActionOutput[] = [];
diff --git a/src/lib/agents/search/researcher/index.ts b/src/lib/agents/search/researcher/index.ts
index d6532819..0a25d91d 100644
--- a/src/lib/agents/search/researcher/index.ts
+++ b/src/lib/agents/search/researcher/index.ts
@@ -167,6 +167,7 @@ class Researcher {
         session: session,
         researchBlockId: researchBlockId,
         fileIds: input.config.fileIds,
+        mode: input.config.mode,
       });
 
       actionOutput.push(...actionResults);
diff --git a/src/lib/agents/search/types.ts b/src/lib/agents/search/types.ts
index 64c967ea..d77db7c8 100644
--- a/src/lib/agents/search/types.ts
+++ b/src/lib/agents/search/types.ts
@@ -117,6 +117,7 @@ export interface ResearchAction<
     additionalConfig: AdditionalConfig & {
       researchBlockId: string;
       fileIds: string[];
+      mode: SearchAgentConfig['mode'];
     },
   ) => Promise<ActionOutput>;
 }