beeai-framework 0.1.28 → 0.1.30

package/dist/agents/base.cjs.map

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

package/dist/agents/base.d.cts

@@ -1,8 +1,8 @@
 import '../context.cjs';
 import '../errors.cjs';
-export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-BqzN2QwB.cjs';
+export { A as AgentError, B as BaseAgent, a as BaseAgentRunOptions } from '../base-s3ioVIE9.cjs';
 import '../internals/serializable.cjs';
-import '../emitter-UkIkNrek.cjs';
+import '../emitter-D4OcelZ9.cjs';
 import '../memory/base.cjs';
 import '../internals/helpers/promise.cjs';
 import '../internals/types.cjs';

package/dist/agents/base.d.ts

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

package/dist/agents/base.js.map

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

package/dist/agents/experimental/replan/agent.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/experimental/replan/agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","_run","_options","context","prompt","memory","add","UserMessage","runner","createRunner","finalMessage","undefined","state","run","nextStep","type","message","toolResults","tools","calls","AssistantMessage","RePlanAssistantPrompt","render","results","JSON","stringify","intermediateMemory","UnconstrainedMemory","addMany","messages","schema","createRePlanOutputSchema","response","llm","createStructure","definition","abortSignal","signal","emit","object","Promise","all","map","call","tool","find","name","AgentError","meta","output","Tool","contextKeys","Memory","error"],"mappings":";;;;;;;;;;;AAiDO,MAAMA,oBAAoBC,kBAAAA,CAAAA;EAjDjC;;;;EAkDSC,OAAAA,GAAUC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAoB;IAChDC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;EAEA,MAAgBC,IAAAA,CACdD,KAAAA,EACAE,QAAAA,EACAC,OAAAA,EAC0B;AAC1B,IAAA,IAAIH,KAAAA,CAAMI,WAAW,IAAA,EAAM;AACzB,MAAA,MAAM,KAAKC,MAAAA,CAAOC,GAAAA,CAAI,IAAIC,uBAAAA,CAAYP,KAAAA,CAAMI,MAAM,CAAA,CAAA;AACpD;AAEA,IAAA,MAAMI,MAAAA,GAAS,MAAM,IAAA,CAAKC,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAAA,GAAoCC,MAAAA;AACxC,IAAA,OAAO,CAACD,YAAAA,EAAc;AACpB,MAAA,MAAME,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAAA,EAAG;AAE9B,MAAA,IAAID,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,SAAA,EAAW;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,uBAAAA,CAAYK,KAAAA,CAAME,QAAAA,CAASE,OAAO,CAAA;OACvD,MAAA,IAAWJ,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,MAAA,EAAQ;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAAA,CAAMN,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAAA,CAAOC,GAAAA,CAClB,IAAIc,4BAAAA,CACFC,kCAAsBC,MAAAA,CAAO;UAC3BC,OAAAA,EAASC,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN;AACF;AAEA,IAAA,MAAM,IAAA,CAAKZ,MAAAA,CAAOC,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAA,OAAO;MACLM,OAAAA,EAASN,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAAA,CAAOH;AAC7B,KAAA;AACF;AAEA,EAAA,MAAgBI,aAAaN,OAAAA,EAA8B;AACzD,IAAA,MAAME,MAAAA,GAAS,IAAIsB,2CAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAAA,CAAOuB,OAAAA,CAAQ,IAAA,CAAKvB,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAAA,GAAS,MAAMC,oCAAAA,CAAyB,IAAA,CAAK/B,MAAMkB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAAA,GAAW,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,eAAAA,CAAgB;AACpDJ,QAAAA,MAAAA,EAAQA,MAAAA,CAAOK,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAAA,CAAQkC,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAAA,CAAOwB;OACnB,CAAA;AACA,MAAA,MAAMxB,MAAAA,CAAOC,IAAI,IAAIc,4BAAAA,CAAiBI,KAAKC,SAAAA,CAAUO,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAA,MAAM7B,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,QAAA,EAAU;AAAE1B,QAAAA,KAAAA,EAAOoB,QAAAA,CAASO;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAAAA,CAASO,MAAAA;KAClB,EAVY,KAAA,CAAA;AAYZ,IAAA,MAAMrB,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAAA,CAAQC,GAAAA,CACnBtB,KAAAA,CAAMuB,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAA,MAAMC,IAAAA,GAAO,IAAA,CAAK5C,KAAAA,CAAMkB,KAAAA,CAAM2B,IAAAA,CAAK,CAACD,KAAAA,KAASA,KAAAA,CAAKE,IAAAA,KAASH,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAAA,EAAM;AACT,UAAA,MAAM,IAAIG,mBAAAA,CAAW,CAAA,KAAA,EAAQJ,IAAAA,CAAKG,IAAI,CAAA,gBAAA,CAAkB,CAAA;AAC1D;AAEA,QAAA,MAAME,IAAAA,GAAO;UAAEhD,KAAAA,EAAO2C,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAA,MAAMhB,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;UAAEvB,IAAAA,EAAM,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAA,IAAI;AACF,UAAA,MAAMC,MAAAA,GAAS,MAAML,IAAAA,CAAK/B,GAAAA,CAAI8B,KAAK3C,KAAAA,EAAO;AAAEqC,YAAAA,MAAAA,EAAQlC,OAAAA,CAAQkC;AAAO,WAAA,EAAGlC,OAAAA,CAAQ;YAC5E,CAAC+C,eAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAA,MAAMF,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAA,OAAOA,MAAAA;AACT,SAAA,CAAA,OAASI,KAAAA,EAAO;AACd,UAAA,MAAMlD,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAA,MAAMA,KAAAA;AACR;AACF,OAAA,CAAA,CAAA;KAEJ,EAtBc,OAAA,CAAA;AAwBd,IAAA,OAAO;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF;AAEA,EAAA,IAAIb,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB;AAEA,EAAA,IAAIA,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport {\n  createRePlanOutputSchema,\n  RePlanState,\n  RePlanAssistantPrompt,\n} from \"@/agents/experimental/replan/prompts.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface RePlanRunInput {\n  prompt: string | null;\n}\n\nexport interface RePlanRunOutput {\n  message: Message;\n  intermediateMemory: BaseMemory;\n}\n\nexport interface RePlanToolCall {\n  name: string;\n  input: any;\n}\n\nexport interface RePlanEvents {\n  update: Callback<{ state: RePlanState }>;\n  tool: Callback<\n    | { type: \"start\"; tool: AnyTool; input: any; calls: RePlanToolCall[] }\n    | { type: \"success\"; tool: AnyTool; input: any; output: any; calls: RePlanToolCall[] }\n    | { type: \"error\"; tool: AnyTool; input: any; error: Error; calls: RePlanToolCall[] }\n  >;\n}\n\ninterface Input {\n  memory: BaseMemory;\n  tools: AnyTool[];\n  llm: ChatModel;\n}\n\nexport class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {\n  public emitter = Emitter.root.child<RePlanEvents>({\n    namespace: [\"agent\", \"rePlan\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RePlanRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RePlanRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(input.prompt));\n    }\n\n    const runner = await this.createRunner(context);\n\n    let finalMessage: Message | undefined = undefined;\n    while (!finalMessage) {\n      const state = await runner.run();\n\n      if (state.nextStep.type === \"message\") {\n        finalMessage = new UserMessage(state.nextStep.message);\n      } else if (state.nextStep.type === \"tool\") {\n        const toolResults = await runner.tools(state.nextStep.calls);\n        await runner.memory.add(\n          new AssistantMessage(\n            RePlanAssistantPrompt.render({\n              results: JSON.stringify(toolResults),\n            }),\n          ),\n        );\n      }\n    }\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n      intermediateMemory: runner.memory,\n    };\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const memory = new UnconstrainedMemory();\n    await memory.addMany(this.memory.messages);\n\n    const run = async (): Promise<RePlanState> => {\n      const schema = await createRePlanOutputSchema(this.input.tools);\n      const response = await this.input.llm.createStructure({\n        schema: schema.definition,\n        abortSignal: context.signal,\n        messages: memory.messages,\n      });\n      await memory.add(new AssistantMessage(JSON.stringify(response)));\n      await context.emitter.emit(\"update\", { state: response.object });\n      return response.object;\n    };\n\n    const tools = async (calls: RePlanToolCall[]) => {\n      return await Promise.all(\n        calls.map(async (call) => {\n          const tool = this.input.tools.find((tool) => tool.name === call.name);\n          if (!tool) {\n            throw new AgentError(`Tool ${call.name} does not exist.`);\n          }\n\n          const meta = { input: call, tool, calls };\n          await context.emitter.emit(\"tool\", { type: \"start\", ...meta });\n          try {\n            const output = await tool.run(call.input, { signal: context.signal }).context({\n              [Tool.contextKeys.Memory]: memory,\n            });\n            await context.emitter.emit(\"tool\", { type: \"success\", ...meta, output });\n            return output;\n          } catch (error) {\n            await context.emitter.emit(\"tool\", { type: \"error\", ...meta, error });\n            throw error;\n          }\n        }),\n      );\n    };\n\n    return {\n      memory,\n      run,\n      tools,\n    };\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/experimental/replan/agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","_run","_options","context","prompt","memory","add","UserMessage","runner","createRunner","finalMessage","undefined","state","run","nextStep","type","message","toolResults","tools","calls","AssistantMessage","RePlanAssistantPrompt","render","results","JSON","stringify","intermediateMemory","UnconstrainedMemory","addMany","messages","schema","createRePlanOutputSchema","response","llm","createStructure","definition","abortSignal","signal","emit","object","Promise","all","map","call","tool","find","name","AgentError","meta","output","Tool","contextKeys","Memory","error"],"mappings":";;;;;;;;;;;AAiDO,MAAMA,oBAAoBC,kBAAAA,CAAAA;EAjDjC;;;;EAkDSC,OAAAA,GAAUC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAoB;IAChDC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B,EAAA;EAEA,MAAgBC,IAAAA,CACdD,KAAAA,EACAE,QAAAA,EACAC,OAAAA,EAC0B;AAC1B,IAAA,IAAIH,KAAAA,CAAMI,WAAW,IAAA,EAAM;AACzB,MAAA,MAAM,KAAKC,MAAAA,CAAOC,GAAAA,CAAI,IAAIC,uBAAAA,CAAYP,KAAAA,CAAMI,MAAM,CAAA,CAAA;AACpD,IAAA;AAEA,IAAA,MAAMI,MAAAA,GAAS,MAAM,IAAA,CAAKC,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAAA,GAAoCC,MAAAA;AACxC,IAAA,OAAO,CAACD,YAAAA,EAAc;AACpB,MAAA,MAAME,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAAA,EAAG;AAE9B,MAAA,IAAID,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,SAAA,EAAW;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,uBAAAA,CAAYK,KAAAA,CAAME,QAAAA,CAASE,OAAO,CAAA;MACvD,CAAA,MAAA,IAAWJ,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,MAAA,EAAQ;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAAA,CAAMN,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAAA,CAAOC,GAAAA,CAClB,IAAIc,4BAAAA,CACFC,kCAAsBC,MAAAA,CAAO;UAC3BC,OAAAA,EAASC,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN,MAAA;AACF,IAAA;AAEA,IAAA,MAAM,IAAA,CAAKZ,MAAAA,CAAOC,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAA,OAAO;MACLM,OAAAA,EAASN,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAAA,CAAOH;AAC7B,KAAA;AACF,EAAA;AAEA,EAAA,MAAgBI,aAAaN,OAAAA,EAA8B;AACzD,IAAA,MAAME,MAAAA,GAAS,IAAIsB,2CAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAAA,CAAOuB,OAAAA,CAAQ,IAAA,CAAKvB,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAAA,GAAS,MAAMC,oCAAAA,CAAyB,IAAA,CAAK/B,MAAMkB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAAA,GAAW,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,eAAAA,CAAgB;AACpDJ,QAAAA,MAAAA,EAAQA,MAAAA,CAAOK,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAAA,CAAQkC,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAAA,CAAOwB;OACnB,CAAA;AACA,MAAA,MAAMxB,MAAAA,CAAOC,IAAI,IAAIc,4BAAAA,CAAiBI,KAAKC,SAAAA,CAAUO,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAA,MAAM7B,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,QAAA,EAAU;AAAE1B,QAAAA,KAAAA,EAAOoB,QAAAA,CAASO;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAAAA,CAASO,MAAAA;IAClB,CAAA,EAVY,KAAA,CAAA;AAYZ,IAAA,MAAMrB,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAAA,CAAQC,GAAAA,CACnBtB,KAAAA,CAAMuB,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAA,MAAMC,IAAAA,GAAO,IAAA,CAAK5C,KAAAA,CAAMkB,KAAAA,CAAM2B,IAAAA,CAAK,CAACD,KAAAA,KAASA,KAAAA,CAAKE,IAAAA,KAASH,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAAA,EAAM;AACT,UAAA,MAAM,IAAIG,mBAAAA,CAAW,CAAA,KAAA,EAAQJ,IAAAA,CAAKG,IAAI,CAAA,gBAAA,CAAkB,CAAA;AAC1D,QAAA;AAEA,QAAA,MAAME,IAAAA,GAAO;UAAEhD,KAAAA,EAAO2C,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAA,MAAMhB,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;UAAEvB,IAAAA,EAAM,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAA,IAAI;AACF,UAAA,MAAMC,MAAAA,GAAS,MAAML,IAAAA,CAAK/B,GAAAA,CAAI8B,KAAK3C,KAAAA,EAAO;AAAEqC,YAAAA,MAAAA,EAAQlC,OAAAA,CAAQkC;AAAO,WAAA,EAAGlC,OAAAA,CAAQ;YAC5E,CAAC+C,eAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAA,MAAMF,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAA,OAAOA,MAAAA;AACT,QAAA,CAAA,CAAA,OAASI,KAAAA,EAAO;AACd,UAAA,MAAMlD,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAA,MAAMA,KAAAA;AACR,QAAA;AACF,MAAA,CAAA,CAAA,CAAA;IAEJ,CAAA,EAtBc,OAAA,CAAA;AAwBd,IAAA,OAAO;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF,EAAA;AAEA,EAAA,IAAIb,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB,EAAA;AAEA,EAAA,IAAIA,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB,EAAA;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport {\n  createRePlanOutputSchema,\n  RePlanState,\n  RePlanAssistantPrompt,\n} from \"@/agents/experimental/replan/prompts.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface RePlanRunInput {\n  prompt: string | null;\n}\n\nexport interface RePlanRunOutput {\n  message: Message;\n  intermediateMemory: BaseMemory;\n}\n\nexport interface RePlanToolCall {\n  name: string;\n  input: any;\n}\n\nexport interface RePlanEvents {\n  update: Callback<{ state: RePlanState }>;\n  tool: Callback<\n    | { type: \"start\"; tool: AnyTool; input: any; calls: RePlanToolCall[] }\n    | { type: \"success\"; tool: AnyTool; input: any; output: any; calls: RePlanToolCall[] }\n    | { type: \"error\"; tool: AnyTool; input: any; error: Error; calls: RePlanToolCall[] }\n  >;\n}\n\ninterface Input {\n  memory: BaseMemory;\n  tools: AnyTool[];\n  llm: ChatModel;\n}\n\nexport class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {\n  public emitter = Emitter.root.child<RePlanEvents>({\n    namespace: [\"agent\", \"rePlan\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RePlanRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RePlanRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(input.prompt));\n    }\n\n    const runner = await this.createRunner(context);\n\n    let finalMessage: Message | undefined = undefined;\n    while (!finalMessage) {\n      const state = await runner.run();\n\n      if (state.nextStep.type === \"message\") {\n        finalMessage = new UserMessage(state.nextStep.message);\n      } else if (state.nextStep.type === \"tool\") {\n        const toolResults = await runner.tools(state.nextStep.calls);\n        await runner.memory.add(\n          new AssistantMessage(\n            RePlanAssistantPrompt.render({\n              results: JSON.stringify(toolResults),\n            }),\n          ),\n        );\n      }\n    }\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n      intermediateMemory: runner.memory,\n    };\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const memory = new UnconstrainedMemory();\n    await memory.addMany(this.memory.messages);\n\n    const run = async (): Promise<RePlanState> => {\n      const schema = await createRePlanOutputSchema(this.input.tools);\n      const response = await this.input.llm.createStructure({\n        schema: schema.definition,\n        abortSignal: context.signal,\n        messages: memory.messages,\n      });\n      await memory.add(new AssistantMessage(JSON.stringify(response)));\n      await context.emitter.emit(\"update\", { state: response.object });\n      return response.object;\n    };\n\n    const tools = async (calls: RePlanToolCall[]) => {\n      return await Promise.all(\n        calls.map(async (call) => {\n          const tool = this.input.tools.find((tool) => tool.name === call.name);\n          if (!tool) {\n            throw new AgentError(`Tool ${call.name} does not exist.`);\n          }\n\n          const meta = { input: call, tool, calls };\n          await context.emitter.emit(\"tool\", { type: \"start\", ...meta });\n          try {\n            const output = await tool.run(call.input, { signal: context.signal }).context({\n              [Tool.contextKeys.Memory]: memory,\n            });\n            await context.emitter.emit(\"tool\", { type: \"success\", ...meta, output });\n            return output;\n          } catch (error) {\n            await context.emitter.emit(\"tool\", { type: \"error\", ...meta, error });\n            throw error;\n          }\n        }),\n      );\n    };\n\n    return {\n      memory,\n      run,\n      tools,\n    };\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n}\n"]}

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

