Merge pull request stakwork#764 from MahtabBukhari/Update-the-chat-st…

…ore-to-include-workspace-id-and-include-in-send Update the chat store to include workspace id and include in send
aliraza556 · Dec 13, 2024 · 72f8efd · 72f8efd
2 parents cb47b56 + 1ed6822
commit 72f8efd
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/src/services/index.ts b/src/services/index.ts
@@ -60,6 +60,8 @@ export class ChatService {
   async sendMessage(
     chat_id: string,
     message: string,
+    sourceWebsocketID: string,
+    workspaceUUID: string,
     contextTags?: ContextTag[]
   ): Promise<ChatMessage | undefined> {
     try {
@@ -76,7 +78,9 @@ export class ChatService {
         body: JSON.stringify({
           chat_id,
           message,
-          context_tags: contextTags
+          context_tags: contextTags,
+          sourceWebsocketID,
+          workspaceUUID
         })
       });
 

diff --git a/src/store/chat.ts b/src/store/chat.ts
@@ -91,9 +91,17 @@ export class ChatHistoryStore implements ChatStore {
   async sendMessage(
     chat_id: string,
     message: string,
+    sourceWebsocketID: string,
+    workspaceUUID: string,
     contextTags?: ContextTag[]
   ): Promise<ChatMessage | undefined> {
-    const newMessage = await chatService.sendMessage(chat_id, message, contextTags);
+    const newMessage = await chatService.sendMessage(
+      chat_id,
+      message,
+      sourceWebsocketID,
+      workspaceUUID,
+      contextTags
+    );
     if (newMessage) {
       this.addMessage(newMessage);
     }

diff --git a/src/store/interface.ts b/src/store/interface.ts
@@ -478,6 +478,7 @@ export interface ChatMessage {
   status: ChatStatus;
   source: ChatSource;
   sourceWebsocketID: string;
+  workspaceUUID: string;
 }
 
 export interface Chat {