@upstash/workflow 0.2.5-agents-2 → 0.2.5

package/astro.d.mts

@@ -1,5 +1,5 @@
 import { APIContext, APIRoute } from 'astro';
-import { b as WorkflowContext, j as PublicServeOptions } from './types-D9gwTj2n.mjs';
+import { b as WorkflowContext, j as PublicServeOptions } from './types-CalpUeFX.mjs';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';

package/astro.d.ts

@@ -1,5 +1,5 @@
 import { APIContext, APIRoute } from 'astro';
-import { b as WorkflowContext, j as PublicServeOptions } from './types-D9gwTj2n.js';
+import { b as WorkflowContext, j as PublicServeOptions } from './types-CalpUeFX.js';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';

package/astro.js

@@ -1711,6 +1711,13 @@ var convertLangchainTool = (langchainTool) => {
 // 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;
@@ -1718,13 +1725,15 @@ var Agent = class {
   background;
   model;
   temparature;
-  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }) {
+  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
@@ -1734,6 +1743,9 @@ var Agent = class {
    */
   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,
@@ -1796,14 +1808,17 @@ var ManagerAgent = class extends Agent {
     model,
     maxSteps,
     name = "manager llm"
-  }) {
-    super({
-      background,
-      maxSteps,
-      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
-      name,
-      model
-    });
+  }, context) {
+    super(
+      {
+        background,
+        maxSteps,
+        tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+        name,
+        model
+      },
+      context
+    );
     this.agents = agents;
   }
 };
@@ -1826,23 +1841,25 @@ var Task = class {
    */
   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
+        prompt
       });
       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 });
+      const managerAgent = new ManagerAgent(
+        {
+          model,
+          maxSteps,
+          agents,
+          name: "Manager LLM",
+          background
+        },
+        this.context
+      );
+      const result = await managerAgent.call({ prompt });
       return { text: result.text };
     }
   }
@@ -1879,10 +1896,13 @@ var WorkflowAgents = class {
    */
   agent(params) {
     const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
-    return new Agent({
-      ...params,
-      tools: wrappedTools
-    });
+    return new Agent(
+      {
+        ...params,
+        tools: wrappedTools
+      },
+      this.context
+    );
   }
   task(taskParameters) {
     return new Task({ context: this.context, taskParameters });
@@ -2366,6 +2386,7 @@ function decodeBase64(base64) {
 var import_qstash8 = require("@upstash/qstash");
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
   static disabledMessage = "disabled-qstash-worklfow-run";
+  disabled = true;
   /**
    * overwrite the WorkflowContext.addStep method to always raise WorkflowAbort
    * error in order to stop the execution whenever we encounter a step.

package/astro.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-VOM3CFYZ.mjs";
+} from "./chunk-42MM2EPQ.mjs";
 
 // platforms/astro.ts
 function serve(routeFunction, options) {

package/{chunk-VOM3CFYZ.mjs → chunk-42MM2EPQ.mjs}

@@ -1724,6 +1724,13 @@ var convertLangchainTool = (langchainTool) => {
 // src/agents/agent.ts
 import { z } from "zod";
 import { generateText, tool as tool2, ToolExecutionError } from "ai";
+
+// src/serve/utils.ts
+var isDisabledWorkflowContext = (context) => {
+  return "disabled" in context;
+};
+
+// src/agents/agent.ts
 var Agent = class {
   name;
   tools;
@@ -1731,13 +1738,15 @@ var Agent = class {
   background;
   model;
   temparature;
-  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }) {
+  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
@@ -1747,6 +1756,9 @@ var Agent = class {
    */
   async call({ prompt }) {
     try {
+      if (isDisabledWorkflowContext(this.context)) {
+        await this.context.sleep("abort", 0);
+      }
       const result = await generateText({
         model: this.model,
         tools: this.tools,
@@ -1809,14 +1821,17 @@ var ManagerAgent = class extends Agent {
     model,
     maxSteps,
     name = "manager llm"
-  }) {
-    super({
-      background,
-      maxSteps,
-      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
-      name,
-      model
-    });
+  }, context) {
+    super(
+      {
+        background,
+        maxSteps,
+        tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+        name,
+        model
+      },
+      context
+    );
     this.agents = agents;
   }
 };
@@ -1839,23 +1854,25 @@ var Task = class {
    */
   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
+        prompt
       });
       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 });
+      const managerAgent = new ManagerAgent(
+        {
+          model,
+          maxSteps,
+          agents,
+          name: "Manager LLM",
+          background
+        },
+        this.context
+      );
+      const result = await managerAgent.call({ prompt });
       return { text: result.text };
     }
   }
@@ -1892,10 +1909,13 @@ var WorkflowAgents = class {
    */
   agent(params) {
     const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
-    return new Agent({
-      ...params,
-      tools: wrappedTools
-    });
+    return new Agent(
+      {
+        ...params,
+        tools: wrappedTools
+      },
+      this.context
+    );
   }
   task(taskParameters) {
     return new Task({ context: this.context, taskParameters });
@@ -2379,6 +2399,7 @@ function decodeBase64(base64) {
 import { Client as Client2 } from "@upstash/qstash";
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
   static disabledMessage = "disabled-qstash-worklfow-run";
+  disabled = true;
   /**
    * overwrite the WorkflowContext.addStep method to always raise WorkflowAbort
    * error in order to stop the execution whenever we encounter a step.

package/cloudflare.d.mts

@@ -1,4 +1,4 @@
-import { R as RouteFunction, j as PublicServeOptions } from './types-D9gwTj2n.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';

package/cloudflare.d.ts

@@ -1,4 +1,4 @@
-import { R as RouteFunction, j as PublicServeOptions } from './types-D9gwTj2n.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';

package/cloudflare.js

@@ -1711,6 +1711,13 @@ var convertLangchainTool = (langchainTool) => {
 // 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;
@@ -1718,13 +1725,15 @@ var Agent = class {
   background;
   model;
   temparature;
-  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }) {
+  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
@@ -1734,6 +1743,9 @@ var Agent = class {
    */
   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,
@@ -1796,14 +1808,17 @@ var ManagerAgent = class extends Agent {
     model,
     maxSteps,
     name = "manager llm"
-  }) {
-    super({
-      background,
-      maxSteps,
-      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
-      name,
-      model
-    });
+  }, context) {
+    super(
+      {
+        background,
+        maxSteps,
+        tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+        name,
+        model
+      },
+      context
+    );
     this.agents = agents;
   }
 };
