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

@ -6,6 +6,10 @@ async function createStream(req: NextRequest) {
const decoder = new TextDecoder();
let apiKey = process.env.OPENAI_API_KEY;
let apiBasePath = process.env.OPENAI_API_BASE_PATH;
if (apiBasePath) {
apiBasePath = apiBasePath.replace(/\/+$/, "");
}
const userApiKey = req.headers.get("token");
if (userApiKey) {
@ -13,7 +17,7 @@ async function createStream(req: NextRequest) {
console.log("[Stream] using user api key");
}
const res = await fetch("https://api.openai.com/v1/chat/completions", {
const res = await fetch(`${apiBasePath ? apiBasePath : 'https://api.openai.com/v1'}/chat/completions`, {
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${apiKey}`,

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;
}
const openai = new OpenAIApi(
new Configuration({
let configuration: ConfigurationParameters = {
apiKey,
})
);
};
if (apiBasePath) {
configuration.basePath = apiBasePath.replace(/\/+$/, "");
}
const openai = new OpenAIApi(new Configuration(configuration));
const requestBody = (await req.json()) as ChatRequest;
const completion = await openai!.createChatCompletion({