beeai-framework 0.1.20 → 0.1.22

package/dist/adapters/watsonx/backend/chat.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/adapters/watsonx/backend/chat.ts"],"names":["Type","WatsonxChatModel","ChatModel","client","emitter","Emitter","root","child","namespace","creator","toolChoiceSupport","providerId","modelId","getEnv","WatsonxClient","ensure","config","name","parameters","undefined","NotImplementedError","_create","input","run","result","instance","textChat","prepareInput","signal","messages","finishReason","usage","extractOutput","choices","ChatModelOutput","_createStream","stream","textChatStream","returnObject","raw","aborted","controller","abort","data","map","delta","choice","message","findLast","Boolean","finish_reason","completionTokens","completion_tokens","promptTokens","prompt_tokens","totalTokens","total_tokens","flatMap","content","msg","AssistantMessage","type","text","push","tool_calls","call","toolCallId","id","toolName","function","args","parseBrokenJson","arguments","refusal","filter","isTruthy","overrides","tools","Promise","all","tool","description","getInputJsonSchema","toolConfig","toolChoice","toolChoiceOption","Tool","includes","ToolMessage","role","JSON","stringify","tool_call_id","SystemMessage","toPlain","spaceId","projectId","isEmpty","responseFormat","JSON_OBJECT","topP","frequencyPenalty","temperature","n","maxTokens","presencePenalty","stop","stopSequences","seed","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;;;;;;;AA4BA,MAAOA,IAAAA,GAAO,iBAAA,CAAkB,sBAAA,CAAuB,SAAA,CAAU,IAAA;AAM1D,MAAMC,yBAAyBC,SAAAA,CAAAA;EAlCtC;;;;AAmCqBC,EAAAA,MAAAA;EACHC,OAAAA,GAA4BC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC7DC,SAAAA,EAAW;AAAC,MAAA,SAAA;AAAW,MAAA,SAAA;AAAW,MAAA;;IAClCC,OAAAA,EAAS;GACX,CAAA;EACgBC,iBAAAA,GAAkD;AAAC,IAAA,MAAA;AAAQ,IAAA;;AAE3E,EAAA,IAAIC,UAAAA,GAAa;AACf,IAAA,OAAO,SAAA;AACT;AAEA,EAAA,WAAA,CACkBC,OAAAA,GAAUC,MAAAA,CAAO,oBAAA,EAAsB,2BAAA,GACvDV,MAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,KAHWS,OAAAA,GAAAA,OAAAA;AAIhB,IAAA,IAAA,CAAKT,MAAAA,GAASW,aAAAA,CAAcC,MAAAA,CAAOZ,MAAAA,CAAAA;AACrC;AAEAa,EAAAA,MAAAA,CAAOA,MAAAA,EAAoB;AACzB,IAAA,KAAA,CAAMA,OAAOA,MAAAA,CAAAA;AACb,IAAA,KAAA,MAAWC,IAAAA,IAAQ;AAAC,MAAA;AAA0C,KAAA,EAAA;AAC5D,MAAA,IAAI,IAAA,CAAKC,UAAAA,CAAWD,IAAAA,CAAAA,KAAUE,MAAAA,EAAW;AACvC,QAAA,IAAA,CAAKD,UAAAA,CAAWD,IAAAA,CAAAA,GAAQE,MAAAA;AACxB,QAAA,MAAM,IAAIC,mBAAAA,CACR,CAAA,SAAA,EAAYH,IAAAA,CAAAA,qDAAAA,CAA2D,CAAA;AAE3E;AACF;AACA,IAAA,OAAO,IAAA;AACT;EAEA,MAAgBI,OAAAA,CAAQC,OAAuBC,GAAAA,EAAyB;AACtE,IAAA,MAAM,EAAEC,MAAAA,EAAM,GAAK,MAAM,IAAA,CAAKrB,MAAAA,CAAOsB,SAASC,QAAAA,CAAS;MACrD,GAAI,MAAM,IAAA,CAAKC,YAAAA,CAAaL,KAAAA,CAAAA;AAC5BM,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK;KACd,CAAA;AACA,IAAA,MAAM,EAAEC,QAAAA,EAAUC,YAAAA,EAAcC,KAAAA,EAAK,GAAK,KAAKC,aAAAA,CAAcR,MAAAA,CAAOS,OAAAA,EAAST,MAAAA,CAAOO,KAAK,CAAA;AACzF,IAAA,OAAO,IAAIG,eAAAA,CAAgBL,QAAAA,EAAUE,KAAAA,EAAOD,YAAAA,CAAAA;AAC9C;EAEA,OAAOK,aAAAA,CAAcb,OAAuBC,GAAAA,EAA0B;AACpE,IAAA,MAAMa,MAAAA,GAAS,MAAM,IAAA,CAAKjC,MAAAA,CAAOsB,SAASY,cAAAA,CAAe;MACvD,GAAI,MAAM,IAAA,CAAKV,YAAAA,CAAaL,KAAAA,CAAAA;AAC5BM,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK,MAAAA;MACZU,YAAAA,EAAc;KAChB,CAAA;AACA,IAAA,WAAA,MAAiBC,OAAOH,MAAAA,EAAQ;AAC9B,MAAA,IAAIb,GAAAA,CAAIK,OAAOY,OAAAA,EAAS;AACtBJ,QAAAA,MAAAA,CAAOK,UAAAA,CAAWC,KAAAA,CAAMnB,GAAAA,CAAIK,MAAAA,CAAOY,OAAO,CAAA;AAC1C,QAAA;AACF;AACA,MAAA,MAAM,EAAEX,QAAAA,EAAUC,YAAAA,EAAcC,KAAAA,EAAK,GAAK,KAAKC,aAAAA,CAC7CO,GAAAA,CAAII,IAAAA,CAAKV,OAAAA,CAAQW,IAAI,CAAC,EAAEC,KAAAA,EAAO,GAAGC,QAAAA,MAAc;QAAE,GAAGA,MAAAA;QAAQC,OAAAA,EAASF;OAAM,CAAA,CAAA,EAC5EN,GAAAA,CAAII,IAAAA,CAAKZ,KAAK,CAAA;AAEhB,MAAA,MAAM,IAAIG,eAAAA,CAAgBL,QAAAA,EAAUE,KAAAA,EAAOD,YAAAA,CAAAA;AAC7C;AACF;AAEUE,EAAAA,aAAAA,CAAcC,SAAiCF,KAAAA,EAAuB;AAC9E,IAAA,OAAO;MACLD,YAAAA,EAAckB,QAAAA,CAASf,SAAS,CAACa,MAAAA,KAAWG,QAAQH,MAAAA,EAAQI,aAAAA,CAAAA,CAAAA,EACxDA,aAAAA;AACJnB,MAAAA,KAAAA,EAAOA,KAAAA,GACH;AACEoB,QAAAA,gBAAAA,EAAkBpB,MAAMqB,iBAAAA,IAAqB,CAAA;AAC7CC,QAAAA,YAAAA,EAActB,MAAMuB,aAAAA,IAAiB,CAAA;AACrCC,QAAAA,WAAAA,EAAaxB,MAAMyB,YAAAA,IAAgB;OACrC,GACArC,MAAAA;AACJU,MAAAA,QAAAA,EAAUI,OAAAA,CACPwB,OAAAA,CAAQ,CAAC,EAAEV,SAAO,KAAE;AACnB,QAAA,MAAMlB,WAAsB,EAAA;AAC5B,QAAA,IAAIkB,SAASW,OAAAA,EAAS;AACpB,UAAA,MAAMC,GAAAA,GAAM,IAAIC,gBAAAA,CAAiB;YAAEC,IAAAA,EAAM,MAAA;AAAQC,YAAAA,IAAAA,EAAMf,OAAAA,CAAQW;WAAQ,CAAA;AAEvE7B,UAAAA,QAAAA,CAASkC,KAAKJ,GAAAA,CAAAA;AAChB;AACA,QAAA,IAAIZ,SAASiB,UAAAA,EAAY;AACvB,UAAA,MAAML,MAAM,IAAIC,gBAAAA,CACdb,QAAQiB,UAAAA,CAAWpB,GAAAA,CACjB,CAACqB,IAAAA,MAAwB;YACvBJ,IAAAA,EAAM,WAAA;AACNK,YAAAA,UAAAA,EAAYD,IAAAA,CAAKE,EAAAA;AACjBC,YAAAA,QAAAA,EAAUH,KAAKI,QAAAA,CAASpD,IAAAA;YACxBqD,IAAAA,EAAMC,eAAAA,CAAgBN,IAAAA,CAAKI,QAAAA,CAASG,SAAS;AAC/C,WAAA,CAAA,CAAA,CAAA;AAIJ3C,UAAAA,QAAAA,CAASkC,KAAKJ,GAAAA,CAAAA;AAChB;AACA,QAAA,IAAIZ,SAAS0B,OAAAA,EAAS;AACpB,UAAA,MAAMd,GAAAA,GAAM,IAAIC,gBAAAA,CAAiB;YAAEC,IAAAA,EAAM,MAAA;AAAQC,YAAAA,IAAAA,EAAMf,OAAAA,CAAQ0B;WAAQ,CAAA;AAEvE5C,UAAAA,QAAAA,CAASkC,KAAKJ,GAAAA,CAAAA;AAChB;AACA,QAAA,OAAO9B,QAAAA;OACT,CAAA,CACC6C,OAAOC,QAAAA;AACZ,KAAA;AACF;AAEA,EAAA,MAAgBhD,aAAaiD,SAAAA,EAAoD;AAC/E,IAAA,MAAMtD,KAAAA,GAAwB;AAAE,MAAA,GAAG,IAAA,CAAKJ,UAAAA;MAAY,GAAG0D;AAAU,KAAA;AACjE,IAAA,MAAMC,KAAAA,GAAkC,MAAMC,OAAAA,CAAQC,GAAAA,CAAAA,CACnDzD,KAAAA,CAAMuD,SAAS,EAAA,EAAIjC,GAAAA,CAAI,OAAOoC,IAAAA,MAAU;MACvCnB,IAAAA,EAAM,UAAA;MACNQ,QAAAA,EAAU;AACRpD,QAAAA,IAAAA,EAAM+D,IAAAA,CAAK/D,IAAAA;AACXgE,QAAAA,WAAAA,EAAaD,IAAAA,CAAKC,WAAAA;QAClB/D,UAAAA,EAAY,MAAM8D,KAAKE,kBAAAA;AACzB;AACF,KAAA,CAAA,CAAA,CAAA;AAGF,IAAA,MAAMC,UAAAA,GAAsE;MAC1EC,UAAAA,EAAYjE,MAAAA;MACZkE,gBAAAA,EAAkBlE;AACpB,KAAA;AACA,IAAA,IAAIG,MAAM8D,UAAAA,EAAY;AACpB,MAAA,IAAI9D,KAAAA,CAAM8D,sBAAsBE,IAAAA,EAAM;AACpCH,QAAAA,UAAAA,CAAWC,UAAAA,GAAa;UACtBvB,IAAAA,EAAM,UAAA;UACNQ,QAAAA,EAAU;AAAEpD,YAAAA,IAAAA,EAAMK,MAAM8D,UAAAA,CAAWnE;AAAK;AAC1C,SAAA;AACF,OAAA,MAAA,IAAW,IAAA,CAAKP,iBAAAA,CAAkB6E,QAAAA,CAASjE,KAAAA,CAAM8D,UAAU,CAAA,EAAG;AAC5DD,QAAAA,UAAAA,CAAWE,mBAAmB/D,KAAAA,CAAM8D,UAAAA;AACtC;AACF;AAEA,IAAA,OAAO;AACLxE,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdiB,MAAAA,QAAAA,EAAUP,KAAAA,CAAMO,QAAAA,CAAS4B,OAAAA,CAAQ,CAACV,OAAAA,KAAAA;AAChC,QAAA,IAAIA,mBAAmByC,WAAAA,EAAa;AAClC,UAAA,OAAOzC,OAAAA,CAAQW,OAAAA,CAAQd,GAAAA,CAAI,CAACc,OAAAA,MAAa;YACvC+B,IAAAA,EAAM,MAAA;YACN/B,OAAAA,EAASgC,IAAAA,CAAKC,SAAAA,CAAUjC,OAAAA,CAAQlC,MAAM,CAAA;AACtCoE,YAAAA,YAAAA,EAAclC,OAAAA,CAAQQ;WACxB,CAAA,CAAA;AACF,SAAA,MAAA,IAAWnB,mBAAmB8C,aAAAA,EAAe;AAC3C,UAAA,OAAO9C,OAAAA,CAAQW,OAAAA,CAAQd,GAAAA,CAAI,CAACc,OAAAA,MAAa;YACvC+B,IAAAA,EAAM,QAAA;AACN/B,YAAAA,OAAAA,EAASA,OAAAA,CAAQI;WACnB,CAAA,CAAA;AACF,SAAA,MAAA,IAAWf,mBAAmBa,gBAAAA,EAAkB;AAC9C,UAAA,OAAOb,OAAAA,CAAQW,OAAAA,CAAQd,GAAAA,CAAI,CAACc,OAAAA,MAAa;YACvC+B,IAAAA,EAAM,WAAA;YACN,GAAI/B,OAAAA,CAAQG,SAAS,MAAA,IAAU;AAC7BH,cAAAA,OAAAA,EAASA,OAAAA,CAAQI;AACnB,aAAA;YACA,GAAIJ,OAAAA,CAAQG,SAAS,WAAA,IAAe;AAClCM,cAAAA,EAAAA,EAAIT,OAAAA,CAAQQ,UAAAA;cACZL,IAAAA,EAAM,UAAA;cACNQ,QAAAA,EAAU;AACRpD,gBAAAA,IAAAA,EAAMyC,OAAAA,CAAQU,QAAAA;gBACdI,SAAAA,EAAWkB,IAAAA,CAAKC,SAAAA,CAAUjC,OAAAA,CAAQY,IAAI;AACxC;AACF;WACF,CAAA,CAAA;SACF,MAAO;AACL,UAAA,OAAO;AAACvB,YAAAA,OAAAA,CAAQ+C,OAAAA;;AAClB;OACF,CAAA;AACAC,MAAAA,OAAAA,EAAS,KAAK5F,MAAAA,CAAO4F,OAAAA;AACrBC,MAAAA,SAAAA,EAAW,KAAK7F,MAAAA,CAAO6F,SAAAA;MACvBnB,KAAAA,EAAOoB,OAAAA,CAAQpB,KAAAA,CAAAA,GAAS1D,MAAAA,GAAY0D,KAAAA;MACpCqB,cAAAA,EAAgB/E,MAAAA;AAChB,MAAA,GAAIG,MAAM4E,cAAAA,IAAkB;QAC1BA,cAAAA,EAAgB;AAAErC,UAAAA,IAAAA,EAAM7D,IAAAA,CAAKmG;AAAY;AAC3C,OAAA;AACAC,MAAAA,IAAAA,EAAM9E,KAAAA,CAAM8E,IAAAA;AACZC,MAAAA,gBAAAA,EAAkB/E,KAAAA,CAAM+E,gBAAAA;AACxBC,MAAAA,WAAAA,EAAahF,KAAAA,CAAMgF,WAAAA;AACnBC,MAAAA,CAAAA,EAAGjF,KAAAA,CAAMiF,CAAAA;AACTC,MAAAA,SAAAA,EAAWlF,KAAAA,CAAMkF,SAAAA;AACjBC,MAAAA,eAAAA,EAAiBnF,KAAAA,CAAMmF,eAAAA;AACvBC,MAAAA,IAAAA,EAAMpF,KAAAA,CAAMqF,aAAAA;AACZC,MAAAA,IAAAA,EAAMtF,KAAAA,CAAMsF,IAAAA;MACZ,GAAGzB;AACL,KAAA;AACF;EAEA0B,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACTjG,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdM,UAAAA,EAAY4F,WAAAA,CAAY,KAAK5F,UAAU,CAAA;AACvCf,MAAAA,MAAAA,EAAQ,IAAA,CAAKA;AACf,KAAA;AACF;AAEA4G,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"chat.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  ChatConfig,\n  ChatModel,\n  ChatModelEmitter,\n  ChatModelInput,\n  ChatModelOutput,\n  ChatModelParameters,\n  ChatModelToolChoiceSupport,\n} from \"@/backend/chat.js\";\nimport { WatsonxClient, WatsonxClientSettings } from \"@/adapters/watsonx/backend/client.js\";\nimport { findLast, isEmpty, isTruthy } from \"remeda\";\nimport WatsonxAiMlVml_v1, {\n  TextChatMessages,\n  TextChatParameterTools,\n  TextChatParams,\n  TextChatResultChoice,\n  TextChatUsage,\n} from \"@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, SystemMessage, ToolMessage } from \"@/backend/message.js\";\nimport { ToolCallPart } from \"ai\";\nimport Type = WatsonxAiMlVml_v1.TextChatResponseFormat.Constants.Type;\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport { NotImplementedError } from \"@/errors.js\";\nimport { Tool } from \"@/tools/base.js\";\n\nexport class WatsonxChatModel extends ChatModel {\n  protected readonly client: WatsonxClient;\n  public readonly emitter: ChatModelEmitter = Emitter.root.child({\n    namespace: [\"backend\", \"watsonx\", \"chat\"],\n    creator: this,\n  });\n  public readonly toolChoiceSupport: ChatModelToolChoiceSupport[] = [\"none\", \"single\"];\n\n  get providerId() {\n    return \"watsonx\";\n  }\n\n  constructor(\n    public readonly modelId = getEnv(\"WATSONX_CHAT_MODEL\", \"ibm/granite-3-8b-instruct\"),\n    client?: WatsonxClient | WatsonxClientSettings,\n  ) {\n    super();\n    this.client = WatsonxClient.ensure(client);\n  }\n\n  config(config: ChatConfig) {\n    super.config(config);\n    for (const name of [\"topK\"] as (keyof ChatModelParameters)[]) {\n      if (this.parameters[name] !== undefined) {\n        this.parameters[name] = undefined;\n        throw new NotImplementedError(\n          `Setting \"${name}\" parameter is not supported for WatsonX Chat Models.`,\n        );\n      }\n    }\n    return this;\n  }\n\n  protected async _create(input: ChatModelInput, run: GetRunContext<any>) {\n    const { result } = await this.client.instance.textChat({\n      ...(await this.prepareInput(input)),\n      signal: run.signal,\n    });\n    const { messages, finishReason, usage } = this.extractOutput(result.choices, result.usage);\n    return new ChatModelOutput(messages, usage, finishReason);\n  }\n\n  async *_createStream(input: ChatModelInput, run: GetRunContext<this>) {\n    const stream = await this.client.instance.textChatStream({\n      ...(await this.prepareInput(input)),\n      signal: run.signal,\n      returnObject: true,\n    });\n    for await (const raw of stream) {\n      if (run.signal.aborted) {\n        stream.controller.abort(run.signal.aborted);\n        break;\n      }\n      const { messages, finishReason, usage } = this.extractOutput(\n        raw.data.choices.map(({ delta, ...choice }) => ({ ...choice, message: delta })),\n        raw.data.usage,\n      );\n      yield new ChatModelOutput(messages, usage, finishReason);\n    }\n  }\n\n  protected extractOutput(choices: TextChatResultChoice[], usage?: TextChatUsage) {\n    return {\n      finishReason: findLast(choices, (choice) => Boolean(choice?.finish_reason))\n        ?.finish_reason as ChatModelOutput[\"finishReason\"],\n      usage: usage\n        ? {\n            completionTokens: usage.completion_tokens ?? 0,\n            promptTokens: usage.prompt_tokens ?? 0,\n            totalTokens: usage.total_tokens ?? 0,\n          }\n        : undefined,\n      messages: choices\n        .flatMap(({ message }) => {\n          const messages: Message[] = [];\n          if (message?.content) {\n            const msg = new AssistantMessage({ type: \"text\", text: message.content });\n            // msg.role = message.role || msg.role;\n            messages.push(msg);\n          }\n          if (message?.tool_calls) {\n            const msg = new AssistantMessage(\n              message.tool_calls.map(\n                (call): ToolCallPart => ({\n                  type: \"tool-call\",\n                  toolCallId: call.id,\n                  toolName: call.function.name,\n                  args: parseBrokenJson(call.function.arguments),\n                }),\n              ),\n            );\n            // msg.role = message.role || msg.role;\n            messages.push(msg);\n          }\n          if (message?.refusal) {\n            const msg = new AssistantMessage({ type: \"text\", text: message.refusal });\n            // msg.role = message.role || msg.role;\n            messages.push(msg);\n          }\n          return messages;\n        })\n        .filter(isTruthy),\n    };\n  }\n\n  protected async prepareInput(overrides: ChatModelInput): Promise<TextChatParams> {\n    const input: ChatModelInput = { ...this.parameters, ...overrides };\n    const tools: TextChatParameterTools[] = await Promise.all(\n      (input.tools ?? []).map(async (tool) => ({\n        type: \"function\",\n        function: {\n          name: tool.name,\n          description: tool.description,\n          parameters: await tool.getInputJsonSchema(),\n        },\n      })),\n    );\n\n    const toolConfig: Pick<TextChatParams, \"toolChoice\" | \"toolChoiceOption\"> = {\n      toolChoice: undefined,\n      toolChoiceOption: undefined,\n    };\n    if (input.toolChoice) {\n      if (input.toolChoice instanceof Tool) {\n        toolConfig.toolChoice = {\n          type: \"function\",\n          function: { name: input.toolChoice.name },\n        };\n      } else if (this.toolChoiceSupport.includes(input.toolChoice)) {\n        toolConfig.toolChoiceOption = input.toolChoice;\n      }\n    }\n\n    return {\n      modelId: this.modelId,\n      messages: input.messages.flatMap((message): TextChatMessages[] => {\n        if (message instanceof ToolMessage) {\n          return message.content.map((content) => ({\n            role: \"tool\",\n            content: JSON.stringify(content.result),\n            tool_call_id: content.toolCallId,\n          }));\n        } else if (message instanceof SystemMessage) {\n          return message.content.map((content) => ({\n            role: \"system\",\n            content: content.text,\n          }));\n        } else if (message instanceof AssistantMessage) {\n          return message.content.map((content) => ({\n            role: \"assistant\",\n            ...(content.type === \"text\" && {\n              content: content.text,\n            }),\n            ...(content.type === \"tool-call\" && {\n              id: content.toolCallId,\n              type: \"function\",\n              function: {\n                name: content.toolName,\n                arguments: JSON.stringify(content.args),\n              },\n            }),\n          }));\n        } else {\n          return [message.toPlain()];\n        }\n      }),\n      spaceId: this.client.spaceId,\n      projectId: this.client.projectId,\n      tools: isEmpty(tools) ? undefined : tools,\n      responseFormat: undefined,\n      ...(input.responseFormat && {\n        responseFormat: { type: Type.JSON_OBJECT },\n      }),\n      topP: input.topP,\n      frequencyPenalty: input.frequencyPenalty,\n      temperature: input.temperature,\n      n: input.n,\n      maxTokens: input.maxTokens,\n      presencePenalty: input.presencePenalty,\n      stop: input.stopSequences,\n      seed: input.seed,\n      ...toolConfig,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      modelId: this.modelId,\n      parameters: shallowCopy(this.parameters),\n      client: this.client,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/adapters/watsonx/backend/chat.ts"],"names":["Type","WatsonxChatModel","ChatModel","client","emitter","Emitter","root","child","namespace","creator","toolChoiceSupport","providerId","modelId","getEnv","WatsonxClient","ensure","config","name","parameters","undefined","NotImplementedError","_create","input","run","result","instance","textChat","prepareInput","signal","messages","finishReason","usage","extractOutput","choices","ChatModelOutput","_createStream","stream","textChatStream","returnObject","raw","aborted","controller","abort","data","map","delta","choice","message","findLast","Boolean","finish_reason","completionTokens","completion_tokens","promptTokens","prompt_tokens","totalTokens","total_tokens","flatMap","content","msg","AssistantMessage","type","text","push","tool_calls","call","toolCallId","id","toolName","function","parseBrokenJson","arguments","refusal","filter","isTruthy","overrides","tools","Promise","all","tool","description","getInputJsonSchema","toolConfig","toolChoice","toolChoiceOption","Tool","includes","ToolMessage","role","output","value","JSON","stringify","tool_call_id","SystemMessage","Error","toPlain","spaceId","projectId","isEmpty","responseFormat","JSON_OBJECT","topP","frequencyPenalty","temperature","n","maxTokens","presencePenalty","stop","stopSequences","seed","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;;;;;;;AA4BA,MAAOA,IAAAA,GAAO,iBAAA,CAAkB,sBAAA,CAAuB,SAAA,CAAU,IAAA;AAW1D,MAAMC,yBAAyBC,SAAAA,CAAAA;EAvCtC;;;;AAwCqBC,EAAAA,MAAAA;EACHC,OAAAA,GAA4BC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC7DC,SAAAA,EAAW;AAAC,MAAA,SAAA;AAAW,MAAA,SAAA;AAAW,MAAA;;IAClCC,OAAAA,EAAS;GACX,CAAA;EACgBC,iBAAAA,GAAkD;AAAC,IAAA,MAAA;AAAQ,IAAA;;AAE3E,EAAA,IAAIC,UAAAA,GAAa;AACf,IAAA,OAAO,SAAA;AACT;AAEA,EAAA,WAAA,CACkBC,OAAAA,GAAUC,MAAAA,CAAO,oBAAA,EAAsB,2BAAA,GACvDV,MAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,KAHWS,OAAAA,GAAAA,OAAAA;AAIhB,IAAA,IAAA,CAAKT,MAAAA,GAASW,aAAAA,CAAcC,MAAAA,CAAOZ,MAAAA,CAAAA;AACrC;AAEAa,EAAAA,MAAAA,CAAOA,MAAAA,EAAoB;AACzB,IAAA,KAAA,CAAMA,OAAOA,MAAAA,CAAAA;AACb,IAAA,KAAA,MAAWC,IAAAA,IAAQ;AAAC,MAAA;AAA0C,KAAA,EAAA;AAC5D,MAAA,IAAI,IAAA,CAAKC,UAAAA,CAAWD,IAAAA,CAAAA,KAAUE,MAAAA,EAAW;AACvC,QAAA,IAAA,CAAKD,UAAAA,CAAWD,IAAAA,CAAAA,GAAQE,MAAAA;AACxB,QAAA,MAAM,IAAIC,mBAAAA,CACR,CAAA,SAAA,EAAYH,IAAAA,CAAAA,qDAAAA,CAA2D,CAAA;AAE3E;AACF;AACA,IAAA,OAAO,IAAA;AACT;EAEA,MAAgBI,OAAAA,CAAQC,OAAuBC,GAAAA,EAAyB;AACtE,IAAA,MAAM,EAAEC,MAAAA,EAAM,GAAK,MAAM,IAAA,CAAKrB,MAAAA,CAAOsB,SAASC,QAAAA,CAAS;MACrD,GAAI,MAAM,IAAA,CAAKC,YAAAA,CAAaL,KAAAA,CAAAA;AAC5BM,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK;KACd,CAAA;AACA,IAAA,MAAM,EAAEC,QAAAA,EAAUC,YAAAA,EAAcC,KAAAA,EAAK,GAAK,KAAKC,aAAAA,CAAcR,MAAAA,CAAOS,OAAAA,EAAST,MAAAA,CAAOO,KAAK,CAAA;AACzF,IAAA,OAAO,IAAIG,eAAAA,CAAgBL,QAAAA,EAAUE,KAAAA,EAAOD,YAAAA,CAAAA;AAC9C;EAEA,OAAOK,aAAAA,CAAcb,OAAuBC,GAAAA,EAA0B;AACpE,IAAA,MAAMa,MAAAA,GAAS,MAAM,IAAA,CAAKjC,MAAAA,CAAOsB,SAASY,cAAAA,CAAe;MACvD,GAAI,MAAM,IAAA,CAAKV,YAAAA,CAAaL,KAAAA,CAAAA;AAC5BM,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK,MAAAA;MACZU,YAAAA,EAAc;KAChB,CAAA;AACA,IAAA,WAAA,MAAiBC,OAAOH,MAAAA,EAAQ;AAC9B,MAAA,IAAIb,GAAAA,CAAIK,OAAOY,OAAAA,EAAS;AACtBJ,QAAAA,MAAAA,CAAOK,UAAAA,CAAWC,KAAAA,CAAMnB,GAAAA,CAAIK,MAAAA,CAAOY,OAAO,CAAA;AAC1C,QAAA;AACF;AACA,MAAA,MAAM,EAAEX,QAAAA,EAAUC,YAAAA,EAAcC,KAAAA,EAAK,GAAK,KAAKC,aAAAA,CAC7CO,GAAAA,CAAII,IAAAA,CAAKV,OAAAA,CAAQW,IAAI,CAAC,EAAEC,KAAAA,EAAO,GAAGC,QAAAA,MAAc;QAAE,GAAGA,MAAAA;QAAQC,OAAAA,EAASF;OAAM,CAAA,CAAA,EAC5EN,GAAAA,CAAII,IAAAA,CAAKZ,KAAK,CAAA;AAEhB,MAAA,MAAM,IAAIG,eAAAA,CAAgBL,QAAAA,EAAUE,KAAAA,EAAOD,YAAAA,CAAAA;AAC7C;AACF;AAEUE,EAAAA,aAAAA,CAAcC,SAAiCF,KAAAA,EAAuB;AAC9E,IAAA,OAAO;MACLD,YAAAA,EAAckB,QAAAA,CAASf,SAAS,CAACa,MAAAA,KAAWG,QAAQH,MAAAA,EAAQI,aAAAA,CAAAA,CAAAA,EACxDA,aAAAA;AACJnB,MAAAA,KAAAA,EAAOA,KAAAA,GACH;AACEoB,QAAAA,gBAAAA,EAAkBpB,MAAMqB,iBAAAA,IAAqB,CAAA;AAC7CC,QAAAA,YAAAA,EAActB,MAAMuB,aAAAA,IAAiB,CAAA;AACrCC,QAAAA,WAAAA,EAAaxB,MAAMyB,YAAAA,IAAgB;OACrC,GACArC,MAAAA;AACJU,MAAAA,QAAAA,EAAUI,OAAAA,CACPwB,OAAAA,CAAQ,CAAC,EAAEV,SAAO,KAAE;AACnB,QAAA,MAAMlB,WAAsB,EAAA;AAC5B,QAAA,IAAIkB,SAASW,OAAAA,EAAS;AACpB,UAAA,MAAMC,GAAAA,GAAM,IAAIC,gBAAAA,CAAiB;YAAEC,IAAAA,EAAM,MAAA;AAAQC,YAAAA,IAAAA,EAAMf,OAAAA,CAAQW;WAAQ,CAAA;AAEvE7B,UAAAA,QAAAA,CAASkC,KAAKJ,GAAAA,CAAAA;AAChB;AACA,QAAA,IAAIZ,SAASiB,UAAAA,EAAY;AACvB,UAAA,MAAML,MAAM,IAAIC,gBAAAA,CACdb,QAAQiB,UAAAA,CAAWpB,GAAAA,CACjB,CAACqB,IAAAA,MAAwB;YACvBJ,IAAAA,EAAM,WAAA;AACNK,YAAAA,UAAAA,EAAYD,IAAAA,CAAKE,EAAAA;AACjBC,YAAAA,QAAAA,EAAUH,KAAKI,QAAAA,CAASpD,IAAAA;YACxBK,KAAAA,EAAOgD,eAAAA,CAAgBL,IAAAA,CAAKI,QAAAA,CAASE,SAAS;AAChD,WAAA,CAAA,CAAA,CAAA;AAIJ1C,UAAAA,QAAAA,CAASkC,KAAKJ,GAAAA,CAAAA;AAChB;AACA,QAAA,IAAIZ,SAASyB,OAAAA,EAAS;AACpB,UAAA,MAAMb,GAAAA,GAAM,IAAIC,gBAAAA,CAAiB;YAAEC,IAAAA,EAAM,MAAA;AAAQC,YAAAA,IAAAA,EAAMf,OAAAA,CAAQyB;WAAQ,CAAA;AAEvE3C,UAAAA,QAAAA,CAASkC,KAAKJ,GAAAA,CAAAA;AAChB;AACA,QAAA,OAAO9B,QAAAA;OACT,CAAA,CACC4C,OAAOC,QAAAA;AACZ,KAAA;AACF;AAEA,EAAA,MAAgB/C,aAAagD,SAAAA,EAAoD;AAC/E,IAAA,MAAMrD,KAAAA,GAAwB;AAAE,MAAA,GAAG,IAAA,CAAKJ,UAAAA;MAAY,GAAGyD;AAAU,KAAA;AACjE,IAAA,MAAMC,KAAAA,GAAkC,MAAMC,OAAAA,CAAQC,GAAAA,CAAAA,CACnDxD,KAAAA,CAAMsD,SAAS,EAAA,EAAIhC,GAAAA,CAAI,OAAOmC,IAAAA,MAAU;MACvClB,IAAAA,EAAM,UAAA;MACNQ,QAAAA,EAAU;AACRpD,QAAAA,IAAAA,EAAM8D,IAAAA,CAAK9D,IAAAA;AACX+D,QAAAA,WAAAA,EAAaD,IAAAA,CAAKC,WAAAA;QAClB9D,UAAAA,EAAY,MAAM6D,KAAKE,kBAAAA;AACzB;AACF,KAAA,CAAA,CAAA,CAAA;AAGF,IAAA,MAAMC,UAAAA,GAAsE;MAC1EC,UAAAA,EAAYhE,MAAAA;MACZiE,gBAAAA,EAAkBjE;AACpB,KAAA;AACA,IAAA,IAAIG,MAAM6D,UAAAA,EAAY;AACpB,MAAA,IAAI7D,KAAAA,CAAM6D,sBAAsBE,IAAAA,EAAM;AACpCH,QAAAA,UAAAA,CAAWC,UAAAA,GAAa;UACtBtB,IAAAA,EAAM,UAAA;UACNQ,QAAAA,EAAU;AAAEpD,YAAAA,IAAAA,EAAMK,MAAM6D,UAAAA,CAAWlE;AAAK;AAC1C,SAAA;AACF,OAAA,MAAA,IAAW,IAAA,CAAKP,iBAAAA,CAAkB4E,QAAAA,CAAShE,KAAAA,CAAM6D,UAAU,CAAA,EAAG;AAC5DD,QAAAA,UAAAA,CAAWE,mBAAmB9D,KAAAA,CAAM6D,UAAAA;AACtC;AACF;AAEA,IAAA,OAAO;AACLvE,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdiB,MAAAA,QAAAA,EAAUP,KAAAA,CAAMO,QAAAA,CAAS4B,OAAAA,CAAQ,CAACV,OAAAA,KAAAA;AAChC,QAAA,IAAIA,mBAAmBwC,WAAAA,EAAa;AAClC,UAAA,OAAOxC,OAAAA,CAAQW,OAAAA,CAAQd,GAAAA,CACrB,CAACc,OAAAA,MAAkC;YACjC8B,IAAAA,EAAM,MAAA;YACN9B,OAAAA,EACE,OAAOA,OAAAA,CAAQ+B,MAAAA,CAAOC,KAAAA,KAAU,QAAA,GAC5BhC,QAAQ+B,MAAAA,CAAOC,KAAAA,GACfC,IAAAA,CAAKC,SAAAA,CAAUlC,OAAAA,CAAAA;AACrBmC,YAAAA,YAAAA,EAAcnC,OAAAA,CAAQQ;WACxB,CAAA,CAAA;AAEJ,SAAA,MAAA,IAAWnB,mBAAmB+C,aAAAA,EAAe;AAC3C,UAAA,OAAO/C,OAAAA,CAAQW,OAAAA,CAAQd,GAAAA,CACrB,CAACc,OAAAA,MAAoC;YACnC8B,IAAAA,EAAM,QAAA;AACN9B,YAAAA,OAAAA,EAASA,OAAAA,CAAQI;WACnB,CAAA,CAAA;AAEJ,SAAA,MAAA,IAAWf,mBAAmBa,gBAAAA,EAAkB;AAC9C,UAAA,OAAOb,OAAAA,CAAQW,OAAAA,CAAQd,GAAAA,CACrB,CAACc,OAAAA,MAAuC;YACtC8B,IAAAA,EAAM,WAAA;YACN,GAAI9B,OAAAA,CAAQG,SAAS,MAAA,IAAU;AAC7BH,cAAAA,OAAAA,EAASA,OAAAA,CAAQI;AACnB,aAAA;YACA,GAAIJ,OAAAA,CAAQG,SAAS,WAAA,IAAe;AAClCM,cAAAA,EAAAA,EAAIT,OAAAA,CAAQQ,UAAAA;cACZL,IAAAA,EAAM,UAAA;cACNQ,QAAAA,EAAU;AACRpD,gBAAAA,IAAAA,EAAMyC,OAAAA,CAAQU,QAAAA;gBACdG,SAAAA,EAAWoB,IAAAA,CAAKC,SAAAA,CAAUlC,OAAAA,CAAQpC,KAAK;AACzC;AACF,aAAA;AACA,YAAA,GAAA,CAAI,MAAA;AACF,cAAA,IAAIoC,OAAAA,CAAQG,SAAS,MAAA,EAAQ;AAC3B,gBAAA,MAAM,IAAIkC,MAAM,8CAAA,CAAA;AAClB;AACA,cAAA,OAAO,EAAC;aACV;WACF,CAAA,CAAA;SAEJ,MAAO;AACL,UAAA,OAAO;AAAChD,YAAAA,OAAAA,CAAQiD,OAAAA;;AAClB;OACF,CAAA;AACAC,MAAAA,OAAAA,EAAS,KAAK9F,MAAAA,CAAO8F,OAAAA;AACrBC,MAAAA,SAAAA,EAAW,KAAK/F,MAAAA,CAAO+F,SAAAA;MACvBtB,KAAAA,EAAOuB,OAAAA,CAAQvB,KAAAA,CAAAA,GAASzD,MAAAA,GAAYyD,KAAAA;MACpCwB,cAAAA,EAAgBjF,MAAAA;AAChB,MAAA,GAAIG,MAAM8E,cAAAA,IAAkB;QAC1BA,cAAAA,EAAgB;AAAEvC,UAAAA,IAAAA,EAAM7D,IAAAA,CAAKqG;AAAY;AAC3C,OAAA;AACAC,MAAAA,IAAAA,EAAMhF,KAAAA,CAAMgF,IAAAA;AACZC,MAAAA,gBAAAA,EAAkBjF,KAAAA,CAAMiF,gBAAAA;AACxBC,MAAAA,WAAAA,EAAalF,KAAAA,CAAMkF,WAAAA;AACnBC,MAAAA,CAAAA,EAAGnF,KAAAA,CAAMmF,CAAAA;AACTC,MAAAA,SAAAA,EAAWpF,KAAAA,CAAMoF,SAAAA;AACjBC,MAAAA,eAAAA,EAAiBrF,KAAAA,CAAMqF,eAAAA;AACvBC,MAAAA,IAAAA,EAAMtF,KAAAA,CAAMuF,aAAAA;AACZC,MAAAA,IAAAA,EAAMxF,KAAAA,CAAMwF,IAAAA;MACZ,GAAG5B;AACL,KAAA;AACF;EAEA6B,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACTnG,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdM,UAAAA,EAAY8F,WAAAA,CAAY,KAAK9F,UAAU,CAAA;AACvCf,MAAAA,MAAAA,EAAQ,IAAA,CAAKA;AACf,KAAA;AACF;AAEA8G,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"chat.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  ChatConfig,\n  ChatModel,\n  ChatModelEmitter,\n  ChatModelInput,\n  ChatModelOutput,\n  ChatModelParameters,\n  ChatModelToolChoiceSupport,\n} from \"@/backend/chat.js\";\nimport { WatsonxClient, WatsonxClientSettings } from \"@/adapters/watsonx/backend/client.js\";\nimport { findLast, isEmpty, isTruthy } from \"remeda\";\nimport WatsonxAiMlVml_v1, {\n  TextChatMessages,\n  TextChatParameterTools,\n  TextChatParams,\n  TextChatResultChoice,\n  TextChatUsage,\n} from \"@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, SystemMessage, ToolMessage } from \"@/backend/message.js\";\nimport { ToolCallPart } from \"ai\";\nimport Type = WatsonxAiMlVml_v1.TextChatResponseFormat.Constants.Type;\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport { NotImplementedError } from \"@/errors.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport type {\n  TextChatMessageAssistant,\n  TextChatMessageSystem,\n  TextChatMessageTool,\n} from \"@ibm-cloud/watsonx-ai\";\n\nexport class WatsonxChatModel extends ChatModel {\n  protected readonly client: WatsonxClient;\n  public readonly emitter: ChatModelEmitter = Emitter.root.child({\n    namespace: [\"backend\", \"watsonx\", \"chat\"],\n    creator: this,\n  });\n  public readonly toolChoiceSupport: ChatModelToolChoiceSupport[] = [\"none\", \"single\"];\n\n  get providerId() {\n    return \"watsonx\";\n  }\n\n  constructor(\n    public readonly modelId = getEnv(\"WATSONX_CHAT_MODEL\", \"ibm/granite-3-8b-instruct\"),\n    client?: WatsonxClient | WatsonxClientSettings,\n  ) {\n    super();\n    this.client = WatsonxClient.ensure(client);\n  }\n\n  config(config: ChatConfig) {\n    super.config(config);\n    for (const name of [\"topK\"] as (keyof ChatModelParameters)[]) {\n      if (this.parameters[name] !== undefined) {\n        this.parameters[name] = undefined;\n        throw new NotImplementedError(\n          `Setting \"${name}\" parameter is not supported for WatsonX Chat Models.`,\n        );\n      }\n    }\n    return this;\n  }\n\n  protected async _create(input: ChatModelInput, run: GetRunContext<any>) {\n    const { result } = await this.client.instance.textChat({\n      ...(await this.prepareInput(input)),\n      signal: run.signal,\n    });\n    const { messages, finishReason, usage } = this.extractOutput(result.choices, result.usage);\n    return new ChatModelOutput(messages, usage, finishReason);\n  }\n\n  async *_createStream(input: ChatModelInput, run: GetRunContext<this>) {\n    const stream = await this.client.instance.textChatStream({\n      ...(await this.prepareInput(input)),\n      signal: run.signal,\n      returnObject: true,\n    });\n    for await (const raw of stream) {\n      if (run.signal.aborted) {\n        stream.controller.abort(run.signal.aborted);\n        break;\n      }\n      const { messages, finishReason, usage } = this.extractOutput(\n        raw.data.choices.map(({ delta, ...choice }) => ({ ...choice, message: delta })),\n        raw.data.usage,\n      );\n      yield new ChatModelOutput(messages, usage, finishReason);\n    }\n  }\n\n  protected extractOutput(choices: TextChatResultChoice[], usage?: TextChatUsage) {\n    return {\n      finishReason: findLast(choices, (choice) => Boolean(choice?.finish_reason))\n        ?.finish_reason as ChatModelOutput[\"finishReason\"],\n      usage: usage\n        ? {\n            completionTokens: usage.completion_tokens ?? 0,\n            promptTokens: usage.prompt_tokens ?? 0,\n            totalTokens: usage.total_tokens ?? 0,\n          }\n        : undefined,\n      messages: choices\n        .flatMap(({ message }) => {\n          const messages: Message[] = [];\n          if (message?.content) {\n            const msg = new AssistantMessage({ type: \"text\", text: message.content });\n            // msg.role = message.role || msg.role;\n            messages.push(msg);\n          }\n          if (message?.tool_calls) {\n            const msg = new AssistantMessage(\n              message.tool_calls.map(\n                (call): ToolCallPart => ({\n                  type: \"tool-call\",\n                  toolCallId: call.id,\n                  toolName: call.function.name,\n                  input: parseBrokenJson(call.function.arguments),\n                }),\n              ),\n            );\n            // msg.role = message.role || msg.role;\n            messages.push(msg);\n          }\n          if (message?.refusal) {\n            const msg = new AssistantMessage({ type: \"text\", text: message.refusal });\n            // msg.role = message.role || msg.role;\n            messages.push(msg);\n          }\n          return messages;\n        })\n        .filter(isTruthy),\n    };\n  }\n\n  protected async prepareInput(overrides: ChatModelInput): Promise<TextChatParams> {\n    const input: ChatModelInput = { ...this.parameters, ...overrides };\n    const tools: TextChatParameterTools[] = await Promise.all(\n      (input.tools ?? []).map(async (tool) => ({\n        type: \"function\",\n        function: {\n          name: tool.name,\n          description: tool.description,\n          parameters: await tool.getInputJsonSchema(),\n        },\n      })),\n    );\n\n    const toolConfig: Pick<TextChatParams, \"toolChoice\" | \"toolChoiceOption\"> = {\n      toolChoice: undefined,\n      toolChoiceOption: undefined,\n    };\n    if (input.toolChoice) {\n      if (input.toolChoice instanceof Tool) {\n        toolConfig.toolChoice = {\n          type: \"function\",\n          function: { name: input.toolChoice.name },\n        };\n      } else if (this.toolChoiceSupport.includes(input.toolChoice)) {\n        toolConfig.toolChoiceOption = input.toolChoice;\n      }\n    }\n\n    return {\n      modelId: this.modelId,\n      messages: input.messages.flatMap((message): TextChatMessages[] => {\n        if (message instanceof ToolMessage) {\n          return message.content.map(\n            (content): TextChatMessageTool => ({\n              role: \"tool\",\n              content:\n                typeof content.output.value === \"string\"\n                  ? content.output.value\n                  : JSON.stringify(content),\n              tool_call_id: content.toolCallId,\n            }),\n          );\n        } else if (message instanceof SystemMessage) {\n          return message.content.map(\n            (content): TextChatMessageSystem => ({\n              role: \"system\",\n              content: content.text,\n            }),\n          );\n        } else if (message instanceof AssistantMessage) {\n          return message.content.map(\n            (content): TextChatMessageAssistant => ({\n              role: \"assistant\",\n              ...(content.type === \"text\" && {\n                content: content.text,\n              }),\n              ...(content.type === \"tool-call\" && {\n                id: content.toolCallId,\n                type: \"function\",\n                function: {\n                  name: content.toolName,\n                  arguments: JSON.stringify(content.input),\n                },\n              }),\n              ...(() => {\n                if (content.type === \"file\") {\n                  throw new Error(\"Assistants cannot send a file in IBM watsonx\");\n                }\n                return {};\n              })(),\n            }),\n          );\n        } else {\n          return [message.toPlain()];\n        }\n      }),\n      spaceId: this.client.spaceId,\n      projectId: this.client.projectId,\n      tools: isEmpty(tools) ? undefined : tools,\n      responseFormat: undefined,\n      ...(input.responseFormat && {\n        responseFormat: { type: Type.JSON_OBJECT },\n      }),\n      topP: input.topP,\n      frequencyPenalty: input.frequencyPenalty,\n      temperature: input.temperature,\n      n: input.n,\n      maxTokens: input.maxTokens,\n      presencePenalty: input.presencePenalty,\n      stop: input.stopSequences,\n      seed: input.seed,\n      ...toolConfig,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      modelId: this.modelId,\n      parameters: shallowCopy(this.parameters),\n      client: this.client,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/adapters/watsonx/backend/embedding.d.cts

