beeai-framework 0.1.15 → 0.1.17

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

@@ -0,0 +1,39 @@
+import { VercelChatModel } from '../../vercel/backend/chat.cjs';
+import { XaiClientSettings, XaiClient } from './client.cjs';
+import { XaiProvider } from '@ai-sdk/xai';
+import '../../../logger/logger.cjs';
+import 'pino';
+import '../../../errors.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import '../../../internals/serializable.cjs';
+import '../../../chat-CR4yfOR9.cjs';
+import '../../../backend/message.cjs';
+import 'ai';
+import '../../../context.cjs';
+import '../../../emitter-DWEsd-ud.cjs';
+import '../../../internals/helpers/promise.cjs';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../backend/constants.cjs';
+import '../../../tools/base.cjs';
+import 'ajv';
+import '../../../internals/helpers/schema.cjs';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../template.cjs';
+import '../../../backend/client.cjs';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+type XaiParameters = Parameters<XaiProvider["languageModel"]>;
+type XAIChatModelId = NonNullable<XaiParameters[0]>;
+type XAIChatModelSettings = NonNullable<XaiParameters[1]>;
+declare class XAIChatModel extends VercelChatModel {
+    constructor(modelId?: XAIChatModelId, settings?: XAIChatModelSettings, client?: XaiClientSettings | XaiClient);
+}
+
+export { XAIChatModel, type XAIChatModelId, type XAIChatModelSettings };

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

@@ -0,0 +1,39 @@
+import { VercelChatModel } from '../../vercel/backend/chat.js';
+import { XaiClientSettings, XaiClient } from './client.js';
+import { XaiProvider } from '@ai-sdk/xai';
+import '../../../logger/logger.js';
+import 'pino';
+import '../../../errors.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+import '../../../internals/serializable.js';
+import '../../../chat-W9j8T6UU.js';
+import '../../../backend/message.js';
+import 'ai';
+import '../../../context.js';
+import '../../../emitter-C9nwGLt6.js';
+import '../../../internals/helpers/promise.js';
+import 'promise-based-task';
+import '../../../cache/base.js';
+import '../../../backend/constants.js';
+import '../../../tools/base.js';
+import 'ajv';
+import '../../../internals/helpers/schema.js';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../template.js';
+import '../../../backend/client.js';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+type XaiParameters = Parameters<XaiProvider["languageModel"]>;
+type XAIChatModelId = NonNullable<XaiParameters[0]>;
+type XAIChatModelSettings = NonNullable<XaiParameters[1]>;
+declare class XAIChatModel extends VercelChatModel {
+    constructor(modelId?: XAIChatModelId, settings?: XAIChatModelSettings, client?: XaiClientSettings | XaiClient);
+}
+
+export { XAIChatModel, type XAIChatModelId, type XAIChatModelSettings };

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

@@ -0,0 +1,22 @@
+import { VercelChatModel } from '../../vercel/backend/chat.js';
+import { XaiClient } from './client.js';
+import { getEnv } from '../../../internals/env.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class XAIChatModel extends VercelChatModel {
+  static {
+    __name(this, "XAIChatModel");
+  }
+  constructor(modelId = getEnv("XAI_CHAT_MODEL", "grok-3-mini"), settings = {}, client) {
+    const model = XaiClient.ensure(client).instance.languageModel(modelId, settings);
+    super(model);
+  }
+  static {
+    XAIChatModel.register();
+  }
+}
+
+export { XAIChatModel };
+//# sourceMappingURL=chat.js.map
+//# sourceMappingURL=chat.js.map

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

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

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

