@vertesia/workflow 1.0.0-dev.20260128.144200 → 1.0.0-dev.20260225.024852Z

package/src/dsl/setup/ActivityContext.ts

@@ -5,10 +5,13 @@ import {
     DSLWorkflowExecutionPayload,
     Project,
     WorkflowExecutionPayload,
+    WorkflowInputType,
+    WorkflowInputFile,
 } from "@vertesia/common";
 import {
     DocumentNotFoundError,
     WorkflowParamNotFoundError,
+    WorkflowExecutionError,
 } from "../../errors.js";
 import { getProjectFromToken } from "../../utils/auth.js";
 import { getVertesiaClient } from "../../utils/client.js";
@@ -99,6 +102,76 @@ export class ActivityContext<ParamsT extends Record<string, any>> {
         return workflowId;
     }
 
+    /**
+     * Get the workflow input type (objectIds or files).
+     * Defaults to 'objectIds' for backward compatibility with legacy format.
+     */
+    get inputType(): WorkflowInputType {
+        return this.payload.input?.inputType || 'objectIds';
+    }
+
+    /**
+     * Get the first file from the workflow input.
+     * Only available when workflow input type is 'files'.
+     * @throws {WorkflowExecutionError} If input type is not 'files'
+     * @throws {WorkflowParamNotFoundError} If files array is empty
+     */
+    get file(): WorkflowInputFile {
+        const input = this.payload.input;
+        if (!input || input.inputType !== 'files') {
+            throw new WorkflowExecutionError(
+                'Activity expects files but received objectIds'
+            );
+        }
+        // TypeScript now knows input is { inputType: 'files', files: WorkflowInputFile[] }
+        const files = input.files;
+        if (!files || files.length === 0) {
+            log.error("No files found in payload");
+            throw new WorkflowParamNotFoundError(
+                "files[0]",
+                (
+                    this
+                        .payload as WorkflowExecutionPayload as DSLWorkflowExecutionPayload
+                ).workflow,
+            );
+        }
+        return files[0];
+    }
+
+    /**
+     * Get all files from the workflow input.
+     * Only available when workflow input type is 'files'.
+     * @throws {WorkflowExecutionError} If input type is not 'files'
+     */
+    get files(): WorkflowInputFile[] {
+        const input = this.payload.input;
+        if (!input || input.inputType !== 'files') {
+            throw new WorkflowExecutionError(
+                'Activity expects files but received objectIds'
+            );
+        }
+        // TypeScript now knows input is { inputType: 'files', files: WorkflowInputFile[] }
+        return input.files;
+    }
+
+    /**
+     * Generic accessor for the first input (objectId or file URL).
+     * Use this in dual-mode activities that support both input types.
+     * For files, returns the URL of the first file.
+     */
+    get input(): string {
+        return this.inputType === 'objectIds' ? this.objectId : this.file.url;
+    }
+
+    /**
+     * Generic accessor for all inputs (objectIds or file URLs).
+     * Use this in dual-mode activities that support both input types.
+     * For files, returns an array of URLs.
+     */
+    get inputs(): string[] {
+        return this.inputType === 'objectIds' ? (this.objectIds || []) : this.files.map(f => f.url);
+    }
+
     fetchProject() {
         if (!this._project) {
             this._project = _fetchProject(this.client, this.payload);

package/src/dsl.ts

@@ -1,3 +1,4 @@
 export * from "./dsl/dslProxyActivities.js";
 export * from "./errors.js";
 export * from "./result-types.js";
+

package/src/errors.ts

@@ -91,6 +91,19 @@ export class TokenExpiredError extends ApplicationFailure {
     }
 }
 
+export class WorkflowExecutionError extends ApplicationFailure {
+    constructor(
+        message: string,
+        public workflow?: DSLWorkflowSpec,
+    ) {
+        super(
+            message,
+            "WorkflowExecutionError",
+            true, // non-retryable
+        );
+    }
+}
+
 export const WF_NON_RETRYABLE_ERRORS = [
     "DocumentNotFoundError",
     "ActivityParamInvalidError",
@@ -98,4 +111,6 @@ export const WF_NON_RETRYABLE_ERRORS = [
     "WorkflowParamNotFoundError",
     "InvalidContentTypeError",
     "TokenExpiredError",
+    "ZenoClientNotFoundError",
+    "WorkflowExecutionError",
 ];

package/src/index.ts

@@ -18,15 +18,15 @@ export * from "./activities/executeInteraction.js";
 export * from "./activities/extractDocumentText.js";
 export * from "./activities/generateDocumentProperties.js";
 export * from "./activities/generateEmbeddings.js";
-export * from "./activities/renditions/generateImageRendition.js";
-export * from "./activities/renditions/generateVideoRendition.js";
 export * from "./activities/generateOrAssignContentType.js";
+export * from "./activities/media/prepareAudio.js";
+export * from "./activities/media/prepareVideo.js";
 export * from "./activities/notifyWebhook.js";
 export * from "./activities/rateLimiter.js";
+export * from "./activities/renditions/generateImageRendition.js";
+export * from "./activities/renditions/generateVideoRendition.js";
 export * from "./activities/setDocumentStatus.js";
 export * from "./iterative-generation/activities/index.js";
-export * from "./activities/media/prepareVideo.js";
-export * from "./activities/media/prepareAudio.js";
 
 export * from "./dsl/setup/ActivityContext.js";
 export * from "./errors.js";
@@ -34,8 +34,9 @@ export * from "./result-types.js";
 export * from "./utils/blobs.js";
 export * from "./utils/client.js";
 export * from "./utils/memory.js";
-export * from "./utils/tokens.js";
 export * from "./utils/renditions.js";
 export * from "./utils/storage.js";
+export * from "./utils/text-preview-utils.js";
+export * from "./utils/tokens.js";
 
 export * from "./conversion/image.js";

package/src/result-types.ts

@@ -5,7 +5,11 @@
  * The result of a text extraction operation.
  */
 export interface TextExtractionResult {
-    objectId: string;
+    objectId?: string;
+    file?: {   
+        source_url: string;
+        result_url?: string;
+    }
     status: TextExtractionStatus;
     hasText: boolean;
     message?: string;

package/src/utils/renditions.ts

@@ -9,6 +9,8 @@ import { imageResizer } from "../conversion/image.js";
 export interface ImageRenditionParams {
     max_hw: number; //maximum size of the longest side of the image
     format: ImageRenditionFormat;
+    /** Custom upload path — overrides default renditions/{etag}/{max_hw}/{page}.{format} path */
+    outputPath?: string;
 }
 
 /**
@@ -53,7 +55,7 @@ export async function uploadRenditionPages(
     params: ImageRenditionParams,
     concurrency?: number,
 ) {
-    log.info(
+    log.debug(
         `Uploading rendition for etag ${contentEtag} with ${files.length} pages (max_hw: ${params.max_hw}, format: ${params.format})`,
         { files },
     );
@@ -61,11 +63,15 @@ export async function uploadRenditionPages(
     const limit = pLimit(concurrency ?? 20);
 
     const uploads = files.map((file, i) => limit(async () => {
-        const pageId = getRenditionPagePath(contentEtag, params, i);
+        const pageId = params.outputPath
+            ? (files.length === 1
+                ? params.outputPath
+                : `${params.outputPath}/${String(i).padStart(4, "0")}.${params.format}`)
+            : getRenditionPagePath(contentEtag, params, i);
         let resizedImagePath = null;
 
         try {
-            log.info(`Resizing image for ${contentEtag} page ${i}`, {
+            log.debug(`Resizing image for ${contentEtag} page ${i}`, {
                 file,
                 params,
             });
@@ -87,7 +93,7 @@ export async function uploadRenditionPages(
                 pageId,
             );
 
-            log.info(
+            log.debug(
                 `Uploading rendition for ${contentEtag} page ${i} with max_hw: ${params.max_hw} and format: ${params.format}`,
                 {
                     resizedImagePath,
@@ -110,7 +116,7 @@ export async function uploadRenditionPages(
                     );
                     return Promise.reject(`Upload failed: ${err.message}`);
                 });
-            log.info(`Rendition uploaded for ${contentEtag} page ${i}`, {
+            log.debug(`Rendition uploaded for ${contentEtag} page ${i}`, {
                 result,
             });
 

package/src/utils/text-preview-utils.ts

@@ -0,0 +1,62 @@
+import { log } from "@temporalio/activity";
+import { VertesiaClient } from "@vertesia/client";
+import { NodeStreamSource } from "@vertesia/client/node";
+import { Readable } from "stream";
+import { TextExtractionResult, TextExtractionStatus } from "../result-types.js";
+
+/**
+ * Uploads extracted text preview to cloud storage
+ */
+export async function uploadTextPreviewToStorage(
+    vertesia: VertesiaClient,
+    text: string,
+    storagePath: string,
+    fileType: string
+): Promise<string> {
+
+    const source = new NodeStreamSource(
+        Readable.from(text),
+        storagePath,
+        'text/markdown',
+        storagePath
+    );
+
+    const uploadedPath = await vertesia.files.uploadFile(source);
+    log.info(`Uploaded ${fileType} preview text to ${uploadedPath}`);
+
+    return uploadedPath;
+}
+
+/**
+ * Creates the TextExtractionResult for file_source mode
+ */
+export function createFileSourceResult(
+    sourceUrl: string,
+    storagePath: string,
+    text: string | null
+): TextExtractionResult {
+    return {
+        hasText: !!text,
+        file: {
+            source_url: sourceUrl,
+            result_url: text && storagePath ? storagePath : undefined,
+        },
+        status: TextExtractionStatus.success,
+    };
+}
+
+/**
+ * Saves extracted text to an object in the object store
+ */
+export async function saveTextToObject(
+    vertesia: VertesiaClient,
+    objectId: string,
+    text: string
+): Promise<void> {
+    const object = await vertesia.objects.retrieve(objectId);
+    await vertesia.objects.update(objectId, {
+        text: text,
+        text_etag: object.content?.etag,
+    });
+    log.info(`Saved text to object ${objectId}`);
+}