🐝 BeeAI agent

1import { Actor, log } from 'apify';
2import { OpenAIChatModel } from 'beeai-framework/adapters/openai/backend/chat';
3import { ReActAgent } from 'beeai-framework/agents/react/agent';
4import { UnconstrainedMemory } from 'beeai-framework/memory/unconstrainedMemory';
5import { z } from 'zod';
6
7import { StructuredOutputGenerator } from './structured_response_generator.js';
8import { CalculatorSumTool } from './tools/calculator.js';
9import { InstagramScrapeTool } from './tools/instagram.js';
10
11// Apify-hosted OpenRouter proxy (https://apify.com/apify/openrouter): an
12// OpenAI-compatible endpoint billed against the user's Apify account, so no
13// provider API key is needed. The proxy authenticates via the `Authorization`
14// header carrying the run's APIFY_TOKEN.
15const OPENROUTER_BASE_URL = 'https://openrouter.apify.actor/api/v1';
16
17// This is an ESM project, and as such, it requires you to specify extensions in your relative imports.
18// Read more about this here: https://nodejs.org/docs/latest-v18.x/api/esm.html#mandatory-file-extensions
19// Note that we need to use `.js` even when inside TS files
20// import { router } from './routes.js';
21
22// Actor input schema
23interface Input {
24    query: string;
25    modelName: string;
26    debug?: boolean;
27}
28
29// The init() call configures the Actor to correctly work with the Apify-provided environment - mainly the storage infrastructure. It is necessary that every Actor performs an init() call.
30await Actor.init();
31
32// Charge for Actor start
33await Actor.charge({ eventName: 'actor-start' });
34
35// Handle input
36const {
37    // The query default value is provided only for template testing purposes.
38    // You can remove it.
39    query,
40    modelName,
41    debug,
42} = (await Actor.getInput()) as Input;
43if (debug) {
44    log.setLevel(log.LEVELS.DEBUG);
45}
46if (!query) {
47    throw new Error('An agent query is required.');
48}
49
50/**
51 * Actor code
52 */
53// Create an agent that can use tools.
54// See https://framework.beeai.dev/modules/agents
55log.debug(`Using model: ${modelName}`);
56const llm = new OpenAIChatModel(
57    modelName,
58    {},
59    {
60        baseURL: OPENROUTER_BASE_URL,
61        apiKey: 'apify-openrouter-proxy', // unused by the proxy; OpenAI SDK rejects an empty string
62        headers: { Authorization: `Bearer ${process.env.APIFY_TOKEN}` },
63    },
64);
65const agent = new ReActAgent({
66    llm,
67    memory: new UnconstrainedMemory(),
68    tools: [new CalculatorSumTool(), new InstagramScrapeTool()],
69});
70
71// Store tool messages for later structured output generation.
72// This can be removed if you don't need structured output.
73const structuredOutputGenerator = new StructuredOutputGenerator(llm);
74
75// Prompt the agent with the query.
76// Debug log agent status updates, e.g., thoughts, tool calls, etc.
77const response = await agent.run({ prompt: query }).observe((emitter) => {
78    emitter.on('update', async ({ update }) => {
79        log.debug(`Agent (${update.key}) 🤖 : ${update.value}`);
80
81        // Save tool messages for later structured output generation.
82        // This can be removed if you don't need structured output.
83        if (['tool_name', 'tool_output', 'tool_input'].includes(update.key as string)) {
84            structuredOutputGenerator.processToolMessage(
85                update.key as 'tool_name' | 'tool_output' | 'tool_input',
86                update.value,
87            );
88        }
89        // End of tool message saving.
90    });
91});
92
93log.info(`Agent 🤖 : ${response.result.text}`);
94
95// Hacky way to get the structured output.
96// Using the stored tool messages and the user query to create a structured output.
97const structuredResponse = await structuredOutputGenerator.generateStructuredOutput(
98    query,
99    z.object({
100        totalLikes: z.number(),
101        totalComments: z.number(),
102        mostPopularPosts: z.array(
103            z.object({
104                url: z.string(),
105                likes: z.number(),
106                comments: z.number(),
107                timestamp: z.string(),
108                caption: z.string().nullable().optional(),
109                alt: z.string().nullable().optional(),
110            }),
111        ),
112    }),
113);
114log.debug(`Structured response: ${JSON.stringify(structuredResponse)}`);
115
116// Charge for task completion
117await Actor.charge({ eventName: 'task-completed' });
118
119// Push results to the dataset.
120await Actor.pushData({
121    query,
122    response: response.result.text,
123    // This can be removed if you don't need structured output.
124    structuredResponse: structuredResponse.object,
125});
126log.info('Pushed the data into the dataset!');
127
128// Gracefully exit the Actor process. It's recommended to quit all Actors with an exit().
129await Actor.exit();