@@ -1,7 +1,7 @@
 import { WatsonxClient, WatsonxClientSettings } from './client.cjs';
 import { EmbeddingModel, EmbeddingModelEvents, EmbeddingModelInput, EmbeddingModelOutput } from '../../../backend/embedding.cjs';
 import { EmbeddingParameters } from '@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js';
-import { E as Emitter } from '../../../emitter-DWEsd-ud.cjs';
+import { E as Emitter } from '../../../emitter-CGKd_-kK.cjs';
 import '@ibm-cloud/watsonx-ai';
 import 'ibm-cloud-sdk-core';
 import '../../../backend/client.cjs';
@@ -11,7 +11,7 @@ import '../../../internals/helpers/guards.cjs';
 import '../../../context.cjs';
 import '../../../internals/helpers/promise.cjs';
 import '../../../errors.cjs';
-import '../../../chat-CR4yfOR9.cjs';
+import '../../../chat-DB_cf15S.cjs';
 import '../../../backend/message.cjs';
 import 'ai';
 import 'promise-based-task';

package/dist/adapters/watsonx/backend/embedding.d.ts

@@ -1,7 +1,7 @@
 import { WatsonxClient, WatsonxClientSettings } from './client.js';
 import { EmbeddingModel, EmbeddingModelEvents, EmbeddingModelInput, EmbeddingModelOutput } from '../../../backend/embedding.js';
 import { EmbeddingParameters } from '@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js';
