ai-world-sdk 1.0.5 → 1.0.7

package/README.md

@@ -98,7 +98,10 @@ console.log('图像 URL:', result.data[0]?.url);
 const geminiClient = new GeminiImageGenerationClient({});
 const geminiResult = await geminiClient.generate({
   prompt: 'A futuristic city',
+  model: 'gemini-3-pro-image-preview', // 使用 Gemini 3 Pro 模型
   aspect_ratio: '16:9',
+  image_size: '2K', // 仅适用于 gemini-3-pro-image-preview
+  response_modalities: ['IMAGE'], // 仅返回图片，不返回文本
 });
 console.log('图像 URL:', geminiResult.data[0]?.url);
 ```
@@ -211,15 +214,49 @@ const result = await client.generate({
 ```typescript
 const client = new GeminiImageGenerationClient({});
 
+// 基础用法（使用 Gemini 2.5 Flash - 快速模型）
 const result = await client.generate({
   prompt: 'A beautiful landscape',              // 必需
-  model: 'gemini-2.0-flash-exp-image-generation', // 可选，默认值
-  number_of_images: 1,                          // 可选
-  aspect_ratio: '16:9',                         // 可选: 1:1, 9:16, 16:9, 4:3, 3:4
-  temperature: 0.7,                             // 可选
+  model: 'gemini-2.5-flash-image',              // 推荐：快速、高效
+  aspect_ratio: '16:9',                         // 可选: 1:1, 2:3, 3:2, 3:4, 4:3, 4:5, 5:4, 9:16, 16:9, 21:9
+  number_of_images: 1,                          // 可选: 1-4
+});
+
+// 高级用法（使用 Gemini 3 Pro - 专业模型，支持高分辨率）
+const result2 = await client.generate({
+  prompt: 'A futuristic city at sunset',
+  model: 'gemini-3-pro-image-preview',         // 专业级模型
+  aspect_ratio: '16:9',
+  image_size: '2K',                             // 可选: 1K, 2K, 4K（仅适用于 gemini-3-pro-image-preview）
+  number_of_images: 1,
+  response_modalities: ['IMAGE'],              // 仅返回图片，不返回文本
+  temperature: 0.7,                             // 可选: 0.0-2.0
+  max_output_tokens: 1000,                      // 可选
 });
 ```
 
+**参数说明：**
+
+| 参数 | 类型 | 说明 | 默认值 |
+|------|------|------|--------|
+| `prompt` | `string` | 图像生成提示词（必需） | - |
+| `model` | `string` | 模型名称 | `gemini-2.0-flash-exp-image-generation` |
+| `aspect_ratio` | `string` | 宽高比 | - |
+| `image_size` | `string` | 图片大小（仅 gemini-3-pro-image-preview） | - |
+| `number_of_images` | `number` | 生成图片数量 | `1` |
+| `response_modalities` | `array` | 响应模态 | `['TEXT', 'IMAGE']` |
+| `temperature` | `number` | 温度参数 | `0.7` |
+| `max_output_tokens` | `number` | 最大输出 token 数 | `1000` |
+
+**支持的宽高比：** `1:1`, `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9`
+
+**模型对比：**
+
+| 模型 | 分辨率 | 特点 | 适用场景 |
+|------|--------|------|----------|
+| `gemini-2.5-flash-image` | 1024px | 快速、高效、成本低 | 日常使用、批量生成 |
+| `gemini-3-pro-image-preview` | 1K/2K/4K | 专业级、高分辨率、高级功能 | 专业设计、高分辨率需求 |
+
 ### 视频生成
 
 #### VideoGenerationClient
@@ -444,6 +481,8 @@ const response = await modelWithTools.invoke([
 
 ### 图像生成工作流
 
+#### 豆包图像生成
+
 ```typescript
 import { DoubaoImageGenerationClient } from 'ai-world-sdk';
 
@@ -470,6 +509,38 @@ multiResult.data.forEach((image, index) => {
 });
 ```
 
+#### Gemini 图像生成
+
+```typescript
+import { GeminiImageGenerationClient } from 'ai-world-sdk';
+
+const client = new GeminiImageGenerationClient({});
+
+// 使用 Gemini 2.5 Flash（快速模型）
+const result = await client.generate({
+  prompt: 'A beautiful sunset over the ocean',
+  model: 'gemini-2.5-flash-image',
+  aspect_ratio: '16:9',
+  number_of_images: 1,
+});
+
+console.log('图像 URL:', result.data[0]?.url || 'Base64 编码');
+if (result.text) {
+  console.log('图像描述:', result.text);
+}
+
+// 使用 Gemini 3 Pro（专业模型，支持高分辨率）
+const result2 = await client.generate({
+  prompt: 'A futuristic city at night',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '21:9', // 超宽屏
+  image_size: '4K',    // 4K 分辨率
+  response_modalities: ['IMAGE'], // 仅返回图片
+});
+
+console.log('4K 图像:', result2.data[0]?.b64_json ? 'Base64 编码' : result2.data[0]?.url);
+```
+
 ### 视频生成工作流
 
 ```typescript
