20 lines
518 B
TypeScript
20 lines
518 B
TypeScript
import OpenAI from 'openai';
|
|
import 'dotenv/config';
|
|
|
|
const client = new OpenAI({
|
|
apiKey: process.env.OPENAI_API_KEY,
|
|
baseURL: process.env.OPENAI_BASE_URL,
|
|
});
|
|
|
|
export async function chat(
|
|
messages: { role: string; content: string }[],
|
|
tools?: any[] // OpenAI 格式的工具定义数组
|
|
) {
|
|
const response = await client.chat.completions.create({
|
|
model: 'deepseek-v4-pro',
|
|
messages: messages as any,
|
|
temperature: 0.7,
|
|
...(tools && { tools }),
|
|
});
|
|
return response.choices[0]!.message!;
|
|
} |