-import { E as Emitter } from '../../../emitter-C9nwGLt6.js';
+import { E as Emitter } from '../../../emitter-CN7I4uSV.js';
 import '@ibm-cloud/watsonx-ai';
 import 'ibm-cloud-sdk-core';
 import '../../../backend/client.js';
@@ -11,7 +11,7 @@ import '../../../internals/helpers/guards.js';
 import '../../../context.js';
 import '../../../internals/helpers/promise.js';
 import '../../../errors.js';
-import '../../../chat-W9j8T6UU.js';
+import '../../../chat-BbCYMAdE.js';
 import '../../../backend/message.js';
 import 'ai';
 import 'promise-based-task';

package/dist/adapters/xai/backend/chat.cjs

@@ -10,9 +10,10 @@ class XAIChatModel extends chat_cjs.VercelChatModel {
   static {
     __name(this, "XAIChatModel");
   }
-  constructor(modelId = env_cjs.getEnv("XAI_CHAT_MODEL", "grok-3-mini"), settings = {}, client) {
-    const model = client_cjs.XaiClient.ensure(client).instance.languageModel(modelId, settings);
+  constructor(modelId = env_cjs.getEnv("XAI_CHAT_MODEL", "grok-3-mini"), parameters = {}, client) {
+    const model = client_cjs.XaiClient.ensure(client).instance.languageModel(modelId);
     super(model);
+    Object.assign(this.parameters, parameters ?? {});
   }
   static {
     XAIChatModel.register();

package/dist/adapters/xai/backend/chat.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/adapters/xai/backend/chat.ts"],"names":["XAIChatModel","VercelChatModel","modelId","getEnv","settings","client","model","XaiClient","ensure","instance","languageModel","register"],"mappings":";;;;;;;;AAcO,MAAMA,qBAAqBC,wBAAAA,CAAAA;EAdlC;;;EAeE,WAAA,CACEC,OAAAA,GAA0BC,eAAO,gBAAA,EAAkB,aAAA,GACnDC,QAAAA,GAAiC,IACjCC,MAAAA,EACA;AACA,IAAA,MAAMC,KAAAA,GAAQC,qBAAUC,MAAAA,CAAOH,MAAAA,EAAQI,QAAAA,CAASC,aAAAA,CAAcR,SAASE,QAAAA,CAAAA;AACvE,IAAA,KAAA,CAAME,KAAAA,CAAAA;AACR;EAEA;AACEN,IAAAA,YAAAA,CAAaW,QAAAA,EAAQ;AACvB;AACF","file":"chat.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { VercelChatModel } from \"@/adapters/vercel/backend/chat.js\";\nimport { XaiClient, type XaiClientSettings } from \"@/adapters/xai/backend/client.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport type { XaiProvider } from \"@ai-sdk/xai\";\n\ntype XaiParameters = Parameters<XaiProvider[\"languageModel\"]>;\nexport type XAIChatModelId = NonNullable<XaiParameters[0]>;\nexport type XAIChatModelSettings = NonNullable<XaiParameters[1]>;\n\nexport class XAIChatModel extends VercelChatModel {\n  constructor(\n    modelId: XAIChatModelId = getEnv(\"XAI_CHAT_MODEL\", \"grok-3-mini\"),\n    settings: XAIChatModelSettings = {},\n    client?: XaiClientSettings | XaiClient,\n  ) {\n    const model = XaiClient.ensure(client).instance.languageModel(modelId, settings);\n    super(model);\n  }\n\n  static {\n    XAIChatModel.register();\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/adapters/xai/backend/chat.ts"],"names":["XAIChatModel","VercelChatModel","modelId","getEnv","parameters","client","model","XaiClient","ensure","instance","languageModel","Object","assign","register"],"mappings":";;;;;;;;AAcO,MAAMA,qBAAqBC,wBAAAA,CAAAA;EAdlC;;;EAeE,WAAA,CACEC,OAAAA,GAA0BC,eAAO,gBAAA,EAAkB,aAAA,GACnDC,UAAAA,GAAkC,IAClCC,MAAAA,EACA;AACA,IAAA,MAAMC,QAAQC,oBAAAA,CAAUC,MAAAA,CAAOH,MAAAA,CAAAA,CAAQI,QAAAA,CAASC,cAAcR,OAAAA,CAAAA;AAC9D,IAAA,KAAA,CAAMI,KAAAA,CAAAA;AACNK,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKR,UAAAA,EAAYA,UAAAA,IAAc,EAAC,CAAA;AAChD;EAEA;AACEJ,IAAAA,YAAAA,CAAaa,QAAAA,EAAQ;AACvB;AACF","file":"chat.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { VercelChatModel } from \"@/adapters/vercel/backend/chat.js\";\nimport { XaiClient, type XaiClientSettings } from \"@/adapters/xai/backend/client.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport type { XaiProvider } from \"@ai-sdk/xai\";\nimport { ChatModelParameters } from \"@/backend/chat.js\";\n\ntype XaiParameters = Parameters<XaiProvider[\"languageModel\"]>;\nexport type XAIChatModelId = NonNullable<XaiParameters[0]>;\n\nexport class XAIChatModel extends VercelChatModel {\n  constructor(\n    modelId: XAIChatModelId = getEnv(\"XAI_CHAT_MODEL\", \"grok-3-mini\"),\n    parameters: ChatModelParameters = {},\n    client?: XaiClientSettings | XaiClient,\n  ) {\n    const model = XaiClient.ensure(client).instance.languageModel(modelId);\n    super(model);\n    Object.assign(this.parameters, parameters ?? {});\n  }\n\n  static {\n    XAIChatModel.register();\n  }\n}\n"]}

