@upstash/workflow 0.2.5-agents → 0.2.5-agents-2

package/h3.js

@@ -1138,29 +1138,16 @@ var triggerWorkflowDelete = async (workflowContext, debug, cancel = false) => {
   await debug?.log("SUBMIT", "SUBMIT_CLEANUP", {
     deletedWorkflowRunId: workflowContext.workflowRunId
   });
-  try {
-    await workflowContext.qstashClient.http.request({
-      path: ["v2", "workflows", "runs", `${workflowContext.workflowRunId}?cancel=${cancel}`],
-      method: "DELETE",
-      parseResponseAsJson: false
-    });
-    await debug?.log(
-      "SUBMIT",
-      "SUBMIT_CLEANUP",
-      `workflow run ${workflowContext.workflowRunId} deleted.`
-    );
-    return { deleted: true };
-  } catch (error) {
-    if (error instanceof import_qstash3.QstashError && error.status === 404) {
-      await debug?.log("WARN", "SUBMIT_CLEANUP", {
-        message: `Failed to remove workflow run ${workflowContext.workflowRunId} as it doesn't exist.`,
-        name: error.name,
-        errorMessage: error.message
-      });
-      return { deleted: false };
-    }
-    throw error;
-  }
+  await workflowContext.qstashClient.http.request({
+    path: ["v2", "workflows", "runs", `${workflowContext.workflowRunId}?cancel=${cancel}`],
+    method: "DELETE",
+    parseResponseAsJson: false
+  });
+  await debug?.log(
+    "SUBMIT",
+    "SUBMIT_CLEANUP",
+    `workflow run ${workflowContext.workflowRunId} deleted.`
+  );
 };
 var recreateUserHeaders = (headers) => {
   const filteredHeaders = new Headers();
@@ -1947,7 +1934,22 @@ var WorkflowApi = class extends BaseWorkflowApi {
 // src/agents/adapters.ts
 var import_openai2 = require("@ai-sdk/openai");
 var import_ai = require("ai");
+
+// 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/agents/adapters.ts
 var createWorkflowOpenAI = (context) => {
   return (0, import_openai2.createOpenAI)({
     compatibility: "strict",
@@ -2014,8 +2016,7 @@ 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)
+    execute: async (...param) => langchainTool.invoke(...param)
   });
 };
 
@@ -2028,16 +2029,24 @@ var Agent = class {
   maxSteps;
   background;
   model;
-  constructor({ tools, maxSteps, background, name, model }) {
+  temparature;
+  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }) {
     this.name = name;
     this.tools = tools ?? {};
     this.maxSteps = maxSteps;
     this.background = background;
     this.model = model;
+    this.temparature = temparature;
   }
+  /**
+   * Trigger the agent by passing a prompt
+   *
+   * @param prompt task to assign to the agent
+   * @returns Response as `{ text: string }`
+   */
   async call({ prompt }) {
     try {
-      return await (0, import_ai2.generateText)({
+      const result = await (0, import_ai2.generateText)({
         model: this.model,
         tools: this.tools,
         maxSteps: this.maxSteps,
@@ -2045,8 +2054,10 @@ var Agent = class {
         prompt,
         headers: {
           [AGENT_NAME_HEADER]: this.name
-        }
+        },
+        temperature: this.temparature
       });
+      return { text: result.text };
     } catch (error) {
       if (error instanceof import_ai2.ToolExecutionError) {
         if (error.cause instanceof Error && error.cause.name === "WorkflowAbort") {
@@ -2061,6 +2072,11 @@ var Agent = class {
       }
     }
   }
+  /**
+   * 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)({
@@ -2072,18 +2088,25 @@ var Agent = class {
     });
   }
 };
-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;
+  /**
+   * 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({
-    maxSteps,
-    background = MANAGER_AGENT_PROMPT,
     agents,
+    background = MANAGER_AGENT_PROMPT,
     model,
+    maxSteps,
     name = "manager llm"
   }) {
     super({
@@ -2108,6 +2131,11 @@ var Task = class {
     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;
     const safePrompt = await this.context.run("Get Prompt", () => prompt);
@@ -2138,6 +2166,29 @@ var WorkflowAgents = class {
   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({
@@ -2148,6 +2199,9 @@ var WorkflowAgents = class {
   task(taskParameters) {
     return new Task({ context: this.context, taskParameters });
   }
+  /**
+   * creates an openai model for agents
+   */
   openai(...params) {
     const openai2 = createWorkflowOpenAI(this.context);
     return openai2(...params);
@@ -2748,7 +2802,6 @@ 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/h3.mjs

@@ -1,8 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-RFX5YRRT.mjs";
-import "./chunk-PU5J4TNC.mjs";
+} from "./chunk-VOM3CFYZ.mjs";
 
 // node_modules/defu/dist/defu.mjs
 function isPlainObject(value) {

package/hono.d.mts

@@ -1,10 +1,9 @@
 import { Context } from 'hono';
-import { R as RouteFunction, j as PublicServeOptions } from './types-BEyIoCRe.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-D9gwTj2n.mjs';
 import { Variables } from 'hono/types';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';
-import 'langchain/tools';
 
 type WorkflowBindings = {
     QSTASH_TOKEN: string;

package/hono.d.ts

@@ -1,10 +1,9 @@
 import { Context } from 'hono';
-import { R as RouteFunction, j as PublicServeOptions } from './types-BEyIoCRe.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-D9gwTj2n.js';
 import { Variables } from 'hono/types';
 import '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';
-import 'langchain/tools';
 
 type WorkflowBindings = {
     QSTASH_TOKEN: string;

package/hono.js

@@ -826,29 +826,16 @@ var triggerWorkflowDelete = async (workflowContext, debug, cancel = false) => {
   await debug?.log("SUBMIT", "SUBMIT_CLEANUP", {
     deletedWorkflowRunId: workflowContext.workflowRunId
   });
-  try {
-    await workflowContext.qstashClient.http.request({
-      path: ["v2", "workflows", "runs", `${workflowContext.workflowRunId}?cancel=${cancel}`],
-      method: "DELETE",
-      parseResponseAsJson: false
-    });
-    await debug?.log(
-      "SUBMIT",
-      "SUBMIT_CLEANUP",
-      `workflow run ${workflowContext.workflowRunId} deleted.`
-    );
-    return { deleted: true };
-  } catch (error) {
-    if (error instanceof import_qstash3.QstashError && error.status === 404) {
-      await debug?.log("WARN", "SUBMIT_CLEANUP", {
-        message: `Failed to remove workflow run ${workflowContext.workflowRunId} as it doesn't exist.`,
-        name: error.name,
-        errorMessage: error.message
-      });
-      return { deleted: false };
-    }
-    throw error;
-  }
+  await workflowContext.qstashClient.http.request({
+    path: ["v2", "workflows", "runs", `${workflowContext.workflowRunId}?cancel=${cancel}`],
+    method: "DELETE",
+    parseResponseAsJson: false
+  });
+  await debug?.log(
+    "SUBMIT",
+    "SUBMIT_CLEANUP",
+    `workflow run ${workflowContext.workflowRunId} deleted.`
+  );
 };
 var recreateUserHeaders = (headers) => {
   const filteredHeaders = new Headers();
@@ -1635,7 +1622,22 @@ var WorkflowApi = class extends BaseWorkflowApi {
 // src/agents/adapters.ts
 var import_openai2 = require("@ai-sdk/openai");
 var import_ai = require("ai");
+
+// 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/agents/adapters.ts
 var createWorkflowOpenAI = (context) => {
   return (0, import_openai2.createOpenAI)({
     compatibility: "strict",
@@ -1702,8 +1704,7 @@ 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)
+    execute: async (...param) => langchainTool.invoke(...param)
   });
 };
 
@@ -1716,16 +1717,24 @@ var Agent = class {
   maxSteps;
   background;
   model;
-  constructor({ tools, maxSteps, background, name, model }) {
+  temparature;
+  constructor({ tools, maxSteps, background, name, model, temparature = 0.1 }) {
     this.name = name;
     this.tools = tools ?? {};
     this.maxSteps = maxSteps;
     this.background = background;
     this.model = model;
+    this.temparature = temparature;
   }
+  /**
+   * Trigger the agent by passing a prompt
+   *
+   * @param prompt task to assign to the agent
+   * @returns Response as `{ text: string }`
+   */
   async call({ prompt }) {
     try {
-      return await (0, import_ai2.generateText)({
+      const result = await (0, import_ai2.generateText)({
         model: this.model,
         tools: this.tools,
         maxSteps: this.maxSteps,
@@ -1733,8 +1742,10 @@ var Agent = class {
         prompt,
         headers: {
           [AGENT_NAME_HEADER]: this.name
-        }
+        },
+        temperature: this.temparature
       });
+      return { text: result.text };
     } catch (error) {
       if (error instanceof import_ai2.ToolExecutionError) {
         if (error.cause instanceof Error && error.cause.name === "WorkflowAbort") {
@@ -1749,6 +1760,11 @@ var Agent = class {
       }
     }
   }
+  /**
+   * 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)({
@@ -1760,18 +1776,25 @@ var Agent = class {
     });
   }
 };
-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;
+  /**
+   * 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({
-    maxSteps,
-    background = MANAGER_AGENT_PROMPT,
     agents,
+    background = MANAGER_AGENT_PROMPT,
     model,
+    maxSteps,
     name = "manager llm"
   }) {
     super({
@@ -1796,6 +1819,11 @@ var Task = class {
     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;
     const safePrompt = await this.context.run("Get Prompt", () => prompt);
@@ -1826,6 +1854,29 @@ var WorkflowAgents = class {
   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({
@@ -1836,6 +1887,9 @@ var WorkflowAgents = class {
   task(taskParameters) {
     return new Task({ context: this.context, taskParameters });
   }
+  /**
+   * creates an openai model for agents
+   */
   openai(...params) {
     const openai2 = createWorkflowOpenAI(this.context);
     return openai2(...params);
@@ -2436,7 +2490,6 @@ 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/hono.mjs

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

package/index.d.mts

@@ -1,9 +1,8 @@
-import { R as RouteFunction, W as WorkflowServeOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-BEyIoCRe.mjs';
-export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-BEyIoCRe.mjs';
+import { R as RouteFunction, W as WorkflowServeOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-D9gwTj2n.mjs';
+export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-D9gwTj2n.mjs';
 import { Client as Client$1, QstashError } from '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';
-import 'langchain/tools';
 
 /**
  * Creates an async method that handles incoming requests and runs the provided

package/index.d.ts

@@ -1,9 +1,8 @@
-import { R as RouteFunction, W as WorkflowServeOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-BEyIoCRe.js';
-export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-BEyIoCRe.js';
+import { R as RouteFunction, W as WorkflowServeOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-D9gwTj2n.js';
+export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-D9gwTj2n.js';
 import { Client as Client$1, QstashError } from '@upstash/qstash';
 import 'ai';
 import '@ai-sdk/openai';
-import 'langchain/tools';
 
 /**
  * Creates an async method that handles incoming requests and runs the provided