@@ -1,12 +1,12 @@
-import { C as Callback, E as Emitter } from '../../../emitter-UkIkNrek.cjs';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BqzN2QwB.cjs';
+import { E as Emitter, C as Callback } from '../../../emitter-D4OcelZ9.cjs';
+import { B as BaseAgent, a as BaseAgentRunOptions } from '../../../base-s3ioVIE9.cjs';
 import { GetRunContext } from '../../../context.cjs';
 import { Message } from '../../../backend/message.cjs';
 import { RePlanState } from './prompts.cjs';
 import { BaseMemory } from '../../../memory/base.cjs';
 import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.cjs';
 import { AnyTool } from '../../../tools/base.cjs';
-import { m as ChatModel } from '../../../chat-DaqV3woF.cjs';
+import { a as ChatModel } from '../../../chat-C20nYj46.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';

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

@@ -1,12 +1,12 @@
-import { C as Callback, E as Emitter } from '../../../emitter-CL56qab4.js';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BfhNzUYf.js';
+import { E as Emitter, C as Callback } from '../../../emitter-kHxkUxco.js';
+import { B as BaseAgent, a as BaseAgentRunOptions } from '../../../base-BjzyQ8k6.js';
 import { GetRunContext } from '../../../context.js';
 import { Message } from '../../../backend/message.js';
 import { RePlanState } from './prompts.js';
 import { BaseMemory } from '../../../memory/base.js';
 import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.js';
 import { AnyTool } from '../../../tools/base.js';
-import { m as ChatModel } from '../../../chat-DTxmaKP0.js';
+import { a as ChatModel } from '../../../chat-CfTNfR60.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';

package/dist/agents/experimental/replan/agent.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/experimental/replan/agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","_run","_options","context","prompt","memory","add","UserMessage","runner","createRunner","finalMessage","undefined","state","run","nextStep","type","message","toolResults","tools","calls","AssistantMessage","RePlanAssistantPrompt","render","results","JSON","stringify","intermediateMemory","UnconstrainedMemory","addMany","messages","schema","createRePlanOutputSchema","response","llm","createStructure","definition","abortSignal","signal","emit","object","Promise","all","map","call","tool","find","name","AgentError","meta","output","Tool","contextKeys","Memory","error"],"mappings":";;;;;;;;;AAiDO,MAAMA,oBAAoBC,SAAAA,CAAAA;EAjDjC;;;;EAkDSC,OAAAA,GAAUC,OAAAA,CAAQC,KAAKC,KAAAA,CAAoB;IAChDC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;EAEA,MAAgBC,IAAAA,CACdD,KAAAA,EACAE,QAAAA,EACAC,OAAAA,EAC0B;AAC1B,IAAA,IAAIH,KAAAA,CAAMI,WAAW,IAAA,EAAM;AACzB,MAAA,MAAM,KAAKC,MAAAA,CAAOC,GAAAA,CAAI,IAAIC,WAAAA,CAAYP,KAAAA,CAAMI,MAAM,CAAA,CAAA;AACpD;AAEA,IAAA,MAAMI,MAAAA,GAAS,MAAM,IAAA,CAAKC,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAAA,GAAoCC,MAAAA;AACxC,IAAA,OAAO,CAACD,YAAAA,EAAc;AACpB,MAAA,MAAME,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAAA,EAAG;AAE9B,MAAA,IAAID,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,SAAA,EAAW;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,WAAAA,CAAYK,KAAAA,CAAME,QAAAA,CAASE,OAAO,CAAA;OACvD,MAAA,IAAWJ,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,MAAA,EAAQ;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAAA,CAAMN,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAAA,CAAOC,GAAAA,CAClB,IAAIc,gBAAAA,CACFC,sBAAsBC,MAAAA,CAAO;UAC3BC,OAAAA,EAASC,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN;AACF;AAEA,IAAA,MAAM,IAAA,CAAKZ,MAAAA,CAAOC,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAA,OAAO;MACLM,OAAAA,EAASN,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAAA,CAAOH;AAC7B,KAAA;AACF;AAEA,EAAA,MAAgBI,aAAaN,OAAAA,EAA8B;AACzD,IAAA,MAAME,MAAAA,GAAS,IAAIsB,mBAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAAA,CAAOuB,OAAAA,CAAQ,IAAA,CAAKvB,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAAA,GAAS,MAAMC,wBAAAA,CAAyB,IAAA,CAAK/B,MAAMkB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAAA,GAAW,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,eAAAA,CAAgB;AACpDJ,QAAAA,MAAAA,EAAQA,MAAAA,CAAOK,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAAA,CAAQkC,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAAA,CAAOwB;OACnB,CAAA;AACA,MAAA,MAAMxB,MAAAA,CAAOC,IAAI,IAAIc,gBAAAA,CAAiBI,KAAKC,SAAAA,CAAUO,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAA,MAAM7B,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,QAAA,EAAU;AAAE1B,QAAAA,KAAAA,EAAOoB,QAAAA,CAASO;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAAAA,CAASO,MAAAA;KAClB,EAVY,KAAA,CAAA;AAYZ,IAAA,MAAMrB,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAAA,CAAQC,GAAAA,CACnBtB,KAAAA,CAAMuB,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAA,MAAMC,IAAAA,GAAO,IAAA,CAAK5C,KAAAA,CAAMkB,KAAAA,CAAM2B,IAAAA,CAAK,CAACD,KAAAA,KAASA,KAAAA,CAAKE,IAAAA,KAASH,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAAA,EAAM;AACT,UAAA,MAAM,IAAIG,UAAAA,CAAW,CAAA,KAAA,EAAQJ,IAAAA,CAAKG,IAAI,CAAA,gBAAA,CAAkB,CAAA;AAC1D;AAEA,QAAA,MAAME,IAAAA,GAAO;UAAEhD,KAAAA,EAAO2C,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAA,MAAMhB,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;UAAEvB,IAAAA,EAAM,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAA,IAAI;AACF,UAAA,MAAMC,MAAAA,GAAS,MAAML,IAAAA,CAAK/B,GAAAA,CAAI8B,KAAK3C,KAAAA,EAAO;AAAEqC,YAAAA,MAAAA,EAAQlC,OAAAA,CAAQkC;AAAO,WAAA,EAAGlC,OAAAA,CAAQ;YAC5E,CAAC+C,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAA,MAAMF,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAA,OAAOA,MAAAA;AACT,SAAA,CAAA,OAASI,KAAAA,EAAO;AACd,UAAA,MAAMlD,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAA,MAAMA,KAAAA;AACR;AACF,OAAA,CAAA,CAAA;KAEJ,EAtBc,OAAA,CAAA;AAwBd,IAAA,OAAO;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF;AAEA,EAAA,IAAIb,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB;AAEA,EAAA,IAAIA,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport {\n  createRePlanOutputSchema,\n  RePlanState,\n  RePlanAssistantPrompt,\n} from \"@/agents/experimental/replan/prompts.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface RePlanRunInput {\n  prompt: string | null;\n}\n\nexport interface RePlanRunOutput {\n  message: Message;\n  intermediateMemory: BaseMemory;\n}\n\nexport interface RePlanToolCall {\n  name: string;\n  input: any;\n}\n\nexport interface RePlanEvents {\n  update: Callback<{ state: RePlanState }>;\n  tool: Callback<\n    | { type: \"start\"; tool: AnyTool; input: any; calls: RePlanToolCall[] }\n    | { type: \"success\"; tool: AnyTool; input: any; output: any; calls: RePlanToolCall[] }\n    | { type: \"error\"; tool: AnyTool; input: any; error: Error; calls: RePlanToolCall[] }\n  >;\n}\n\ninterface Input {\n  memory: BaseMemory;\n  tools: AnyTool[];\n  llm: ChatModel;\n}\n\nexport class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {\n  public emitter = Emitter.root.child<RePlanEvents>({\n    namespace: [\"agent\", \"rePlan\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RePlanRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RePlanRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(input.prompt));\n    }\n\n    const runner = await this.createRunner(context);\n\n    let finalMessage: Message | undefined = undefined;\n    while (!finalMessage) {\n      const state = await runner.run();\n\n      if (state.nextStep.type === \"message\") {\n        finalMessage = new UserMessage(state.nextStep.message);\n      } else if (state.nextStep.type === \"tool\") {\n        const toolResults = await runner.tools(state.nextStep.calls);\n        await runner.memory.add(\n          new AssistantMessage(\n            RePlanAssistantPrompt.render({\n              results: JSON.stringify(toolResults),\n            }),\n          ),\n        );\n      }\n    }\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n      intermediateMemory: runner.memory,\n    };\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const memory = new UnconstrainedMemory();\n    await memory.addMany(this.memory.messages);\n\n    const run = async (): Promise<RePlanState> => {\n      const schema = await createRePlanOutputSchema(this.input.tools);\n      const response = await this.input.llm.createStructure({\n        schema: schema.definition,\n        abortSignal: context.signal,\n        messages: memory.messages,\n      });\n      await memory.add(new AssistantMessage(JSON.stringify(response)));\n      await context.emitter.emit(\"update\", { state: response.object });\n      return response.object;\n    };\n\n    const tools = async (calls: RePlanToolCall[]) => {\n      return await Promise.all(\n        calls.map(async (call) => {\n          const tool = this.input.tools.find((tool) => tool.name === call.name);\n          if (!tool) {\n            throw new AgentError(`Tool ${call.name} does not exist.`);\n          }\n\n          const meta = { input: call, tool, calls };\n          await context.emitter.emit(\"tool\", { type: \"start\", ...meta });\n          try {\n            const output = await tool.run(call.input, { signal: context.signal }).context({\n              [Tool.contextKeys.Memory]: memory,\n            });\n            await context.emitter.emit(\"tool\", { type: \"success\", ...meta, output });\n            return output;\n          } catch (error) {\n            await context.emitter.emit(\"tool\", { type: \"error\", ...meta, error });\n            throw error;\n          }\n        }),\n      );\n    };\n\n    return {\n      memory,\n      run,\n      tools,\n    };\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/experimental/replan/agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","_run","_options","context","prompt","memory","add","UserMessage","runner","createRunner","finalMessage","undefined","state","run","nextStep","type","message","toolResults","tools","calls","AssistantMessage","RePlanAssistantPrompt","render","results","JSON","stringify","intermediateMemory","UnconstrainedMemory","addMany","messages","schema","createRePlanOutputSchema","response","llm","createStructure","definition","abortSignal","signal","emit","object","Promise","all","map","call","tool","find","name","AgentError","meta","output","Tool","contextKeys","Memory","error"],"mappings":";;;;;;;;;AAiDO,MAAMA,oBAAoBC,SAAAA,CAAAA;EAjDjC;;;;EAkDSC,OAAAA,GAAUC,OAAAA,CAAQC,KAAKC,KAAAA,CAAoB;IAChDC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B,EAAA;EAEA,MAAgBC,IAAAA,CACdD,KAAAA,EACAE,QAAAA,EACAC,OAAAA,EAC0B;AAC1B,IAAA,IAAIH,KAAAA,CAAMI,WAAW,IAAA,EAAM;AACzB,MAAA,MAAM,KAAKC,MAAAA,CAAOC,GAAAA,CAAI,IAAIC,WAAAA,CAAYP,KAAAA,CAAMI,MAAM,CAAA,CAAA;AACpD,IAAA;AAEA,IAAA,MAAMI,MAAAA,GAAS,MAAM,IAAA,CAAKC,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAAA,GAAoCC,MAAAA;AACxC,IAAA,OAAO,CAACD,YAAAA,EAAc;AACpB,MAAA,MAAME,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAAA,EAAG;AAE9B,MAAA,IAAID,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,SAAA,EAAW;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,WAAAA,CAAYK,KAAAA,CAAME,QAAAA,CAASE,OAAO,CAAA;MACvD,CAAA,MAAA,IAAWJ,KAAAA,CAAME,QAAAA,CAASC,IAAAA,KAAS,MAAA,EAAQ;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAAA,CAAMN,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAAA,CAAOC,GAAAA,CAClB,IAAIc,gBAAAA,CACFC,sBAAsBC,MAAAA,CAAO;UAC3BC,OAAAA,EAASC,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN,MAAA;AACF,IAAA;AAEA,IAAA,MAAM,IAAA,CAAKZ,MAAAA,CAAOC,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAA,OAAO;MACLM,OAAAA,EAASN,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAAA,CAAOH;AAC7B,KAAA;AACF,EAAA;AAEA,EAAA,MAAgBI,aAAaN,OAAAA,EAA8B;AACzD,IAAA,MAAME,MAAAA,GAAS,IAAIsB,mBAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAAA,CAAOuB,OAAAA,CAAQ,IAAA,CAAKvB,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAAA,GAAS,MAAMC,wBAAAA,CAAyB,IAAA,CAAK/B,MAAMkB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAAA,GAAW,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,eAAAA,CAAgB;AACpDJ,QAAAA,MAAAA,EAAQA,MAAAA,CAAOK,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAAA,CAAQkC,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAAA,CAAOwB;OACnB,CAAA;AACA,MAAA,MAAMxB,MAAAA,CAAOC,IAAI,IAAIc,gBAAAA,CAAiBI,KAAKC,SAAAA,CAAUO,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAA,MAAM7B,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,QAAA,EAAU;AAAE1B,QAAAA,KAAAA,EAAOoB,QAAAA,CAASO;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAAAA,CAASO,MAAAA;IAClB,CAAA,EAVY,KAAA,CAAA;AAYZ,IAAA,MAAMrB,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAAA,CAAQC,GAAAA,CACnBtB,KAAAA,CAAMuB,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAA,MAAMC,IAAAA,GAAO,IAAA,CAAK5C,KAAAA,CAAMkB,KAAAA,CAAM2B,IAAAA,CAAK,CAACD,KAAAA,KAASA,KAAAA,CAAKE,IAAAA,KAASH,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAAA,EAAM;AACT,UAAA,MAAM,IAAIG,UAAAA,CAAW,CAAA,KAAA,EAAQJ,IAAAA,CAAKG,IAAI,CAAA,gBAAA,CAAkB,CAAA;AAC1D,QAAA;AAEA,QAAA,MAAME,IAAAA,GAAO;UAAEhD,KAAAA,EAAO2C,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAA,MAAMhB,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;UAAEvB,IAAAA,EAAM,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAA,IAAI;AACF,UAAA,MAAMC,MAAAA,GAAS,MAAML,IAAAA,CAAK/B,GAAAA,CAAI8B,KAAK3C,KAAAA,EAAO;AAAEqC,YAAAA,MAAAA,EAAQlC,OAAAA,CAAQkC;AAAO,WAAA,EAAGlC,OAAAA,CAAQ;YAC5E,CAAC+C,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAA,MAAMF,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAA,OAAOA,MAAAA;AACT,QAAA,CAAA,CAAA,OAASI,KAAAA,EAAO;AACd,UAAA,MAAMlD,OAAAA,CAAQT,OAAAA,CAAQ4C,IAAAA,CAAK,MAAA,EAAQ;YAAEvB,IAAAA,EAAM,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAA,MAAMA,KAAAA;AACR,QAAA;AACF,MAAA,CAAA,CAAA,CAAA;IAEJ,CAAA,EAtBc,OAAA,CAAA;AAwBd,IAAA,OAAO;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF,EAAA;AAEA,EAAA,IAAIb,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB,EAAA;AAEA,EAAA,IAAIA,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB,EAAA;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport {\n  createRePlanOutputSchema,\n  RePlanState,\n  RePlanAssistantPrompt,\n} from \"@/agents/experimental/replan/prompts.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface RePlanRunInput {\n  prompt: string | null;\n}\n\nexport interface RePlanRunOutput {\n  message: Message;\n  intermediateMemory: BaseMemory;\n}\n\nexport interface RePlanToolCall {\n  name: string;\n  input: any;\n}\n\nexport interface RePlanEvents {\n  update: Callback<{ state: RePlanState }>;\n  tool: Callback<\n    | { type: \"start\"; tool: AnyTool; input: any; calls: RePlanToolCall[] }\n    | { type: \"success\"; tool: AnyTool; input: any; output: any; calls: RePlanToolCall[] }\n    | { type: \"error\"; tool: AnyTool; input: any; error: Error; calls: RePlanToolCall[] }\n  >;\n}\n\ninterface Input {\n  memory: BaseMemory;\n  tools: AnyTool[];\n  llm: ChatModel;\n}\n\nexport class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {\n  public emitter = Emitter.root.child<RePlanEvents>({\n    namespace: [\"agent\", \"rePlan\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RePlanRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RePlanRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(input.prompt));\n    }\n\n    const runner = await this.createRunner(context);\n\n    let finalMessage: Message | undefined = undefined;\n    while (!finalMessage) {\n      const state = await runner.run();\n\n      if (state.nextStep.type === \"message\") {\n        finalMessage = new UserMessage(state.nextStep.message);\n      } else if (state.nextStep.type === \"tool\") {\n        const toolResults = await runner.tools(state.nextStep.calls);\n        await runner.memory.add(\n          new AssistantMessage(\n            RePlanAssistantPrompt.render({\n              results: JSON.stringify(toolResults),\n            }),\n          ),\n        );\n      }\n    }\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n      intermediateMemory: runner.memory,\n    };\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const memory = new UnconstrainedMemory();\n    await memory.addMany(this.memory.messages);\n\n    const run = async (): Promise<RePlanState> => {\n      const schema = await createRePlanOutputSchema(this.input.tools);\n      const response = await this.input.llm.createStructure({\n        schema: schema.definition,\n        abortSignal: context.signal,\n        messages: memory.messages,\n      });\n      await memory.add(new AssistantMessage(JSON.stringify(response)));\n      await context.emitter.emit(\"update\", { state: response.object });\n      return response.object;\n    };\n\n    const tools = async (calls: RePlanToolCall[]) => {\n      return await Promise.all(\n        calls.map(async (call) => {\n          const tool = this.input.tools.find((tool) => tool.name === call.name);\n          if (!tool) {\n            throw new AgentError(`Tool ${call.name} does not exist.`);\n          }\n\n          const meta = { input: call, tool, calls };\n          await context.emitter.emit(\"tool\", { type: \"start\", ...meta });\n          try {\n            const output = await tool.run(call.input, { signal: context.signal }).context({\n              [Tool.contextKeys.Memory]: memory,\n            });\n            await context.emitter.emit(\"tool\", { type: \"success\", ...meta, output });\n            return output;\n          } catch (error) {\n            await context.emitter.emit(\"tool\", { type: \"error\", ...meta, error });\n            throw error;\n          }\n        }),\n      );\n    };\n\n    return {\n      memory,\n      run,\n      tools,\n    };\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n}\n"]}