@@ -0,0 +1,26 @@
+'use strict';
+
+var xai = require('@ai-sdk/xai');
+var client_cjs = require('../../../backend/client.cjs');
+var env_cjs = require('../../../internals/env.cjs');
+var utils_cjs = require('../../vercel/backend/utils.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class XaiClient extends client_cjs.BackendClient {
+  static {
+    __name(this, "XaiClient");
+  }
+  create() {
+    return xai.createXai({
+      ...this.settings,
+      baseURL: this.settings.baseURL || env_cjs.getEnv("XAI_API_BASE_URL"),
+      apiKey: this.settings.apiKey || env_cjs.getEnv("XAI_API_KEY"),
+      fetch: utils_cjs.vercelFetcher(this.settings?.fetch)
+    });
+  }
+}
+
+exports.XaiClient = XaiClient;
+//# sourceMappingURL=client.cjs.map
+//# sourceMappingURL=client.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/adapters/xai/backend/client.ts"],"names":["XaiClient","BackendClient","create","createXai","settings","baseURL","getEnv","apiKey","fetch","vercelFetcher"],"mappings":";;;;;;;;;AAYO,MAAMA,kBAAkBC,wBAAAA,CAAAA;EAZ/B;;;EAaYC,MAAAA,GAAsB;AAC9B,IAAA,OAAOC,aAAAA,CAAU;AACf,MAAA,GAAG,IAAA,CAAKC,QAAAA;AACRC,MAAAA,OAAAA,EAAS,IAAA,CAAKD,QAAAA,CAASC,OAAAA,IAAWC,cAAAA,CAAO,kBAAA,CAAA;AACzCC,MAAAA,MAAAA,EAAQ,IAAA,CAAKH,QAAAA,CAASG,MAAAA,IAAUD,cAAAA,CAAO,aAAA,CAAA;MACvCE,KAAAA,EAAOC,uBAAAA,CAAc,IAAA,CAAKL,QAAAA,EAAUI,KAAAA;KACtC,CAAA;AACF;AACF","file":"client.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { createXai, type XaiProvider, type XaiProviderSettings } from \"@ai-sdk/xai\";\nimport { BackendClient } from \"@/backend/client.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport { vercelFetcher } from \"@/adapters/vercel/backend/utils.js\";\n\nexport type XaiClientSettings = XaiProviderSettings;\n\nexport class XaiClient extends BackendClient<XaiClientSettings, XaiProvider> {\n  protected create(): XaiProvider {\n    return createXai({\n      ...this.settings,\n      baseURL: this.settings.baseURL || getEnv(\"XAI_API_BASE_URL\"),\n      apiKey: this.settings.apiKey || getEnv(\"XAI_API_KEY\"),\n      fetch: vercelFetcher(this.settings?.fetch),\n    });\n  }\n}\n"]}

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

@@ -0,0 +1,17 @@
+import { XaiProviderSettings, XaiProvider } from '@ai-sdk/xai';
+import { BackendClient } from '../../../backend/client.cjs';
+import '../../../internals/serializable.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+type XaiClientSettings = XaiProviderSettings;
+declare class XaiClient extends BackendClient<XaiClientSettings, XaiProvider> {
+    protected create(): XaiProvider;
+}
+
+export { XaiClient, type XaiClientSettings };

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

@@ -0,0 +1,17 @@
+import { XaiProviderSettings, XaiProvider } from '@ai-sdk/xai';
+import { BackendClient } from '../../../backend/client.js';
+import '../../../internals/serializable.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+type XaiClientSettings = XaiProviderSettings;
+declare class XaiClient extends BackendClient<XaiClientSettings, XaiProvider> {
+    protected create(): XaiProvider;
+}
+
+export { XaiClient, type XaiClientSettings };

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

