@upstash/workflow 0.2.5-agents-2 → 0.2.5

package/solidjs.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/solidjs.mjs

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

package/svelte.d.mts

@@ -1,5 +1,5 @@
 import { RequestHandler } from '@sveltejs/kit';
-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/svelte.d.ts

@@ -1,5 +1,5 @@
 import { RequestHandler } from '@sveltejs/kit';
-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/svelte.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/svelte.mjs

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

package/{types-D9gwTj2n.d.mts → types-CalpUeFX.d.mts}

@@ -410,7 +410,8 @@ declare class Agent {
     readonly background: AgentParameters["background"];
     readonly model: AgentParameters["model"];
     readonly temparature: AgentParameters["temparature"];
-    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters);
+    private readonly context;
+    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters, context: WorkflowContext);
     /**
      * Trigger the agent by passing a prompt
      *

package/{types-D9gwTj2n.d.ts → types-CalpUeFX.d.ts}

@@ -410,7 +410,8 @@ declare class Agent {
     readonly background: AgentParameters["background"];
     readonly model: AgentParameters["model"];
     readonly temparature: AgentParameters["temparature"];
-    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters);
+    private readonly context;
+    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters, context: WorkflowContext);
     /**
      * Trigger the agent by passing a prompt
      *