package/dist/agents/experimental/replan/prompts.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/experimental/replan/prompts.ts"],"names":["createRePlanOutputSchema","tools","toolSchemas","Promise","all","map","tool","name","description","input","z","object","passthrough","inputSchema","getInputJsonSchema","zodSchemaToJsonSchema","WeakMap","shape","definition","information","record","string","describe","lookback","plan","array","title","decision","research","boolean","computation","nextStep","discriminatedUnion","type","literal","message","hasAtLeast","calls","json","toJsonSchema","override","_def","def","typeName","ZodFirstPartyTypeKind","ZodObject","schema","get","delete","ignoreOverride","RePlanSystemPrompt","PromptTemplate","template","RePlanAssistantPrompt","results"],"mappings":";;;;;;;;;;AAYA,eAAsBA,yBAA4CC,KAAAA,EAAU;AAC1E,EAAA,MAAMC,cAAc,MAAMC,OAAAA,CAAQC,IAChCH,KAAAA,CAAMI,GAAAA,CAAI,OAAOC,IAAAA,MAAU;AACzBC,IAAAA,IAAAA,EAAMD,IAAAA,CAAKC,IAAAA;AACXC,IAAAA,WAAAA,EAAaF,IAAAA,CAAKE,WAAAA;AAClBC,IAAAA,KAAAA,EAAOC,KAAAA,CAAEC,MAAAA,CAAO,EAAC,EAAGC,WAAAA,EAAW;IAC/BC,WAAAA,EAAa,MAAMP,KAAKQ,kBAAAA;AAC1B,GAAA,CAAA,CAAA,CAAA;AAEF,EAAA,MAAMC,wBAAwB,IAAIC,OAAAA,CAChCd,WAAAA,CAAYG,GAAAA,CAAI,CAACC,IAAAA,KAAS;AAACA,IAAAA,IAAAA,CAAKG,KAAAA,CAAMQ,KAAAA;IAAOX,IAAAA,CAAKO;GAAY,CAAA,CAAA;AAGhE,EAAA,MAAMK,UAAAA,GAAaR,MAAEC,MAAAA,CAAO;AAC1BQ,IAAAA,WAAAA,EAAaT,MACVU,MAAAA,CAAOV,KAAAA,CAAEW,QAAM,CAAA,CACfC,SACC,mQAAA,CAAA;AAEJC,IAAAA,QAAAA,EAAUb,KAAAA,CACPW,MAAAA,EAAM,CACNC,QAAAA,CACC,yMAAA,CAAA;IAEJE,IAAAA,EAAMd,KAAAA,CACHe,KAAAA,CACCf,KAAAA,CAAEC,MAAAA,CAAO;AACPe,MAAAA,KAAAA,EAAOhB,KAAAA,CACJW,MAAAA,EAAM,CACNC,QAAAA,CAAS,+DAAA,CAAA;AACZK,MAAAA,QAAAA,EAAUjB,KAAAA,CAAEW,MAAAA,EAAM,CAAGC,QAAAA,CAAS,kDAAA,CAAA;AAC9BM,MAAAA,QAAAA,EAAUlB,KAAAA,CACPmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,sEAAA,CAAA;AACZQ,MAAAA,WAAAA,EAAapB,KAAAA,CACVmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,4EAAA;KACd,CAAA,CAAA,CAEDA,QAAAA,CACC,yQAAA,CAAA;IAEJS,QAAAA,EAAUrB,KAAAA,CAAEsB,mBAAmB,MAAA,EAAQ;AACrCtB,MAAAA,KAAAA,CACGC,MAAAA,CAAO;QACNsB,IAAAA,EAAMvB,KAAAA,CAAEwB,QAAQ,SAAA,CAAA;AAChBC,QAAAA,OAAAA,EAASzB,KAAAA,CACNW,MAAAA,EAAM,CACNC,QAAAA,CACC,mJAAA;OAEN,CAAA,CACCA,SAAS,gFAAA,CAAA;SACRc,iBAAAA,CAAWlC,WAAAA,EAAa,CAAA,CAAA,GACxB;AACEQ,QAAAA,KAAAA,CACGC,MAAAA,CAAO;UACNsB,IAAAA,EAAMvB,KAAAA,CAAEwB,QAAQ,MAAA,CAAA;UAChBG,KAAAA,EAAO3B,KAAAA,CAAEe,KAAAA,CACPf,KAAAA,CAAEsB,kBAAAA,CACA,MAAA,EACA3B,WAAIH,WAAAA,EAAa,CAACI,IAAAA,KAChBI,KAAAA,CACGC,MAAAA,CAAO;YACNJ,IAAAA,EAAMG,KAAAA,CAAEwB,OAAAA,CAAQ5B,IAAAA,CAAKC,IAAI,CAAA;AACzBE,YAAAA,KAAAA,EAAOH,IAAAA,CAAKG;AACd,WAAA,EACCa,QAAAA,CAAShB,IAAAA,CAAKE,WAAW,CAAA,CAAA,CAAA;SAIpC,CAAA,CACCc,SAAS,sCAAA;UAEd;AACL,KAAA;GACH,CAAA;AAEA,EAAA,OAAO;AACLJ,IAAAA,UAAAA;AACAoB,IAAAA,IAAAA,EAAMC,wBAAarB,UAAAA,EAAY;AAC7BsB,MAAAA,QAAAA,0BAAWC,IAAAA,KAAAA;AACT,QAAA,MAAMC,GAAAA,GAAMD,IAAAA;AACZ,QAAA,IAAIC,GAAAA,CAAIC,QAAAA,KAAaC,yBAAAA,CAAsBC,SAAAA,EAAW;AACpD,UAAA,MAAM5B,KAAAA,GAAQyB,IAAIzB,KAAAA,EAAK;AACvB,UAAA,MAAM6B,MAAAA,GAAS/B,qBAAAA,CAAsBgC,GAAAA,CAAI9B,KAAAA,CAAAA;AACzC,UAAA,IAAI6B,MAAAA,EAAQ;AACV/B,YAAAA,qBAAAA,CAAsBiC,OAAO/B,KAAAA,CAAAA;AAC7B,YAAA,OAAO6B,MAAAA;AACT;AACF;AACA,QAAA,OAAOG,8BAAAA;OACT,EAXU,UAAA;KAYZ;AACF,GAAA;AACF;AA/FsBjD,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAqGf,MAAMkD,kBAAAA,GAAqB,IAAIC,2BAAAA,CAAe;AACnDL,EAAAA,MAAAA,EAAQpC,MAAEC,MAAAA,CAAO;AACfmC,IAAAA,MAAAA,EAAQpC,MAAEW,MAAAA;GACZ,CAAA;EACA+B,QAAAA,EAAU,CAAA;;;;;;;AAOZ,CAAA;AAEO,MAAMC,qBAAAA,GAAwB,IAAIF,2BAAAA,CAAe;AACtDL,EAAAA,MAAAA,EAAQpC,MAAEC,MAAAA,CAAO;AAAE2C,IAAAA,OAAAA,EAAS5C,MAAEW,MAAAA;GAAS,CAAA;EACvC+B,QAAAA,EAAU,CAAA,WAAA;AACZ,CAAA","file":"prompts.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodFirstPartyTypeKind, ZodObjectDef, ZodTypeDef } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { hasAtLeast, map } from \"remeda\";\nimport { toJsonSchema } from \"@/internals/helpers/schema.js\";\nimport { ignoreOverride, JsonSchema7Type } from \"zod-to-json-schema\";\n\nexport async function createRePlanOutputSchema<T extends AnyTool>(tools: T[]) {\n  const toolSchemas = await Promise.all(\n    tools.map(async (tool) => ({\n      name: tool.name,\n      description: tool.description,\n      input: z.object({}).passthrough(),\n      inputSchema: await tool.getInputJsonSchema(),\n    })),\n  );\n  const zodSchemaToJsonSchema = new WeakMap(\n    toolSchemas.map((tool) => [tool.input.shape, tool.inputSchema]),\n  );\n\n  const definition = z.object({\n    information: z\n      .record(z.string())\n      .describe(\n        \"Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included.\",\n      ),\n    lookback: z\n      .string()\n      .describe(\n        \"A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point.\",\n      ),\n    plan: z\n      .array(\n        z.object({\n          title: z\n            .string()\n            .describe(\"Title of this step, shortly describing what needs to be done.\"),\n          decision: z.string().describe(\"Assistant's decision of how to tackle this step.\"),\n          research: z\n            .boolean()\n            .describe(\"Does this step involve looking up factual information through tools?\"),\n          computation: z\n            .boolean()\n            .describe(\"Does this step involve calculating or computing information through tools?\"),\n        }),\n      )\n      .describe(\n        \"Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on.\",\n      ),\n    nextStep: z.discriminatedUnion(\"type\", [\n      z\n        .object({\n          type: z.literal(\"message\"),\n          message: z\n            .string()\n            .describe(\n              \"Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.\",\n            ),\n        })\n        .describe(\"Message the user -- either to give the answer, or to ask for more information.\"),\n      ...(hasAtLeast(toolSchemas, 1)\n        ? [\n            z\n              .object({\n                type: z.literal(\"tool\"),\n                calls: z.array(\n                  z.discriminatedUnion(\n                    \"name\",\n                    map(toolSchemas, (tool) =>\n                      z\n                        .object({\n                          name: z.literal(tool.name),\n                          input: tool.input,\n                        })\n                        .describe(tool.description),\n                    ),\n                  ),\n                ),\n              })\n              .describe(\"Obtain more information using tools.\"),\n          ]\n        : []),\n    ]),\n  });\n\n  return {\n    definition,\n    json: toJsonSchema(definition, {\n      override: (_def: ZodTypeDef) => {\n        const def = _def as ZodObjectDef;\n        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {\n          const shape = def.shape();\n          const schema = zodSchemaToJsonSchema.get(shape) as JsonSchema7Type;\n          if (schema) {\n            zodSchemaToJsonSchema.delete(shape);\n            return schema;\n          }\n        }\n        return ignoreOverride;\n      },\n    }),\n  };\n}\n\nexport type RePlanState = z.output<\n  Awaited<ReturnType<typeof createRePlanOutputSchema>>[\"definition\"]\n>;\n\nexport const RePlanSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    schema: z.string(),\n  }),\n  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.\n\nThe assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.\n\nThe assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.\n\nOutput Schema: {{schema}}`,\n});\n\nexport const RePlanAssistantPrompt = new PromptTemplate({\n  schema: z.object({ results: z.string() }),\n  template: `{{results}}`,\n});\n"]}
+{"version":3,"sources":["../../../../src/agents/experimental/replan/prompts.ts"],"names":["createRePlanOutputSchema","tools","toolSchemas","Promise","all","map","tool","name","description","input","z","object","passthrough","inputSchema","getInputJsonSchema","zodSchemaToJsonSchema","WeakMap","shape","definition","information","record","string","describe","lookback","plan","array","title","decision","research","boolean","computation","nextStep","discriminatedUnion","type","literal","message","hasAtLeast","calls","json","toJsonSchema","override","_def","def","typeName","ZodFirstPartyTypeKind","ZodObject","schema","get","delete","ignoreOverride","RePlanSystemPrompt","PromptTemplate","template","RePlanAssistantPrompt","results"],"mappings":";;;;;;;;;;AAYA,eAAsBA,yBAA4CC,KAAAA,EAAU;AAC1E,EAAA,MAAMC,cAAc,MAAMC,OAAAA,CAAQC,IAChCH,KAAAA,CAAMI,GAAAA,CAAI,OAAOC,IAAAA,MAAU;AACzBC,IAAAA,IAAAA,EAAMD,IAAAA,CAAKC,IAAAA;AACXC,IAAAA,WAAAA,EAAaF,IAAAA,CAAKE,WAAAA;AAClBC,IAAAA,KAAAA,EAAOC,KAAAA,CAAEC,MAAAA,CAAO,EAAC,EAAGC,WAAAA,EAAW;IAC/BC,WAAAA,EAAa,MAAMP,KAAKQ,kBAAAA;AAC1B,GAAA,CAAA,CAAA,CAAA;AAEF,EAAA,MAAMC,wBAAwB,IAAIC,OAAAA,CAChCd,WAAAA,CAAYG,GAAAA,CAAI,CAACC,IAAAA,KAAS;AAACA,IAAAA,IAAAA,CAAKG,KAAAA,CAAMQ,KAAAA;IAAOX,IAAAA,CAAKO;GAAY,CAAA,CAAA;AAGhE,EAAA,MAAMK,UAAAA,GAAaR,MAAEC,MAAAA,CAAO;AAC1BQ,IAAAA,WAAAA,EAAaT,MACVU,MAAAA,CAAOV,KAAAA,CAAEW,QAAM,CAAA,CACfC,SACC,mQAAA,CAAA;AAEJC,IAAAA,QAAAA,EAAUb,KAAAA,CACPW,MAAAA,EAAM,CACNC,QAAAA,CACC,yMAAA,CAAA;IAEJE,IAAAA,EAAMd,KAAAA,CACHe,KAAAA,CACCf,KAAAA,CAAEC,MAAAA,CAAO;AACPe,MAAAA,KAAAA,EAAOhB,KAAAA,CACJW,MAAAA,EAAM,CACNC,QAAAA,CAAS,+DAAA,CAAA;AACZK,MAAAA,QAAAA,EAAUjB,KAAAA,CAAEW,MAAAA,EAAM,CAAGC,QAAAA,CAAS,kDAAA,CAAA;AAC9BM,MAAAA,QAAAA,EAAUlB,KAAAA,CACPmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,sEAAA,CAAA;AACZQ,MAAAA,WAAAA,EAAapB,KAAAA,CACVmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,4EAAA;KACd,CAAA,CAAA,CAEDA,QAAAA,CACC,yQAAA,CAAA;IAEJS,QAAAA,EAAUrB,KAAAA,CAAEsB,mBAAmB,MAAA,EAAQ;AACrCtB,MAAAA,KAAAA,CACGC,MAAAA,CAAO;QACNsB,IAAAA,EAAMvB,KAAAA,CAAEwB,QAAQ,SAAA,CAAA;AAChBC,QAAAA,OAAAA,EAASzB,KAAAA,CACNW,MAAAA,EAAM,CACNC,QAAAA,CACC,mJAAA;OAEN,CAAA,CACCA,SAAS,gFAAA,CAAA;SACRc,iBAAAA,CAAWlC,WAAAA,EAAa,CAAA,CAAA,GACxB;AACEQ,QAAAA,KAAAA,CACGC,MAAAA,CAAO;UACNsB,IAAAA,EAAMvB,KAAAA,CAAEwB,QAAQ,MAAA,CAAA;UAChBG,KAAAA,EAAO3B,KAAAA,CAAEe,KAAAA,CACPf,KAAAA,CAAEsB,kBAAAA,CACA,MAAA,EACA3B,WAAIH,WAAAA,EAAa,CAACI,IAAAA,KAChBI,KAAAA,CACGC,MAAAA,CAAO;YACNJ,IAAAA,EAAMG,KAAAA,CAAEwB,OAAAA,CAAQ5B,IAAAA,CAAKC,IAAI,CAAA;AACzBE,YAAAA,KAAAA,EAAOH,IAAAA,CAAKG;AACd,WAAA,EACCa,QAAAA,CAAShB,IAAAA,CAAKE,WAAW,CAAA,CAAA,CAAA;SAIpC,CAAA,CACCc,SAAS,sCAAA;UAEd;AACL,KAAA;GACH,CAAA;AAEA,EAAA,OAAO;AACLJ,IAAAA,UAAAA;AACAoB,IAAAA,IAAAA,EAAMC,wBAAarB,UAAAA,EAAY;AAC7BsB,MAAAA,QAAAA,0BAAWC,IAAAA,KAAAA;AACT,QAAA,MAAMC,GAAAA,GAAMD,IAAAA;AACZ,QAAA,IAAIC,GAAAA,CAAIC,QAAAA,KAAaC,yBAAAA,CAAsBC,SAAAA,EAAW;AACpD,UAAA,MAAM5B,KAAAA,GAAQyB,IAAIzB,KAAAA,EAAK;AACvB,UAAA,MAAM6B,MAAAA,GAAS/B,qBAAAA,CAAsBgC,GAAAA,CAAI9B,KAAAA,CAAAA;AACzC,UAAA,IAAI6B,MAAAA,EAAQ;AACV/B,YAAAA,qBAAAA,CAAsBiC,OAAO/B,KAAAA,CAAAA;AAC7B,YAAA,OAAO6B,MAAAA;AACT,UAAA;AACF,QAAA;AACA,QAAA,OAAOG,8BAAAA;MACT,CAAA,EAXU,UAAA;KAYZ;AACF,GAAA;AACF;AA/FsBjD,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAqGf,MAAMkD,kBAAAA,GAAqB,IAAIC,2BAAAA,CAAe;AACnDL,EAAAA,MAAAA,EAAQpC,MAAEC,MAAAA,CAAO;AACfmC,IAAAA,MAAAA,EAAQpC,MAAEW,MAAAA;GACZ,CAAA;EACA+B,QAAAA,EAAU,CAAA;;;;;;;AAOZ,CAAA;AAEO,MAAMC,qBAAAA,GAAwB,IAAIF,2BAAAA,CAAe;AACtDL,EAAAA,MAAAA,EAAQpC,MAAEC,MAAAA,CAAO;AAAE2C,IAAAA,OAAAA,EAAS5C,MAAEW,MAAAA;GAAS,CAAA;EACvC+B,QAAAA,EAAU,CAAA,WAAA;AACZ,CAAA","file":"prompts.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodFirstPartyTypeKind, ZodObjectDef, ZodTypeDef } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { hasAtLeast, map } from \"remeda\";\nimport { toJsonSchema } from \"@/internals/helpers/schema.js\";\nimport { ignoreOverride, JsonSchema7Type } from \"zod-to-json-schema\";\n\nexport async function createRePlanOutputSchema<T extends AnyTool>(tools: T[]) {\n  const toolSchemas = await Promise.all(\n    tools.map(async (tool) => ({\n      name: tool.name,\n      description: tool.description,\n      input: z.object({}).passthrough(),\n      inputSchema: await tool.getInputJsonSchema(),\n    })),\n  );\n  const zodSchemaToJsonSchema = new WeakMap(\n    toolSchemas.map((tool) => [tool.input.shape, tool.inputSchema]),\n  );\n\n  const definition = z.object({\n    information: z\n      .record(z.string())\n      .describe(\n        \"Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included.\",\n      ),\n    lookback: z\n      .string()\n      .describe(\n        \"A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point.\",\n      ),\n    plan: z\n      .array(\n        z.object({\n          title: z\n            .string()\n            .describe(\"Title of this step, shortly describing what needs to be done.\"),\n          decision: z.string().describe(\"Assistant's decision of how to tackle this step.\"),\n          research: z\n            .boolean()\n            .describe(\"Does this step involve looking up factual information through tools?\"),\n          computation: z\n            .boolean()\n            .describe(\"Does this step involve calculating or computing information through tools?\"),\n        }),\n      )\n      .describe(\n        \"Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on.\",\n      ),\n    nextStep: z.discriminatedUnion(\"type\", [\n      z\n        .object({\n          type: z.literal(\"message\"),\n          message: z\n            .string()\n            .describe(\n              \"Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.\",\n            ),\n        })\n        .describe(\"Message the user -- either to give the answer, or to ask for more information.\"),\n      ...(hasAtLeast(toolSchemas, 1)\n        ? [\n            z\n              .object({\n                type: z.literal(\"tool\"),\n                calls: z.array(\n                  z.discriminatedUnion(\n                    \"name\",\n                    map(toolSchemas, (tool) =>\n                      z\n                        .object({\n                          name: z.literal(tool.name),\n                          input: tool.input,\n                        })\n                        .describe(tool.description),\n                    ),\n                  ),\n                ),\n              })\n              .describe(\"Obtain more information using tools.\"),\n          ]\n        : []),\n    ]),\n  });\n\n  return {\n    definition,\n    json: toJsonSchema(definition, {\n      override: (_def: ZodTypeDef) => {\n        const def = _def as ZodObjectDef;\n        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {\n          const shape = def.shape();\n          const schema = zodSchemaToJsonSchema.get(shape) as JsonSchema7Type;\n          if (schema) {\n            zodSchemaToJsonSchema.delete(shape);\n            return schema;\n          }\n        }\n        return ignoreOverride;\n      },\n    }),\n  };\n}\n\nexport type RePlanState = z.output<\n  Awaited<ReturnType<typeof createRePlanOutputSchema>>[\"definition\"]\n>;\n\nexport const RePlanSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    schema: z.string(),\n  }),\n  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.\n\nThe assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.\n\nThe assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.\n\nOutput Schema: {{schema}}`,\n});\n\nexport const RePlanAssistantPrompt = new PromptTemplate({\n  schema: z.object({ results: z.string() }),\n  template: `{{results}}`,\n});\n"]}

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

