Update openAI.ts

babaohuang · Dec 17, 2023 · c3c911a · c3c911a · vercel · Dec 17, 2023
1 parent a51abd2
commit c3c911a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils/openAI.ts b/src/utils/openAI.ts
@@ -1,4 +1,4 @@
-import { GoogleGenerativeAI } from '@fuyun/generative-ai'
+import { GoogleGenerativeAI, HarmCategory } from '@fuyun/generative-ai'
 
 const apiKey = process.env.GEMINI_API_KEY
 const apiBaseUrl = process.env.API_BASE_URL