@vertesia/client 0.66.0 → 0.68.0

package/src/store/WorkflowsApi.ts

@@ -31,6 +31,11 @@ export class WorkflowsApi extends ApiTopic {
         return this.post(`/runs`, { payload: { documentId, eventName, ruleId } });
     }
 
+    /** List conversations the users has access to */
+    listConversations(): Promise<ListWorkflowRunsResponse> {
+        return this.get(`/conversations`);
+    }
+
     searchRuns(payload: ListWorkflowRunsPayload): Promise<ListWorkflowRunsResponse> {
         return this.post(`/runs`, { payload: payload });
     }
@@ -70,14 +75,14 @@ export class WorkflowsApi extends ApiTopic {
         return this.post(`/runs/${runId}/updates`, { payload: msg });
     }
 
-    retrieveMessages(runId: string, since?: number): Promise<AgentMessage[]> {
+    retrieveMessages(workflowId: string, runId: string, since?: number): Promise<AgentMessage[]> {
         const query = {
             since,
         };
-        return this.get(`/runs/${runId}/updates`, { query });
+        return this.get(`/runs/${workflowId}/${runId}/updates`, { query });
     }
 
-    async streamMessages(runId: string, onMessage?: (message: AgentMessage, exitFn?: (payload: unknown) => void) => void, since?: number): Promise<unknown> {
+    async streamMessages(workflowId: string, runId: string, onMessage?: (message: AgentMessage, exitFn?: (payload: unknown) => void) => void, since?: number): Promise<unknown> {
         return new Promise<unknown>((resolve, reject) => {
             let reconnectAttempts = 0;
             let lastMessageTimestamp = since || 0;
@@ -121,7 +126,7 @@ export class WorkflowsApi extends ApiTopic {
                 try {
                     const EventSourceImpl = await EventSourceProvider();
                     const client = this.client as VertesiaClient;
-                    const streamUrl = new URL(client.workflows.baseUrl + "/runs/" + runId + "/stream");
+                    const streamUrl = new URL(client.workflows.baseUrl + `/runs/${workflowId}/${runId}/stream`);
 
                     // Use the timestamp of the last received message for reconnection
                     if (lastMessageTimestamp > 0) {
@@ -171,8 +176,12 @@ export class WorkflowsApi extends ApiTopic {
 
                             if (onMessage) onMessage(message, exit);
 
-                            // Only close the stream when the main workstream completes
-                            if (message.type === AgentMessageType.COMPLETE && (!message.workstream_id || message.workstream_id === 'main')) {
+                            const streamIsOver = message.type === AgentMessageType.TERMINATED || 
+                                (message.type === AgentMessageType.COMPLETE && 
+                                    (!message.workstream_id || message.workstream_id === 'main'));
+
+                            // Only close the stream when the main workstream completes or terminates
+                            if (streamIsOver) {
                                 console.log("Closing stream due to COMPLETE message from main workstream");
                                 if (!isClosed) {
                                     isClosed = true;