@vertesia/client 1.5.0 → 1.5.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vertesia/client",
-  "version": "1.5.0",
+  "version": "1.5.1",
   "type": "module",
   "main": "./lib/index.js",
   "types": "./lib/index.d.ts",
@@ -19,9 +19,9 @@
   },
   "dependencies": {
     "eventsource": "^5.0.0",
-    "@llumiverse/common": "1.5.0",
-    "@vertesia/api-fetch-client": "1.5.0",
-    "@vertesia/common": "1.5.0"
+    "@llumiverse/common": "1.5.1",
+    "@vertesia/api-fetch-client": "1.5.1",
+    "@vertesia/common": "1.5.1"
   },
   "exports": {
     ".": {
@@ -48,7 +48,7 @@
     "ai",
     "typescript"
   ],
-  "gitHead": "3485742020066ab1fd3292487cf3a9577b5e4d91",
+  "gitHead": "7ab3acd95af12b8c254e1b3852e67d5078790f98",
   "scripts": {
     "lint": "biome lint src",
     "clean:lib": "rimraf ./lib ./tsconfig.tsbuildinfo",

package/src/EnvironmentsApi.ts

@@ -1,6 +1,11 @@
 import type { AIModel, ModelSearchPayload } from '@llumiverse/common';
 import { ApiTopic, type ClientBase } from '@vertesia/api-fetch-client';
 import type {
+    EmbeddingBatchCapabilityRequest,
+    EmbeddingBatchCapabilityResponse,
+    EmbeddingBatchCreateRequest,
+    EmbeddingBatchJobRequest,
+    EmbeddingBatchJobResponse,
     EmbeddingsApiRequest,
     EmbeddingsApiResult,
     EnableEnvironmentModelPayload,
@@ -84,6 +89,29 @@ export default class EnvironmentsApi extends ApiTopic {
         });
     }
 
+    embeddingBatchCapability(
+        id: string,
+        payload: EmbeddingBatchCapabilityRequest,
+    ): Promise<EmbeddingBatchCapabilityResponse> {
+        return this.post(`/${id}/embedding-batches/capability`, { payload });
+    }
+
+    createEmbeddingBatch(id: string, payload: EmbeddingBatchCreateRequest): Promise<EmbeddingBatchJobResponse> {
+        return this.post(`/${id}/embedding-batches`, { payload });
+    }
+
+    getEmbeddingBatch(id: string, payload: EmbeddingBatchJobRequest): Promise<EmbeddingBatchJobResponse> {
+        return this.post(`/${id}/embedding-batches/get`, { payload });
+    }
+
+    cancelEmbeddingBatch(id: string, payload: EmbeddingBatchJobRequest): Promise<EmbeddingBatchJobResponse> {
+        return this.post(`/${id}/embedding-batches/cancel`, { payload });
+    }
+
+    deleteEmbeddingBatch(id: string, payload: EmbeddingBatchJobRequest): Promise<EmbeddingBatchJobResponse> {
+        return this.post(`/${id}/embedding-batches/delete`, { payload });
+    }
+
     /**
      * Batch update the LLM Environment for multiple Interactions, including the model used and the
      * Environment that they belong to.

package/src/InteractionsApi.ts

@@ -30,6 +30,7 @@ import type {
     RateLimitRequestPayload,
     RateLimitRequestResponse,
     ResolvedInteractionExecutionInfo,
+    ResolveInteractionQuery,
 } from '@vertesia/common';
 import type { VertesiaClient } from './client.js';
 import { checkRateLimit, executeInteraction, executeInteractionAsync, executeInteractionByName } from './execute.js';
@@ -339,15 +340,7 @@ export default class InteractionsApi extends ApiTopic {
      * @param options Optional environment and/or model to resolve with
      * @returns ResolvedInteractionExecutionInfo with the resolved environment and model
      */
-    resolve(
-        nameOrId: string,
-        options?: {
-            environment?: string;
-            model?: string;
-            hasImage?: boolean;
-            hasVideo?: boolean;
-        },
-    ): Promise<ResolvedInteractionExecutionInfo> {
+    resolve(nameOrId: string, options?: ResolveInteractionQuery): Promise<ResolvedInteractionExecutionInfo> {
         return this.get(`/resolve/${encodeURIComponent(nameOrId)}`, {
             query: options,
         });

package/src/InternalAccountsApi.ts

@@ -0,0 +1,12 @@
+import { ApiTopic, type ClientBase } from '@vertesia/api-fetch-client';
+import type { AccountApiVersionPolicy } from '@vertesia/common';
+
+export default class InternalAccountsApi extends ApiTopic {
+    constructor(parent: ClientBase) {
+        super(parent, '/internal/accounts');
+    }
+
+    getApiVersionPolicy(accountId: string): Promise<AccountApiVersionPolicy> {
+        return this.get(`/${encodeURIComponent(accountId)}/api-version-policy`);
+    }
+}

package/src/index.ts

@@ -6,6 +6,7 @@ export * from './client.js';
 export type { GroupsQueryOptions } from './GroupsApi.js';
 export * from './InteractionBase.js';
 export * from './InteractionOutput.js';
+export { default as InternalAccountsApi } from './InternalAccountsApi.js';
 export { default as InternalSecretsApi } from './InternalSecretsApi.js';
 export * from './managed-sse.js';
 export { default as OAuthClientsApi } from './OAuthClientsApi.js';

package/src/store/AgentsApi.stream.test.ts

@@ -72,7 +72,16 @@ function createHarness(): Harness {
             }
             const since = Number(url.searchParams.get('since') ?? 0);
             pollSince.push(since);
-            return Response.json({ messages: pollResponse(since) });
+            try {
+                return Response.json({ messages: pollResponse(since) });
+            } catch (err) {
+                // A thrown error carrying `status` models an HTTP error response (e.g. 404).
+                const status = (err as { status?: unknown }).status;
+                if (typeof status === 'number') {
+                    return Response.json({ error: (err as Error).message }, { status });
+                }
+                throw err;
+            }
         }
         if (url.pathname.endsWith(`/${AGENT_RUN_ID}`)) {
             return Response.json({ id: AGENT_RUN_ID, status: runStatus });
@@ -337,6 +346,29 @@ describe('AgentsApi.streamMessages reconnection', () => {
         await settle();
     });
 
+    it('stops the polling fallback when the run no longer exists (404)', async () => {
+        const harness = createHarness();
+        harness.setHistory([{ t: AgentMessageType.UPDATE, m: 'history', ts: 1000 }]);
+        harness.setPollResponse(() => {
+            throw Object.assign(new Error('Not Found: Agent run not found'), { status: 404 });
+        });
+
+        const abort = new AbortController();
+        const done = harness.client.agents.streamMessages(AGENT_RUN_ID, undefined, undefined, abort.signal);
+        await settle();
+
+        for (let i = 0; i < 11; i++) {
+            await failCurrentConnection();
+        }
+        await settle(6000);
+
+        await expect(done).resolves.toBeNull();
+        expect(warnings.some((line) => line.includes('the run no longer exists (404)'))).toBe(true);
+        // A single 404 ends the loop: no retry warning, no further poll after the exit.
+        expect(warnings.filter((line) => line.includes('GET /updates failed while polling'))).toEqual([]);
+        expect(harness.pollSince).toHaveLength(1);
+    });
+
     it('closes the stream when polling observes a terminal run status', async () => {
         const harness = createHarness();
         const abort = new AbortController();

package/src/store/AgentsApi.ts

@@ -417,6 +417,15 @@ export class AgentsApi extends ApiTopic {
             );
         };
 
+        // A 404 while polling means the run itself is gone (deleted, or never persisted): nothing
+        // will ever arrive, so stop instead of polling the missing resource every 5s forever.
+        const isRunGone = (error: unknown): boolean =>
+            typeof error === 'object' && error !== null && (error as { status?: unknown }).status === 404;
+        const exitBecauseRunGone = () => {
+            console.warn(`Agent stream ${id}: the run no longer exists (404); stopping the polling fallback.`);
+            exit(null);
+        };
+
         const pollTick = async () => {
             if (isClosed || !isPolling) return;
             let polledMessages = false;
@@ -438,6 +447,10 @@ export class AgentsApi extends ApiTopic {
                     }
                 }
             } catch (err) {
+                if (isRunGone(err)) {
+                    exitBecauseRunGone();
+                    return;
+                }
                 warnPollFailure('GET /updates', err);
             }
             if (isClosed || !isPolling) return;
@@ -450,6 +463,10 @@ export class AgentsApi extends ApiTopic {
                 }
                 if (polledMessages) consecutivePollFailures = 0;
             } catch (err) {
+                if (isRunGone(err)) {
+                    exitBecauseRunGone();
+                    return;
+                }
                 warnPollFailure('run status check', err);
             }
             if (isClosed || !isPolling) return;

package/src/store/EmbeddingsApi.test.ts

@@ -0,0 +1,31 @@
+import { SupportedEmbeddingTypes } from '@vertesia/common';
+import { describe, expect, it, vi } from 'vitest';
+import { ZenoClient } from './client.js';
+
+describe('EmbeddingsApi', () => {
+    it('omits the mode by default and sends the explicit sync override when requested', async () => {
+        const requests: Request[] = [];
+        const client = new ZenoClient({
+            serverUrl: 'https://store.example.com',
+            apikey: 'token',
+            fetch: vi.fn(
+                async () =>
+                    new Response(JSON.stringify({ status: 'success', message: 'started' }), {
+                        status: 200,
+                        headers: { 'content-type': 'application/json' },
+                    }),
+            ),
+            onRequest: (request) => requests.push(request),
+        });
+
+        await client.embeddings.recalculate(SupportedEmbeddingTypes.text);
+        await client.embeddings.recalculate(SupportedEmbeddingTypes.text, { mode: 'sync' });
+        await client.embeddings.recalculate(SupportedEmbeddingTypes.image, { force: true });
+
+        expect(requests.map(({ url }) => url)).toEqual([
+            'https://store.example.com/api/v1/embeddings/text/recalculate',
+            'https://store.example.com/api/v1/embeddings/text/recalculate?mode=sync',
+            'https://store.example.com/api/v1/embeddings/image/recalculate?force=true',
+        ]);
+    });
+});

package/src/store/EmbeddingsApi.ts

@@ -1,8 +1,16 @@
 import { ApiTopic, type ClientBase } from '@vertesia/api-fetch-client';
 import type {
+    EmbeddingBatchApplyRequest,
+    EmbeddingBatchApplyResponse,
+    EmbeddingBatchPrepareRequest,
+    EmbeddingBatchPrepareResponse,
+    EmbeddingBatchRenditionPageRequest,
+    EmbeddingBatchRenditionPageResponse,
+    EmbeddingBatchUpdateRequest,
     EmbeddingsStatusResponse,
     GenericCommandResponse,
     ProjectConfigurationEmbeddingEnablePayload,
+    RecalculateEmbeddingsQuery,
     SupportedEmbeddingTypes,
 } from '@vertesia/common';
 
@@ -33,7 +41,44 @@ export class EmbeddingsApi extends ApiTopic {
         return this.post(`${type}/disable`);
     }
 
-    async recalculate(type: SupportedEmbeddingTypes): Promise<GenericCommandResponse> {
-        return this.post(`${type}/recalculate`);
+    async recalculate(
+        type: SupportedEmbeddingTypes,
+        query: RecalculateEmbeddingsQuery = {},
+    ): Promise<GenericCommandResponse> {
+        return this.post(`${type}/recalculate`, Object.keys(query).length ? { query } : undefined);
+    }
+
+    async prepareBatch(
+        payload: EmbeddingBatchPrepareRequest,
+        timeoutMs: number | false | null = false,
+    ): Promise<EmbeddingBatchPrepareResponse> {
+        return this.post('/batch/prepare', { payload, timeoutMs });
+    }
+
+    async updateBatch(payload: EmbeddingBatchUpdateRequest): Promise<GenericCommandResponse> {
+        return this.post('/batch/update', { payload });
+    }
+
+    async prepareBatchRenditionPage(
+        payload: EmbeddingBatchRenditionPageRequest,
+    ): Promise<EmbeddingBatchRenditionPageResponse> {
+        return this.post('/batch/renditions/prepare', { payload });
+    }
+
+    async pollBatchRenditionPage(
+        payload: EmbeddingBatchRenditionPageRequest,
+    ): Promise<EmbeddingBatchRenditionPageResponse> {
+        return this.post('/batch/renditions/status', { payload });
+    }
+
+    async cancelBatchRenditionPage(payload: EmbeddingBatchRenditionPageRequest): Promise<GenericCommandResponse> {
+        return this.post('/batch/renditions/cancel', { payload });
+    }
+
+    async applyBatch(
+        payload: EmbeddingBatchApplyRequest,
+        timeoutMs: number | false | null = false,
+    ): Promise<EmbeddingBatchApplyResponse> {
+        return this.post('/batch/apply', { payload, timeoutMs });
     }
 }