@@ -0,0 +1,24 @@
+import { createXai } from '@ai-sdk/xai';
+import { BackendClient } from '../../../backend/client.js';
+import { getEnv } from '../../../internals/env.js';
+import { vercelFetcher } from '../../vercel/backend/utils.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class XaiClient extends BackendClient {
+  static {
+    __name(this, "XaiClient");
+  }
+  create() {
+    return createXai({
+      ...this.settings,
+      baseURL: this.settings.baseURL || getEnv("XAI_API_BASE_URL"),
+      apiKey: this.settings.apiKey || getEnv("XAI_API_KEY"),
+      fetch: vercelFetcher(this.settings?.fetch)
+    });
+  }
+}
+
+export { XaiClient };
+//# sourceMappingURL=client.js.map
+//# sourceMappingURL=client.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/adapters/xai/backend/client.ts"],"names":["XaiClient","BackendClient","create","createXai","settings","baseURL","getEnv","apiKey","fetch","vercelFetcher"],"mappings":";;;;;;;AAYO,MAAMA,kBAAkBC,aAAAA,CAAAA;EAZ/B;;;EAaYC,MAAAA,GAAsB;AAC9B,IAAA,OAAOC,SAAAA,CAAU;AACf,MAAA,GAAG,IAAA,CAAKC,QAAAA;AACRC,MAAAA,OAAAA,EAAS,IAAA,CAAKD,QAAAA,CAASC,OAAAA,IAAWC,MAAAA,CAAO,kBAAA,CAAA;AACzCC,MAAAA,MAAAA,EAAQ,IAAA,CAAKH,QAAAA,CAASG,MAAAA,IAAUD,MAAAA,CAAO,aAAA,CAAA;MACvCE,KAAAA,EAAOC,aAAAA,CAAc,IAAA,CAAKL,QAAAA,EAAUI,KAAAA;KACtC,CAAA;AACF;AACF","file":"client.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { createXai, type XaiProvider, type XaiProviderSettings } from \"@ai-sdk/xai\";\nimport { BackendClient } from \"@/backend/client.js\";\nimport { getEnv } from \"@/internals/env.js\";\nimport { vercelFetcher } from \"@/adapters/vercel/backend/utils.js\";\n\nexport type XaiClientSettings = XaiProviderSettings;\n\nexport class XaiClient extends BackendClient<XaiClientSettings, XaiProvider> {\n  protected create(): XaiProvider {\n    return createXai({\n      ...this.settings,\n      baseURL: this.settings.baseURL || getEnv(\"XAI_API_BASE_URL\"),\n      apiKey: this.settings.apiKey || getEnv(\"XAI_API_KEY\"),\n      fetch: vercelFetcher(this.settings?.fetch),\n    });\n  }\n}\n"]}

package/dist/{agent-CgsoqVL5.d.ts → agent-CUzHyTQd.d.ts}

@@ -1,12 +1,12 @@
 import { GetRunContext, RunContext } from './context.js';
-import { b as AgentMeta, a as BaseAgent } from './base-BOCDDBvG.js';
+import { b as AgentMeta, a as BaseAgent } from './base-BrMAtrc_.js';
 import { AnyTool } from './tools/base.js';
 import { BaseMemory } from './memory/base.js';
-import { E as Emitter } from './emitter-C3dO-s2P.js';
+import { E as Emitter } from './emitter-C9nwGLt6.js';
 import { ReActAgentRunOptions, ReActAgentRunIteration, ReActAgentCallbacks, ReActAgentParserInput, ReActAgentMeta, ReActAgentRunInput, ReActAgentIterationToolResult, ReActAgentTemplates, ReActAgentExecutionConfig, ReActAgentRunOutput } from './agents/react/types.js';
 import { LinePrefixParser } from './parsers/linePrefix.js';
 import { Serializable } from './internals/serializable.js';