@@ -7,7 +7,7 @@ import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';
 import '../../../context.cjs';
-import '../../../emitter-UkIkNrek.cjs';
+import '../../../emitter-D4OcelZ9.cjs';
 import '../../../internals/helpers/promise.cjs';
 import 'promise-based-task';
 import '../../../cache/base.cjs';

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

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

package/dist/agents/experimental/replan/prompts.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/experimental/replan/prompts.ts"],"names":["createRePlanOutputSchema","tools","toolSchemas","Promise","all","map","tool","name","description","input","z","object","passthrough","inputSchema","getInputJsonSchema","zodSchemaToJsonSchema","WeakMap","shape","definition","information","record","string","describe","lookback","plan","array","title","decision","research","boolean","computation","nextStep","discriminatedUnion","type","literal","message","hasAtLeast","calls","json","toJsonSchema","override","_def","def","typeName","ZodFirstPartyTypeKind","ZodObject","schema","get","delete","ignoreOverride","RePlanSystemPrompt","PromptTemplate","template","RePlanAssistantPrompt","results"],"mappings":";;;;;;;;AAYA,eAAsBA,yBAA4CC,KAAAA,EAAU;AAC1E,EAAA,MAAMC,cAAc,MAAMC,OAAAA,CAAQC,IAChCH,KAAAA,CAAMI,GAAAA,CAAI,OAAOC,IAAAA,MAAU;AACzBC,IAAAA,IAAAA,EAAMD,IAAAA,CAAKC,IAAAA;AACXC,IAAAA,WAAAA,EAAaF,IAAAA,CAAKE,WAAAA;AAClBC,IAAAA,KAAAA,EAAOC,CAAAA,CAAEC,MAAAA,CAAO,EAAC,EAAGC,WAAAA,EAAW;IAC/BC,WAAAA,EAAa,MAAMP,KAAKQ,kBAAAA;AAC1B,GAAA,CAAA,CAAA,CAAA;AAEF,EAAA,MAAMC,wBAAwB,IAAIC,OAAAA,CAChCd,WAAAA,CAAYG,GAAAA,CAAI,CAACC,IAAAA,KAAS;AAACA,IAAAA,IAAAA,CAAKG,KAAAA,CAAMQ,KAAAA;IAAOX,IAAAA,CAAKO;GAAY,CAAA,CAAA;AAGhE,EAAA,MAAMK,UAAAA,GAAaR,EAAEC,MAAAA,CAAO;AAC1BQ,IAAAA,WAAAA,EAAaT,EACVU,MAAAA,CAAOV,CAAAA,CAAEW,QAAM,CAAA,CACfC,SACC,mQAAA,CAAA;AAEJC,IAAAA,QAAAA,EAAUb,CAAAA,CACPW,MAAAA,EAAM,CACNC,QAAAA,CACC,yMAAA,CAAA;IAEJE,IAAAA,EAAMd,CAAAA,CACHe,KAAAA,CACCf,CAAAA,CAAEC,MAAAA,CAAO;AACPe,MAAAA,KAAAA,EAAOhB,CAAAA,CACJW,MAAAA,EAAM,CACNC,QAAAA,CAAS,+DAAA,CAAA;AACZK,MAAAA,QAAAA,EAAUjB,CAAAA,CAAEW,MAAAA,EAAM,CAAGC,QAAAA,CAAS,kDAAA,CAAA;AAC9BM,MAAAA,QAAAA,EAAUlB,CAAAA,CACPmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,sEAAA,CAAA;AACZQ,MAAAA,WAAAA,EAAapB,CAAAA,CACVmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,4EAAA;KACd,CAAA,CAAA,CAEDA,QAAAA,CACC,yQAAA,CAAA;IAEJS,QAAAA,EAAUrB,CAAAA,CAAEsB,mBAAmB,MAAA,EAAQ;AACrCtB,MAAAA,CAAAA,CACGC,MAAAA,CAAO;QACNsB,IAAAA,EAAMvB,CAAAA,CAAEwB,QAAQ,SAAA,CAAA;AAChBC,QAAAA,OAAAA,EAASzB,CAAAA,CACNW,MAAAA,EAAM,CACNC,QAAAA,CACC,mJAAA;OAEN,CAAA,CACCA,SAAS,gFAAA,CAAA;SACRc,UAAAA,CAAWlC,WAAAA,EAAa,CAAA,CAAA,GACxB;AACEQ,QAAAA,CAAAA,CACGC,MAAAA,CAAO;UACNsB,IAAAA,EAAMvB,CAAAA,CAAEwB,QAAQ,MAAA,CAAA;UAChBG,KAAAA,EAAO3B,CAAAA,CAAEe,KAAAA,CACPf,CAAAA,CAAEsB,kBAAAA,CACA,MAAA,EACA3B,IAAIH,WAAAA,EAAa,CAACI,IAAAA,KAChBI,CAAAA,CACGC,MAAAA,CAAO;YACNJ,IAAAA,EAAMG,CAAAA,CAAEwB,OAAAA,CAAQ5B,IAAAA,CAAKC,IAAI,CAAA;AACzBE,YAAAA,KAAAA,EAAOH,IAAAA,CAAKG;AACd,WAAA,EACCa,QAAAA,CAAShB,IAAAA,CAAKE,WAAW,CAAA,CAAA,CAAA;SAIpC,CAAA,CACCc,SAAS,sCAAA;UAEd;AACL,KAAA;GACH,CAAA;AAEA,EAAA,OAAO;AACLJ,IAAAA,UAAAA;AACAoB,IAAAA,IAAAA,EAAMC,aAAarB,UAAAA,EAAY;AAC7BsB,MAAAA,QAAAA,0BAAWC,IAAAA,KAAAA;AACT,QAAA,MAAMC,GAAAA,GAAMD,IAAAA;AACZ,QAAA,IAAIC,GAAAA,CAAIC,QAAAA,KAAaC,qBAAAA,CAAsBC,SAAAA,EAAW;AACpD,UAAA,MAAM5B,KAAAA,GAAQyB,IAAIzB,KAAAA,EAAK;AACvB,UAAA,MAAM6B,MAAAA,GAAS/B,qBAAAA,CAAsBgC,GAAAA,CAAI9B,KAAAA,CAAAA;AACzC,UAAA,IAAI6B,MAAAA,EAAQ;AACV/B,YAAAA,qBAAAA,CAAsBiC,OAAO/B,KAAAA,CAAAA;AAC7B,YAAA,OAAO6B,MAAAA;AACT;AACF;AACA,QAAA,OAAOG,cAAAA;OACT,EAXU,UAAA;KAYZ;AACF,GAAA;AACF;AA/FsBjD,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAqGf,MAAMkD,kBAAAA,GAAqB,IAAIC,cAAAA,CAAe;AACnDL,EAAAA,MAAAA,EAAQpC,EAAEC,MAAAA,CAAO;AACfmC,IAAAA,MAAAA,EAAQpC,EAAEW,MAAAA;GACZ,CAAA;EACA+B,QAAAA,EAAU,CAAA;;;;;;;AAOZ,CAAA;AAEO,MAAMC,qBAAAA,GAAwB,IAAIF,cAAAA,CAAe;AACtDL,EAAAA,MAAAA,EAAQpC,EAAEC,MAAAA,CAAO;AAAE2C,IAAAA,OAAAA,EAAS5C,EAAEW,MAAAA;GAAS,CAAA;EACvC+B,QAAAA,EAAU,CAAA,WAAA;AACZ,CAAA","file":"prompts.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodFirstPartyTypeKind, ZodObjectDef, ZodTypeDef } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { hasAtLeast, map } from \"remeda\";\nimport { toJsonSchema } from \"@/internals/helpers/schema.js\";\nimport { ignoreOverride, JsonSchema7Type } from \"zod-to-json-schema\";\n\nexport async function createRePlanOutputSchema<T extends AnyTool>(tools: T[]) {\n  const toolSchemas = await Promise.all(\n    tools.map(async (tool) => ({\n      name: tool.name,\n      description: tool.description,\n      input: z.object({}).passthrough(),\n      inputSchema: await tool.getInputJsonSchema(),\n    })),\n  );\n  const zodSchemaToJsonSchema = new WeakMap(\n    toolSchemas.map((tool) => [tool.input.shape, tool.inputSchema]),\n  );\n\n  const definition = z.object({\n    information: z\n      .record(z.string())\n      .describe(\n        \"Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included.\",\n      ),\n    lookback: z\n      .string()\n      .describe(\n        \"A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point.\",\n      ),\n    plan: z\n      .array(\n        z.object({\n          title: z\n            .string()\n            .describe(\"Title of this step, shortly describing what needs to be done.\"),\n          decision: z.string().describe(\"Assistant's decision of how to tackle this step.\"),\n          research: z\n            .boolean()\n            .describe(\"Does this step involve looking up factual information through tools?\"),\n          computation: z\n            .boolean()\n            .describe(\"Does this step involve calculating or computing information through tools?\"),\n        }),\n      )\n      .describe(\n        \"Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on.\",\n      ),\n    nextStep: z.discriminatedUnion(\"type\", [\n      z\n        .object({\n          type: z.literal(\"message\"),\n          message: z\n            .string()\n            .describe(\n              \"Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.\",\n            ),\n        })\n        .describe(\"Message the user -- either to give the answer, or to ask for more information.\"),\n      ...(hasAtLeast(toolSchemas, 1)\n        ? [\n            z\n              .object({\n                type: z.literal(\"tool\"),\n                calls: z.array(\n                  z.discriminatedUnion(\n                    \"name\",\n                    map(toolSchemas, (tool) =>\n                      z\n                        .object({\n                          name: z.literal(tool.name),\n                          input: tool.input,\n                        })\n                        .describe(tool.description),\n                    ),\n                  ),\n                ),\n              })\n              .describe(\"Obtain more information using tools.\"),\n          ]\n        : []),\n    ]),\n  });\n\n  return {\n    definition,\n    json: toJsonSchema(definition, {\n      override: (_def: ZodTypeDef) => {\n        const def = _def as ZodObjectDef;\n        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {\n          const shape = def.shape();\n          const schema = zodSchemaToJsonSchema.get(shape) as JsonSchema7Type;\n          if (schema) {\n            zodSchemaToJsonSchema.delete(shape);\n            return schema;\n          }\n        }\n        return ignoreOverride;\n      },\n    }),\n  };\n}\n\nexport type RePlanState = z.output<\n  Awaited<ReturnType<typeof createRePlanOutputSchema>>[\"definition\"]\n>;\n\nexport const RePlanSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    schema: z.string(),\n  }),\n  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.\n\nThe assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.\n\nThe assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.\n\nOutput Schema: {{schema}}`,\n});\n\nexport const RePlanAssistantPrompt = new PromptTemplate({\n  schema: z.object({ results: z.string() }),\n  template: `{{results}}`,\n});\n"]}
+{"version":3,"sources":["../../../../src/agents/experimental/replan/prompts.ts"],"names":["createRePlanOutputSchema","tools","toolSchemas","Promise","all","map","tool","name","description","input","z","object","passthrough","inputSchema","getInputJsonSchema","zodSchemaToJsonSchema","WeakMap","shape","definition","information","record","string","describe","lookback","plan","array","title","decision","research","boolean","computation","nextStep","discriminatedUnion","type","literal","message","hasAtLeast","calls","json","toJsonSchema","override","_def","def","typeName","ZodFirstPartyTypeKind","ZodObject","schema","get","delete","ignoreOverride","RePlanSystemPrompt","PromptTemplate","template","RePlanAssistantPrompt","results"],"mappings":";;;;;;;;AAYA,eAAsBA,yBAA4CC,KAAAA,EAAU;AAC1E,EAAA,MAAMC,cAAc,MAAMC,OAAAA,CAAQC,IAChCH,KAAAA,CAAMI,GAAAA,CAAI,OAAOC,IAAAA,MAAU;AACzBC,IAAAA,IAAAA,EAAMD,IAAAA,CAAKC,IAAAA;AACXC,IAAAA,WAAAA,EAAaF,IAAAA,CAAKE,WAAAA;AAClBC,IAAAA,KAAAA,EAAOC,CAAAA,CAAEC,MAAAA,CAAO,EAAC,EAAGC,WAAAA,EAAW;IAC/BC,WAAAA,EAAa,MAAMP,KAAKQ,kBAAAA;AAC1B,GAAA,CAAA,CAAA,CAAA;AAEF,EAAA,MAAMC,wBAAwB,IAAIC,OAAAA,CAChCd,WAAAA,CAAYG,GAAAA,CAAI,CAACC,IAAAA,KAAS;AAACA,IAAAA,IAAAA,CAAKG,KAAAA,CAAMQ,KAAAA;IAAOX,IAAAA,CAAKO;GAAY,CAAA,CAAA;AAGhE,EAAA,MAAMK,UAAAA,GAAaR,EAAEC,MAAAA,CAAO;AAC1BQ,IAAAA,WAAAA,EAAaT,EACVU,MAAAA,CAAOV,CAAAA,CAAEW,QAAM,CAAA,CACfC,SACC,mQAAA,CAAA;AAEJC,IAAAA,QAAAA,EAAUb,CAAAA,CACPW,MAAAA,EAAM,CACNC,QAAAA,CACC,yMAAA,CAAA;IAEJE,IAAAA,EAAMd,CAAAA,CACHe,KAAAA,CACCf,CAAAA,CAAEC,MAAAA,CAAO;AACPe,MAAAA,KAAAA,EAAOhB,CAAAA,CACJW,MAAAA,EAAM,CACNC,QAAAA,CAAS,+DAAA,CAAA;AACZK,MAAAA,QAAAA,EAAUjB,CAAAA,CAAEW,MAAAA,EAAM,CAAGC,QAAAA,CAAS,kDAAA,CAAA;AAC9BM,MAAAA,QAAAA,EAAUlB,CAAAA,CACPmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,sEAAA,CAAA;AACZQ,MAAAA,WAAAA,EAAapB,CAAAA,CACVmB,OAAAA,EAAO,CACPP,QAAAA,CAAS,4EAAA;KACd,CAAA,CAAA,CAEDA,QAAAA,CACC,yQAAA,CAAA;IAEJS,QAAAA,EAAUrB,CAAAA,CAAEsB,mBAAmB,MAAA,EAAQ;AACrCtB,MAAAA,CAAAA,CACGC,MAAAA,CAAO;QACNsB,IAAAA,EAAMvB,CAAAA,CAAEwB,QAAQ,SAAA,CAAA;AAChBC,QAAAA,OAAAA,EAASzB,CAAAA,CACNW,MAAAA,EAAM,CACNC,QAAAA,CACC,mJAAA;OAEN,CAAA,CACCA,SAAS,gFAAA,CAAA;SACRc,UAAAA,CAAWlC,WAAAA,EAAa,CAAA,CAAA,GACxB;AACEQ,QAAAA,CAAAA,CACGC,MAAAA,CAAO;UACNsB,IAAAA,EAAMvB,CAAAA,CAAEwB,QAAQ,MAAA,CAAA;UAChBG,KAAAA,EAAO3B,CAAAA,CAAEe,KAAAA,CACPf,CAAAA,CAAEsB,kBAAAA,CACA,MAAA,EACA3B,IAAIH,WAAAA,EAAa,CAACI,IAAAA,KAChBI,CAAAA,CACGC,MAAAA,CAAO;YACNJ,IAAAA,EAAMG,CAAAA,CAAEwB,OAAAA,CAAQ5B,IAAAA,CAAKC,IAAI,CAAA;AACzBE,YAAAA,KAAAA,EAAOH,IAAAA,CAAKG;AACd,WAAA,EACCa,QAAAA,CAAShB,IAAAA,CAAKE,WAAW,CAAA,CAAA,CAAA;SAIpC,CAAA,CACCc,SAAS,sCAAA;UAEd;AACL,KAAA;GACH,CAAA;AAEA,EAAA,OAAO;AACLJ,IAAAA,UAAAA;AACAoB,IAAAA,IAAAA,EAAMC,aAAarB,UAAAA,EAAY;AAC7BsB,MAAAA,QAAAA,0BAAWC,IAAAA,KAAAA;AACT,QAAA,MAAMC,GAAAA,GAAMD,IAAAA;AACZ,QAAA,IAAIC,GAAAA,CAAIC,QAAAA,KAAaC,qBAAAA,CAAsBC,SAAAA,EAAW;AACpD,UAAA,MAAM5B,KAAAA,GAAQyB,IAAIzB,KAAAA,EAAK;AACvB,UAAA,MAAM6B,MAAAA,GAAS/B,qBAAAA,CAAsBgC,GAAAA,CAAI9B,KAAAA,CAAAA;AACzC,UAAA,IAAI6B,MAAAA,EAAQ;AACV/B,YAAAA,qBAAAA,CAAsBiC,OAAO/B,KAAAA,CAAAA;AAC7B,YAAA,OAAO6B,MAAAA;AACT,UAAA;AACF,QAAA;AACA,QAAA,OAAOG,cAAAA;MACT,CAAA,EAXU,UAAA;KAYZ;AACF,GAAA;AACF;AA/FsBjD,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAqGf,MAAMkD,kBAAAA,GAAqB,IAAIC,cAAAA,CAAe;AACnDL,EAAAA,MAAAA,EAAQpC,EAAEC,MAAAA,CAAO;AACfmC,IAAAA,MAAAA,EAAQpC,EAAEW,MAAAA;GACZ,CAAA;EACA+B,QAAAA,EAAU,CAAA;;;;;;;AAOZ,CAAA;AAEO,MAAMC,qBAAAA,GAAwB,IAAIF,cAAAA,CAAe;AACtDL,EAAAA,MAAAA,EAAQpC,EAAEC,MAAAA,CAAO;AAAE2C,IAAAA,OAAAA,EAAS5C,EAAEW,MAAAA;GAAS,CAAA;EACvC+B,QAAAA,EAAU,CAAA,WAAA;AACZ,CAAA","file":"prompts.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodFirstPartyTypeKind, ZodObjectDef, ZodTypeDef } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { hasAtLeast, map } from \"remeda\";\nimport { toJsonSchema } from \"@/internals/helpers/schema.js\";\nimport { ignoreOverride, JsonSchema7Type } from \"zod-to-json-schema\";\n\nexport async function createRePlanOutputSchema<T extends AnyTool>(tools: T[]) {\n  const toolSchemas = await Promise.all(\n    tools.map(async (tool) => ({\n      name: tool.name,\n      description: tool.description,\n      input: z.object({}).passthrough(),\n      inputSchema: await tool.getInputJsonSchema(),\n    })),\n  );\n  const zodSchemaToJsonSchema = new WeakMap(\n    toolSchemas.map((tool) => [tool.input.shape, tool.inputSchema]),\n  );\n\n  const definition = z.object({\n    information: z\n      .record(z.string())\n      .describe(\n        \"Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included.\",\n      ),\n    lookback: z\n      .string()\n      .describe(\n        \"A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point.\",\n      ),\n    plan: z\n      .array(\n        z.object({\n          title: z\n            .string()\n            .describe(\"Title of this step, shortly describing what needs to be done.\"),\n          decision: z.string().describe(\"Assistant's decision of how to tackle this step.\"),\n          research: z\n            .boolean()\n            .describe(\"Does this step involve looking up factual information through tools?\"),\n          computation: z\n            .boolean()\n            .describe(\"Does this step involve calculating or computing information through tools?\"),\n        }),\n      )\n      .describe(\n        \"Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on.\",\n      ),\n    nextStep: z.discriminatedUnion(\"type\", [\n      z\n        .object({\n          type: z.literal(\"message\"),\n          message: z\n            .string()\n            .describe(\n              \"Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.\",\n            ),\n        })\n        .describe(\"Message the user -- either to give the answer, or to ask for more information.\"),\n      ...(hasAtLeast(toolSchemas, 1)\n        ? [\n            z\n              .object({\n                type: z.literal(\"tool\"),\n                calls: z.array(\n                  z.discriminatedUnion(\n                    \"name\",\n                    map(toolSchemas, (tool) =>\n                      z\n                        .object({\n                          name: z.literal(tool.name),\n                          input: tool.input,\n                        })\n                        .describe(tool.description),\n                    ),\n                  ),\n                ),\n              })\n              .describe(\"Obtain more information using tools.\"),\n          ]\n        : []),\n    ]),\n  });\n\n  return {\n    definition,\n    json: toJsonSchema(definition, {\n      override: (_def: ZodTypeDef) => {\n        const def = _def as ZodObjectDef;\n        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {\n          const shape = def.shape();\n          const schema = zodSchemaToJsonSchema.get(shape) as JsonSchema7Type;\n          if (schema) {\n            zodSchemaToJsonSchema.delete(shape);\n            return schema;\n          }\n        }\n        return ignoreOverride;\n      },\n    }),\n  };\n}\n\nexport type RePlanState = z.output<\n  Awaited<ReturnType<typeof createRePlanOutputSchema>>[\"definition\"]\n>;\n\nexport const RePlanSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    schema: z.string(),\n  }),\n  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.\n\nThe assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.\n\nThe assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.\n\nOutput Schema: {{schema}}`,\n});\n\nexport const RePlanAssistantPrompt = new PromptTemplate({\n  schema: z.object({ results: z.string() }),\n  template: `{{results}}`,\n});\n"]}

