ai-world-sdk 1.0.13 → 1.0.15

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.15",
   "description": "TypeScript SDK for AI World Platform - Chat Models, Image Generation, and Video Generation",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",