Add OPENAI_BASE_URL

lvwzhen · May 4, 2023 · 6ae9d68 · 6ae9d68
1 parent bfacffe
commit 6ae9d68
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/.env.example b/.env.example
@@ -5,4 +5,5 @@ NEXT_PUBLIC_SUPABASE_ANON_KEY=
 SUPABASE_SERVICE_ROLE_KEY=
 
 # Get your key at https://platform.openai.com/account/api-keys
-OPENAI_KEY=
+OPENAI_KEY=
+OPENAI_BASE_URL=
diff --git a/pages/api/vector-search.ts b/pages/api/vector-search.ts
@@ -11,6 +11,7 @@ export const config = {
 }
 
 const openAiKey = process.env.OPENAI_KEY
+const openAiBaseUrl = process.env.OPENAI_BASE_URL
 const supabaseUrl = process.env.NEXT_PUBLIC_SUPABASE_URL
 const supabaseServiceKey = process.env.SUPABASE_SERVICE_ROLE_KEY
 
@@ -44,7 +45,7 @@ export default async function handler(req: NextRequest) {
 
     // Moderate the content to comply with OpenAI T&C
     const sanitizedQuery = query.trim()
-    const moderationResponse = await fetch('https://api.openai.com/v1/moderations', {
+    const moderationResponse = await fetch('https://' + openAiBaseUrl + '/v1/v1/moderations', {
       method: 'POST',
       headers: {
         Authorization: `Bearer ${openAiKey}`,
@@ -64,7 +65,7 @@ export default async function handler(req: NextRequest) {
       })
     }
 
-    const embeddingResponse = await fetch('https://api.openai.com/v1/embeddings', {
+    const embeddingResponse = await fetch('https://' + openAiBaseUrl + '/v1/v1/embeddings', {
       method: 'POST',
       headers: {
         Authorization: `Bearer ${openAiKey}`,
@@ -143,7 +144,7 @@ export default async function handler(req: NextRequest) {
       stream: true,
     }
 
-    const response = await fetch('https://api.openai.com/v1/chat/completions', {
+    const response = await fetch('https://' + openAiBaseUrl + '/v1/chat/completions', {
       method: 'POST',
       headers: {
         Authorization: `Bearer ${openAiKey}`,