package/dist/adapters/xai/backend/chat.d.cts

@@ -1,17 +1,17 @@
 import { VercelChatModel } from '../../vercel/backend/chat.cjs';
 import { XaiClientSettings, XaiClient } from './client.cjs';
 import { XaiProvider } from '@ai-sdk/xai';
+import { C as ChatModelParameters } from '../../../chat-DB_cf15S.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
 import '../../../errors.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';
-import '../../../chat-CR4yfOR9.cjs';
-import '../../../backend/message.cjs';
 import 'ai';
+import '../../../emitter-CGKd_-kK.cjs';
+import '../../../backend/message.cjs';
 import '../../../context.cjs';
-import '../../../emitter-DWEsd-ud.cjs';
 import '../../../internals/helpers/promise.cjs';
 import 'promise-based-task';
 import '../../../cache/base.cjs';
@@ -31,9 +31,8 @@ import '../../../backend/client.cjs';
 
 type XaiParameters = Parameters<XaiProvider["languageModel"]>;
 type XAIChatModelId = NonNullable<XaiParameters[0]>;
-type XAIChatModelSettings = NonNullable<XaiParameters[1]>;
 declare class XAIChatModel extends VercelChatModel {
-    constructor(modelId?: XAIChatModelId, settings?: XAIChatModelSettings, client?: XaiClientSettings | XaiClient);
+    constructor(modelId?: XAIChatModelId, parameters?: ChatModelParameters, client?: XaiClientSettings | XaiClient);
 }
 
