refactor(api): extract OpenAI client to shared module

This commit is contained in:
cfngc4594 2025-05-14 22:30:45 +08:00
parent 1c031ce24e
commit 9443ab6d6d
2 changed files with 10 additions and 7 deletions

View File

@ -1,14 +1,9 @@
import { streamText } from 'ai';
import { createOpenAI } from '@ai-sdk/openai';
import { streamText } from "ai";
import { openai } from "@/lib/ai";
// Allow streaming responses up to 30 seconds
export const maxDuration = 30;
const openai = createOpenAI({
apiKey: process.env.OPENAI_API_KEY || "",
baseURL: process.env.OPENAI_BASE_URL || "",
});
export async function POST(req: Request) {
const { messages } = await req.json();

8
src/lib/ai.ts Normal file
View File

@ -0,0 +1,8 @@
import "server-only";
import { createOpenAI } from "@ai-sdk/openai";
export const openai = createOpenAI({
apiKey: process.env.OPENAI_API_KEY || "",
baseURL: process.env.OPENAI_BASE_URL || "",
});