diff --git a/frontend/src/app/chat/types.ts b/frontend/src/app/chat/types.ts
index 507dfe09..5f9c54d6 100644
--- a/frontend/src/app/chat/types.ts
+++ b/frontend/src/app/chat/types.ts
@@ -4,6 +4,7 @@ export interface Message {
   timestamp: Date;
   functionCalls?: FunctionCall[];
   isStreaming?: boolean;
+  source?: "langflow" | "chat";
 }
 
 export interface FunctionCall {