-export { XAIChatModel, type XAIChatModelId, type XAIChatModelSettings };
+export { XAIChatModel, type XAIChatModelId };

package/dist/adapters/xai/backend/chat.d.ts

@@ -1,17 +1,17 @@
 import { VercelChatModel } from '../../vercel/backend/chat.js';
 import { XaiClientSettings, XaiClient } from './client.js';
 import { XaiProvider } from '@ai-sdk/xai';
+import { C as ChatModelParameters } from '../../../chat-BbCYMAdE.js';
 import '../../../logger/logger.js';
 import 'pino';
 import '../../../errors.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';
-import '../../../chat-W9j8T6UU.js';
-import '../../../backend/message.js';
 import 'ai';
+import '../../../emitter-CN7I4uSV.js';
+import '../../../backend/message.js';
 import '../../../context.js';
-import '../../../emitter-C9nwGLt6.js';
 import '../../../internals/helpers/promise.js';
 import 'promise-based-task';
 import '../../../cache/base.js';
@@ -31,9 +31,8 @@ import '../../../backend/client.js';
 
 type XaiParameters = Parameters<XaiProvider["languageModel"]>;
 type XAIChatModelId = NonNullable<XaiParameters[0]>;
-type XAIChatModelSettings = NonNullable<XaiParameters[1]>;
 declare class XAIChatModel extends VercelChatModel {
-    constructor(modelId?: XAIChatModelId, settings?: XAIChatModelSettings, client?: XaiClientSettings | XaiClient);
+    constructor(modelId?: XAIChatModelId, parameters?: ChatModelParameters, client?: XaiClientSettings | XaiClient);
 }
 
