@vertesia/client 0.79.0 → 0.80.0-dev-20251118

package/src/InteractionResult.example.ts

@@ -0,0 +1,72 @@
+/**
+ * Example usage of InteractionOutput
+ * This file demonstrates the Proxy-based approach where the result acts as both
+ * an array and has convenience methods.
+ */
+
+import { CompletionResult } from '@llumiverse/common';
+import { InteractionOutput } from './InteractionOutput.js';
+
+// Sample data
+const sampleResults: CompletionResult[] = [
+    { type: 'text', value: 'Hello, ' },
+    { type: 'text', value: 'World!' },
+    { type: 'json', value: { name: 'Alice', age: 30 } },
+    { type: 'json', value: { title: 'Engineer', level: 'Senior' } },
+    { type: 'image', value: 'data:image/png;base64,iVBORw0K...' }
+];
+
+// Example 1: Using InteractionOutput.from() (Proxy approach - RECOMMENDED)
+console.log('=== Example 1: Proxy Approach ===');
+interface Person { name: string; age: number; }
+const output = InteractionOutput.from<Person>(sampleResults);
+
+// Works as an array
+console.log('Array length:', output.length);           // 5
+console.log('First item:', output[0]);                 // { type: 'text', value: 'Hello, ' }
+console.log('Types:', output.map(r => r.type));        // ['text', 'text', 'json', 'json', 'image']
+
+// Has convenience methods
+console.log('All text:', output.text);                 // 'Hello, World!'
+console.log('First object:', output.object());         // { name: 'Alice', age: 30 } (typed as Person)
+console.log('All objects:', output.objects());         // [{ name: 'Alice', age: 30 }, { title: 'Engineer', level: 'Senior' }]
+console.log('First image:', output.image);             // 'data:image/png;base64,iVBORw0K...'
+
+// Override type for specific objects
+interface Job { title: string; level: string; }
+console.log('Second object:', output.objectAt<Job>(1)); // { title: 'Engineer', level: 'Senior' } (typed as Job)
+
+// Example 2: Using InteractionOutput class directly
+console.log('\n=== Example 2: Class Approach ===');
+const wrapper = new InteractionOutput<Person>(sampleResults);
+
+// Access through wrapper properties (no array access)
+console.log('All text:', wrapper.text);
+console.log('Raw results:', wrapper.results);          // Original array
+console.log('First object:', wrapper.object());
+
+// Example 3: Type safety demonstration
+console.log('\n=== Example 3: Type Safety ===');
+
+interface Contract { title: string; parties: string[]; }
+const contractResults: CompletionResult[] = [
+    { type: 'json', value: { title: 'Sales Agreement', parties: ['Alice', 'Bob'] } }
+];
+
+const contractOutput = InteractionOutput.from<Contract>(contractResults);
+const contract = contractOutput.object(); // TypeScript knows this is Contract
+console.log('Contract title:', contract.title);
+console.log('Parties:', contract.parties.join(', '));
+
+// Example 4: Mixed content
+console.log('\n=== Example 4: Mixed Content ===');
+const mixedResults: CompletionResult[] = [
+    { type: 'text', value: 'Analysis complete. Results: ' },
+    { type: 'json', value: { score: 0.95, confidence: 'high' } },
+    { type: 'text', value: '\nThank you!' }
+];
+
+const mixed = InteractionOutput.from(mixedResults);
+console.log('Full text:', mixed.text);                 // 'Analysis complete. Results: \nThank you!'
+console.log('Analysis:', mixed.object());              // { score: 0.95, confidence: 'high' }
+console.log('Text parts:', mixed.texts);               // ['Analysis complete. Results: ', '\nThank you!']

package/src/InteractionsApi.ts

@@ -1,7 +1,17 @@
 import { ApiTopic, ClientBase, ServerError } from "@vertesia/api-fetch-client";
