@vertesia/client 0.59.0 → 0.61.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -71,8 +71,8 @@ export class WorkflowsApi extends ApiTopic {
71
71
  return this.get(`/runs/${runId}/updates`, { query });
72
72
  }
73
73
 
74
- async streamMessages(runId: string, onMessage?: (message: AgentMessage) => void, since?: number): Promise<void> {
75
- return new Promise((resolve, reject) => {
74
+ async streamMessages(runId: string, onMessage?: (message: AgentMessage, exitFn?: (payload: unknown) => void) => void, since?: number): Promise<unknown> {
75
+ return new Promise<unknown>((resolve, reject) => {
76
76
  let reconnectAttempts = 0;
77
77
  let lastMessageTimestamp = since || 0;
78
78
  let isClosed = false;
@@ -101,6 +101,14 @@ export class WorkflowsApi extends ApiTopic {
101
101
  }
102
102
  };
103
103
 
104
+ const exit = (payload: unknown) => {
105
+ if (!isClosed) {
106
+ isClosed = true;
107
+ cleanup();
108
+ resolve(payload);
109
+ }
110
+ };
111
+
104
112
  const setupStream = async (isReconnect: boolean = false) => {
105
113
  if (isClosed) return;
106
114
 
@@ -155,7 +163,7 @@ export class WorkflowsApi extends ApiTopic {
155
163
  lastMessageTimestamp = Math.max(lastMessageTimestamp, message.timestamp);
156
164
  }
157
165
 
158
- if (onMessage) onMessage(message);
166
+ if (onMessage) onMessage(message, exit);
159
167
 
160
168
  // Only close the stream when the main workstream completes
161
169
  if (message.type === AgentMessageType.COMPLETE && (!message.workstream_id || message.workstream_id === 'main')) {
@@ -163,7 +171,7 @@ export class WorkflowsApi extends ApiTopic {
163
171
  if (!isClosed) {
164
172
  isClosed = true;
165
173
  cleanup();
166
- resolve();
174
+ resolve(null);
167
175
  }
168
176
  } else if (message.type === AgentMessageType.COMPLETE) {
169
177
  console.log(`Received COMPLETE message from non-main workstream: ${message.workstream_id || 'unknown'}, keeping stream open`);