CWC

Streaming fix

src/codewhispererChat/clients/chat/v0/chat.ts

@@ -29,6 +29,10 @@ export class ChatSession {
         this.tokenSource = new vscode.CancellationTokenSource()
     }
 
+    public setSessionID(id?: string){
+        this.sessionId = id
+    }
+
     async chat(chatRequest: GenerateAssistantResponseRequest): Promise<GenerateAssistantResponseCommandOutput> {
         this.client = await new CodeWhispererStreamingClient().createSdkClient()
 
@@ -43,15 +47,6 @@ export class ChatSession {
             )
         }
 
-        // read the first event to get conversation id.
-        // this assumes that the metadataEvent is the first event in the response stream.
-        for await (const event of response.generateAssistantResponseResponse) {
-            if (event.messageMetadataEvent !== undefined) {
-                this.sessionId = event.messageMetadataEvent!.conversationId
-            }
-            break
-        }
-
         return response
     }
 }

src/codewhispererChat/controllers/chat/messenger/messenger.ts

@@ -115,6 +115,10 @@ export class Messenger {
                         }
                     }
 
+                    if (chatEvent.messageMetadataEvent !== undefined) {
+                        session.setSessionID(chatEvent.messageMetadataEvent!.conversationId)
+                    }
+
                     if (session.tokenSource.token.isCancellationRequested) {
                         return true
                     }