ChatGPT-Next-Web/app/api/chat/route.ts
2023-03-11 02:47:29 +08:00

48 lines
979 B
TypeScript

import { OpenAIApi, Configuration } from "openai";
import { ChatRequest } from "./typing";
const isProd = process.env.NODE_ENV === "production";
let openai: OpenAIApi | undefined;
async function initService() {
let apiKey = process.env.OPENAI_API_KEY;
if (!isProd) {
apiKey = await (await import("./config")).apiKey;
}
openai = new OpenAIApi(
new Configuration({
apiKey,
})
);
}
export async function POST(req: Request) {
if (!openai) {
await initService();
}
try {
const requestBody = (await req.json()) as ChatRequest;
const completion = await openai!.createChatCompletion(
{
...requestBody,
},
isProd
? {}
: {
proxy: {
protocol: "socks",
host: "127.0.0.1",
port: 7890,
},
}
);
return new Response(JSON.stringify(completion.data));
} catch (e) {
return new Response(JSON.stringify(e));
}
}