Revert "change sdk"

This reverts commit 31e1dab.
babaohuang · Dec 17, 2023 · 5fbfb89 · 5fbfb89 · vercel · Dec 17, 2023
1 parent 8a88633
commit 5fbfb89
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 18 deletions.
diff --git a/.env.example b/.env.example
@@ -3,7 +3,7 @@ GEMINI_API_KEY=
 # Provide proxy for OpenAI API. e.g. http://127.0.0.1:7890
 HTTPS_PROXY=
 # Custom base url for OpenAI API. default: https://api.openai.com
-API_BASE_URL=
+OPENAI_API_BASE_URL=
 # Inject analytics or other scripts before </head> of the page
 HEAD_SCRIPTS=
 # Secret string for the project. Use for generating signatures for API calls

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -18,7 +18,7 @@
     "@astrojs/node": "^5.3.0",
     "@astrojs/solid-js": "^2.2.0",
     "@astrojs/vercel": "^3.5.0",
-    "@fuyun/generative-ai": "^0.1.2",
+    "@google/generative-ai": "^0.1.1",
     "@zag-js/slider": "^0.16.0",
     "@zag-js/solid": "^0.16.0",
     "astro": "^2.7.0",

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/utils/openAI.ts b/src/utils/openAI.ts
@@ -1,11 +1,7 @@
-import { GoogleGenerativeAI } from '@fuyun/generative-ai'
+import { GoogleGenerativeAI } from '@google/generative-ai'
 
 const apiKey = process.env.GEMINI_API_KEY
-const apiBaseUrl = process.env.API_BASE_URL
-
-// 根据是否有 API_BASE_URL 环境变量来决定参数
-const genAIOptions = apiBaseUrl ? { apiKey, apiBaseUrl } : { apiKey }
-const genAI = new GoogleGenerativeAI(genAIOptions)
+const genAI = new GoogleGenerativeAI(apiKey)
 
 export const startChatAndSendMessageStream = async(history: ChatMessage[], newMessage: string) => {
   const model = genAI.getGenerativeModel({ model: 'gemini-pro' })