diff --git a/packages/core/src/services/llm/service.ts b/packages/core/src/services/llm/service.ts
index e4df3a03..97eb84dc 100644
--- a/packages/core/src/services/llm/service.ts
+++ b/packages/core/src/services/llm/service.ts
@@ -107,13 +107,9 @@ export class LLMService implements ILLMService {
    */
   private getGeminiModel(modelConfig: ModelConfig, systemInstruction?: string, isStream: boolean = false): GenerativeModel {
     const apiKey = modelConfig.apiKey || '';
-    
-    // 创建GoogleGenerativeAI实例配置
-    const genAIConfig: any = {
-      apiKey: apiKey
-    };
 
-    const genAI = new GoogleGenerativeAI(genAIConfig);
+    // 创建GoogleGenerativeAI实例 - 旧版本直接传入字符串API key
+    const genAI = new GoogleGenerativeAI(apiKey);
 
     // 创建模型配置
     const modelOptions: any = {
diff --git a/packages/core/tests/integration/prompt/service.integration.test.ts b/packages/core/tests/integration/prompt/service.integration.test.ts
index 00812cb3..428dd545 100644
--- a/packages/core/tests/integration/prompt/service.integration.test.ts
+++ b/packages/core/tests/integration/prompt/service.integration.test.ts
@@ -56,8 +56,8 @@ describe('PromptService Integration Tests', () => {
         provider: 'gemini',
         apiKey: process.env.VITE_GEMINI_API_KEY!,
         baseURL: 'https://generativelanguage.googleapis.com/v1beta',
-        defaultModel: 'gemini-2.0-flash-exp',
-        models: ['gemini-2.0-flash-exp'],
+        defaultModel: 'gemini-2.0-flash',
+        models: ['gemini-2.0-flash'],
         enabled: true,
         llmParams: {
           temperature: 0.7,
diff --git a/packages/core/tests/integration/real-api.test.ts b/packages/core/tests/integration/real-api.test.ts
index e01361c8..1efaab49 100644
--- a/packages/core/tests/integration/real-api.test.ts
+++ b/packages/core/tests/integration/real-api.test.ts
@@ -158,8 +158,8 @@ describe('Real API Integration Tests', () => {
         name: 'Google Gemini',
         baseURL: 'https://generativelanguage.googleapis.com/v1beta',
         apiKey: process.env.VITE_GEMINI_API_KEY!,
-        models: ['gemini-2.0-flash-exp'],
-        defaultModel: 'gemini-2.0-flash-exp',
+        models: ['gemini-2.0-flash'],
+        defaultModel: 'gemini-2.0-flash',
         enabled: true,
         provider: 'gemini' as const
       }
diff --git a/packages/core/tests/integration/storage-implementations.test.ts b/packages/core/tests/integration/storage-implementations.test.ts
index b94e9984..cb89f175 100644
--- a/packages/core/tests/integration/storage-implementations.test.ts
+++ b/packages/core/tests/integration/storage-implementations.test.ts
@@ -243,9 +243,11 @@ describe('存储实现通用测试', () => {
 
       describe('HistoryManager 集成测试', () => {
         let historyManager: HistoryManager;
+        let modelManager: ModelManager;
 
         beforeEach(() => {
-          historyManager = new HistoryManager(storageProvider);
+          modelManager = createModelManager(storageProvider);
+          historyManager = new HistoryManager(storageProvider, modelManager);
         });
 
         it('应该能够添加和获取历史记录', async () => {
@@ -341,7 +343,7 @@ describe('存储实现通用测试', () => {
           };
 
           await templateManager.saveTemplate(template);
-          const retrieved = templateManager.getTemplate('test-template');
+          const retrieved = await templateManager.getTemplate('test-template');
 
           expect(retrieved.id).toBe('test-template');
           expect(retrieved.name).toBe('Test Template');
@@ -377,7 +379,7 @@ describe('存储实现通用测试', () => {
           await templateManager.saveTemplate(template1);
           await templateManager.saveTemplate(template2);
 
-          const templates = templateManager.listTemplates();
+          const templates = await templateManager.listTemplates();
           const userTemplates = templates.filter(t => !t.isBuiltin);
 
           expect(userTemplates).toHaveLength(2);