-import { m as ChatModel } from './chat-BdN8hGWn.js';
+import { m as ChatModel } from './chat-W9j8T6UU.js';
 import { RetryCounter } from './internals/helpers/counter.js';
 
 interface ReActAgentRunnerLLMInput {

package/dist/{agent-Ca5UE2P1.d.cts → agent-Cqh2zuKJ.d.cts}

@@ -1,12 +1,12 @@
 import { GetRunContext, RunContext } from './context.cjs';
-import { b as AgentMeta, a as BaseAgent } from './base-szR2Izbt.cjs';
+import { b as AgentMeta, a as BaseAgent } from './base-Bc-Xv1jG.cjs';
 import { AnyTool } from './tools/base.cjs';
 import { BaseMemory } from './memory/base.cjs';
-import { E as Emitter } from './emitter-D5Mu0EEH.cjs';
+import { E as Emitter } from './emitter-DWEsd-ud.cjs';
 import { ReActAgentRunOptions, ReActAgentRunIteration, ReActAgentCallbacks, ReActAgentParserInput, ReActAgentMeta, ReActAgentRunInput, ReActAgentIterationToolResult, ReActAgentTemplates, ReActAgentExecutionConfig, ReActAgentRunOutput } from './agents/react/types.cjs';
 import { LinePrefixParser } from './parsers/linePrefix.cjs';
 import { Serializable } from './internals/serializable.cjs';
-import { m as ChatModel } from './chat-DvMzAHj6.cjs';
+import { m as ChatModel } from './chat-CR4yfOR9.cjs';
 import { RetryCounter } from './internals/helpers/counter.cjs';
 
 interface ReActAgentRunnerLLMInput {

package/dist/agents/base.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","destroy","emitter","meta","name","constructor","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;AAuBO,MAAMA,mBAAmBC,yBAAAA,CAAAA;EAvBhC;;;AAuBgD;AAMzC,MAAeC,kBAIZC,6BAAAA,CAAAA;EAjCV;;;EAkCYC,SAAY,GAAA,KAAA;EAIfC,GACF,CAAA,GAAA,CAACC,KAAOC,EAAAA,OAAAA,CAGX,EAAA;AACA,IAAA,IAAI,KAAKH,SAAW,EAAA;AAClB,MAAM,MAAA,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAOQ,OAAAA,sBAAAA,CAAWC,MAChB,IACA,EAAA;AAAEC,MAAAA,MAAAA,EAAQH,OAASG,EAAAA,MAAAA;MAAQC,MAAQ,EAAA;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAI,IAAA;AACF,QAAA,IAAA,CAAKR,SAAY,GAAA,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKS,IAAKP,CAAAA,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAG,EAAA;AACV,QAAA,IAAIA,aAAad,UAAY,EAAA;AAC3B,UAAMc,MAAAA,CAAAA;SACD,MAAA;AACL,UAAM,MAAA,IAAId,WAAW,CAAuC,mCAAA,CAAA,EAAA;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAY,GAAA,KAAA;AACnB;KACF,CAAA;AAEJ;EAQAW,OAAU,GAAA;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAO,EAAA;AACtB;AAKA,EAAA,IAAWE,IAAkB,GAAA;AAC3B,IAAO,OAAA;MACLC,IAAM,EAAA,IAAA,CAAKC,YAAYD,IAAQ,IAAA,WAAA;MAC/BE,WAAa,EAAA,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;MAAElB,SAAW,EAAA,KAAA;AAAOY,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAAQ,KAAA;AACnD;AAEAO,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    );\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return { isRunning: false, emitter: this.emitter };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","destroy","emitter","meta","name","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;AAYO,MAAMA,mBAAmBC,yBAAAA,CAAAA;EAZhC;;;AAYgD;AAMzC,MAAeC,kBAIZC,6BAAAA,CAAAA;EAtBV;;;EAuBYC,SAAAA,GAAY,KAAA;EAIfC,GAAAA,CAAAA,GACF,CAACC,KAAAA,EAAOC,OAAAA,CAAAA,EAGX;AACA,IAAA,IAAI,KAAKH,SAAAA,EAAW;AAClB,MAAA,MAAM,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAA,OAAOQ,sBAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAA,IAAI;AACF,QAAA,IAAA,CAAKR,SAAAA,GAAY,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKS,IAAAA,CAAKP,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAAA,EAAG;AACV,QAAA,IAAIA,aAAad,UAAAA,EAAY;AAC3B,UAAA,MAAMc,CAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAId,WAAW,CAAA,mCAAA,CAAA,EAAuC;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAAA,GAAY,KAAA;AACnB;KACF,CAAA;AAEJ;EAQAW,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAAA,EAAO;AACtB;AAKA,EAAA,IAAWE,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,IAAA,CAAK,YAAYA,IAAAA,IAAQ,WAAA;MAC/BC,WAAAA,EAAa,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEjB,SAAAA,EAAW,KAAA;AAAOY,MAAAA,OAAAA,EAAS,IAAA,CAAKA;AAAQ,KAAA;AACnD;AAEAM,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    );\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return { isRunning: false, emitter: this.emitter };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

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-szR2Izbt.cjs';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-Bc-Xv1jG.cjs';
 import '../internals/serializable.cjs';
-import '../emitter-D5Mu0EEH.cjs';
+import '../emitter-DWEsd-ud.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-BOCDDBvG.js';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-BrMAtrc_.js';
 import '../internals/serializable.js';
