add basePath for api env inject

This commit is contained in:
Rentoo
2023-03-28 03:43:26 +08:00
parent c93a46a02f
commit e6241bf405
2 changed files with 15 additions and 8 deletions

View File

@@ -1,20 +1,23 @@
import { OpenAIApi, Configuration } from "openai";
import { OpenAIApi, Configuration, ConfigurationParameters } from "openai";
import { ChatRequest } from "./typing";
export async function POST(req: Request) {
try {
let apiKey = process.env.OPENAI_API_KEY;
let apiBasePath = process.env.OPENAI_API_BASE_PATH;
const userApiKey = req.headers.get("token");
if (userApiKey) {
apiKey = userApiKey;
}
let configuration: ConfigurationParameters = {
apiKey,
};
if (apiBasePath) {
configuration.basePath = apiBasePath.replace(/\/+$/, "");
}
const openai = new OpenAIApi(
new Configuration({
apiKey,
})
);
const openai = new OpenAIApi(new Configuration(configuration));
const requestBody = (await req.json()) as ChatRequest;
const completion = await openai!.createChatCompletion({