@upstash/workflow 0.2.12 → 0.2.14

package/svelte.mjs

@@ -2,7 +2,7 @@ import {
   SDK_TELEMETRY,
   serveBase,
   serveManyBase
-} from "./chunk-4GTHIL7S.mjs";
+} from "./chunk-RMS2NQ3K.mjs";
 
 // platforms/svelte.ts
 var telemetry = {

package/{types-D1W0VOpy.d.ts → types-C1WIgVLA.d.mts}

@@ -2,8 +2,47 @@ import { PublishRequest, FlowControl, Client, Receiver, HTTPMethods as HTTPMetho
 import { ZodType, z } from 'zod';
 import * as ai from 'ai';
 import { CoreTool, generateText } from 'ai';
-import * as _ai_sdk_openai from '@ai-sdk/openai';
+import { createOpenAI } from '@ai-sdk/openai';
 
+declare const LOG_LEVELS: readonly ["DEBUG", "INFO", "SUBMIT", "WARN", "ERROR"];
+type LogLevel = (typeof LOG_LEVELS)[number];
+type ChatLogEntry = {
+    timestamp: number;
+    workflowRunId: string;
+    logLevel: LogLevel;
+    eventType: "ENDPOINT_START" | "SUBMIT_THIRD_PARTY_RESULT" | "CREATE_CONTEXT" | "SUBMIT_FIRST_INVOCATION" | "RUN_SINGLE" | "RUN_PARALLEL" | "SUBMIT_STEP" | "SUBMIT_CLEANUP" | "RESPONSE_WORKFLOW" | "RESPONSE_DEFAULT" | "ERROR";
+    details: unknown;
+};
+type WorkflowLoggerOptions = {
+    logLevel: LogLevel;
+    logOutput: "console";
+};
+declare class WorkflowLogger {
+    private logs;
+    private options;
+    private workflowRunId?;
+    constructor(options: WorkflowLoggerOptions);
+    log(level: LogLevel, eventType: ChatLogEntry["eventType"], details?: unknown): Promise<void>;
+    setWorkflowRunId(workflowRunId: string): void;
+    private writeToConsole;
+    private shouldLog;
+    static getLogger(verbose?: boolean | WorkflowLogger): WorkflowLogger | undefined;
+}
+
+type HeadersResponse = {
+    headers: Record<string, string>;
+    contentType: string;
+};
+
+type StepParams = {
+    context: WorkflowContext;
+} & Pick<HeaderParams, "telemetry"> & Required<Pick<HeaderParams, "step" | "invokeCount">>;
+type GetHeaderParams = StepParams;
+type GetBodyParams = StepParams & Omit<HeadersResponse, "contentType">;
+type SubmitStepParams = StepParams & Pick<HeadersResponse, "headers"> & {
+    body: string;
+    isParallel: boolean;
+};
 /**
  * Base class outlining steps. Basically, each step kind (run/sleep/sleepUntil)
  * should have two methods: getPlanStep & getResultStep.
@@ -44,31 +83,11 @@ declare abstract class BaseLazyStep<TResult = unknown> {
     parseOut(out: unknown): TResult;
     protected safeParseOut(out: string): TResult;
     protected static tryParsing(stepOut: unknown): any;
-}
-
-declare const LOG_LEVELS: readonly ["DEBUG", "INFO", "SUBMIT", "WARN", "ERROR"];
-type LogLevel = (typeof LOG_LEVELS)[number];
-type ChatLogEntry = {
-    timestamp: number;
-    workflowRunId: string;
-    logLevel: LogLevel;
-    eventType: "ENDPOINT_START" | "SUBMIT_THIRD_PARTY_RESULT" | "CREATE_CONTEXT" | "SUBMIT_FIRST_INVOCATION" | "RUN_SINGLE" | "RUN_PARALLEL" | "SUBMIT_STEP" | "SUBMIT_CLEANUP" | "RESPONSE_WORKFLOW" | "RESPONSE_DEFAULT" | "ERROR";
-    details: unknown;
-};
-type WorkflowLoggerOptions = {
-    logLevel: LogLevel;
-    logOutput: "console";
-};
-declare class WorkflowLogger {
-    private logs;
-    private options;
-    private workflowRunId?;
-    constructor(options: WorkflowLoggerOptions);
-    log(level: LogLevel, eventType: ChatLogEntry["eventType"], details?: unknown): Promise<void>;
-    setWorkflowRunId(workflowRunId: string): void;
-    private writeToConsole;
-    private shouldLog;
-    static getLogger(verbose?: boolean | WorkflowLogger): WorkflowLogger | undefined;
+    getBody({ step }: GetBodyParams): string;
+    getHeaders({ context, telemetry, invokeCount, step }: GetHeaderParams): HeadersResponse;
+    submitStep({ context, body, headers }: SubmitStepParams): Promise<{
+        messageId: string;
+    }[]>;
 }
 
 declare class AutoExecutor {
@@ -156,12 +175,6 @@ declare class AutoExecutor {
      * @returns parallel call state
      */
     protected getParallelCallState(parallelStepCount: number, initialStepCount: number): ParallelCallState;
-    /**
-     * sends the steps to QStash as batch
-     *
-     * @param steps steps to send
-     */
-    private submitStepsToQStash;
     /**
      * Get the promise by executing the lazt steps list. If there is a single
      * step, we call `runSingle`. Otherwise `runParallel` is called.
@@ -446,17 +459,6 @@ declare class Agent {
     asTool(): AISDKTool;
 }
 
-/**
- * creates an AI SDK openai client with a custom
- * fetch implementation which uses context.call.
- *
- * @param context workflow context
- * @returns ai sdk openai
- */
-declare const createWorkflowOpenAI: (context: WorkflowContext, config?: {
-    baseURL?: string;
-    apiKey?: string;
-}) => _ai_sdk_openai.OpenAIProvider;
 declare class WorkflowTool<TSchema extends ZodType = ZodType> implements LangchainTool {
     /**
      * description of the tool
@@ -582,10 +584,13 @@ type MultiAgentTaskParams = TaskParams & {
      */
     background?: string;
 };
-type ModelParams = Parameters<ReturnType<typeof createWorkflowOpenAI>>;
+type ModelParams = Parameters<ReturnType<typeof createOpenAI>>;
+type AgentCallParams = Pick<CallSettings, "flowControl" | "retries" | "timeout">;
 type CustomModelSettings = ModelParams["1"] & {
     baseURL?: string;
     apiKey?: string;
+} & {
+    callSettings: AgentCallParams;
 };
 type CustomModelParams = [ModelParams[0], CustomModelSettings?];
 type ProviderFunction = (params: {
@@ -703,10 +708,13 @@ declare class WorkflowAgents {
      * creates an openai model for agents
      */
     openai(...params: CustomModelParams): ai.LanguageModelV1;
-    AISDKModel: <TProvider extends ProviderFunction>({ context, provider, providerParams, }: {
+    AISDKModel: <TProvider extends ProviderFunction>({ context, provider, providerParams, agentCallParams, }: {
         context: WorkflowContext;
         provider: TProvider;
         providerParams?: Omit<Required<Parameters<TProvider>>[0], "fetch">;
+        agentCallParams? /**
+         * creates an openai model for agents
+         */: AgentCallParams;
     }) => ReturnType<TProvider>;
 }
 
@@ -942,6 +950,11 @@ declare class WorkflowContext<TInitialPayload = unknown> {
      *   }
      */
     call<TResult = unknown, TBody = unknown>(stepName: string, settings: CallSettings<TBody>): Promise<CallResponse<TResult>>;
+    call<TResult extends {
+        workflowRunId: string;
+    } = {
+        workflowRunId: string;
+    }, TBody = unknown>(stepName: string, settings: LazyInvokeStepParams<TBody, unknown> & Pick<CallSettings, "timeout">): Promise<CallResponse<TResult>>;
     /**
      * Pauses workflow execution until a specific event occurs or a timeout is reached.
      *

package/{types-D1W0VOpy.d.mts → types-C1WIgVLA.d.ts}

@@ -2,8 +2,47 @@ import { PublishRequest, FlowControl, Client, Receiver, HTTPMethods as HTTPMetho
 import { ZodType, z } from 'zod';
 import * as ai from 'ai';
 import { CoreTool, generateText } from 'ai';
-import * as _ai_sdk_openai from '@ai-sdk/openai';
+import { createOpenAI } from '@ai-sdk/openai';
 
+declare const LOG_LEVELS: readonly ["DEBUG", "INFO", "SUBMIT", "WARN", "ERROR"];
+type LogLevel = (typeof LOG_LEVELS)[number];
+type ChatLogEntry = {
+    timestamp: number;
+    workflowRunId: string;
+    logLevel: LogLevel;
+    eventType: "ENDPOINT_START" | "SUBMIT_THIRD_PARTY_RESULT" | "CREATE_CONTEXT" | "SUBMIT_FIRST_INVOCATION" | "RUN_SINGLE" | "RUN_PARALLEL" | "SUBMIT_STEP" | "SUBMIT_CLEANUP" | "RESPONSE_WORKFLOW" | "RESPONSE_DEFAULT" | "ERROR";
+    details: unknown;
+};
+type WorkflowLoggerOptions = {
+    logLevel: LogLevel;
+    logOutput: "console";
+};
+declare class WorkflowLogger {
+    private logs;
+    private options;
+    private workflowRunId?;
+    constructor(options: WorkflowLoggerOptions);
+    log(level: LogLevel, eventType: ChatLogEntry["eventType"], details?: unknown): Promise<void>;
+    setWorkflowRunId(workflowRunId: string): void;
+    private writeToConsole;
+    private shouldLog;
+    static getLogger(verbose?: boolean | WorkflowLogger): WorkflowLogger | undefined;
+}
+
+type HeadersResponse = {
+    headers: Record<string, string>;
+    contentType: string;
+};
+
+type StepParams = {
+    context: WorkflowContext;
+} & Pick<HeaderParams, "telemetry"> & Required<Pick<HeaderParams, "step" | "invokeCount">>;
+type GetHeaderParams = StepParams;
+type GetBodyParams = StepParams & Omit<HeadersResponse, "contentType">;
+type SubmitStepParams = StepParams & Pick<HeadersResponse, "headers"> & {
+    body: string;
+    isParallel: boolean;
+};
 /**
  * Base class outlining steps. Basically, each step kind (run/sleep/sleepUntil)
  * should have two methods: getPlanStep & getResultStep.
@@ -44,31 +83,11 @@ declare abstract class BaseLazyStep<TResult = unknown> {
     parseOut(out: unknown): TResult;
     protected safeParseOut(out: string): TResult;
     protected static tryParsing(stepOut: unknown): any;
-}
-
-declare const LOG_LEVELS: readonly ["DEBUG", "INFO", "SUBMIT", "WARN", "ERROR"];
-type LogLevel = (typeof LOG_LEVELS)[number];
-type ChatLogEntry = {
-    timestamp: number;
-    workflowRunId: string;
-    logLevel: LogLevel;
-    eventType: "ENDPOINT_START" | "SUBMIT_THIRD_PARTY_RESULT" | "CREATE_CONTEXT" | "SUBMIT_FIRST_INVOCATION" | "RUN_SINGLE" | "RUN_PARALLEL" | "SUBMIT_STEP" | "SUBMIT_CLEANUP" | "RESPONSE_WORKFLOW" | "RESPONSE_DEFAULT" | "ERROR";
-    details: unknown;
-};
-type WorkflowLoggerOptions = {
-    logLevel: LogLevel;
-    logOutput: "console";
-};
-declare class WorkflowLogger {
-    private logs;
-    private options;
-    private workflowRunId?;
-    constructor(options: WorkflowLoggerOptions);
-    log(level: LogLevel, eventType: ChatLogEntry["eventType"], details?: unknown): Promise<void>;
-    setWorkflowRunId(workflowRunId: string): void;
-    private writeToConsole;
-    private shouldLog;
-    static getLogger(verbose?: boolean | WorkflowLogger): WorkflowLogger | undefined;
+    getBody({ step }: GetBodyParams): string;
+    getHeaders({ context, telemetry, invokeCount, step }: GetHeaderParams): HeadersResponse;
+    submitStep({ context, body, headers }: SubmitStepParams): Promise<{
+        messageId: string;
+    }[]>;
 }
 
 declare class AutoExecutor {
@@ -156,12 +175,6 @@ declare class AutoExecutor {
      * @returns parallel call state
      */
     protected getParallelCallState(parallelStepCount: number, initialStepCount: number): ParallelCallState;
-    /**
-     * sends the steps to QStash as batch
-     *
-     * @param steps steps to send
-     */
-    private submitStepsToQStash;
     /**
      * Get the promise by executing the lazt steps list. If there is a single
      * step, we call `runSingle`. Otherwise `runParallel` is called.
@@ -446,17 +459,6 @@ declare class Agent {
     asTool(): AISDKTool;
 }
 
-/**
- * creates an AI SDK openai client with a custom
- * fetch implementation which uses context.call.
- *
- * @param context workflow context
- * @returns ai sdk openai
- */
-declare const createWorkflowOpenAI: (context: WorkflowContext, config?: {
-    baseURL?: string;
-    apiKey?: string;
-}) => _ai_sdk_openai.OpenAIProvider;
 declare class WorkflowTool<TSchema extends ZodType = ZodType> implements LangchainTool {
     /**
      * description of the tool
@@ -582,10 +584,13 @@ type MultiAgentTaskParams = TaskParams & {
      */
     background?: string;
 };
-type ModelParams = Parameters<ReturnType<typeof createWorkflowOpenAI>>;
+type ModelParams = Parameters<ReturnType<typeof createOpenAI>>;
+type AgentCallParams = Pick<CallSettings, "flowControl" | "retries" | "timeout">;
 type CustomModelSettings = ModelParams["1"] & {
     baseURL?: string;
     apiKey?: string;
+} & {
+    callSettings: AgentCallParams;
 };
 type CustomModelParams = [ModelParams[0], CustomModelSettings?];
 type ProviderFunction = (params: {
@@ -703,10 +708,13 @@ declare class WorkflowAgents {
      * creates an openai model for agents
      */
     openai(...params: CustomModelParams): ai.LanguageModelV1;
-    AISDKModel: <TProvider extends ProviderFunction>({ context, provider, providerParams, }: {
+    AISDKModel: <TProvider extends ProviderFunction>({ context, provider, providerParams, agentCallParams, }: {
         context: WorkflowContext;
         provider: TProvider;
         providerParams?: Omit<Required<Parameters<TProvider>>[0], "fetch">;
+        agentCallParams? /**
+         * creates an openai model for agents
+         */: AgentCallParams;
     }) => ReturnType<TProvider>;
 }
 
@@ -942,6 +950,11 @@ declare class WorkflowContext<TInitialPayload = unknown> {
      *   }
      */
     call<TResult = unknown, TBody = unknown>(stepName: string, settings: CallSettings<TBody>): Promise<CallResponse<TResult>>;
+    call<TResult extends {
+        workflowRunId: string;
+    } = {
+        workflowRunId: string;
+    }, TBody = unknown>(stepName: string, settings: LazyInvokeStepParams<TBody, unknown> & Pick<CallSettings, "timeout">): Promise<CallResponse<TResult>>;
     /**
      * Pauses workflow execution until a specific event occurs or a timeout is reached.
      *