-import { AsyncExecutionPayload, ComputeInteractionFacetPayload, ExecutionRun, GenerateInteractionPayload, GenerateTestDataPayload, ImprovePromptPayload, Interaction, InteractionCreatePayload, InteractionEndpoint, InteractionEndpointQuery, InteractionExecutionPayload, InteractionExecutionResult, InteractionForkPayload, InteractionPublishPayload, InteractionRef, InteractionRefWithSchema, InteractionSearchPayload, InteractionSearchQuery, InteractionUpdatePayload, InteractionsExportPayload, RateLimitRequestPayload, RateLimitRequestResponse } from "@vertesia/common";
+import {
+    AsyncExecutionPayload, ComputeInteractionFacetPayload, GenerateInteractionPayload, GenerateTestDataPayload, ImprovePromptPayload,
+    ImprovePromptPayloadConfig,
+    Interaction, InteractionCreatePayload, InteractionEndpoint, InteractionEndpointQuery,
+    InteractionExecutionPayload, InteractionForkPayload,
+    InteractionPublishPayload, InteractionRef, InteractionRefWithSchema, InteractionSearchPayload, InteractionSearchQuery,
+    InteractionsExportPayload, InteractionUpdatePayload,
+    RateLimitRequestPayload, RateLimitRequestResponse
+} from "@vertesia/common";
 import { VertesiaClient } from "./client.js";
 import { checkRateLimit, executeInteraction, executeInteractionAsync, executeInteractionByName } from "./execute.js";
