diff --git a/src/utils/openAI.ts b/src/utils/openAI.ts
index 8a3f4932..c4de5151 100644
--- 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