@upstash/workflow 0.2.22 → 0.3.0-rc

package/index.js

@@ -28,7 +28,6 @@ __export(src_exports, {
   WorkflowLogger: () => WorkflowLogger,
   WorkflowNonRetryableError: () => WorkflowNonRetryableError,
   WorkflowRetryAfterError: () => WorkflowRetryAfterError,
-  WorkflowTool: () => WorkflowTool,
   serve: () => serve
 });
 module.exports = __toCommonJS(src_exports);
@@ -194,12 +193,11 @@ var WORKFLOW_PROTOCOL_VERSION_HEADER = "Upstash-Workflow-Sdk-Version";
 var DEFAULT_CONTENT_TYPE = "application/json";
 var NO_CONCURRENCY = 1;
 var DEFAULT_RETRIES = 3;
-var VERSION = "v0.2.22";
+var VERSION = "v0.3.0-rc";
 var SDK_TELEMETRY = `@upstash/workflow@${VERSION}`;
 var TELEMETRY_HEADER_SDK = "Upstash-Telemetry-Sdk";
 var TELEMETRY_HEADER_FRAMEWORK = "Upstash-Telemetry-Framework";
 var TELEMETRY_HEADER_RUNTIME = "Upstash-Telemetry-Runtime";
-var TELEMETRY_HEADER_AGENT = "Upstash-Telemetry-Agent";
 
 // src/context/auto-executor.ts
 var import_qstash5 = require("@upstash/qstash");
