vite get enviroment variable

fujia-apps · Dec 18, 2023 · 45965c9 · 45965c9
1 parent f7a84d3
commit 45965c9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/utils/openAI.ts b/src/utils/openAI.ts
@@ -1,11 +1,11 @@
 import { GoogleGenerativeAI } from '@fuyun/generative-ai'
 
-const apiKey = process.env.GEMINI_API_KEY
-const apiBaseUrl = process.env.API_BASE_URL?.trim().replace(/\/$/, '')
+const apiKey = (import.meta.env.GEMINI_API_KEY)
+const apiBaseUrl = (import.meta.env.API_BASE_URL)?.trim().replace(/\/$/, '')
 
 const genAI = apiBaseUrl
   ? new GoogleGenerativeAI(apiKey, apiBaseUrl)
-  : new GoogleGenerativeAI(apiKey)
+  : new GoogleGenerativeAI(apiKey)  
 
 export const startChatAndSendMessageStream = async(history: ChatMessage[], newMessage: string) => {
   const model = genAI.getGenerativeModel({ model: 'gemini-pro' })