@lobehub/chat 1.101.2 → 1.102.1

package/CHANGELOG.md

@@ -2,6 +2,64 @@
 
 # Changelog
 
+### [Version 1.102.1](https://github.com/lobehub/lobe-chat/compare/v1.102.0...v1.102.1)
+
+<sup>Released on **2025-07-21**</sup>
+
+#### 🐛 Bug Fixes
+
+- **groq**: Enable streaming for tool calls and add Kimi K2 model.
+
+#### 💄 Styles
+
+- **misc**: Modal list header sticky style.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **groq**: Enable streaming for tool calls and add Kimi K2 model, closes [#8510](https://github.com/lobehub/lobe-chat/issues/8510) ([60739bc](https://github.com/lobehub/lobe-chat/commit/60739bc))
+
+#### Styles
+
+- **misc**: Modal list header sticky style, closes [#8514](https://github.com/lobehub/lobe-chat/issues/8514) ([75273d5](https://github.com/lobehub/lobe-chat/commit/75273d5))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+## [Version 1.102.0](https://github.com/lobehub/lobe-chat/compare/v1.101.2...v1.102.0)
+
+<sup>Released on **2025-07-21**</sup>
+
+#### ✨ Features
+
+- **misc**: Add image generation capabilities using Google AI Imagen API.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's improved
+
+- **misc**: Add image generation capabilities using Google AI Imagen API, closes [#8503](https://github.com/lobehub/lobe-chat/issues/8503) ([cef8208](https://github.com/lobehub/lobe-chat/commit/cef8208))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.101.2](https://github.com/lobehub/lobe-chat/compare/v1.101.1...v1.101.2)
 
 <sup>Released on **2025-07-21**</sup>

package/README.md

@@ -383,8 +383,8 @@ In addition, these plugins are not limited to news aggregation, but can also ext
 
 | Recent Submits                                                                                                               | Description                                                                                                             |
 | ---------------------------------------------------------------------------------------------------------------------------- | ----------------------------------------------------------------------------------------------------------------------- |
+| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2025-07-21**</sup>        | Analyze stocks and get comprehensive real-time investment data and analytics.<br/>`stock`                               |
 | [Speak](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2025-07-18**</sup>                            | Learn how to say anything in another language with Speak, your AI-powered language tutor.<br/>`education` `language`    |
-| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2025-05-27**</sup>        | Analyze stocks and get comprehensive real-time investment data and analytics.<br/>`stock`                               |
 | [Web](https://lobechat.com/discover/plugin/web)<br/><sup>By **Proghit** on **2025-01-24**</sup>                              | Smart web search that reads and analyzes pages to deliver comprehensive answers from Google results.<br/>`web` `search` |
 | [Bing_websearch](https://lobechat.com/discover/plugin/Bingsearch-identifier)<br/><sup>By **FineHow** on **2024-12-22**</sup> | Search for information from the internet base BingApi<br/>`bingsearch`                                                  |
 

package/README.zh-CN.md

@@ -376,8 +376,8 @@ LobeChat 的插件生态系统是其核心功能的重要扩展，它极大地
 
 | 最近新增                                                                                                                   | 描述                                                                               |
 | -------------------------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------- |
+| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2025-07-21**</sup>      | 分析股票并获取全面的实时投资数据和分析。<br/>`股票`                                |
 | [Speak](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2025-07-18**</sup>                          | 使用 Speak，您的 AI 语言导师，学习如何用另一种语言说任何事情。<br/>`教育` `语言`   |
-| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2025-05-27**</sup>      | 分析股票并获取全面的实时投资数据和分析。<br/>`股票`                                |
 | [网页](https://lobechat.com/discover/plugin/web)<br/><sup>By **Proghit** on **2025-01-24**</sup>                           | 智能网页搜索，读取和分析页面，以提供来自 Google 结果的全面答案。<br/>`网页` `搜索` |
 | [必应网页搜索](https://lobechat.com/discover/plugin/Bingsearch-identifier)<br/><sup>By **FineHow** on **2024-12-22**</sup> | 通过 BingApi 搜索互联网上的信息<br/>`bingsearch`                                   |
 

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "improvements": [
+        "Modal list header sticky style."
+      ]
+    },
+    "date": "2025-07-21",
+    "version": "1.102.1"
+  },
+  {
+    "children": {
+      "features": [
+        "Add image generation capabilities using Google AI Imagen API."
+      ]
+    },
+    "date": "2025-07-21",
+    "version": "1.102.0"
+  },
   {
     "children": {
       "improvements": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.101.2",
+  "version": "1.102.1",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",

package/src/app/[variants]/(main)/settings/provider/features/ModelList/ModelTitle/index.tsx

@@ -58,8 +58,10 @@ const ModelTitle = memo<ModelFetcherProps>(
         paddingBlock={8}
         style={{
           background: theme.colorBgContainer,
+          marginTop: mobile ? 0 : -12,
+          paddingTop: mobile ? 0 : 20,
           position: 'sticky',
-          top: mobile ? -2 : -16,
+          top: mobile ? -2 : -32,
           zIndex: 15,
         }}
       >

package/src/config/aiModels/google.ts

@@ -1,4 +1,5 @@
-import { AIChatModelCard } from '@/types/aiModel';
+import { ModelParamsSchema } from '@/libs/standard-parameters';
+import { AIChatModelCard, AIImageModelCard } from '@/types/aiModel';
 
 const googleChatModels: AIChatModelCard[] = [
   {
@@ -496,6 +497,38 @@ const googleChatModels: AIChatModelCard[] = [
   },
 ];
 
-export const allModels = [...googleChatModels];
+// Common parameters for Imagen models
+const imagenBaseParameters: ModelParamsSchema = {
+  aspectRatio: {
+    default: '1:1',
+    enum: ['1:1', '16:9', '9:16', '3:4', '4:3'],
+  },
+  prompt: { default: '' },
+};
+
+const googleImageModels: AIImageModelCard[] = [
+  {
+    description: 'Imagen 4th generation text-to-image model series',
+    displayName: 'Imagen4 Preview 06-06',
+    enabled: true,
+    id: 'imagen-4.0-generate-preview-06-06',
+    organization: 'Deepmind',
+    parameters: imagenBaseParameters,
+    releasedAt: '2024-06-06',
+    type: 'image',
+  },
+  {
+    description: 'Imagen 4th generation text-to-image model series Ultra version',
+    displayName: 'Imagen4 Ultra Preview 06-06',
+    enabled: true,
+    id: 'imagen-4.0-ultra-generate-preview-06-06',
+    organization: 'Deepmind',
+    parameters: imagenBaseParameters,
+    releasedAt: '2024-06-06',
+    type: 'image',
+  },
+];
+
+export const allModels = [...googleChatModels, ...googleImageModels];
 
 export default allModels;

package/src/config/aiModels/groq.ts

@@ -23,6 +23,23 @@ const groqChatModels: AIChatModelCard[] = [
     maxOutput: 8192,
     type: 'chat',
   },
+  {
+    abilities: {
+      functionCall: true,
+    },
+    contextWindowTokens: 131_072,
+    description:
+      'kimi-k2 是一款具备超强代码和 Agent 能力的 MoE 架构基础模型，总参数 1T，激活参数 32B。在通用知识推理、编程、数学、Agent 等主要类别的基准性能测试中，K2 模型的性能超过其他主流开源模型。',
+    displayName: 'Kimi K2 Instruct',
+    enabled: true,
+    id: 'moonshotai/kimi-k2-instruct',
+    pricing: {
+      input: 1,
+      output: 3,
+    },
+    releasedAt: '2025-07-11',
+    type: 'chat',
+  },
   {
     contextWindowTokens: 131_072,
     displayName: 'Llama 4 Scout (17Bx16E)',

package/src/libs/model-runtime/google/index.test.ts

@@ -4,6 +4,7 @@ import OpenAI from 'openai';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { OpenAIChatMessage } from '@/libs/model-runtime';
+import { CreateImagePayload } from '@/libs/model-runtime/types/image';
 import { ChatStreamPayload } from '@/types/openai/chat';
 import * as imageToBase64Module from '@/utils/imageToBase64';
 
@@ -22,7 +23,7 @@ let instance: LobeGoogleAI;
 beforeEach(() => {
   instance = new LobeGoogleAI({ apiKey: 'test' });
 
-  // 使用 vi.spyOn 来模拟 chat.completions.create 方法
+  // Use vi.spyOn to mock the chat.completions.create method
   const mockStreamData = (async function* (): AsyncGenerator<GenerateContentResponse> {})();
   vi.spyOn(instance['client'].models, 'generateContentStream').mockResolvedValue(mockStreamData);
 });
@@ -53,7 +54,7 @@ describe('LobeGoogleAI', () => {
       expect(result).toBeInstanceOf(Response);
     });
     it('should handle text messages correctly', async () => {
-      // 模拟 Google AI SDK 的 generateContentStream 方法返回一个成功的响应流
+      // Mock Google AI SDK's generateContentStream method to return a successful response stream
       const mockStream = new ReadableStream({
         start(controller) {
           controller.enqueue('Hello, world!');
@@ -71,7 +72,7 @@ describe('LobeGoogleAI', () => {
       });
 
       expect(result).toBeInstanceOf(Response);
-      // 额外的断言可以加入，比如验证返回的流内容等
+      // Additional assertions can be added, such as verifying the returned stream content
     });
 
     it('should withGrounding', () => {
@@ -214,10 +215,10 @@ describe('LobeGoogleAI', () => {
     });
 
     it('should call debugStream in DEBUG mode', async () => {
-      // 设置环境变量以启用DEBUG模式
+      // Set environment variable to enable DEBUG mode
       process.env.DEBUG_GOOGLE_CHAT_COMPLETION = '1';
 
-      // 模拟 Google AI SDK 的 generateContentStream 方法返回一个成功的响应流
+      // Mock Google AI SDK's generateContentStream method to return a successful response stream
       const mockStream = new ReadableStream({
         start(controller) {
           controller.enqueue('Debug mode test');
@@ -239,13 +240,13 @@ describe('LobeGoogleAI', () => {
 
       expect(debugStreamSpy).toHaveBeenCalled();
 
-      // 清理环境变量
+      // Clean up environment variable
       delete process.env.DEBUG_GOOGLE_CHAT_COMPLETION;
     });
 
     describe('Error', () => {
       it('should throw InvalidGoogleAPIKey error on API_KEY_INVALID error', async () => {
-        // 模拟 Google AI SDK 抛出异常
+        // Mock Google AI SDK throwing an exception
         const message = `[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:streamGenerateContent?alt=sse: [400 Bad Request] API key not valid. Please pass a valid API key. [{"@type":"type.googleapis.com/google.rpc.ErrorInfo","reason":"API_KEY_INVALID","domain":"googleapis.com","metadata":{"service":"generativelanguage.googleapis.com"}}]`;
 
         const apiError = new Error(message);
@@ -264,7 +265,7 @@ describe('LobeGoogleAI', () => {
       });
 
       it('should throw LocationNotSupportError error on location not support error', async () => {
-        // 模拟 Google AI SDK 抛出异常
+        // Mock Google AI SDK throwing an exception
         const message = `[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:streamGenerateContent?alt=sse: [400 Bad Request] User location is not supported for the API use.`;
 
         const apiError = new Error(message);
@@ -283,7 +284,7 @@ describe('LobeGoogleAI', () => {
       });
 
       it('should throw BizError error', async () => {
-        // 模拟 Google AI SDK 抛出异常
+        // Mock Google AI SDK throwing an exception
         const message = `[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:streamGenerateContent?alt=sse: [400 Bad Request] API key not valid. Please pass a valid API key. [{"@type":"type.googleapis.com/google.rpc.ErrorInfo","reason":"Error","domain":"googleapis.com","metadata":{"service":"generativelanguage.googleapis.com"}}]`;
 
         const apiError = new Error(message);
@@ -315,7 +316,7 @@ describe('LobeGoogleAI', () => {
       });
 
       it('should throw DefaultError error', async () => {
-        // 模拟 Google AI SDK 抛出异常
+        // Mock Google AI SDK throwing an exception
         const message = `[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:streamGenerateContent?alt=sse: [400 Bad Request] API key not valid. Please pass a valid API key. [{"@type":"type.googleapis.com/google.rpc.ErrorInfo","reason":"Error","domain":"googleapis.com","metadata":{"service":"generativelanguage.googleapis.com}}]`;
 
         const apiError = new Error(message);
@@ -345,7 +346,7 @@ describe('LobeGoogleAI', () => {
         // Arrange
         const apiError = new Error('Error message');
 
-        // 使用 vi.spyOn 来模拟 chat.completions.create 方法
+        // Use vi.spyOn to mock the chat.completions.create method
         vi.spyOn(instance['client'].models, 'generateContentStream').mockRejectedValue(apiError);
 
         // Act
@@ -537,7 +538,7 @@ describe('LobeGoogleAI', () => {
           },
         ];
 
-        // 调用 buildGoogleMessages 方法
+        // Call the buildGoogleMessages method
         const contents = await instance['buildGoogleMessages'](messages);
 
         expect(contents).toHaveLength(1);
@@ -826,4 +827,331 @@ describe('LobeGoogleAI', () => {
       });
     });
   });
+
+  describe('createImage', () => {
+    it('should create image successfully with basic parameters', async () => {
+      // Arrange - Use real base64 image data (5x5 red pixel PNG)
+      const realBase64ImageData =
+        'iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAYAAACNbyblAAAAHElEQVQI12P4//8/w38GIAXDIBKE0DHxgljNBAAO9TXL0Y4OHwAAAABJRU5ErkJggg==';
+      const mockImageResponse = {
+        generatedImages: [
+          {
+            image: {
+              imageBytes: realBase64ImageData,
+            },
+          },
+        ],
+      };
+      vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+        mockImageResponse as any,
+      );
+
+      const payload: CreateImagePayload = {
+        model: 'imagen-4.0-generate-preview-06-06',
+        params: {
+          prompt: 'A beautiful landscape with mountains and trees',
+          aspectRatio: '1:1',
+        },
+      };
+
+      // Act
+      const result = await instance.createImage(payload);
+
+      // Assert
+      expect(instance['client'].models.generateImages).toHaveBeenCalledWith({
+        model: 'imagen-4.0-generate-preview-06-06',
+        prompt: 'A beautiful landscape with mountains and trees',
+        config: {
+          aspectRatio: '1:1',
+          numberOfImages: 1,
+        },
+      });
+      expect(result).toEqual({
+        imageUrl: `data:image/png;base64,${realBase64ImageData}`,
+      });
+    });
+
+    it('should support different aspect ratios like 16:9 for widescreen images', async () => {
+      // Arrange - Use real base64 data
+      const realBase64Data =
+        'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==';
+      const mockImageResponse = {
+        generatedImages: [
+          {
+            image: {
+              imageBytes: realBase64Data,
+            },
+          },
+        ],
+      };
+      vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+        mockImageResponse as any,
+      );
+
+      const payload: CreateImagePayload = {
+        model: 'imagen-4.0-ultra-generate-preview-06-06',
+        params: {
+          prompt: 'Cinematic landscape shot with dramatic lighting',
+          aspectRatio: '16:9',
+        },
+      };
+
+      // Act
+      await instance.createImage(payload);
+
+      // Assert
+      expect(instance['client'].models.generateImages).toHaveBeenCalledWith({
+        model: 'imagen-4.0-ultra-generate-preview-06-06',
+        prompt: 'Cinematic landscape shot with dramatic lighting',
+        config: {
+          aspectRatio: '16:9',
+          numberOfImages: 1,
+        },
+      });
+    });
+
+    it('should work with only prompt when aspect ratio is not specified', async () => {
+      // Arrange
+      const realBase64Data =
+        'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==';
+      const mockImageResponse = {
+        generatedImages: [
+          {
+            image: {
+              imageBytes: realBase64Data,
+            },
+          },
+        ],
+      };
+      vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+        mockImageResponse as any,
+      );
+
+      const payload: CreateImagePayload = {
+        model: 'imagen-4.0-generate-preview-06-06',
+        params: {
+          prompt: 'A cute cat sitting in a garden',
+        },
+      };
+
+      // Act
+      await instance.createImage(payload);
+
+      // Assert
+      expect(instance['client'].models.generateImages).toHaveBeenCalledWith({
+        model: 'imagen-4.0-generate-preview-06-06',
+        prompt: 'A cute cat sitting in a garden',
+        config: {
+          aspectRatio: undefined,
+          numberOfImages: 1,
+        },
+      });
+    });
+
+    describe('Error handling', () => {
+      it('should throw InvalidProviderAPIKey error when API key is invalid', async () => {
+        // Arrange - Use real Google AI error format
+        const message = `[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/imagen-4.0:generateImages: [400 Bad Request] API key not valid. Please pass a valid API key. [{"@type":"type.googleapis.com/google.rpc.ErrorInfo","reason":"API_KEY_INVALID","domain":"googleapis.com","metadata":{"service":"generativelanguage.googleapis.com"}}]`;
+        const apiError = new Error(message);
+        vi.spyOn(instance['client'].models, 'generateImages').mockRejectedValue(apiError);
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-4.0-generate-preview-06-06',
+          params: {
+            prompt: 'A realistic landscape photo',
+          },
+        };
+
+        // Act & Assert - Test error type rather than specific text
+        await expect(instance.createImage(payload)).rejects.toEqual(
+          expect.objectContaining({
+            errorType: invalidErrorType,
+            provider,
+          }),
+        );
+      });
+
+      it('should throw ProviderBizError for network and API errors', async () => {
+        // Arrange
+        const apiError = new Error('Network connection failed');
+        vi.spyOn(instance['client'].models, 'generateImages').mockRejectedValue(apiError);
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-4.0-generate-preview-06-06',
+          params: {
+            prompt: 'A digital art portrait',
+          },
+        };
+
+        // Act & Assert - Test error type and basic structure
+        await expect(instance.createImage(payload)).rejects.toEqual(
+          expect.objectContaining({
+            errorType: bizErrorType,
+            provider,
+            error: expect.objectContaining({
+              message: expect.any(String),
+            }),
+          }),
+        );
+      });
+
+      it('should throw error when API response is malformed - missing generatedImages', async () => {
+        // Arrange
+        const mockImageResponse = {};
+        vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+          mockImageResponse as any,
+        );
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-4.0-generate-preview-06-06',
+          params: {
+            prompt: 'Abstract geometric patterns',
+          },
+        };
+
+        // Act & Assert - Test error behavior rather than specific text
+        await expect(instance.createImage(payload)).rejects.toEqual(
+          expect.objectContaining({
+            errorType: bizErrorType,
+            provider,
+          }),
+        );
+      });
+
+      it('should throw error when API response contains empty image array', async () => {
+        // Arrange
+        const mockImageResponse = {
+          generatedImages: [],
+        };
+        vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+          mockImageResponse as any,
+        );
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-4.0-generate-preview-06-06',
+          params: {
+            prompt: 'Minimalist design poster',
+          },
+        };
+
+        // Act & Assert
+        await expect(instance.createImage(payload)).rejects.toEqual(
+          expect.objectContaining({
+            errorType: bizErrorType,
+            provider,
+          }),
+        );
+      });
+
+      it('should throw error when generated image lacks required data', async () => {
+        // Arrange
+        const mockImageResponse = {
+          generatedImages: [
+            {
+              image: {}, // Missing imageBytes
+            },
+          ],
+        };
+        vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+          mockImageResponse as any,
+        );
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-4.0-generate-preview-06-06',
+          params: {
+            prompt: 'Watercolor painting style',
+          },
+        };
+
+        // Act & Assert
+        await expect(instance.createImage(payload)).rejects.toEqual(
+          expect.objectContaining({
+            errorType: bizErrorType,
+            provider,
+          }),
+        );
+      });
+    });
+
+    describe('Edge cases', () => {
+      it('should return first image when API returns multiple generated images', async () => {
+        // Arrange - Use two different real base64 image data
+        const firstImageData =
+          'iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAYAAACNbyblAAAAHElEQVQI12P4//8/w38GIAXDIBKE0DHxgljNBAAO9TXL0Y4OHwAAAABJRU5ErkJggg==';
+        const secondImageData =
+          'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==';
+        const mockImageResponse = {
+          generatedImages: [
+            {
+              image: {
+                imageBytes: firstImageData,
+              },
+            },
+            {
+              image: {
+                imageBytes: secondImageData,
+              },
+            },
+          ],
+        };
+        vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+          mockImageResponse as any,
+        );
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-4.0-generate-preview-06-06',
+          params: {
+            prompt: 'Generate multiple variations of a sunset',
+          },
+        };
+
+        // Act
+        const result = await instance.createImage(payload);
+
+        // Assert - Should return the first image
+        expect(result).toEqual({
+          imageUrl: `data:image/png;base64,${firstImageData}`,
+        });
+      });
+
+      it('should work with custom future Imagen model versions', async () => {
+        // Arrange
+        const realBase64Data =
+          'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==';
+        const mockImageResponse = {
+          generatedImages: [
+            {
+              image: {
+                imageBytes: realBase64Data,
+              },
+            },
+          ],
+        };
+        vi.spyOn(instance['client'].models, 'generateImages').mockResolvedValue(
+          mockImageResponse as any,
+        );
+
+        const payload: CreateImagePayload = {
+          model: 'imagen-5.0-future-model',
+          params: {
+            prompt: 'Photorealistic portrait with soft lighting',
+            aspectRatio: '4:3',
+          },
+        };
+
+        // Act
+        await instance.createImage(payload);
+
+        // Assert
+        expect(instance['client'].models.generateImages).toHaveBeenCalledWith({
+          model: 'imagen-5.0-future-model',
+          prompt: 'Photorealistic portrait with soft lighting',
+          config: {
+            aspectRatio: '4:3',
+            numberOfImages: 1,
+          },
+        });
+      });
+    });
+  });
 });

package/src/libs/model-runtime/google/index.ts

@@ -21,6 +21,7 @@ import {
   OpenAIChatMessage,
   UserMessageContentPart,
 } from '../types';
+import { CreateImagePayload, CreateImageResponse } from '../types/image';
 import { AgentRuntimeError } from '../utils/createError';
 import { debugStream } from '../utils/debugStream';
 import { StreamingResponse } from '../utils/response';
@@ -243,6 +244,52 @@ export class LobeGoogleAI implements LobeRuntimeAI {
     }
   }
 
+  /**
+   * Generate images using Google AI Imagen API
+   * @see https://ai.google.dev/gemini-api/docs/image-generation#imagen
+   */
+  async createImage(payload: CreateImagePayload): Promise<CreateImageResponse> {
+    try {
+      const { model, params } = payload;
+
+      const response = await this.client.models.generateImages({
+        config: {
+          aspectRatio: params.aspectRatio,
+          numberOfImages: 1,
+        },
+        model,
+        prompt: params.prompt,
+      });
+
+      if (!response.generatedImages || response.generatedImages.length === 0) {
+        throw new Error('No images generated');
+      }
+
+      const generatedImage = response.generatedImages[0];
+      if (!generatedImage.image || !generatedImage.image.imageBytes) {
+        throw new Error('Invalid image data');
+      }
+
+      const { imageBytes } = generatedImage.image;
+      // 1. official doc use png as example
+      // 2. no responseType param support like openai now.
+      // I think we can just hard code png now
+      const imageUrl = `data:image/png;base64,${imageBytes}`;
+
+      return { imageUrl };
+    } catch (error) {
+      const err = error as Error;
+      console.error('Google AI image generation error:', err);
+
+      const { errorType, error: parsedError } = this.parseErrorMessage(err.message);
+      throw AgentRuntimeError.createImage({
+        error: parsedError,
+        errorType,
+        provider: this.provider,
+      });
+    }
+  }
+
   private createEnhancedStream(originalStream: any, signal: AbortSignal): ReadableStream {
     return new ReadableStream({
       async start(controller) {

package/src/libs/model-runtime/groq/index.test.ts

@@ -34,25 +34,7 @@ afterEach(() => {
 
 describe('LobeGroqAI Temperature Tests', () => {
   describe('handlePayload option', () => {
-    it('should set stream to false when payload contains tools', async () => {
-      const mockCreateMethod = vi
-        .spyOn(instance['client'].chat.completions, 'create')
-        .mockResolvedValue({
-          id: 'chatcmpl-8xDx5AETP8mESQN7UB30GxTN2H1SO',
-          object: 'chat.completion',
-          created: 1709125675,
-          model: 'mistralai/mistral-7b-instruct:free',
-          system_fingerprint: 'fp_86156a94a0',
-          choices: [
-            {
-              index: 0,
-              message: { role: 'assistant', content: 'hello', refusal: null },
-              logprobs: null,
-              finish_reason: 'stop',
-            },
-          ],
-        });
-
+    it('should not set stream to false when payload contains tools', async () => {
       await instance.chat({
         messages: [{ content: 'Hello', role: 'user' }],
         model: 'mistralai/mistral-7b-instruct:free',
@@ -65,8 +47,8 @@ describe('LobeGroqAI Temperature Tests', () => {
         ],
       });
 
-      expect(mockCreateMethod).toHaveBeenCalledWith(
-        expect.objectContaining({ stream: false }),
+      expect(instance['client'].chat.completions.create).toHaveBeenCalledWith(
+        expect.objectContaining({ stream: true }),
         expect.anything(),
       );
     });

package/src/libs/model-runtime/groq/index.ts

@@ -21,8 +21,7 @@ export const LobeGroq = createOpenAICompatibleRuntime({
       const { temperature, ...restPayload } = payload;
       return {
         ...restPayload,
-        // disable stream for tools due to groq dont support
-        stream: !payload.tools,
+        stream: payload.stream ?? true,
 
         temperature: temperature <= 0 ? undefined : temperature,
       } as any;

package/src/libs/model-runtime/types/type.ts

@@ -16,6 +16,12 @@ export interface ChatCompletionErrorPayload {
   provider: string;
 }
 
+export interface CreateImageErrorPayload {
+  error: object;
+  errorType: ILobeAgentRuntimeErrorType;
+  provider: string;
+}
+
 export interface CreateChatCompletionOptions {
   chatModel: OpenAI;
   payload: ChatStreamPayload;

package/src/libs/model-runtime/utils/createError.ts

@@ -1,5 +1,9 @@
 import { ILobeAgentRuntimeErrorType } from '../error';
-import { AgentInitErrorPayload, ChatCompletionErrorPayload } from '../types';
+import {
+  AgentInitErrorPayload,
+  ChatCompletionErrorPayload,
+  CreateImageErrorPayload,
+} from '../types';
 
 export const AgentRuntimeError = {
   chat: (error: ChatCompletionErrorPayload): ChatCompletionErrorPayload => error,
@@ -7,5 +11,6 @@ export const AgentRuntimeError = {
     errorType: ILobeAgentRuntimeErrorType | string | number,
     error?: any,
   ): AgentInitErrorPayload => ({ error, errorType }),
+  createImage: (error: CreateImageErrorPayload): CreateImageErrorPayload => error,
   textToImage: (error: any): any => error,
 };

package/src/store/user/slices/auth/action.ts

@@ -39,10 +39,10 @@ export const createAuthSlice: StateCreator<
   },
   openLogin: async () => {
     if (enableClerk) {
-      const reditectUrl = location.toString();
+      const redirectUrl = location.toString();
       get().clerkSignIn?.({
-        fallbackRedirectUrl: reditectUrl,
-        signUpForceRedirectUrl: reditectUrl,
+        fallbackRedirectUrl: redirectUrl,
+        signUpForceRedirectUrl: redirectUrl,
         signUpUrl: '/signup',
       });