-export { XAIChatModel, type XAIChatModelId, type XAIChatModelSettings };
+export { XAIChatModel, type XAIChatModelId };

package/dist/adapters/xai/backend/chat.js

@@ -8,9 +8,10 @@ class XAIChatModel extends VercelChatModel {
   static {
     __name(this, "XAIChatModel");
   }
-  constructor(modelId = getEnv("XAI_CHAT_MODEL", "grok-3-mini"), settings = {}, client) {
-    const model = XaiClient.ensure(client).instance.languageModel(modelId, settings);
+  constructor(modelId = getEnv("XAI_CHAT_MODEL", "grok-3-mini"), parameters = {}, client) {
+    const model = XaiClient.ensure(client).instance.languageModel(modelId);
     super(model);
+    Object.assign(this.parameters, parameters ?? {});
   }
   static {
     XAIChatModel.register();

package/dist/adapters/xai/backend/chat.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/adapters/xai/backend/chat.ts"],"names":["XAIChatModel","VercelChatModel","modelId","getEnv","settings","client","model","XaiClient","ensure","instance","languageModel","register"],"mappings":";;;;;;AAcO,MAAMA,qBAAqBC,eAAAA,CAAAA;EAdlC;;;EAeE,WAAA,CACEC,OAAAA,GAA0BC,OAAO,gBAAA,EAAkB,aAAA,GACnDC,QAAAA,GAAiC,IACjCC,MAAAA,EACA;AACA,IAAA,MAAMC,KAAAA,GAAQC,UAAUC,MAAAA,CAAOH,MAAAA,EAAQI,QAAAA,CAASC,aAAAA,CAAcR,SAASE,QAAAA,CAAAA;AACvE,IAAA,KAAA,CAAME,KAAAA,CAAAA;AACR;EAEA;AACEN,IAAAA,YAAAA,CAAaW,QAAAA,EAAQ;AACvB;AACF","file":"chat.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { VercelChatModel } from \"@/adapters/vercel/backend/chat.js\";\nimport { XaiClient, type XaiClientSettings } from \"@/adapters/xai/backend/client.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport type { XaiProvider } from \"@ai-sdk/xai\";\n\ntype XaiParameters = Parameters<XaiProvider[\"languageModel\"]>;\nexport type XAIChatModelId = NonNullable<XaiParameters[0]>;\nexport type XAIChatModelSettings = NonNullable<XaiParameters[1]>;\n\nexport class XAIChatModel extends VercelChatModel {\n  constructor(\n    modelId: XAIChatModelId = getEnv(\"XAI_CHAT_MODEL\", \"grok-3-mini\"),\n    settings: XAIChatModelSettings = {},\n    client?: XaiClientSettings | XaiClient,\n  ) {\n    const model = XaiClient.ensure(client).instance.languageModel(modelId, settings);\n    super(model);\n  }\n\n  static {\n    XAIChatModel.register();\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/adapters/xai/backend/chat.ts"],"names":["XAIChatModel","VercelChatModel","modelId","getEnv","parameters","client","model","XaiClient","ensure","instance","languageModel","Object","assign","register"],"mappings":";;;;;;AAcO,MAAMA,qBAAqBC,eAAAA,CAAAA;EAdlC;;;EAeE,WAAA,CACEC,OAAAA,GAA0BC,OAAO,gBAAA,EAAkB,aAAA,GACnDC,UAAAA,GAAkC,IAClCC,MAAAA,EACA;AACA,IAAA,MAAMC,QAAQC,SAAAA,CAAUC,MAAAA,CAAOH,MAAAA,CAAAA,CAAQI,QAAAA,CAASC,cAAcR,OAAAA,CAAAA;AAC9D,IAAA,KAAA,CAAMI,KAAAA,CAAAA;AACNK,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKR,UAAAA,EAAYA,UAAAA,IAAc,EAAC,CAAA;AAChD;EAEA;AACEJ,IAAAA,YAAAA,CAAaa,QAAAA,EAAQ;AACvB;AACF","file":"chat.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { VercelChatModel } from \"@/adapters/vercel/backend/chat.js\";\nimport { XaiClient, type XaiClientSettings } from \"@/adapters/xai/backend/client.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport type { XaiProvider } from \"@ai-sdk/xai\";\nimport { ChatModelParameters } from \"@/backend/chat.js\";\n\ntype XaiParameters = Parameters<XaiProvider[\"languageModel\"]>;\nexport type XAIChatModelId = NonNullable<XaiParameters[0]>;\n\nexport class XAIChatModel extends VercelChatModel {\n  constructor(\n    modelId: XAIChatModelId = getEnv(\"XAI_CHAT_MODEL\", \"grok-3-mini\"),\n    parameters: ChatModelParameters = {},\n    client?: XaiClientSettings | XaiClient,\n  ) {\n    const model = XaiClient.ensure(client).instance.languageModel(modelId);\n    super(model);\n    Object.assign(this.parameters, parameters ?? {});\n  }\n\n  static {\n    XAIChatModel.register();\n  }\n}\n"]}

