@upstash/workflow 0.2.4 → 0.2.5-agents

package/nextjs.js

@@ -1633,6 +1633,216 @@ var WorkflowApi = class extends BaseWorkflowApi {
   }
 };
 
+// src/agents/adapters.ts
+var import_openai2 = require("@ai-sdk/openai");
+var import_ai = require("ai");
+var AGENT_NAME_HEADER = "upstash-agent-name";
+var createWorkflowOpenAI = (context) => {
+  return (0, import_openai2.createOpenAI)({
+    compatibility: "strict",
+    fetch: async (input, init) => {
+      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
+        });
+        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 && error.name === "WorkflowAbort") {
+          throw error;
+        } else {
+          console.error("Error in fetch implementation:", error);
+          throw error;
+        }
+      }
+    }
+  });
+};
+var wrapTools = ({
+  context,
+  tools
+}) => {
+  return Object.fromEntries(
+    Object.entries(tools).map((toolInfo) => {
+      const [toolName, tool3] = toolInfo;
+      const aiSDKTool = convertToAISDKTool(tool3);
+      const execute = aiSDKTool.execute;
+      if (execute) {
+        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,
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    execute: async (param) => langchainTool.invoke(param)
+  });
+};
+
+// src/agents/agent.ts
+var import_zod = require("zod");
+var import_ai2 = require("ai");
+var Agent = class {
+  name;
+  tools;
+  maxSteps;
+  background;
+  model;
+  constructor({ tools, maxSteps, background, name, model }) {
+    this.name = name;
+    this.tools = tools ?? {};
+    this.maxSteps = maxSteps;
+    this.background = background;
+    this.model = model;
+  }
+  async call({ prompt }) {
+    try {
+      return await (0, import_ai2.generateText)({
+        model: this.model,
+        tools: this.tools,
+        maxSteps: this.maxSteps,
+        system: this.background,
+        prompt,
+        headers: {
+          [AGENT_NAME_HEADER]: this.name
+        }
+      });
+    } catch (error) {
+      if (error instanceof import_ai2.ToolExecutionError) {
+        if (error.cause instanceof Error && error.cause.name === "WorkflowAbort") {
+          throw error.cause;
+        } else if (error.cause instanceof import_ai2.ToolExecutionError && error.cause.cause instanceof Error && error.cause.cause.name === "WorkflowAbort") {
+          throw error.cause.cause;
+        } else {
+          throw error;
+        }
+      } else {
+        throw error;
+      }
+    }
+  }
+  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 MANAGER_AGENT_PROMPT = `You are an AI agent who orchestrates 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. 
+`;
+var ManagerAgent = class extends Agent {
+  agents;
+  constructor({
+    maxSteps,
+    background = MANAGER_AGENT_PROMPT,
+    agents,
+    model,
+    name = "manager llm"
+  }) {
+    super({
+      background,
+      maxSteps,
+      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+      name,
+      model
+    });
+    this.agents = agents;
+  }
+};
+
+// src/agents/task.ts
+var Task = class {
+  context;
+  taskParameters;
+  constructor({
+    context,
+    taskParameters
+  }) {
+    this.context = context;
+    this.taskParameters = taskParameters;
+  }
+  async run() {
+    const { prompt, ...otherParams } = this.taskParameters;
+    const safePrompt = await this.context.run("Get Prompt", () => prompt);
+    if ("agent" in otherParams) {
+      const agent = otherParams.agent;
+      const result = await agent.call({
+        prompt: safePrompt
+      });
+      return { text: result.text };
+    } else {
+      const { agents, maxSteps, model, background } = otherParams;
+      const managerAgent = new ManagerAgent({
+        model,
+        maxSteps,
+        agents,
+        name: "Manager LLM",
+        background
+      });
+      const result = await managerAgent.call({ prompt: safePrompt });
+      return { text: result.text };
+    }
+  }
+};
+
+// src/agents/index.ts
+var WorkflowAgents = class {
+  context;
+  constructor({ context }) {
+    this.context = context;
+  }
+  agent(params) {
+    const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
+    return new Agent({
+      ...params,
+      tools: wrappedTools
+    });
+  }
+  task(taskParameters) {
+    return new Task({ context: this.context, taskParameters });
+  }
+  openai(...params) {
+    const openai2 = createWorkflowOpenAI(this.context);
+    return openai2(...params);
+  }
+};
+
 // src/context/context.ts
 var WorkflowContext = class {
   executor;
@@ -2018,6 +2228,11 @@ var WorkflowContext = class {
       context: this
     });
   }
+  get agents() {
+    return new WorkflowAgents({
+      context: this
+    });
+  }
 };
 
 // src/logger.ts
@@ -2222,6 +2437,7 @@ var checkIfLastOneIsDuplicate = async (steps, debug) => {
     if (step.stepId === lastStepId && step.targetStep === lastTargetStepId) {
       const message = `Upstash Workflow: The step '${step.stepName}' with id '${step.stepId}'  has run twice during workflow execution. Rest of the workflow will continue running as usual.`;
       await debug?.log("WARN", "RESPONSE_DEFAULT", message);
+      console.log(steps);
       console.warn(message);
       return true;
     }

package/nextjs.mjs

@@ -1,7 +1,8 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-ETDFMXER.mjs";
+} from "./chunk-RFX5YRRT.mjs";
+import "./chunk-PU5J4TNC.mjs";
 
 // platforms/nextjs.ts
 var serve = (routeFunction, options) => {

package/package.json

@@ -1 +1 @@
-{"name":"@upstash/workflow","version":"v0.2.4","description":"Durable, Reliable and Performant Serverless Functions","main":"./index.js","module":"./index.mjs","types":"./index.d.ts","files":["./*"],"exports":{".":{"import":"./index.mjs","require":"./index.js"},"./dist/nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./h3":{"import":"./h3.mjs","require":"./h3.js"},"./svelte":{"import":"./svelte.mjs","require":"./svelte.js"},"./solidjs":{"import":"./solidjs.mjs","require":"./solidjs.js"},"./workflow":{"import":"./workflow.mjs","require":"./workflow.js"},"./hono":{"import":"./hono.mjs","require":"./hono.js"},"./cloudflare":{"import":"./cloudflare.mjs","require":"./cloudflare.js"},"./astro":{"import":"./astro.mjs","require":"./astro.js"},"./express":{"import":"./express.mjs","require":"./express.js"}},"scripts":{"build":"tsup && cp README.md ./dist/ && cp package.json ./dist/ && cp LICENSE ./dist/","test":"bun test src","fmt":"prettier --write .","lint":"tsc && eslint \"{src,platforms}/**/*.{js,ts,tsx}\" --quiet --fix","check-exports":"bun run build && cd dist && attw -P"},"repository":{"type":"git","url":"git+https://github.com/upstash/workflow-ts.git"},"keywords":["upstash","qstash","workflow","serverless"],"author":"Cahid Arda Oz","license":"MIT","bugs":{"url":"https://github.com/upstash/workflow-ts/issues"},"homepage":"https://github.com/upstash/workflow-ts#readme","devDependencies":{"@commitlint/cli":"^19.5.0","@commitlint/config-conventional":"^19.5.0","@eslint/js":"^9.11.1","@solidjs/start":"^1.0.8","@sveltejs/kit":"^2.6.1","@types/bun":"^1.1.10","@types/express":"^5.0.0","astro":"^4.16.7","eslint":"^9.11.1","eslint-plugin-unicorn":"^55.0.0","express":"^4.21.1","globals":"^15.10.0","h3":"^1.12.0","hono":"^4.6.3","husky":"^9.1.6","next":"^14.2.14","prettier":"3.3.3","tsup":"^8.3.0","typescript":"^5.7.2","typescript-eslint":"^8.18.0"},"dependencies":{"@upstash/qstash":"^2.7.20"},"directories":{"example":"examples"}}
+{"name":"@upstash/workflow","version":"v0.2.5-agents","description":"Durable, Reliable and Performant Serverless Functions","main":"./index.js","module":"./index.mjs","types":"./index.d.ts","files":["./*"],"exports":{".":{"import":"./index.mjs","require":"./index.js"},"./dist/nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./h3":{"import":"./h3.mjs","require":"./h3.js"},"./svelte":{"import":"./svelte.mjs","require":"./svelte.js"},"./solidjs":{"import":"./solidjs.mjs","require":"./solidjs.js"},"./workflow":{"import":"./workflow.mjs","require":"./workflow.js"},"./hono":{"import":"./hono.mjs","require":"./hono.js"},"./cloudflare":{"import":"./cloudflare.mjs","require":"./cloudflare.js"},"./astro":{"import":"./astro.mjs","require":"./astro.js"},"./express":{"import":"./express.mjs","require":"./express.js"},"./agents":{"import":"./agents.mjs","require":"./agents.js"}},"scripts":{"build":"tsup && cp README.md ./dist/ && cp package.json ./dist/ && cp LICENSE ./dist/","test":"bun test src","fmt":"prettier --write .","lint":"tsc && eslint \"{src,platforms}/**/*.{js,ts,tsx}\" --quiet --fix","check-exports":"bun run build && cd dist && attw -P"},"repository":{"type":"git","url":"git+https://github.com/upstash/workflow-ts.git"},"keywords":["upstash","qstash","workflow","serverless"],"author":"Cahid Arda Oz","license":"MIT","bugs":{"url":"https://github.com/upstash/workflow-ts/issues"},"homepage":"https://github.com/upstash/workflow-ts#readme","devDependencies":{"@commitlint/cli":"^19.5.0","@commitlint/config-conventional":"^19.5.0","@eslint/js":"^9.11.1","@solidjs/start":"^1.0.8","@sveltejs/kit":"^2.6.1","@types/bun":"^1.1.10","@types/express":"^5.0.0","astro":"^4.16.7","eslint":"^9.11.1","eslint-plugin-unicorn":"^55.0.0","express":"^4.21.1","globals":"^15.10.0","h3":"^1.12.0","hono":"^4.6.3","husky":"^9.1.6","next":"^14.2.14","prettier":"3.3.3","tsup":"^8.3.0","typescript":"^5.7.2","typescript-eslint":"^8.18.0"},"dependencies":{"@ai-sdk/openai":"^1.0.15","@upstash/qstash":"^2.7.20","ai":"^4.0.30","langchain":"^0.3.11","zod":"^3.24.1"},"directories":{"example":"examples"}}

package/solidjs.d.mts

@@ -1,6 +1,9 @@
 import { APIEvent } from '@solidjs/start/server';
-import { R as RouteFunction, j as PublicServeOptions } from './types-Bt4-paRy.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-BEyIoCRe.mjs';
 import '@upstash/qstash';
+import 'ai';
+import '@ai-sdk/openai';
+import 'langchain/tools';
 
 /**
  * Serve method to serve a Upstash Workflow in a Nextjs project

package/solidjs.d.ts

@@ -1,6 +1,9 @@
 import { APIEvent } from '@solidjs/start/server';
-import { R as RouteFunction, j as PublicServeOptions } from './types-Bt4-paRy.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-BEyIoCRe.js';
 import '@upstash/qstash';
+import 'ai';
+import '@ai-sdk/openai';
+import 'langchain/tools';
 
 /**
  * Serve method to serve a Upstash Workflow in a Nextjs project

package/solidjs.js

@@ -1632,6 +1632,216 @@ var WorkflowApi = class extends BaseWorkflowApi {
   }
 };
 
+// src/agents/adapters.ts
+var import_openai2 = require("@ai-sdk/openai");
+var import_ai = require("ai");
+var AGENT_NAME_HEADER = "upstash-agent-name";
+var createWorkflowOpenAI = (context) => {
+  return (0, import_openai2.createOpenAI)({
+    compatibility: "strict",
+    fetch: async (input, init) => {
+      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
+        });
+        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 && error.name === "WorkflowAbort") {
+          throw error;
+        } else {
+          console.error("Error in fetch implementation:", error);
+          throw error;
+        }
+      }
+    }
+  });
+};
+var wrapTools = ({
+  context,
+  tools
+}) => {
+  return Object.fromEntries(
+    Object.entries(tools).map((toolInfo) => {
+      const [toolName, tool3] = toolInfo;
+      const aiSDKTool = convertToAISDKTool(tool3);
+      const execute = aiSDKTool.execute;
+      if (execute) {
+        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,
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    execute: async (param) => langchainTool.invoke(param)
+  });
+};
+
+// src/agents/agent.ts
+var import_zod = require("zod");
+var import_ai2 = require("ai");
+var Agent = class {
+  name;
+  tools;
+  maxSteps;
+  background;
+  model;
+  constructor({ tools, maxSteps, background, name, model }) {
+    this.name = name;
+    this.tools = tools ?? {};
+    this.maxSteps = maxSteps;
+    this.background = background;
+    this.model = model;
+  }
+  async call({ prompt }) {
+    try {
+      return await (0, import_ai2.generateText)({
+        model: this.model,
+        tools: this.tools,
+        maxSteps: this.maxSteps,
+        system: this.background,
+        prompt,
+        headers: {
+          [AGENT_NAME_HEADER]: this.name
+        }
+      });
+    } catch (error) {
+      if (error instanceof import_ai2.ToolExecutionError) {
+        if (error.cause instanceof Error && error.cause.name === "WorkflowAbort") {
+          throw error.cause;
+        } else if (error.cause instanceof import_ai2.ToolExecutionError && error.cause.cause instanceof Error && error.cause.cause.name === "WorkflowAbort") {
+          throw error.cause.cause;
+        } else {
+          throw error;
+        }
+      } else {
+        throw error;
+      }
+    }
+  }
+  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 MANAGER_AGENT_PROMPT = `You are an AI agent who orchestrates 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. 
+`;
+var ManagerAgent = class extends Agent {
+  agents;
+  constructor({
+    maxSteps,
+    background = MANAGER_AGENT_PROMPT,
+    agents,
+    model,
+    name = "manager llm"
+  }) {
+    super({
+      background,
+      maxSteps,
+      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+      name,
+      model
+    });
+    this.agents = agents;
+  }
+};
+
+// src/agents/task.ts
+var Task = class {
+  context;
+  taskParameters;
+  constructor({
+    context,
+    taskParameters
+  }) {
+    this.context = context;
+    this.taskParameters = taskParameters;
+  }
+  async run() {
+    const { prompt, ...otherParams } = this.taskParameters;
+    const safePrompt = await this.context.run("Get Prompt", () => prompt);
+    if ("agent" in otherParams) {
+      const agent = otherParams.agent;
+      const result = await agent.call({
+        prompt: safePrompt
+      });
+      return { text: result.text };
+    } else {
+      const { agents, maxSteps, model, background } = otherParams;
+      const managerAgent = new ManagerAgent({
+        model,
+        maxSteps,
+        agents,
+        name: "Manager LLM",
+        background
+      });
+      const result = await managerAgent.call({ prompt: safePrompt });
+      return { text: result.text };
+    }
+  }
+};
+
+// src/agents/index.ts
+var WorkflowAgents = class {
+  context;
+  constructor({ context }) {
+    this.context = context;
+  }
+  agent(params) {
+    const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
+    return new Agent({
+      ...params,
+      tools: wrappedTools
+    });
+  }
+  task(taskParameters) {
+    return new Task({ context: this.context, taskParameters });
+  }
+  openai(...params) {
+    const openai2 = createWorkflowOpenAI(this.context);
+    return openai2(...params);
+  }
+};
+
 // src/context/context.ts
 var WorkflowContext = class {
   executor;
@@ -2017,6 +2227,11 @@ var WorkflowContext = class {
       context: this
     });
   }
+  get agents() {
+    return new WorkflowAgents({
+      context: this
+    });
+  }
 };
 
 // src/logger.ts
@@ -2221,6 +2436,7 @@ var checkIfLastOneIsDuplicate = async (steps, debug) => {
     if (step.stepId === lastStepId && step.targetStep === lastTargetStepId) {
       const message = `Upstash Workflow: The step '${step.stepName}' with id '${step.stepId}'  has run twice during workflow execution. Rest of the workflow will continue running as usual.`;
       await debug?.log("WARN", "RESPONSE_DEFAULT", message);
+      console.log(steps);
       console.warn(message);
       return true;
     }

package/solidjs.mjs

@@ -1,7 +1,8 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-ETDFMXER.mjs";
+} from "./chunk-RFX5YRRT.mjs";
+import "./chunk-PU5J4TNC.mjs";
 
 // platforms/solidjs.ts
 var serve = (routeFunction, options) => {

package/svelte.d.mts

@@ -1,6 +1,9 @@
 import { RequestHandler } from '@sveltejs/kit';
-import { R as RouteFunction, j as PublicServeOptions } from './types-Bt4-paRy.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-BEyIoCRe.mjs';
 import '@upstash/qstash';
+import 'ai';
+import '@ai-sdk/openai';
+import 'langchain/tools';
 
 /**
  * Serve method to serve a Upstash Workflow in a Nextjs project

package/svelte.d.ts

@@ -1,6 +1,9 @@
 import { RequestHandler } from '@sveltejs/kit';
-import { R as RouteFunction, j as PublicServeOptions } from './types-Bt4-paRy.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-BEyIoCRe.js';
 import '@upstash/qstash';
+import 'ai';
+import '@ai-sdk/openai';
+import 'langchain/tools';
 
 /**
  * Serve method to serve a Upstash Workflow in a Nextjs project