beeai-framework

import "dotenv/config"; import { Agent, ChatModel, Tool, Message } from 'beeai-framework'; // Define a simple tool that the agent can use class MyCalculatorTool extends Tool { constructor() { super({ name: "calculator", description: "A simple calculator tool that can add two numbers.", parameters: { type: "object", properties: { a: { type: "number", description: "The first number" }, b: { type: "number", description: "The second number" } }, required: ["a", "b"] } }); } async execute(params: { a: number; b: number }): Promise<any> { return { result: params.a + params.b }; } } async function runAgent() { // Initialize a chat model (e.g., using OpenAI via @ai-sdk/openai peer dep) // Ensure OPENAI_API_KEY is set in your .env file const chatModel = new ChatModel({ model: process.env.OPENAI_MODEL ?? 'gpt-4o', provider: 'openai' // This would depend on specific @ai-sdk integration }); // Instantiate the agent with the chat model and available tools const agent = new Agent({ model: chatModel, tools: [new MyCalculatorTool()] // Add our custom tool // Other agent configurations like memory, system prompts etc. }); console.log("Agent initialized. Asking it to perform a calculation..."); // Interact with the agent const response = await agent.run([ { role: "user", content: "What is 5 plus 3?" } ]); console.log("Agent's response:", response.messages[response.messages.length - 1].content); } runAgent().catch(console.error);