@@ -572,9 +570,9 @@ var Ok = class {
   }
   safeUnwrap() {
     const value = this.value;
-    return function* () {
+    return (function* () {
       return value;
-    }();
+    })();
   }
   _unsafeUnwrap(_) {
     return this.value;
@@ -633,10 +631,10 @@ var Err = class {
   }
   safeUnwrap() {
     const error = this.error;
-    return function* () {
+    return (function* () {
       yield err(error);
       throw new Error("Do not use this generator out of `safeTry`");
-    }();
+    })();
   }
   _unsafeUnwrap(config) {
     throw createNeverThrowError("Called `_unsafeUnwrap` on an Err", this, config);
@@ -1603,20 +1601,6 @@ var LazyInvokeStep = class extends BaseLazyStep {
   }
 };
 
-// src/agents/constants.ts
-var AGENT_NAME_HEADER = "upstash-agent-name";
-var MANAGER_AGENT_PROMPT = `You are an agent orchestrating other AI Agents.
-
-These other agents have tools available to them.
-
-Given a prompt, utilize these agents to address requests.
-
-Don't always call all the agents provided to you at the same time. You can call one and use it's response to call another.
-
-Avoid calling the same agent twice in one turn. Instead, prefer to call it once but provide everything
-you need from that agent.
-`;
-
 // src/qstash/headers.ts
 var WorkflowHeaders = class {
   userHeaders;
@@ -1665,8 +1649,7 @@ var WorkflowHeaders = class {
       [WORKFLOW_URL_HEADER]: this.workflowConfig.workflowUrl,
       [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody,WF_DetectTrigger" + (this.keepTriggerConfig ? ",WF_TriggerOnConfig" : ""),
       [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION,
-      ...this.workflowConfig.telemetry ? getTelemetryHeaders(this.workflowConfig.telemetry) : {},
-      ...this.workflowConfig.telemetry && this.stepInfo?.lazyStep instanceof LazyCallStep && this.stepInfo.lazyStep.headers[AGENT_NAME_HEADER] ? { [TELEMETRY_HEADER_AGENT]: "true" } : {}
+      ...this.workflowConfig.telemetry ? getTelemetryHeaders(this.workflowConfig.telemetry) : {}
     };
     if (this.stepInfo?.lazyStep.stepType !== "Call") {
       this.headers.rawHeaders[`Upstash-Forward-${WORKFLOW_PROTOCOL_VERSION_HEADER}`] = WORKFLOW_PROTOCOL_VERSION;
@@ -2329,340 +2312,6 @@ var WorkflowApi = class extends BaseWorkflowApi {
   }
 };
 
-// src/agents/index.ts
-var import_openai2 = require("@ai-sdk/openai");
-
-// src/agents/adapters.ts
-var import_ai = require("ai");
-var fetchWithContextCall = async (context, agentCallParams, ...params) => {
-  const [input, init] = params;
-  try {
-    const headers = init?.headers ? Object.fromEntries(new Headers(init.headers).entries()) : {};
-    const body = init?.body ? JSON.parse(init.body) : void 0;
-    const agentName = headers[AGENT_NAME_HEADER];
-    const stepName = agentName ? `Call Agent ${agentName}` : "Call Agent";
-    const responseInfo = await context.call(stepName, {
-      url: input.toString(),
-      method: init?.method,
-      headers,
-      body,
-      timeout: agentCallParams?.timeout,
-      retries: agentCallParams?.retries,
-      retryDelay: agentCallParams?.retryDelay,
-      flowControl: agentCallParams?.flowControl
-    });
-    const responseHeaders = new Headers(
-      Object.entries(responseInfo.header).reduce(
-        (acc, [key, values]) => {
-          acc[key] = values.join(", ");
-          return acc;
-        },
-        {}
-      )
-    );
-    return new Response(JSON.stringify(responseInfo.body), {
-      status: responseInfo.status,
-      headers: responseHeaders
-    });
-  } catch (error) {
-    if (error instanceof Error && isInstanceOf(error, WorkflowAbort)) {
-      throw error;
-    } else {
-      console.error("Error in fetch implementation:", error);
-      throw error;
-    }
-  }
-};
-var createWorkflowModel = ({
-  context,
-  provider,
-  providerParams,
-  agentCallParams
-}) => {
-  return provider({
-    fetch: (...params) => fetchWithContextCall(context, agentCallParams, ...params),
-    ...providerParams
-  });
-};
-var wrapTools = ({
-  context,
-  tools
-}) => {
-  return Object.fromEntries(
-    Object.entries(tools).map((toolInfo) => {
-      const [toolName, tool3] = toolInfo;
-      const executeAsStep = "executeAsStep" in tool3 ? tool3.executeAsStep : true;
-      const aiSDKTool = convertToAISDKTool(tool3);
-      const execute = aiSDKTool.execute;
-      if (execute && executeAsStep) {
-        const wrappedExecute = (...params) => {
-          return context.run(`Run tool ${toolName}`, () => execute(...params));
-        };
-        aiSDKTool.execute = wrappedExecute;
-      }
-      return [toolName, aiSDKTool];
-    })
-  );
-};
-var convertToAISDKTool = (tool3) => {
-  const isLangchainTool = "invoke" in tool3;
-  return isLangchainTool ? convertLangchainTool(tool3) : tool3;
-};
-var convertLangchainTool = (langchainTool) => {
-  return (0, import_ai.tool)({
-    description: langchainTool.description,
-    parameters: langchainTool.schema,
-    execute: async (...param) => langchainTool.invoke(...param)
-  });
-};
-var WorkflowTool = class {
-  /**
-   * description of the tool
-   */
-  description;
-  /**
-   * schema of the tool
-   */
-  schema;
-  /**
-   * function to invoke the tool
-   */
-  invoke;
-  /**
-   * whether the invoke method of the tool is to be wrapped with `context.run`
-   */
-  executeAsStep;
-  /**
-   *
-   * @param description description of the tool
-   * @param schema schema of the tool
-   * @param invoke function to invoke the tool
-   * @param executeAsStep whether the invoke method of the tool is to be wrapped with `context.run`
-   */
-  constructor(params) {
-    this.description = params.description;
-    this.schema = params.schema;
-    this.invoke = params.invoke;
-    this.executeAsStep = params.executeAsStep ?? true;
-  }
-};
-
-// src/agents/agent.ts
-var import_zod = require("zod");
-var import_ai2 = require("ai");
-
-// src/serve/utils.ts
-var isDisabledWorkflowContext = (context) => {
-  return "disabled" in context;
-};
-
-// src/agents/agent.ts
-var Agent = class {
-  name;
-  tools;
-  maxSteps;
-  background;
-  model;
-  temparature;
-  context;
-  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }, context) {
-    this.name = name;
-    this.tools = tools ?? {};
-    this.maxSteps = maxSteps;
-    this.background = background;
-    this.model = model;
-    this.temparature = temparature;
-    this.context = context;
-  }
-  /**
-   * Trigger the agent by passing a prompt
-   *
-   * @param prompt task to assign to the agent
-   * @returns Response as `{ text: string }`
-   */
-  async call({ prompt }) {
-    try {
-      if (isDisabledWorkflowContext(this.context)) {
-        await this.context.sleep("abort", 0);
-      }
-      const result = await (0, import_ai2.generateText)({
-        model: this.model,
-        tools: this.tools,
-        maxSteps: this.maxSteps,
-        system: this.background,
-        prompt,
-        headers: {
-          [AGENT_NAME_HEADER]: this.name
-        },
-        temperature: this.temparature
-      });
-      return { text: result.text };
-    } catch (error) {
-      if (isInstanceOf(error, import_ai2.ToolExecutionError)) {
-        if (error.cause instanceof Error && isInstanceOf(error.cause, WorkflowAbort)) {
-          throw error.cause;
-        } else if (isInstanceOf(error.cause, import_ai2.ToolExecutionError) && isInstanceOf(error.cause.cause, WorkflowAbort)) {
-          throw error.cause.cause;
-        } else {
-          throw error;
-        }
-      } else {
-        throw error;
-      }
-    }
-  }
-  /**
-   * Convert the agent to a tool which can be used by other agents.
-   *
-   * @returns the agent as a tool
-   */
-  asTool() {
-    const toolDescriptions = Object.values(this.tools).map((tool3) => tool3.description).join("\n");
-    return (0, import_ai2.tool)({
-      parameters: import_zod.z.object({ prompt: import_zod.z.string() }),
-      execute: async ({ prompt }) => {
-        return await this.call({ prompt });
-      },
-      description: `An AI Agent with the following background: ${this.background}Has access to the following tools: ${toolDescriptions}`
-    });
-  }
-};
-var ManagerAgent = class extends Agent {
-  agents;
-  /**
-   * A manager agent which coordinates agents available to it to achieve a
-   * given task
-   *
-   * @param name Name of the agent
-   * @param background Background of the agent. If not passed, default will be used.
-   * @param model LLM model to use
-   * @param agents: List of agents available to the agent
-   * @param maxSteps number of times the manager agent can call the LLM at most.
-   *   If the agent abruptly stops execution after calling other agents, you may
-   *   need to increase maxSteps
-   */
-  constructor({
-    agents,
-    background = MANAGER_AGENT_PROMPT,
-    model,
-    maxSteps,
-    name = "manager llm"
-  }, context) {
-    super(
-      {
-        background,
-        maxSteps,
-        tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
-        name,
-        model
-      },
-      context
-    );
-    this.agents = agents;
-  }
-};
-
-// src/agents/task.ts
-var Task = class {
-  context;
-  taskParameters;
-  constructor({
-    context,
-    taskParameters
-  }) {
-    this.context = context;
-    this.taskParameters = taskParameters;
-  }
-  /**
-   * Run the agents to complete the task
-   *
-   * @returns Result of the task as { text: string }
-   */
-  async run() {
-    const { prompt, ...otherParams } = this.taskParameters;
-    if ("agent" in otherParams) {
-      const agent = otherParams.agent;
-      const result = await agent.call({
-        prompt
-      });
-      return { text: result.text };
-    } else {
-      const { agents, maxSteps, model, background } = otherParams;
-      const managerAgent = new ManagerAgent(
-        {
-          model,
-          maxSteps,
-          agents,
-          name: "Manager LLM",
-          background
-        },
-        this.context
-      );
-      const result = await managerAgent.call({ prompt });
-      return { text: result.text };
-    }
-  }
-};
-
-// src/agents/index.ts
-var WorkflowAgents = class {
-  context;
-  constructor({ context }) {
-    this.context = context;
-  }
-  /**
-   * Defines an agent
-   *
-   * ```ts
-   * const researcherAgent = context.agents.agent({
-   *   model,
-   *   name: 'academic',
-   *   maxSteps: 2,
-   *   tools: {
-   *     wikiTool: new WikipediaQueryRun({
-   *       topKResults: 1,
-   *       maxDocContentLength: 500,
-   *     })
-   *   },
-   *   background:
-   *     'You are researcher agent with access to Wikipedia. ' +
-   *     'Utilize Wikipedia as much as possible for correct information',
-   * });
-   * ```
-   *
-   * @param params agent parameters
-   * @returns
-   */
-  agent(params) {
-    const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
-    return new Agent(
-      {
-        ...params,
-        tools: wrappedTools
-      },
-      this.context
-    );
-  }
-  task(taskParameters) {
-    return new Task({ context: this.context, taskParameters });
-  }
-  /**
-   * creates an openai model for agents
-   */
-  openai(...params) {
-    const [model, settings] = params;
-    const { baseURL, apiKey, callSettings, ...otherSettings } = settings ?? {};
-    const openaiModel = this.AISDKModel({
-      context: this.context,
-      provider: import_openai2.createOpenAI,
-      providerParams: { baseURL, apiKey, compatibility: "strict" },
-      agentCallParams: callSettings
-    });
-    return openaiModel(model, otherSettings);
-  }
-  AISDKModel = createWorkflowModel;
-};
-
 // src/serve/serve-many.ts
 var getNewUrlFromWorkflowId = (url, workflowId) => {
   if (!workflowId) {
@@ -2904,7 +2553,7 @@ var WorkflowContext = class {
    * @returns result of the step function
    */
   async run(stepName, stepFunction) {
-    const wrappedStepFunction = () => this.executor.wrapStep(stepName, stepFunction);
+    const wrappedStepFunction = (() => this.executor.wrapStep(stepName, stepFunction));
     return await this.addStep(new LazyFunctionStep(stepName, wrappedStepFunction));
   }
   /**
@@ -3075,11 +2724,6 @@ var WorkflowContext = class {
       context: this
     });
   }
-  get agents() {
-    return new WorkflowAgents({
-      context: this
-    });
-  }
 };
 
 // src/logger.ts
@@ -3963,7 +3607,6 @@ var Client4 = class {
       const finalWorkflowRunId = getWorkflowRunId(option.workflowRunId);
       const context = new WorkflowContext({
         qstashClient: this.client,
-        // @ts-expect-error header type mismatch because of bun
         headers: new Headers({
           ...option.headers ?? {},
           ...option.label ? { [WORKFLOW_LABEL_HEADER]: option.label } : {}
@@ -4067,6 +3710,5 @@ var Client4 = class {
   WorkflowLogger,
   WorkflowNonRetryableError,
   WorkflowRetryAfterError,
-  WorkflowTool,
   serve
 });

package/index.mjs

@@ -8,14 +8,13 @@ import {
   WorkflowLogger,
   WorkflowNonRetryableError,
   WorkflowRetryAfterError,
-  WorkflowTool,
   getWorkflowRunId,
   makeGetWaitersRequest,
   makeNotifyRequest,
   prepareFlowControl,
   serve,
   triggerFirstInvocation
-} from "./chunk-BON2RKOR.mjs";
+} from "./chunk-AGYYZKP7.mjs";
 
 // src/client/index.ts
 import { Client as QStashClient } from "@upstash/qstash";
@@ -247,7 +246,6 @@ var Client = class {
       const finalWorkflowRunId = getWorkflowRunId(option.workflowRunId);
       const context = new WorkflowContext({
         qstashClient: this.client,
-        // @ts-expect-error header type mismatch because of bun
         headers: new Headers({
           ...option.headers ?? {},
           ...option.label ? { [WORKFLOW_LABEL_HEADER]: option.label } : {}
@@ -350,6 +348,5 @@ export {
   WorkflowLogger,
   WorkflowNonRetryableError,
   WorkflowRetryAfterError,
-  WorkflowTool,
   serve
 };

package/nextjs.d.mts

@@ -1,10 +1,8 @@
 import { NextApiHandler, NextApiRequest, NextApiResponse } from 'next';
-import { R as RouteFunction, o as PublicServeOptions, z as InvokableWorkflow } from './types-9nCq6bRP.mjs';
-import { s as serveManyBase } from './serve-many-CctdYIfB.mjs';
+import { R as RouteFunction, n as PublicServeOptions, y as InvokableWorkflow } from './types-DESkn7K9.mjs';
+import { s as serveManyBase } from './serve-many-DEwKPF6H.mjs';
 import '@upstash/qstash';
 import 'zod';
-import 'ai';
-import '@ai-sdk/openai';
 
 /**
  * Serve method to serve a Upstash Workflow in a Nextjs project
@@ -18,14 +16,14 @@ import '@ai-sdk/openai';
 declare const serve: <TInitialPayload = unknown, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload>) => {
     POST: (request: Request) => Promise<Response>;
 };
-declare const createWorkflow: <TInitialPayload, TResult>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload> | undefined) => InvokableWorkflow<TInitialPayload, TResult>;
+declare const createWorkflow: <TInitialPayload, TResult>(...params: Parameters<typeof serve<TInitialPayload, TResult>>) => InvokableWorkflow<TInitialPayload, TResult>;
 declare const serveMany: (workflows: Parameters<typeof serveManyBase>[0]["workflows"], options?: Parameters<typeof serveManyBase>[0]["options"]) => {
     POST: (request: Request) => Promise<any>;
 };
 declare const servePagesRouter: <TInitialPayload = unknown, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload>) => {
     handler: NextApiHandler;
 };
-declare const createWorkflowPagesRouter: <TInitialPayload, TResult>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload> | undefined) => InvokableWorkflow<TInitialPayload, TResult>;
+declare const createWorkflowPagesRouter: <TInitialPayload, TResult>(...params: Parameters<typeof servePagesRouter<TInitialPayload, TResult>>) => InvokableWorkflow<TInitialPayload, TResult>;
 declare const serveManyPagesRouter: (workflows: Parameters<typeof serveManyBase>[0]["workflows"], options?: Parameters<typeof serveManyBase>[0]["options"]) => {
     handler: (req: NextApiRequest, res: NextApiResponse<any>) => Promise<any>;
 };

package/nextjs.d.ts

@@ -1,10 +1,8 @@
 import { NextApiHandler, NextApiRequest, NextApiResponse } from 'next';
-import { R as RouteFunction, o as PublicServeOptions, z as InvokableWorkflow } from './types-9nCq6bRP.js';
-import { s as serveManyBase } from './serve-many-BXDr30rl.js';
+import { R as RouteFunction, n as PublicServeOptions, y as InvokableWorkflow } from './types-DESkn7K9.js';
+import { s as serveManyBase } from './serve-many-DVtHRxeg.js';
 import '@upstash/qstash';
 import 'zod';
-import 'ai';
-import '@ai-sdk/openai';
 
 /**
  * Serve method to serve a Upstash Workflow in a Nextjs project
@@ -18,14 +16,14 @@ import '@ai-sdk/openai';
 declare const serve: <TInitialPayload = unknown, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload>) => {
     POST: (request: Request) => Promise<Response>;
 };
-declare const createWorkflow: <TInitialPayload, TResult>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload> | undefined) => InvokableWorkflow<TInitialPayload, TResult>;
+declare const createWorkflow: <TInitialPayload, TResult>(...params: Parameters<typeof serve<TInitialPayload, TResult>>) => InvokableWorkflow<TInitialPayload, TResult>;
 declare const serveMany: (workflows: Parameters<typeof serveManyBase>[0]["workflows"], options?: Parameters<typeof serveManyBase>[0]["options"]) => {
     POST: (request: Request) => Promise<any>;
 };
 declare const servePagesRouter: <TInitialPayload = unknown, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload>) => {
     handler: NextApiHandler;
 };
-declare const createWorkflowPagesRouter: <TInitialPayload, TResult>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload> | undefined) => InvokableWorkflow<TInitialPayload, TResult>;
+declare const createWorkflowPagesRouter: <TInitialPayload, TResult>(...params: Parameters<typeof servePagesRouter<TInitialPayload, TResult>>) => InvokableWorkflow<TInitialPayload, TResult>;
 declare const serveManyPagesRouter: (workflows: Parameters<typeof serveManyBase>[0]["workflows"], options?: Parameters<typeof serveManyBase>[0]["options"]) => {
     handler: (req: NextApiRequest, res: NextApiResponse<any>) => Promise<any>;
 };