@@ -519,7 +590,14 @@ if (result.status === 'succeeded') {
 ### 图像生成模型
 
 - **豆包 Seedream**: `doubao-seedream-4-5-251128` (默认)
-- **Google Gemini**: `gemini-2.0-flash-exp-image-generation` (默认)
+- **Google Gemini**:
+  - `gemini-2.5-flash-image` (Nano Banana) - **推荐**，快速、高效，1024px 分辨率，支持所有宽高比
+  - `gemini-3-pro-image-preview` (Nano Banana Pro) - 专业级，支持 1K/2K/4K 分辨率，支持 Google 搜索、思考模式，最多 14 张参考图片
+  - `gemini-2.0-flash-exp-image-generation` (已弃用，建议使用 `gemini-2.5-flash-image`)
+
+**模型选择建议：**
+- **日常使用**: `gemini-2.5-flash-image` - 速度快，成本低
+- **专业需求**: `gemini-3-pro-image-preview` - 高分辨率、高级功能（需要 `image_size` 参数）
 
 ### 视频生成模型
 

package/dist/__tests__/example.test.js

@@ -589,10 +589,12 @@ describe("Langchain SDK Tests", () => {
     test("GeminiImageGenerationClient - 基础图像生成", async () => {
         const imageClient = new index_1.GeminiImageGenerationClient({});
         const result = await imageClient.generate({
-            prompt: "A beautiful sunset over the ocean with vibrant colors",
-            model: "gemini-2.0-flash-exp-image-generation",
+            prompt: 'A beautiful sunset over the ocean',
+            model: 'gemini-3-pro-image-preview',
+            aspect_ratio: '16:9',
+            image_size: '2K', // 仅适用于 gemini-3-pro-image-preview
             number_of_images: 1,
-            aspect_ratio: "16:9",
+            response_modalities: ['IMAGE'], // 仅返回图片
         });
         expect(result).toBeDefined();
         expect(result.created).toBeDefined();
@@ -606,7 +608,7 @@ describe("Langchain SDK Tests", () => {
         });
         console.log("✅ GeminiImageGenerationClient 基础测试成功");
         console.log(`生成图像数量: ${result.data.length}`);
-        console.log("图像 URL:", result.data[0]?.url || "Base64 编码");
+        console.log("图像 URL:", result.data[0]?.url || ("Base64 编码" + result.data[0]?.b64_json));
         if (result.text) {
             console.log("图像描述:", result.text);
         }

package/dist/gemini-image-generation.d.ts

@@ -10,9 +10,11 @@ export interface GeminiImageGenerationRequest {
     prompt: string;
     model?: string;
     number_of_images?: number;
-    aspect_ratio?: "1:1" | "9:16" | "16:9" | "4:3" | "3:4";
+    aspect_ratio?: "1:1" | "2:3" | "3:2" | "3:4" | "4:3" | "4:5" | "5:4" | "9:16" | "16:9" | "21:9";
+    image_size?: "1K" | "2K" | "4K";
     temperature?: number;
     max_output_tokens?: number;
+    response_modalities?: ("TEXT" | "IMAGE")[];
     user?: string;
 }
 export interface GeminiImageData {
@@ -31,6 +33,12 @@ export declare class GeminiImageGenerationClient {
     /**
      * Generate images
      * 生成图像
+     *
+     * 支持的参数：
+     * - aspect_ratio: 宽高比，支持 1:1, 2:3, 3:2, 3:4, 4:3, 4:5, 5:4, 9:16, 16:9, 21:9
+     * - image_size: 图片大小（仅 gemini-3-pro-image-preview），支持 1K, 2K, 4K
+     * - response_modalities: 响应模态，["TEXT", "IMAGE"] 或 ["IMAGE"]
+     * - number_of_images: 生成图片数量（1-4）
      */
     generate(request: GeminiImageGenerationRequest): Promise<GeminiImageGenerationResponse>;
 }

package/dist/gemini-image-generation.js

@@ -22,6 +22,12 @@ class GeminiImageGenerationClient {
     /**
      * Generate images
      * 生成图像
+     *
+     * 支持的参数：
+     * - aspect_ratio: 宽高比，支持 1:1, 2:3, 3:2, 3:4, 4:3, 4:5, 5:4, 9:16, 16:9, 21:9
+     * - image_size: 图片大小（仅 gemini-3-pro-image-preview），支持 1K, 2K, 4K
+     * - response_modalities: 响应模态，["TEXT", "IMAGE"] 或 ["IMAGE"]
+     * - number_of_images: 生成图片数量（1-4）
      */
     async generate(request) {
         const requestBody = {
@@ -33,12 +39,18 @@ class GeminiImageGenerationClient {
         if (request.aspect_ratio) {
             requestBody.aspect_ratio = request.aspect_ratio;
         }
+        if (request.image_size) {
+            requestBody.image_size = request.image_size;
+        }
         if (request.temperature !== undefined) {
             requestBody.temperature = request.temperature;
         }
         if (request.max_output_tokens) {
             requestBody.max_output_tokens = request.max_output_tokens;
         }
+        if (request.response_modalities) {
+            requestBody.response_modalities = request.response_modalities;
+        }
         if (request.user) {
             requestBody.user = request.user;
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-world-sdk",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "description": "TypeScript SDK for AI World Platform - Chat Models, Image Generation, and Video Generation",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",