package/dist/agents/experimental/streamlit/agent.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/experimental/streamlit/agent.ts"],"names":["StreamlitAgent","BaseAgent","emitter","Emitter","namespace","creator","input","meta","name","tools","description","memory","_run","_options","run","userMessage","runMemory","prepare","content","raw","llm","create","stream","messages","abortSignal","signal","observe","on","value","chunk","delta","getTextContent","emit","state","result","parse","assistantMessage","AssistantMessage","addMany","filter","isTruthy","message","systemMessage","SystemMessage","templates","system","StreamlitAgentSystemPrompt","render","prompt","isEmpty","UserMessage","createdAt","Date","inputMessages","TokenMemory","capacityThreshold","syncThreshold","handlers","removalSelector","msgs","prevConversationMsg","find","msg","includes","last","lastMsg","length","m","role","Role","ASSISTANT","AgentError","blocks","i","text","substring","code","findFirstPair","allowOverlap","push","start","end","inner","createSnapshot"],"mappings":";;;;;;;;;;;;AA0DO,MAAMA,uBAAuBC,kBAAAA,CAAAA;EA1DpC;;;;AA2DSC,EAAAA,OAAAA,GAAU,IAAIC,mBAAAA,CAAyB;IAC5CC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA,cAAA;AAAgB,MAAA;;IACrCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAA4B;AACzD,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;AAEA,EAAA,IAAWC,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,CAAA,SAAA,CAAA;AACNC,MAAAA,KAAAA,EAAO,EAAA;MACPC,WAAAA,EAAa,CAAA,iLAAA;AACf,KAAA;AACF;AAEA,EAAA,IAAIC,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB;AAEA,EAAA,IAAWA,MAAAA,GAAS;AAClB,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB;EAEA,MAAgBC,IAAAA,CACdN,KAAAA,EACAO,QAAAA,EACAC,GAAAA,EAC6B;AAC7B,IAAA,MAAM,EAAEC,WAAAA,EAAaC,SAAAA,KAAc,MAAM,IAAA,CAAKC,QAAQX,KAAAA,CAAAA;AAEtD,IAAA,IAAIY,OAAAA,GAAU,EAAA;AACd,IAAA,MAAMC,GAAAA,GAAM,MAAM,IAAA,CAAKb,KAAAA,CAAMc,IAC1BC,MAAAA,CAAO;MACNC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,QAAAA,EAAUP,SAAAA,CAAUO,QAAAA;AACpBC,MAAAA,WAAAA,EAAaV,GAAAA,CAAIW;KACnB,CAAA,CACCC,OAAAA,CAAQ,CAACxB,OAAAA,KAAAA;AACRA,MAAAA,OAAAA,CAAQyB,GAAG,UAAA,EAAY,OAAO,EAAEC,KAAAA,EAAOC,OAAK,KAAE;AAC5C,QAAA,MAAMC,KAAAA,GAAQD,MAAME,cAAAA,EAAc;AAClC,QAAA,IAAID,KAAAA,EAAO;AACTZ,UAAAA,OAAAA,IAAWY,KAAAA;AACX,UAAA,MAAMhB,GAAAA,CAAIZ,OAAAA,CAAQ8B,IAAAA,CAAK,UAAA,EAAY;AAAEF,YAAAA,KAAAA;YAAOG,KAAAA,EAAO;AAAEf,cAAAA;AAAQ,aAAA;AAAGW,YAAAA;WAAM,CAAA;AACxE;OACF,CAAA;KACF,CAAA;AACF,IAAA,MAAMK,SAAS,IAAA,CAAKC,KAAAA,CAAMjB,OAAAA,IAAWC,GAAAA,CAAIY,gBAAc,CAAA;AAEvD,IAAA,MAAMK,gBAAAA,GAAmB,IAAIC,4BAAAA,CAAiBnB,OAAAA,CAAAA;AAC9C,IAAA,MAAM,IAAA,CAAKP,OAAO2B,OAAAA,CAAQ;AAACvB,MAAAA,WAAAA;AAAaqB,MAAAA;AAAkBG,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AACLN,MAAAA,MAAAA;MACAO,OAAAA,EAASL,gBAAAA;MACTzB,MAAAA,EAAQK;AACV,KAAA;AACF;AAEA,EAAA,MAAgBC,QAAQX,KAAAA,EAA0B;AAChD,IAAA,MAAMoC,aAAAA,GAAgB,IAAIC,yBAAAA,CAAAA,CACvB,IAAA,CAAKrC,KAAAA,CAAMsC,SAAAA,EAAWC,MAAAA,IAAUC,sCAAAA,EAA4BC,MAAAA,CAAO,EAAC,CAAA,CAAA;AAGvE,IAAA,MAAMhC,WAAAA,GACJT,KAAAA,CAAM0C,MAAAA,KAAW,IAAA,IAAQ,IAAA,CAAKrC,MAAAA,CAAOsC,OAAAA,EAAO,GACxC,IAAIC,uBAAAA,CAAY5C,KAAAA,CAAM0C,MAAAA,IAAU,aAAA,EAAe;AAAEG,MAAAA,SAAAA,sBAAeC,IAAAA;AAAO,KAAA,CAAA,GACvE,IAAA;AAEN,IAAA,MAAMC,aAAAA,GAAgB;AAAI,MAAA,GAAA,IAAA,CAAK1C,MAAAA,CAAOY,QAAAA;AAAUR,MAAAA;AAAawB,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA;AAEpE,IAAA,MAAMxB,SAAAA,GAAY,IAAIsC,2BAAAA,CAAY;MAChCC,iBAAAA,EAAmB,IAAA;MACnBC,aAAAA,EAAe,GAAA;MACfC,QAAAA,EAAU;AACRC,QAAAA,eAAAA,CAAgBC,IAAAA,EAAI;AAElB,UAAA,MAAMC,mBAAAA,GAAsBD,KAAKE,IAAAA,CAAK,CAACC,QAAQT,aAAAA,CAAcU,QAAAA,CAASD,GAAAA,CAAAA,CAAAA;AACtE,UAAA,IAAIF,mBAAAA,IAAuBA,mBAAAA,KAAwBI,WAAAA,CAAKX,aAAAA,CAAAA,EAAgB;AACtE,YAAA,OAAOO,mBAAAA;AACT;AAEA,UAAA,MAAMK,OAAAA,GAAUN,IAAAA,CAAKO,MAAAA,GAAS,CAAA,IAAKP,IAAAA,CAAKE,IAAAA,CAAK,CAACM,CAAAA,KAAMA,CAAAA,CAAEC,IAAAA,KAASC,gBAAAA,CAAKC,SAAS,CAAA;AAC7E,UAAA,IAAI,CAACL,OAAAA,EAAS;AACZ,YAAA,MAAM,IAAIM,oBAAW,8DAAA,CAAA;AACvB;AACA,UAAA,OAAON,OAAAA;AACT;AACF;KACF,CAAA;AACA,IAAA,MAAMjD,UAAUsB,OAAAA,CAAQ;AAACI,MAAAA,aAAAA;AAAkBW,MAAAA,GAAAA;AAAed,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AAAExB,MAAAA,SAAAA;AAAWD,MAAAA;AAAY,KAAA;AAClC;AAEUoB,EAAAA,KAAAA,CAAMhB,GAAAA,EAAqB;AACnC,IAAA,MAAMqD,SAAkB,EAAA;AAExB,IAAA,KAAA,IAASC,CAAAA,GAAI,CAAA,EAAGA,CAAAA,GAAItD,GAAAA,CAAI+C,MAAAA,IAAU;AAChC,MAAA,MAAMQ,IAAAA,GAAOvD,GAAAA,CAAIwD,SAAAA,CAAUF,CAAAA,CAAAA;AAE3B,MAAA,MAAMG,IAAAA,GAAOC,yBAAcH,IAAAA,EAAM;AAAC,QAAA,iBAAA;AAAmB,QAAA;AAAU,OAAA,EAAA;QAAEI,YAAAA,EAAc;OAAK,CAAA;AACpF,MAAA,IAAI,CAACF,IAAAA,EAAM;AACTJ,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UAAEC,KAAAA,EAAOP,CAAAA;AAAGQ,UAAAA,GAAAA,EAAKR,IAAIC,IAAAA,CAAKR,MAAAA;UAAQhD,OAAAA,EAASwD,IAAAA;UAAMlE,IAAAA,EAAM;SAAO,CAAA;AAC1E,QAAA;AACF;AAEA,MAAA,IAAIoE,IAAAA,CAAKI,QAAQ,CAAA,EAAG;AAClBR,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UACVvE,IAAAA,EAAM,MAAA;UACNwE,KAAAA,EAAOP,CAAAA;AACPQ,UAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKI,KAAAA;AACd9D,UAAAA,OAAAA,EAASwD,IAAAA,CAAKC,SAAAA,CAAU,CAAA,EAAGC,IAAAA,CAAKI,KAAK;SACvC,CAAA;AACF;AACAR,MAAAA,MAAAA,CAAOO,IAAAA,CAAK;QACVvE,IAAAA,EAAM,KAAA;AACNU,QAAAA,OAAAA,EAAS0D,IAAAA,CAAKM,KAAAA;AACdF,QAAAA,KAAAA,EAAOP,IAAIG,IAAAA,CAAKI,KAAAA;AAChBC,QAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKK;OAChB,CAAA;AACAR,MAAAA,CAAAA,IAAKG,IAAAA,CAAKK,GAAAA;AACZ;AAEA,IAAA,OAAO;AACL9D,MAAAA,GAAAA;AACAqD,MAAAA;AACF,KAAA;AACF;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACT7E,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { isTruthy, last } from \"remeda\";\nimport { AssistantMessage, Message, Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport {\n  StreamlitAgentSystemPrompt,\n  StreamlitAgentTemplates,\n} from \"@/agents/experimental/streamlit/prompts.js\";\nimport { TokenMemory } from \"@/memory/tokenMemory.js\";\nimport { findFirstPair } from \"@/internals/helpers/string.js\";\nimport { ChatModel, ChatModelOutput } from \"@/backend/chat.js\";\n\nexport interface StreamlitAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  templates?: Partial<StreamlitAgentTemplates>;\n}\n\nexport interface StreamlitRunInput {\n  prompt: string | null;\n}\n\ninterface Block {\n  name: \"text\" | \"app\";\n  content: string;\n  start: number;\n  end: number;\n}\n\ninterface Result {\n  raw: string;\n  blocks: Block[];\n}\n\nexport interface StreamlitRunOutput {\n  result: Result;\n  message: Message;\n  memory: BaseMemory;\n}\n\nexport interface StreamlitEvents {\n  newToken: Callback<{\n    delta: string;\n    state: Readonly<{\n      content: string;\n    }>;\n    chunk: ChatModelOutput;\n  }>;\n}\n\nexport class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOutput> {\n  public emitter = new Emitter<StreamlitEvents>({\n    namespace: [\"agent\", \"experimental\", \"streamlit\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: StreamlitAgentInput) {\n    super();\n  }\n\n  public get meta(): AgentMeta {\n    return {\n      name: `Streamlit`,\n      tools: [],\n      description: `StreamlitAgent is an experimental meta-app agent that uses \\`Meta LLaMa 3.1 70B\\` to build \\`IBM Granite 3 8B\\`-powered apps using Streamlit -- a simple UI framework for Python.`,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  public get memory() {\n    return this.input.memory;\n  }\n\n  protected async _run(\n    input: StreamlitRunInput,\n    _options: BaseAgentRunOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<StreamlitRunOutput> {\n    const { userMessage, runMemory } = await this.prepare(input);\n\n    let content = \"\";\n    const raw = await this.input.llm\n      .create({\n        stream: true,\n        messages: runMemory.messages,\n        abortSignal: run.signal,\n      })\n      .observe((emitter) => {\n        emitter.on(\"newToken\", async ({ value: chunk }) => {\n          const delta = chunk.getTextContent();\n          if (delta) {\n            content += delta;\n            await run.emitter.emit(\"newToken\", { delta, state: { content }, chunk });\n          }\n        });\n      });\n    const result = this.parse(content || raw.getTextContent());\n\n    const assistantMessage = new AssistantMessage(content);\n    await this.memory.addMany([userMessage, assistantMessage].filter(isTruthy));\n\n    return {\n      result,\n      message: assistantMessage,\n      memory: runMemory,\n    };\n  }\n\n  protected async prepare(input: StreamlitRunInput) {\n    const systemMessage = new SystemMessage(\n      (this.input.templates?.system ?? StreamlitAgentSystemPrompt).render({}),\n    );\n\n    const userMessage =\n      input.prompt !== null || this.memory.isEmpty()\n        ? new UserMessage(input.prompt ?? \"No message.\", { createdAt: new Date() })\n        : null;\n\n    const inputMessages = [...this.memory.messages, userMessage].filter(isTruthy);\n\n    const runMemory = new TokenMemory({\n      capacityThreshold: 0.85,\n      syncThreshold: 0.6,\n      handlers: {\n        removalSelector(msgs) {\n          // First we remove messages from the past conversations\n          const prevConversationMsg = msgs.find((msg) => inputMessages.includes(msg));\n          if (prevConversationMsg && prevConversationMsg !== last(inputMessages)) {\n            return prevConversationMsg;\n          }\n\n          const lastMsg = msgs.length > 3 && msgs.find((m) => m.role === Role.ASSISTANT);\n          if (!lastMsg) {\n            throw new AgentError(\"Cannot fit the current conversation into the context window!\");\n          }\n          return lastMsg;\n        },\n      },\n    });\n    await runMemory.addMany([systemMessage, ...inputMessages].filter(isTruthy));\n\n    return { runMemory, userMessage };\n  }\n\n  protected parse(raw: string): Result {\n    const blocks: Block[] = [];\n\n    for (let i = 0; i < raw.length; ) {\n      const text = raw.substring(i);\n\n      const code = findFirstPair(text, [\"```python-app\\n\", \"\\n```\"], { allowOverlap: true });\n      if (!code) {\n        blocks.push({ start: i, end: i + text.length, content: text, name: \"text\" });\n        break;\n      }\n\n      if (code.start > 0) {\n        blocks.push({\n          name: \"text\",\n          start: i,\n          end: i + code.start,\n          content: text.substring(0, code.start),\n        });\n      }\n      blocks.push({\n        name: \"app\",\n        content: code.inner,\n        start: i + code.start,\n        end: i + code.end,\n      });\n      i += code.end;\n    }\n\n    return {\n      raw,\n      blocks,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: this.input,\n    };\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/experimental/streamlit/agent.ts"],"names":["StreamlitAgent","BaseAgent","emitter","Emitter","namespace","creator","input","meta","name","tools","description","memory","_run","_options","run","userMessage","runMemory","prepare","content","raw","llm","create","stream","messages","abortSignal","signal","observe","on","value","chunk","delta","getTextContent","emit","state","result","parse","assistantMessage","AssistantMessage","addMany","filter","isTruthy","message","systemMessage","SystemMessage","templates","system","StreamlitAgentSystemPrompt","render","prompt","isEmpty","UserMessage","createdAt","Date","inputMessages","TokenMemory","capacityThreshold","syncThreshold","handlers","removalSelector","msgs","prevConversationMsg","find","msg","includes","last","lastMsg","length","m","role","Role","ASSISTANT","AgentError","blocks","i","text","substring","code","findFirstPair","allowOverlap","push","start","end","inner","createSnapshot"],"mappings":";;;;;;;;;;;;AA0DO,MAAMA,uBAAuBC,kBAAAA,CAAAA;EA1DpC;;;;AA2DSC,EAAAA,OAAAA,GAAU,IAAIC,mBAAAA,CAAyB;IAC5CC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA,cAAA;AAAgB,MAAA;;IACrCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAA4B;AACzD,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B,EAAA;AAEA,EAAA,IAAWC,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,CAAA,SAAA,CAAA;AACNC,MAAAA,KAAAA,EAAO,EAAA;MACPC,WAAAA,EAAa,CAAA,iLAAA;AACf,KAAA;AACF,EAAA;AAEA,EAAA,IAAIC,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB,EAAA;AAEA,EAAA,IAAWA,MAAAA,GAAS;AAClB,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB,EAAA;EAEA,MAAgBC,IAAAA,CACdN,KAAAA,EACAO,QAAAA,EACAC,GAAAA,EAC6B;AAC7B,IAAA,MAAM,EAAEC,WAAAA,EAAaC,SAAAA,KAAc,MAAM,IAAA,CAAKC,QAAQX,KAAAA,CAAAA;AAEtD,IAAA,IAAIY,OAAAA,GAAU,EAAA;AACd,IAAA,MAAMC,GAAAA,GAAM,MAAM,IAAA,CAAKb,KAAAA,CAAMc,IAC1BC,MAAAA,CAAO;MACNC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,QAAAA,EAAUP,SAAAA,CAAUO,QAAAA;AACpBC,MAAAA,WAAAA,EAAaV,GAAAA,CAAIW;KACnB,CAAA,CACCC,OAAAA,CAAQ,CAACxB,OAAAA,KAAAA;AACRA,MAAAA,OAAAA,CAAQyB,GAAG,UAAA,EAAY,OAAO,EAAEC,KAAAA,EAAOC,OAAK,KAAE;AAC5C,QAAA,MAAMC,KAAAA,GAAQD,MAAME,cAAAA,EAAc;AAClC,QAAA,IAAID,KAAAA,EAAO;AACTZ,UAAAA,OAAAA,IAAWY,KAAAA;AACX,UAAA,MAAMhB,GAAAA,CAAIZ,OAAAA,CAAQ8B,IAAAA,CAAK,UAAA,EAAY;AAAEF,YAAAA,KAAAA;YAAOG,KAAAA,EAAO;AAAEf,cAAAA;AAAQ,aAAA;AAAGW,YAAAA;WAAM,CAAA;AACxE,QAAA;MACF,CAAA,CAAA;IACF,CAAA,CAAA;AACF,IAAA,MAAMK,SAAS,IAAA,CAAKC,KAAAA,CAAMjB,OAAAA,IAAWC,GAAAA,CAAIY,gBAAc,CAAA;AAEvD,IAAA,MAAMK,gBAAAA,GAAmB,IAAIC,4BAAAA,CAAiBnB,OAAAA,CAAAA;AAC9C,IAAA,MAAM,IAAA,CAAKP,OAAO2B,OAAAA,CAAQ;AAACvB,MAAAA,WAAAA;AAAaqB,MAAAA;AAAkBG,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AACLN,MAAAA,MAAAA;MACAO,OAAAA,EAASL,gBAAAA;MACTzB,MAAAA,EAAQK;AACV,KAAA;AACF,EAAA;AAEA,EAAA,MAAgBC,QAAQX,KAAAA,EAA0B;AAChD,IAAA,MAAMoC,aAAAA,GAAgB,IAAIC,yBAAAA,CAAAA,CACvB,IAAA,CAAKrC,KAAAA,CAAMsC,SAAAA,EAAWC,MAAAA,IAAUC,sCAAAA,EAA4BC,MAAAA,CAAO,EAAC,CAAA,CAAA;AAGvE,IAAA,MAAMhC,WAAAA,GACJT,KAAAA,CAAM0C,MAAAA,KAAW,IAAA,IAAQ,IAAA,CAAKrC,MAAAA,CAAOsC,OAAAA,EAAO,GACxC,IAAIC,uBAAAA,CAAY5C,KAAAA,CAAM0C,MAAAA,IAAU,aAAA,EAAe;AAAEG,MAAAA,SAAAA,sBAAeC,IAAAA;AAAO,KAAA,CAAA,GACvE,IAAA;AAEN,IAAA,MAAMC,aAAAA,GAAgB;AAAI,MAAA,GAAA,IAAA,CAAK1C,MAAAA,CAAOY,QAAAA;AAAUR,MAAAA;AAAawB,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA;AAEpE,IAAA,MAAMxB,SAAAA,GAAY,IAAIsC,2BAAAA,CAAY;MAChCC,iBAAAA,EAAmB,IAAA;MACnBC,aAAAA,EAAe,GAAA;MACfC,QAAAA,EAAU;AACRC,QAAAA,eAAAA,CAAgBC,IAAAA,EAAI;AAElB,UAAA,MAAMC,mBAAAA,GAAsBD,KAAKE,IAAAA,CAAK,CAACC,QAAQT,aAAAA,CAAcU,QAAAA,CAASD,GAAAA,CAAAA,CAAAA;AACtE,UAAA,IAAIF,mBAAAA,IAAuBA,mBAAAA,KAAwBI,WAAAA,CAAKX,aAAAA,CAAAA,EAAgB;AACtE,YAAA,OAAOO,mBAAAA;AACT,UAAA;AAEA,UAAA,MAAMK,OAAAA,GAAUN,IAAAA,CAAKO,MAAAA,GAAS,CAAA,IAAKP,IAAAA,CAAKE,IAAAA,CAAK,CAACM,CAAAA,KAAMA,CAAAA,CAAEC,IAAAA,KAASC,gBAAAA,CAAKC,SAAS,CAAA;AAC7E,UAAA,IAAI,CAACL,OAAAA,EAAS;AACZ,YAAA,MAAM,IAAIM,oBAAW,8DAAA,CAAA;AACvB,UAAA;AACA,UAAA,OAAON,OAAAA;AACT,QAAA;AACF;KACF,CAAA;AACA,IAAA,MAAMjD,UAAUsB,OAAAA,CAAQ;AAACI,MAAAA,aAAAA;AAAkBW,MAAAA,GAAAA;AAAed,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AAAExB,MAAAA,SAAAA;AAAWD,MAAAA;AAAY,KAAA;AAClC,EAAA;AAEUoB,EAAAA,KAAAA,CAAMhB,GAAAA,EAAqB;AACnC,IAAA,MAAMqD,SAAkB,EAAA;AAExB,IAAA,KAAA,IAASC,CAAAA,GAAI,CAAA,EAAGA,CAAAA,GAAItD,GAAAA,CAAI+C,MAAAA,IAAU;AAChC,MAAA,MAAMQ,IAAAA,GAAOvD,GAAAA,CAAIwD,SAAAA,CAAUF,CAAAA,CAAAA;AAE3B,MAAA,MAAMG,IAAAA,GAAOC,yBAAcH,IAAAA,EAAM;AAAC,QAAA,iBAAA;AAAmB,QAAA;AAAU,OAAA,EAAA;QAAEI,YAAAA,EAAc;OAAK,CAAA;AACpF,MAAA,IAAI,CAACF,IAAAA,EAAM;AACTJ,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UAAEC,KAAAA,EAAOP,CAAAA;AAAGQ,UAAAA,GAAAA,EAAKR,IAAIC,IAAAA,CAAKR,MAAAA;UAAQhD,OAAAA,EAASwD,IAAAA;UAAMlE,IAAAA,EAAM;SAAO,CAAA;AAC1E,QAAA;AACF,MAAA;AAEA,MAAA,IAAIoE,IAAAA,CAAKI,QAAQ,CAAA,EAAG;AAClBR,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UACVvE,IAAAA,EAAM,MAAA;UACNwE,KAAAA,EAAOP,CAAAA;AACPQ,UAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKI,KAAAA;AACd9D,UAAAA,OAAAA,EAASwD,IAAAA,CAAKC,SAAAA,CAAU,CAAA,EAAGC,IAAAA,CAAKI,KAAK;SACvC,CAAA;AACF,MAAA;AACAR,MAAAA,MAAAA,CAAOO,IAAAA,CAAK;QACVvE,IAAAA,EAAM,KAAA;AACNU,QAAAA,OAAAA,EAAS0D,IAAAA,CAAKM,KAAAA;AACdF,QAAAA,KAAAA,EAAOP,IAAIG,IAAAA,CAAKI,KAAAA;AAChBC,QAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKK;OAChB,CAAA;AACAR,MAAAA,CAAAA,IAAKG,IAAAA,CAAKK,GAAAA;AACZ,IAAA;AAEA,IAAA,OAAO;AACL9D,MAAAA,GAAAA;AACAqD,MAAAA;AACF,KAAA;AACF,EAAA;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACT7E,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF,EAAA;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { isTruthy, last } from \"remeda\";\nimport { AssistantMessage, Message, Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport {\n  StreamlitAgentSystemPrompt,\n  StreamlitAgentTemplates,\n} from \"@/agents/experimental/streamlit/prompts.js\";\nimport { TokenMemory } from \"@/memory/tokenMemory.js\";\nimport { findFirstPair } from \"@/internals/helpers/string.js\";\nimport { ChatModel, ChatModelOutput } from \"@/backend/chat.js\";\n\nexport interface StreamlitAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  templates?: Partial<StreamlitAgentTemplates>;\n}\n\nexport interface StreamlitRunInput {\n  prompt: string | null;\n}\n\ninterface Block {\n  name: \"text\" | \"app\";\n  content: string;\n  start: number;\n  end: number;\n}\n\ninterface Result {\n  raw: string;\n  blocks: Block[];\n}\n\nexport interface StreamlitRunOutput {\n  result: Result;\n  message: Message;\n  memory: BaseMemory;\n}\n\nexport interface StreamlitEvents {\n  newToken: Callback<{\n    delta: string;\n    state: Readonly<{\n      content: string;\n    }>;\n    chunk: ChatModelOutput;\n  }>;\n}\n\nexport class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOutput> {\n  public emitter = new Emitter<StreamlitEvents>({\n    namespace: [\"agent\", \"experimental\", \"streamlit\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: StreamlitAgentInput) {\n    super();\n  }\n\n  public get meta(): AgentMeta {\n    return {\n      name: `Streamlit`,\n      tools: [],\n      description: `StreamlitAgent is an experimental meta-app agent that uses \\`Meta LLaMa 3.1 70B\\` to build \\`IBM Granite 3 8B\\`-powered apps using Streamlit -- a simple UI framework for Python.`,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  public get memory() {\n    return this.input.memory;\n  }\n\n  protected async _run(\n    input: StreamlitRunInput,\n    _options: BaseAgentRunOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<StreamlitRunOutput> {\n    const { userMessage, runMemory } = await this.prepare(input);\n\n    let content = \"\";\n    const raw = await this.input.llm\n      .create({\n        stream: true,\n        messages: runMemory.messages,\n        abortSignal: run.signal,\n      })\n      .observe((emitter) => {\n        emitter.on(\"newToken\", async ({ value: chunk }) => {\n          const delta = chunk.getTextContent();\n          if (delta) {\n            content += delta;\n            await run.emitter.emit(\"newToken\", { delta, state: { content }, chunk });\n          }\n        });\n      });\n    const result = this.parse(content || raw.getTextContent());\n\n    const assistantMessage = new AssistantMessage(content);\n    await this.memory.addMany([userMessage, assistantMessage].filter(isTruthy));\n\n    return {\n      result,\n      message: assistantMessage,\n      memory: runMemory,\n    };\n  }\n\n  protected async prepare(input: StreamlitRunInput) {\n    const systemMessage = new SystemMessage(\n      (this.input.templates?.system ?? StreamlitAgentSystemPrompt).render({}),\n    );\n\n    const userMessage =\n      input.prompt !== null || this.memory.isEmpty()\n        ? new UserMessage(input.prompt ?? \"No message.\", { createdAt: new Date() })\n        : null;\n\n    const inputMessages = [...this.memory.messages, userMessage].filter(isTruthy);\n\n    const runMemory = new TokenMemory({\n      capacityThreshold: 0.85,\n      syncThreshold: 0.6,\n      handlers: {\n        removalSelector(msgs) {\n          // First we remove messages from the past conversations\n          const prevConversationMsg = msgs.find((msg) => inputMessages.includes(msg));\n          if (prevConversationMsg && prevConversationMsg !== last(inputMessages)) {\n            return prevConversationMsg;\n          }\n\n          const lastMsg = msgs.length > 3 && msgs.find((m) => m.role === Role.ASSISTANT);\n          if (!lastMsg) {\n            throw new AgentError(\"Cannot fit the current conversation into the context window!\");\n          }\n          return lastMsg;\n        },\n      },\n    });\n    await runMemory.addMany([systemMessage, ...inputMessages].filter(isTruthy));\n\n    return { runMemory, userMessage };\n  }\n\n  protected parse(raw: string): Result {\n    const blocks: Block[] = [];\n\n    for (let i = 0; i < raw.length; ) {\n      const text = raw.substring(i);\n\n      const code = findFirstPair(text, [\"```python-app\\n\", \"\\n```\"], { allowOverlap: true });\n      if (!code) {\n        blocks.push({ start: i, end: i + text.length, content: text, name: \"text\" });\n        break;\n      }\n\n      if (code.start > 0) {\n        blocks.push({\n          name: \"text\",\n          start: i,\n          end: i + code.start,\n          content: text.substring(0, code.start),\n        });\n      }\n      blocks.push({\n        name: \"app\",\n        content: code.inner,\n        start: i + code.start,\n        end: i + code.end,\n      });\n      i += code.end;\n    }\n\n    return {\n      raw,\n      blocks,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: this.input,\n    };\n  }\n}\n"]}

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

