@upstash/workflow 0.2.3 → 0.2.5-agents

package/svelte.mjs

@@ -1,14 +1,23 @@
 import {
+  SDK_TELEMETRY,
   serveBase
-} from "./chunk-QBJ3LQIO.mjs";
+} from "./chunk-RFX5YRRT.mjs";
+import "./chunk-PU5J4TNC.mjs";
 
 // platforms/svelte.ts
 var serve = (routeFunction, options) => {
   const handler = async ({ request }) => {
-    const { handler: serveHandler } = serveBase(routeFunction, {
-      ...options,
-      useJSONContent: true
-    });
+    const { handler: serveHandler } = serveBase(
+      routeFunction,
+      {
+        sdk: SDK_TELEMETRY,
+        framework: "svelte"
+      },
+      {
+        ...options,
+        useJSONContent: true
+      }
+    );
     return await serveHandler(request);
   };
   return { POST: handler };

package/{types-R9q4MUwl.d.mts → types-BEyIoCRe.d.mts}

@@ -1,4 +1,8 @@
 import { PublishRequest, Client, Receiver, HTTPMethods as HTTPMethods$1 } from '@upstash/qstash';
+import * as ai from 'ai';
+import { CoreTool, generateText } from 'ai';
+import * as _ai_sdk_openai from '@ai-sdk/openai';
+import { Tool } from 'langchain/tools';
 
 /**
  * Base class outlining steps. Basically, each step kind (run/sleep/sleepUntil)
@@ -63,10 +67,11 @@ declare class AutoExecutor {
     private readonly nonPlanStepCount;
     private readonly steps;
     private indexInCurrentList;
+    private telemetry?;
     stepCount: number;
     planStepCount: number;
     protected executingStep: string | false;
-    constructor(context: WorkflowContext, steps: Step[], debug?: WorkflowLogger);
+    constructor(context: WorkflowContext, steps: Step[], telemetry?: Telemetry, debug?: WorkflowLogger);
     /**
      * Adds the step function to the list of step functions to run in
      * parallel. After adding the function, defers the execution, so
@@ -386,6 +391,76 @@ declare class WorkflowApi extends BaseWorkflowApi {
     get anthropic(): AnthropicAPI;
 }
 
+declare class Agent {
+    readonly name: AgentParameters["name"];
+    readonly tools: AgentParameters["tools"];
+    readonly maxSteps: AgentParameters["maxSteps"];
+    readonly background: AgentParameters["background"];
+    readonly model: AgentParameters["model"];
+    constructor({ tools, maxSteps, background, name, model }: AgentParameters);
+    call({ prompt }: {
+        prompt: string;
+    }): Promise<ai.GenerateTextResult<Record<string, AISDKTool>, never>>;
+    asTool(): AISDKTool;
+}
+type ManagerAgentParameters = {
+    agents: Agent[];
+    model: Model;
+} & Pick<Partial<AgentParameters>, "name" | "background"> & Pick<AgentParameters, "maxSteps">;
+declare class ManagerAgent extends Agent {
+    agents: ManagerAgentParameters["agents"];
+    constructor({ maxSteps, background, agents, model, name, }: ManagerAgentParameters);
+}
+
+type AISDKTool = CoreTool;
+type LangchainTool = Tool;
+type GenerateTextParams = Parameters<typeof generateText>[0];
+type Model = GenerateTextParams["model"];
+type AgentParameters<TTool extends AISDKTool | LangchainTool = AISDKTool> = {
+    maxSteps: number;
+    background: string;
+    tools: Record<string, TTool>;
+    name: string;
+    model: Model;
+};
+type TaskParams = {
+    prompt: string;
+};
+type SingleAgentTaskParams = TaskParams & {
+    agent: Agent;
+};
+type MultiAgentTaskParams = TaskParams & {
+    agents: Agent[];
+    maxSteps: number;
+    model: Model;
+    background?: string;
+};
+
+declare const createWorkflowOpenAI: (context: WorkflowContext) => _ai_sdk_openai.OpenAIProvider;
+
+declare class Task {
+    private readonly context;
+    private readonly taskParameters;
+    constructor({ context, taskParameters, }: {
+        context: WorkflowContext;
+        taskParameters: SingleAgentTaskParams | MultiAgentTaskParams;
+    });
+    run(): Promise<{
+        text: string;
+    }>;
+}
+
+declare class WorkflowAgents {
+    private context;
+    constructor({ context }: {
+        context: WorkflowContext;
+    });
+    agent(params: AgentParameters<AISDKTool | LangchainTool>): Agent;
+    task(taskParameters: SingleAgentTaskParams): Task;
+    task(taskParameters: MultiAgentTaskParams): Task;
+    openai(...params: Parameters<ReturnType<typeof createWorkflowOpenAI>>): ai.LanguageModelV1;
+}
+
 /**
  * Upstash Workflow context
  *
@@ -512,7 +587,7 @@ declare class WorkflowContext<TInitialPayload = unknown> {
      * Number of retries
      */
     readonly retries: number;
-    constructor({ qstashClient, workflowRunId, headers, steps, url, failureUrl, debug, initialPayload, env, retries, }: {
+    constructor({ qstashClient, workflowRunId, headers, steps, url, failureUrl, debug, initialPayload, env, retries, telemetry, }: {
         qstashClient: WorkflowClient;
         workflowRunId: string;
         headers: Headers;
@@ -523,6 +598,7 @@ declare class WorkflowContext<TInitialPayload = unknown> {
         initialPayload: TInitialPayload;
         env?: Record<string, string | undefined>;
         retries?: number;
+        telemetry?: Telemetry;
     });
     /**
      * Executes a workflow step
@@ -680,6 +756,7 @@ declare class WorkflowContext<TInitialPayload = unknown> {
      */
     protected addStep<TResult = unknown>(step: BaseLazyStep<TResult>): Promise<TResult>;
     get api(): WorkflowApi;
+    get agents(): WorkflowAgents;
 }
 
 /**
@@ -838,7 +915,7 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * @returns void
      */
     failureFunction?: (failureData: {
-        context: Omit<WorkflowContext<TInitialPayload>, "run" | "sleepUntil" | "sleep" | "call" | "waitForEvent" | "notify">;
+        context: Omit<WorkflowContext<TInitialPayload>, "run" | "sleepUntil" | "sleep" | "call" | "waitForEvent" | "notify" | "cancel" | "api">;
         failStatus: number;
         failResponse: string;
         failHeaders: Record<string, string[]>;
@@ -876,6 +953,26 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * Not part of the public API. Only available in serveBase, which is not exported.
      */
     useJSONContent?: boolean;
+    /**
+     * By default, Workflow SDK sends telemetry about SDK version, framework or runtime.
+     *
+     * Set `disableTelemetry` to disable this behavior.
+     */
+    disableTelemetry?: boolean;
+};
+type Telemetry = {
+    /**
+     * sdk version
+     */
+    sdk: string;
+    /**
+     * platform (such as nextjs/cloudflare)
+     */
+    framework: string;
+    /**
+     * node version
+     */
+    runtime?: string;
 };
 type PublicServeOptions<TInitialPayload = unknown, TResponse extends Response = Response> = Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "onStepFinish" | "useJSONContent">;
 /**
@@ -971,5 +1068,68 @@ type CallSettings<TBody = unknown> = {
     retries?: number;
     timeout?: Duration | number;
 };
+type HeaderParams = {
+    /**
+     * whether the request is a first invocation request.
+     */
+    initHeaderValue: "true" | "false";
+    /**
+     * run id of the workflow
+     */
+    workflowRunId: string;
+    /**
+     * url where the workflow is hosted
+     */
+    workflowUrl: string;
+    /**
+     * user headers which will be forwarded in the request
+     */
+    userHeaders?: Headers;
+    /**
+     * failure url to call incase of failure
+     */
+    failureUrl?: WorkflowServeOptions["failureUrl"];
+    /**
+     * retry setting of requests except context.call
+     */
+    retries?: number;
+    /**
+     * telemetry to include in timeoutHeaders.
+     *
+     * Only needed/used when the step is a waitForEvent step
+     */
+    telemetry?: Telemetry;
+} & ({
+    /**
+     * step to generate headers for
+     */
+    step: Step;
+    /**
+     * number of retries in context.call
+     */
+    callRetries?: number;
+    /**
+     * timeout duration in context.call
+     */
+    callTimeout?: number | Duration;
+} | {
+    /**
+     * step not passed. Either first invocation or simply getting headers for
+     * third party callack.
+     */
+    step?: never;
+    /**
+     * number of retries in context.call
+     *
+     * set to never because this is not a context.call step
+     */
+    callRetries?: never;
+    /**
+     * timeout duration in context.call
+     *
+     * set to never because this is not a context.call step
+     */
+    callTimeout?: never;
+});
 
-export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type FinishCondition as F, type LogLevel as L, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s };
+export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type FinishCondition as F, type HeaderParams as H, type LogLevel as L, ManagerAgent as M, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type Telemetry as T, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s, WorkflowAgents as t, createWorkflowOpenAI as u, Agent as v };

package/{types-R9q4MUwl.d.ts → types-BEyIoCRe.d.ts}

@@ -1,4 +1,8 @@
 import { PublishRequest, Client, Receiver, HTTPMethods as HTTPMethods$1 } from '@upstash/qstash';
+import * as ai from 'ai';
+import { CoreTool, generateText } from 'ai';
+import * as _ai_sdk_openai from '@ai-sdk/openai';
+import { Tool } from 'langchain/tools';
 
 /**
  * Base class outlining steps. Basically, each step kind (run/sleep/sleepUntil)
@@ -63,10 +67,11 @@ declare class AutoExecutor {
     private readonly nonPlanStepCount;
     private readonly steps;
     private indexInCurrentList;
+    private telemetry?;
     stepCount: number;
     planStepCount: number;
     protected executingStep: string | false;
-    constructor(context: WorkflowContext, steps: Step[], debug?: WorkflowLogger);
+    constructor(context: WorkflowContext, steps: Step[], telemetry?: Telemetry, debug?: WorkflowLogger);
     /**
      * Adds the step function to the list of step functions to run in
      * parallel. After adding the function, defers the execution, so
@@ -386,6 +391,76 @@ declare class WorkflowApi extends BaseWorkflowApi {
     get anthropic(): AnthropicAPI;
 }
 
+declare class Agent {
+    readonly name: AgentParameters["name"];
+    readonly tools: AgentParameters["tools"];
+    readonly maxSteps: AgentParameters["maxSteps"];
+    readonly background: AgentParameters["background"];
+    readonly model: AgentParameters["model"];
+    constructor({ tools, maxSteps, background, name, model }: AgentParameters);
+    call({ prompt }: {
+        prompt: string;
+    }): Promise<ai.GenerateTextResult<Record<string, AISDKTool>, never>>;
+    asTool(): AISDKTool;
+}
+type ManagerAgentParameters = {
+    agents: Agent[];
+    model: Model;
+} & Pick<Partial<AgentParameters>, "name" | "background"> & Pick<AgentParameters, "maxSteps">;
+declare class ManagerAgent extends Agent {
+    agents: ManagerAgentParameters["agents"];
+    constructor({ maxSteps, background, agents, model, name, }: ManagerAgentParameters);
+}
+
+type AISDKTool = CoreTool;
+type LangchainTool = Tool;
+type GenerateTextParams = Parameters<typeof generateText>[0];
+type Model = GenerateTextParams["model"];
+type AgentParameters<TTool extends AISDKTool | LangchainTool = AISDKTool> = {
+    maxSteps: number;
+    background: string;
+    tools: Record<string, TTool>;
+    name: string;
+    model: Model;
+};
+type TaskParams = {
+    prompt: string;
+};
+type SingleAgentTaskParams = TaskParams & {
+    agent: Agent;
+};
+type MultiAgentTaskParams = TaskParams & {
+    agents: Agent[];
+    maxSteps: number;
+    model: Model;
+    background?: string;
+};
+
+declare const createWorkflowOpenAI: (context: WorkflowContext) => _ai_sdk_openai.OpenAIProvider;
+
+declare class Task {
+    private readonly context;
+    private readonly taskParameters;
+    constructor({ context, taskParameters, }: {
+        context: WorkflowContext;
+        taskParameters: SingleAgentTaskParams | MultiAgentTaskParams;
+    });
+    run(): Promise<{
+        text: string;
+    }>;
+}
+
+declare class WorkflowAgents {
+    private context;
+    constructor({ context }: {
+        context: WorkflowContext;
+    });
+    agent(params: AgentParameters<AISDKTool | LangchainTool>): Agent;
+    task(taskParameters: SingleAgentTaskParams): Task;
+    task(taskParameters: MultiAgentTaskParams): Task;
+    openai(...params: Parameters<ReturnType<typeof createWorkflowOpenAI>>): ai.LanguageModelV1;
+}
+
 /**
  * Upstash Workflow context
  *
@@ -512,7 +587,7 @@ declare class WorkflowContext<TInitialPayload = unknown> {
      * Number of retries
      */
     readonly retries: number;
-    constructor({ qstashClient, workflowRunId, headers, steps, url, failureUrl, debug, initialPayload, env, retries, }: {
+    constructor({ qstashClient, workflowRunId, headers, steps, url, failureUrl, debug, initialPayload, env, retries, telemetry, }: {
         qstashClient: WorkflowClient;
         workflowRunId: string;
         headers: Headers;
@@ -523,6 +598,7 @@ declare class WorkflowContext<TInitialPayload = unknown> {
         initialPayload: TInitialPayload;
         env?: Record<string, string | undefined>;
         retries?: number;
+        telemetry?: Telemetry;
     });
     /**
      * Executes a workflow step
@@ -680,6 +756,7 @@ declare class WorkflowContext<TInitialPayload = unknown> {
      */
     protected addStep<TResult = unknown>(step: BaseLazyStep<TResult>): Promise<TResult>;
     get api(): WorkflowApi;
+    get agents(): WorkflowAgents;
 }
 
 /**
@@ -838,7 +915,7 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * @returns void
      */
     failureFunction?: (failureData: {
-        context: Omit<WorkflowContext<TInitialPayload>, "run" | "sleepUntil" | "sleep" | "call" | "waitForEvent" | "notify">;
+        context: Omit<WorkflowContext<TInitialPayload>, "run" | "sleepUntil" | "sleep" | "call" | "waitForEvent" | "notify" | "cancel" | "api">;
         failStatus: number;
         failResponse: string;
         failHeaders: Record<string, string[]>;
@@ -876,6 +953,26 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * Not part of the public API. Only available in serveBase, which is not exported.
      */
     useJSONContent?: boolean;
+    /**
+     * By default, Workflow SDK sends telemetry about SDK version, framework or runtime.
+     *
+     * Set `disableTelemetry` to disable this behavior.
+     */
+    disableTelemetry?: boolean;
+};
+type Telemetry = {
+    /**
+     * sdk version
+     */
+    sdk: string;
+    /**
+     * platform (such as nextjs/cloudflare)
+     */
+    framework: string;
+    /**
+     * node version
+     */
+    runtime?: string;
 };
 type PublicServeOptions<TInitialPayload = unknown, TResponse extends Response = Response> = Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "onStepFinish" | "useJSONContent">;
 /**
@@ -971,5 +1068,68 @@ type CallSettings<TBody = unknown> = {
     retries?: number;
     timeout?: Duration | number;
 };
+type HeaderParams = {
+    /**
+     * whether the request is a first invocation request.
+     */
+    initHeaderValue: "true" | "false";
+    /**
+     * run id of the workflow
+     */
+    workflowRunId: string;
+    /**
+     * url where the workflow is hosted
+     */
+    workflowUrl: string;
+    /**
+     * user headers which will be forwarded in the request
+     */
+    userHeaders?: Headers;
+    /**
+     * failure url to call incase of failure
+     */
+    failureUrl?: WorkflowServeOptions["failureUrl"];
+    /**
+     * retry setting of requests except context.call
+     */
+    retries?: number;
+    /**
+     * telemetry to include in timeoutHeaders.
+     *
+     * Only needed/used when the step is a waitForEvent step
+     */
+    telemetry?: Telemetry;
+} & ({
+    /**
+     * step to generate headers for
+     */
+    step: Step;
+    /**
+     * number of retries in context.call
+     */
+    callRetries?: number;
+    /**
+     * timeout duration in context.call
+     */
+    callTimeout?: number | Duration;
+} | {
+    /**
+     * step not passed. Either first invocation or simply getting headers for
+     * third party callack.
+     */
+    step?: never;
+    /**
+     * number of retries in context.call
+     *
+     * set to never because this is not a context.call step
+     */
+    callRetries?: never;
+    /**
+     * timeout duration in context.call
+     *
+     * set to never because this is not a context.call step
+     */
+    callTimeout?: never;
+});
 
-export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type FinishCondition as F, type LogLevel as L, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s };
+export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type FinishCondition as F, type HeaderParams as H, type LogLevel as L, ManagerAgent as M, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type Telemetry as T, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s, WorkflowAgents as t, createWorkflowOpenAI as u, Agent as v };