beeai-framework 0.1.23 → 0.1.25

package/dist/agents/requirement/utils/tool.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","result","tool","input","output","StringToolOutput","error","find","name","toolName","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage","JSON","stringify"],"mappings":";;;;;;;;AAiCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,MAAMC,MAAAA,GAA+B;AACnCF,IAAAA,GAAAA;IACAG,IAAAA,EAAM,IAAA;AACNC,IAAAA,KAAAA,EAAOJ,GAAAA,CAAII,KAAAA;IACXC,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFL,IAAAA,MAAAA,CAAOC,IAAAA,GAAOJ,MAAMS,IAAAA,CAAK,CAACL,SAASA,IAAAA,CAAKM,IAAAA,KAAST,GAAAA,CAAIU,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACR,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIQ,SAAAA,CAAU,CAAA,MAAA,EAASX,GAAAA,CAAIU,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAR,IAAAA,MAAAA,CAAOG,MAAAA,GAAS,MAAMH,MAAAA,CAAOC,IAAAA,CAAKS,IAAIV,MAAAA,CAAOE,KAAK,EAAEH,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHY,WAAAA,EAAab,GAAAA;AACb,MAAA,CAACc,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGf,QAAQgB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,SAAAA,EAAW;AAC1BT,MAAAA,MAAAA,CAAOK,KAAAA,GAAQa,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOjB,MAAAA;AACT;AAjCsBJ,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAoCtB,MAAMwB,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,IAAAA,CAAAA;EA5ErC;;;EA6EkBL,IAAAA,GAAO,cAAA;EACPoB,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,SAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,SAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJtC,KAAAA,EACAuC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMf,MAAAA,GAASE,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK6B,0BAA0BE,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,iBAAiBC,IAAAA,CAAKC,SAAAA,CAAU5C,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKa,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,gBAAAA,CAAiB1C,MAAMqB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAInB,iBAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,eAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,sBAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,SAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,IAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,SAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,SAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,iBAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,gBAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAIjB,iBAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}

package/dist/agents/toolCalling/agent.d.cts

@@ -1,10 +1,10 @@
 import { GetRunContext, MiddlewareType } from '../../context.cjs';
-import { c as AgentMeta, a as BaseAgent } from '../../base-XfWeXQ-S.cjs';
+import { c as AgentMeta, a as BaseAgent } from '../../base-BqzN2QwB.cjs';
 import { AnyTool } from '../../tools/base.cjs';
 import { BaseMemory } from '../../memory/base.cjs';
-import { E as Emitter } from '../../emitter-jN55XZZq.cjs';
+import { E as Emitter } from '../../emitter-UkIkNrek.cjs';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.cjs';
-import { m as ChatModel } from '../../chat-BOb9n7AJ.cjs';
+import { m as ChatModel } from '../../chat-DaqV3woF.cjs';
 import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';

package/dist/agents/toolCalling/agent.d.ts

@@ -1,10 +1,10 @@
 import { GetRunContext, MiddlewareType } from '../../context.js';
-import { c as AgentMeta, a as BaseAgent } from '../../base-CyFx7BRL.js';
+import { c as AgentMeta, a as BaseAgent } from '../../base-BfhNzUYf.js';
 import { AnyTool } from '../../tools/base.js';
 import { BaseMemory } from '../../memory/base.js';
-import { E as Emitter } from '../../emitter-36-9MnvA.js';
+import { E as Emitter } from '../../emitter-CL56qab4.js';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.js';
-import { m as ChatModel } from '../../chat-DrqwUgD8.js';
+import { m as ChatModel } from '../../chat-DTxmaKP0.js';
 import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';

package/dist/agents/toolCalling/types.d.cts

@@ -1,6 +1,6 @@
 import { BaseMemory } from '../../memory/base.cjs';
 import { AssistantMessage } from '../../backend/message.cjs';
-import { C as Callback } from '../../emitter-jN55XZZq.cjs';
+import { C as Callback } from '../../emitter-UkIkNrek.cjs';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.cjs';
 import { ZodSchema } from 'zod';
 import '../../errors.cjs';

package/dist/agents/toolCalling/types.d.ts

@@ -1,6 +1,6 @@
 import { BaseMemory } from '../../memory/base.js';
 import { AssistantMessage } from '../../backend/message.js';
-import { C as Callback } from '../../emitter-36-9MnvA.js';
+import { C as Callback } from '../../emitter-CL56qab4.js';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.js';
 import { ZodSchema } from 'zod';
 import '../../errors.js';

package/dist/agents/types.d.cts

@@ -1,8 +1,8 @@
-export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-XfWeXQ-S.cjs';
+export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BqzN2QwB.cjs';
 import '../internals/types.cjs';
 import '../tools/base.cjs';
 import '../context.cjs';
-import '../emitter-jN55XZZq.cjs';
+import '../emitter-UkIkNrek.cjs';
 import '../internals/serializable.cjs';
 import '../internals/helpers/guards.cjs';
 import '../internals/helpers/promise.cjs';

package/dist/agents/types.d.ts

@@ -1,8 +1,8 @@
-export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-CyFx7BRL.js';
+export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BfhNzUYf.js';
 import '../internals/types.js';
 import '../tools/base.js';
 import '../context.js';
-import '../emitter-36-9MnvA.js';
+import '../emitter-CL56qab4.js';
 import '../internals/serializable.js';
 import '../internals/helpers/guards.js';
 import '../internals/helpers/promise.js';

package/dist/backend/backend.d.cts

@@ -1,12 +1,12 @@
 import { Serializable } from '../internals/serializable.cjs';
-import { m as ChatModel, F as FullModelName } from '../chat-BOb9n7AJ.cjs';
+import { m as ChatModel, F as FullModelName } from '../chat-DaqV3woF.cjs';
 import { EmbeddingModel } from './embedding.cjs';
 import { ProviderName } from './constants.cjs';
 import { OptionalExcept } from '../internals/types.cjs';
 import './message.cjs';
 import 'ai';
 import '../context.cjs';
-import '../emitter-jN55XZZq.cjs';
+import '../emitter-UkIkNrek.cjs';
 import '../internals/helpers/promise.cjs';
 import '../errors.cjs';
 import 'promise-based-task';

package/dist/backend/backend.d.ts

@@ -1,12 +1,12 @@
 import { Serializable } from '../internals/serializable.js';
-import { m as ChatModel, F as FullModelName } from '../chat-DrqwUgD8.js';
+import { m as ChatModel, F as FullModelName } from '../chat-DTxmaKP0.js';
 import { EmbeddingModel } from './embedding.js';
 import { ProviderName } from './constants.js';
 import { OptionalExcept } from '../internals/types.js';
 import './message.js';
 import 'ai';
 import '../context.js';
-import '../emitter-36-9MnvA.js';
+import '../emitter-CL56qab4.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';
 import 'promise-based-task';

package/dist/backend/chat.cjs

@@ -20,6 +20,7 @@ var template_cjs = require('../template.cjs');
 var promise_cjs = require('../internals/helpers/promise.cjs');
 var serializer_cjs = require('../serializer/serializer.cjs');
 var logger_cjs = require('../logger/logger.cjs');
+var utils_cjs$2 = require('../adapters/vercel/backend/utils.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -29,6 +30,7 @@ class ChatModel extends serializable_cjs.Serializable {
   }
   cache = new nullCache_cjs.NullCache();
   parameters = {};
+  middlewares = [];
   logger = logger_cjs.Logger.root.child({
     name: this.constructor.name
   });
@@ -40,6 +42,8 @@ class ChatModel extends serializable_cjs.Serializable {
   ];
   toolCallFallbackViaResponseFormat = true;
   modelSupportsToolCalling = true;
