@lobehub/chat 1.101.1 → 1.102.0

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+## [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>
+
+#### 💄 Styles
+
+- **misc**: Fix lobehub provider `/chat` in desktop.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Fix lobehub provider `/chat` in desktop, closes [#8508](https://github.com/lobehub/lobe-chat/issues/8508) ([c801f9c](https://github.com/lobehub/lobe-chat/commit/c801f9c))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.101.1](https://github.com/lobehub/lobe-chat/compare/v1.101.0...v1.101.1)
 
 <sup>Released on **2025-07-19**</sup>

package/Dockerfile

@@ -53,6 +53,8 @@ ENV NEXT_PUBLIC_SENTRY_DSN="${NEXT_PUBLIC_SENTRY_DSN}" \
     SENTRY_ORG="" \
     SENTRY_PROJECT=""
 
+ENV APP_URL="http://app.com"
+
 # Posthog
 ENV NEXT_PUBLIC_ANALYTICS_POSTHOG="${NEXT_PUBLIC_ANALYTICS_POSTHOG}" \
     NEXT_PUBLIC_POSTHOG_HOST="${NEXT_PUBLIC_POSTHOG_HOST}" \

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": {
+      "features": [
+        "Add image generation capabilities using Google AI Imagen API."
+      ]
+    },
+    "date": "2025-07-21",
+    "version": "1.102.0"
+  },
+  {
+    "children": {
+      "improvements": [
+        "Fix lobehub provider /chat in desktop."
+      ]
+    },
+    "date": "2025-07-21",
+    "version": "1.101.2"
+  },
   {
     "children": {
       "fixes": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.101.1",
+  "version": "1.102.0",
   "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/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/index.ts

@@ -156,6 +156,7 @@ export { default as infiniai } from './infiniai';
 export { default as internlm } from './internlm';
 export { default as jina } from './jina';
 export { default as lmstudio } from './lmstudio';
+export { default as lobehub } from './lobehub';
 export { default as minimax } from './minimax';
 export { default as mistral } from './mistral';
 export { default as modelscope } from './modelscope';

package/src/config/aiModels/lobehub.ts

@@ -0,0 +1,7 @@
+import { AIChatModelCard } from '@/types/aiModel';
+
+const lobehubChatModels: AIChatModelCard[] = [];
+
+export const allModels = [...lobehubChatModels];
+
+export default allModels;

package/src/config/modelProviders/index.ts

@@ -200,6 +200,7 @@ export { default as InfiniAIProviderCard } from './infiniai';
 export { default as InternLMProviderCard } from './internlm';
 export { default as JinaProviderCard } from './jina';
 export { default as LMStudioProviderCard } from './lmstudio';
+export { default as LobeHubProviderCard } from './lobehub';
 export { default as MinimaxProviderCard } from './minimax';
 export { default as MistralProviderCard } from './mistral';
 export { default as ModelScopeProviderCard } from './modelscope';

package/src/config/modelProviders/lobehub.ts

@@ -0,0 +1,22 @@
+import { ModelProviderCard } from '@/types/llm';
+
+const LobeHub: ModelProviderCard = {
+  chatModels: [],
+  description:
+    'LobeChat Cloud 通过官方部署的 API 来实现 AI 模型的调用，并采用 Credits 计算积分的方式来衡量 AI 模型的用量，对应大模型使用的 Tokens。',
+  enabled: true,
+  id: 'lobehub',
+  modelsUrl: 'https://lobehub.com/zh/docs/usage/subscription/model-pricing',
+  name: 'LobeHub',
+  settings: {
+    modelEditable: false,
+    showAddNewModel: false,
+    showModelFetcher: false,
+  },
+  showConfig: false,
+  url: 'https://lobehub.com',
+};
+
+export default LobeHub;
+
+export const planCardModels = ['gpt-4o-mini', 'deepseek-reasoner', 'claude-3-5-sonnet-latest'];

package/src/features/ModelSwitchPanel/index.tsx

@@ -148,7 +148,7 @@ const ModelSwitchPanel = memo<IProps>(({ children, onOpenChange, open }) => {
         // 不加限高就会导致面板超长，顶部的内容会被隐藏
         // https://github.com/user-attachments/assets/9c043c47-42c5-46ef-b5c1-bee89376f042
         style: {
-          maxHeight: 500,
+          maxHeight: 550,
           overflowY: 'scroll',
         },
       }}

package/src/libs/model-runtime/ModelRuntime.test.ts

@@ -72,7 +72,7 @@ beforeEach(async () => {
 
 describe('AgentRuntime', () => {
   describe('should initialize with various providers', () => {
-    const providers = Object.values(ModelProvider);
+    const providers = Object.values(ModelProvider).filter((i) => i !== 'lobehub');
 
     const specialProviderIds = [ModelProvider.VertexAI, ...specialProviders.map((p) => p.id)];
 

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/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;
@@ -45,6 +51,7 @@ export enum ModelProvider {
   InternLM = 'internlm',
   Jina = 'jina',
   LMStudio = 'lmstudio',
+  LobeHub = 'lobehub',
   Minimax = 'minimax',
   Mistral = 'mistral',
   ModelScope = 'modelscope',

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,
 };