package/dist/{agent-I0Saf4A1.d.cts → agent-BMfyig7X.d.cts}

@@ -1,12 +1,12 @@
-import { GetRunContext, RunContext } from './context.cjs';
-import { b as AgentMeta, a as BaseAgent } from './base-BP7v-NSL.cjs';
+import { GetRunContext, RunContext, MiddlewareType } from './context.cjs';
+import { b as AgentMeta, a as BaseAgent } from './base-v-PYIq5G.cjs';
 import { AnyTool } from './tools/base.cjs';
 import { BaseMemory } from './memory/base.cjs';
-import { E as Emitter } from './emitter-DWEsd-ud.cjs';
+import { E as Emitter } from './emitter-CGKd_-kK.cjs';
 import { ReActAgentRunOptions, ReActAgentRunIteration, ReActAgentCallbacks, ReActAgentParserInput, ReActAgentMeta, ReActAgentRunInput, ReActAgentIterationToolResult, ReActAgentTemplates, ReActAgentExecutionConfig, ReActAgentRunOutput } from './agents/react/types.cjs';
 import { LinePrefixParser } from './parsers/linePrefix.cjs';
 import { Serializable } from './internals/serializable.cjs';
-import { m as ChatModel } from './chat-CR4yfOR9.cjs';
+import { m as ChatModel } from './chat-DB_cf15S.cjs';
 import { RetryCounter } from './internals/helpers/counter.cjs';
 
 interface ReActAgentRunnerLLMInput {
@@ -78,6 +78,7 @@ declare class ReActAgent extends BaseAgent<ReActAgentRunInput, ReActAgentRunOutp
         emitter: Emitter<ReActAgentCallbacks>;
         runner: new (input: ReActAgentInput, options: ReActAgentRunOptions, run: RunContext<ReActAgent, any>) => BaseRunner;
         isRunning: boolean;
+        middlewares: MiddlewareType<any>[];
     };
 }
 

package/dist/{agent-CtrCeN7o.d.ts → agent-Dxw8AGWb.d.ts}

@@ -1,12 +1,12 @@
-import { GetRunContext, RunContext } from './context.js';
-import { b as AgentMeta, a as BaseAgent } from './base-BsY2baWy.js';
+import { GetRunContext, RunContext, MiddlewareType } from './context.js';
+import { b as AgentMeta, a as BaseAgent } from './base-BurRTBA7.js';
 import { AnyTool } from './tools/base.js';
 import { BaseMemory } from './memory/base.js';
-import { E as Emitter } from './emitter-C9nwGLt6.js';
+import { E as Emitter } from './emitter-CN7I4uSV.js';
 import { ReActAgentRunOptions, ReActAgentRunIteration, ReActAgentCallbacks, ReActAgentParserInput, ReActAgentMeta, ReActAgentRunInput, ReActAgentIterationToolResult, ReActAgentTemplates, ReActAgentExecutionConfig, ReActAgentRunOutput } from './agents/react/types.js';
 import { LinePrefixParser } from './parsers/linePrefix.js';
 import { Serializable } from './internals/serializable.js';
-import { m as ChatModel } from './chat-W9j8T6UU.js';
+import { m as ChatModel } from './chat-BbCYMAdE.js';
 import { RetryCounter } from './internals/helpers/counter.js';
 
 interface ReActAgentRunnerLLMInput {
@@ -78,6 +78,7 @@ declare class ReActAgent extends BaseAgent<ReActAgentRunInput, ReActAgentRunOutp
         emitter: Emitter<ReActAgentCallbacks>;
         runner: new (input: ReActAgentInput, options: ReActAgentRunOptions, run: RunContext<ReActAgent, any>) => BaseRunner;
         isRunning: boolean;
+        middlewares: MiddlewareType<any>[];
     };
 }
 

package/dist/agents/base.cjs

@@ -3,6 +3,7 @@
 var errors_cjs = require('../errors.cjs');
 var serializable_cjs = require('../internals/serializable.cjs');
 var context_cjs = require('../context.cjs');
+var utils_cjs = require('../serializer/utils.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -16,6 +17,7 @@ class BaseAgent extends serializable_cjs.Serializable {
     __name(this, "BaseAgent");
   }
   isRunning = false;