+  fixInvalidToolCalls = true;
+  retryOnEmptyResponse = true;
   create(input) {
     input = utils_cjs.shallowCopy(input);
     if (input.stream === void 0) {
@@ -63,56 +67,124 @@ class ChatModel extends serializable_cjs.Serializable {
         this.logger.warn(`The following tool choice value '${input.toolChoice}' is not supported. Ignoring.`);
         input.toolChoice = void 0;
       }
+      const modelInputMessagesBackup = input.messages.slice();
       const cacheEntry = await this.createCacheAccessor(input);
       try {
         await run.emitter.emit("start", {
           input
         });
-        const chunks = [];
-        const generator = cacheEntry.value ?? (input.stream ? this._createStream(input, run) : promise_cjs.toAsyncGenerator(this._create(input, run)));
-        const controller = new AbortController();
-        for await (const value of generator) {
-          chunks.push(value);
-          await run.emitter.emit("newToken", {
-            value,
-            callbacks: {
-              abort: /* @__PURE__ */ __name(() => controller.abort(), "abort")
+        const result = await new retryable_cjs.Retryable({
+          executor: /* @__PURE__ */ __name(async () => {
+            const chunks = [];
+            const generator = cacheEntry.value ?? (input.stream ? this._createStream(input, run) : promise_cjs.toAsyncGenerator(this._create(input, run)));
+            const controller = new AbortController();
+            for await (const value of generator) {
+              chunks.push(value);
+              await run.emitter.emit("newToken", {
+                value,
+                callbacks: {
+                  abort: /* @__PURE__ */ __name(() => controller.abort(), "abort")
+                }
+              });
+              if (controller.signal.aborted) {
+                break;
+              }
             }
-          });
-          if (controller.signal.aborted) {
-            break;
-          }
-        }
-        cacheEntry.resolve(chunks);
-        const result = ChatModelOutput.fromChunks(chunks);
-        for (const toolCall of result.getToolCalls()) {
-          if (remeda.isString(toolCall.input)) {
-            toolCall.input = JSON.parse(toolCall.input);
-          }
-        }
-        if (forceToolCallViaResponseFormat && remeda.isEmpty(result.getToolCalls())) {
-          const lastMsg = result.messages.at(-1);
-          const toolCall = schema_cjs.parseBrokenJson(lastMsg.text, {
-            pair: [
-              "{",
-              "}"
-            ]
-          });
-          if (!toolCall || !toolCall.name || !toolCall.parameters) {
-            throw new errors_cjs.ChatModelError(`Failed to produce a valid tool call. Generate output: ${lastMsg.text}`, [], {
-              isFatal: true,
-              isRetryable: false
-            });
-          }
-          lastMsg.content.length = 0;
-          lastMsg.content.push({
-            type: "tool-call",
-            toolCallId: `call_${remeda.randomString(8).toLowerCase()}`,
-            toolName: toolCall.name,
-            // todo: add types
-            input: toolCall.parameters
-          });
-        }
+            const result2 = ChatModelOutput.fromChunks(chunks);
+            if (result2.isEmpty()) {
+              throw new errors_cjs.EmptyChatModelResponseError();
+            }
+            if (remeda.isEmpty(result2.getToolCalls()) && (forceToolCallViaResponseFormat || input.toolChoice === "required" || input.toolChoice instanceof base_cjs.Tool)) {
+              const lastMsg = result2.messages.at(-1);
+              let toolCall = schema_cjs.parseBrokenJson(lastMsg.text, {
+                pair: [
+                  "{",
+                  "}"
+                ]
+              });
+              if (toolCall && !toolCall.name && !toolCall.parameters && input.toolChoice instanceof base_cjs.Tool) {
+                toolCall = {
+                  name: input.toolChoice.name,
+                  parameters: toolCall
+                };
+              }
+              if (!toolCall || !toolCall.name || !toolCall.parameters) {
+                throw new errors_cjs.ChatModelToolCallError(`Failed to produce a valid tool call. Generate output: ${lastMsg.text}`, [], {
+                  generatedContent: lastMsg.text,
+                  generatedError: "Tool call was not produced.",
+                  response: result2
+                });
+              }
+              lastMsg.content.length = 0;
+              lastMsg.content.push({
+                type: "tool-call",
+                toolCallId: `call_${remeda.randomString(8).toLowerCase()}`,
+                toolName: toolCall.name,
+                // todo: add types
+                input: toolCall.parameters
+              });
+            }
+            for (const toolCall of result2.getToolCalls()) {
+              const tool = input.tools?.find((t) => t.name === toolCall.toolName);
+              if (!tool) {
+                const availableTools = input.tools?.map((t) => t.name).join(",") || "None";
+                throw new errors_cjs.ChatModelToolCallError("Non existing tool call.", [], {
+                  generatedError: `Error: Unknown tool '${toolCall.toolName}'.
+Use on of the available tools: ${availableTools}`,
+                  generatedContent: JSON.stringify({
+                    name: toolCall.toolName,
+                    input: remeda.isString(toolCall.input) ? toolCall.input : JSON.stringify(toolCall.input)
+                  }),
+                  response: result2
+                });
+              }
+              if (!utils_cjs$2.isToolCallValid(toolCall)) {
+                throw new errors_cjs.ChatModelToolCallError("Malformed tool call.", [], {
+                  generatedContent: remeda.isString(toolCall.input) ? toolCall.input : JSON.stringify(toolCall.input),
+                  generatedError: `The tool call for the '${toolCall.toolName}' tool has malformed parameters. It must be a valid JSON.`,
+                  response: result2
+                });
+              }
+              if (remeda.isString(toolCall.input)) {
+                toolCall.input = JSON.parse(toolCall.input);
+              }
+            }
+            cacheEntry.resolve(chunks);
+            return result2;
+          }, "executor"),
+          config: {
+            maxRetries: input.maxRetries ?? 0
+          },
+          onRetry: /* @__PURE__ */ __name(async (_, lastError) => {
+            if (this.fixInvalidToolCalls && lastError instanceof errors_cjs.ChatModelToolCallError) {
+              input.messages = input.messages.slice();
+              if (lastError.data.generatedContent) {
+                input.messages.push(new message_cjs.AssistantMessage(lastError.data.generatedContent, {
+                  tempMessage: true
+                }));
+              }
+              const toolNames = input.tools?.map((t) => t.name).join(", ") || "None";
+              input.messages.push(new message_cjs.UserMessage(`${lastError.data.generatedError}
+
+Available Tools: ${toolNames}`, {
+                tempMessage: true
+              }));
+            } else if (this.retryOnEmptyResponse && lastError instanceof errors_cjs.EmptyChatModelResponseError) {
+              input.messages = input.messages.slice();
+              const lastMessage = input.messages.at(-1);
+              if (lastMessage && lastMessage instanceof message_cjs.AssistantMessage && lastMessage.meta["tempMessage"] && lastMessage.text === "") {
+                input.messages.push(new message_cjs.UserMessage("No output received. Please regenerate your previous response.", {
+                  tempMessage: true
+                }));
+              } else {
+                input.messages.push(new message_cjs.AssistantMessage("", {
+                  tempMessage: true
+                }));
+              }
+            }
+          }, "onRetry")
+        }).get();
+        input.messages = modelInputMessagesBackup;
         await run.emitter.emit("success", {
           value: result
         });
@@ -133,7 +205,7 @@ class ChatModel extends serializable_cjs.Serializable {
       } finally {
         await run.emitter.emit("finish", null);
       }
-    });
+    }).middleware(...this.middlewares);
   }
   createStructure(input) {
     return context_cjs.RunContext.enter(this, {
@@ -237,11 +309,14 @@ Validation Errors: {{errors}}`
     return {
       cache: this.cache,
       emitter: this.emitter,
+      middlewares: utils_cjs.shallowCopy(this.middlewares),
       parameters: utils_cjs.shallowCopy(this.parameters),
       logger: this.logger,
       toolChoiceSupport: this.toolChoiceSupport.slice(),
       toolCallFallbackViaResponseFormat: this.toolCallFallbackViaResponseFormat,
-      modelSupportsToolCalling: this.modelSupportsToolCalling
+      modelSupportsToolCalling: this.modelSupportsToolCalling,
+      retryOnEmptyResponse: this.retryOnEmptyResponse,
+      fixInvalidToolCalls: this.fixInvalidToolCalls
     };
   }
   destroy() {
@@ -301,6 +376,12 @@ class ChatModelOutput extends serializable_cjs.Serializable {
     chunks.forEach((cur) => final.merge(cur));
     return final;
   }
+  isEmpty() {
+    if (this.messages.length === 0) {
+      return true;
+    }
+    return this.getTextContent() === "" && this.getToolCalls().length === 0;
+  }
   merge(other) {
     if (other.messages.length > 0) {
       const clones = other.messages.map(cloneSync);
@@ -335,7 +416,7 @@ class ChatModelOutput extends serializable_cjs.Serializable {
               continue;
             }
             if (!chunk.toolCallId && messagesByToolCallId.size > 0) {
-              const lastToolCallId = Array.from(messagesByToolCallId.keys()).pop();
+              const lastToolCallId = object_cjs.getLast(messagesByToolCallId.keys(), "");
               if (lastToolCallId) {
                 chunk.toolCallId = lastToolCallId;
               }

package/dist/backend/chat.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/backend/chat.ts"],"names":["ChatModel","Serializable","cache","NullCache","parameters","logger","Logger","root","child","name","toolChoiceSupport","toolCallFallbackViaResponseFormat","modelSupportsToolCalling","create","input","shallowCopy","stream","undefined","RunContext","enter","params","signal","abortSignal","run","tools","forceToolCallViaResponseFormat","shouldForceToolCallViaResponseFormat","isEmpty","responseFormat","generateToolUnionSchema","filterToolsByToolChoice","toolChoice","isToolChoiceSupported","warn","cacheEntry","createCacheAccessor","emitter","emit","chunks","generator","value","_createStream","toAsyncGenerator","_create","controller","AbortController","push","callbacks","abort","aborted","resolve","result","ChatModelOutput","fromChunks","toolCall","getToolCalls","isString","JSON","parse","lastMsg","messages","at","parseBrokenJson","text","pair","ChatModelError","isFatal","isRetryable","content","length","type","toolCallId","randomString","toLowerCase","toolName","error","reject","createStructure","_createStructure","config","isFunction","fromName","options","providerId","modelId","parseModel","Target","loadModel","instance","Object","assign","schema","jsonSchema","toJsonSchema","systemTemplate","systemPromptTemplate","PromptTemplate","z","object","string","min","template","SystemMessage","render","stringify","errorTemplate","errors","expected","received","Retryable","executor","response","textResponse","getTextContent","validator","createSchemaValidator","success","context","UserMessage","output","maxRetries","get","createSnapshot","slice","destroy","_","key","ObjectHashKeyFn","Serializer","serialize","map","msg","toPlain","isNew","task","Task","set","delete","Boolean","toolChoiceSupported","choice","Tool","includes","usage","finishReason","dedupe","final","forEach","cur","merge","other","clones","cloneSync","customMerge","totalTokens","takeBigger","promptTokens","completionTokens","messagesById","Map","messagesByToolCallId","msgId","id","AssistantMessage","filteredChunks","chunk","size","lastToolCallId","Array","from","keys","pop","has","values","main","shift","finalToolCalls","lastId","excludedIndexes","index","lastToolCall","Error","reverse","splice","filter","r","flatMap","getTextMessages","join","toString","loadSnapshot","snapshot","serializable","isPromise","target","prototype","load"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AA+HO,MAAeA,kBAAkBC,6BAAAA,CAAAA;EA/HxC;;;AAiISC,EAAAA,KAAAA,GAAwB,IAAIC,uBAAAA,EAAAA;AAC5BC,EAAAA,UAAAA,GAAkC,EAAC;EACvBC,MAAAA,GAASC,iBAAAA,CAAOC,KAAKC,KAAAA,CAAM;AAC5CC,IAAAA,IAAAA,EAAM,KAAK,WAAA,CAAYA;GACzB,CAAA;EAEgBC,iBAAAA,GAAkD;AAChE,IAAA,UAAA;AACA,IAAA,MAAA;AACA,IAAA,QAAA;AACA,IAAA;;EAEKC,iCAAAA,GAAoC,IAAA;EAC3BC,wBAAAA,GAAoC,IAAA;AAKpDC,EAAAA,MAAAA,CAAOC,KAAAA,EAAuB;AAC5BA,IAAAA,KAAAA,GAAQC,sBAAYD,KAAAA,CAAAA;AACpB,IAAA,IAAIA,KAAAA,CAAME,WAAWC,MAAAA,EAAW;AAC9BH,MAAAA,KAAAA,CAAME,MAAAA,GAAS,KAAKZ,UAAAA,CAAWY,MAAAA;AACjC;AAEA,IAAA,OAAOE,sBAAAA,CAAWC,MAChB,IAAA,EACA;MAAEC,MAAAA,EAAQ;AAACN,QAAAA;;AAAiBO,MAAAA,MAAAA,EAAQP,KAAAA,EAAOQ;AAAY,KAAA,EACvD,OAAOC,GAAAA,KAAAA;AACL,MAAA,IAAI,CAAC,KAAKX,wBAAAA,EAA0B;AAClCE,QAAAA,KAAAA,CAAMU,QAAQ,EAAA;AAChB;AAEA,MAAA,MAAMC,8BAAAA,GAAiC,IAAA,CAAKC,oCAAAA,CAAqCZ,KAAAA,CAAAA;AACjF,MAAA,IAAIW,kCAAkCX,KAAAA,CAAMU,KAAAA,IAAS,CAACG,cAAAA,CAAQb,KAAAA,CAAMU,KAAK,CAAA,EAAG;AAC1EV,QAAAA,KAAAA,CAAMc,cAAAA,GAAiB,MAAMC,mCAAAA,CAC3BC,mCAAAA,CAAwBhB,MAAMU,KAAAA,EAAOV,KAAAA,CAAMiB,UAAU,CAAA,CAAA;AAEvDjB,QAAAA,KAAAA,CAAMiB,UAAAA,GAAad,MAAAA;AACrB;AAEA,MAAA,IAAI,CAAC,IAAA,CAAKe,qBAAAA,CAAsBlB,KAAAA,CAAMiB,UAAU,CAAA,EAAG;AACjD,QAAA,IAAA,CAAK1B,MAAAA,CAAO4B,IAAAA,CACV,CAAA,iCAAA,EAAoCnB,KAAAA,CAAMiB,UAAU,CAAA,6BAAA,CAA+B,CAAA;AAErFjB,QAAAA,KAAAA,CAAMiB,UAAAA,GAAad,MAAAA;AACrB;AAEA,MAAA,MAAMiB,UAAAA,GAAa,MAAM,IAAA,CAAKC,mBAAAA,CAAoBrB,KAAAA,CAAAA;AAElD,MAAA,IAAI;AACF,QAAA,MAAMS,GAAAA,CAAIa,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEvB,UAAAA;SAAM,CAAA;AACxC,QAAA,MAAMwB,SAA4B,EAAA;AAElC,QAAA,MAAMC,SAAAA,GACJL,UAAAA,CAAWM,KAAAA,KACV1B,KAAAA,CAAME,SACH,IAAA,CAAKyB,aAAAA,CAAc3B,KAAAA,EAAOS,GAAAA,IAC1BmB,4BAAAA,CAAiB,IAAA,CAAKC,OAAAA,CAAQ7B,KAAAA,EAAOS,GAAAA,CAAAA,CAAAA,CAAAA;AAE3C,QAAA,MAAMqB,UAAAA,GAAa,IAAIC,eAAAA,EAAAA;AACvB,QAAA,WAAA,MAAiBL,SAASD,SAAAA,EAAW;AACnCD,UAAAA,MAAAA,CAAOQ,KAAKN,KAAAA,CAAAA;AACZ,UAAA,MAAMjB,GAAAA,CAAIa,OAAAA,CAAQC,IAAAA,CAAK,UAAA,EAAY;AACjCG,YAAAA,KAAAA;YACAO,SAAAA,EAAW;cAAEC,KAAAA,kBAAO,MAAA,CAAA,MAAMJ,UAAAA,CAAWI,KAAAA,EAAK,EAAtB,OAAA;AAAyB;WAC/C,CAAA;AACA,UAAA,IAAIJ,UAAAA,CAAWvB,OAAO4B,OAAAA,EAAS;AAC7B,YAAA;AACF;AACF;AAEAf,QAAAA,UAAAA,CAAWgB,QAAQZ,MAAAA,CAAAA;AACnB,QAAA,MAAMa,MAAAA,GAASC,eAAAA,CAAgBC,UAAAA,CAAWf,MAAAA,CAAAA;AAC1C,QAAA,KAAA,MAAWgB,QAAAA,IAAYH,MAAAA,CAAOI,YAAAA,EAAY,EAAI;AAC5C,UAAA,IAAIC,eAAAA,CAASF,QAAAA,CAASxC,KAAK,CAAA,EAAG;AAC5BwC,YAAAA,QAAAA,CAASxC,KAAAA,GAAQ2C,IAAAA,CAAKC,KAAAA,CAAMJ,QAAAA,CAASxC,KAAK,CAAA;AAC5C;AACF;AAEA,QAAA,IAAIW,8BAAAA,IAAkCE,cAAAA,CAAQwB,MAAAA,CAAOI,YAAAA,EAAY,CAAA,EAAK;AACpE,UAAA,MAAMI,OAAAA,GAAUR,MAAAA,CAAOS,QAAAA,CAASC,EAAAA,CAAG,CAAA,CAAC,CAAA;AACpC,UAAA,MAAMP,QAAAA,GAAWQ,0BAAAA,CAAgBH,OAAAA,CAAQI,IAAAA,EAAM;YAAEC,IAAAA,EAAM;AAAC,cAAA,GAAA;AAAK,cAAA;;WAAK,CAAA;AAClE,UAAA,IAAI,CAACV,QAAAA,IAAY,CAACA,SAAS7C,IAAAA,IAAQ,CAAC6C,SAASlD,UAAAA,EAAY;AACvD,YAAA,MAAM,IAAI6D,yBAAAA,CACR,CAAA,sDAAA,EAAyDN,QAAQI,IAAI,CAAA,CAAA,EACrE,EAAA,EACA;cACEG,OAAAA,EAAS,IAAA;cACTC,WAAAA,EAAa;aACf,CAAA;AAEJ;AACAR,UAAAA,OAAAA,CAAQS,QAAQC,MAAAA,GAAS,CAAA;AACzBV,UAAAA,OAAAA,CAAQS,QAAQtB,IAAAA,CAAK;YACnBwB,IAAAA,EAAM,WAAA;AACNC,YAAAA,UAAAA,EAAY,CAAA,KAAA,EAAQC,mBAAAA,CAAa,CAAA,CAAA,CAAGC,aAAW,CAAA,CAAA;AAC/CC,YAAAA,QAAAA,EAAUpB,QAAAA,CAAS7C,IAAAA;;AACnBK,YAAAA,KAAAA,EAAOwC,QAAAA,CAASlD;WAClB,CAAA;AACF;AAEA,QAAA,MAAMmB,GAAAA,CAAIa,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;UAAEG,KAAAA,EAAOW;SAAO,CAAA;AAClD,QAAA,OAAOA,MAAAA;AACT,OAAA,CAAA,OAASwB,KAAAA,EAAO;AACd,QAAA,MAAMpD,GAAAA,CAAIa,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEvB,UAAAA,KAAAA;AAAO6D,UAAAA;SAAM,CAAA;AAC/C,QAAA,MAAMzC,UAAAA,CAAW0C,OAAOD,KAAAA,CAAAA;AACxB,QAAA,IAAIA,iBAAiBV,yBAAAA,EAAgB;AACnC,UAAA,MAAMU,KAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAIV,0BAAe,CAAA,wCAAA,CAAA,EAA4C;AAACU,YAAAA;AAAM,WAAA,CAAA;AAC9E;OACF,SAAA;AACE,QAAA,MAAMpD,GAAAA,CAAIa,OAAAA,CAAQC,IAAAA,CAAK,QAAA,EAAU,IAAA,CAAA;AACnC;KACF,CAAA;AAEJ;AAEAwC,EAAAA,eAAAA,CAAmB/D,KAAAA,EAAgC;AACjD,IAAA,OAAOI,sBAAAA,CAAWC,MAChB,IAAA,EACA;MAAEC,MAAAA,EAAQ;AAACN,QAAAA;;AAAiBO,MAAAA,MAAAA,EAAQP,KAAAA,EAAOQ;AAAY,KAAA,EACvD,OAAOC,GAAAA,KAAAA;AACL,MAAA,OAAO,MAAM,IAAA,CAAKuD,gBAAAA,CAAoBhE,KAAAA,EAAOS,GAAAA,CAAAA;KAC/C,CAAA;AAEJ;EAEAwD,MAAAA,CAAO,EAAE7E,KAAAA,EAAOE,UAAAA,EAAU,EAAsB;AAC9C,IAAA,IAAIF,KAAAA,EAAO;AACT,MAAA,IAAA,CAAKA,QAAQ8E,iBAAAA,CAAW9E,KAAAA,IAASA,KAAAA,CAAM,IAAA,CAAKA,KAAK,CAAA,GAAIA,KAAAA;AACvD;AACA,IAAA,IAAIE,UAAAA,EAAY;AACd,MAAA,IAAA,CAAKA,aAAa4E,iBAAAA,CAAW5E,UAAAA,IAAcA,UAAAA,CAAW,IAAA,CAAKA,UAAU,CAAA,GAAIA,UAAAA;AAC3E;AACF;EAEA,aAAa6E,QAAAA,CAASxE,MAAoCyE,OAAAA,EAA+B;AACvF,IAAA,MAAM,EAAEC,UAAAA,EAAYC,OAAAA,EAAO,GAAKC,uBAAW5E,IAAAA,CAAAA;AAC3C,IAAA,MAAM6E,MAAAA,GAAS,MAAMC,qBAAAA,CAAqBJ,UAAAA,EAAY,MAAA,CAAA;AACtD,IAAA,MAAMK,QAAAA,GAAW,IAAIF,MAAAA,CAAOF,OAAAA,IAAWnE,MAAAA,CAAAA;AACvC,IAAA,IAAIiE,OAAAA,EAAS;AACXO,MAAAA,MAAAA,CAAOC,MAAAA,CAAOF,QAAAA,CAASpF,UAAAA,EAAY8E,OAAAA,CAAAA;AACrC;AACA,IAAA,OAAOM,QAAAA;AACT;EAWA,MAAgBV,gBAAAA,CACdhE,OACAS,GAAAA,EACmC;AACnC,IAAA,MAAM,EAAEoE,MAAAA,EAAQ,GAAGT,OAAAA,EAAAA,GAAYpE,KAAAA;AAC/B,IAAA,MAAM8E,UAAAA,GAAaC,wBAAaF,MAAAA,CAAAA;AAEhC,IAAA,MAAMG,cAAAA,GACJhF,KAAAA,CAAMiF,oBAAAA,IACN,IAAIC,2BAAAA,CAAe;AACjBL,MAAAA,MAAAA,EAAQM,MAAEC,MAAAA,CAAO;AACfP,QAAAA,MAAAA,EAAQM,KAAAA,CAAEE,MAAAA,EAAM,CAAGC,GAAAA,CAAI,CAAA;OACzB,CAAA;MACAC,QAAAA,EAAU,CAAA;;;;;;;KAOZ,CAAA;AAEF,IAAA,MAAMzC,QAAAA,GAAsB;MAC1B,IAAI0C,yBAAAA,CAAcR,eAAeS,MAAAA,CAAO;AAAEZ,QAAAA,MAAAA,EAAQlC,IAAAA,CAAK+C,SAAAA,CAAUZ,UAAAA,EAAY,IAAA,EAAM,CAAA;AAAG,OAAA,CAAA,CAAA;SACnF9E,KAAAA,CAAM8C;;AAGX,IAAA,MAAM6C,aAAAA,GAAgB,IAAIT,2BAAAA,CAAe;AACvCL,MAAAA,MAAAA,EAAQM,MAAEC,MAAAA,CAAO;AACfQ,QAAAA,MAAAA,EAAQT,MAAEE,MAAAA,EAAM;AAChBQ,QAAAA,QAAAA,EAAUV,MAAEE,MAAAA,EAAM;AAClBS,QAAAA,QAAAA,EAAUX,MAAEE,MAAAA;OACd,CAAA;MACAE,QAAAA,EAAU,CAAA;;;KAGZ,CAAA;AAEA,IAAA,OAAO,IAAIQ,uBAAAA,CAAoC;AAC7CC,MAAAA,QAAAA,kBAAU,MAAA,CAAA,YAAA;AACR,QAAA,MAAMC,QAAAA,GAAW,MAAM,IAAA,CAAKpE,OAAAA,CAC1B;UACE,GAAGuC,OAAAA;AACHtB,UAAAA,QAAAA;UACAhC,cAAAA,EAAgB;YAAE0C,IAAAA,EAAM;AAAc;AACxC,SAAA,EACA/C,GAAAA,CAAAA;AAGF,QAAA,MAAMyF,YAAAA,GAAeD,SAASE,cAAAA,EAAc;AAC5C,QAAA,MAAMf,MAAAA,GAAYpC,2BAAgBkD,YAAAA,EAAc;UAAEhD,IAAAA,EAAM;AAAC,YAAA,GAAA;AAAK,YAAA;;SAAK,CAAA;AACnE,QAAA,MAAMkD,SAAAA,GAAYC,iCAAsBxB,MAAAA,CAAAA;AAExC,QAAA,MAAMyB,OAAAA,GAAUF,UAAUhB,MAAAA,CAAAA;AAC1B,QAAA,IAAI,CAACkB,OAAAA,EAAS;AACZ,UAAA,MAAMC,OAAAA,GAAU;YACdV,QAAAA,EAAUlD,IAAAA,CAAK+C,UAAUZ,UAAAA,CAAAA;YACzBgB,QAAAA,EAAUI,YAAAA;AACVN,YAAAA,MAAAA,EAAQjD,IAAAA,CAAK+C,SAAAA,CAAUU,SAAAA,CAAUR,MAAAA,IAAU,EAAE;AAC/C,WAAA;AAEA9C,UAAAA,QAAAA,CAASd,KAAK,IAAIwE,uBAAAA,CAAYb,cAAcF,MAAAA,CAAOc,OAAAA,CAAAA,CAAAA,CAAAA;AACnD,UAAA,MAAM,IAAIpD,yBAAAA,CAAe,CAAA,mCAAA,CAAA,EAAuC,EAAA,EAAI;AAClEoD,YAAAA;WACF,CAAA;AACF;AAEA,QAAA,OAAO;AACLnB,UAAAA,MAAAA;UACAqB,MAAAA,EAAQR;AACV,SAAA;OACF,EAhCU,UAAA,CAAA;MAiCVhC,MAAAA,EAAQ;AACN1D,QAAAA,MAAAA,EAAQE,GAAAA,CAAIF,MAAAA;AACZmG,QAAAA,UAAAA,EAAY1G,OAAO0G,UAAAA,IAAc;AACnC;AACF,KAAA,EAAGC,GAAAA,EAAG;AACR;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLxH,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZkC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdhC,UAAAA,EAAYW,qBAAAA,CAAY,KAAKX,UAAU,CAAA;AACvCC,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;MACbK,iBAAAA,EAAmB,IAAA,CAAKA,kBAAkBiH,KAAAA,EAAK;AAC/ChH,MAAAA,iCAAAA,EAAmC,IAAA,CAAKA,iCAAAA;AACxCC,MAAAA,wBAAAA,EAA0B,IAAA,CAAKA;AACjC,KAAA;AACF;EAEAgH,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKxF,QAAQwF,OAAAA,EAAO;AACtB;EAEA,MAAgBzF,mBAAAA,CAAoB,EAClCb,WAAAA,EAAauG,CAAAA,EACbjE,QAAAA,EACApC,QAAQ,EAAA,EACR,GAAGV,KAAAA,EAAAA,EACc;AACjB,IAAA,MAAMgH,MAAMC,kCAAAA,CAAgB;MAC1B,GAAGjH,KAAAA;MACH8C,QAAAA,EAAU,MAAMoE,yBAAAA,CAAWC,SAAAA,CAAUrE,QAAAA,CAASsE,GAAAA,CAAI,CAACC,GAAAA,KAAQA,GAAAA,CAAIC,OAAAA,EAAO,CAAA,CAAA;MACtE5G,KAAAA,EAAO,MAAMwG,yBAAAA,CAAWC,SAAAA,CAAUzG,KAAAA;KACpC,CAAA;AACA,IAAA,MAAMgB,KAAAA,GAAQ,MAAM,IAAA,CAAKtC,KAAAA,CAAMuH,IAAIK,GAAAA,CAAAA;AACnC,IAAA,MAAMO,QAAQ7F,KAAAA,KAAUvB,MAAAA;AAExB,IAAA,IAAIqH,IAAAA,GAAuC,IAAA;AAC3C,IAAA,IAAID,KAAAA,EAAO;AACTC,MAAAA,IAAAA,GAAO,IAAIC,qBAAAA,EAAAA;AACX,MAAA,MAAM,IAAA,CAAKrI,KAAAA,CAAMsI,GAAAA,CAAIV,GAAAA,EAAKQ,IAAAA,CAAAA;AAC5B;AAEA,IAAA,OAAO;AACLR,MAAAA,GAAAA;AACAtF,MAAAA,KAAAA;AACAU,MAAAA,OAAAA,0BAAsCV,MAAAA,KAAAA;AACpC8F,QAAAA,IAAAA,EAAMpF,UAAUV,MAAAA,CAAAA;OAClB,EAFS,SAAA,CAAA;AAGToC,MAAAA,MAAAA,gCAAeD,KAAAA,KAAAA;AACb2D,QAAAA,IAAAA,EAAM1D,SAASD,KAAAA,CAAAA;AACf,QAAA,IAAI0D,KAAAA,EAAO;AACT,UAAA,MAAM,IAAA,CAAKnI,KAAAA,CAAMuI,MAAAA,CAAOX,GAAAA,CAAAA;AAC1B;OACF,EALQ,QAAA;AAMV,KAAA;AACF;AAEUpG,EAAAA,oCAAAA,CAAqC,EAC7CF,KAAAA,GAAQ,EAAA,EACRO,UAAAA,EACAH,gBAAc,EACG;AACjB,IAAA,IACED,cAAAA,CAAQH,KAAAA,CAAAA,IACR,CAACO,cACDA,UAAAA,KAAe,MAAA,IACfA,UAAAA,KAAe,MAAA,IACf,CAAC,IAAA,CAAKpB,iCAAAA,IACN+H,OAAAA,CAAQ9G,cAAAA,CAAAA,EACR;AACA,MAAA,OAAO,KAAA;AACT;AAEA,IAAA,MAAM+G,mBAAAA,GAAsB,IAAA,CAAK3G,qBAAAA,CAAsBD,UAAAA,CAAAA;AACvD,IAAA,OAAO,CAAC,IAAA,CAAKnB,wBAAAA,IAA4B,CAAC+H,mBAAAA;AAC5C;AAEU3G,EAAAA,qBAAAA,CAAsB4G,MAAAA,EAAuC;AACrE,IAAA,OACE,CAACA,MAAAA,KACAA,MAAAA,YAAkBC,aAAAA,GACf,IAAA,CAAKnI,iBAAAA,CAAkBoI,QAAAA,CAAS,QAAA,CAAA,GAChC,IAAA,CAAKpI,iBAAAA,CAAkBoI,QAAAA,CAASF,MAAAA,CAAAA,CAAAA;AAExC;AACF;AAEO,MAAMxF,wBAAwBnD,6BAAAA,CAAAA;EA7brC;;;;;;EA8bE,WAAA,CACkB2D,QAAAA,EACTmF,OACAC,YAAAA,EACP;AACA,IAAA,KAAA,EAAK,EAAA,KAJWpF,QAAAA,GAAAA,QAAAA,EAAAA,KACTmF,KAAAA,GAAAA,KAAAA,EAAAA,KACAC,YAAAA,GAAAA,YAAAA;AAGP,IAAA,IAAA,CAAKC,MAAAA,EAAM;AACb;AAEA,EAAA,OAAO5F,WAAWf,MAAAA,EAA2B;AAC3C,IAAA,MAAM4G,KAAAA,GAAQ,IAAI9F,eAAAA,CAAgB,EAAE,CAAA;AACpCd,IAAAA,MAAAA,CAAO6G,QAAQ,CAACC,GAAAA,KAAQF,KAAAA,CAAMG,KAAAA,CAAMD,GAAAA,CAAAA,CAAAA;AACpC,IAAA,OAAOF,KAAAA;AACT;AAEAG,EAAAA,KAAAA,CAAMC,KAAAA,EAAwB;AAC5B,IAAA,IAAIA,KAAAA,CAAM1F,QAAAA,CAASS,MAAAA,GAAS,CAAA,EAAG;AAC7B,MAAA,MAAMkF,MAAAA,GAASD,KAAAA,CAAM1F,QAAAA,CAASsE,GAAAA,CAAIsB,SAAAA,CAAAA;AAClC,MAAA,IAAA,CAAK5F,QAAAA,CAASd,IAAAA,CAAI,GAAIyG,MAAAA,CAAAA;AACtB,MAAA,IAAA,CAAKN,MAAAA,EAAM;AACb;AAEA,IAAA,IAAA,CAAKD,eAAeM,KAAAA,CAAMN,YAAAA;AAC1B,IAAA,IAAI,IAAA,CAAKD,KAAAA,IAASO,KAAAA,CAAMP,KAAAA,EAAO;AAC7B,MAAA,IAAA,CAAKA,QAAQU,sBAAAA,CAAY;QAAC,IAAA,CAAKV,KAAAA;QAAOO,KAAAA,CAAMP;AAAQ,OAAA,EAAA;QAClDW,WAAAA,EAAaC,qBAAAA;QACbC,YAAAA,EAAcD,qBAAAA;QACdE,gBAAAA,EAAkBF;OACpB,CAAA;AACF,KAAA,MAAA,IAAWL,MAAMP,KAAAA,EAAO;AACtB,MAAA,IAAA,CAAKA,KAAAA,GAAQhI,qBAAAA,CAAYuI,KAAAA,CAAMP,KAAK,CAAA;AACtC;AACF;EAEAE,MAAAA,GAAe;AAEb,IAAA,IAAI,IAAA,CAAKrF,QAAAA,CAASS,MAAAA,GAAS,CAAA,EAAG;AAC5B,MAAA,MAAMyF,YAAAA,uBAAmBC,GAAAA,EAAAA;AACzB,MAAA,MAAMC,oBAAAA,uBAA2BD,GAAAA,EAAAA;AAEjC,MAAA,KAAA,MAAW5B,GAAAA,IAAO,KAAKvE,QAAAA,EAAU;AAC/B,QAAA,MAAMqG,KAAAA,GAAQ9B,IAAI+B,EAAAA,IAAM,EAAA;AAExB,QAAA,IAAI/B,eAAegC,4BAAAA,IAAoBhC,GAAAA,CAAI5E,YAAAA,EAAY,CAAGc,SAAS,CAAA,EAAG;AACpE,UAAA,MAAM+F,iBAA8C,EAAA;AACpD,UAAA,KAAA,MAAWC,KAAAA,IAASlC,IAAI/D,OAAAA,EAAS;AAC/B,YAAA,IAAIiG,KAAAA,CAAM/F,SAAS,WAAA,EAAa;AAC9B8F,cAAAA,cAAAA,CAAetH,KAAKuH,KAAAA,CAAAA;AACpB,cAAA;AACF;AAGA,YAAA,IAAI,CAACA,KAAAA,CAAM9F,UAAAA,IAAcyF,oBAAAA,CAAqBM,OAAO,CAAA,EAAG;AACtD,cAAA,MAAMC,iBAAiBC,KAAAA,CAAMC,IAAAA,CAAKT,qBAAqBU,IAAAA,EAAI,EAAIC,GAAAA,EAAG;AAClE,cAAA,IAAIJ,cAAAA,EAAgB;AAClBF,gBAAAA,KAAAA,CAAM9F,UAAAA,GAAagG,cAAAA;AACrB;AACF;AAEA,YAAA,IAAIF,MAAM9F,UAAAA,IAAcyF,oBAAAA,CAAqBY,GAAAA,CAAIP,KAAAA,CAAM9F,UAAU,CAAA,EAAG;AAClEyF,cAAAA,oBAAAA,CAAqBvC,IAAI4C,KAAAA,CAAM9F,UAAU,CAAA,CAAGH,OAAAA,CAAQtB,KAAKuH,KAAAA,CAAAA;AAC3D,aAAA,MAAA,IAAWA,MAAM9F,UAAAA,EAAY;AAC3ByF,cAAAA,oBAAAA,CAAqBxB,GAAAA,CAAI6B,KAAAA,CAAM9F,UAAAA,EAAY4D,GAAAA,CAAAA;AAC3CiC,cAAAA,cAAAA,CAAetH,KAAKuH,KAAAA,CAAAA;AACtB;AACF;AAEAlC,UAAAA,GAAAA,CAAI/D,QAAQC,MAAAA,GAAS,CAAA;AACrB8D,UAAAA,GAAAA,CAAI/D,OAAAA,CAAQtB,IAAAA,CAAI,GAAIsH,cAAAA,CAAAA;AAEpB,UAAA,IAAIA,cAAAA,CAAe/F,WAAW,CAAA,EAAG;AAC/B,YAAA;AACF;AACF;AAEA,QAAA,IAAI,CAACyF,YAAAA,CAAac,GAAAA,CAAIX,KAAAA,CAAAA,EAAQ;AAC5BH,UAAAA,YAAAA,CAAatB,IAAIyB,KAAAA,EAAO;AAAC9B,YAAAA;AAAI,WAAA,CAAA;SAC/B,MAAO;AACL2B,UAAAA,YAAAA,CAAarC,GAAAA,CAAIwC,KAAAA,CAAAA,CAAQnH,IAAAA,CAAKqF,GAAAA,CAAAA;AAChC;AACF;AAEA,MAAA,IAAA,CAAKvE,SAASS,MAAAA,GAAS,CAAA;AAEvB,MAAA,KAAA,MAAWT,QAAAA,IAAYkG,YAAAA,CAAae,MAAAA,EAAM,EAAI;AAC5C,QAAA,MAAMC,IAAAA,GAAOlH,SAASmH,KAAAA,EAAK;AAC3B,QAAA,KAAA,MAAWzB,SAAS1F,QAAAA,EAAU;AAC5BkH,UAAAA,IAAAA,CAAKzB,MAAMC,KAAAA,CAAAA;AACb;AACA,QAAA,IAAA,CAAK1F,QAAAA,CAASd,KAAKgI,IAAAA,CAAAA;AACrB;AACF;AAGA,IAAA,KAAA,MAAW3C,GAAAA,IAAO,KAAKvE,QAAAA,EAAU;AAC/B,MAAA,IAAI,EAAEuE,eAAegC,4BAAAA,CAAAA,EAAmB;AACtC,QAAA;AACF;AAEA,MAAA,MAAMa,iBAA+C,EAAC;AACtD,MAAA,IAAIC,MAAAA,GAAS,EAAA;AAEb,MAAA,MAAMC,kBAA4B,EAAA;AAClC/C,MAAAA,GAAAA,CAAI/D,OAAAA,CAAQ+E,OAAAA,CAAQ,CAACkB,KAAAA,EAAOc,KAAAA,KAAAA;AAC1B,QAAA,IAAId,KAAAA,CAAM/F,SAAS,WAAA,EAAa;AAC9B,UAAA;AACF;AACA,QAAA,MAAM4F,EAAAA,GAAKG,MAAM9F,UAAAA,IAAc0G,MAAAA;AAC/B,QAAA,IAAI,EAAEf,MAAMc,cAAAA,CAAAA,EAAiB;AAC3BA,UAAAA,cAAAA,CAAed,EAAAA,CAAAA,GAAMnJ,qBAAAA,CAAYsJ,KAAAA,CAAAA;AACjClC,UAAAA,GAAAA,CAAI/D,OAAAA,CAAQ+G,KAAAA,CAAAA,GAASH,cAAAA,CAAed,EAAAA,CAAAA;SACtC,MAAO;AACLgB,UAAAA,eAAAA,CAAgBpI,KAAKqI,KAAAA,CAAAA;AACrB,UAAA,MAAMC,YAAAA,GAAeJ,eAAed,EAAAA,CAAAA;AACpC,UAAA,IAAI1G,gBAAS4H,YAAAA,CAAatK,KAAK,KAAK0C,eAAAA,CAAS6G,KAAAA,CAAMvJ,KAAK,CAAA,EAAG;AACzDsK,YAAAA,YAAAA,CAAatK,SAASuJ,KAAAA,CAAMvJ,KAAAA;WAC9B,MAAO;AACL,YAAA,MAAM,IAAIuK,MAAM,0BAAA,CAAA;AAClB;AACA,UAAA,IAAI,CAACD,aAAa1G,QAAAA,EAAU;AAC1B0G,YAAAA,YAAAA,CAAa1G,WAAW2F,KAAAA,CAAM3F,QAAAA;AAChC;AACF;AACAuG,QAAAA,MAAAA,GAASf,EAAAA;OACX,CAAA;AACAgB,MAAAA,eAAAA,CAAgBI,OAAAA,EAAO,CAAGnC,OAAAA,CAAQ,CAACgC,KAAAA,KAAAA;AACjChD,QAAAA,GAAAA,CAAI/D,OAAAA,CAAQmH,MAAAA,CAAOJ,KAAAA,EAAO,CAAA,CAAA;OAC5B,CAAA;AACF;AACF;EAEA5H,YAAAA,GAAe;AACb,IAAA,OAAO,KAAKK,QAAAA,CACT4H,MAAAA,CAAO,CAACC,CAAAA,KAAMA,aAAatB,4BAAAA,CAAAA,CAC3BuB,OAAAA,CAAQ,CAACD,MAAMA,CAAAA,CAAElI,YAAAA,EAAY,CAAA,CAC7BiI,OAAO9C,OAAAA,CAAAA;AACZ;EAEAiD,eAAAA,GAAsC;AACpC,IAAA,OAAO,IAAA,CAAK/H,QAAAA,CAAS4H,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,YAAatB,4BAAAA,CAAAA,CAAkBqB,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAE1H,IAAI,CAAA;AACxF;EAEAkD,cAAAA,GAAyB;AACvB,IAAA,OAAO,KAAKrD,QAAAA,CACT4H,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,YAAatB,4BAAAA,CAAAA,CAC3BuB,OAAAA,CAAQ,CAACD,CAAAA,KAAMA,EAAE1H,IAAI,CAAA,CACrByH,OAAO9C,OAAAA,CAAAA,CACPkD,KAAK,EAAA,CAAA;AACV;EAEAC,QAAAA,GAAW;AACT,IAAA,OAAO,KAAK5E,cAAAA,EAAc;AAC5B;EAEAS,cAAAA,GAAiB;AACf,IAAA,OAAO;MACL9D,QAAAA,EAAU7C,qBAAAA,CAAY,KAAK6C,QAAQ,CAAA;MACnCmF,KAAAA,EAAOhI,qBAAAA,CAAY,KAAKgI,KAAK,CAAA;AAC7BC,MAAAA,YAAAA,EAAc,IAAA,CAAKA;AACrB,KAAA;AACF;AAEA8C,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DtG,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMqG,QAAAA,CAAAA;AACtB;AACF;AAEA,SAASvC,UAAkCwC,YAAAA,EAAe;AACxD,EAAA,MAAMD,QAAAA,GAAWC,aAAatE,cAAAA,EAAc;AAC5C,EAAA,IAAIuE,gBAAAA,CAAUF,QAAAA,CAAAA,EAAW;AACvB,IAAA,MAAM,IAAIV,MAAM,CAAA,8BAAA,CAAgC,CAAA;AAClD;AAEA,EAAA,MAAMa,MAAAA,GAASzG,MAAAA,CAAO5E,MAAAA,CAAOmL,YAAAA,CAAa,YAAYG,SAAS,CAAA;AAC/D,EAAA,MAAMC,IAAAA,GAAOF,MAAAA,CAAOJ,YAAAA,CAAaC,QAAAA,CAAAA;AACjC,EAAA,IAAIE,gBAAAA,CAAUG,IAAAA,CAAAA,EAAO;AACnB,IAAA,MAAM,IAAIf,MAAM,CAAA,4BAAA,CAA8B,CAAA;AAChD;AAEA,EAAA,OAAOa,MAAAA;AACT;AAbS1C,MAAAA,CAAAA,SAAAA,EAAAA,WAAAA,CAAAA","file":"chat.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { customMerge } from \"@/internals/helpers/object.js\";\nimport { takeBigger } from \"@/internals/helpers/number.js\";\nimport { Callback } from \"@/emitter/types.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { isEmpty, isFunction, isPromise, isString, randomString } from \"remeda\";\nimport { ObjectHashKeyFn } from \"@/cache/decoratorCache.js\";\nimport { Task } from \"promise-based-task\";\nimport { NullCache } from \"@/cache/nullCache.js\";\nimport { BaseCache } from \"@/cache/base.js\";\nimport {\n  filterToolsByToolChoice,\n  FullModelName,\n  generateToolUnionSchema,\n  loadModel,\n  parseModel,\n} from \"@/backend/utils.js\";\nimport { ProviderName } from \"@/backend/constants.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { AssistantMessage, Message, SystemMessage, UserMessage } from \"@/backend/message.js\";\n\nimport { ChatModelError } from \"@/backend/errors.js\";\nimport { z, ZodSchema, ZodType } from \"zod\";\nimport {\n  createSchemaValidator,\n  parseBrokenJson,\n  toJsonSchema,\n} from \"@/internals/helpers/schema.js\";\nimport { Retryable } from \"@/internals/helpers/retryable.js\";\nimport { SchemaObject, ValidateFunction } from \"ajv\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { toAsyncGenerator } from \"@/internals/helpers/promise.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { ToolCallPart } from \"ai\";\n\nexport interface ChatModelParameters {\n  maxTokens?: number;\n  topP?: number;\n  frequencyPenalty?: number;\n  temperature?: number;\n  topK?: number;\n  n?: number;\n  presencePenalty?: number;\n  seed?: number;\n  stopSequences?: string[];\n  stream?: boolean;\n}\n\ninterface ResponseObjectJson {\n  type: \"object-json\";\n  schema?: SchemaObject;\n  name?: string;\n  description?: string;\n}\n\nexport interface ChatModelObjectInput<T> extends ChatModelParameters {\n  schema: z.ZodSchema<T, any, any> | ResponseObjectJson;\n  systemPromptTemplate?: PromptTemplate<ZodType<{ schema: string }>>;\n  messages: Message[];\n  abortSignal?: AbortSignal;\n  maxRetries?: number;\n}\n\nexport interface ChatModelObjectOutput<T> {\n  object: T;\n  output: ChatModelOutput;\n}\n\nexport type ChatModelToolChoice = \"auto\" | \"none\" | \"required\" | AnyTool;\n\nexport interface ChatModelInput extends ChatModelParameters {\n  tools?: AnyTool[];\n  abortSignal?: AbortSignal;\n  stopSequences?: string[];\n  responseFormat?: ZodSchema | ResponseObjectJson;\n  toolChoice?: ChatModelToolChoice;\n  messages: Message[];\n  streamPartialToolCalls?: boolean;\n}\n\nexport type ChatModelFinishReason =\n  | \"stop\"\n  | \"length\"\n  | \"content-filter\"\n  | \"tool-calls\"\n  | \"error\"\n  | \"other\"\n  | \"unknown\";\n\nexport interface ChatModelUsage {\n  promptTokens: number;\n  completionTokens: number;\n  totalTokens: number;\n  reasoningTokens?: number;\n  cachedPromptTokens?: number;\n}\n\nexport interface ChatModelEvents {\n  newToken?: Callback<{ value: ChatModelOutput; callbacks: { abort: () => void } }>;\n  success?: Callback<{ value: ChatModelOutput }>;\n  start?: Callback<{ input: ChatModelInput }>;\n  error?: Callback<{ input: ChatModelInput; error: FrameworkError }>;\n  finish?: Callback<null>;\n}\n\nexport type ChatModelEmitter<A = Record<never, never>> = Emitter<\n  ChatModelEvents & Omit<A, keyof ChatModelEvents>\n>;\n\nexport type ChatModelCache = BaseCache<Task<ChatModelOutput[]>>;\nexport type ConfigFn<T> = (value: T) => T;\nexport interface ChatConfig {\n  cache?: ChatModelCache | ConfigFn<ChatModelCache>;\n  parameters?: ChatModelParameters | ConfigFn<ChatModelParameters>;\n}\n\nexport type ChatModelToolChoiceSupport = \"required\" | \"none\" | \"single\" | \"auto\";\n\nexport abstract class ChatModel extends Serializable {\n  public abstract readonly emitter: Emitter<ChatModelEvents>;\n  public cache: ChatModelCache = new NullCache();\n  public parameters: ChatModelParameters = {};\n  protected readonly logger = Logger.root.child({\n    name: this.constructor.name,\n  });\n\n  public readonly toolChoiceSupport: ChatModelToolChoiceSupport[] = [\n    \"required\",\n    \"none\",\n    \"single\",\n    \"auto\",\n  ];\n  public toolCallFallbackViaResponseFormat = true;\n  public readonly modelSupportsToolCalling: boolean = true;\n\n  abstract get modelId(): string;\n  abstract get providerId(): string;\n\n  create(input: ChatModelInput) {\n    input = shallowCopy(input);\n    if (input.stream === undefined) {\n      input.stream = this.parameters.stream;\n    }\n\n    return RunContext.enter(\n      this,\n      { params: [input] as const, signal: input?.abortSignal },\n      async (run) => {\n        if (!this.modelSupportsToolCalling) {\n          input.tools = [];\n        }\n\n        const forceToolCallViaResponseFormat = this.shouldForceToolCallViaResponseFormat(input);\n        if (forceToolCallViaResponseFormat && input.tools && !isEmpty(input.tools)) {\n          input.responseFormat = await generateToolUnionSchema(\n            filterToolsByToolChoice(input.tools, input.toolChoice),\n          );\n          input.toolChoice = undefined;\n        }\n\n        if (!this.isToolChoiceSupported(input.toolChoice)) {\n          this.logger.warn(\n            `The following tool choice value '${input.toolChoice}' is not supported. Ignoring.`,\n          );\n          input.toolChoice = undefined;\n        }\n\n        const cacheEntry = await this.createCacheAccessor(input);\n\n        try {\n          await run.emitter.emit(\"start\", { input });\n          const chunks: ChatModelOutput[] = [];\n\n          const generator =\n            cacheEntry.value ??\n            (input.stream\n              ? this._createStream(input, run)\n              : toAsyncGenerator(this._create(input, run)));\n\n          const controller = new AbortController();\n          for await (const value of generator) {\n            chunks.push(value);\n            await run.emitter.emit(\"newToken\", {\n              value,\n              callbacks: { abort: () => controller.abort() },\n            });\n            if (controller.signal.aborted) {\n              break;\n            }\n          }\n\n          cacheEntry.resolve(chunks);\n          const result = ChatModelOutput.fromChunks(chunks);\n          for (const toolCall of result.getToolCalls()) {\n            if (isString(toolCall.input)) {\n              toolCall.input = JSON.parse(toolCall.input);\n            }\n          }\n\n          if (forceToolCallViaResponseFormat && isEmpty(result.getToolCalls())) {\n            const lastMsg = result.messages.at(-1)!;\n            const toolCall = parseBrokenJson(lastMsg.text, { pair: [\"{\", \"}\"] });\n            if (!toolCall || !toolCall.name || !toolCall.parameters) {\n              throw new ChatModelError(\n                `Failed to produce a valid tool call. Generate output: ${lastMsg.text}`,\n                [],\n                {\n                  isFatal: true,\n                  isRetryable: false,\n                },\n              );\n            }\n            lastMsg.content.length = 0;\n            lastMsg.content.push({\n              type: \"tool-call\",\n              toolCallId: `call_${randomString(8).toLowerCase()}`,\n              toolName: toolCall.name, // todo: add types\n              input: toolCall.parameters,\n            });\n          }\n\n          await run.emitter.emit(\"success\", { value: result });\n          return result;\n        } catch (error) {\n          await run.emitter.emit(\"error\", { input, error });\n          await cacheEntry.reject(error);\n          if (error instanceof ChatModelError) {\n            throw error;\n          } else {\n            throw new ChatModelError(`The Chat Model has encountered an error.`, [error]);\n          }\n        } finally {\n          await run.emitter.emit(\"finish\", null);\n        }\n      },\n    );\n  }\n\n  createStructure<T>(input: ChatModelObjectInput<T>) {\n    return RunContext.enter(\n      this,\n      { params: [input] as const, signal: input?.abortSignal },\n      async (run) => {\n        return await this._createStructure<T>(input, run);\n      },\n    );\n  }\n\n  config({ cache, parameters }: ChatConfig): void {\n    if (cache) {\n      this.cache = isFunction(cache) ? cache(this.cache) : cache;\n    }\n    if (parameters) {\n      this.parameters = isFunction(parameters) ? parameters(this.parameters) : parameters;\n    }\n  }\n\n  static async fromName(name: FullModelName | ProviderName, options?: ChatModelParameters) {\n    const { providerId, modelId } = parseModel(name);\n    const Target = await loadModel<ChatModel>(providerId, \"chat\");\n    const instance = new Target(modelId || undefined);\n    if (options) {\n      Object.assign(instance.parameters, options);\n    }\n    return instance;\n  }\n\n  protected abstract _create(\n    input: ChatModelInput,\n    run: GetRunContext<typeof this>,\n  ): Promise<ChatModelOutput>;\n  protected abstract _createStream(\n    input: ChatModelInput,\n    run: GetRunContext<typeof this>,\n  ): AsyncGenerator<ChatModelOutput, void>;\n\n  protected async _createStructure<T>(\n    input: ChatModelObjectInput<T>,\n    run: GetRunContext<typeof this>,\n  ): Promise<ChatModelObjectOutput<T>> {\n    const { schema, ...options } = input;\n    const jsonSchema = toJsonSchema(schema);\n\n    const systemTemplate =\n      input.systemPromptTemplate ??\n      new PromptTemplate({\n        schema: z.object({\n          schema: z.string().min(1),\n        }),\n        template: `You are a helpful assistant that generates only valid JSON adhering to the following JSON Schema.\n\n\\`\\`\\`\n{{schema}}\n\\`\\`\\`\n\nIMPORTANT: You MUST answer with a JSON object that matches the JSON schema above.`,\n      });\n\n    const messages: Message[] = [\n      new SystemMessage(systemTemplate.render({ schema: JSON.stringify(jsonSchema, null, 2) })),\n      ...input.messages,\n    ];\n\n    const errorTemplate = new PromptTemplate({\n      schema: z.object({\n        errors: z.string(),\n        expected: z.string(),\n        received: z.string(),\n      }),\n      template: `Generated object does not match the expected JSON schema!\n\nValidation Errors: {{errors}}`,\n    });\n\n    return new Retryable<ChatModelObjectOutput<T>>({\n      executor: async () => {\n        const response = await this._create(\n          {\n            ...options,\n            messages,\n            responseFormat: { type: \"object-json\" },\n          },\n          run,\n        );\n\n        const textResponse = response.getTextContent();\n        const object: T = parseBrokenJson(textResponse, { pair: [\"{\", \"}\"] });\n        const validator = createSchemaValidator(schema) as ValidateFunction<T>;\n\n        const success = validator(object);\n        if (!success) {\n          const context = {\n            expected: JSON.stringify(jsonSchema),\n            received: textResponse,\n            errors: JSON.stringify(validator.errors ?? []),\n          };\n\n          messages.push(new UserMessage(errorTemplate.render(context)));\n          throw new ChatModelError(`LLM did not produce a valid output.`, [], {\n            context,\n          });\n        }\n\n        return {\n          object,\n          output: response,\n        };\n      },\n      config: {\n        signal: run.signal,\n        maxRetries: input?.maxRetries || 1,\n      },\n    }).get();\n  }\n\n  createSnapshot() {\n    return {\n      cache: this.cache,\n      emitter: this.emitter,\n      parameters: shallowCopy(this.parameters),\n      logger: this.logger,\n      toolChoiceSupport: this.toolChoiceSupport.slice(),\n      toolCallFallbackViaResponseFormat: this.toolCallFallbackViaResponseFormat,\n      modelSupportsToolCalling: this.modelSupportsToolCalling,\n    };\n  }\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  protected async createCacheAccessor({\n    abortSignal: _,\n    messages,\n    tools = [],\n    ...input\n  }: ChatModelInput) {\n    const key = ObjectHashKeyFn({\n      ...input,\n      messages: await Serializer.serialize(messages.map((msg) => msg.toPlain())),\n      tools: await Serializer.serialize(tools),\n    });\n    const value = await this.cache.get(key);\n    const isNew = value === undefined;\n\n    let task: Task<ChatModelOutput[]> | null = null;\n    if (isNew) {\n      task = new Task();\n      await this.cache.set(key, task);\n    }\n\n    return {\n      key,\n      value,\n      resolve: <T2 extends ChatModelOutput>(value: T2[]) => {\n        task?.resolve?.(value);\n      },\n      reject: async (error: Error) => {\n        task?.reject?.(error);\n        if (isNew) {\n          await this.cache.delete(key);\n        }\n      },\n    };\n  }\n\n  protected shouldForceToolCallViaResponseFormat({\n    tools = [],\n    toolChoice,\n    responseFormat,\n  }: ChatModelInput) {\n    if (\n      isEmpty(tools) ||\n      !toolChoice ||\n      toolChoice === \"none\" ||\n      toolChoice === \"auto\" ||\n      !this.toolCallFallbackViaResponseFormat ||\n      Boolean(responseFormat)\n    ) {\n      return false;\n    }\n\n    const toolChoiceSupported = this.isToolChoiceSupported(toolChoice);\n    return !this.modelSupportsToolCalling || !toolChoiceSupported;\n  }\n\n  protected isToolChoiceSupported(choice?: ChatModelToolChoice): boolean {\n    return (\n      !choice ||\n      (choice instanceof Tool\n        ? this.toolChoiceSupport.includes(\"single\")\n        : this.toolChoiceSupport.includes(choice))\n    );\n  }\n}\n\nexport class ChatModelOutput extends Serializable {\n  constructor(\n    public readonly messages: Message[],\n    public usage?: ChatModelUsage,\n    public finishReason?: ChatModelFinishReason,\n  ) {\n    super();\n    this.dedupe();\n  }\n\n  static fromChunks(chunks: ChatModelOutput[]) {\n    const final = new ChatModelOutput([]);\n    chunks.forEach((cur) => final.merge(cur));\n    return final;\n  }\n\n  merge(other: ChatModelOutput) {\n    if (other.messages.length > 0) {\n      const clones = other.messages.map(cloneSync);\n      this.messages.push(...clones);\n      this.dedupe();\n    }\n\n    this.finishReason = other.finishReason;\n    if (this.usage && other.usage) {\n      this.usage = customMerge([this.usage, other.usage], {\n        totalTokens: takeBigger,\n        promptTokens: takeBigger,\n        completionTokens: takeBigger,\n      });\n    } else if (other.usage) {\n      this.usage = shallowCopy(other.usage);\n    }\n  }\n\n  dedupe(): void {\n    // Dedupe messages\n    if (this.messages.length > 1) {\n      const messagesById = new Map<string, Message[]>();\n      const messagesByToolCallId = new Map<string, AssistantMessage>();\n\n      for (const msg of this.messages) {\n        const msgId = msg.id || \"\";\n\n        if (msg instanceof AssistantMessage && msg.getToolCalls().length > 0) {\n          const filteredChunks: AssistantMessage[\"content\"] = [];\n          for (const chunk of msg.content) {\n            if (chunk.type !== \"tool-call\") {\n              filteredChunks.push(chunk);\n              continue;\n            }\n\n            // Assume tool calls with no id refer to the most recent tool call\n            if (!chunk.toolCallId && messagesByToolCallId.size > 0) {\n              const lastToolCallId = Array.from(messagesByToolCallId.keys()).pop();\n              if (lastToolCallId) {\n                chunk.toolCallId = lastToolCallId;\n              }\n            }\n\n            if (chunk.toolCallId && messagesByToolCallId.has(chunk.toolCallId)) {\n              messagesByToolCallId.get(chunk.toolCallId)!.content.push(chunk);\n            } else if (chunk.toolCallId) {\n              messagesByToolCallId.set(chunk.toolCallId, msg);\n              filteredChunks.push(chunk);\n            }\n          }\n\n          msg.content.length = 0;\n          msg.content.push(...filteredChunks);\n\n          if (filteredChunks.length === 0) {\n            continue; // nothing to process\n          }\n        }\n\n        if (!messagesById.has(msgId)) {\n          messagesById.set(msgId, [msg]);\n        } else {\n          messagesById.get(msgId)!.push(msg);\n        }\n      }\n\n      this.messages.length = 0;\n\n      for (const messages of messagesById.values()) {\n        const main = messages.shift()!;\n        for (const other of messages) {\n          main.merge(other);\n        }\n        this.messages.push(main);\n      }\n    }\n\n    // Dedupe tool calls\n    for (const msg of this.messages) {\n      if (!(msg instanceof AssistantMessage)) {\n        continue;\n      }\n\n      const finalToolCalls: Record<string, ToolCallPart> = {};\n      let lastId = \"\";\n\n      const excludedIndexes: number[] = [];\n      msg.content.forEach((chunk, index) => {\n        if (chunk.type !== \"tool-call\") {\n          return;\n        }\n        const id = chunk.toolCallId || lastId;\n        if (!(id in finalToolCalls)) {\n          finalToolCalls[id] = shallowCopy(chunk);\n          msg.content[index] = finalToolCalls[id];\n        } else {\n          excludedIndexes.push(index);\n          const lastToolCall = finalToolCalls[id];\n          if (isString(lastToolCall.input) && isString(chunk.input)) {\n            lastToolCall.input += chunk.input;\n          } else {\n            throw new Error(\"Chunks cannot be merged.\");\n          }\n          if (!lastToolCall.toolName) {\n            lastToolCall.toolName = chunk.toolName;\n          }\n        }\n        lastId = id;\n      });\n      excludedIndexes.reverse().forEach((index) => {\n        msg.content.splice(index, 1);\n      });\n    }\n  }\n\n  getToolCalls() {\n    return this.messages\n      .filter((r) => r instanceof AssistantMessage)\n      .flatMap((r) => r.getToolCalls())\n      .filter(Boolean);\n  }\n\n  getTextMessages(): AssistantMessage[] {\n    return this.messages.filter((r) => r instanceof AssistantMessage).filter((r) => r.text);\n  }\n\n  getTextContent(): string {\n    return this.messages\n      .filter((r) => r instanceof AssistantMessage)\n      .flatMap((r) => r.text)\n      .filter(Boolean)\n      .join(\"\");\n  }\n\n  toString() {\n    return this.getTextContent();\n  }\n\n  createSnapshot() {\n    return {\n      messages: shallowCopy(this.messages),\n      usage: shallowCopy(this.usage),\n      finishReason: this.finishReason,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n\nfunction cloneSync<T extends Serializable>(serializable: T): T {\n  const snapshot = serializable.createSnapshot();\n  if (isPromise(snapshot)) {\n    throw new Error(`createSnapshot cannot be async`);\n  }\n\n  const target = Object.create(serializable.constructor.prototype) as T;\n  const load = target.loadSnapshot(snapshot);\n  if (isPromise(load)) {\n    throw new Error(`loadSnapshot cannot be async`);\n  }\n\n  return target;\n}\n"]}
+{"version":3,"sources":["../../src/backend/chat.ts"],"names":["ChatModel","Serializable","cache","NullCache","parameters","middlewares","logger","Logger","root","child","name","toolChoiceSupport","toolCallFallbackViaResponseFormat","modelSupportsToolCalling","fixInvalidToolCalls","retryOnEmptyResponse","create","input","shallowCopy","stream","undefined","RunContext","enter","params","signal","abortSignal","run","tools","forceToolCallViaResponseFormat","shouldForceToolCallViaResponseFormat","isEmpty","responseFormat","generateToolUnionSchema","filterToolsByToolChoice","toolChoice","isToolChoiceSupported","warn","modelInputMessagesBackup","messages","slice","cacheEntry","createCacheAccessor","emitter","emit","result","Retryable","executor","chunks","generator","value","_createStream","toAsyncGenerator","_create","controller","AbortController","push","callbacks","abort","aborted","ChatModelOutput","fromChunks","EmptyChatModelResponseError","getToolCalls","Tool","lastMsg","at","toolCall","parseBrokenJson","text","pair","ChatModelToolCallError","generatedContent","generatedError","response","content","length","type","toolCallId","randomString","toLowerCase","toolName","tool","find","t","availableTools","map","join","JSON","stringify","isString","isToolCallValid","parse","resolve","config","maxRetries","onRetry","_","lastError","data","AssistantMessage","tempMessage","toolNames","UserMessage","lastMessage","meta","get","error","reject","ChatModelError","middleware","createStructure","_createStructure","isFunction","fromName","options","providerId","modelId","parseModel","Target","loadModel","instance","Object","assign","schema","jsonSchema","toJsonSchema","systemTemplate","systemPromptTemplate","PromptTemplate","z","object","string","min","template","SystemMessage","render","errorTemplate","errors","expected","received","textResponse","getTextContent","validator","createSchemaValidator","success","context","output","createSnapshot","destroy","key","ObjectHashKeyFn","Serializer","serialize","msg","toPlain","isNew","task","Task","set","delete","Boolean","toolChoiceSupported","choice","includes","usage","finishReason","dedupe","final","forEach","cur","merge","other","clones","cloneSync","customMerge","totalTokens","takeBigger","promptTokens","completionTokens","messagesById","Map","messagesByToolCallId","msgId","id","filteredChunks","chunk","size","lastToolCallId","getLast","keys","has","values","main","shift","finalToolCalls","lastId","excludedIndexes","index","lastToolCall","Error","reverse","splice","filter","r","flatMap","getTextMessages","toString","loadSnapshot","snapshot","serializable","isPromise","target","prototype","load"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAqIO,MAAeA,kBAAkBC,6BAAAA,CAAAA;EArIxC;;;AAuISC,EAAAA,KAAAA,GAAwB,IAAIC,uBAAAA,EAAAA;AAC5BC,EAAAA,UAAAA,GAAkC,EAAC;AAC1BC,EAAAA,WAAAA,GAA6C,EAAA;EAC1CC,MAAAA,GAASC,iBAAAA,CAAOC,KAAKC,KAAAA,CAAM;AAC5CC,IAAAA,IAAAA,EAAM,KAAK,WAAA,CAAYA;GACzB,CAAA;EAEgBC,iBAAAA,GAAkD;AAChE,IAAA,UAAA;AACA,IAAA,MAAA;AACA,IAAA,QAAA;AACA,IAAA;;EAEKC,iCAAAA,GAAoC,IAAA;EAC3BC,wBAAAA,GAAoC,IAAA;EACpCC,mBAAAA,GAA+B,IAAA;EAC/BC,oBAAAA,GAAgC,IAAA;AAKhDC,EAAAA,MAAAA,CAAOC,KAAAA,EAAuB;AAC5BA,IAAAA,KAAAA,GAAQC,sBAAYD,KAAAA,CAAAA;AACpB,IAAA,IAAIA,KAAAA,CAAME,WAAWC,MAAAA,EAAW;AAC9BH,MAAAA,KAAAA,CAAME,MAAAA,GAAS,KAAKf,UAAAA,CAAWe,MAAAA;AACjC;AAEA,IAAA,OAAOE,sBAAAA,CAAWC,MAChB,IAAA,EACA;MAAEC,MAAAA,EAAQ;AAACN,QAAAA;;AAAiBO,MAAAA,MAAAA,EAAQP,KAAAA,EAAOQ;AAAY,KAAA,EACvD,OAAOC,GAAAA,KAAAA;AACL,MAAA,IAAI,CAAC,KAAKb,wBAAAA,EAA0B;AAClCI,QAAAA,KAAAA,CAAMU,QAAQ,EAAA;AAChB;AAEA,MAAA,MAAMC,8BAAAA,GAAiC,IAAA,CAAKC,oCAAAA,CAAqCZ,KAAAA,CAAAA;AACjF,MAAA,IAAIW,kCAAkCX,KAAAA,CAAMU,KAAAA,IAAS,CAACG,cAAAA,CAAQb,KAAAA,CAAMU,KAAK,CAAA,EAAG;AAC1EV,QAAAA,KAAAA,CAAMc,cAAAA,GAAiB,MAAMC,mCAAAA,CAC3BC,mCAAAA,CAAwBhB,MAAMU,KAAAA,EAAOV,KAAAA,CAAMiB,UAAU,CAAA,CAAA;AAEvDjB,QAAAA,KAAAA,CAAMiB,UAAAA,GAAad,MAAAA;AACrB;AAEA,MAAA,IAAI,CAAC,IAAA,CAAKe,qBAAAA,CAAsBlB,KAAAA,CAAMiB,UAAU,CAAA,EAAG;AACjD,QAAA,IAAA,CAAK5B,MAAAA,CAAO8B,IAAAA,CACV,CAAA,iCAAA,EAAoCnB,KAAAA,CAAMiB,UAAU,CAAA,6BAAA,CAA+B,CAAA;AAErFjB,QAAAA,KAAAA,CAAMiB,UAAAA,GAAad,MAAAA;AACrB;AAEA,MAAA,MAAMiB,wBAAAA,GAA2BpB,KAAAA,CAAMqB,QAAAA,CAASC,KAAAA,EAAK;AACrD,MAAA,MAAMC,UAAAA,GAAa,MAAM,IAAA,CAAKC,mBAAAA,CAAoBxB,KAAAA,CAAAA;AAElD,MAAA,IAAI;AACF,QAAA,MAAMS,GAAAA,CAAIgB,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAE1B,UAAAA;SAAM,CAAA;AAExC,QAAA,MAAM2B,MAAAA,GAAS,MAAM,IAAIC,uBAAAA,CAAU;AACjCC,UAAAA,QAAAA,kBAAU,MAAA,CAAA,YAAA;AACR,YAAA,MAAMC,SAA4B,EAAA;AAElC,YAAA,MAAMC,SAAAA,GACJR,UAAAA,CAAWS,KAAAA,KACVhC,KAAAA,CAAME,SACH,IAAA,CAAK+B,aAAAA,CAAcjC,KAAAA,EAAOS,GAAAA,IAC1ByB,4BAAAA,CAAiB,IAAA,CAAKC,OAAAA,CAAQnC,KAAAA,EAAOS,GAAAA,CAAAA,CAAAA,CAAAA;AAE3C,YAAA,MAAM2B,UAAAA,GAAa,IAAIC,eAAAA,EAAAA;AACvB,YAAA,WAAA,MAAiBL,SAASD,SAAAA,EAAW;AACnCD,cAAAA,MAAAA,CAAOQ,KAAKN,KAAAA,CAAAA;AACZ,cAAA,MAAMvB,GAAAA,CAAIgB,OAAAA,CAAQC,IAAAA,CAAK,UAAA,EAAY;AACjCM,gBAAAA,KAAAA;gBACAO,SAAAA,EAAW;kBAAEC,KAAAA,kBAAO,MAAA,CAAA,MAAMJ,UAAAA,CAAWI,KAAAA,EAAK,EAAtB,OAAA;AAAyB;eAC/C,CAAA;AACA,cAAA,IAAIJ,UAAAA,CAAW7B,OAAOkC,OAAAA,EAAS;AAC7B,gBAAA;AACF;AACF;AAEA,YAAA,MAAMd,OAAAA,GAASe,eAAAA,CAAgBC,UAAAA,CAAWb,MAAAA,CAAAA;AAC1C,YAAA,IAAIH,OAAAA,CAAOd,SAAO,EAAI;AACpB,cAAA,MAAM,IAAI+B,sCAAAA,EAAAA;AACZ;AAEA,YAAA,IACE/B,cAAAA,CAAQc,OAAAA,CAAOkB,YAAAA,EAAY,CAAA,KAC1BlC,8BAAAA,IACCX,KAAAA,CAAMiB,UAAAA,KAAe,UAAA,IACrBjB,KAAAA,CAAMiB,UAAAA,YAAsB6B,aAAAA,CAAAA,EAC9B;AACA,cAAA,MAAMC,OAAAA,GAAUpB,OAAAA,CAAON,QAAAA,CAAS2B,EAAAA,CAAG,CAAA,CAAC,CAAA;AACpC,cAAA,IAAIC,QAAAA,GAAWC,0BAAAA,CAAgBH,OAAAA,CAAQI,IAAAA,EAAM;gBAAEC,IAAAA,EAAM;AAAC,kBAAA,GAAA;AAAK,kBAAA;;eAAK,CAAA;AAChE,cAAA,IACEH,QAAAA,IACA,CAACA,QAAAA,CAASxD,IAAAA,IACV,CAACwD,QAAAA,CAAS9D,UAAAA,IACVa,KAAAA,CAAMiB,UAAAA,YAAsB6B,aAAAA,EAC5B;AACAG,gBAAAA,QAAAA,GAAW;AAAExD,kBAAAA,IAAAA,EAAMO,MAAMiB,UAAAA,CAAWxB,IAAAA;kBAAMN,UAAAA,EAAY8D;AAAS,iBAAA;AACjE;AACA,cAAA,IAAI,CAACA,QAAAA,IAAY,CAACA,SAASxD,IAAAA,IAAQ,CAACwD,SAAS9D,UAAAA,EAAY;AACvD,gBAAA,MAAM,IAAIkE,iCAAAA,CACR,CAAA,sDAAA,EAAyDN,QAAQI,IAAI,CAAA,CAAA,EACrE,EAAA,EACA;AACEG,kBAAAA,gBAAAA,EAAkBP,OAAAA,CAAQI,IAAAA;kBAC1BI,cAAAA,EAAgB,6BAAA;kBAChBC,QAAAA,EAAU7B;iBACZ,CAAA;AAEJ;AACAoB,cAAAA,OAAAA,CAAQU,QAAQC,MAAAA,GAAS,CAAA;AACzBX,cAAAA,OAAAA,CAAQU,QAAQnB,IAAAA,CAAK;gBACnBqB,IAAAA,EAAM,WAAA;AACNC,gBAAAA,UAAAA,EAAY,CAAA,KAAA,EAAQC,mBAAAA,CAAa,CAAA,CAAA,CAAGC,aAAW,CAAA,CAAA;AAC/CC,gBAAAA,QAAAA,EAAUd,QAAAA,CAASxD,IAAAA;;AACnBO,gBAAAA,KAAAA,EAAOiD,QAAAA,CAAS9D;eAClB,CAAA;AACF;AAEA,YAAA,KAAA,MAAW8D,QAAAA,IAAYtB,OAAAA,CAAOkB,YAAAA,EAAY,EAAI;AAC5C,cAAA,MAAMmB,IAAAA,GAAOhE,MAAMU,KAAAA,EAAOuD,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAEzE,IAAAA,KAASwD,QAAAA,CAASc,QAAQ,CAAA;AAClE,cAAA,IAAI,CAACC,IAAAA,EAAM;AACT,gBAAA,MAAMG,cAAAA,GAAiBnE,KAAAA,CAAMU,KAAAA,EAAO0D,GAAAA,CAAI,CAACF,CAAAA,KAAMA,CAAAA,CAAEzE,IAAI,CAAA,CAAE4E,IAAAA,CAAK,GAAA,CAAA,IAAQ,MAAA;AACpE,gBAAA,MAAM,IAAIhB,iCAAAA,CAAuB,yBAAA,EAA2B,EAAA,EAAI;kBAC9DE,cAAAA,EAAgB,CAAA,qBAAA,EAAwBN,SAASc,QAAQ,CAAA;iCAAsCI,cAAAA,CAAAA,CAAAA;AAC/Fb,kBAAAA,gBAAAA,EAAkBgB,KAAKC,SAAAA,CAAU;AAC/B9E,oBAAAA,IAAAA,EAAMwD,QAAAA,CAASc,QAAAA;oBACf/D,KAAAA,EAAOwE,eAAAA,CAASvB,SAASjD,KAAK,CAAA,GAC1BiD,SAASjD,KAAAA,GACTsE,IAAAA,CAAKC,SAAAA,CAAUtB,QAAAA,CAASjD,KAAK;mBACnC,CAAA;kBACAwD,QAAAA,EAAU7B;iBACZ,CAAA;AACF;AAEA,cAAA,IAAI,CAAC8C,2BAAAA,CAAgBxB,QAAAA,CAAAA,EAAW;AAC9B,gBAAA,MAAM,IAAII,iCAAAA,CAAuB,sBAAA,EAAwB,EAAA,EAAI;kBAC3DC,gBAAAA,EAAkBkB,eAAAA,CAASvB,SAASjD,KAAK,CAAA,GACrCiD,SAASjD,KAAAA,GACTsE,IAAAA,CAAKC,SAAAA,CAAUtB,QAAAA,CAASjD,KAAK,CAAA;kBACjCuD,cAAAA,EAAgB,CAAA,uBAAA,EAA0BN,SAASc,QAAQ,CAAA,yDAAA,CAAA;kBAC3DP,QAAAA,EAAU7B;iBACZ,CAAA;AACF;AAEA,cAAA,IAAI6C,eAAAA,CAASvB,QAAAA,CAASjD,KAAK,CAAA,EAAG;AAC5BiD,gBAAAA,QAAAA,CAASjD,KAAAA,GAAQsE,IAAAA,CAAKI,KAAAA,CAAMzB,QAAAA,CAASjD,KAAK,CAAA;AAC5C;AACF;AAEAuB,YAAAA,UAAAA,CAAWoD,QAAQ7C,MAAAA,CAAAA;AAEnB,YAAA,OAAOH,OAAAA;WACT,EAhGU,UAAA,CAAA;UAiGViD,MAAAA,EAAQ;AAAEC,YAAAA,UAAAA,EAAY7E,MAAM6E,UAAAA,IAAc;AAAE,WAAA;UAC5CC,OAAAA,kBAAS,MAAA,CAAA,OAAOC,GAAGC,SAAAA,KAAAA;AACjB,YAAA,IAAI,IAAA,CAAKnF,mBAAAA,IAAuBmF,SAAAA,YAAqB3B,iCAAAA,EAAwB;AAC3ErD,cAAAA,KAAAA,CAAMqB,QAAAA,GAAWrB,KAAAA,CAAMqB,QAAAA,CAASC,KAAAA,EAAK;AACrC,cAAA,IAAI0D,SAAAA,CAAUC,KAAK3B,gBAAAA,EAAkB;AACnCtD,gBAAAA,KAAAA,CAAMqB,SAASiB,IAAAA,CACb,IAAI4C,4BAAAA,CAAiBF,SAAAA,CAAUC,KAAK3B,gBAAAA,EAAkB;kBACpD6B,WAAAA,EAAa;AACf,iBAAA,CAAA,CAAA;AAEJ;AAEA,cAAA,MAAMC,SAAAA,GAAYpF,KAAAA,CAAMU,KAAAA,EAAO0D,GAAAA,CAAI,CAACF,CAAAA,KAAMA,CAAAA,CAAEzE,IAAI,CAAA,CAAE4E,IAAAA,CAAK,IAAA,CAAA,IAAS,MAAA;AAChErE,cAAAA,KAAAA,CAAMqB,SAASiB,IAAAA,CACb,IAAI+C,wBACF,CAAA,EAAGL,SAAAA,CAAUC,KAAK1B,cAAc;;AAAwB6B,iBAAAA,EAAAA,SAAAA,CAAAA,CAAAA,EACxD;gBACED,WAAAA,EAAa;AACf,eAAA,CAAA,CAAA;aAGN,MAAA,IACE,IAAA,CAAKrF,oBAAAA,IACLkF,SAAAA,YAAqBpC,sCAAAA,EACrB;AACA5C,cAAAA,KAAAA,CAAMqB,QAAAA,GAAWrB,KAAAA,CAAMqB,QAAAA,CAASC,KAAAA,EAAK;AACrC,cAAA,MAAMgE,WAAAA,GAActF,KAAAA,CAAMqB,QAAAA,CAAS2B,EAAAA,CAAG,CAAA,CAAC,CAAA;AACvC,cAAA,IACEsC,WAAAA,IACAA,uBAAuBJ,4BAAAA,IACvBI,WAAAA,CAAYC,KAAK,aAAA,CAAA,IACjBD,WAAAA,CAAYnC,IAAAA,KAAS,EAAA,EACrB;AACAnD,gBAAAA,KAAAA,CAAMqB,QAAAA,CAASiB,IAAAA,CACb,IAAI+C,uBAAAA,CACF,+DAAA,EACA;kBAAEF,WAAAA,EAAa;AAAK,iBAAA,CAAA,CAAA;eAG1B,MAAO;AAELnF,gBAAAA,KAAAA,CAAMqB,QAAAA,CAASiB,IAAAA,CAAK,IAAI4C,4BAAAA,CAAiB,EAAA,EAAI;kBAAEC,WAAAA,EAAa;AAAK,iBAAA,CAAA,CAAA;AACnE;AACF;WACF,EA3CS,SAAA;AA4CX,SAAA,EAAGK,GAAAA,EAAG;AAENxF,QAAAA,KAAAA,CAAMqB,QAAAA,GAAWD,wBAAAA;AACjB,QAAA,MAAMX,GAAAA,CAAIgB,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;UAAEM,KAAAA,EAAOL;SAAO,CAAA;AAClD,QAAA,OAAOA,MAAAA;AACT,OAAA,CAAA,OAAS8D,KAAAA,EAAO;AACd,QAAA,MAAMhF,GAAAA,CAAIgB,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAE1B,UAAAA,KAAAA;AAAOyF,UAAAA;SAAM,CAAA;AAC/C,QAAA,MAAMlE,UAAAA,CAAWmE,OAAOD,KAAAA,CAAAA;AACxB,QAAA,IAAIA,iBAAiBE,yBAAAA,EAAgB;AACnC,UAAA,MAAMF,KAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAIE,0BAAe,CAAA,wCAAA,CAAA,EAA4C;AAACF,YAAAA;AAAM,WAAA,CAAA;AAC9E;OACF,SAAA;AACE,QAAA,MAAMhF,GAAAA,CAAIgB,OAAAA,CAAQC,IAAAA,CAAK,QAAA,EAAU,IAAA,CAAA;AACnC;AACF,KAAA,CAAA,CACAkE,UAAAA,CAAU,GAAI,IAAA,CAAKxG,WAAW,CAAA;AAClC;AAEAyG,EAAAA,eAAAA,CAAmB7F,KAAAA,EAAgC;AACjD,IAAA,OAAOI,sBAAAA,CAAWC,MAChB,IAAA,EACA;MAAEC,MAAAA,EAAQ;AAACN,QAAAA;;AAAiBO,MAAAA,MAAAA,EAAQP,KAAAA,EAAOQ;AAAY,KAAA,EACvD,OAAOC,GAAAA,KAAAA;AACL,MAAA,OAAO,MAAM,IAAA,CAAKqF,gBAAAA,CAAoB9F,KAAAA,EAAOS,GAAAA,CAAAA;KAC/C,CAAA;AAEJ;EAEAmE,MAAAA,CAAO,EAAE3F,KAAAA,EAAOE,UAAAA,EAAU,EAAsB;AAC9C,IAAA,IAAIF,KAAAA,EAAO;AACT,MAAA,IAAA,CAAKA,QAAQ8G,iBAAAA,CAAW9G,KAAAA,IAASA,KAAAA,CAAM,IAAA,CAAKA,KAAK,CAAA,GAAIA,KAAAA;AACvD;AACA,IAAA,IAAIE,UAAAA,EAAY;AACd,MAAA,IAAA,CAAKA,aAAa4G,iBAAAA,CAAW5G,UAAAA,IAAcA,UAAAA,CAAW,IAAA,CAAKA,UAAU,CAAA,GAAIA,UAAAA;AAC3E;AACF;EAEA,aAAa6G,QAAAA,CAASvG,MAAoCwG,OAAAA,EAA+B;AACvF,IAAA,MAAM,EAAEC,UAAAA,EAAYC,OAAAA,EAAO,GAAKC,uBAAW3G,IAAAA,CAAAA;AAC3C,IAAA,MAAM4G,MAAAA,GAAS,MAAMC,qBAAAA,CAAqBJ,UAAAA,EAAY,MAAA,CAAA;AACtD,IAAA,MAAMK,QAAAA,GAAW,IAAIF,MAAAA,CAAOF,OAAAA,IAAWhG,MAAAA,CAAAA;AACvC,IAAA,IAAI8F,OAAAA,EAAS;AACXO,MAAAA,MAAAA,CAAOC,MAAAA,CAAOF,QAAAA,CAASpH,UAAAA,EAAY8G,OAAAA,CAAAA;AACrC;AACA,IAAA,OAAOM,QAAAA;AACT;EAWA,MAAgBT,gBAAAA,CACd9F,OACAS,GAAAA,EACmC;AACnC,IAAA,MAAM,EAAEiG,MAAAA,EAAQ,GAAGT,OAAAA,EAAAA,GAAYjG,KAAAA;AAC/B,IAAA,MAAM2G,UAAAA,GAAaC,wBAAaF,MAAAA,CAAAA;AAEhC,IAAA,MAAMG,cAAAA,GACJ7G,KAAAA,CAAM8G,oBAAAA,IACN,IAAIC,2BAAAA,CAAe;AACjBL,MAAAA,MAAAA,EAAQM,MAAEC,MAAAA,CAAO;AACfP,QAAAA,MAAAA,EAAQM,KAAAA,CAAEE,MAAAA,EAAM,CAAGC,GAAAA,CAAI,CAAA;OACzB,CAAA;MACAC,QAAAA,EAAU,CAAA;;;;;;;KAOZ,CAAA;AAEF,IAAA,MAAM/F,QAAAA,GAAsB;MAC1B,IAAIgG,yBAAAA,CAAcR,eAAeS,MAAAA,CAAO;AAAEZ,QAAAA,MAAAA,EAAQpC,IAAAA,CAAKC,SAAAA,CAAUoC,UAAAA,EAAY,IAAA,EAAM,CAAA;AAAG,OAAA,CAAA,CAAA;SACnF3G,KAAAA,CAAMqB;;AAGX,IAAA,MAAMkG,aAAAA,GAAgB,IAAIR,2BAAAA,CAAe;AACvCL,MAAAA,MAAAA,EAAQM,MAAEC,MAAAA,CAAO;AACfO,QAAAA,MAAAA,EAAQR,MAAEE,MAAAA,EAAM;AAChBO,QAAAA,QAAAA,EAAUT,MAAEE,MAAAA,EAAM;AAClBQ,QAAAA,QAAAA,EAAUV,MAAEE,MAAAA;OACd,CAAA;MACAE,QAAAA,EAAU,CAAA;;;KAGZ,CAAA;AAEA,IAAA,OAAO,IAAIxF,uBAAAA,CAAoC;AAC7CC,MAAAA,QAAAA,kBAAU,MAAA,CAAA,YAAA;AACR,QAAA,MAAM2B,QAAAA,GAAW,MAAM,IAAA,CAAKrB,OAAAA,CAC1B;UACE,GAAG8D,OAAAA;AACH5E,UAAAA,QAAAA;UACAP,cAAAA,EAAgB;YAAE6C,IAAAA,EAAM;AAAc;AACxC,SAAA,EACAlD,GAAAA,CAAAA;AAGF,QAAA,MAAMkH,YAAAA,GAAenE,SAASoE,cAAAA,EAAc;AAC5C,QAAA,MAAMX,MAAAA,GAAY/D,2BAAgByE,YAAAA,EAAc;UAAEvE,IAAAA,EAAM;AAAC,YAAA,GAAA;AAAK,YAAA;;SAAK,CAAA;AACnE,QAAA,MAAMyE,SAAAA,GAAYC,iCAAsBpB,MAAAA,CAAAA;AAExC,QAAA,MAAMqB,OAAAA,GAAUF,UAAUZ,MAAAA,CAAAA;AAC1B,QAAA,IAAI,CAACc,OAAAA,EAAS;AACZ,UAAA,MAAMC,OAAAA,GAAU;YACdP,QAAAA,EAAUnD,IAAAA,CAAKC,UAAUoC,UAAAA,CAAAA;YACzBe,QAAAA,EAAUC,YAAAA;AACVH,YAAAA,MAAAA,EAAQlD,IAAAA,CAAKC,SAAAA,CAAUsD,SAAAA,CAAUL,MAAAA,IAAU,EAAE;AAC/C,WAAA;AAEAnG,UAAAA,QAAAA,CAASiB,KAAK,IAAI+C,uBAAAA,CAAYkC,cAAcD,MAAAA,CAAOU,OAAAA,CAAAA,CAAAA,CAAAA;AACnD,UAAA,MAAM,IAAIrC,yBAAAA,CAAe,CAAA,mCAAA,CAAA,EAAuC,EAAA,EAAI;AAClEqC,YAAAA;WACF,CAAA;AACF;AAEA,QAAA,OAAO;AACLf,UAAAA,MAAAA;UACAgB,MAAAA,EAAQzE;AACV,SAAA;OACF,EAhCU,UAAA,CAAA;MAiCVoB,MAAAA,EAAQ;AACNrE,QAAAA,MAAAA,EAAQE,GAAAA,CAAIF,MAAAA;AACZsE,QAAAA,UAAAA,EAAY7E,OAAO6E,UAAAA,IAAc;AACnC;AACF,KAAA,EAAGW,GAAAA,EAAG;AACR;EAEA0C,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLjJ,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZwC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdrC,WAAAA,EAAaa,qBAAAA,CAAY,KAAKb,WAAW,CAAA;MACzCD,UAAAA,EAAYc,qBAAAA,CAAY,KAAKd,UAAU,CAAA;AACvCE,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;MACbK,iBAAAA,EAAmB,IAAA,CAAKA,kBAAkB4B,KAAAA,EAAK;AAC/C3B,MAAAA,iCAAAA,EAAmC,IAAA,CAAKA,iCAAAA;AACxCC,MAAAA,wBAAAA,EAA0B,IAAA,CAAKA,wBAAAA;AAC/BE,MAAAA,oBAAAA,EAAsB,IAAA,CAAKA,oBAAAA;AAC3BD,MAAAA,mBAAAA,EAAqB,IAAA,CAAKA;AAC5B,KAAA;AACF;EAEAsI,OAAAA,GAAU;AACR,IAAA,IAAA,CAAK1G,QAAQ0G,OAAAA,EAAO;AACtB;EAEA,MAAgB3G,mBAAAA,CAAoB,EAClChB,WAAAA,EAAauE,CAAAA,EACb1D,QAAAA,EACAX,QAAQ,EAAA,EACR,GAAGV,KAAAA,EAAAA,EACc;AACjB,IAAA,MAAMoI,MAAMC,kCAAAA,CAAgB;MAC1B,GAAGrI,KAAAA;MACHqB,QAAAA,EAAU,MAAMiH,yBAAAA,CAAWC,SAAAA,CAAUlH,QAAAA,CAAS+C,GAAAA,CAAI,CAACoE,GAAAA,KAAQA,GAAAA,CAAIC,OAAAA,EAAO,CAAA,CAAA;MACtE/H,KAAAA,EAAO,MAAM4H,yBAAAA,CAAWC,SAAAA,CAAU7H,KAAAA;KACpC,CAAA;AACA,IAAA,MAAMsB,KAAAA,GAAQ,MAAM,IAAA,CAAK/C,KAAAA,CAAMuG,IAAI4C,GAAAA,CAAAA;AACnC,IAAA,MAAMM,QAAQ1G,KAAAA,KAAU7B,MAAAA;AAExB,IAAA,IAAIwI,IAAAA,GAAuC,IAAA;AAC3C,IAAA,IAAID,KAAAA,EAAO;AACTC,MAAAA,IAAAA,GAAO,IAAIC,qBAAAA,EAAAA;AACX,MAAA,MAAM,IAAA,CAAK3J,KAAAA,CAAM4J,GAAAA,CAAIT,GAAAA,EAAKO,IAAAA,CAAAA;AAC5B;AAEA,IAAA,OAAO;AACLP,MAAAA,GAAAA;AACApG,MAAAA,KAAAA;AACA2C,MAAAA,OAAAA,0BAAsC3C,MAAAA,KAAAA;AACpC2G,QAAAA,IAAAA,EAAMhE,UAAU3C,MAAAA,CAAAA;OAClB,EAFS,SAAA,CAAA;AAGT0D,MAAAA,MAAAA,gCAAeD,KAAAA,KAAAA;AACbkD,QAAAA,IAAAA,EAAMjD,SAASD,KAAAA,CAAAA;AACf,QAAA,IAAIiD,KAAAA,EAAO;AACT,UAAA,MAAM,IAAA,CAAKzJ,KAAAA,CAAM6J,MAAAA,CAAOV,GAAAA,CAAAA;AAC1B;OACF,EALQ,QAAA;AAMV,KAAA;AACF;AAEUxH,EAAAA,oCAAAA,CAAqC,EAC7CF,KAAAA,GAAQ,EAAA,EACRO,UAAAA,EACAH,gBAAc,EACG;AACjB,IAAA,IACED,cAAAA,CAAQH,KAAAA,CAAAA,IACR,CAACO,cACDA,UAAAA,KAAe,MAAA,IACfA,UAAAA,KAAe,MAAA,IACf,CAAC,IAAA,CAAKtB,iCAAAA,IACNoJ,OAAAA,CAAQjI,cAAAA,CAAAA,EACR;AACA,MAAA,OAAO,KAAA;AACT;AAEA,IAAA,MAAMkI,mBAAAA,GAAsB,IAAA,CAAK9H,qBAAAA,CAAsBD,UAAAA,CAAAA;AACvD,IAAA,OAAO,CAAC,IAAA,CAAKrB,wBAAAA,IAA4B,CAACoJ,mBAAAA;AAC5C;AAEU9H,EAAAA,qBAAAA,CAAsB+H,MAAAA,EAAuC;AACrE,IAAA,OACE,CAACA,MAAAA,KACAA,MAAAA,YAAkBnG,aAAAA,GACf,IAAA,CAAKpD,iBAAAA,CAAkBwJ,QAAAA,CAAS,QAAA,CAAA,GAChC,IAAA,CAAKxJ,iBAAAA,CAAkBwJ,QAAAA,CAASD,MAAAA,CAAAA,CAAAA;AAExC;AACF;AAEO,MAAMvG,wBAAwB1D,6BAAAA,CAAAA;EA3iBrC;;;;;;EA4iBE,WAAA,CACkBqC,QAAAA,EACT8H,OACAC,YAAAA,EACP;AACA,IAAA,KAAA,EAAK,EAAA,KAJW/H,QAAAA,GAAAA,QAAAA,EAAAA,KACT8H,KAAAA,GAAAA,KAAAA,EAAAA,KACAC,YAAAA,GAAAA,YAAAA;AAGP,IAAA,IAAA,CAAKC,MAAAA,EAAM;AACb;AAEA,EAAA,OAAO1G,WAAWb,MAAAA,EAA2B;AAC3C,IAAA,MAAMwH,KAAAA,GAAQ,IAAI5G,eAAAA,CAAgB,EAAE,CAAA;AACpCZ,IAAAA,MAAAA,CAAOyH,QAAQ,CAACC,GAAAA,KAAQF,KAAAA,CAAMG,KAAAA,CAAMD,GAAAA,CAAAA,CAAAA;AACpC,IAAA,OAAOF,KAAAA;AACT;EAEAzI,OAAAA,GAAU;AACR,IAAA,IAAI,IAAA,CAAKQ,QAAAA,CAASqC,MAAAA,KAAW,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AACT;AACA,IAAA,OAAO,KAAKkE,cAAAA,EAAc,KAAO,MAAM,IAAA,CAAK/E,YAAAA,GAAea,MAAAA,KAAW,CAAA;AACxE;AAEA+F,EAAAA,KAAAA,CAAMC,KAAAA,EAAwB;AAC5B,IAAA,IAAIA,KAAAA,CAAMrI,QAAAA,CAASqC,MAAAA,GAAS,CAAA,EAAG;AAC7B,MAAA,MAAMiG,MAAAA,GAASD,KAAAA,CAAMrI,QAAAA,CAAS+C,GAAAA,CAAIwF,SAAAA,CAAAA;AAClC,MAAA,IAAA,CAAKvI,QAAAA,CAASiB,IAAAA,CAAI,GAAIqH,MAAAA,CAAAA;AACtB,MAAA,IAAA,CAAKN,MAAAA,EAAM;AACb;AAEA,IAAA,IAAA,CAAKD,eAAeM,KAAAA,CAAMN,YAAAA;AAC1B,IAAA,IAAI,IAAA,CAAKD,KAAAA,IAASO,KAAAA,CAAMP,KAAAA,EAAO;AAC7B,MAAA,IAAA,CAAKA,QAAQU,sBAAAA,CAAY;QAAC,IAAA,CAAKV,KAAAA;QAAOO,KAAAA,CAAMP;AAAQ,OAAA,EAAA;QAClDW,WAAAA,EAAaC,qBAAAA;QACbC,YAAAA,EAAcD,qBAAAA;QACdE,gBAAAA,EAAkBF;OACpB,CAAA;AACF,KAAA,MAAA,IAAWL,MAAMP,KAAAA,EAAO;AACtB,MAAA,IAAA,CAAKA,KAAAA,GAAQlJ,qBAAAA,CAAYyJ,KAAAA,CAAMP,KAAK,CAAA;AACtC;AACF;EAEAE,MAAAA,GAAe;AAEb,IAAA,IAAI,IAAA,CAAKhI,QAAAA,CAASqC,MAAAA,GAAS,CAAA,EAAG;AAC5B,MAAA,MAAMwG,YAAAA,uBAAmBC,GAAAA,EAAAA;AACzB,MAAA,MAAMC,oBAAAA,uBAA2BD,GAAAA,EAAAA;AAEjC,MAAA,KAAA,MAAW3B,GAAAA,IAAO,KAAKnH,QAAAA,EAAU;AAC/B,QAAA,MAAMgJ,KAAAA,GAAQ7B,IAAI8B,EAAAA,IAAM,EAAA;AAExB,QAAA,IAAI9B,eAAetD,4BAAAA,IAAoBsD,GAAAA,CAAI3F,YAAAA,EAAY,CAAGa,SAAS,CAAA,EAAG;AACpE,UAAA,MAAM6G,iBAA8C,EAAA;AACpD,UAAA,KAAA,MAAWC,KAAAA,IAAShC,IAAI/E,OAAAA,EAAS;AAC/B,YAAA,IAAI+G,KAAAA,CAAM7G,SAAS,WAAA,EAAa;AAC9B4G,cAAAA,cAAAA,CAAejI,KAAKkI,KAAAA,CAAAA;AACpB,cAAA;AACF;AAGA,YAAA,IAAI,CAACA,KAAAA,CAAM5G,UAAAA,IAAcwG,oBAAAA,CAAqBK,OAAO,CAAA,EAAG;AACtD,cAAA,MAAMC,cAAAA,GAAiBC,kBAAAA,CAAQP,oBAAAA,CAAqBQ,IAAAA,IAAQ,EAAA,CAAA;AAC5D,cAAA,IAAIF,cAAAA,EAAgB;AAClBF,gBAAAA,KAAAA,CAAM5G,UAAAA,GAAa8G,cAAAA;AACrB;AACF;AAEA,YAAA,IAAIF,MAAM5G,UAAAA,IAAcwG,oBAAAA,CAAqBS,GAAAA,CAAIL,KAAAA,CAAM5G,UAAU,CAAA,EAAG;AAClEwG,cAAAA,oBAAAA,CAAqB5E,IAAIgF,KAAAA,CAAM5G,UAAU,CAAA,CAAGH,OAAAA,CAAQnB,KAAKkI,KAAAA,CAAAA;AAC3D,aAAA,MAAA,IAAWA,MAAM5G,UAAAA,EAAY;AAC3BwG,cAAAA,oBAAAA,CAAqBvB,GAAAA,CAAI2B,KAAAA,CAAM5G,UAAAA,EAAY4E,GAAAA,CAAAA;AAC3C+B,cAAAA,cAAAA,CAAejI,KAAKkI,KAAAA,CAAAA;AACtB;AACF;AAEAhC,UAAAA,GAAAA,CAAI/E,QAAQC,MAAAA,GAAS,CAAA;AACrB8E,UAAAA,GAAAA,CAAI/E,OAAAA,CAAQnB,IAAAA,CAAI,GAAIiI,cAAAA,CAAAA;AAEpB,UAAA,IAAIA,cAAAA,CAAe7G,WAAW,CAAA,EAAG;AAC/B,YAAA;AACF;AACF;AAEA,QAAA,IAAI,CAACwG,YAAAA,CAAaW,GAAAA,CAAIR,KAAAA,CAAAA,EAAQ;AAC5BH,UAAAA,YAAAA,CAAarB,IAAIwB,KAAAA,EAAO;AAAC7B,YAAAA;AAAI,WAAA,CAAA;SAC/B,MAAO;AACL0B,UAAAA,YAAAA,CAAa1E,GAAAA,CAAI6E,KAAAA,CAAAA,CAAQ/H,IAAAA,CAAKkG,GAAAA,CAAAA;AAChC;AACF;AAEA,MAAA,IAAA,CAAKnH,SAASqC,MAAAA,GAAS,CAAA;AAEvB,MAAA,KAAA,MAAWrC,QAAAA,IAAY6I,YAAAA,CAAaY,MAAAA,EAAM,EAAI;AAC5C,QAAA,MAAMC,IAAAA,GAAO1J,SAAS2J,KAAAA,EAAK;AAC3B,QAAA,KAAA,MAAWtB,SAASrI,QAAAA,EAAU;AAC5B0J,UAAAA,IAAAA,CAAKtB,MAAMC,KAAAA,CAAAA;AACb;AACA,QAAA,IAAA,CAAKrI,QAAAA,CAASiB,KAAKyI,IAAAA,CAAAA;AACrB;AACF;AAGA,IAAA,KAAA,MAAWvC,GAAAA,IAAO,KAAKnH,QAAAA,EAAU;AAC/B,MAAA,IAAI,EAAEmH,eAAetD,4BAAAA,CAAAA,EAAmB;AACtC,QAAA;AACF;AAEA,MAAA,MAAM+F,iBAA+C,EAAC;AACtD,MAAA,IAAIC,MAAAA,GAAS,EAAA;AAEb,MAAA,MAAMC,kBAA4B,EAAA;AAClC3C,MAAAA,GAAAA,CAAI/E,OAAAA,CAAQ8F,OAAAA,CAAQ,CAACiB,KAAAA,EAAOY,KAAAA,KAAAA;AAC1B,QAAA,IAAIZ,KAAAA,CAAM7G,SAAS,WAAA,EAAa;AAC9B,UAAA;AACF;AACA,QAAA,MAAM2G,EAAAA,GAAKE,MAAM5G,UAAAA,IAAcsH,MAAAA;AAC/B,QAAA,IAAI,EAAEZ,MAAMW,cAAAA,CAAAA,EAAiB;AAC3BA,UAAAA,cAAAA,CAAeX,EAAAA,CAAAA,GAAMrK,qBAAAA,CAAYuK,KAAAA,CAAAA;AACjChC,UAAAA,GAAAA,CAAI/E,OAAAA,CAAQ2H,KAAAA,CAAAA,GAASH,cAAAA,CAAeX,EAAAA,CAAAA;SACtC,MAAO;AACLa,UAAAA,eAAAA,CAAgB7I,KAAK8I,KAAAA,CAAAA;AACrB,UAAA,MAAMC,YAAAA,GAAeJ,eAAeX,EAAAA,CAAAA;AACpC,UAAA,IAAI9F,gBAAS6G,YAAAA,CAAarL,KAAK,KAAKwE,eAAAA,CAASgG,KAAAA,CAAMxK,KAAK,CAAA,EAAG;AACzDqL,YAAAA,YAAAA,CAAarL,SAASwK,KAAAA,CAAMxK,KAAAA;WAC9B,MAAO;AACL,YAAA,MAAM,IAAIsL,MAAM,0BAAA,CAAA;AAClB;AACA,UAAA,IAAI,CAACD,aAAatH,QAAAA,EAAU;AAC1BsH,YAAAA,YAAAA,CAAatH,WAAWyG,KAAAA,CAAMzG,QAAAA;AAChC;AACF;AACAmH,QAAAA,MAAAA,GAASZ,EAAAA;OACX,CAAA;AACAa,MAAAA,eAAAA,CAAgBI,OAAAA,EAAO,CAAGhC,OAAAA,CAAQ,CAAC6B,KAAAA,KAAAA;AACjC5C,QAAAA,GAAAA,CAAI/E,OAAAA,CAAQ+H,MAAAA,CAAOJ,KAAAA,EAAO,CAAA,CAAA;OAC5B,CAAA;AACF;AACF;EAEAvI,YAAAA,GAAe;AACb,IAAA,OAAO,KAAKxB,QAAAA,CACToK,MAAAA,CAAO,CAACC,CAAAA,KAAMA,aAAaxG,4BAAAA,CAAAA,CAC3ByG,OAAAA,CAAQ,CAACD,MAAMA,CAAAA,CAAE7I,YAAAA,EAAY,CAAA,CAC7B4I,OAAO1C,OAAAA,CAAAA;AACZ;EAEA6C,eAAAA,GAAsC;AACpC,IAAA,OAAO,IAAA,CAAKvK,QAAAA,CAASoK,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,YAAaxG,4BAAAA,CAAAA,CAAkBuG,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEvI,IAAI,CAAA;AACxF;EAEAyE,cAAAA,GAAyB;AACvB,IAAA,OAAO,KAAKvG,QAAAA,CACToK,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,YAAaxG,4BAAAA,CAAAA,CAC3ByG,OAAAA,CAAQ,CAACD,CAAAA,KAAMA,EAAEvI,IAAI,CAAA,CACrBsI,OAAO1C,OAAAA,CAAAA,CACP1E,KAAK,EAAA,CAAA;AACV;EAEAwH,QAAAA,GAAW;AACT,IAAA,OAAO,KAAKjE,cAAAA,EAAc;AAC5B;EAEAM,cAAAA,GAAiB;AACf,IAAA,OAAO;MACL7G,QAAAA,EAAUpB,qBAAAA,CAAY,KAAKoB,QAAQ,CAAA;MACnC8H,KAAAA,EAAOlJ,qBAAAA,CAAY,KAAKkJ,KAAK,CAAA;AAC7BC,MAAAA,YAAAA,EAAc,IAAA,CAAKA;AACrB,KAAA;AACF;AAEA0C,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DvF,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMsF,QAAAA,CAAAA;AACtB;AACF;AAEA,SAASnC,UAAkCoC,YAAAA,EAAe;AACxD,EAAA,MAAMD,QAAAA,GAAWC,aAAa9D,cAAAA,EAAc;AAC5C,EAAA,IAAI+D,gBAAAA,CAAUF,QAAAA,CAAAA,EAAW;AACvB,IAAA,MAAM,IAAIT,MAAM,CAAA,8BAAA,CAAgC,CAAA;AAClD;AAEA,EAAA,MAAMY,MAAAA,GAAS1F,MAAAA,CAAOzG,MAAAA,CAAOiM,YAAAA,CAAa,YAAYG,SAAS,CAAA;AAC/D,EAAA,MAAMC,IAAAA,GAAOF,MAAAA,CAAOJ,YAAAA,CAAaC,QAAAA,CAAAA;AACjC,EAAA,IAAIE,gBAAAA,CAAUG,IAAAA,CAAAA,EAAO;AACnB,IAAA,MAAM,IAAId,MAAM,CAAA,4BAAA,CAA8B,CAAA;AAChD;AAEA,EAAA,OAAOY,MAAAA;AACT;AAbStC,MAAAA,CAAAA,SAAAA,EAAAA,WAAAA,CAAAA","file":"chat.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { customMerge, getLast } from \"@/internals/helpers/object.js\";\nimport { takeBigger } from \"@/internals/helpers/number.js\";\nimport { Callback } from \"@/emitter/types.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext, MiddlewareType, RunContext } from \"@/context.js\";\nimport { isEmpty, isFunction, isPromise, isString, randomString } from \"remeda\";\nimport { ObjectHashKeyFn } from \"@/cache/decoratorCache.js\";\nimport { Task } from \"promise-based-task\";\nimport { NullCache } from \"@/cache/nullCache.js\";\nimport { BaseCache } from \"@/cache/base.js\";\nimport {\n  filterToolsByToolChoice,\n  FullModelName,\n  generateToolUnionSchema,\n  loadModel,\n  parseModel,\n} from \"@/backend/utils.js\";\nimport { ProviderName } from \"@/backend/constants.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { AssistantMessage, Message, SystemMessage, UserMessage } from \"@/backend/message.js\";\n\nimport {\n  ChatModelError,\n  ChatModelToolCallError,\n  EmptyChatModelResponseError,\n} from \"@/backend/errors.js\";\nimport { z, ZodSchema, ZodType } from \"zod\";\nimport {\n  createSchemaValidator,\n  parseBrokenJson,\n  toJsonSchema,\n} from \"@/internals/helpers/schema.js\";\nimport { Retryable } from \"@/internals/helpers/retryable.js\";\nimport { SchemaObject, ValidateFunction } from \"ajv\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { toAsyncGenerator } from \"@/internals/helpers/promise.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { ToolCallPart } from \"ai\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\n\nexport interface ChatModelParameters {\n  maxTokens?: number;\n  topP?: number;\n  frequencyPenalty?: number;\n  temperature?: number;\n  topK?: number;\n  n?: number;\n  presencePenalty?: number;\n  seed?: number;\n  stopSequences?: string[];\n  stream?: boolean;\n}\n\ninterface ResponseObjectJson {\n  type: \"object-json\";\n  schema?: SchemaObject;\n  name?: string;\n  description?: string;\n}\n\nexport interface ChatModelObjectInput<T> extends ChatModelParameters {\n  schema: z.ZodSchema<T, any, any> | ResponseObjectJson;\n  systemPromptTemplate?: PromptTemplate<ZodType<{ schema: string }>>;\n  messages: Message[];\n  abortSignal?: AbortSignal;\n  maxRetries?: number;\n}\n\nexport interface ChatModelObjectOutput<T> {\n  object: T;\n  output: ChatModelOutput;\n}\n\nexport type ChatModelToolChoice = \"auto\" | \"none\" | \"required\" | AnyTool;\n\nexport interface ChatModelInput extends ChatModelParameters {\n  tools?: AnyTool[];\n  abortSignal?: AbortSignal;\n  stopSequences?: string[];\n  responseFormat?: ZodSchema | ResponseObjectJson;\n  toolChoice?: ChatModelToolChoice;\n  messages: Message[];\n  streamPartialToolCalls?: boolean;\n  maxRetries?: number;\n}\n\nexport type ChatModelFinishReason =\n  | \"stop\"\n  | \"length\"\n  | \"content-filter\"\n  | \"tool-calls\"\n  | \"error\"\n  | \"other\"\n  | \"unknown\";\n\nexport interface ChatModelUsage {\n  promptTokens: number;\n  completionTokens: number;\n  totalTokens: number;\n  reasoningTokens?: number;\n  cachedPromptTokens?: number;\n}\n\nexport interface ChatModelEvents {\n  newToken?: Callback<{ value: ChatModelOutput; callbacks: { abort: () => void } }>;\n  success?: Callback<{ value: ChatModelOutput }>;\n  start?: Callback<{ input: ChatModelInput }>;\n  error?: Callback<{ input: ChatModelInput; error: FrameworkError }>;\n  finish?: Callback<null>;\n}\n\nexport type ChatModelEmitter<A = Record<never, never>> = Emitter<\n  ChatModelEvents & Omit<A, keyof ChatModelEvents>\n>;\n\nexport type ChatModelCache = BaseCache<Task<ChatModelOutput[]>>;\nexport type ConfigFn<T> = (value: T) => T;\nexport interface ChatConfig {\n  cache?: ChatModelCache | ConfigFn<ChatModelCache>;\n  parameters?: ChatModelParameters | ConfigFn<ChatModelParameters>;\n}\n\nexport type ChatModelToolChoiceSupport = \"required\" | \"none\" | \"single\" | \"auto\";\n\nexport abstract class ChatModel extends Serializable {\n  public abstract readonly emitter: Emitter<ChatModelEvents>;\n  public cache: ChatModelCache = new NullCache();\n  public parameters: ChatModelParameters = {};\n  public readonly middlewares: MiddlewareType<typeof this>[] = [];\n  protected readonly logger = Logger.root.child({\n    name: this.constructor.name,\n  });\n\n  public readonly toolChoiceSupport: ChatModelToolChoiceSupport[] = [\n    \"required\",\n    \"none\",\n    \"single\",\n    \"auto\",\n  ];\n  public toolCallFallbackViaResponseFormat = true;\n  public readonly modelSupportsToolCalling: boolean = true;\n  public readonly fixInvalidToolCalls: boolean = true;\n  public readonly retryOnEmptyResponse: boolean = true;\n\n  abstract get modelId(): string;\n  abstract get providerId(): string;\n\n  create(input: ChatModelInput) {\n    input = shallowCopy(input);\n    if (input.stream === undefined) {\n      input.stream = this.parameters.stream;\n    }\n\n    return RunContext.enter(\n      this,\n      { params: [input] as const, signal: input?.abortSignal },\n      async (run): Promise<ChatModelOutput> => {\n        if (!this.modelSupportsToolCalling) {\n          input.tools = [];\n        }\n\n        const forceToolCallViaResponseFormat = this.shouldForceToolCallViaResponseFormat(input);\n        if (forceToolCallViaResponseFormat && input.tools && !isEmpty(input.tools)) {\n          input.responseFormat = await generateToolUnionSchema(\n            filterToolsByToolChoice(input.tools, input.toolChoice),\n          );\n          input.toolChoice = undefined;\n        }\n\n        if (!this.isToolChoiceSupported(input.toolChoice)) {\n          this.logger.warn(\n            `The following tool choice value '${input.toolChoice}' is not supported. Ignoring.`,\n          );\n          input.toolChoice = undefined;\n        }\n\n        const modelInputMessagesBackup = input.messages.slice();\n        const cacheEntry = await this.createCacheAccessor(input);\n\n        try {\n          await run.emitter.emit(\"start\", { input });\n\n          const result = await new Retryable({\n            executor: async () => {\n              const chunks: ChatModelOutput[] = [];\n\n              const generator =\n                cacheEntry.value ??\n                (input.stream\n                  ? this._createStream(input, run)\n                  : toAsyncGenerator(this._create(input, run)));\n\n              const controller = new AbortController();\n              for await (const value of generator) {\n                chunks.push(value);\n                await run.emitter.emit(\"newToken\", {\n                  value,\n                  callbacks: { abort: () => controller.abort() },\n                });\n                if (controller.signal.aborted) {\n                  break;\n                }\n              }\n\n              const result = ChatModelOutput.fromChunks(chunks);\n              if (result.isEmpty()) {\n                throw new EmptyChatModelResponseError();\n              }\n\n              if (\n                isEmpty(result.getToolCalls()) &&\n                (forceToolCallViaResponseFormat ||\n                  input.toolChoice === \"required\" ||\n                  input.toolChoice instanceof Tool)\n              ) {\n                const lastMsg = result.messages.at(-1)!;\n                let toolCall = parseBrokenJson(lastMsg.text, { pair: [\"{\", \"}\"] });\n                if (\n                  toolCall &&\n                  !toolCall.name &&\n                  !toolCall.parameters &&\n                  input.toolChoice instanceof Tool\n                ) {\n                  toolCall = { name: input.toolChoice.name, parameters: toolCall };\n                }\n                if (!toolCall || !toolCall.name || !toolCall.parameters) {\n                  throw new ChatModelToolCallError(\n                    `Failed to produce a valid tool call. Generate output: ${lastMsg.text}`,\n                    [],\n                    {\n                      generatedContent: lastMsg.text,\n                      generatedError: \"Tool call was not produced.\",\n                      response: result,\n                    },\n                  );\n                }\n                lastMsg.content.length = 0;\n                lastMsg.content.push({\n                  type: \"tool-call\",\n                  toolCallId: `call_${randomString(8).toLowerCase()}`,\n                  toolName: toolCall.name, // todo: add types\n                  input: toolCall.parameters,\n                });\n              }\n\n              for (const toolCall of result.getToolCalls()) {\n                const tool = input.tools?.find((t) => t.name === toolCall.toolName);\n                if (!tool) {\n                  const availableTools = input.tools?.map((t) => t.name).join(\",\") || \"None\";\n                  throw new ChatModelToolCallError(\"Non existing tool call.\", [], {\n                    generatedError: `Error: Unknown tool '${toolCall.toolName}'.\\nUse on of the available tools: ${availableTools}`,\n                    generatedContent: JSON.stringify({\n                      name: toolCall.toolName,\n                      input: isString(toolCall.input)\n                        ? toolCall.input\n                        : JSON.stringify(toolCall.input),\n                    }),\n                    response: result,\n                  });\n                }\n\n                if (!isToolCallValid(toolCall)) {\n                  throw new ChatModelToolCallError(\"Malformed tool call.\", [], {\n                    generatedContent: isString(toolCall.input)\n                      ? toolCall.input\n                      : JSON.stringify(toolCall.input),\n                    generatedError: `The tool call for the '${toolCall.toolName}' tool has malformed parameters. It must be a valid JSON.`,\n                    response: result,\n                  });\n                }\n\n                if (isString(toolCall.input)) {\n                  toolCall.input = JSON.parse(toolCall.input);\n                }\n              }\n\n              cacheEntry.resolve(chunks);\n\n              return result;\n            },\n            config: { maxRetries: input.maxRetries ?? 0 },\n            onRetry: async (_, lastError) => {\n              if (this.fixInvalidToolCalls && lastError instanceof ChatModelToolCallError) {\n                input.messages = input.messages.slice();\n                if (lastError.data.generatedContent) {\n                  input.messages.push(\n                    new AssistantMessage(lastError.data.generatedContent, {\n                      tempMessage: true,\n                    }),\n                  );\n                }\n\n                const toolNames = input.tools?.map((t) => t.name).join(\", \") || \"None\";\n                input.messages.push(\n                  new UserMessage(\n                    `${lastError.data.generatedError}\\n\\nAvailable Tools: ${toolNames}`,\n                    {\n                      tempMessage: true,\n                    },\n                  ),\n                );\n              } else if (\n                this.retryOnEmptyResponse &&\n                lastError instanceof EmptyChatModelResponseError\n              ) {\n                input.messages = input.messages.slice();\n                const lastMessage = input.messages.at(-1);\n                if (\n                  lastMessage &&\n                  lastMessage instanceof AssistantMessage &&\n                  lastMessage.meta[\"tempMessage\"] &&\n                  lastMessage.text === \"\"\n                ) {\n                  input.messages.push(\n                    new UserMessage(\n                      \"No output received. Please regenerate your previous response.\",\n                      { tempMessage: true },\n                    ),\n                  );\n                } else {\n                  // Python compatibility\n                  input.messages.push(new AssistantMessage(\"\", { tempMessage: true }));\n                }\n              }\n            },\n          }).get();\n\n          input.messages = modelInputMessagesBackup;\n          await run.emitter.emit(\"success\", { value: result });\n          return result;\n        } catch (error) {\n          await run.emitter.emit(\"error\", { input, error });\n          await cacheEntry.reject(error);\n          if (error instanceof ChatModelError) {\n            throw error;\n          } else {\n            throw new ChatModelError(`The Chat Model has encountered an error.`, [error]);\n          }\n        } finally {\n          await run.emitter.emit(\"finish\", null);\n        }\n      },\n    ).middleware(...this.middlewares);\n  }\n\n  createStructure<T>(input: ChatModelObjectInput<T>) {\n    return RunContext.enter(\n      this,\n      { params: [input] as const, signal: input?.abortSignal },\n      async (run) => {\n        return await this._createStructure<T>(input, run);\n      },\n    );\n  }\n\n  config({ cache, parameters }: ChatConfig): void {\n    if (cache) {\n      this.cache = isFunction(cache) ? cache(this.cache) : cache;\n    }\n    if (parameters) {\n      this.parameters = isFunction(parameters) ? parameters(this.parameters) : parameters;\n    }\n  }\n\n  static async fromName(name: FullModelName | ProviderName, options?: ChatModelParameters) {\n    const { providerId, modelId } = parseModel(name);\n    const Target = await loadModel<ChatModel>(providerId, \"chat\");\n    const instance = new Target(modelId || undefined);\n    if (options) {\n      Object.assign(instance.parameters, options);\n    }\n    return instance;\n  }\n\n  protected abstract _create(\n    input: ChatModelInput,\n    run: GetRunContext<typeof this>,\n  ): Promise<ChatModelOutput>;\n  protected abstract _createStream(\n    input: ChatModelInput,\n    run: GetRunContext<typeof this>,\n  ): AsyncGenerator<ChatModelOutput, void>;\n\n  protected async _createStructure<T>(\n    input: ChatModelObjectInput<T>,\n    run: GetRunContext<typeof this>,\n  ): Promise<ChatModelObjectOutput<T>> {\n    const { schema, ...options } = input;\n    const jsonSchema = toJsonSchema(schema);\n\n    const systemTemplate =\n      input.systemPromptTemplate ??\n      new PromptTemplate({\n        schema: z.object({\n          schema: z.string().min(1),\n        }),\n        template: `You are a helpful assistant that generates only valid JSON adhering to the following JSON Schema.\n\n\\`\\`\\`\n{{schema}}\n\\`\\`\\`\n\nIMPORTANT: You MUST answer with a JSON object that matches the JSON schema above.`,\n      });\n\n    const messages: Message[] = [\n      new SystemMessage(systemTemplate.render({ schema: JSON.stringify(jsonSchema, null, 2) })),\n      ...input.messages,\n    ];\n\n    const errorTemplate = new PromptTemplate({\n      schema: z.object({\n        errors: z.string(),\n        expected: z.string(),\n        received: z.string(),\n      }),\n      template: `Generated object does not match the expected JSON schema!\n\nValidation Errors: {{errors}}`,\n    });\n\n    return new Retryable<ChatModelObjectOutput<T>>({\n      executor: async () => {\n        const response = await this._create(\n          {\n            ...options,\n            messages,\n            responseFormat: { type: \"object-json\" },\n          },\n          run,\n        );\n\n        const textResponse = response.getTextContent();\n        const object: T = parseBrokenJson(textResponse, { pair: [\"{\", \"}\"] });\n        const validator = createSchemaValidator(schema) as ValidateFunction<T>;\n\n        const success = validator(object);\n        if (!success) {\n          const context = {\n            expected: JSON.stringify(jsonSchema),\n            received: textResponse,\n            errors: JSON.stringify(validator.errors ?? []),\n          };\n\n          messages.push(new UserMessage(errorTemplate.render(context)));\n          throw new ChatModelError(`LLM did not produce a valid output.`, [], {\n            context,\n          });\n        }\n\n        return {\n          object,\n          output: response,\n        };\n      },\n      config: {\n        signal: run.signal,\n        maxRetries: input?.maxRetries || 1,\n      },\n    }).get();\n  }\n\n  createSnapshot() {\n    return {\n      cache: this.cache,\n      emitter: this.emitter,\n      middlewares: shallowCopy(this.middlewares) as MiddlewareType<any>[],\n      parameters: shallowCopy(this.parameters),\n      logger: this.logger,\n      toolChoiceSupport: this.toolChoiceSupport.slice(),\n      toolCallFallbackViaResponseFormat: this.toolCallFallbackViaResponseFormat,\n      modelSupportsToolCalling: this.modelSupportsToolCalling,\n      retryOnEmptyResponse: this.retryOnEmptyResponse,\n      fixInvalidToolCalls: this.fixInvalidToolCalls,\n    };\n  }\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  protected async createCacheAccessor({\n    abortSignal: _,\n    messages,\n    tools = [],\n    ...input\n  }: ChatModelInput) {\n    const key = ObjectHashKeyFn({\n      ...input,\n      messages: await Serializer.serialize(messages.map((msg) => msg.toPlain())),\n      tools: await Serializer.serialize(tools),\n    });\n    const value = await this.cache.get(key);\n    const isNew = value === undefined;\n\n    let task: Task<ChatModelOutput[]> | null = null;\n    if (isNew) {\n      task = new Task();\n      await this.cache.set(key, task);\n    }\n\n    return {\n      key,\n      value,\n      resolve: <T2 extends ChatModelOutput>(value: T2[]) => {\n        task?.resolve?.(value);\n      },\n      reject: async (error: Error) => {\n        task?.reject?.(error);\n        if (isNew) {\n          await this.cache.delete(key);\n        }\n      },\n    };\n  }\n\n  protected shouldForceToolCallViaResponseFormat({\n    tools = [],\n    toolChoice,\n    responseFormat,\n  }: ChatModelInput) {\n    if (\n      isEmpty(tools) ||\n      !toolChoice ||\n      toolChoice === \"none\" ||\n      toolChoice === \"auto\" ||\n      !this.toolCallFallbackViaResponseFormat ||\n      Boolean(responseFormat)\n    ) {\n      return false;\n    }\n\n    const toolChoiceSupported = this.isToolChoiceSupported(toolChoice);\n    return !this.modelSupportsToolCalling || !toolChoiceSupported;\n  }\n\n  protected isToolChoiceSupported(choice?: ChatModelToolChoice): boolean {\n    return (\n      !choice ||\n      (choice instanceof Tool\n        ? this.toolChoiceSupport.includes(\"single\")\n        : this.toolChoiceSupport.includes(choice))\n    );\n  }\n}\n\nexport class ChatModelOutput extends Serializable {\n  constructor(\n    public readonly messages: Message[],\n    public usage?: ChatModelUsage,\n    public finishReason?: ChatModelFinishReason,\n  ) {\n    super();\n    this.dedupe();\n  }\n\n  static fromChunks(chunks: ChatModelOutput[]) {\n    const final = new ChatModelOutput([]);\n    chunks.forEach((cur) => final.merge(cur));\n    return final;\n  }\n\n  isEmpty() {\n    if (this.messages.length === 0) {\n      return true;\n    }\n    return this.getTextContent() === \"\" && this.getToolCalls().length === 0;\n  }\n\n  merge(other: ChatModelOutput) {\n    if (other.messages.length > 0) {\n      const clones = other.messages.map(cloneSync);\n      this.messages.push(...clones);\n      this.dedupe();\n    }\n\n    this.finishReason = other.finishReason;\n    if (this.usage && other.usage) {\n      this.usage = customMerge([this.usage, other.usage], {\n        totalTokens: takeBigger,\n        promptTokens: takeBigger,\n        completionTokens: takeBigger,\n      });\n    } else if (other.usage) {\n      this.usage = shallowCopy(other.usage);\n    }\n  }\n\n  dedupe(): void {\n    // Dedupe messages\n    if (this.messages.length > 1) {\n      const messagesById = new Map<string, Message[]>();\n      const messagesByToolCallId = new Map<string, AssistantMessage>();\n\n      for (const msg of this.messages) {\n        const msgId = msg.id || \"\";\n\n        if (msg instanceof AssistantMessage && msg.getToolCalls().length > 0) {\n          const filteredChunks: AssistantMessage[\"content\"] = [];\n          for (const chunk of msg.content) {\n            if (chunk.type !== \"tool-call\") {\n              filteredChunks.push(chunk);\n              continue;\n            }\n\n            // Assume tool calls with no id refer to the most recent tool call\n            if (!chunk.toolCallId && messagesByToolCallId.size > 0) {\n              const lastToolCallId = getLast(messagesByToolCallId.keys(), \"\");\n              if (lastToolCallId) {\n                chunk.toolCallId = lastToolCallId;\n              }\n            }\n\n            if (chunk.toolCallId && messagesByToolCallId.has(chunk.toolCallId)) {\n              messagesByToolCallId.get(chunk.toolCallId)!.content.push(chunk);\n            } else if (chunk.toolCallId) {\n              messagesByToolCallId.set(chunk.toolCallId, msg);\n              filteredChunks.push(chunk);\n            }\n          }\n\n          msg.content.length = 0;\n          msg.content.push(...filteredChunks);\n\n          if (filteredChunks.length === 0) {\n            continue; // nothing to process\n          }\n        }\n\n        if (!messagesById.has(msgId)) {\n          messagesById.set(msgId, [msg]);\n        } else {\n          messagesById.get(msgId)!.push(msg);\n        }\n      }\n\n      this.messages.length = 0;\n\n      for (const messages of messagesById.values()) {\n        const main = messages.shift()!;\n        for (const other of messages) {\n          main.merge(other);\n        }\n        this.messages.push(main);\n      }\n    }\n\n    // Dedupe tool calls\n    for (const msg of this.messages) {\n      if (!(msg instanceof AssistantMessage)) {\n        continue;\n      }\n\n      const finalToolCalls: Record<string, ToolCallPart> = {};\n      let lastId = \"\";\n\n      const excludedIndexes: number[] = [];\n      msg.content.forEach((chunk, index) => {\n        if (chunk.type !== \"tool-call\") {\n          return;\n        }\n        const id = chunk.toolCallId || lastId;\n        if (!(id in finalToolCalls)) {\n          finalToolCalls[id] = shallowCopy(chunk);\n          msg.content[index] = finalToolCalls[id];\n        } else {\n          excludedIndexes.push(index);\n          const lastToolCall = finalToolCalls[id];\n          if (isString(lastToolCall.input) && isString(chunk.input)) {\n            lastToolCall.input += chunk.input;\n          } else {\n            throw new Error(\"Chunks cannot be merged.\");\n          }\n          if (!lastToolCall.toolName) {\n            lastToolCall.toolName = chunk.toolName;\n          }\n        }\n        lastId = id;\n      });\n      excludedIndexes.reverse().forEach((index) => {\n        msg.content.splice(index, 1);\n      });\n    }\n  }\n\n  getToolCalls() {\n    return this.messages\n      .filter((r) => r instanceof AssistantMessage)\n      .flatMap((r) => r.getToolCalls())\n      .filter(Boolean);\n  }\n\n  getTextMessages(): AssistantMessage[] {\n    return this.messages.filter((r) => r instanceof AssistantMessage).filter((r) => r.text);\n  }\n\n  getTextContent(): string {\n    return this.messages\n      .filter((r) => r instanceof AssistantMessage)\n      .flatMap((r) => r.text)\n      .filter(Boolean)\n      .join(\"\");\n  }\n\n  toString() {\n    return this.getTextContent();\n  }\n\n  createSnapshot() {\n    return {\n      messages: shallowCopy(this.messages),\n      usage: shallowCopy(this.usage),\n      finishReason: this.finishReason,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n\nfunction cloneSync<T extends Serializable>(serializable: T): T {\n  const snapshot = serializable.createSnapshot();\n  if (isPromise(snapshot)) {\n    throw new Error(`createSnapshot cannot be async`);\n  }\n\n  const target = Object.create(serializable.constructor.prototype) as T;\n  const load = target.loadSnapshot(snapshot);\n  if (isPromise(load)) {\n    throw new Error(`loadSnapshot cannot be async`);\n  }\n\n  return target;\n}\n"]}

package/dist/backend/chat.d.cts

@@ -1,11 +1,11 @@
 import './message.cjs';
 import '../context.cjs';
 import '../internals/serializable.cjs';
-import '../emitter-jN55XZZq.cjs';
+import '../emitter-UkIkNrek.cjs';
 import '../errors.cjs';
 import 'promise-based-task';
 import '../cache/base.cjs';
-export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-BOb9n7AJ.cjs';
+export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-DaqV3woF.cjs';
 import './constants.cjs';
 import '../tools/base.cjs';
 import 'zod';