Files
firecrawl/apps/api/src/lib/llm/generate.ts
T
Nicolas 5c47e97db2 (feat/deep-research) Alpha implementation of deep research (#1202)
* Nick:

* Revert "fix(v1/types): fix extract -> json rename (FIR-1072) (#1195)"

This reverts commit 586a10f40d.

* Update deep-research-service.ts

* Nick:

* Nick:

* Nick:

* Nick:

* Nick:

* Nick:

* Update deep-research-service.ts

* Nick:

* Update deep-research-service.ts

* Apply suggestions from code review

---------

Co-authored-by: Gergő Móricz <mo.geryy@gmail.com>
2025-02-19 12:44:21 -03:00

33 lines
680 B
TypeScript

import OpenAI from "openai";
const openai = new OpenAI({
apiKey: process.env.OPENAI_API_KEY,
});
interface Message {
role: "system" | "user" | "assistant";
content: string;
}
interface GenerateTextOptions {
model: string;
messages: Message[];
temperature?: number;
maxTokens?: number;
}
export async function generateText(options: GenerateTextOptions) {
const { model, messages, temperature = 0.7, maxTokens } = options;
const completion = await openai.chat.completions.create({
model,
messages,
temperature,
max_tokens: maxTokens,
});
return {
text: completion.choices[0].message.content || "",
usage: completion.usage,
};
}