diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index 3a8563cf1..465c81f1f 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -524,6 +524,11 @@ export type chatMessagePropsType = {
   chat: ChatMessageType;
   lockChat: boolean;
   lastMessage: boolean;
+  updateChat: (
+    chat: ChatMessageType,
+    message: string,
+    stream_url: string | null
+  ) => void;
 };
 
 export type formModalPropsType = {