+import { InteractionCatalogApi } from "./InteractionCatalogApi.js";
+import { EnhancedExecutionRun, EnhancedInteractionExecutionResult, enhanceExecutionRun, enhanceInteractionExecutionResult } from "./InteractionOutput.js";
 
 export interface ComputeInteractionFacetsResponse {
     tags?: { _id: string, count: number }[];
@@ -14,8 +24,11 @@ export interface AsyncExecutionResult {
 }
 
 export default class InteractionsApi extends ApiTopic {
+    catalog: InteractionCatalogApi;
+
     constructor(parent: ClientBase) {
         super(parent, "/api/v1/interactions");
+        this.catalog = new InteractionCatalogApi(parent);
     }
 
     /**
@@ -132,15 +145,16 @@ export default class InteractionsApi extends ApiTopic {
      * @throws 500 if interaction execution fails
      * @throws 500 if interaction execution times out
      **/
-    execute<P = any>(id: string, payload: InteractionExecutionPayload = {},
-        onChunk?: (chunk: string) => void): Promise<ExecutionRun<P>> {
-        return executeInteraction<P>(this.client as VertesiaClient, id, payload, onChunk).catch(err => {
+    async execute<ResultT = any, ParamsT = any>(id: string, payload: InteractionExecutionPayload = {},
+        onChunk?: (chunk: string) => void): Promise<EnhancedInteractionExecutionResult<ResultT, ParamsT>> {
+        const r = await executeInteraction<ParamsT>(this.client as VertesiaClient, id, payload, onChunk).catch(err => {
             if (err instanceof ServerError && err.payload?.id) {
                 throw err.updateDetails({ run_id: err.payload.id });
             } else {
                 throw err;
             }
         });
+        return enhanceInteractionExecutionResult<ResultT, ParamsT>(r);
     }
 
     /**
@@ -158,15 +172,16 @@ export default class InteractionsApi extends ApiTopic {
      * @param onChunk
      * @returns
      */
-    executeByName<P = any>(nameWithTag: string, payload: InteractionExecutionPayload = {},
-        onChunk?: (chunk: string) => void): Promise<InteractionExecutionResult<P>> {
-        return executeInteractionByName<P>(this.client as VertesiaClient, nameWithTag, payload, onChunk).catch(err => {
+    async executeByName<ResultT = any, ParamsT = any>(nameWithTag: string, payload: InteractionExecutionPayload = {},
+        onChunk?: (chunk: string) => void): Promise<EnhancedInteractionExecutionResult<ResultT, ParamsT>> {
+        const r = await executeInteractionByName<ParamsT>(this.client as VertesiaClient, nameWithTag, payload, onChunk).catch(err => {
             if (err instanceof ServerError && err.payload?.id) {
                 throw err.updateDetails({ run_id: err.payload.id });
             } else {
                 throw err;
             }
         });
+        return enhanceInteractionExecutionResult<ResultT, ParamsT>(r);
     }
 
     /**
@@ -213,13 +228,23 @@ export default class InteractionsApi extends ApiTopic {
 
     /**
      * Suggest Improvement for a prompt
+     * @deprecated use suggestPromptImprovements instead
      */
-    suggestImprovements(id: string, payload: ImprovePromptPayload): Promise<{ result: string; }> {
-        return this.post(`${id}/suggest-prompt-improvements`, {
+    async suggestImprovements<ResultT = any, ParamsT = any>(id: string, payload: ImprovePromptPayloadConfig): Promise<EnhancedExecutionRun<ResultT, ParamsT>> {
+        const r = await this.post(`${id}/suggest-prompt-improvements`, {
+            payload
+        });
+        return enhanceExecutionRun<ResultT, ParamsT>(r);
+    }
+
+    async suggestPromptImprovements<ResultT = any, ParamsT = any>(payload: ImprovePromptPayload): Promise<EnhancedInteractionExecutionResult<ResultT, ParamsT>> {
+        const r = await this.post(`/improve`, {
             payload
         });
+        return enhanceInteractionExecutionResult<ResultT, ParamsT>(r);
     }
 
+
     /**
      * List the versions of the interaction. Returns an empty array if no versions are found
      * @param id

package/src/ProjectsApi.ts

@@ -1,5 +1,5 @@
 import { ApiTopic, ClientBase } from "@vertesia/api-fetch-client";
-import { AwsConfiguration, GithubConfiguration, GladiaConfiguration, ICreateProjectPayload, MagicPdfConfiguration, Project, ProjectIntegrationListEntry, ProjectRef, SupportedIntegrations } from "@vertesia/common";
+import { AwsConfiguration, GithubConfiguration, GladiaConfiguration, ICreateProjectPayload, MagicPdfConfiguration, Project, ProjectConfiguration, ProjectIntegrationListEntry, ProjectRef, SupportedIntegrations } from "@vertesia/common";
 
 export default class ProjectsApi extends ApiTopic {
     constructor(parent: ClientBase) {
@@ -26,6 +26,12 @@ export default class ProjectsApi extends ApiTopic {
         });
     }
 
+    updateConfiguration(projectId: string, payload: Partial<ProjectConfiguration>): Promise<ProjectConfiguration> {
+        return this.put(`/${projectId}/configuration`, {
+            payload
+        });
+    }
+
     integrations: IntegrationsConfigurationApi = new IntegrationsConfigurationApi(this);
 
 }

package/src/RunsApi.ts

@@ -1,3 +1,4 @@
+import type { ExecutionResponse } from "@llumiverse/common";
 import { ApiTopic, ClientBase } from "@vertesia/api-fetch-client";
 import {
     CheckpointConversationPayload,
@@ -5,6 +6,7 @@ import {
     ExecutionRun,
     ExecutionRunRef,
     FindPayload,
+    PopulatedExecutionRun,
     RunCreatePayload,
     RunListingFilters,
     RunListingQueryOptions,
@@ -13,7 +15,7 @@ import {
     UserMessagePayload,
 } from "@vertesia/common";
 import { VertesiaClient } from "./client.js";
-import type { ExecutionResponse } from "@llumiverse/common";
+import { EnhancedExecutionRun, enhanceExecutionRun } from "./InteractionOutput.js";
 
 export interface FilterOption {
     id: string;
@@ -63,8 +65,15 @@ export class RunsApi extends ApiTopic {
      * @param id
      * @returns InteractionResult
      **/
-    retrieve<P = any>(id: string): Promise<ExecutionRun<P>> {
-        return this.get("/" + id);
+    async retrieve<ResultT = any, ParamsT = any>(id: string): Promise<EnhancedExecutionRun<ResultT, ParamsT>> {
+        const r = await this.get("/" + id);
+        return enhanceExecutionRun<ResultT, ParamsT>(r);
+    }
+
+    retrievePopulated<P = any>(id: string): Promise<PopulatedExecutionRun<P>> {
+        return this.get("/" + id, {
+            query: { populate: "true" },
+        });
     }
 
     /**
@@ -78,7 +87,7 @@ export class RunsApi extends ApiTopic {
         return this.get(`/filter-options/${field}`, { query });
     }
 
-    create(payload: RunCreatePayload): Promise<ExecutionRun> {
+    async create<ResultT = any, ParamsT = any>(payload: RunCreatePayload): Promise<EnhancedExecutionRun<ResultT, ParamsT>> {
         const sessionTags = (this.client as VertesiaClient).sessionTags;
         if (sessionTags) {
             let tags = Array.isArray(sessionTags) ? sessionTags : [sessionTags];
@@ -89,9 +98,10 @@ export class RunsApi extends ApiTopic {
             }
             payload = { ...payload, tags };
         }
-        return this.post("/", {
+        const r = await this.post("/", {
             payload,
         });
+        return enhanceExecutionRun<ResultT, ParamsT>(r);
     }
 
     /**

package/src/client.test.ts

@@ -6,6 +6,7 @@ describe('Test Vertesia Client', () => {
         const client = new VertesiaClient({
             serverUrl: 'https://api.vertesia.io',
             storeUrl: 'https://api.vertesia.io',
+            tokenServerUrl: 'https://sts.vertesia.io',
             apikey: '1234',
         });
         expect(client).toBeDefined();
@@ -73,6 +74,7 @@ describe('Test Vertesia Client', () => {
         const client = new VertesiaClient({
             serverUrl: 'http://localhost:8091',
             storeUrl: 'http://localhost:8092',
+            tokenServerUrl: 'http://localhost:8093',
         });
 
         expect(client).toBeDefined();

package/src/client.ts

@@ -37,18 +37,21 @@ export type VertesiaClientProps = {
      * @default api.vertesia.io
      * @since 0.52.0
      */
-    site?: 'api.vertesia.io' | 'api-preview.vertesia.io' | 'api-staging.vertesia.io';
+    site?:
+        | "api.vertesia.io"
+        | "api-preview.vertesia.io"
+        | "api-staging.vertesia.io";
     serverUrl?: string;
     storeUrl?: string;
+    tokenServerUrl?: string;
     apikey?: string;
     projectId?: string;
     sessionTags?: string | string[];
     onRequest?: (request: Request) => void;
     onResponse?: (response: Response) => void;
-}
+};
 
 export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
-
     /**
      * The JWT token linked to the API KEY (sk or pk)
      */
@@ -64,11 +67,15 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
      */
     sessionTags?: string | string[];
 
+    /**
+     * tokenServerUrl
+     */
+    tokenServerUrl: string;
 
     /**
-     * Create a client from the given token. 
+     * Create a client from the given token.
      * If you already have the decoded token you can pass it as the second argument to avoid decodinf it again.
-     * 
+     *
      * @param token the raw JWT token
      * @param payload the decoded JWT token as an AuthTokenPayload - optional
      */
@@ -76,21 +83,21 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
         if (!payload) {
             payload = decodeJWT(token);
         }
-        const endpoints = decodeEndpoints(payload!.endpoints);
+
+        const endpoints = decodeEndpoints(payload.endpoints);
         return await new VertesiaClient({
             serverUrl: endpoints.studio,
-            storeUrl: endpoints.store
+            storeUrl: endpoints.store,
+            tokenServerUrl: payload.iss,
         }).withApiKey(token);
     }
 
-    static decodeEndpoints() {
-
-    }
+    static decodeEndpoints() {}
 
     constructor(
         opts: VertesiaClientProps = {
-            site: 'api.vertesia.io',
-        }
+            site: "api.vertesia.io",
+        },
     ) {
         let studioServerUrl: string;
         let zenoServerUrl: string;
@@ -100,7 +107,9 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
         } else if (opts.site) {
             studioServerUrl = `https://${opts.site}`;
         } else {
-            throw new Error("Parameter 'site' or 'serverUrl' is required for VertesiaClient");
+            throw new Error(
+                "Parameter 'site' or 'serverUrl' is required for VertesiaClient",
+            );
         }
 
         if (opts.storeUrl) {
@@ -108,25 +117,67 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
         } else if (opts.site) {
             zenoServerUrl = `https://${opts.site}`;
         } else {
-            throw new Error("Parameter 'site' or 'storeUrl' is required for VertesiaClient");
+            throw new Error(
+                "Parameter 'site' or 'storeUrl' is required for VertesiaClient",
+            );
         }
 
         super(studioServerUrl);
 
+        if (opts.tokenServerUrl) {
+            this.tokenServerUrl = opts.tokenServerUrl;
+        } else if (opts.site) {
+            this.tokenServerUrl = `https://${opts.site.replace(/^api/, "sts")}`;
+        } else if (opts.serverUrl || opts.storeUrl) {
+            // Determine STS URL based on environment in serverUrl or storeUrl
+            const urlToCheck = opts.serverUrl || opts.storeUrl || "";
+            try {
+                const url = new URL(urlToCheck);
+                // Check for environment patterns
+                if (url.hostname.includes("-production.")) {
+                    // zeno-server-production.api.vertesia.io -> sts.vertesia.io
+                    this.tokenServerUrl = "https://sts.vertesia.io";
+                } else if (url.hostname.includes("-preview.")) {
+                    // zeno-server-preview.api.vertesia.io -> sts-preview.vertesia.io
+                    this.tokenServerUrl = "https://sts-preview.vertesia.io";
+                } else if (url.hostname === "api.vertesia.io") {
+                    // api.vertesia.io -> sts.vertesia.io
+                    this.tokenServerUrl = "https://sts.vertesia.io";
+                } else if (url.hostname === "api-preview.vertesia.io") {
+                    // api-preview.vertesia.io -> sts-preview.vertesia.io
+                    this.tokenServerUrl = "https://sts-preview.vertesia.io";
+                } else if (url.hostname === "api-staging.vertesia.io") {
+                    // api-staging.vertesia.io -> sts-staging.vertesia.io
+                    this.tokenServerUrl = "https://sts-staging.vertesia.io";
+                } else if (url.hostname.startsWith("api")) {
+                    // Generic api.* pattern replacement
+                    url.hostname = url.hostname.replace(/^api/, "sts");
+                    this.tokenServerUrl = url.toString();
+                } else {
+                    // Default to staging for everything else
+                    this.tokenServerUrl = "https://sts-staging.vertesia.io";
+                }
+            } catch (e) {
+                // Default to staging if URL parsing fails
+                this.tokenServerUrl = "https://sts-staging.vertesia.io";
+            }
+        } else {
+            // Default to staging if no URL provided
+            this.tokenServerUrl = "https://sts-staging.vertesia.io";
+        }
+
         this.store = new ZenoClient({
             serverUrl: zenoServerUrl,
+            tokenServerUrl: this.tokenServerUrl,
             apikey: opts.apikey,
             onRequest: opts.onRequest,
-            onResponse: opts.onResponse
+            onResponse: opts.onResponse,
         });
 
         if (opts.apikey) {
             this.withApiKey(opts.apikey);
         }
-        //TODO: this is no more used, remove in next major version
-        if (opts.projectId) {
-            this.headers["x-project-id"] = opts.projectId;
-        }
+
         this.onRequest = opts.onRequest;
         this.onResponse = opts.onResponse;
         this.sessionTags = opts.sessionTags;
@@ -153,25 +204,28 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
 
     async withApiKey(apiKey: string | null) {
         return this.withAuthCallback(
-            apiKey ? async () => {
-                if (!isApiKey(apiKey)) {
-                    return `Bearer ${apiKey}`
-                }
-
-                if (isTokenExpired(this._jwt)) {
-                    const jwt = await this.getAuthToken(apiKey);
-                    this._jwt = jwt.token;
-                }
-                return `Bearer ${this._jwt}`
-            } : undefined
+            apiKey
+                ? async () => {
+                      if (!isApiKey(apiKey)) {
+                          return `Bearer ${apiKey}`;
+                      }
+
+                      if (isTokenExpired(this._jwt)) {
+                          const jwt = await this.getAuthToken(apiKey);
+                          this._jwt = jwt.token;
+                      }
+                      return `Bearer ${this._jwt}`;
+                  }
+                : undefined,
         );
     }
 
     async getRawJWT() {
         if (!this._jwt && this._auth) {
             const auth = await this._auth();
-            if (!this._jwt) { // the _jwt may be set by the auth callback
-                this._jwt = auth.trim().split(' ')[1]; // remove Bearer prefix
+            if (!this._jwt) {
+                // the _jwt may be set by the auth callback
+                this._jwt = auth.trim().split(" ")[1]; // remove Bearer prefix
             }
         }
         return this._jwt || null;
@@ -221,21 +275,29 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
         return this.store.baseUrl;
     }
 
-
     /**
      *
-     * Generate a token for use with other Composable's services
+     * Generate a token for use with other Vertesia's services
      *
-     * @param accountId: selected account to generate the token for
      * @returns AuthTokenResponse
      */
-    async getAuthToken(token?: string, accountId?: string): Promise<AuthTokenResponse> {
-        const query = {
-            accountId,
-            token
-        };
-
-        return this.get('/auth/token', { query: query, headers: { "authorization": undefined } as any });
+    async getAuthToken(token?: string): Promise<AuthTokenResponse> {
+        return fetch(`${this.tokenServerUrl}/token/issue`, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${token}`,
+            },
+        })
+            .then((response) => response.json())
+            .then((data) => data as AuthTokenResponse)
+            .catch((error) => {
+                console.error(
+                    `Error fetching token from ${this.tokenServerUrl}:`,
+                    { error },
+                );
+                throw error;
+            });
     }
 
     get initialHeaders() {
@@ -263,7 +325,7 @@ export class VertesiaClient extends AbstractFetchClient<VertesiaClient> {
 }
 
 function isApiKey(apiKey: string) {
-    return (apiKey.startsWith('pk-') || apiKey.startsWith('sk-'));
+    return apiKey.startsWith("pk-") || apiKey.startsWith("sk-");
 }
 
 function isTokenExpired(token: string | null) {
@@ -274,38 +336,45 @@ function isTokenExpired(token: string | null) {
     const decoded = decodeJWT(token);
     const exp = decoded.exp;
     const currentTime = Date.now();
-    return (currentTime <= exp * 1000 - EXPIRATION_THRESHOLD);
+    return currentTime <= exp * 1000 - EXPIRATION_THRESHOLD;
 }
 
 export function decodeJWT(jwt: string): AuthTokenPayload {
-    const payloadBase64 = jwt.split('.')[1];
+    const payloadBase64 = jwt.split(".")[1];
     const decodedJson = base64UrlDecode(payloadBase64);
-    return JSON.parse(decodedJson)
+    return JSON.parse(decodedJson);
 }
 
 function base64UrlDecode(input: string): string {
     // Convert base64url to base64
-    const base64 = input.replace(/-/g, '+').replace(/_/g, '/')
+    const base64 = input
+        .replace(/-/g, "+")
+        .replace(/_/g, "/")
         // Pad with '=' to make length a multiple of 4
-        .padEnd(Math.ceil(input.length / 4) * 4, '=');
+        .padEnd(Math.ceil(input.length / 4) * 4, "=");
 
-    if (typeof Buffer !== 'undefined') {
+    if (typeof Buffer !== "undefined") {
         // Node.js
-        return Buffer.from(base64, 'base64').toString('utf-8');
-    } else if (typeof atob !== 'undefined' && typeof TextDecoder !== 'undefined') {
+        return Buffer.from(base64, "base64").toString("utf-8");
+    } else if (
+        typeof atob !== "undefined" &&
+        typeof TextDecoder !== "undefined"
+    ) {
         // Browser
         const binary = atob(base64);
-        const bytes = Uint8Array.from(binary, c => c.charCodeAt(0));
+        const bytes = Uint8Array.from(binary, (c) => c.charCodeAt(0));
         // decode to utf8
         return new TextDecoder().decode(bytes);
     } else {
-        throw new Error('No base64 decoder available');
+        throw new Error("No base64 decoder available");
     }
 }
 
-export function decodeEndpoints(endpoints: string | Record<string, string> | undefined): Record<string, string> {
+export function decodeEndpoints(
+    endpoints: string | Record<string, string> | undefined,
+): Record<string, string> {
     if (!endpoints) {
-        return getEndpointsFromDomain("api.vertesia.io")
+        return getEndpointsFromDomain("api.vertesia.io");
     }
     if (typeof endpoints === "string") {
         return getEndpointsFromDomain(endpoints);
@@ -319,12 +388,14 @@ function getEndpointsFromDomain(domain: string) {
         return {
             studio: `http://localhost:8091`,
             store: `http://localhost:8092`,
-        }
+            token: process.env.STS_URL ?? "https://sts-staging.vertesia.io",
+        };
     } else {
         const url = `https://${domain}`;
         return {
             studio: url,
             store: url,
-        }
+            token: url.replace("api", "sts"),
+        };
     }
 }

package/src/index.ts

@@ -1,8 +1,10 @@
 export * from './client.js';
 export * from './InteractionBase.js';
+export * from './InteractionOutput.js';
 export type { AsyncExecutionResult, ComputeInteractionFacetsResponse } from './InteractionsApi.js';
 export type { ComputePromptFacetsResponse, ListInteractionsResponse } from './PromptsApi.js';
 export type { ComputeRunFacetsResponse, FilterOption } from './RunsApi.js';
 export type { GroupsQueryOptions } from './GroupsApi.js';
 export * from "./store/index.js";
 export * from "./StreamSource.js";
+export type { OrphanedAppInstallation } from "./AppsApi.js";

package/src/store/CollectionsApi.ts

@@ -91,6 +91,30 @@ export class CollectionsApi extends ApiTopic {
         });
     }
 
+    addChildren(collectionId: string, children: string[]): Promise<{ count: number }> {
+        return this.post(`/${collectionId}/children`, {
+            payload: {
+                action: 'add',
+                children
+            }
+        });
+    }
+
+    deleteChildren(collectionId: string, children: string[]): Promise<{ count: number }> {
+        return this.post(`/${collectionId}/children`, {
+            payload: {
+                action: 'delete',
+                children
+            }
+        });
+    }
+
+    searchChildren(collectionId: string, query: ComplexCollectionSearchQuery = {}): Promise<CollectionItem[]> {
+        return this.post(`/${collectionId}/children/search`, {
+            payload: query
+        });
+    }
+
     delete(id: string) {
         return this.del(`/${id}`);
     }
@@ -104,7 +128,6 @@ export class CollectionsApi extends ApiTopic {
     updatePermissions(collectionId: string, permissions: Record<string, string[]>): Promise<{
         id: string;
         security: Record<string, string[]>;
-        objectsUpdated: number;
     }> {
         return this.put(`/${collectionId}/permissions`, {
             payload: permissions
@@ -121,9 +144,23 @@ export class CollectionsApi extends ApiTopic {
         id: string;
         message: string;
         security?: Record<string, string[]>;
-        objectsUpdated: number;
     }> {
         return this.post(`/${collectionId}/propagate-permissions`);
     }
 
+    /**
+     * Manually trigger shared properties propagation from collection to member objects
+     * Useful for debugging and fixing shared properties issues
+     * @param collectionId - The collection ID
+     * @returns Object with collection id, message, and number of objects updated
+     */
+    propagateSharedProperties(collectionId: string): Promise<{
+        id: string;
+        message: string;
+        shared_properties: string[]
+    }> {
+        return this.post(`/${collectionId}/propagate-shared-props`);
+    }
+
+
 }