@vertesia/client 1.1.1-dev.20260505.163000Z → 1.3.0-dev.20260620.061059Z

package/src/store/WorkflowsApi.ts

@@ -1,58 +1,61 @@
-import { ApiTopic, ClientBase } from "@vertesia/api-fetch-client";
+import { ApiTopic, type ClientBase, type IRequestParams } from '@vertesia/api-fetch-client';
 import {
-    ActivityCatalog,
-    AgentEvent,
-    AgentMessage,
+    type ActivityCatalog,
+    type AgentEvent,
+    type AgentMessage,
     AgentMessageType,
-    CompactMessage,
-    CreateWorkflowRulePayload,
-    DSLWorkflowDefinition,
-    DSLWorkflowSpec,
-    ErrorAnalyticsResponse,
-    ExecuteWorkflowPayload,
-    FirstResponseBehaviorAnalyticsResponse,
-    LatencyAnalyticsResponse,
-    ListWorkflowInteractionsResponse,
-    ListWorkflowRunsPayload,
-    ListWorkflowRunsResponse,
+    type CompactMessage,
+    type CreateWorkflowRulePayload,
+    type DSLWorkflowDefinition,
+    type DSLWorkflowSpec,
+    type ErrorAnalyticsResponse,
+    type ExecuteWorkflowPayload,
+    type FirstResponseBehaviorAnalyticsResponse,
+    type LatencyAnalyticsResponse,
+    type ListWorkflowInteractionsResponse,
+    type ListWorkflowRunsPayload,
+    type ListWorkflowRunsResponse,
+    type PromptSizeAnalyticsResponse,
     parseMessage,
+    type RunsByAgentAnalyticsResponse,
+    type SignalAgentPayload,
+    type SignalAgentResponse,
+    type TimeToFirstResponseAnalyticsResponse,
+    type TokenUsageAnalyticsResponse,
+    type ToolAnalyticsResponse,
+    type ToolParameterAnalyticsResponse,
+    type TopPrincipalsAnalyticsResponse,
     toAgentMessage,
-    PromptSizeAnalyticsResponse,
-    RunsByAgentAnalyticsResponse,
-    SignalAgentResponse,
-    TimeToFirstResponseAnalyticsResponse,
-    TokenUsageAnalyticsResponse,
-    ToolAnalyticsResponse,
-    ToolParameterAnalyticsResponse,
-    TopPrincipalsAnalyticsResponse,
-    WebSocketClientMessage,
-    WebSocketServerMessage,
-    WorkflowActionPayload,
-    WorkflowAnalyticsFilterOptionsResponse,
-    WorkflowAnalyticsSummaryQuery,
-    WorkflowAnalyticsSummaryResponse,
-    WorkflowAnalyticsTimeSeriesQuery,
-    WorkflowActionResponse,
-    WorkflowDefinitionRef,
-    WorkflowExecutionStartResult,
-    WorkflowQueryResult,
-    WorkflowRule,
-    WorkflowRuleItem,
-    WorkflowRunUpdatesResponse,
-    WorkflowRunWithDetails,
-    WorkflowToolParametersQuery,
-    WorkflowUpdatePublishResponse,
-} from "@vertesia/common";
-import { VertesiaClient } from "../client.js";
-import { EventSourceProvider } from "../execute.js";
+    type UploadWorkflowRulePayload,
+    type WebSocketClientMessage,
+    type WebSocketServerMessage,
+    type WorkflowActionPayload,
+    type WorkflowActionResponse,
+    type WorkflowAnalyticsFilterOptionsResponse,
+    type WorkflowAnalyticsSummaryQuery,
+    type WorkflowAnalyticsSummaryResponse,
+    type WorkflowAnalyticsTimeSeriesQuery,
+    type WorkflowDefinitionRef,
+    type WorkflowExecutionStartResult,
+    type WorkflowQueryResult,
+    type WorkflowRule,
+    type WorkflowRuleItem,
+    type WorkflowRunUpdatesResponse,
+    type WorkflowRunWithDetails,
+    type WorkflowToolParametersQuery,
+    type WorkflowUpdatePublishResponse,
+} from '@vertesia/common';
+import type { VertesiaClient } from '../client.js';
+import { EventSourceProvider } from '../execute.js';
+import { shouldCloseAgentRunStream, shouldCloseCompactRunStream } from './stream-termination.js';
 
 export class WorkflowsApi extends ApiTopic {
     constructor(parent: ClientBase) {
-        super(parent, "/api/v1/workflows");
+        super(parent, '/api/v1/workflows');
     }
 
     getActivityCatalog(): Promise<ActivityCatalog> {
-        return this.get("/activity-catalog");
+        return this.get('/activity-catalog');
     }
 
     listRuns(documentId: string, eventName: string, ruleId: string): Promise<ListWorkflowRunsResponse> {
@@ -62,7 +65,7 @@ export class WorkflowsApi extends ApiTopic {
     /** List conversations the users has access to */
     listConversations(payload: ListWorkflowRunsPayload): Promise<ListWorkflowRunsResponse> {
         return this.post(`/conversations`, {
-            payload
+            payload,
         });
     }
 
@@ -70,7 +73,12 @@ export class WorkflowsApi extends ApiTopic {
         return this.post(`/runs`, { payload: payload });
     }
 
-    sendSignal(workflowId: string, runId: string, signal: string, payload?: any): Promise<SignalAgentResponse> {
+    sendSignal(
+        workflowId: string,
+        runId: string,
+        signal: string,
+        payload?: SignalAgentPayload | null,
+    ): Promise<SignalAgentResponse> {
         return this.post(`/runs/${workflowId}/${runId}/signal/${signal}`, { payload });
     }
 
@@ -80,9 +88,10 @@ export class WorkflowsApi extends ApiTopic {
         options?: {
             includeHistory?: boolean;
             historyFormat?: 'events' | 'tasks' | 'agent';
-        }
+            hydratePayloads?: boolean;
+        },
     ): Promise<WorkflowRunWithDetails> {
-        const query: Record<string, any> = {};
+        const query: NonNullable<IRequestParams['query']> = {};
 
         // Support legacy includeHistory parameter
         if (options?.includeHistory !== undefined) {
@@ -94,6 +103,10 @@ export class WorkflowsApi extends ApiTopic {
             query.history_format = options.historyFormat;
         }
 
+        if (options?.hydratePayloads !== undefined) {
+            query.hydrate_payloads = options.hydratePayloads;
+        }
+
         return this.get(`/runs/${workflowId}/${runId}`, { query });
     }
 
@@ -123,16 +136,13 @@ export class WorkflowsApi extends ApiTopic {
         return this.get(`/runs/${workflowId}/${runId}/query/${queryName}`);
     }
 
-    execute(
-        name: string,
-        payload: ExecuteWorkflowPayload = {},
-    ): Promise<(WorkflowExecutionStartResult | undefined)[]> {
+    execute(name: string, payload: ExecuteWorkflowPayload = {}): Promise<(WorkflowExecutionStartResult | undefined)[]> {
         return this.post(`/execute/${name}`, { payload });
     }
 
     postMessage(runId: string, msg: AgentMessage): Promise<WorkflowUpdatePublishResponse> {
         if (!runId) {
-            throw new Error("runId is required");
+            throw new Error('runId is required');
         }
         return this.post(`/runs/${runId}/updates`, { payload: msg });
     }
@@ -144,9 +154,11 @@ export class WorkflowsApi extends ApiTopic {
      */
     async retrieveMessages(workflowId: string, runId: string, since?: number): Promise<AgentMessage[]> {
         const query = { since };
-        const response = await this.get(`/runs/${workflowId}/${runId}/updates`, { query }) as WorkflowRunUpdatesResponse;
-        // Convert compact messages to AgentMessage for backward compatibility
-        return response.messages.map((m: CompactMessage) => toAgentMessage(m, runId));
+        const response = (await this.get(`/runs/${workflowId}/${runId}/updates`, {
+            query,
+        })) as WorkflowRunUpdatesResponse;
+        // Normalize compact and legacy messages to AgentMessage for backward compatibility.
+        return response.messages.map((m) => toAgentMessage(parseMessage(m), runId));
     }
 
     /**
@@ -168,6 +180,7 @@ export class WorkflowsApi extends ApiTopic {
         return new Promise<unknown>((resolve, reject) => {
             let reconnectAttempts = 0;
             let lastMessageTimestamp = since || 0;
+            const historyFetchStartedAt = Date.now();
             let isClosed = false;
             let currentSse: EventSource | null = null;
             let interval: ReturnType<typeof setInterval> | null = null;
@@ -179,7 +192,7 @@ export class WorkflowsApi extends ApiTopic {
             const maxDelay = 30000; // 30 seconds max delay
 
             const calculateBackoffDelay = (attempts: number): number => {
-                const exponentialDelay = Math.min(baseDelay * Math.pow(2, attempts), maxDelay);
+                const exponentialDelay = Math.min(baseDelay * 2 ** attempts, maxDelay);
                 // Add jitter to prevent thundering herd
                 const jitter = Math.random() * 0.1 * exponentialDelay;
                 return exponentialDelay + jitter;
@@ -228,37 +241,39 @@ export class WorkflowsApi extends ApiTopic {
                 try {
                     const EventSourceImpl = await EventSourceProvider();
                     const client = this.client as VertesiaClient;
-                    const streamUrl = new URL(client.workflows.baseUrl + `/runs/${workflowId}/${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) {
-                        streamUrl.searchParams.set("since", lastMessageTimestamp.toString());
+                        streamUrl.searchParams.set('since', lastMessageTimestamp.toString());
                     }
 
                     // Skip historical messages - we already fetched them via GET /updates
-                    streamUrl.searchParams.set("skipHistory", "true");
+                    streamUrl.searchParams.set('skipHistory', 'true');
 
                     const bearerToken = client._auth ? await client._auth() : undefined;
                     if (isClosed) return;
                     if (!bearerToken) {
                         isClosed = true;
                         cleanup();
-                        reject(new Error("No auth token available"));
+                        reject(new Error('No auth token available'));
                         return;
                     }
 
-                    const token = bearerToken.split(" ")[1];
-                    streamUrl.searchParams.set("access_token", token);
+                    const token = bearerToken.split(' ')[1];
+                    streamUrl.searchParams.set('access_token', token);
 
                     if (isReconnect) {
-                        console.log(`Reconnecting to SSE stream for run ${runId} (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`);
+                        console.log(
+                            `Reconnecting to SSE stream for run ${runId} (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`,
+                        );
                     }
 
                     const sse = new EventSourceImpl(streamUrl.href);
                     currentSse = sse;
 
                     // Prevent Node from exiting prematurely
-                    interval = setInterval(() => { }, 1000);
+                    interval = setInterval(() => {}, 1000);
 
                     sse.onopen = () => {
                         if (isReconnect) {
@@ -269,8 +284,8 @@ export class WorkflowsApi extends ApiTopic {
                     };
 
                     sse.onmessage = (ev: MessageEvent) => {
-                        if (!ev.data || ev.data.startsWith(":")) {
-                            console.log("Received comment or heartbeat; ignoring it.: ", ev.data);
+                        if (!ev.data || ev.data.startsWith(':')) {
+                            console.log('Received comment or heartbeat; ignoring it.: ', ev.data);
                             return;
                         }
 
@@ -291,29 +306,24 @@ export class WorkflowsApi extends ApiTopic {
                                 onMessage(agentMessage, exit);
                             }
 
-                            // Get workstream ID (defaults to 'main' if not set)
-                            const workstreamId = compactMessage.w || 'main';
-
-                            const streamIsOver = compactMessage.t === AgentMessageType.TERMINATED ||
-                                (compactMessage.t === AgentMessageType.COMPLETE && workstreamId === '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 (shouldCloseCompactRunStream(compactMessage, runId)) {
+                                console.log('Closing stream due to COMPLETE message from main workstream');
                                 if (!isClosed) {
                                     isClosed = true;
                                     cleanup();
                                     resolve(null);
                                 }
                             } else if (compactMessage.t === AgentMessageType.COMPLETE) {
-                                console.log(`Received COMPLETE message from non-main workstream: ${workstreamId}, keeping stream open`);
+                                console.log(
+                                    `Received COMPLETE message that does not close the root stream for run ${runId}`,
+                                );
                             }
                         } catch (err) {
-                            console.error("Failed to parse SSE message:", err, ev.data);
+                            console.error('Failed to parse SSE message:', err, ev.data);
                         }
                     };
 
-                    sse.onerror = (err: any) => {
+                    sse.onerror = (err: unknown) => {
                         if (isClosed) return;
 
                         console.warn(`SSE stream error for run ${runId}:`, err);
@@ -322,30 +332,36 @@ export class WorkflowsApi extends ApiTopic {
                         // Check if we should attempt reconnection
                         if (reconnectAttempts < maxReconnectAttempts) {
                             const delay = calculateBackoffDelay(reconnectAttempts);
-                            console.log(`Attempting to reconnect in ${delay}ms (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`);
+                            console.log(
+                                `Attempting to reconnect in ${delay}ms (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`,
+                            );
 
                             reconnectAttempts++;
                             reconnectTimer = setTimeout(() => {
                                 reconnectTimer = null;
                                 if (!isClosed) {
-                                    setupStream(true);
+                                    void setupStream(true);
                                 }
                             }, delay);
                         } else {
-                            console.error(`Failed to reconnect to SSE stream for run ${runId} after ${maxReconnectAttempts} attempts`);
+                            console.error(
+                                `Failed to reconnect to SSE stream for run ${runId} after ${maxReconnectAttempts} attempts`,
+                            );
                             isClosed = true;
-                            reject(new Error(`SSE connection failed after ${maxReconnectAttempts} reconnection attempts`));
+                            reject(
+                                new Error(`SSE connection failed after ${maxReconnectAttempts} reconnection attempts`),
+                            );
                         }
                     };
                 } catch (err) {
-                    console.error("Error setting up SSE stream:", err);
+                    console.error('Error setting up SSE stream:', err);
                     if (reconnectAttempts < maxReconnectAttempts) {
                         const delay = calculateBackoffDelay(reconnectAttempts);
                         reconnectAttempts++;
                         reconnectTimer = setTimeout(() => {
                             reconnectTimer = null;
                             if (!isClosed) {
-                                setupStream(true);
+                                void setupStream(true);
                             }
                         }, delay);
                     } else {
@@ -359,25 +375,33 @@ export class WorkflowsApi extends ApiTopic {
                 // 1. Fetch historical messages via GET /updates (gzip-compressed if > 3KB)
                 try {
                     const historical = await this.retrieveMessages(workflowId, runId, since);
-                    for (const msg of historical) {
+                    let shouldCloseAfterHistory = false;
+                    for (let index = 0; index < historical.length; index++) {
+                        const msg = historical[index];
                         lastMessageTimestamp = Math.max(lastMessageTimestamp, msg.timestamp || 0);
                         if (onMessage) {
                             onMessage(msg, exit);
                         }
-                        const workstreamId = msg.workstream_id || 'main';
-                        const streamIsOver = msg.type === AgentMessageType.TERMINATED ||
-                            (msg.type === AgentMessageType.COMPLETE && workstreamId === 'main');
-                        if (streamIsOver) {
-                            exit(null);
-                            return;
-                        }
+                        shouldCloseAfterHistory =
+                            index === historical.length - 1 && shouldCloseAgentRunStream(msg, runId);
+                    }
+                    if (shouldCloseAfterHistory) {
+                        exit(null);
+                        return;
                     }
                 } catch (err) {
-                    console.warn("Failed to fetch historical messages, continuing with SSE:", err);
+                    console.warn('Failed to fetch historical messages, continuing with SSE:', err);
+                }
+                if (!isClosed && lastMessageTimestamp <= 0) {
+                    // The server only replays the GET-to-SSE handoff gap when
+                    // `since > 0`. New runs can have no history at the first
+                    // fetch, so use the GET start time as the cursor to avoid
+                    // dropping messages emitted before the SSE subscription is active.
+                    lastMessageTimestamp = Math.max(1, historyFetchStartedAt - 1);
                 }
 
                 // 2. Connect to SSE
-                setupStream(false);
+                void setupStream(false);
             };
 
             init().catch(reject);
@@ -396,8 +420,8 @@ export class WorkflowsApi extends ApiTopic {
         workflowId: string,
         runId: string,
         onMessage?: (message: CompactMessage) => void,
-        since?: number
-    ): Promise<{ cleanup: () => void; sendSignal: (signalName: string, data: any) => void }> {
+        since?: number,
+    ): Promise<{ cleanup: () => void; sendSignal: (signalName: string, data: unknown) => void }> {
         return new Promise((resolve, reject) => {
             let reconnectAttempts = 0;
             const maxReconnectAttempts = 10;
@@ -408,7 +432,7 @@ export class WorkflowsApi extends ApiTopic {
             let isClosed = false;
 
             const calculateBackoffDelay = (attempts: number): number => {
-                const exponentialDelay = Math.min(baseDelay * Math.pow(2, attempts), maxDelay);
+                const exponentialDelay = Math.min(baseDelay * 2 ** attempts, maxDelay);
                 const jitter = Math.random() * 0.1 * exponentialDelay;
                 return exponentialDelay + jitter;
             };
@@ -418,7 +442,7 @@ export class WorkflowsApi extends ApiTopic {
 
                 try {
                     const client = this.client as VertesiaClient;
-                    const wsUrl = new URL(client.workflows.baseUrl + `/runs/${workflowId}/${runId}/ws`);
+                    const wsUrl = new URL(`${client.workflows.baseUrl}/runs/${workflowId}/${runId}/ws`);
 
                     // Replace http/https with ws/wss
                     wsUrl.protocol = wsUrl.protocol.replace('http', 'ws');
@@ -438,7 +462,9 @@ export class WorkflowsApi extends ApiTopic {
                     wsUrl.searchParams.set('access_token', token);
 
                     if (reconnectAttempts > 0) {
-                        console.log(`Reconnecting to WebSocket for run ${runId} (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`);
+                        console.log(
+                            `Reconnecting to WebSocket for run ${runId} (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`,
+                        );
                     }
 
                     ws = new WebSocket(wsUrl.href);
@@ -459,19 +485,19 @@ export class WorkflowsApi extends ApiTopic {
                                         ws = null;
                                     }
                                 },
-                                sendSignal: (signalName: string, data: any) => {
+                                sendSignal: (signalName: string, data: unknown) => {
                                     if (ws?.readyState === WebSocket.OPEN) {
                                         const message: WebSocketClientMessage = {
                                             type: 'signal',
                                             signalName,
                                             data,
-                                            requestId: Date.now()
+                                            requestId: Date.now(),
                                         };
                                         ws.send(JSON.stringify(message));
                                     } else {
                                         console.warn('WebSocket not open, cannot send signal');
                                     }
-                                }
+                                },
                             });
                         }
                     };
@@ -534,7 +560,9 @@ export class WorkflowsApi extends ApiTopic {
                     ws.onclose = () => {
                         if (!isClosed && reconnectAttempts < maxReconnectAttempts) {
                             const delay = calculateBackoffDelay(reconnectAttempts);
-                            console.log(`WebSocket closed, reconnecting in ${delay}ms (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`);
+                            console.log(
+                                `WebSocket closed, reconnecting in ${delay}ms (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`,
+                            );
                             reconnectAttempts++;
                             setTimeout(connect, delay);
                         } else if (reconnectAttempts >= maxReconnectAttempts) {
@@ -553,7 +581,7 @@ export class WorkflowsApi extends ApiTopic {
                 }
             };
 
-            connect();
+            void connect();
         });
     }
 
@@ -564,7 +592,7 @@ export class WorkflowsApi extends ApiTopic {
     ingestEvents(
         workflowId: string,
         runId: string,
-        events: AgentEvent[]
+        events: AgentEvent[],
     ): Promise<{ ingested: number; status?: string; error?: string }> {
         return this.post(`/runs/${workflowId}/${runId}/events`, {
             payload: { events },
@@ -579,9 +607,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get workflow analytics summary.
      * Returns overall metrics including token usage, success rates, and run counts.
      */
-    getAnalyticsSummary(
-        query: WorkflowAnalyticsSummaryQuery = {}
-    ): Promise<WorkflowAnalyticsSummaryResponse> {
+    getAnalyticsSummary(query: WorkflowAnalyticsSummaryQuery = {}): Promise<WorkflowAnalyticsSummaryResponse> {
         return this.post('/analytics/summary', { payload: query });
     }
 
@@ -589,9 +615,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get token usage analytics.
      * Returns token consumption metrics by model, agent, tool, or over time.
      */
-    getTokenUsageAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
-    ): Promise<TokenUsageAnalyticsResponse> {
+    getTokenUsageAnalytics(query: WorkflowAnalyticsTimeSeriesQuery = {}): Promise<TokenUsageAnalyticsResponse> {
         return this.post('/analytics/tokens', { payload: query });
     }
 
@@ -599,9 +623,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get LLM latency analytics.
      * Returns duration/latency metrics for LLM calls.
      */
-    getLlmLatencyAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
-    ): Promise<LatencyAnalyticsResponse> {
+    getLlmLatencyAnalytics(query: WorkflowAnalyticsTimeSeriesQuery = {}): Promise<LatencyAnalyticsResponse> {
         return this.post('/analytics/latency/llm', { payload: query });
     }
 
@@ -609,9 +631,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get tool latency analytics.
      * Returns duration/latency metrics for tool calls.
      */
-    getToolLatencyAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
-    ): Promise<LatencyAnalyticsResponse> {
+    getToolLatencyAnalytics(query: WorkflowAnalyticsTimeSeriesQuery = {}): Promise<LatencyAnalyticsResponse> {
         return this.post('/analytics/latency/tools', { payload: query });
     }
 
@@ -619,9 +639,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get agent/workflow latency analytics.
      * Returns duration metrics for complete workflow runs.
      */
-    getAgentLatencyAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
-    ): Promise<LatencyAnalyticsResponse> {
+    getAgentLatencyAnalytics(query: WorkflowAnalyticsTimeSeriesQuery = {}): Promise<LatencyAnalyticsResponse> {
         return this.post('/analytics/latency/agents', { payload: query });
     }
 
@@ -629,9 +647,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get error analytics.
      * Returns error rates, types, and trends.
      */
-    getErrorAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
-    ): Promise<ErrorAnalyticsResponse> {
+    getErrorAnalytics(query: WorkflowAnalyticsTimeSeriesQuery = {}): Promise<ErrorAnalyticsResponse> {
         return this.post('/analytics/errors', { payload: query });
     }
 
@@ -639,9 +655,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get tool usage analytics.
      * Returns tool invocation counts, success rates, and performance metrics.
      */
-    getToolAnalytics(
-        query: WorkflowAnalyticsSummaryQuery = {}
-    ): Promise<ToolAnalyticsResponse> {
+    getToolAnalytics(query: WorkflowAnalyticsSummaryQuery = {}): Promise<ToolAnalyticsResponse> {
         return this.post('/analytics/tools', { payload: query });
     }
 
@@ -649,9 +663,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get tool parameter analytics.
      * Returns parameter value distributions for a specific tool.
      */
-    getToolParameterAnalytics(
-        query: WorkflowToolParametersQuery
-    ): Promise<ToolParameterAnalyticsResponse> {
+    getToolParameterAnalytics(query: WorkflowToolParametersQuery): Promise<ToolParameterAnalyticsResponse> {
         return this.post('/analytics/tools/parameters', { payload: query });
     }
 
@@ -660,7 +672,7 @@ export class WorkflowsApi extends ApiTopic {
      * Returns unique agents, environments, and models from telemetry data.
      */
     getAnalyticsFilterOptions(
-        query: WorkflowAnalyticsSummaryQuery = {}
+        query: WorkflowAnalyticsSummaryQuery = {},
     ): Promise<WorkflowAnalyticsFilterOptionsResponse> {
         return this.post('/analytics/filter-options', { payload: query });
     }
@@ -669,9 +681,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get average prompt size (input tokens) by agent for startConversation calls.
      * This represents the initial prompt + tools size.
      */
-    getPromptSizeAnalytics(
-        query: WorkflowAnalyticsSummaryQuery = {}
-    ): Promise<PromptSizeAnalyticsResponse> {
+    getPromptSizeAnalytics(query: WorkflowAnalyticsSummaryQuery = {}): Promise<PromptSizeAnalyticsResponse> {
         return this.post('/analytics/prompt-size', { payload: query });
     }
 
@@ -679,9 +689,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get top principals (users/API keys) who started the most agent runs.
      * Returns the top N principals sorted by run count descending.
      */
-    getTopPrincipalsAnalytics(
-        query: WorkflowAnalyticsSummaryQuery = {}
-    ): Promise<TopPrincipalsAnalyticsResponse> {
+    getTopPrincipalsAnalytics(query: WorkflowAnalyticsSummaryQuery = {}): Promise<TopPrincipalsAnalyticsResponse> {
         return this.post('/analytics/top-principals', { payload: query });
     }
 
@@ -689,9 +697,7 @@ export class WorkflowsApi extends ApiTopic {
      * Get agent run distribution - how many runs per agent/interaction type.
      * Returns the top N agents sorted by run count descending.
      */
-    getRunsByAgentAnalytics(
-        query: WorkflowAnalyticsSummaryQuery = {}
-    ): Promise<RunsByAgentAnalyticsResponse> {
+    getRunsByAgentAnalytics(query: WorkflowAnalyticsSummaryQuery = {}): Promise<RunsByAgentAnalyticsResponse> {
         return this.post('/analytics/runs-by-agent', { payload: query });
     }
 
@@ -701,7 +707,7 @@ export class WorkflowsApi extends ApiTopic {
      * Returns average, min, max, median, p95, and p99 metrics.
      */
     getTimeToFirstResponseAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
+        query: WorkflowAnalyticsTimeSeriesQuery = {},
     ): Promise<TimeToFirstResponseAnalyticsResponse> {
         return this.post('/analytics/time-to-first-response', { payload: query });
     }
@@ -713,7 +719,7 @@ export class WorkflowsApi extends ApiTopic {
      * - Percentage of agents that return no tool calls at start
      */
     getFirstResponseBehaviorAnalytics(
-        query: WorkflowAnalyticsTimeSeriesQuery = {}
+        query: WorkflowAnalyticsTimeSeriesQuery = {},
     ): Promise<FirstResponseBehaviorAnalyticsResponse> {
         return this.post('/analytics/first-response-behavior', { payload: query });
     }
@@ -724,25 +730,25 @@ export class WorkflowsApi extends ApiTopic {
 
 export class WorkflowsRulesApi extends ApiTopic {
     constructor(parent: WorkflowsApi) {
-        super(parent, "/rules");
+        super(parent, '/rules');
     }
 
     list(): Promise<WorkflowRuleItem[]> {
-        return this.get("/");
+        return this.get('/');
     }
 
     retrieve(id: string): Promise<WorkflowRule> {
         return this.get(`/${id}`);
     }
 
-    update(id: string, payload: any): Promise<WorkflowRule> {
+    update(id: string, payload: UploadWorkflowRulePayload): Promise<WorkflowRule> {
         return this.put(`/${id}`, {
             payload,
         });
     }
 
     create(payload: CreateWorkflowRulePayload): Promise<WorkflowRule> {
-        return this.post("/", {
+        return this.post('/', {
             payload,
         });
     }
@@ -754,7 +760,7 @@ export class WorkflowsRulesApi extends ApiTopic {
     execute(
         id: string,
         objectIds?: string[],
-        vars?: Record<string, any>,
+        vars?: Record<string, unknown>,
     ): Promise<({ run_id: string; workflow_id: string } | undefined)[]> {
         const payload: ExecuteWorkflowPayload = {
             objectIds,
@@ -768,25 +774,25 @@ export class WorkflowsDefinitionApi extends ApiTopic {
     //model: DSLWorkflowDefinition;
 
     constructor(parent: WorkflowsApi) {
-        super(parent, "/definitions");
+        super(parent, '/definitions');
     }
 
     list(): Promise<WorkflowDefinitionRef[]> {
-        return this.get("/");
+        return this.get('/');
     }
 
     retrieve(id: string): Promise<DSLWorkflowDefinition> {
         return this.get(`/${id}`);
     }
 
-    update(id: string, payload: any): Promise<DSLWorkflowDefinition> {
+    update(id: string, payload: DSLWorkflowSpec): Promise<DSLWorkflowDefinition> {
         return this.put(`/${id}`, {
             payload,
         });
     }
 
     create(payload: DSLWorkflowSpec): Promise<DSLWorkflowDefinition> {
-        return this.post("/", {
+        return this.post('/', {
             payload,
         });
     }