@vertesia/client 0.80.0-dev.20251121 → 0.80.0

package/src/store/FilesApi.ts

@@ -3,16 +3,27 @@ import {
     GetFileUrlPayload,
     GetFileUrlResponse,
     GetUploadUrlPayload,
+    SetFileMetadataPayload,
 } from "@vertesia/common";
 import { StreamSource } from "../StreamSource.js";
 
 export const MEMORIES_PREFIX = "memories";
+export const ARTIFACTS_PREFIX = "agents";
 
 export function getMemoryFilePath(name: string) {
     const nameWithExt = name.endsWith(".tar.gz") ? name : name + ".tar.gz";
     return `${MEMORIES_PREFIX}/${nameWithExt}`;
 }
 
+/**
+ * Build the storage path for an agent artifact
+ * @param runId - The workflow run ID
+ * @param name - The artifact filename
+ */
+export function getAgentArtifactPath(runId: string, name: string): string {
+    return `${ARTIFACTS_PREFIX}/${runId}/${name}`;
+}
+
 export class FilesApi extends ApiTopic {
     constructor(parent: ClientBase) {
         super(parent, "/api/v1/files");
@@ -36,6 +47,7 @@ export class FilesApi extends ApiTopic {
         contentType: string;
         contentDisposition?: string;
         etag?: string;
+        customMetadata?: Record<string, string>;
     }> {
         return this.get("/metadata", {
             query: {
@@ -44,6 +56,17 @@ export class FilesApi extends ApiTopic {
         });
     }
 
+    /**
+     * Set custom metadata on a file
+     * @param file - The file path or URI
+     * @param metadata - Custom metadata key-value pairs
+     * @returns Success status
+     */
+    setFileMetadata(file: string, metadata: Record<string, string>): Promise<{ success: boolean; file: string }> {
+        const payload: SetFileMetadataPayload = { file, metadata };
+        return this.put("/metadata", { payload });
+    }
+
     /**
      * Get or create a bucket for the project. If the bucket already exists, it does nothing.
      * The bucket URI is returned.
@@ -179,4 +202,74 @@ export class FilesApi extends ApiTopic {
         }
         return stream;
     }
+
+    // ==================== Agent Artifact Methods ====================
+
+    /**
+     * List files by prefix
+     * @param prefix - Path prefix to filter files
+     * @returns Array of file paths matching the prefix
+     */
+    listByPrefix(prefix: string): Promise<{ files: string[] }> {
+        return this.get("/list", { query: { prefix } });
+    }
+
+    /**
+     * Upload an artifact for an agent run
+     * @param runId - The workflow run ID
+     * @param name - Artifact name (e.g., "output.json")
+     * @param source - File content source
+     * @returns The full path of the uploaded artifact
+     */
+    async uploadArtifact(runId: string, name: string, source: StreamSource | File): Promise<string> {
+        const artifactPath = getAgentArtifactPath(runId, name);
+        if (source instanceof File) {
+            const file = source as File;
+            return this.uploadFile(
+                new StreamSource(file.stream(), name, file.type, artifactPath)
+            );
+        } else {
+            return this.uploadFile(
+                new StreamSource(source.stream, name, source.type, artifactPath)
+            );
+        }
+    }
+
+    /**
+     * Download an artifact from an agent run
+     * @param runId - The workflow run ID
+     * @param name - Artifact name
+     * @returns ReadableStream of the artifact content
+     */
+    async downloadArtifact(runId: string, name: string): Promise<ReadableStream<Uint8Array>> {
+        const artifactPath = getAgentArtifactPath(runId, name);
+        return this.downloadFile(artifactPath);
+    }
+
+    /**
+     * Get download URL for an artifact
+     * @param runId - The workflow run ID
+     * @param name - Artifact name
+     * @param disposition - Content disposition (inline or attachment)
+     * @returns Signed URL response
+     */
+    getArtifactDownloadUrl(
+        runId: string,
+        name: string,
+        disposition?: "inline" | "attachment"
+    ): Promise<GetFileUrlResponse> {
+        const artifactPath = getAgentArtifactPath(runId, name);
+        return this.getDownloadUrl(artifactPath, name, disposition);
+    }
+
+    /**
+     * List artifacts for an agent run
+     * @param runId - The workflow run ID
+     * @returns Array of artifact file paths
+     */
+    async listArtifacts(runId: string): Promise<string[]> {
+        const prefix = `${ARTIFACTS_PREFIX}/${runId}/`;
+        const result = await this.listByPrefix(prefix);
+        return result.files;
+    }
 }

package/src/store/ObjectsApi.ts

@@ -364,6 +364,7 @@ export class ObjectsApi extends ApiTopic {
             max_hw: options.max_hw,
             generate_if_missing: options.generate_if_missing,
             sign_url: options.sign_url,
+            block_on_generation: options.block_on_generation,
         };
 
         return this.get(`/${documentId}/renditions/${options.format}`, {

package/src/store/index.ts

@@ -1,5 +1,6 @@
 export * from "./client.js";
 export * from "./CommandsApi.js";
+export * from "./FilesApi.js";
 export * from "./ObjectsApi.js";
 export * from "./TypesApi.js";
 export * from "./WorkflowsApi.js";