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