@@ -1,11 +1,11 @@
 import { GetRunContext, MiddlewareType } from '../../../context.cjs';
-import { a as BaseAgent, c as AgentMeta, B as BaseAgentRunOptions } from '../../../base-BqzN2QwB.cjs';
-import { C as Callback, E as Emitter } from '../../../emitter-UkIkNrek.cjs';
+import { B as BaseAgent, c as AgentMeta, a as BaseAgentRunOptions } from '../../../base-s3ioVIE9.cjs';
+import { E as Emitter, C as Callback } from '../../../emitter-D4OcelZ9.cjs';
 import { BaseMemory } from '../../../memory/base.cjs';
 import { Message, UserMessage } from '../../../backend/message.cjs';
 import { StreamlitAgentTemplates } from './prompts.cjs';
 import { TokenMemory } from '../../../memory/tokenMemory.cjs';
-import { m as ChatModel, n as ChatModelOutput } from '../../../chat-DaqV3woF.cjs';
+import { a as ChatModel, i as ChatModelOutput } from '../../../chat-C20nYj46.cjs';
 import '../../../internals/serializable.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';

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

@@ -1,11 +1,11 @@
 import { GetRunContext, MiddlewareType } from '../../../context.js';
-import { a as BaseAgent, c as AgentMeta, B as BaseAgentRunOptions } from '../../../base-BfhNzUYf.js';
-import { C as Callback, E as Emitter } from '../../../emitter-CL56qab4.js';
+import { B as BaseAgent, c as AgentMeta, a as BaseAgentRunOptions } from '../../../base-BjzyQ8k6.js';
+import { E as Emitter, C as Callback } from '../../../emitter-kHxkUxco.js';
 import { BaseMemory } from '../../../memory/base.js';
 import { Message, UserMessage } from '../../../backend/message.js';
 import { StreamlitAgentTemplates } from './prompts.js';
 import { TokenMemory } from '../../../memory/tokenMemory.js';
