ai-world-sdk 1.0.13 → 1.0.16

package/README.md

@@ -46,21 +46,21 @@ import { ChatGoogleGenerativeAI, ChatOpenAI, ChatAnthropic, HumanMessage, System
 const geminiModel = new ChatGoogleGenerativeAI({
   modelName: 'gemini-2.5-flash-image',
   temperature: 0.7,
-  provider: 'gemini', // 或 'aihubmix'
+  provider: 'gemini', // 或 'aihubmix', 'api2img'
 });
 
 // GPT 模型（使用 aihubmix provider）
 const gptModel = new ChatOpenAI({
   modelName: 'gpt-4o-mini',
   temperature: 0.7,
-  provider: 'aihubmix',
+  provider: 'aihubmix', // 或 'api2img'
 });
 
 // Claude 模型（使用 aihubmix provider）
 const claudeModel = new ChatAnthropic({
   modelName: 'claude-3-sonnet-20240229',
   temperature: 0.7,
-  provider: 'aihubmix',
+  provider: 'aihubmix', // 或 'api2img'
 });
 
 // 非流式调用
@@ -208,7 +208,7 @@ import { createChatModel } from 'ai-world-sdk';
 // 根据模型名称自动选择正确的类
 const model = createChatModel('gemini-2.5-flash-image', {
   temperature: 0.7,
-  provider: 'gemini', // 或 'aihubmix', 'doubao'
+  provider: 'gemini', // 或 'aihubmix', 'api2img', 'doubao'
 });
 ```
 
@@ -220,6 +220,7 @@ const model = createChatModel('gemini-2.5-flash-image', {
 
 **Provider 参数说明:**
 - `provider: 'aihubmix'` - 使用 aihubmix 聚合服务（推荐，支持所有模型）
+- `provider: 'api2img'` - 使用 api2img 聚合服务（推荐，支持所有模型）
 - `provider: 'gemini'` - 直接使用 Google Gemini API
 - `provider: 'doubao'` - 使用豆包服务
 
@@ -330,6 +331,7 @@ const secondResponse = await client.chat({
 
 - `provider: 'gemini'`（默认）- 直接使用 Google Gemini 官方 API
 - `provider: 'aihubmix'` - 通过 aihubmix 代理服务调用，适合需要统一管理多个 API 的场景
+- `provider: 'api2img'` - 通过 api2img 代理服务调用，适合需要统一管理多个 API 的场景
 
 **模型对比：**
 
@@ -661,6 +663,7 @@ const response = await doubaoModel.invoke([
 |----------|----------|------------|
 | `aihubmix` | 多模型聚合，统一接口 | GPT、Gemini、Claude、Doubao 等 |
 | `gemini` | 直接使用 Google Gemini API | 所有 Gemini 模型 |
+| `api2img` | 使用 api2img 聚合服务 | 所有模型 |
 | `doubao` | 使用豆包服务 | Doubao 系列模型 |
 
 **推荐使用 `aihubmix` provider**，因为它提供了统一的接口和更好的模型选择灵活性。
@@ -1025,7 +1028,7 @@ for await (const chunk of client.streamDownload({
 | Doubao | `doubao` 或 `aihubmix` | `doubao-pro-4k`, `doubao-seedream-4-5-251128` | `ChatOpenAI` |
 
 **注意:** 
-- 使用 `aihubmix` provider 可以访问所有模型，推荐用于多模型场景
+- 使用 `aihubmix` 或 `api2img` provider 可以访问所有模型，推荐用于多模型场景
 - 使用特定 provider（如 `gemini`、`doubao`）会直接调用对应的官方 API
 
 ### 图像生成模型

package/dist/__tests__/example.test.js

@@ -670,6 +670,68 @@ describe("Langchain SDK Tests", () => {
             console.log("图像描述:", result.text);
         }
     }, 120000);
+    test("ChatGoogleGenerativeAI - api2img.com no stream", async () => {
+        const gemini = new index_1.ChatGoogleGenerativeAI({
+            modelName: "gemini-2.5-flash-image",
+            temperature: 0.7,
+            provider: "api2img",
+        });
+        const response = await gemini.invoke([
+            new index_1.HumanMessage("api2img.com是做什么的"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        expect(typeof response.content).toBe("string");
+        console.log("✅ api2img.com no stream 测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("ChatGoogleGenerativeAI - api2img.com stream", async () => {
+        const gemini = new index_1.ChatGoogleGenerativeAI({
+            modelName: "gemini-2.5-flash-image",
+            temperature: 0.7,
+            provider: "api2img",
+        });
+        let fullText = "";
+        for await (const chunk of gemini.stream([
+            new index_1.HumanMessage("api2img.com是做什么的"),
+        ])) {
+            fullText += extractTextFromChunk(chunk);
+            console.log("AI stream chunk:", chunk);
+            console.log("AI stream text:", fullText);
+        }
+        expect(fullText.length).toBeGreaterThan(0);
+        console.log("✅ api2img.com stream 测试成功");
+        console.log("AI:", fullText);
+        console.log(`完整回复长度: ${fullText.length} 字符`);
+    }, 30000);
+    test("GeminiImageGenerationClient - 使用 api2img provider", async () => {
+        const imageClient = new index_1.GeminiImageGenerationClient({
+            provider: "api2img",
+        });
+        const result = await imageClient.generate({
+            prompt: 'A beautiful sunset over the ocean',
+            model: 'gemini-3-pro-image-preview',
+            aspect_ratio: '16:9',
+            image_size: '1K',
+            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).toBeGreaterThan(0);
+        result.data.forEach((item) => {
+            expect(item).toBeDefined();
+            expect(item.url || item.b64_json).toBeDefined();
+        });
+        console.log("✅ GeminiImageGenerationClient (api2img provider) 测试成功");
+        console.log(`生成图像数量: ${result.data.length}`);
+        console.log("图像 URL:", result.data[0]?.url || ("Base64 编码" + result.data[0]?.b64_json));
+        if (result.text) {
+            console.log("图像描述:", result.text);
+        }
+    }, 120000);
     test("GeminiImageGenerationClient - 多轮图片修改", async () => {
         const imageClient = new index_1.GeminiImageGenerationClient({});
         // 第一轮：创建初始图片

package/dist/base.d.ts

@@ -3,7 +3,7 @@
  * Similar to LangChain.js BaseChatModel
  */
 import { BaseMessage, AIMessage, AIMessageChunk } from "./messages";
-export type AIModelProvider = "aihubmix" | "doubao" | "gemini";
+export type AIModelProvider = "aihubmix" | "api2img" | "doubao" | "gemini";
 export interface BaseChatModelParams {
     provider: AIModelProvider;
     baseUrl?: string;

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

@@ -2,8 +2,9 @@
  * Gemini Image Generation Client
  * Google Gemini 图像生成客户端
  */
+export type GeminiImageGenerationProvider = "aihubmix" | "api2img" | "gemini";
 export interface GeminiImageGenerationConfig {
-    provider?: "aihubmix" | "gemini";
+    provider?: GeminiImageGenerationProvider;
     baseUrl?: string;
     headers?: Record<string, string>;
 }
@@ -32,7 +33,7 @@ export interface GeminiImageChatRequest {
     message: string;
     chat_id?: string;
     model?: string;
-    provider?: "aihubmix" | "gemini";
+    provider?: GeminiImageGenerationProvider;
     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";
     response_modalities?: ("TEXT" | "IMAGE")[];

package/dist/index.js

@@ -52,6 +52,7 @@ function createChatModel(model, config) {
     switch (config.provider) {
         case "doubao":
         case "aihubmix":
+        case "api2img":
             return new openai_1.ChatOpenAI({
                 modelName: model,
                 ...finalConfig,

package/package.json

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