Observability for BeeAI (TypeScript) with Opik

1
import "dotenv/config";
2
import { NodeSDK } from "@opentelemetry/sdk-node";
3
import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
4
import { BeeAIInstrumentation } from "@arizeai/openinference-instrumentation-beeai";
5
import * as beeaiFramework from "beeai-framework";
6
7
// Initialize BeeAI Instrumentation
8
const beeAIInstrumentation = new BeeAIInstrumentation();
9
10
// Configure and start the OpenTelemetry SDK
11
const sdk = new NodeSDK({
12
  traceExporter: new OTLPTraceExporter(),
13
  instrumentations: [beeAIInstrumentation],
14
});
15
sdk.start();
16
17
// Manually patch BeeAI framework (required for trace collection)
18
beeAIInstrumentation.manuallyInstrument(beeaiFramework);
19
20
// Now you can use BeeAI as normal
21
import { ReActAgent } from "beeai-framework/agents/react";
22
import { OpenAIChatModel } from "beeai-framework/adapters/openai/backend/chat";
23
import { WikipediaTool } from "beeai-framework/tools/search/wikipedia";
24
import { OpenMeteoTool } from "beeai-framework/tools/weather/openmeteo";
25
import { TokenMemory } from "beeai-framework/memory";
26
27
// Initialize the OpenAI language model
28
const llm = new OpenAIChatModel("gpt-4o-mini", {
29
  temperature: 0.7,
30
});
31
32
// Create tools for the agent
33
const tools = [
34
  new WikipediaTool(),
35
  new OpenMeteoTool(),
36
];
37
38
// Create a ReAct agent with memory
39
const agent = new ReActAgent({
40
  llm,
41
  tools,
42
  memory: new TokenMemory({ llm }),
43
});
44
45
// Run the agent
46
async function main() {
47
  const response = await agent.run({
48
    prompt: "I'm planning a trip to Barcelona, Spain. Can you research key attractions and landmarks I should visit, and also tell me what the current weather conditions are like there?",
49
    execution: {
50
      maxRetriesPerStep: 3,
51
      totalMaxRetries: 10,
52
      maxIterations: 5,
53
    },
54
  });
55
  
56
  console.log("Agent Response:", response.result.text);
57
  return response;
58
}
59
60
// Run the example
61
main();