@@ -1826,23 +1841,25 @@ var Task = class {
    */
   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
+        prompt
       });
       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 });
+      const managerAgent = new ManagerAgent(
+        {
+          model,
+          maxSteps,
+          agents,
+          name: "Manager LLM",
+          background
+        },
+        this.context
+      );
+      const result = await managerAgent.call({ prompt });
       return { text: result.text };
     }
   }
@@ -1879,10 +1896,13 @@ var WorkflowAgents = class {
    */
   agent(params) {
     const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
-    return new Agent({
-      ...params,
-      tools: wrappedTools
-    });
+    return new Agent(
+      {
+        ...params,
+        tools: wrappedTools
+      },
+      this.context
+    );
   }
   task(taskParameters) {
     return new Task({ context: this.context, taskParameters });
@@ -2366,6 +2386,7 @@ function decodeBase64(base64) {
 var import_qstash8 = require("@upstash/qstash");
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
   static disabledMessage = "disabled-qstash-worklfow-run";
+  disabled = true;
   /**
    * overwrite the WorkflowContext.addStep method to always raise WorkflowAbort
    * error in order to stop the execution whenever we encounter a step.

package/cloudflare.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-VOM3CFYZ.mjs";
+} from "./chunk-42MM2EPQ.mjs";
 
 // platforms/cloudflare.ts
 var getArgs = (args) => {

package/express.d.mts

@@ -1,4 +1,4 @@
-import { R as RouteFunction, W as WorkflowServeOptions } from './types-D9gwTj2n.mjs';
+import { R as RouteFunction, W as WorkflowServeOptions } from './types-CalpUeFX.mjs';
 import { Router } from 'express';
 import '@upstash/qstash';
 import 'ai';

package/express.d.ts

@@ -1,4 +1,4 @@
-import { R as RouteFunction, W as WorkflowServeOptions } from './types-D9gwTj2n.js';
+import { R as RouteFunction, W as WorkflowServeOptions } from './types-CalpUeFX.js';
 import { Router } from 'express';
 import '@upstash/qstash';
 import 'ai';

package/express.js

@@ -25383,6 +25383,13 @@ var convertLangchainTool = (langchainTool) => {
 // 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;
@@ -25390,13 +25397,15 @@ var Agent = class {
   background;
   model;
   temparature;
-  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }) {
+  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
@@ -25406,6 +25415,9 @@ var Agent = class {
    */
   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,
@@ -25468,14 +25480,17 @@ var ManagerAgent = class extends Agent {
     model,
     maxSteps,
     name = "manager llm"
-  }) {
-    super({
-      background,
-      maxSteps,
-      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
-      name,
-      model
-    });
+  }, context) {
+    super(
+      {
+        background,
+        maxSteps,
+        tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+        name,
+        model
+      },
+      context
+    );
     this.agents = agents;
   }
 };
@@ -25498,23 +25513,25 @@ var Task = class {
    */
   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
+        prompt
       });
       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 });
+      const managerAgent = new ManagerAgent(
+        {
+          model,
+          maxSteps,
+          agents,
+          name: "Manager LLM",
+          background
+        },
+        this.context
+      );
+      const result = await managerAgent.call({ prompt });
       return { text: result.text };
     }
   }
@@ -25551,10 +25568,13 @@ var WorkflowAgents = class {
    */
   agent(params) {
     const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
-    return new Agent({
-      ...params,
-      tools: wrappedTools
-    });
+    return new Agent(
+      {
+        ...params,
+        tools: wrappedTools
+      },
+      this.context
+    );
   }
   task(taskParameters) {
     return new Task({ context: this.context, taskParameters });
@@ -26038,6 +26058,7 @@ function decodeBase64(base64) {
 var import_qstash8 = require("@upstash/qstash");
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
   static disabledMessage = "disabled-qstash-worklfow-run";
+  disabled = true;
   /**
    * overwrite the WorkflowContext.addStep method to always raise WorkflowAbort
    * error in order to stop the execution whenever we encounter a step.

package/express.mjs

@@ -4,7 +4,7 @@ import {
   __require,
   __toESM,
   serveBase
-} from "./chunk-VOM3CFYZ.mjs";
+} from "./chunk-42MM2EPQ.mjs";
 
 // node_modules/depd/index.js
 var require_depd = __commonJS({

package/h3.d.mts

@@ -1,5 +1,5 @@
 import * as h3 from 'h3';
-import { R as RouteFunction, j as PublicServeOptions } from './types-D9gwTj2n.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';

package/h3.d.ts

@@ -1,5 +1,5 @@
 import * as h3 from 'h3';
-import { R as RouteFunction, j as PublicServeOptions } from './types-D9gwTj2n.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';