beeai-framework 0.1.2

package/dist/adapters/watsonx/backend/embedding.js

@@ -0,0 +1,68 @@
+import { WatsonxClient } from './client.js';
+import { EmbeddingModel } from '../../../backend/embedding.js';
+import { Emitter } from '../../../emitter/emitter.js';
+import { getEnv } from '../../../internals/env.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class WatsonxEmbeddingModel extends EmbeddingModel {
+  static {
+    __name(this, "WatsonxEmbeddingModel");
+  }
+  modelId;
+  parameters;
+  client;
+  emitter;
+  get providerId() {
+    return "watsonx";
+  }
+  constructor(modelId = getEnv("WATSONX_EMBEDDING_MODEL", "ibm/granite-embedding-107m-multilingual"), parameters = {}, client) {
+    super(), this.modelId = modelId, this.parameters = parameters;
+    this.client = WatsonxClient.ensure(client);
+    this.emitter = Emitter.root.child({
+      namespace: [
+        "backend",
+        "watsonx",
+        "embedding"
+      ],
+      creator: this
+    });
+  }
+  async _create(input) {
+    const response = await this.client.instance.embedText({
+      modelId: this.modelId,
+      spaceId: this.client.spaceId,
+      projectId: this.client.projectId,
+      inputs: input.values,
+      parameters: {
+        return_options: {
+          input_text: true
+        },
+        ...this.parameters
+      }
+    });
+    const embeddings = response.result.results.map((e) => e.embedding);
+    const values = response.result.results.map((e, i) => e.input || input.values.at(i));
+    return {
+      embeddings,
+      values,
+      usage: {
+        tokens: response.result.input_token_count
+      }
+    };
+  }
+  createSnapshot() {
+    return {
+      ...super.createSnapshot(),
+      modelId: this.modelId,
+      client: this.client
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+
+export { WatsonxEmbeddingModel };
+//# sourceMappingURL=embedding.js.map
+//# sourceMappingURL=embedding.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/adapters/watsonx/backend/embedding.ts"],"names":["WatsonxEmbeddingModel","EmbeddingModel","client","emitter","providerId","constructor","modelId","getEnv","parameters","WatsonxClient","ensure","Emitter","root","child","namespace","creator","_create","input","response","instance","embedText","spaceId","projectId","inputs","values","return_options","input_text","embeddings","result","results","map","e","embedding","i","at","usage","tokens","input_token_count","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;AA6BO,MAAMA,8BAA8BC,cAAAA,CAAAA;EA7B3C;;;;;AA8BqBC,EAAAA,MAAAA;AACHC,EAAAA,OAAAA;AAEhB,EAAA,IAAIC,UAAa,GAAA;AACf,IAAO,OAAA,SAAA;AACT;EAEAC,WACkBC,CAAAA,OAAAA,GAAkBC,OAChC,yBACA,EAAA,yCAAA,GAEcC,UAA8C,GAAA,IAC9DN,MACA,EAAA;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CAPWI,OAAAA,GAAAA,OAAAA,EAAAA,KAIAE,UAAAA,GAAAA,UAAAA;AAIhB,IAAKN,IAAAA,CAAAA,MAAAA,GAASO,aAAcC,CAAAA,MAAAA,CAAOR,MAAAA,CAAAA;AACnC,IAAKC,IAAAA,CAAAA,OAAAA,GAAUQ,OAAQC,CAAAA,IAAAA,CAAKC,KAAM,CAAA;MAChCC,SAAW,EAAA;AAAC,QAAA,SAAA;AAAW,QAAA,SAAA;AAAW,QAAA;;MAClCC,OAAS,EAAA;KACX,CAAA;AACF;AAEA,EAAA,MAAgBC,QAAQC,KAA2D,EAAA;AACjF,IAAA,MAAMC,QAAW,GAAA,MAAM,IAAKhB,CAAAA,MAAAA,CAAOiB,SAASC,SAAU,CAAA;AACpDd,MAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;AACde,MAAAA,OAAAA,EAAS,KAAKnB,MAAOmB,CAAAA,OAAAA;AACrBC,MAAAA,SAAAA,EAAW,KAAKpB,MAAOoB,CAAAA,SAAAA;AACvBC,MAAAA,MAAAA,EAAQN,KAAMO,CAAAA,MAAAA;MACdhB,UAAY,EAAA;QACViB,cAAgB,EAAA;UAAEC,UAAY,EAAA;AAAK,SAAA;AACnC,QAAA,GAAG,IAAKlB,CAAAA;AACV;KACF,CAAA;AAEA,IAAMmB,MAAAA,UAAAA,GAAaT,SAASU,MAAOC,CAAAA,OAAAA,CAAQC,IAAI,CAACC,CAAAA,KAAMA,EAAEC,SAAS,CAAA;AACjE,IAAA,MAAMR,MAASN,GAAAA,QAAAA,CAASU,MAAOC,CAAAA,OAAAA,CAAQC,IAAI,CAACC,CAAAA,EAAGE,CAAMF,KAAAA,CAAAA,CAAEd,KAASA,IAAAA,KAAAA,CAAMO,MAAOU,CAAAA,EAAAA,CAAGD,CAAAA,CAAAA,CAAAA;AAEhF,IAAO,OAAA;AACLN,MAAAA,UAAAA;AACAH,MAAAA,MAAAA;MACAW,KAAO,EAAA;AACLC,QAAAA,MAAAA,EAAQlB,SAASU,MAAOS,CAAAA;AAC1B;AACF,KAAA;AACF;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACThC,MAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;AACdJ,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA;AACf,KAAA;AACF;AAEAqC,EAAAA,YAAAA,CAAaC,QAAwD,EAAA;AACnEC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"embedding.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\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 { WatsonxClient, WatsonxClientSettings } from \"@/adapters/watsonx/backend/client.js\";\nimport {\n  EmbeddingModel,\n  EmbeddingModelInput,\n  EmbeddingModelOutput,\n  EmbeddingModelEvents,\n} from \"@/backend/embedding.js\";\nimport { EmbeddingParameters as WXEmbeddingParameters } from \"@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { getEnv } from \"@/internals/env.js\";\n\nexport type WatsonxEmbeddingModelParameters = WXEmbeddingParameters;\n\nexport class WatsonxEmbeddingModel extends EmbeddingModel {\n  protected readonly client: WatsonxClient;\n  public readonly emitter: Emitter<EmbeddingModelEvents>;\n\n  get providerId() {\n    return \"watsonx\";\n  }\n\n  constructor(\n    public readonly modelId: string = getEnv(\n      \"WATSONX_EMBEDDING_MODEL\",\n      \"ibm/granite-embedding-107m-multilingual\",\n    ),\n    public readonly parameters: WatsonxEmbeddingModelParameters = {},\n    client?: WatsonxClient | WatsonxClientSettings,\n  ) {\n    super();\n    this.client = WatsonxClient.ensure(client);\n    this.emitter = Emitter.root.child({\n      namespace: [\"backend\", \"watsonx\", \"embedding\"],\n      creator: this,\n    });\n  }\n\n  protected async _create(input: EmbeddingModelInput): Promise<EmbeddingModelOutput> {\n    const response = await this.client.instance.embedText({\n      modelId: this.modelId,\n      spaceId: this.client.spaceId,\n      projectId: this.client.projectId,\n      inputs: input.values,\n      parameters: {\n        return_options: { input_text: true },\n        ...this.parameters,\n      },\n    });\n\n    const embeddings = response.result.results.map((e) => e.embedding);\n    const values = response.result.results.map((e, i) => e.input || input.values.at(i)!);\n\n    return {\n      embeddings,\n      values,\n      usage: {\n        tokens: response.result.input_token_count,\n      },\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      modelId: this.modelId,\n      client: this.client,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/agent-BbaHVlob.d.cts

@@ -0,0 +1,84 @@
+import { GetRunContext, RunContext } from './context.cjs';
+import { b as AgentMeta, a as BaseAgent } from './base-Csr9guq2.cjs';
+import { AnyTool } from './tools/base.cjs';
+import { BaseMemory } from './memory/base.cjs';
+import { E as Emitter } from './emitter-BWtGHYn0.cjs';
+import { BeeMeta, BeeCallbacks, BeeIterationToolResult, BeeRunOptions, BeeAgentRunIteration, BeeParserInput, BeeRunInput, BeeAgentTemplates, BeeAgentExecutionConfig, BeeRunOutput } from './agents/bee/types.cjs';
+import { LinePrefixParser } from './parsers/linePrefix.cjs';
+import { Serializable } from './internals/serializable.cjs';
+import { ChatModel } from './backend/chat.cjs';
+import { RetryCounter } from './internals/helpers/counter.cjs';
+
+interface BeeRunnerLLMInput {
+    meta: BeeMeta;
+    signal: AbortSignal;
+    emitter: Emitter<BeeCallbacks>;
+}
+interface BeeRunnerToolInput {
+    state: BeeIterationToolResult;
+    meta: BeeMeta;
+    signal: AbortSignal;
+    emitter: Emitter<BeeCallbacks>;
+}
+declare abstract class BaseRunner extends Serializable {
+    protected readonly input: BeeInput;
+    protected readonly options: BeeRunOptions;
+    protected readonly run: GetRunContext<BeeAgent>;
+    memory: BaseMemory;
+    readonly iterations: BeeAgentRunIteration[];
+    protected readonly failedAttemptsCounter: RetryCounter;
+    constructor(input: BeeInput, options: BeeRunOptions, run: GetRunContext<BeeAgent>);
+    createIteration(): Promise<{
+        emitter: Emitter<BeeCallbacks>;
+        state: LinePrefixParser.infer<BeeParserInput>;
+        meta: BeeMeta;
+        signal: AbortSignal;
+    }>;
+    init(input: BeeRunInput): Promise<void>;
+    abstract llm(input: BeeRunnerLLMInput): Promise<BeeAgentRunIteration>;
+    abstract tool(input: BeeRunnerToolInput): Promise<{
+        output: string;
+        success: boolean;
+    }>;
+    abstract get defaultTemplates(): BeeAgentTemplates;
+    get templates(): BeeAgentTemplates;
+    protected abstract initMemory(input: BeeRunInput): Promise<BaseMemory>;
+    createSnapshot(): {
+        input: BeeInput;
+        options: BeeRunOptions;
+        memory: BaseMemory<unknown>;
+        failedAttemptsCounter: RetryCounter;
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+
+type BeeTemplateFactory<K extends keyof BeeAgentTemplates> = (template: BeeAgentTemplates[K]) => BeeAgentTemplates[K];
+interface BeeInput {
+    llm: ChatModel;
+    tools: AnyTool[];
+    memory: BaseMemory;
+    meta?: Omit<AgentMeta, "tools">;
+    templates?: Partial<{
+        [K in keyof BeeAgentTemplates]: BeeAgentTemplates[K] | BeeTemplateFactory<K>;
+    }>;
+    execution?: BeeAgentExecutionConfig;
+    stream?: boolean;
+}
+declare class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions> {
+    protected readonly input: BeeInput;
+    readonly emitter: Emitter<BeeCallbacks>;
+    protected runner: new (...args: ConstructorParameters<typeof BaseRunner>) => BaseRunner;
+    constructor(input: BeeInput);
+    set memory(memory: BaseMemory);
+    get memory(): BaseMemory;
+    get meta(): AgentMeta;
+    protected _run(input: BeeRunInput, options: BeeRunOptions | undefined, run: GetRunContext<typeof this>): Promise<BeeRunOutput>;
+    createSnapshot(): {
+        input: BeeInput;
+        emitter: Emitter<BeeCallbacks>;
+        runner: new (input: BeeInput, options: BeeRunOptions, run: RunContext<BeeAgent, any>) => BaseRunner;
+        isRunning: boolean;
+    };
+}
+
+export { type BeeInput as B, BeeAgent as a, BaseRunner as b, type BeeRunnerLLMInput as c, type BeeRunnerToolInput as d, type BeeTemplateFactory as e };

package/dist/agent-Bks2fOUl.d.ts

@@ -0,0 +1,84 @@
+import { GetRunContext, RunContext } from './context.js';
+import { b as AgentMeta, a as BaseAgent } from './base-DHhpWHQm.js';
+import { AnyTool } from './tools/base.js';
+import { BaseMemory } from './memory/base.js';
+import { E as Emitter } from './emitter-l0W9gC1A.js';
+import { BeeMeta, BeeCallbacks, BeeIterationToolResult, BeeRunOptions, BeeAgentRunIteration, BeeParserInput, BeeRunInput, BeeAgentTemplates, BeeAgentExecutionConfig, BeeRunOutput } from './agents/bee/types.js';
+import { LinePrefixParser } from './parsers/linePrefix.js';
+import { Serializable } from './internals/serializable.js';
+import { ChatModel } from './backend/chat.js';
+import { RetryCounter } from './internals/helpers/counter.js';
+
+interface BeeRunnerLLMInput {
+    meta: BeeMeta;
+    signal: AbortSignal;
+    emitter: Emitter<BeeCallbacks>;
+}
+interface BeeRunnerToolInput {
+    state: BeeIterationToolResult;
+    meta: BeeMeta;
+    signal: AbortSignal;
+    emitter: Emitter<BeeCallbacks>;
+}
+declare abstract class BaseRunner extends Serializable {
+    protected readonly input: BeeInput;
+    protected readonly options: BeeRunOptions;
+    protected readonly run: GetRunContext<BeeAgent>;
+    memory: BaseMemory;
+    readonly iterations: BeeAgentRunIteration[];
+    protected readonly failedAttemptsCounter: RetryCounter;
+    constructor(input: BeeInput, options: BeeRunOptions, run: GetRunContext<BeeAgent>);
+    createIteration(): Promise<{
+        emitter: Emitter<BeeCallbacks>;
+        state: LinePrefixParser.infer<BeeParserInput>;
+        meta: BeeMeta;
+        signal: AbortSignal;
+    }>;
+    init(input: BeeRunInput): Promise<void>;
+    abstract llm(input: BeeRunnerLLMInput): Promise<BeeAgentRunIteration>;
+    abstract tool(input: BeeRunnerToolInput): Promise<{
+        output: string;
+        success: boolean;
+    }>;
+    abstract get defaultTemplates(): BeeAgentTemplates;
+    get templates(): BeeAgentTemplates;
+    protected abstract initMemory(input: BeeRunInput): Promise<BaseMemory>;
+    createSnapshot(): {
+        input: BeeInput;
+        options: BeeRunOptions;
+        memory: BaseMemory<unknown>;
+        failedAttemptsCounter: RetryCounter;
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+
+type BeeTemplateFactory<K extends keyof BeeAgentTemplates> = (template: BeeAgentTemplates[K]) => BeeAgentTemplates[K];
+interface BeeInput {
+    llm: ChatModel;
+    tools: AnyTool[];
+    memory: BaseMemory;
+    meta?: Omit<AgentMeta, "tools">;
+    templates?: Partial<{
+        [K in keyof BeeAgentTemplates]: BeeAgentTemplates[K] | BeeTemplateFactory<K>;
+    }>;
+    execution?: BeeAgentExecutionConfig;
+    stream?: boolean;
+}
+declare class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions> {
+    protected readonly input: BeeInput;
+    readonly emitter: Emitter<BeeCallbacks>;
+    protected runner: new (...args: ConstructorParameters<typeof BaseRunner>) => BaseRunner;
+    constructor(input: BeeInput);
+    set memory(memory: BaseMemory);
+    get memory(): BaseMemory;
+    get meta(): AgentMeta;
+    protected _run(input: BeeRunInput, options: BeeRunOptions | undefined, run: GetRunContext<typeof this>): Promise<BeeRunOutput>;
+    createSnapshot(): {
+        input: BeeInput;
+        emitter: Emitter<BeeCallbacks>;
+        runner: new (input: BeeInput, options: BeeRunOptions, run: RunContext<BeeAgent, any>) => BaseRunner;
+        isRunning: boolean;
+    };
+}
+
+export { type BeeInput as B, BeeAgent as a, BaseRunner as b, type BeeRunnerLLMInput as c, type BeeRunnerToolInput as d, type BeeTemplateFactory as e };

package/dist/agents/base.cjs

@@ -0,0 +1,73 @@
+'use strict';
+
+var errors_cjs = require('../errors.cjs');
+var serializable_cjs = require('../internals/serializable.cjs');
+var context_cjs = require('../context.cjs');
+var createTelemetryMiddleware_cjs = require('../instrumentation/create-telemetry-middleware.cjs');
+var config_cjs = require('../instrumentation/config.cjs');
+var remeda = require('remeda');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class AgentError extends errors_cjs.FrameworkError {
+  static {
+    __name(this, "AgentError");
+  }
+}
+class BaseAgent extends serializable_cjs.Serializable {
+  static {
+    __name(this, "BaseAgent");
+  }
+  isRunning = false;
+  run(...[input, options]) {
+    if (this.isRunning) {
+      throw new AgentError("Agent is already running!");
+    }
+    return context_cjs.RunContext.enter(this, {
+      signal: options?.signal,
+      params: [
+        input,
+        options
+      ]
+    }, async (context) => {
+      try {
+        this.isRunning = true;
+        return await this._run(input, options, context);
+      } catch (e) {
+        if (e instanceof AgentError) {
+          throw e;
+        } else {
+          throw new AgentError(`Error has occurred!`, [
+            e
+          ]);
+        }
+      } finally {
+        this.isRunning = false;
+      }
+    }).middleware(config_cjs.INSTRUMENTATION_ENABLED ? createTelemetryMiddleware_cjs.createTelemetryMiddleware() : remeda.doNothing());
+  }
+  destroy() {
+    this.emitter.destroy();
+  }
+  get meta() {
+    return {
+      name: this.constructor.name ?? "BaseAgent",
+      description: "",
+      tools: []
+    };
+  }
+  createSnapshot() {
+    return {
+      isRunning: false,
+      emitter: this.emitter
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+
+exports.AgentError = AgentError;
+exports.BaseAgent = BaseAgent;
+//# sourceMappingURL=base.cjs.map
+//# sourceMappingURL=base.cjs.map

package/dist/agents/base.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","middleware","INSTRUMENTATION_ENABLED","createTelemetryMiddleware","doNothing","destroy","emitter","meta","name","constructor","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;;;;AA0BO,MAAMA,mBAAmBC,yBAAAA,CAAAA;EA1BhC;;;AA0BgD;AAMzC,MAAeC,kBAIZC,6BAAAA,CAAAA;EApCV;;;EAqCYC,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,CAAuB,mBAAA,CAAA,EAAA;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjD;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAY,GAAA,KAAA;AACnB;AACF,KAAA,EACAW,UAAWC,CAAAA,kCAAAA,GAA0BC,uDAAAA,EAAAA,GAA8BC,kBAAAA,CAAAA;AACvE;EAQAC,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;MAAEtB,SAAW,EAAA,KAAA;AAAOgB,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 IBM Corp.\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 { createTelemetryMiddleware } from \"@/instrumentation/create-telemetry-middleware.js\";\nimport { INSTRUMENTATION_ENABLED } from \"@/instrumentation/config.js\";\nimport { doNothing } from \"remeda\";\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(`Error has occurred!`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    ).middleware(INSTRUMENTATION_ENABLED ? createTelemetryMiddleware() : doNothing());\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

@@ -0,0 +1,18 @@
+import '../context.cjs';
+import '../errors.cjs';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-Csr9guq2.cjs';
+import '../internals/serializable.cjs';
+import '../emitter-BWtGHYn0.cjs';
+import '../memory/base.cjs';
+import '../internals/helpers/promise.cjs';
+import '../internals/types.cjs';
+import '../internals/helpers/guards.cjs';
+import '../tools/base.cjs';
+import 'ajv';
+import 'promise-based-task';
+import '../cache/base.cjs';
+import '../internals/helpers/schema.cjs';
+import 'zod';
+import 'zod-to-json-schema';
+import '../backend/message.cjs';
+import 'ai';

package/dist/agents/base.d.ts

@@ -0,0 +1,18 @@
+import '../context.js';
+import '../errors.js';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-DHhpWHQm.js';
+import '../internals/serializable.js';
+import '../emitter-l0W9gC1A.js';
+import '../memory/base.js';
+import '../internals/helpers/promise.js';
+import '../internals/types.js';
+import '../internals/helpers/guards.js';
+import '../tools/base.js';
+import 'ajv';
+import 'promise-based-task';
+import '../cache/base.js';
+import '../internals/helpers/schema.js';
+import 'zod';
+import 'zod-to-json-schema';
+import '../backend/message.js';
+import 'ai';

package/dist/agents/base.js

@@ -0,0 +1,70 @@
+import { FrameworkError } from '../errors.js';
+import { Serializable } from '../internals/serializable.js';
+import { RunContext } from '../context.js';
+import { createTelemetryMiddleware } from '../instrumentation/create-telemetry-middleware.js';
+import { INSTRUMENTATION_ENABLED } from '../instrumentation/config.js';
+import { doNothing } from 'remeda';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class AgentError extends FrameworkError {
+  static {
+    __name(this, "AgentError");
+  }
+}
+class BaseAgent extends Serializable {
+  static {
+    __name(this, "BaseAgent");
+  }
+  isRunning = false;
+  run(...[input, options]) {
+    if (this.isRunning) {
+      throw new AgentError("Agent is already running!");
+    }
+    return RunContext.enter(this, {
+      signal: options?.signal,
+      params: [
+        input,
+        options
+      ]
+    }, async (context) => {
+      try {
+        this.isRunning = true;
+        return await this._run(input, options, context);
+      } catch (e) {
+        if (e instanceof AgentError) {
+          throw e;
+        } else {
+          throw new AgentError(`Error has occurred!`, [
+            e
+          ]);
+        }
+      } finally {
+        this.isRunning = false;
+      }
+    }).middleware(INSTRUMENTATION_ENABLED ? createTelemetryMiddleware() : doNothing());
+  }
+  destroy() {
+    this.emitter.destroy();
+  }
+  get meta() {
+    return {
+      name: this.constructor.name ?? "BaseAgent",
+      description: "",
+      tools: []
+    };
+  }
+  createSnapshot() {
+    return {
+      isRunning: false,
+      emitter: this.emitter
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+
+export { AgentError, BaseAgent };
+//# sourceMappingURL=base.js.map
+//# sourceMappingURL=base.js.map

package/dist/agents/base.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/agents/base.ts"],"names":["AgentError","FrameworkError","BaseAgent","Serializable","isRunning","run","input","options","RunContext","enter","signal","params","context","_run","e","middleware","INSTRUMENTATION_ENABLED","createTelemetryMiddleware","doNothing","destroy","emitter","meta","name","constructor","description","tools","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;;;;AA0BO,MAAMA,mBAAmBC,cAAAA,CAAAA;EA1BhC;;;AA0BgD;AAMzC,MAAeC,kBAIZC,YAAAA,CAAAA;EApCV;;;EAqCYC,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,CAAuB,mBAAA,CAAA,EAAA;AAACc,YAAAA;AAAE,WAAA,CAAA;AACjD;OACF,SAAA;AACE,QAAA,IAAA,CAAKV,SAAY,GAAA,KAAA;AACnB;AACF,KAAA,EACAW,UAAWC,CAAAA,uBAAAA,GAA0BC,yBAAAA,EAAAA,GAA8BC,WAAAA,CAAAA;AACvE;EAQAC,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;MAAEtB,SAAW,EAAA,KAAA;AAAOgB,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 IBM Corp.\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 { createTelemetryMiddleware } from \"@/instrumentation/create-telemetry-middleware.js\";\nimport { INSTRUMENTATION_ENABLED } from \"@/instrumentation/config.js\";\nimport { doNothing } from \"remeda\";\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(`Error has occurred!`, [e]);\n          }\n        } finally {\n          this.isRunning = false;\n        }\n      },\n    ).middleware(INSTRUMENTATION_ENABLED ? createTelemetryMiddleware() : doNothing());\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/bee/agent.cjs

@@ -0,0 +1,193 @@
+'use strict';
+
+var base_cjs = require('../base.cjs');
+var message_cjs = require('../../backend/message.cjs');
+var emitter_cjs = require('../../emitter/emitter.cjs');
+var object_cjs = require('../../internals/helpers/object.cjs');
+var R = require('remeda');
+var runner_cjs = require('./runners/granite/runner.cjs');
+var runner_cjs$1 = require('./runners/deep-think/runner.cjs');
+var errors_cjs = require('../../errors.cjs');
+var runner_cjs$2 = require('./runners/default/runner.cjs');
+
+function _interopNamespace(e) {
+  if (e && e.__esModule) return e;
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
+        });
+      }
+    });
+  }
+  n.default = e;
+  return Object.freeze(n);
+}
+
+var R__namespace = /*#__PURE__*/_interopNamespace(R);
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class BeeAgent extends base_cjs.BaseAgent {
+  static {
+    __name(this, "BeeAgent");
+  }
+  input;
+  emitter;
+  runner;
+  constructor(input) {
+    super(), this.input = input, this.emitter = emitter_cjs.Emitter.root.child({
+      namespace: [
+        "agent",
+        "bee"
+      ],
+      creator: this
+    });
+    const duplicate = input.tools.find((a, i, arr) => arr.find((b, j) => i !== j && a.name.toUpperCase() === b.name.toUpperCase()));
+    if (duplicate) {
+      throw new errors_cjs.ValueError(`Agent's tools must all have different names. Conflicting tool: ${duplicate.name}.`);
+    }
+    const modelId = this.input.llm.modelId.toLowerCase();
+    this.runner = [
+      {
+        tag: "granite",
+        runner: runner_cjs.GraniteRunner
+      },
+      {
+        tag: "deepseek-r1",
+        runner: runner_cjs$1.DeepThinkRunner
+      }
+    ].find(({ tag }) => modelId.includes(tag))?.runner ?? runner_cjs$2.DefaultRunner;
+  }
+  static {
+    this.register();
+  }
+  set memory(memory) {
+    this.input.memory = memory;
+  }
+  get memory() {
+    return this.input.memory;
+  }
+  get meta() {
+    const tools = this.input.tools.slice();
+    if (this.input.meta) {
+      return {
+        ...this.input.meta,
+        tools
+      };
+    }
+    return {
+      name: "Bee",
+      tools,
+      description: "The Bee framework demonstrates its ability to auto-correct and adapt in real-time, improving the overall reliability and resilience of the system.",
+      ...tools.length > 0 && {
+        extraDescription: [
+          `Tools that I can use to accomplish given task.`,
+          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`)
+        ].join("\n")
+      }
+    };
+  }
+  async _run(input, options = {}, run) {
+    const runner = new this.runner(this.input, {
+      ...options,
+      execution: this.input.execution ?? options?.execution ?? {
+        maxRetriesPerStep: 3,
+        totalMaxRetries: 20,
+        maxIterations: 10
+      }
+    }, run);
+    await runner.init(input);
+    let finalMessage;
+    while (!finalMessage) {
+      const { state, meta, emitter, signal } = await runner.createIteration();
+      if (state.tool_name && state.tool_input) {
+        const { output, success } = await runner.tool({
+          state,
+          emitter,
+          meta,
+          signal
+        });
+        await runner.memory.add(new message_cjs.AssistantMessage(runner.templates.assistant.render({
+          thought: [
+            state.thought
+          ].filter(R__namespace.isTruthy),
+          toolName: [
+            state.tool_name
+          ].filter(R__namespace.isTruthy),
+          toolInput: [
+            state.tool_input
+          ].filter(R__namespace.isTruthy).map((call) => JSON.stringify(call)),
+          toolOutput: [
+            output
+          ],
+          finalAnswer: [
+            state.final_answer
+          ].filter(R__namespace.isTruthy)
+        }), {
+          success
+        }));
+        object_cjs.assign(state, {
+          tool_output: output
+        });
+        for (const key of [
+          "partialUpdate",
+          "update"
+        ]) {
+          await emitter.emit(key, {
+            data: state,
+            update: {
+              key: "tool_output",
+              value: output,
+              parsedValue: output
+            },
+            meta: {
+              success,
+              ...meta
+            },
+            memory: runner.memory
+          });
+        }
+      }
+      if (state.final_answer) {
+        finalMessage = new message_cjs.AssistantMessage(state.final_answer, {
+          createdAt: /* @__PURE__ */ new Date()
+        });
+        await runner.memory.add(finalMessage);
+        await emitter.emit("success", {
+          data: finalMessage,
+          iterations: runner.iterations,
+          memory: runner.memory,
+          meta
+        });
+      }
+    }
+    if (input.prompt !== null) {
+      await this.input.memory.add(new message_cjs.UserMessage(input.prompt, {
+        createdAt: run.createdAt
+      }));
+    }
+    await this.input.memory.add(finalMessage);
+    return {
+      result: finalMessage,
+      iterations: runner.iterations,
+      memory: runner.memory
+    };
+  }
+  createSnapshot() {
+    return {
+      ...super.createSnapshot(),
+      input: this.input,
+      emitter: this.emitter,
+      runner: this.runner
+    };
+  }
+}
+
+exports.BeeAgent = BeeAgent;
+//# sourceMappingURL=agent.cjs.map
+//# sourceMappingURL=agent.cjs.map