-import '../emitter-C3dO-s2P.js';
+import '../emitter-C9nwGLt6.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":["base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","destroy","emitter","meta","name","constructor","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;AAuBO,MAAMA,mBAAmBC,cAAAA,CAAAA;EAvBhC;;;AAuBgD;AAMzC,MAAeC,kBAIZC,YAAAA,CAAAA;EAjCV;;;EAkCYC,SAAY,GAAA,KAAA;EAIfC,GACF,CAAA,GAAA,CAACC,KAAOC,EAAAA,OAAAA,CAGX,EAAA;AACA,IAAA,IAAI,KAAKH,SAAW,EAAA;AAClB,MAAM,MAAA,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAOQ,OAAAA,UAAAA,CAAWC,MAChB,IACA,EAAA;AAAEC,MAAAA,MAAAA,EAAQH,OAASG,EAAAA,MAAAA;MAAQC,MAAQ,EAAA;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAI,IAAA;AACF,QAAA,IAAA,CAAKR,SAAY,GAAA,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKS,IAAKP,CAAAA,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAG,EAAA;AACV,QAAA,IAAIA,aAAad,UAAY,EAAA;AAC3B,UAAMc,MAAAA,CAAAA;SACD,MAAA;AACL,UAAM,MAAA,IAAId,WAAW,CAAuC,mCAAA,CAAA,EAAA;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAY,GAAA,KAAA;AACnB;KACF,CAAA;AAEJ;EAQAW,OAAU,GAAA;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAO,EAAA;AACtB;AAKA,EAAA,IAAWE,IAAkB,GAAA;AAC3B,IAAO,OAAA;MACLC,IAAM,EAAA,IAAA,CAAKC,YAAYD,IAAQ,IAAA,WAAA;MAC/BE,WAAa,EAAA,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;MAAElB,SAAW,EAAA,KAAA;AAAOY,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAAQ,KAAA;AACnD;AAEAO,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    );\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return { isRunning: false, emitter: this.emitter };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","destroy","emitter","meta","name","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;AAYO,MAAMA,mBAAmBC,cAAAA,CAAAA;EAZhC;;;AAYgD;AAMzC,MAAeC,kBAIZC,YAAAA,CAAAA;EAtBV;;;EAuBYC,SAAAA,GAAY,KAAA;EAIfC,GAAAA,CAAAA,GACF,CAACC,KAAAA,EAAOC,OAAAA,CAAAA,EAGX;AACA,IAAA,IAAI,KAAKH,SAAAA,EAAW;AAClB,MAAA,MAAM,IAAIJ,WAAW,2BAAA,CAAA;AACvB;AAEA,IAAA,OAAOQ,UAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,OAAAA,KAAAA;AACL,MAAA,IAAI;AACF,QAAA,IAAA,CAAKR,SAAAA,GAAY,IAAA;AAEjB,QAAA,OAAO,MAAM,IAAA,CAAKS,IAAAA,CAAKP,KAAAA,EAAOC,SAASK,OAAAA,CAAAA;AACzC,OAAA,CAAA,OAASE,CAAAA,EAAG;AACV,QAAA,IAAIA,aAAad,UAAAA,EAAY;AAC3B,UAAA,MAAMc,CAAAA;SACR,MAAO;AACL,UAAA,MAAM,IAAId,WAAW,CAAA,mCAAA,CAAA,EAAuC;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjE;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAAA,GAAY,KAAA;AACnB;KACF,CAAA;AAEJ;EAQAW,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKC,QAAQD,OAAAA,EAAO;AACtB;AAKA,EAAA,IAAWE,IAAAA,GAAkB;AAC3B,IAAA,OAAO;MACLC,IAAAA,EAAM,IAAA,CAAK,YAAYA,IAAAA,IAAQ,WAAA;MAC/BC,WAAAA,EAAa,EAAA;AACbC,MAAAA,KAAAA,EAAO;AACT,KAAA;AACF;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEjB,SAAAA,EAAW,KAAA;AAAOY,MAAAA,OAAAA,EAAS,IAAA,CAAKA;AAAQ,KAAA;AACnD;AAEAM,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"base.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { FrameworkError } from \"@/errors.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { GetRunContext, RunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\n\nexport class AgentError extends FrameworkError {}\n\nexport interface BaseAgentRunOptions {\n  signal?: AbortSignal;\n}\n\nexport abstract class BaseAgent<\n  TInput,\n  TOutput,\n  TOptions extends BaseAgentRunOptions = BaseAgentRunOptions,\n> extends Serializable {\n  protected isRunning = false;\n\n  public abstract readonly emitter: Emitter<unknown>;\n\n  public run(\n    ...[input, options]: Partial<TOptions> extends TOptions\n      ? [input: TInput, options?: TOptions]\n      : [input: TInput, options: TOptions]\n  ) {\n    if (this.isRunning) {\n      throw new AgentError(\"Agent is already running!\");\n    }\n\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [input, options] as const },\n      async (context) => {\n        try {\n          this.isRunning = true;\n          // @ts-expect-error\n          return await this._run(input, options, context);\n        } catch (e) {\n          if (e instanceof AgentError) {\n            throw e;\n          } else {\n            throw new AgentError(`The Agent has encountered an error.`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    );\n  }\n\n  protected abstract _run(\n    input: TInput,\n    options: TOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<TOutput>;\n\n  destroy() {\n    this.emitter.destroy();\n  }\n\n  public abstract set memory(memory: BaseMemory);\n  public abstract get memory(): BaseMemory;\n\n  public get meta(): AgentMeta {\n    return {\n      name: this.constructor.name ?? \"BaseAgent\",\n      description: \"\",\n      tools: [],\n    };\n  }\n\n  createSnapshot() {\n    return { isRunning: false, emitter: this.emitter };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

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