-import { m as ChatModel, n as ChatModelOutput } from '../../../chat-DTxmaKP0.js';
+import { a as ChatModel, i as ChatModelOutput } from '../../../chat-CfTNfR60.js';
 import '../../../internals/serializable.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';

package/dist/agents/experimental/streamlit/agent.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/experimental/streamlit/agent.ts"],"names":["StreamlitAgent","BaseAgent","emitter","Emitter","namespace","creator","input","meta","name","tools","description","memory","_run","_options","run","userMessage","runMemory","prepare","content","raw","llm","create","stream","messages","abortSignal","signal","observe","on","value","chunk","delta","getTextContent","emit","state","result","parse","assistantMessage","AssistantMessage","addMany","filter","isTruthy","message","systemMessage","SystemMessage","templates","system","StreamlitAgentSystemPrompt","render","prompt","isEmpty","UserMessage","createdAt","Date","inputMessages","TokenMemory","capacityThreshold","syncThreshold","handlers","removalSelector","msgs","prevConversationMsg","find","msg","includes","last","lastMsg","length","m","role","Role","ASSISTANT","AgentError","blocks","i","text","substring","code","findFirstPair","allowOverlap","push","start","end","inner","createSnapshot"],"mappings":";;;;;;;;;;AA0DO,MAAMA,uBAAuBC,SAAAA,CAAAA;EA1DpC;;;;AA2DSC,EAAAA,OAAAA,GAAU,IAAIC,OAAAA,CAAyB;IAC5CC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA,cAAA;AAAgB,MAAA;;IACrCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAA4B;AACzD,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;AAEA,EAAA,IAAWC,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,CAAA,SAAA,CAAA;AACNC,MAAAA,KAAAA,EAAO,EAAA;MACPC,WAAAA,EAAa,CAAA,iLAAA;AACf,KAAA;AACF;AAEA,EAAA,IAAIC,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB;AAEA,EAAA,IAAWA,MAAAA,GAAS;AAClB,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB;EAEA,MAAgBC,IAAAA,CACdN,KAAAA,EACAO,QAAAA,EACAC,GAAAA,EAC6B;AAC7B,IAAA,MAAM,EAAEC,WAAAA,EAAaC,SAAAA,KAAc,MAAM,IAAA,CAAKC,QAAQX,KAAAA,CAAAA;AAEtD,IAAA,IAAIY,OAAAA,GAAU,EAAA;AACd,IAAA,MAAMC,GAAAA,GAAM,MAAM,IAAA,CAAKb,KAAAA,CAAMc,IAC1BC,MAAAA,CAAO;MACNC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,QAAAA,EAAUP,SAAAA,CAAUO,QAAAA;AACpBC,MAAAA,WAAAA,EAAaV,GAAAA,CAAIW;KACnB,CAAA,CACCC,OAAAA,CAAQ,CAACxB,OAAAA,KAAAA;AACRA,MAAAA,OAAAA,CAAQyB,GAAG,UAAA,EAAY,OAAO,EAAEC,KAAAA,EAAOC,OAAK,KAAE;AAC5C,QAAA,MAAMC,KAAAA,GAAQD,MAAME,cAAAA,EAAc;AAClC,QAAA,IAAID,KAAAA,EAAO;AACTZ,UAAAA,OAAAA,IAAWY,KAAAA;AACX,UAAA,MAAMhB,GAAAA,CAAIZ,OAAAA,CAAQ8B,IAAAA,CAAK,UAAA,EAAY;AAAEF,YAAAA,KAAAA;YAAOG,KAAAA,EAAO;AAAEf,cAAAA;AAAQ,aAAA;AAAGW,YAAAA;WAAM,CAAA;AACxE;OACF,CAAA;KACF,CAAA;AACF,IAAA,MAAMK,SAAS,IAAA,CAAKC,KAAAA,CAAMjB,OAAAA,IAAWC,GAAAA,CAAIY,gBAAc,CAAA;AAEvD,IAAA,MAAMK,gBAAAA,GAAmB,IAAIC,gBAAAA,CAAiBnB,OAAAA,CAAAA;AAC9C,IAAA,MAAM,IAAA,CAAKP,OAAO2B,OAAAA,CAAQ;AAACvB,MAAAA,WAAAA;AAAaqB,MAAAA;AAAkBG,KAAAA,CAAAA,MAAAA,CAAOC,QAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AACLN,MAAAA,MAAAA;MACAO,OAAAA,EAASL,gBAAAA;MACTzB,MAAAA,EAAQK;AACV,KAAA;AACF;AAEA,EAAA,MAAgBC,QAAQX,KAAAA,EAA0B;AAChD,IAAA,MAAMoC,aAAAA,GAAgB,IAAIC,aAAAA,CAAAA,CACvB,IAAA,CAAKrC,KAAAA,CAAMsC,SAAAA,EAAWC,MAAAA,IAAUC,0BAAAA,EAA4BC,MAAAA,CAAO,EAAC,CAAA,CAAA;AAGvE,IAAA,MAAMhC,WAAAA,GACJT,KAAAA,CAAM0C,MAAAA,KAAW,IAAA,IAAQ,IAAA,CAAKrC,MAAAA,CAAOsC,OAAAA,EAAO,GACxC,IAAIC,WAAAA,CAAY5C,KAAAA,CAAM0C,MAAAA,IAAU,aAAA,EAAe;AAAEG,MAAAA,SAAAA,sBAAeC,IAAAA;AAAO,KAAA,CAAA,GACvE,IAAA;AAEN,IAAA,MAAMC,aAAAA,GAAgB;AAAI,MAAA,GAAA,IAAA,CAAK1C,MAAAA,CAAOY,QAAAA;AAAUR,MAAAA;AAAawB,KAAAA,CAAAA,MAAAA,CAAOC,QAAAA,CAAAA;AAEpE,IAAA,MAAMxB,SAAAA,GAAY,IAAIsC,WAAAA,CAAY;MAChCC,iBAAAA,EAAmB,IAAA;MACnBC,aAAAA,EAAe,GAAA;MACfC,QAAAA,EAAU;AACRC,QAAAA,eAAAA,CAAgBC,IAAAA,EAAI;AAElB,UAAA,MAAMC,mBAAAA,GAAsBD,KAAKE,IAAAA,CAAK,CAACC,QAAQT,aAAAA,CAAcU,QAAAA,CAASD,GAAAA,CAAAA,CAAAA;AACtE,UAAA,IAAIF,mBAAAA,IAAuBA,mBAAAA,KAAwBI,IAAAA,CAAKX,aAAAA,CAAAA,EAAgB;AACtE,YAAA,OAAOO,mBAAAA;AACT;AAEA,UAAA,MAAMK,OAAAA,GAAUN,IAAAA,CAAKO,MAAAA,GAAS,CAAA,IAAKP,IAAAA,CAAKE,IAAAA,CAAK,CAACM,CAAAA,KAAMA,CAAAA,CAAEC,IAAAA,KAASC,IAAAA,CAAKC,SAAS,CAAA;AAC7E,UAAA,IAAI,CAACL,OAAAA,EAAS;AACZ,YAAA,MAAM,IAAIM,WAAW,8DAAA,CAAA;AACvB;AACA,UAAA,OAAON,OAAAA;AACT;AACF;KACF,CAAA;AACA,IAAA,MAAMjD,UAAUsB,OAAAA,CAAQ;AAACI,MAAAA,aAAAA;AAAkBW,MAAAA,GAAAA;AAAed,KAAAA,CAAAA,MAAAA,CAAOC,QAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AAAExB,MAAAA,SAAAA;AAAWD,MAAAA;AAAY,KAAA;AAClC;AAEUoB,EAAAA,KAAAA,CAAMhB,GAAAA,EAAqB;AACnC,IAAA,MAAMqD,SAAkB,EAAA;AAExB,IAAA,KAAA,IAASC,CAAAA,GAAI,CAAA,EAAGA,CAAAA,GAAItD,GAAAA,CAAI+C,MAAAA,IAAU;AAChC,MAAA,MAAMQ,IAAAA,GAAOvD,GAAAA,CAAIwD,SAAAA,CAAUF,CAAAA,CAAAA;AAE3B,MAAA,MAAMG,IAAAA,GAAOC,cAAcH,IAAAA,EAAM;AAAC,QAAA,iBAAA;AAAmB,QAAA;AAAU,OAAA,EAAA;QAAEI,YAAAA,EAAc;OAAK,CAAA;AACpF,MAAA,IAAI,CAACF,IAAAA,EAAM;AACTJ,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UAAEC,KAAAA,EAAOP,CAAAA;AAAGQ,UAAAA,GAAAA,EAAKR,IAAIC,IAAAA,CAAKR,MAAAA;UAAQhD,OAAAA,EAASwD,IAAAA;UAAMlE,IAAAA,EAAM;SAAO,CAAA;AAC1E,QAAA;AACF;AAEA,MAAA,IAAIoE,IAAAA,CAAKI,QAAQ,CAAA,EAAG;AAClBR,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UACVvE,IAAAA,EAAM,MAAA;UACNwE,KAAAA,EAAOP,CAAAA;AACPQ,UAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKI,KAAAA;AACd9D,UAAAA,OAAAA,EAASwD,IAAAA,CAAKC,SAAAA,CAAU,CAAA,EAAGC,IAAAA,CAAKI,KAAK;SACvC,CAAA;AACF;AACAR,MAAAA,MAAAA,CAAOO,IAAAA,CAAK;QACVvE,IAAAA,EAAM,KAAA;AACNU,QAAAA,OAAAA,EAAS0D,IAAAA,CAAKM,KAAAA;AACdF,QAAAA,KAAAA,EAAOP,IAAIG,IAAAA,CAAKI,KAAAA;AAChBC,QAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKK;OAChB,CAAA;AACAR,MAAAA,CAAAA,IAAKG,IAAAA,CAAKK,GAAAA;AACZ;AAEA,IAAA,OAAO;AACL9D,MAAAA,GAAAA;AACAqD,MAAAA;AACF,KAAA;AACF;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACT7E,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { isTruthy, last } from \"remeda\";\nimport { AssistantMessage, Message, Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport {\n  StreamlitAgentSystemPrompt,\n  StreamlitAgentTemplates,\n} from \"@/agents/experimental/streamlit/prompts.js\";\nimport { TokenMemory } from \"@/memory/tokenMemory.js\";\nimport { findFirstPair } from \"@/internals/helpers/string.js\";\nimport { ChatModel, ChatModelOutput } from \"@/backend/chat.js\";\n\nexport interface StreamlitAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  templates?: Partial<StreamlitAgentTemplates>;\n}\n\nexport interface StreamlitRunInput {\n  prompt: string | null;\n}\n\ninterface Block {\n  name: \"text\" | \"app\";\n  content: string;\n  start: number;\n  end: number;\n}\n\ninterface Result {\n  raw: string;\n  blocks: Block[];\n}\n\nexport interface StreamlitRunOutput {\n  result: Result;\n  message: Message;\n  memory: BaseMemory;\n}\n\nexport interface StreamlitEvents {\n  newToken: Callback<{\n    delta: string;\n    state: Readonly<{\n      content: string;\n    }>;\n    chunk: ChatModelOutput;\n  }>;\n}\n\nexport class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOutput> {\n  public emitter = new Emitter<StreamlitEvents>({\n    namespace: [\"agent\", \"experimental\", \"streamlit\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: StreamlitAgentInput) {\n    super();\n  }\n\n  public get meta(): AgentMeta {\n    return {\n      name: `Streamlit`,\n      tools: [],\n      description: `StreamlitAgent is an experimental meta-app agent that uses \\`Meta LLaMa 3.1 70B\\` to build \\`IBM Granite 3 8B\\`-powered apps using Streamlit -- a simple UI framework for Python.`,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  public get memory() {\n    return this.input.memory;\n  }\n\n  protected async _run(\n    input: StreamlitRunInput,\n    _options: BaseAgentRunOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<StreamlitRunOutput> {\n    const { userMessage, runMemory } = await this.prepare(input);\n\n    let content = \"\";\n    const raw = await this.input.llm\n      .create({\n        stream: true,\n        messages: runMemory.messages,\n        abortSignal: run.signal,\n      })\n      .observe((emitter) => {\n        emitter.on(\"newToken\", async ({ value: chunk }) => {\n          const delta = chunk.getTextContent();\n          if (delta) {\n            content += delta;\n            await run.emitter.emit(\"newToken\", { delta, state: { content }, chunk });\n          }\n        });\n      });\n    const result = this.parse(content || raw.getTextContent());\n\n    const assistantMessage = new AssistantMessage(content);\n    await this.memory.addMany([userMessage, assistantMessage].filter(isTruthy));\n\n    return {\n      result,\n      message: assistantMessage,\n      memory: runMemory,\n    };\n  }\n\n  protected async prepare(input: StreamlitRunInput) {\n    const systemMessage = new SystemMessage(\n      (this.input.templates?.system ?? StreamlitAgentSystemPrompt).render({}),\n    );\n\n    const userMessage =\n      input.prompt !== null || this.memory.isEmpty()\n        ? new UserMessage(input.prompt ?? \"No message.\", { createdAt: new Date() })\n        : null;\n\n    const inputMessages = [...this.memory.messages, userMessage].filter(isTruthy);\n\n    const runMemory = new TokenMemory({\n      capacityThreshold: 0.85,\n      syncThreshold: 0.6,\n      handlers: {\n        removalSelector(msgs) {\n          // First we remove messages from the past conversations\n          const prevConversationMsg = msgs.find((msg) => inputMessages.includes(msg));\n          if (prevConversationMsg && prevConversationMsg !== last(inputMessages)) {\n            return prevConversationMsg;\n          }\n\n          const lastMsg = msgs.length > 3 && msgs.find((m) => m.role === Role.ASSISTANT);\n          if (!lastMsg) {\n            throw new AgentError(\"Cannot fit the current conversation into the context window!\");\n          }\n          return lastMsg;\n        },\n      },\n    });\n    await runMemory.addMany([systemMessage, ...inputMessages].filter(isTruthy));\n\n    return { runMemory, userMessage };\n  }\n\n  protected parse(raw: string): Result {\n    const blocks: Block[] = [];\n\n    for (let i = 0; i < raw.length; ) {\n      const text = raw.substring(i);\n\n      const code = findFirstPair(text, [\"```python-app\\n\", \"\\n```\"], { allowOverlap: true });\n      if (!code) {\n        blocks.push({ start: i, end: i + text.length, content: text, name: \"text\" });\n        break;\n      }\n\n      if (code.start > 0) {\n        blocks.push({\n          name: \"text\",\n          start: i,\n          end: i + code.start,\n          content: text.substring(0, code.start),\n        });\n      }\n      blocks.push({\n        name: \"app\",\n        content: code.inner,\n        start: i + code.start,\n        end: i + code.end,\n      });\n      i += code.end;\n    }\n\n    return {\n      raw,\n      blocks,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: this.input,\n    };\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/experimental/streamlit/agent.ts"],"names":["StreamlitAgent","BaseAgent","emitter","Emitter","namespace","creator","input","meta","name","tools","description","memory","_run","_options","run","userMessage","runMemory","prepare","content","raw","llm","create","stream","messages","abortSignal","signal","observe","on","value","chunk","delta","getTextContent","emit","state","result","parse","assistantMessage","AssistantMessage","addMany","filter","isTruthy","message","systemMessage","SystemMessage","templates","system","StreamlitAgentSystemPrompt","render","prompt","isEmpty","UserMessage","createdAt","Date","inputMessages","TokenMemory","capacityThreshold","syncThreshold","handlers","removalSelector","msgs","prevConversationMsg","find","msg","includes","last","lastMsg","length","m","role","Role","ASSISTANT","AgentError","blocks","i","text","substring","code","findFirstPair","allowOverlap","push","start","end","inner","createSnapshot"],"mappings":";;;;;;;;;;AA0DO,MAAMA,uBAAuBC,SAAAA,CAAAA;EA1DpC;;;;AA2DSC,EAAAA,OAAAA,GAAU,IAAIC,OAAAA,CAAyB;IAC5CC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA,cAAA;AAAgB,MAAA;;IACrCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAA+BC,KAAAA,EAA4B;AACzD,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B,EAAA;AAEA,EAAA,IAAWC,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,CAAA,SAAA,CAAA;AACNC,MAAAA,KAAAA,EAAO,EAAA;MACPC,WAAAA,EAAa,CAAA,iLAAA;AACf,KAAA;AACF,EAAA;AAEA,EAAA,IAAIC,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKL,MAAMK,MAAAA,GAASA,MAAAA;AACtB,EAAA;AAEA,EAAA,IAAWA,MAAAA,GAAS;AAClB,IAAA,OAAO,KAAKL,KAAAA,CAAMK,MAAAA;AACpB,EAAA;EAEA,MAAgBC,IAAAA,CACdN,KAAAA,EACAO,QAAAA,EACAC,GAAAA,EAC6B;AAC7B,IAAA,MAAM,EAAEC,WAAAA,EAAaC,SAAAA,KAAc,MAAM,IAAA,CAAKC,QAAQX,KAAAA,CAAAA;AAEtD,IAAA,IAAIY,OAAAA,GAAU,EAAA;AACd,IAAA,MAAMC,GAAAA,GAAM,MAAM,IAAA,CAAKb,KAAAA,CAAMc,IAC1BC,MAAAA,CAAO;MACNC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,QAAAA,EAAUP,SAAAA,CAAUO,QAAAA;AACpBC,MAAAA,WAAAA,EAAaV,GAAAA,CAAIW;KACnB,CAAA,CACCC,OAAAA,CAAQ,CAACxB,OAAAA,KAAAA;AACRA,MAAAA,OAAAA,CAAQyB,GAAG,UAAA,EAAY,OAAO,EAAEC,KAAAA,EAAOC,OAAK,KAAE;AAC5C,QAAA,MAAMC,KAAAA,GAAQD,MAAME,cAAAA,EAAc;AAClC,QAAA,IAAID,KAAAA,EAAO;AACTZ,UAAAA,OAAAA,IAAWY,KAAAA;AACX,UAAA,MAAMhB,GAAAA,CAAIZ,OAAAA,CAAQ8B,IAAAA,CAAK,UAAA,EAAY;AAAEF,YAAAA,KAAAA;YAAOG,KAAAA,EAAO;AAAEf,cAAAA;AAAQ,aAAA;AAAGW,YAAAA;WAAM,CAAA;AACxE,QAAA;MACF,CAAA,CAAA;IACF,CAAA,CAAA;AACF,IAAA,MAAMK,SAAS,IAAA,CAAKC,KAAAA,CAAMjB,OAAAA,IAAWC,GAAAA,CAAIY,gBAAc,CAAA;AAEvD,IAAA,MAAMK,gBAAAA,GAAmB,IAAIC,gBAAAA,CAAiBnB,OAAAA,CAAAA;AAC9C,IAAA,MAAM,IAAA,CAAKP,OAAO2B,OAAAA,CAAQ;AAACvB,MAAAA,WAAAA;AAAaqB,MAAAA;AAAkBG,KAAAA,CAAAA,MAAAA,CAAOC,QAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AACLN,MAAAA,MAAAA;MACAO,OAAAA,EAASL,gBAAAA;MACTzB,MAAAA,EAAQK;AACV,KAAA;AACF,EAAA;AAEA,EAAA,MAAgBC,QAAQX,KAAAA,EAA0B;AAChD,IAAA,MAAMoC,aAAAA,GAAgB,IAAIC,aAAAA,CAAAA,CACvB,IAAA,CAAKrC,KAAAA,CAAMsC,SAAAA,EAAWC,MAAAA,IAAUC,0BAAAA,EAA4BC,MAAAA,CAAO,EAAC,CAAA,CAAA;AAGvE,IAAA,MAAMhC,WAAAA,GACJT,KAAAA,CAAM0C,MAAAA,KAAW,IAAA,IAAQ,IAAA,CAAKrC,MAAAA,CAAOsC,OAAAA,EAAO,GACxC,IAAIC,WAAAA,CAAY5C,KAAAA,CAAM0C,MAAAA,IAAU,aAAA,EAAe;AAAEG,MAAAA,SAAAA,sBAAeC,IAAAA;AAAO,KAAA,CAAA,GACvE,IAAA;AAEN,IAAA,MAAMC,aAAAA,GAAgB;AAAI,MAAA,GAAA,IAAA,CAAK1C,MAAAA,CAAOY,QAAAA;AAAUR,MAAAA;AAAawB,KAAAA,CAAAA,MAAAA,CAAOC,QAAAA,CAAAA;AAEpE,IAAA,MAAMxB,SAAAA,GAAY,IAAIsC,WAAAA,CAAY;MAChCC,iBAAAA,EAAmB,IAAA;MACnBC,aAAAA,EAAe,GAAA;MACfC,QAAAA,EAAU;AACRC,QAAAA,eAAAA,CAAgBC,IAAAA,EAAI;AAElB,UAAA,MAAMC,mBAAAA,GAAsBD,KAAKE,IAAAA,CAAK,CAACC,QAAQT,aAAAA,CAAcU,QAAAA,CAASD,GAAAA,CAAAA,CAAAA;AACtE,UAAA,IAAIF,mBAAAA,IAAuBA,mBAAAA,KAAwBI,IAAAA,CAAKX,aAAAA,CAAAA,EAAgB;AACtE,YAAA,OAAOO,mBAAAA;AACT,UAAA;AAEA,UAAA,MAAMK,OAAAA,GAAUN,IAAAA,CAAKO,MAAAA,GAAS,CAAA,IAAKP,IAAAA,CAAKE,IAAAA,CAAK,CAACM,CAAAA,KAAMA,CAAAA,CAAEC,IAAAA,KAASC,IAAAA,CAAKC,SAAS,CAAA;AAC7E,UAAA,IAAI,CAACL,OAAAA,EAAS;AACZ,YAAA,MAAM,IAAIM,WAAW,8DAAA,CAAA;AACvB,UAAA;AACA,UAAA,OAAON,OAAAA;AACT,QAAA;AACF;KACF,CAAA;AACA,IAAA,MAAMjD,UAAUsB,OAAAA,CAAQ;AAACI,MAAAA,aAAAA;AAAkBW,MAAAA,GAAAA;AAAed,KAAAA,CAAAA,MAAAA,CAAOC,QAAAA,CAAAA,CAAAA;AAEjE,IAAA,OAAO;AAAExB,MAAAA,SAAAA;AAAWD,MAAAA;AAAY,KAAA;AAClC,EAAA;AAEUoB,EAAAA,KAAAA,CAAMhB,GAAAA,EAAqB;AACnC,IAAA,MAAMqD,SAAkB,EAAA;AAExB,IAAA,KAAA,IAASC,CAAAA,GAAI,CAAA,EAAGA,CAAAA,GAAItD,GAAAA,CAAI+C,MAAAA,IAAU;AAChC,MAAA,MAAMQ,IAAAA,GAAOvD,GAAAA,CAAIwD,SAAAA,CAAUF,CAAAA,CAAAA;AAE3B,MAAA,MAAMG,IAAAA,GAAOC,cAAcH,IAAAA,EAAM;AAAC,QAAA,iBAAA;AAAmB,QAAA;AAAU,OAAA,EAAA;QAAEI,YAAAA,EAAc;OAAK,CAAA;AACpF,MAAA,IAAI,CAACF,IAAAA,EAAM;AACTJ,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UAAEC,KAAAA,EAAOP,CAAAA;AAAGQ,UAAAA,GAAAA,EAAKR,IAAIC,IAAAA,CAAKR,MAAAA;UAAQhD,OAAAA,EAASwD,IAAAA;UAAMlE,IAAAA,EAAM;SAAO,CAAA;AAC1E,QAAA;AACF,MAAA;AAEA,MAAA,IAAIoE,IAAAA,CAAKI,QAAQ,CAAA,EAAG;AAClBR,QAAAA,MAAAA,CAAOO,IAAAA,CAAK;UACVvE,IAAAA,EAAM,MAAA;UACNwE,KAAAA,EAAOP,CAAAA;AACPQ,UAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKI,KAAAA;AACd9D,UAAAA,OAAAA,EAASwD,IAAAA,CAAKC,SAAAA,CAAU,CAAA,EAAGC,IAAAA,CAAKI,KAAK;SACvC,CAAA;AACF,MAAA;AACAR,MAAAA,MAAAA,CAAOO,IAAAA,CAAK;QACVvE,IAAAA,EAAM,KAAA;AACNU,QAAAA,OAAAA,EAAS0D,IAAAA,CAAKM,KAAAA;AACdF,QAAAA,KAAAA,EAAOP,IAAIG,IAAAA,CAAKI,KAAAA;AAChBC,QAAAA,GAAAA,EAAKR,IAAIG,IAAAA,CAAKK;OAChB,CAAA;AACAR,MAAAA,CAAAA,IAAKG,IAAAA,CAAKK,GAAAA;AACZ,IAAA;AAEA,IAAA,OAAO;AACL9D,MAAAA,GAAAA;AACAqD,MAAAA;AACF,KAAA;AACF,EAAA;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACT7E,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF,EAAA;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { isTruthy, last } from \"remeda\";\nimport { AssistantMessage, Message, Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport {\n  StreamlitAgentSystemPrompt,\n  StreamlitAgentTemplates,\n} from \"@/agents/experimental/streamlit/prompts.js\";\nimport { TokenMemory } from \"@/memory/tokenMemory.js\";\nimport { findFirstPair } from \"@/internals/helpers/string.js\";\nimport { ChatModel, ChatModelOutput } from \"@/backend/chat.js\";\n\nexport interface StreamlitAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  templates?: Partial<StreamlitAgentTemplates>;\n}\n\nexport interface StreamlitRunInput {\n  prompt: string | null;\n}\n\ninterface Block {\n  name: \"text\" | \"app\";\n  content: string;\n  start: number;\n  end: number;\n}\n\ninterface Result {\n  raw: string;\n  blocks: Block[];\n}\n\nexport interface StreamlitRunOutput {\n  result: Result;\n  message: Message;\n  memory: BaseMemory;\n}\n\nexport interface StreamlitEvents {\n  newToken: Callback<{\n    delta: string;\n    state: Readonly<{\n      content: string;\n    }>;\n    chunk: ChatModelOutput;\n  }>;\n}\n\nexport class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOutput> {\n  public emitter = new Emitter<StreamlitEvents>({\n    namespace: [\"agent\", \"experimental\", \"streamlit\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: StreamlitAgentInput) {\n    super();\n  }\n\n  public get meta(): AgentMeta {\n    return {\n      name: `Streamlit`,\n      tools: [],\n      description: `StreamlitAgent is an experimental meta-app agent that uses \\`Meta LLaMa 3.1 70B\\` to build \\`IBM Granite 3 8B\\`-powered apps using Streamlit -- a simple UI framework for Python.`,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  public get memory() {\n    return this.input.memory;\n  }\n\n  protected async _run(\n    input: StreamlitRunInput,\n    _options: BaseAgentRunOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<StreamlitRunOutput> {\n    const { userMessage, runMemory } = await this.prepare(input);\n\n    let content = \"\";\n    const raw = await this.input.llm\n      .create({\n        stream: true,\n        messages: runMemory.messages,\n        abortSignal: run.signal,\n      })\n      .observe((emitter) => {\n        emitter.on(\"newToken\", async ({ value: chunk }) => {\n          const delta = chunk.getTextContent();\n          if (delta) {\n            content += delta;\n            await run.emitter.emit(\"newToken\", { delta, state: { content }, chunk });\n          }\n        });\n      });\n    const result = this.parse(content || raw.getTextContent());\n\n    const assistantMessage = new AssistantMessage(content);\n    await this.memory.addMany([userMessage, assistantMessage].filter(isTruthy));\n\n    return {\n      result,\n      message: assistantMessage,\n      memory: runMemory,\n    };\n  }\n\n  protected async prepare(input: StreamlitRunInput) {\n    const systemMessage = new SystemMessage(\n      (this.input.templates?.system ?? StreamlitAgentSystemPrompt).render({}),\n    );\n\n    const userMessage =\n      input.prompt !== null || this.memory.isEmpty()\n        ? new UserMessage(input.prompt ?? \"No message.\", { createdAt: new Date() })\n        : null;\n\n    const inputMessages = [...this.memory.messages, userMessage].filter(isTruthy);\n\n    const runMemory = new TokenMemory({\n      capacityThreshold: 0.85,\n      syncThreshold: 0.6,\n      handlers: {\n        removalSelector(msgs) {\n          // First we remove messages from the past conversations\n          const prevConversationMsg = msgs.find((msg) => inputMessages.includes(msg));\n          if (prevConversationMsg && prevConversationMsg !== last(inputMessages)) {\n            return prevConversationMsg;\n          }\n\n          const lastMsg = msgs.length > 3 && msgs.find((m) => m.role === Role.ASSISTANT);\n          if (!lastMsg) {\n            throw new AgentError(\"Cannot fit the current conversation into the context window!\");\n          }\n          return lastMsg;\n        },\n      },\n    });\n    await runMemory.addMany([systemMessage, ...inputMessages].filter(isTruthy));\n\n    return { runMemory, userMessage };\n  }\n\n  protected parse(raw: string): Result {\n    const blocks: Block[] = [];\n\n    for (let i = 0; i < raw.length; ) {\n      const text = raw.substring(i);\n\n      const code = findFirstPair(text, [\"```python-app\\n\", \"\\n```\"], { allowOverlap: true });\n      if (!code) {\n        blocks.push({ start: i, end: i + text.length, content: text, name: \"text\" });\n        break;\n      }\n\n      if (code.start > 0) {\n        blocks.push({\n          name: \"text\",\n          start: i,\n          end: i + code.start,\n          content: text.substring(0, code.start),\n        });\n      }\n      blocks.push({\n        name: \"app\",\n        content: code.inner,\n        start: i + code.start,\n        end: i + code.end,\n      });\n      i += code.end;\n    }\n\n    return {\n      raw,\n      blocks,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: this.input,\n    };\n  }\n}\n"]}