+  middlewares = [];
   run(...[input, options]) {
     if (this.isRunning) {
       throw new AgentError("Agent is already running!");
@@ -41,7 +43,7 @@ class BaseAgent extends serializable_cjs.Serializable {
       } finally {
         this.isRunning = false;
       }
-    });
+    }).middleware(...this.middlewares);
   }
   destroy() {
     this.emitter.destroy();
@@ -56,7 +58,8 @@ class BaseAgent extends serializable_cjs.Serializable {
   createSnapshot() {
     return {
       isRunning: false,
-      emitter: this.emitter
+      emitter: this.emitter,
+      middlewares: utils_cjs.shallowCopy(this.middlewares)
     };
   }
   loadSnapshot(snapshot) {

package/dist/agents/base.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","destroy","emitter","meta","name","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;AAYO,MAAMA,mBAAmBC,yBAAAA,CAAAA;EAZhC;;;AAYgD;AAMzC,MAAeC,kBAIZC,6BAAAA,CAAAA;EAtBV;;;EAuBYC,SAAAA,GAAY,KAAA;EAIfC,GAAAA,CAAAA,GACF,CAACC,KAAAA,EAAOC,OAAAA,CAAAA,EAGX;AACA,IAAA,IAAI,KAAKH,SAAAA,EAAW;AAClB,MAAA,MAAM,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAA,OAAOQ,sBAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAA,IAAI;AACF,QAAA,IAAA,CAAKR,SAAAA,GAAY,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKS,IAAAA,CAAKP,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAAA,EAAG;AACV,QAAA,IAAIA,aAAad,UAAAA,EAAY;AAC3B,UAAA,MAAMc,CAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAId,WAAW,CAAA,mCAAA,CAAA,EAAuC;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAAA,GAAY,KAAA;AACnB;KACF,CAAA;AAEJ;EAQAW,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAAA,EAAO;AACtB;AAKA,EAAA,IAAWE,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,IAAA,CAAK,YAAYA,IAAAA,IAAQ,WAAA;MAC/BC,WAAAA,EAAa,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEjB,SAAAA,EAAW,KAAA;AAAOY,MAAAA,OAAAA,EAAS,IAAA,CAAKA;AAAQ,KAAA;AACnD;AAEAM,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    );\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return { isRunning: false, emitter: this.emitter };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","middlewares","run","input","options","RunContext","enter","signal","params","context","_run","e","middleware","destroy","emitter","meta","name","description","tools","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;;AAaO,MAAMA,mBAAmBC,yBAAAA,CAAAA;EAbhC;;;AAagD;AAMzC,MAAeC,kBAIZC,6BAAAA,CAAAA;EAvBV;;;EAwBYC,SAAAA,GAAY,KAAA;AAGNC,EAAAA,WAAAA,GAA6C,EAAA;EAEtDC,GAAAA,CAAAA,GACF,CAACC,KAAAA,EAAOC,OAAAA,CAAAA,EAGX;AACA,IAAA,IAAI,KAAKJ,SAAAA,EAAW;AAClB,MAAA,MAAM,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAA,OAAOS,sBAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAA,IAAI;AACF,QAAA,IAAA,CAAKT,SAAAA,GAAY,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKU,IAAAA,CAAKP,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAAA,EAAG;AACV,QAAA,IAAIA,aAAaf,UAAAA,EAAY;AAC3B,UAAA,MAAMe,CAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAIf,WAAW,CAAA,mCAAA,CAAA,EAAuC;AAACe,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKX,SAAAA,GAAY,KAAA;AACnB;AACF,KAAA,CAAA,CACAY,UAAAA,CAAU,GAAI,IAAA,CAAKX,WAAW,CAAA;AAClC;EAQAY,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAAA,EAAO;AACtB;AAKA,EAAA,IAAWE,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,IAAA,CAAK,YAAYA,IAAAA,IAAQ,WAAA;MAC/BC,WAAAA,EAAa,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;MACLnB,SAAAA,EAAW,KAAA;AACXc,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdb,WAAAA,EAAamB,qBAAAA,CAAY,KAAKnB,WAAW;AAC3C,KAAA;AACF;AAEAoB,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, MiddlewareType, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n  public readonly middlewares: MiddlewareType<typeof this>[] = [];\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    ).middleware(...this.middlewares);\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return {\n      isRunning: false,\n      emitter: this.emitter,\n      middlewares: shallowCopy(this.middlewares) as MiddlewareType<any>[],\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/agents/base.d.cts

@@ -1,8 +1,8 @@
 import '../context.cjs';
 import '../errors.cjs';
-export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-BP7v-NSL.cjs';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-v-PYIq5G.cjs';
 import '../internals/serializable.cjs';
-import '../emitter-DWEsd-ud.cjs';
+import '../emitter-CGKd_-kK.cjs';
 import '../memory/base.cjs';
 import '../internals/helpers/promise.cjs';
 import '../internals/types.cjs';

package/dist/agents/base.d.ts

@@ -1,8 +1,8 @@
 import '../context.js';
 import '../errors.js';
-export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-BsY2baWy.js';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-BurRTBA7.js';
 import '../internals/serializable.js';
-import '../emitter-C9nwGLt6.js';
+import '../emitter-CN7I4uSV.js';
 import '../memory/base.js';
 import '../internals/helpers/promise.js';
 import '../internals/types.js';

package/dist/agents/base.js

@@ -1,6 +1,7 @@
 import { FrameworkError } from '../errors.js';
 import { Serializable } from '../internals/serializable.js';
 import { RunContext } from '../context.js';
+import { shallowCopy } from '../serializer/utils.js';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -14,6 +15,7 @@ class BaseAgent extends Serializable {
     __name(this, "BaseAgent");
   }
   isRunning = false;
+  middlewares = [];
   run(...[input, options]) {
     if (this.isRunning) {
       throw new AgentError("Agent is already running!");
@@ -39,7 +41,7 @@ class BaseAgent extends Serializable {
       } finally {
         this.isRunning = false;
       }
-    });
+    }).middleware(...this.middlewares);
   }
   destroy() {
     this.emitter.destroy();
@@ -54,7 +56,8 @@ class BaseAgent extends Serializable {
   createSnapshot() {
     return {
       isRunning: false,
-      emitter: this.emitter
+      emitter: this.emitter,
+      middlewares: shallowCopy(this.middlewares)
     };
   }
   loadSnapshot(snapshot) {

package/dist/agents/base.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","destroy","emitter","meta","name","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;AAYO,MAAMA,mBAAmBC,cAAAA,CAAAA;EAZhC;;;AAYgD;AAMzC,MAAeC,kBAIZC,YAAAA,CAAAA;EAtBV;;;EAuBYC,SAAAA,GAAY,KAAA;EAIfC,GAAAA,CAAAA,GACF,CAACC,KAAAA,EAAOC,OAAAA,CAAAA,EAGX;AACA,IAAA,IAAI,KAAKH,SAAAA,EAAW;AAClB,MAAA,MAAM,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAA,OAAOQ,UAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAA,IAAI;AACF,QAAA,IAAA,CAAKR,SAAAA,GAAY,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKS,IAAAA,CAAKP,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAAA,EAAG;AACV,QAAA,IAAIA,aAAad,UAAAA,EAAY;AAC3B,UAAA,MAAMc,CAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAId,WAAW,CAAA,mCAAA,CAAA,EAAuC;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAAA,GAAY,KAAA;AACnB;KACF,CAAA;AAEJ;EAQAW,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAAA,EAAO;AACtB;AAKA,EAAA,IAAWE,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,IAAA,CAAK,YAAYA,IAAAA,IAAQ,WAAA;MAC/BC,WAAAA,EAAa,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEjB,SAAAA,EAAW,KAAA;AAAOY,MAAAA,OAAAA,EAAS,IAAA,CAAKA;AAAQ,KAAA;AACnD;AAEAM,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    );\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return { isRunning: false, emitter: this.emitter };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","middlewares","run","input","options","RunContext","enter","signal","params","context","_run","e","middleware","destroy","emitter","meta","name","description","tools","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;AAaO,MAAMA,mBAAmBC,cAAAA,CAAAA;EAbhC;;;AAagD;AAMzC,MAAeC,kBAIZC,YAAAA,CAAAA;EAvBV;;;EAwBYC,SAAAA,GAAY,KAAA;AAGNC,EAAAA,WAAAA,GAA6C,EAAA;EAEtDC,GAAAA,CAAAA,GACF,CAACC,KAAAA,EAAOC,OAAAA,CAAAA,EAGX;AACA,IAAA,IAAI,KAAKJ,SAAAA,EAAW;AAClB,MAAA,MAAM,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAA,OAAOS,UAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAA,IAAI;AACF,QAAA,IAAA,CAAKT,SAAAA,GAAY,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKU,IAAAA,CAAKP,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAAA,EAAG;AACV,QAAA,IAAIA,aAAaf,UAAAA,EAAY;AAC3B,UAAA,MAAMe,CAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAIf,WAAW,CAAA,mCAAA,CAAA,EAAuC;AAACe,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKX,SAAAA,GAAY,KAAA;AACnB;AACF,KAAA,CAAA,CACAY,UAAAA,CAAU,GAAI,IAAA,CAAKX,WAAW,CAAA;AAClC;EAQAY,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAAA,EAAO;AACtB;AAKA,EAAA,IAAWE,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,IAAA,CAAK,YAAYA,IAAAA,IAAQ,WAAA;MAC/BC,WAAAA,EAAa,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;MACLnB,SAAAA,EAAW,KAAA;AACXc,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdb,WAAAA,EAAamB,WAAAA,CAAY,KAAKnB,WAAW;AAC3C,KAAA;AACF;AAEAoB,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, MiddlewareType, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n  public readonly middlewares: MiddlewareType<typeof this>[] = [];\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    ).middleware(...this.middlewares);\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return {\n      isRunning: false,\n      emitter: this.emitter,\n      middlewares: shallowCopy(this.middlewares) as MiddlewareType<any>[],\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/agents/experimental/replan/agent.d.cts

@@ -1,12 +1,12 @@
-import { C as Callback, E as Emitter } from '../../../emitter-DWEsd-ud.cjs';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BP7v-NSL.cjs';
+import { C as Callback, E as Emitter } from '../../../emitter-CGKd_-kK.cjs';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-v-PYIq5G.cjs';
 import { GetRunContext } from '../../../context.cjs';
 import { Message } from '../../../backend/message.cjs';
 import { RePlanState } from './prompts.cjs';
 import { BaseMemory } from '../../../memory/base.cjs';
 import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.cjs';
 import { AnyTool } from '../../../tools/base.cjs';
-import { m as ChatModel } from '../../../chat-CR4yfOR9.cjs';
+import { m as ChatModel } from '../../../chat-DB_cf15S.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';

package/dist/agents/experimental/replan/agent.d.ts

@@ -1,12 +1,12 @@
-import { C as Callback, E as Emitter } from '../../../emitter-C9nwGLt6.js';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BsY2baWy.js';
+import { C as Callback, E as Emitter } from '../../../emitter-CN7I4uSV.js';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BurRTBA7.js';
 import { GetRunContext } from '../../../context.js';
 import { Message } from '../../../backend/message.js';
 import { RePlanState } from './prompts.js';
 import { BaseMemory } from '../../../memory/base.js';
 import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.js';
 import { AnyTool } from '../../../tools/base.js';
-import { m as ChatModel } from '../../../chat-W9j8T6UU.js';
+import { m as ChatModel } from '../../../chat-BbCYMAdE.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';

package/dist/agents/experimental/replan/prompts.d.cts

@@ -7,7 +7,7 @@ import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';
 import '../../../context.cjs';
-import '../../../emitter-DWEsd-ud.cjs';
+import '../../../emitter-CGKd_-kK.cjs';
 import '../../../internals/helpers/promise.cjs';
 import 'promise-based-task';
 import '../../../cache/base.cjs';

package/dist/agents/experimental/replan/prompts.d.ts

@@ -7,7 +7,7 @@ import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';
 import '../../../context.js';
-import '../../../emitter-C9nwGLt6.js';
+import '../../../emitter-CN7I4uSV.js';
 import '../../../internals/helpers/promise.js';
 import 'promise-based-task';
 import '../../../cache/base.js';

package/dist/agents/experimental/streamlit/agent.d.cts

@@ -1,15 +1,16 @@
-import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-BP7v-NSL.cjs';
-import { GetRunContext } from '../../../context.cjs';
-import { C as Callback, E as Emitter } from '../../../emitter-DWEsd-ud.cjs';
+import { GetRunContext, MiddlewareType } from '../../../context.cjs';
+import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-v-PYIq5G.cjs';
+import { C as Callback, E as Emitter } from '../../../emitter-CGKd_-kK.cjs';
 import { BaseMemory } from '../../../memory/base.cjs';
 import { Message, UserMessage } from '../../../backend/message.cjs';
 import { StreamlitAgentTemplates } from './prompts.cjs';
 import { TokenMemory } from '../../../memory/tokenMemory.cjs';
-import { m as ChatModel, n as ChatModelOutput } from '../../../chat-CR4yfOR9.cjs';
-import '../../../errors.cjs';
+import { m as ChatModel, n as ChatModelOutput } from '../../../chat-DB_cf15S.cjs';
+import '../../../internals/serializable.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
-import '../../../internals/serializable.cjs';
+import '../../../internals/helpers/promise.cjs';
+import '../../../errors.cjs';
 import '../../../tools/base.cjs';
 import 'ajv';
 import 'promise-based-task';
@@ -17,18 +18,12 @@ import '../../../cache/base.cjs';
 import '../../../internals/helpers/schema.cjs';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../../internals/helpers/promise.cjs';
 import 'ai';
 import '../../../template.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
 
-/**
- * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-
 interface StreamlitAgentInput {
     llm: ChatModel;
     memory: BaseMemory;
@@ -78,6 +73,7 @@ declare class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOu
         input: StreamlitAgentInput;
         isRunning: boolean;
         emitter: Emitter<unknown>;
+        middlewares: MiddlewareType<any>[];
     };
 }
 

package/dist/agents/experimental/streamlit/agent.d.ts

@@ -1,15 +1,16 @@
-import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-BsY2baWy.js';
-import { GetRunContext } from '../../../context.js';
-import { C as Callback, E as Emitter } from '../../../emitter-C9nwGLt6.js';
+import { GetRunContext, MiddlewareType } from '../../../context.js';
+import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-BurRTBA7.js';
+import { C as Callback, E as Emitter } from '../../../emitter-CN7I4uSV.js';
 import { BaseMemory } from '../../../memory/base.js';
 import { Message, UserMessage } from '../../../backend/message.js';
 import { StreamlitAgentTemplates } from './prompts.js';
 import { TokenMemory } from '../../../memory/tokenMemory.js';
-import { m as ChatModel, n as ChatModelOutput } from '../../../chat-W9j8T6UU.js';
-import '../../../errors.js';
+import { m as ChatModel, n as ChatModelOutput } from '../../../chat-BbCYMAdE.js';
+import '../../../internals/serializable.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
-import '../../../internals/serializable.js';
+import '../../../internals/helpers/promise.js';
+import '../../../errors.js';
 import '../../../tools/base.js';
 import 'ajv';
 import 'promise-based-task';
@@ -17,18 +18,12 @@ import '../../../cache/base.js';
 import '../../../internals/helpers/schema.js';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../../internals/helpers/promise.js';
 import 'ai';
 import '../../../template.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
 
-/**
- * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-
 interface StreamlitAgentInput {
     llm: ChatModel;
     memory: BaseMemory;
@@ -78,6 +73,7 @@ declare class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOu
         input: StreamlitAgentInput;
         isRunning: boolean;
         emitter: Emitter<unknown>;
+        middlewares: MiddlewareType<any>[];
     };
 }