BeeAI

npm install "beeai-framework@0.1.13" "@ai-sdk/openai" \
  @arizeai/openinference-instrumentation-beeai \
  @arizeai/openinference-semantic-conventions \
  @opentelemetry/exporter-trace-otlp-proto \
  @opentelemetry/instrumentation \
  @opentelemetry/resources \
  @opentelemetry/sdk-trace-base \
  @opentelemetry/sdk-trace-node \
  @opentelemetry/semantic-conventions

export ARIZE_SPACE_ID="<your-space-id>"
export ARIZE_API_KEY="<your-api-key>"
export ARIZE_PROJECT_NAME="beeai-tracing-example"
export OPENAI_API_KEY="<your-openai-api-key>"

// instrumentation.ts
import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
import { resourceFromAttributes } from "@opentelemetry/resources";
import { SimpleSpanProcessor } from "@opentelemetry/sdk-trace-base";
import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
import { ATTR_SERVICE_NAME } from "@opentelemetry/semantic-conventions";
import {
  SEMRESATTRS_PROJECT_NAME,
} from "@arizeai/openinference-semantic-conventions";
import { registerInstrumentations } from "@opentelemetry/instrumentation";
import {
  BeeAIInstrumentation,
} from "@arizeai/openinference-instrumentation-beeai";
import * as beeaiFramework from "beeai-framework";

const projectName =
  process.env.ARIZE_PROJECT_NAME ?? "beeai-tracing-example";

export const provider = new NodeTracerProvider({
  resource: resourceFromAttributes({
    [ATTR_SERVICE_NAME]: projectName,
    [SEMRESATTRS_PROJECT_NAME]: projectName,
  }),
  spanProcessors: [
    new SimpleSpanProcessor(
      new OTLPTraceExporter({
        url: "https://otlp.arize.com/v1/traces",
        headers: {
          "arize-space-id": process.env.ARIZE_SPACE_ID ?? "",
          "arize-api-key": process.env.ARIZE_API_KEY ?? "",
        },
      }),
    ),
  ],
});

provider.register();

const instrumentation = new BeeAIInstrumentation();
instrumentation.manuallyInstrument(beeaiFramework);

registerInstrumentations({ instrumentations: [instrumentation] });

console.log("Arize AX tracing initialized for BeeAI.");

// example.ts

// Importing instrumentation first ensures tracing is set up before any
// BeeAI agent is created.
import { provider } from "./instrumentation";

import {
  ToolCallingAgent,
} from "beeai-framework/agents/toolCalling/agent";
import { TokenMemory } from "beeai-framework/memory/tokenMemory";
import {
  OpenAIChatModel,
} from "beeai-framework/adapters/openai/backend/chat";

// OpenAIChatModel reads OPENAI_API_KEY from the environment.
const llm = new OpenAIChatModel("gpt-5");

const agent = new ToolCallingAgent({
  llm,
  memory: new TokenMemory(),
  tools: [],
});

const response = await agent.run({
  prompt: "Why is the ocean salty? Answer in two sentences.",
});

console.log(response.result.text);

// Flush any pending spans before the process exits.
await provider.forceFlush();