ai-world-sdk 1.0.7 → 1.0.8

package/README.md

@@ -29,7 +29,7 @@ npm install ai-world-sdk
 import { sdkConfig } from 'ai-world-sdk';
 
 // 设置全局配置（只需一次）
-sdkConfig.setBaseUrl('http://localhost:8000');
+sdkConfig.setBaseUrl('your-base-url');
 sdkConfig.setToken('your-jwt-token');
 sdkConfig.setDebug(true); // 可选：启用调试模式
 
@@ -233,6 +233,20 @@ const result2 = await client.generate({
   temperature: 0.7,                             // 可选: 0.0-2.0
   max_output_tokens: 1000,                      // 可选
 });
+
+// 使用 aihubmix provider（通过 aihubmix 代理服务）
+const aihubmixClient = new GeminiImageGenerationClient({
+  provider: 'aihubmix',  // 使用 aihubmix 代理
+});
+
+const result3 = await aihubmixClient.generate({
+  prompt: 'A beautiful landscape with mountains',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '16:9',
+  image_size: '1K',
+  number_of_images: 1,
+  response_modalities: ['IMAGE'],
+});
 ```
 
 **参数说明：**
@@ -250,6 +264,11 @@ const result2 = await client.generate({
 
 **支持的宽高比：** `1:1`, `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9`
 
+**Provider 说明：**
+
+- `provider: 'gemini'`（默认）- 直接使用 Google Gemini 官方 API
+- `provider: 'aihubmix'` - 通过 aihubmix 代理服务调用，适合需要统一管理多个 API 的场景
+
 **模型对比：**
 
 | 模型 | 分辨率 | 特点 | 适用场景 |
@@ -290,7 +309,7 @@ const result = await client.poll(task.id, {
 import { sdkConfig } from 'ai-world-sdk';
 
 // 设置配置
-sdkConfig.setBaseUrl('http://localhost:8000');
+sdkConfig.setBaseUrl('your-base-url');
 sdkConfig.setToken('your-jwt-token');
 sdkConfig.setHeaders({ 'X-Custom-Header': 'value' });
 sdkConfig.setDebug(true);
@@ -345,11 +364,26 @@ const gptModel = new ChatOpenAI({
 
 // Gemini 模型（通过 aihubmix）
 const geminiModel = new ChatGoogleGenerativeAI({
-  modelName: 'gemini-2.5-flash-image',
+  modelName: 'gemini-2.5-flash',
   temperature: 0.7,
   provider: 'aihubmix',
 });
 
+// Gemini 图像生成（通过 aihubmix）
+import { GeminiImageGenerationClient } from 'ai-world-sdk';
+
+const geminiImageClient = new GeminiImageGenerationClient({
+  provider: 'aihubmix',
+});
+
+const imageResult = await geminiImageClient.generate({
+  prompt: 'A beautiful landscape',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '16:9',
+  image_size: '1K',
+  response_modalities: ['IMAGE'],
+});
+
 // Claude 模型（通过 aihubmix）
 const claudeModel = new ChatAnthropic({
   modelName: 'claude-3-sonnet-20240229',

package/dist/__tests__/example.test.js

@@ -228,38 +228,6 @@ describe("Langchain SDK Tests", () => {
             console.log("响应内容:", content);
         }
     }, 60000);
-    test("ChatGoogleGenerativeAI - aihubmix.com 图像生成", async () => {
-        const gemini = new index_1.ChatGoogleGenerativeAI({
-            modelName: "gemini-3-pro-image-preview",
-            temperature: 0.7,
-            provider: "aihubmix",
-        });
-        const response = await gemini.invoke([
-            new index_1.HumanMessage("请生成一张展示人工智能概念的图片"),
-        ]);
-        expect(response).toBeDefined();
-        expect(response.content).toBeDefined();
-        // 图像生成响应可能包含文本和图像数据
-        // 检查响应内容（可能是字符串、数组或包含图像的对象）
-        const content = response.content;
-        if (typeof content === "string") {
-            console.log("✅ aihubmix.com 图像生成测试成功（文本响应）");
-            console.log("AI:", content);
-        }
-        else if (Array.isArray(content)) {
-            // 多模态响应，可能包含图像
-            const hasImage = content.some((item) => item?.type === "image_url" ||
-                item?.image_url ||
-                (typeof item === "string" && item.startsWith("data:image")));
-            console.log("✅ aihubmix.com 图像生成测试成功（多模态响应）");
-            console.log("响应类型: 数组，包含图像:", hasImage);
-            console.log("响应内容:", JSON.stringify(content, null, 2));
-        }
-        else {
-            console.log("✅ aihubmix.com 图像生成测试成功（其他格式）");
-            console.log("响应内容:", content);
-        }
-    }, 60000);
     test("DoubaoImageGenerationClient - 基础图像生成", async () => {
         const imageClient = new index_1.DoubaoImageGenerationClient({});
         const result = await imageClient.generate({
@@ -592,7 +560,7 @@ describe("Langchain SDK Tests", () => {
             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
+            image_size: '1K', // 仅适用于 gemini-3-pro-image-preview
             number_of_images: 1,
             response_modalities: ['IMAGE'], // 仅返回图片
         });
@@ -613,41 +581,35 @@ describe("Langchain SDK Tests", () => {
             console.log("图像描述:", result.text);
         }
     }, 120000);
-    test("GeminiImageGenerationClient - 多图像生成", async () => {
-        const imageClient = new index_1.GeminiImageGenerationClient({});
+    test("GeminiImageGenerationClient - 使用 aihubmix provider", async () => {
+        const imageClient = new index_1.GeminiImageGenerationClient({
+            provider: "aihubmix",
+        });
         const result = await imageClient.generate({
-            prompt: "A futuristic city skyline",
-            model: "gemini-2.0-flash-exp-image-generation",
-            number_of_images: 2,
-            aspect_ratio: "16:9",
+            prompt: 'A beautiful sunset over the ocean',
+            model: 'gemini-3-pro-image-preview',
+            aspect_ratio: '16:9',
+            image_size: '1K',
+            number_of_images: 1,
+            response_modalities: ['IMAGE'], // 仅返回图片
         });
         expect(result).toBeDefined();
+        expect(result.created).toBeDefined();
+        expect(typeof result.created).toBe("number");
         expect(result.data).toBeDefined();
         expect(Array.isArray(result.data)).toBe(true);
-        expect(result.data.length).toBe(2);
-        console.log("✅ GeminiImageGenerationClient 多图像生成测试成功");
+        expect(result.data.length).toBeGreaterThan(0);
+        result.data.forEach((item) => {
+            expect(item).toBeDefined();
+            expect(item.url || item.b64_json).toBeDefined();
+        });
+        console.log("✅ GeminiImageGenerationClient (aihubmix provider) 测试成功");
         console.log(`生成图像数量: ${result.data.length}`);
-    }, 120000);
-    test("GeminiImageGenerationClient - 不同宽高比测试", async () => {
-        const imageClient = new index_1.GeminiImageGenerationClient({});
-        const aspectRatios = [
-            "1:1",
-            "16:9",
-            "9:16",
-        ];
-        for (const aspectRatio of aspectRatios) {
-            const result = await imageClient.generate({
-                prompt: "A beautiful landscape",
-                model: "gemini-2.0-flash-exp-image-generation",
-                aspect_ratio: aspectRatio,
-                number_of_images: 1,
-            });
-            expect(result).toBeDefined();
-            expect(result.data).toBeDefined();
-            expect(result.data.length).toBeGreaterThan(0);
-            console.log(`✅ Gemini 宽高比 ${aspectRatio} 测试成功`);
+        console.log("图像 URL:", result.data[0]?.url || ("Base64 编码" + result.data[0]?.b64_json));
+        if (result.text) {
+            console.log("图像描述:", result.text);
         }
-    }, 180000);
+    }, 120000);
     test("DoubaoImageGenerationClient - quality 参数测试", async () => {
         const imageClient = new index_1.DoubaoImageGenerationClient({});
         const qualities = ["standard", "hd"];
@@ -662,7 +624,7 @@ describe("Langchain SDK Tests", () => {
             expect(result).toBeDefined();
             expect(result.data).toBeDefined();
             expect(result.data.length).toBeGreaterThan(0);
-            console.log(`✅ quality ${quality} 测试成功`);
+            console.log(`✅ quality ${quality} 测试成功`, result.data);
         }
     }, 180000);
     test("DoubaoImageGenerationClient - style 参数测试", async () => {
@@ -679,7 +641,7 @@ describe("Langchain SDK Tests", () => {
             expect(result).toBeDefined();
             expect(result.data).toBeDefined();
             expect(result.data.length).toBeGreaterThan(0);
-            console.log(`✅ style ${style} 测试成功`);
+            console.log(`✅ style ${style} 测试成功`, result.data);
         }
     }, 180000);
     test("DoubaoImageGenerationClient - response_format 参数测试", async () => {
@@ -719,7 +681,7 @@ describe("Langchain SDK Tests", () => {
         expect(result).toBeDefined();
         expect(result.data).toBeDefined();
         expect(result.data.length).toBeGreaterThan(0);
-        console.log("✅ watermark 参数测试成功");
+        console.log("✅ watermark 参数测试成功", result.data);
     }, 120000);
     test("sdkConfig - 全局配置测试", () => {
         const originalBaseUrl = index_1.sdkConfig.getServerUrl();

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

@@ -3,6 +3,7 @@
  * Google Gemini 图像生成客户端
  */
 export interface GeminiImageGenerationConfig {
+    provider?: "aihubmix" | "gemini";
     baseUrl?: string;
     headers?: Record<string, string>;
 }
@@ -29,6 +30,7 @@ export interface GeminiImageGenerationResponse {
 }
 export declare class GeminiImageGenerationClient {
     private headers;
+    private provider;
     constructor(config?: GeminiImageGenerationConfig);
     /**
      * Generate images

package/dist/gemini-image-generation.js

@@ -9,15 +9,21 @@ const config_1 = require("./config");
 const log_1 = require("./log");
 class GeminiImageGenerationClient {
     constructor(config) {
+        // 确定 provider（默认为 gemini）
+        this.provider = config?.provider || "gemini";
         // 合并全局 headers 和配置 headers
         const globalHeaders = config_1.sdkConfig.getHeaders();
         this.headers = {
             "Content-Type": "application/json",
             "Authorization": `Bearer ${config_1.sdkConfig.getToken()}`,
-            "X-Base-Url": config?.baseUrl || "",
+            "X-Provider": this.provider, // 在 header 中传递 provider，由后端确定 baseUrl
             ...globalHeaders,
             ...config?.headers,
         };
+        // 如果明确指定了 baseUrl，也添加到 header 中
+        if (config?.baseUrl) {
+            this.headers["X-Base-Url"] = config.baseUrl;
+        }
     }
     /**
      * Generate images

package/dist/index.d.ts

@@ -10,7 +10,7 @@ import { GeminiImageGenerationClient, type GeminiImageGenerationConfig, type Gem
 import { VideoGenerationClient, type VideoGenerationConfig, type VideoGenerationRequest, type ContentGenerationTaskID, type ContentGenerationTask } from "./video_generation";
 import { sdkConfig } from "./config";
 export { BaseMessage, HumanMessage, AIMessage, SystemMessage, AIMessageChunk, type MessageContent, type AIMessageChunkData, } from "./messages";
-export { BaseChatModel, type BaseChatModelParams, type ToolDefinition, type BindOptions, } from "./base";
+export { BaseChatModel, type BaseChatModelParams, type AIModelProvider, type ToolDefinition, type BindOptions, } from "./base";
 export { ChatOpenAI } from "./chat_models/openai";
 export { ChatGoogleGenerativeAI } from "./chat_models/google";
 export { ChatAnthropic } from "./chat_models/anthropic";

package/package.json

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