package/dist/agents/experimental/streamlit/prompts.cjs

@@ -122,7 +122,7 @@ async def main():
 - Use \`@st.llm_function\` for complex text-based tasks such as summarization, analysis, and keyword extraction.
 - Always define explicit and detailed instructions for the LLM to ensure accurate results.
 - For structured outputs, use a \`dataclass\` to define the return type and describe every field in the instructions.
-- For unstructured output, use \`str\` as the return type, and the fuction will simply return the raw LLM output as a string.
+- For unstructured output, use \`str\` as the return type, and the function will simply return the raw LLM output as a string.
 - **Never** use multiple LLM functions sequentially with the same input; combine tasks into one function instead.
 - Be verbose and specific in instructions for LLM functions, describing each expected output field in detail.
 
@@ -130,7 +130,7 @@ async def main():
 
 - Avoid deprecated functions like \`st.cache\`; use \`st.cache_data\` for serializable data and \`st.cache_resource\` for persistent resources (e.g., database connections).
 - When querying GitHub API, make sure to parse ISO timestamps from strings like \`created_at\`, \`closed_at\`, \`merged_at\`.
-- When using any external API that returns a JSON reply, always check that the needes fields are present in the response.
+- When using any external API that returns a JSON reply, always check that the needed fields are present in the response.
 - Do not simultaneously set value of input element using \`st.session_state.<key>\` and directly using \`st.text_input(key="<key>", value="...")\`. This results in an error.
 - If you need to clear an input field after submitting, use \`with st.form("form_name", clear_on_submit=True):\` to wrap the input elements. Do not modify \`st.session_state.<key>\` after rendering the element, as that will result in an error.
 - When a button that is **not** part of the form modifies \`st.session_state\`, it has to call \`st.rerun()\` afterwards to ensure proper UI refresh.