@vertesia/client 1.1.0 → 1.1.1-dev.20260505.163000Z

package/src/store/DataApi.ts

@@ -1,21 +1,27 @@
 import { ApiTopic, ClientBase } from "@vertesia/api-fetch-client";
 import {
     AlterTablePayload,
+    BatchQueryResult,
     CreateDataStorePayload,
     CreateSnapshotPayload,
     CreateTablesPayload,
     DataSchema,
     DataSchemaForAI,
+    DataStoreArchiveResult,
     DataStore,
     DataStoreApiHeaders,
+    DataStoreDownloadInfo,
     DataStoreItem,
     DataStoreVersion,
     DataTable,
+    DataStoreTableDetail,
+    DataStoreTableDropResult,
     DataTableSummary,
     ImportDataPayload,
     ImportJob,
     QueryPayload,
     QueryResult,
+    QueryValidationResult,
     UpdateSchemaPayload,
 } from "@vertesia/common";
 import { DashboardApi } from "./DashboardApi.js";
@@ -89,7 +95,7 @@ export class DataApi extends ApiTopic {
      * @param id - Data store ID
      * @returns Object with the archived store ID
      */
-    delete(id: string): Promise<{ id: string }> {
+    delete(id: string): Promise<DataStoreArchiveResult> {
         return this.del(`/${id}`, { headers: this.storeHeaders(id) });
     }
 
@@ -204,7 +210,7 @@ export class DataApi extends ApiTopic {
      * @param sample - If true, includes sample rows
      * @returns The table with metadata and optional sample data
      */
-    getTable(id: string, tableName: string, sample?: boolean): Promise<DataTable & { sampleRows?: Record<string, unknown>[] }> {
+    getTable(id: string, tableName: string, sample?: boolean): Promise<DataStoreTableDetail> {
         const query = sample ? '?sample=true' : '';
         return this.get(`/${id}/tables/${tableName}${query}`, { headers: this.storeHeaders(id) });
     }
@@ -227,7 +233,7 @@ export class DataApi extends ApiTopic {
      * @param id - Data store ID
      * @param tableName - Table name
      */
-    dropTable(id: string, tableName: string): Promise<void> {
+    dropTable(id: string, tableName: string): Promise<DataStoreTableDropResult> {
         return this.del(`/${id}/tables/${tableName}`, { headers: this.storeHeaders(id) });
     }
 
@@ -383,6 +389,10 @@ export class DataApi extends ApiTopic {
         return this.post(`/${id}/query`, { payload, headers: this.storeHeaders(id) });
     }
 
+    queryBatch(id: string, queries: QueryPayload[]): Promise<BatchQueryResult> {
+        return this.post(`/${id}/query/batch`, { payload: { queries }, headers: this.storeHeaders(id) });
+    }
+
     /**
      * Validate SQL queries without executing them.
      *
@@ -479,38 +489,3 @@ export class DataApi extends ApiTopic {
         return new DashboardApi(this.client, storeId);
     }
 }
-
-/**
- * Response from the download endpoint.
- */
-export interface DataStoreDownloadInfo {
-    /** Signed download URL (expires in 15 min) */
-    url: string;
-    /** GCS generation number for cache validation */
-    gcs_generation: number;
-    /** Schema version */
-    schema_version: string;
-    /** Store ID */
-    store_id: string;
-    /** Store name */
-    store_name: string;
-    /** List of table names */
-    tables: string[];
-    /** URL expiry time in seconds */
-    expires_in: number;
-}
-
-/**
- * Result from SQL query validation.
- */
-export interface QueryValidationResult {
-    /** Whether all queries are valid */
-    valid: boolean;
-    /** Validation errors (if any) */
-    errors: Array<{
-        /** Query name that failed validation */
-        query: string;
-        /** Error message describing the issue */
-        error: string;
-    }>;
-}

package/src/store/FilesApi.ts

@@ -2,6 +2,11 @@ import { ApiTopic, ClientBase } from "@vertesia/api-fetch-client";
 import {
     BulkUploadUrlsPayload,
     BulkUploadUrlsResponse,
+    DeleteFileResult,
+    FileBucketResponse,
+    FileListResponse,
+    FileMetadataResponse,
+    FileMetadataUpdateResult,
     GetFileUrlPayload,
     GetFileUrlResponse,
     GetUploadUrlPayload,
@@ -31,9 +36,8 @@ export class FilesApi extends ApiTopic {
         super(parent, "/api/v1/files");
     }
 
-    async deleteFile(path: string, prefix?: boolean): Promise<void | number> {
-        const res = await this.delete(`/${path}`, { query: { prefix } });
-        return res.count;
+    async deleteFile(path: string, prefix?: boolean): Promise<DeleteFileResult> {
+        return this.delete(`/${path}`, { query: { prefix } });
     }
 
     /**
@@ -43,14 +47,7 @@ export class FilesApi extends ApiTopic {
      * @param uri
      * @returns
      */
-    getMetadata(uri: string): Promise<{
-        name: string;
-        size: number;
-        contentType: string;
-        contentDisposition?: string;
-        etag?: string;
-        customMetadata?: Record<string, string>;
-    }> {
+    getMetadata(uri: string): Promise<FileMetadataResponse> {
         return this.get("/metadata", {
             query: {
                 file: uri,
@@ -64,7 +61,7 @@ export class FilesApi extends ApiTopic {
      * @param metadata - Custom metadata key-value pairs
      * @returns Success status
      */
-    setFileMetadata(file: string, metadata: Record<string, string>): Promise<{ success: boolean; file: string }> {
+    setFileMetadata(file: string, metadata: Record<string, string>): Promise<FileMetadataUpdateResult> {
         const payload: SetFileMetadataPayload = { file, metadata };
         return this.put("/metadata", { payload });
     }
@@ -74,10 +71,14 @@ export class FilesApi extends ApiTopic {
      * The bucket URI is returned.
      * @returns
      */
-    getOrCreateBucket(): Promise<{ bucket: string }> {
+    getOrCreateBucket(): Promise<FileBucketResponse> {
         return this.post("/bucket");
     }
 
+    list(prefix: string): Promise<FileListResponse> {
+        return this.get('/list', { query: { prefix } });
+    }
+
     getUploadUrl(payload: GetUploadUrlPayload): Promise<GetFileUrlResponse> {
         return this.post("/upload-url", {
             payload,

package/src/store/IndexingApi.ts

@@ -16,6 +16,7 @@ import {
     EnsureIndexResult,
     AnalyzeDriftBatchResult,
     DriftAnalysisStatusResponse,
+    StartProjectReindexPayload,
     ComputeShardsRequest,
     ComputeShardsResult,
     IndexShardParams,
@@ -57,13 +58,7 @@ export class IndexingApi extends ApiTopic {
      *
      * @param options Optional workflow tuning parameters
      */
-    async reindex(options?: {
-        shard_size?: number;
-        parallel_shard_count?: number;
-        concurrency?: number;
-        bulk_size_bytes?: number;
-        bulk_concurrency?: number;
-    }): Promise<GenericCommandResponse> {
+    async reindex(options?: StartProjectReindexPayload): Promise<GenericCommandResponse> {
         return this.post("/reindex", { payload: options });
     }
 

package/src/store/ObjectsApi.ts

@@ -4,6 +4,7 @@ import {
     BulkObjectDeleteResult,
     BulkObjectUpdateResult,
     canGenerateRendition,
+    Collection,
     ComplexSearchPayload,
     ComputeObjectFacetPayload,
     ContentObject,
@@ -11,8 +12,12 @@ import {
     ContentObjectItem,
     ContentObjectProcessingPriority,
     ContentSource,
+    ContentObjectTextResponse,
     CreateContentObjectPayload,
+    DeleteContentObjectResult,
     Embedding,
+    ObjectSearchResponse,
+    ComputedFacetResponse,
     ExportPropertiesPayload,
     ExportPropertiesResponse,
     FindPayload,
@@ -25,6 +30,7 @@ import {
     ListWorkflowRunsResponse,
     ObjectSearchPayload,
     ObjectSearchQuery,
+    SetObjectEmbeddingsResponse,
     SupportedEmbeddingTypes,
 } from "@vertesia/common";
 
@@ -36,24 +42,6 @@ import { StreamSource } from "../StreamSource.js";
 import { AnalyzeDocApi } from "./AnalyzeDocApi.js";
 import { ZenoClient } from "./client.js";
 
-export interface ComputeFacetsResponse {
-    type?: { _id: string; count: number }[];
-    location?: { _id: string; count: number }[];
-    status?: { _id: string; count: number }[];
-    [key: string]:
-        | { _id: string; count: number }[]
-        | number
-        | undefined;
-    total?: number;
-}
-
-export interface SearchResponse {
-    results: ContentObjectItem[];
-    facets: ComputeFacetsResponse;
-    /** Raw ES aggregation results. Only present when aggs were requested and ES backend was used. */
-    aggregations?: Record<string, unknown>;
-}
-
 export class ObjectsApi extends ApiTopic {
     declare client: ZenoClient;
 
@@ -112,7 +100,7 @@ export class ObjectsApi extends ApiTopic {
 
     computeFacets(
         query: ComputeObjectFacetPayload,
-    ): Promise<ComputeFacetsResponse> {
+    ): Promise<ComputedFacetResponse> {
         return this.post("/facets", {
             payload: query,
         });
@@ -137,7 +125,7 @@ export class ObjectsApi extends ApiTopic {
     }
 
     /** Search object — different from find because allow full text search */
-    search(payload: ComplexSearchPayload): Promise<SearchResponse> {
+    search(payload: ComplexSearchPayload): Promise<ObjectSearchResponse> {
         return this.post("/search", {
             payload,
         });
@@ -151,7 +139,7 @@ export class ObjectsApi extends ApiTopic {
         });
     }
 
-    getObjectText(id: string): Promise<{ text: string }> {
+    getObjectText(id: string): Promise<ContentObjectTextResponse> {
         return this.get(`/${id}/text`);
     }
 
@@ -367,13 +355,13 @@ export class ObjectsApi extends ApiTopic {
      * @param id The ID of the object
      * @returns Array of collections containing this object (both static and dynamic)
      */
-    getCollections(id: string): Promise<any[]> {
+    getCollections(id: string): Promise<Collection[]> {
         return this.get(`/${id}/collections`);
     }
 
-    delete(id: string): Promise<{ id: string }>;
+    delete(id: string): Promise<DeleteContentObjectResult>;
     delete(ids: string[]): Promise<BulkObjectDeleteResult>;
-    delete(idOrIds: string | string[]): Promise<{ id: string } | BulkObjectDeleteResult> {
+    delete(idOrIds: string | string[]): Promise<DeleteContentObjectResult | BulkObjectDeleteResult> {
         if (Array.isArray(idOrIds)) {
             return this.client.runOperation({
                 name: 'delete',
@@ -469,7 +457,7 @@ export class ObjectsApi extends ApiTopic {
         id: string,
         type: SupportedEmbeddingTypes,
         payload: Embedding,
-    ): Promise<Record<SupportedEmbeddingTypes, Embedding>> {
+    ): Promise<SetObjectEmbeddingsResponse> {
         return this.put(`/${id}/embeddings/${type}`, {
             payload,
         });

package/src/store/WorkflowsApi.ts

@@ -19,6 +19,7 @@ import {
     toAgentMessage,
     PromptSizeAnalyticsResponse,
     RunsByAgentAnalyticsResponse,
+    SignalAgentResponse,
     TimeToFirstResponseAnalyticsResponse,
     TokenUsageAnalyticsResponse,
     ToolAnalyticsResponse,
@@ -31,11 +32,16 @@ import {
     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";
@@ -64,7 +70,7 @@ export class WorkflowsApi extends ApiTopic {
         return this.post(`/runs`, { payload: payload });
     }
 
-    sendSignal(workflowId: string, runId: string, signal: string, payload?: any): Promise<{ message: string }> {
+    sendSignal(workflowId: string, runId: string, signal: string, payload?: any): Promise<SignalAgentResponse> {
         return this.post(`/runs/${workflowId}/${runId}/signal/${signal}`, { payload });
     }
 
@@ -95,12 +101,12 @@ export class WorkflowsApi extends ApiTopic {
         return this.get(`/runs/${workflowId}/${runId}/interaction`);
     }
 
-    terminate(workflowId: string, runId: string, reason?: string): Promise<{ message: string }> {
+    terminate(workflowId: string, runId: string, reason?: string): Promise<WorkflowActionResponse> {
         const payload: WorkflowActionPayload = { reason };
         return this.post(`/runs/${workflowId}/${runId}/actions/terminate`, { payload });
     }
 
-    cancel(workflowId: string, runId: string, reason?: string): Promise<{ message: string }> {
+    cancel(workflowId: string, runId: string, reason?: string): Promise<WorkflowActionResponse> {
         const payload: WorkflowActionPayload = { reason };
         return this.post(`/runs/${workflowId}/${runId}/actions/cancel`, { payload });
     }
@@ -111,20 +117,20 @@ export class WorkflowsApi extends ApiTopic {
      * @param workflowId The workflow ID
      * @param runId The run ID
      * @param queryName The name of the query to execute (e.g., "BatchAgentProgress")
-     * @returns The query result
+     * @returns The workflow query result as a JSON value.
      */
-    query<T = unknown>(workflowId: string, runId: string, queryName: string): Promise<T> {
+    query(workflowId: string, runId: string, queryName: string): Promise<WorkflowQueryResult> {
         return this.get(`/runs/${workflowId}/${runId}/query/${queryName}`);
     }
 
     execute(
         name: string,
         payload: ExecuteWorkflowPayload = {},
-    ): Promise<({ run_id: string; workflow_id: string } | undefined)[]> {
+    ): Promise<(WorkflowExecutionStartResult | undefined)[]> {
         return this.post(`/execute/${name}`, { payload });
     }
 
-    postMessage(runId: string, msg: AgentMessage): Promise<void> {
+    postMessage(runId: string, msg: AgentMessage): Promise<WorkflowUpdatePublishResponse> {
         if (!runId) {
             throw new Error("runId is required");
         }
@@ -138,7 +144,7 @@ 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 { messages: CompactMessage[] };
+        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));
     }
@@ -152,13 +158,21 @@ export class WorkflowsApi extends ApiTopic {
      * This approach provides better performance for conversations with large historical messages
      * since HTTP responses are compressed while SSE streams cannot be compressed.
      */
-    async streamMessages(workflowId: string, 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,
+        signal?: AbortSignal,
+    ): Promise<unknown> {
         return new Promise<unknown>((resolve, reject) => {
             let reconnectAttempts = 0;
             let lastMessageTimestamp = since || 0;
             let isClosed = false;
             let currentSse: EventSource | null = null;
             let interval: ReturnType<typeof setInterval> | null = null;
+            let reconnectTimer: ReturnType<typeof setTimeout> | null = null;
+            let abortHandler: (() => void) | null = null;
 
             const maxReconnectAttempts = 10;
             const baseDelay = 1000; // 1 second base delay
@@ -172,6 +186,10 @@ export class WorkflowsApi extends ApiTopic {
             };
 
             const cleanup = () => {
+                if (reconnectTimer) {
+                    clearTimeout(reconnectTimer);
+                    reconnectTimer = null;
+                }
                 if (interval) {
                     clearInterval(interval);
                     interval = null;
@@ -180,6 +198,10 @@ export class WorkflowsApi extends ApiTopic {
                     currentSse.close();
                     currentSse = null;
                 }
+                if (signal && abortHandler) {
+                    signal.removeEventListener('abort', abortHandler);
+                    abortHandler = null;
+                }
             };
 
             const exit = (payload: unknown) => {
@@ -190,6 +212,15 @@ export class WorkflowsApi extends ApiTopic {
                 }
             };
 
+            if (signal) {
+                if (signal.aborted) {
+                    exit(null);
+                    return;
+                }
+                abortHandler = () => exit(null);
+                signal.addEventListener('abort', abortHandler, { once: true });
+            }
+
             // 2. Connect to SSE for real-time updates only (skipHistory=true)
             const setupStream = async (isReconnect: boolean = false) => {
                 if (isClosed) return;
@@ -208,7 +239,10 @@ export class WorkflowsApi extends ApiTopic {
                     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"));
                         return;
                     }
@@ -291,7 +325,8 @@ export class WorkflowsApi extends ApiTopic {
                             console.log(`Attempting to reconnect in ${delay}ms (attempt ${reconnectAttempts + 1}/${maxReconnectAttempts})`);
 
                             reconnectAttempts++;
-                            setTimeout(() => {
+                            reconnectTimer = setTimeout(() => {
+                                reconnectTimer = null;
                                 if (!isClosed) {
                                     setupStream(true);
                                 }
@@ -307,7 +342,8 @@ export class WorkflowsApi extends ApiTopic {
                     if (reconnectAttempts < maxReconnectAttempts) {
                         const delay = calculateBackoffDelay(reconnectAttempts);
                         reconnectAttempts++;
-                        setTimeout(() => {
+                        reconnectTimer = setTimeout(() => {
+                            reconnectTimer = null;
                             if (!isClosed) {
                                 setupStream(true);
                             }
@@ -332,7 +368,7 @@ export class WorkflowsApi extends ApiTopic {
                         const streamIsOver = msg.type === AgentMessageType.TERMINATED ||
                             (msg.type === AgentMessageType.COMPLETE && workstreamId === 'main');
                         if (streamIsOver) {
-                            resolve(null);
+                            exit(null);
                             return;
                         }
                     }

package/src/store/client.ts

@@ -1,5 +1,5 @@
 import { AbstractFetchClient, RequestError } from "@vertesia/api-fetch-client";
-import { BulkOperationPayload, BulkOperationResult } from "@vertesia/common";
+import { BulkOperationPayload, BulkOperationResponse } from "@vertesia/common";
 import { AgentsApi } from "./AgentsApi.js";
 import { CollectionsApi } from "./CollectionsApi.js";
 import { CostApi } from "./CostApi.js";
@@ -72,7 +72,7 @@ export class ZenoClient extends AbstractFetchClient<ZenoClient> {
         );
     }
 
-    runOperation(payload: BulkOperationPayload): Promise<BulkOperationResult> {
+    runOperation(payload: BulkOperationPayload): Promise<BulkOperationResponse> {
         return this.post("/api/v1/operations", {
             payload
         });