diff --git a/src/lib/agents/search/types.ts b/src/lib/agents/search/types.ts
index 04cbe19..ac511e6 100644
--- a/src/lib/agents/search/types.ts
+++ b/src/lib/agents/search/types.ts
@@ -18,6 +18,8 @@ export type SearchAgentInput = {
   chatHistory: ChatTurnMessage[];
   followUp: string;
   config: SearchAgentConfig;
+  chatId: string;
+  messageId: string;
 };
 
 export type WidgetInput = {