import FirecrawlApp from '@mendable/firecrawl-js';
import OpenAI from 'openai';
import { z } from 'zod';
const firecrawl = new FirecrawlApp({ apiKey: process.env.FIRECRAWL_API_KEY });
const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
const ScrapeArgsSchema = z.object({
url: z.string().describe('A URL do site para extrair dados')
});
const tools = [{
type: 'function' as const,
function: {
name: 'scrape_website',
description: 'Extrai conteúdo de qualquer URL de site',
parameters: z.toJSONSchema(ScrapeArgsSchema)
}
}];
const response = await openai.chat.completions.create({
model: 'gpt-5-nano',
messages: [{
role: 'user',
content: 'O que é o Firecrawl? Acesse firecrawl.dev e me fale sobre ele.'
}],
tools
});
const message = response.choices[0]?.message;
if (message?.tool_calls && message.tool_calls.length > 0) {
for (const toolCall of message.tool_calls) {
if (toolCall.type === 'function') {
console.log('Ferramenta chamada:', toolCall.function.name);
const args = ScrapeArgsSchema.parse(JSON.parse(toolCall.function.arguments));
const result = await firecrawl.scrape(args.url, {
formats: ['markdown'] // Outros formatos: html, links, etc.
});
console.log('Conteúdo extraído:', result.markdown?.substring(0, 200) + '...');
// Envia o conteúdo extraído de volta ao modelo para resposta final
const finalResponse = await openai.chat.completions.create({
model: 'gpt-5-nano',
messages: [
{
role: 'user',
content: 'O que é o Firecrawl? Acesse firecrawl.dev e me fale sobre ele.'
},
message,
{
role: 'tool',
tool_call_id: toolCall.id,
content: result.markdown || 'Nenhum conteúdo extraído'
}
],
tools
});
console.log('Resposta final:', finalResponse.choices[0]?.message?.content);
}
}
} else {
console.log('Resposta direta:', message?.content);
}