@@ -14,15 +14,15 @@ class RePlanAgent extends base_cjs.BaseAgent {
     __name(this, "RePlanAgent");
   }
   input;
-  emitter;
+  emitter = emitter_cjs.Emitter.root.child({
+    namespace: [
+      "agent",
+      "rePlan"
+    ],
+    creator: this
+  });
   constructor(input) {
-    super(), this.input = input, this.emitter = emitter_cjs.Emitter.root.child({
-      namespace: [
-        "agent",
-        "rePlan"
-      ],
-      creator: this
-    });
+    super(), this.input = input;
   }
   async _run(input, _options, context) {
     if (input.prompt !== null) {

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

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_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":";;;;;;;;;;;AA4DO,MAAMA,oBAAoBC,kBAAAA,CAAAA;EA5DjC;;;;AA6DSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,mBAAAA,CAAQC,KAAKC,KAAoB,CAAA;MAChDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC0B,EAAA;AAC1B,IAAIR,IAAAA,KAAAA,CAAMS,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,KAAKC,MAAOC,CAAAA,GAAAA,CAAI,IAAIC,uBAAYZ,CAAAA,KAAAA,CAAMS,MAAM,CAAA,CAAA;AACpD;AAEA,IAAA,MAAMI,MAAS,GAAA,MAAM,IAAKC,CAAAA,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAoCC,GAAAA,MAAAA;AACxC,IAAA,OAAO,CAACD,YAAc,EAAA;AACpB,MAAME,MAAAA,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAG,EAAA;AAE9B,MAAID,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,SAAW,EAAA;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,uBAAAA,CAAYK,KAAME,CAAAA,QAAAA,CAASE,OAAO,CAAA;OAC5CJ,MAAAA,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,MAAQ,EAAA;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAMN,CAAAA,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAOC,CAAAA,GAAAA,CAClB,IAAIc,4BAAAA,CACFC,kCAAsBC,MAAO,CAAA;UAC3BC,OAASC,EAAAA,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN;AACF;AAEA,IAAM,MAAA,IAAA,CAAKZ,MAAOC,CAAAA,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAO,OAAA;MACLM,OAASN,EAAAA,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAOH,CAAAA;AAC7B,KAAA;AACF;AAEA,EAAA,MAAgBI,aAAaN,OAA8B,EAAA;AACzD,IAAME,MAAAA,MAAAA,GAAS,IAAIsB,2CAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAOuB,CAAAA,OAAAA,CAAQ,IAAKvB,CAAAA,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAS,GAAA,MAAMC,oCAAyB,CAAA,IAAA,CAAKpC,MAAMuB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAW,GAAA,MAAM,IAAKrC,CAAAA,KAAAA,CAAMsC,IAAIC,eAAgB,CAAA;AACpDJ,QAAAA,MAAAA,EAAQA,MAAOK,CAAAA,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAQkC,CAAAA,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAOwB,CAAAA;OACnB,CAAA;AACA,MAAMxB,MAAAA,MAAAA,CAAOC,IAAI,IAAIc,4BAAAA,CAAiBI,KAAKC,SAAUO,CAAAA,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAM7B,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAE1B,QAAAA,KAAAA,EAAOoB,QAASO,CAAAA;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAASO,CAAAA,MAAAA;KATN,EAAA,KAAA,CAAA;AAYZ,IAAMrB,MAAAA,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAQC,CAAAA,GAAAA,CACnBtB,KAAMuB,CAAAA,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAMC,MAAAA,IAAAA,GAAO,IAAKjD,CAAAA,KAAAA,CAAMuB,KAAM2B,CAAAA,IAAAA,CAAK,CAACD,KAASA,KAAAA,KAAAA,CAAKE,IAASH,KAAAA,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAM,EAAA;AACT,UAAA,MAAM,IAAIG,mBAAAA,CAAW,CAAQJ,KAAAA,EAAAA,IAAAA,CAAKG,IAAI,CAAkB,gBAAA,CAAA,CAAA;AAC1D;AAEA,QAAA,MAAME,IAAO,GAAA;UAAErD,KAAOgD,EAAAA,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAMhB,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;UAAEvB,IAAM,EAAA,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAI,IAAA;AACF,UAAA,MAAMC,MAAS,GAAA,MAAML,IAAK/B,CAAAA,GAAAA,CAAI8B,KAAKhD,KAAO,EAAA;AAAE0C,YAAAA,MAAAA,EAAQlC,OAAQkC,CAAAA;AAAO,WAAA,EAAGlC,OAAQ,CAAA;YAC5E,CAAC+C,eAAAA,CAAKC,WAAYC,CAAAA,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAMF,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAOA,OAAAA,MAAAA;AACT,SAAA,CAAA,OAASI,KAAO,EAAA;AACd,UAAMlD,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAMA,MAAAA,KAAAA;AACR;AACF,OAAA,CAAA,CAAA;KApBU,EAAA,OAAA,CAAA;AAwBd,IAAO,OAAA;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF;AAEA,EAAA,IAAIb,MAAS,GAAA;AACX,IAAA,OAAO,KAAKV,KAAMU,CAAAA,MAAAA;AACpB;AAEA,EAAA,IAAIA,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKV,MAAMU,MAASA,GAAAA,MAAAA;AACtB;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\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;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"]}

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

@@ -1,12 +1,12 @@
-import { C as Callback, E as Emitter } from '../../../emitter-D5Mu0EEH.cjs';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-szR2Izbt.cjs';
+import { C as Callback, E as Emitter } from '../../../emitter-DWEsd-ud.cjs';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-Bc-Xv1jG.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-DvMzAHj6.cjs';
+import { m as ChatModel } from '../../../chat-CR4yfOR9.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';
@@ -26,18 +26,7 @@ import 'pino';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface RePlanRunInput {

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

@@ -1,12 +1,12 @@
-import { C as Callback, E as Emitter } from '../../../emitter-C3dO-s2P.js';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BOCDDBvG.js';
+import { C as Callback, E as Emitter } from '../../../emitter-C9nwGLt6.js';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-BrMAtrc_.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-BdN8hGWn.js';
+import { m as ChatModel } from '../../../chat-W9j8T6UU.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';
@@ -26,18 +26,7 @@ import 'pino';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface RePlanRunInput {

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

@@ -12,15 +12,15 @@ class RePlanAgent extends BaseAgent {
     __name(this, "RePlanAgent");
   }
   input;
-  emitter;
+  emitter = Emitter.root.child({
+    namespace: [
+      "agent",
+      "rePlan"
+    ],
+    creator: this
+  });
   constructor(input) {
-    super(), this.input = input, this.emitter = Emitter.root.child({
-      namespace: [
-        "agent",
-        "rePlan"
-      ],
-      creator: this
-    });
+    super(), this.input = input;
   }
   async _run(input, _options, context) {
     if (input.prompt !== null) {

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

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_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":";;;;;;;;;AA4DO,MAAMA,oBAAoBC,SAAAA,CAAAA;EA5DjC;;;;AA6DSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,OAAAA,CAAQC,KAAKC,KAAoB,CAAA;MAChDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC0B,EAAA;AAC1B,IAAIR,IAAAA,KAAAA,CAAMS,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,KAAKC,MAAOC,CAAAA,GAAAA,CAAI,IAAIC,WAAYZ,CAAAA,KAAAA,CAAMS,MAAM,CAAA,CAAA;AACpD;AAEA,IAAA,MAAMI,MAAS,GAAA,MAAM,IAAKC,CAAAA,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAoCC,GAAAA,MAAAA;AACxC,IAAA,OAAO,CAACD,YAAc,EAAA;AACpB,MAAME,MAAAA,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAG,EAAA;AAE9B,MAAID,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,SAAW,EAAA;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,WAAAA,CAAYK,KAAME,CAAAA,QAAAA,CAASE,OAAO,CAAA;OAC5CJ,MAAAA,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,MAAQ,EAAA;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAMN,CAAAA,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAOC,CAAAA,GAAAA,CAClB,IAAIc,gBAAAA,CACFC,sBAAsBC,MAAO,CAAA;UAC3BC,OAASC,EAAAA,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN;AACF;AAEA,IAAM,MAAA,IAAA,CAAKZ,MAAOC,CAAAA,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAO,OAAA;MACLM,OAASN,EAAAA,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAOH,CAAAA;AAC7B,KAAA;AACF;AAEA,EAAA,MAAgBI,aAAaN,OAA8B,EAAA;AACzD,IAAME,MAAAA,MAAAA,GAAS,IAAIsB,mBAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAOuB,CAAAA,OAAAA,CAAQ,IAAKvB,CAAAA,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAS,GAAA,MAAMC,wBAAyB,CAAA,IAAA,CAAKpC,MAAMuB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAW,GAAA,MAAM,IAAKrC,CAAAA,KAAAA,CAAMsC,IAAIC,eAAgB,CAAA;AACpDJ,QAAAA,MAAAA,EAAQA,MAAOK,CAAAA,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAQkC,CAAAA,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAOwB,CAAAA;OACnB,CAAA;AACA,MAAMxB,MAAAA,MAAAA,CAAOC,IAAI,IAAIc,gBAAAA,CAAiBI,KAAKC,SAAUO,CAAAA,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAM7B,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAE1B,QAAAA,KAAAA,EAAOoB,QAASO,CAAAA;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAASO,CAAAA,MAAAA;KATN,EAAA,KAAA,CAAA;AAYZ,IAAMrB,MAAAA,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAQC,CAAAA,GAAAA,CACnBtB,KAAMuB,CAAAA,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAMC,MAAAA,IAAAA,GAAO,IAAKjD,CAAAA,KAAAA,CAAMuB,KAAM2B,CAAAA,IAAAA,CAAK,CAACD,KAASA,KAAAA,KAAAA,CAAKE,IAASH,KAAAA,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAM,EAAA;AACT,UAAA,MAAM,IAAIG,UAAAA,CAAW,CAAQJ,KAAAA,EAAAA,IAAAA,CAAKG,IAAI,CAAkB,gBAAA,CAAA,CAAA;AAC1D;AAEA,QAAA,MAAME,IAAO,GAAA;UAAErD,KAAOgD,EAAAA,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAMhB,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;UAAEvB,IAAM,EAAA,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAI,IAAA;AACF,UAAA,MAAMC,MAAS,GAAA,MAAML,IAAK/B,CAAAA,GAAAA,CAAI8B,KAAKhD,KAAO,EAAA;AAAE0C,YAAAA,MAAAA,EAAQlC,OAAQkC,CAAAA;AAAO,WAAA,EAAGlC,OAAQ,CAAA;YAC5E,CAAC+C,IAAAA,CAAKC,WAAYC,CAAAA,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAMF,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAOA,OAAAA,MAAAA;AACT,SAAA,CAAA,OAASI,KAAO,EAAA;AACd,UAAMlD,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAMA,MAAAA,KAAAA;AACR;AACF,OAAA,CAAA,CAAA;KApBU,EAAA,OAAA,CAAA;AAwBd,IAAO,OAAA;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF;AAEA,EAAA,IAAIb,MAAS,GAAA;AACX,IAAA,OAAO,KAAKV,KAAMU,CAAAA,MAAAA;AACpB;AAEA,EAAA,IAAIA,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKV,MAAMU,MAASA,GAAAA,MAAAA;AACtB;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\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;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"]}