ai-world-sdk 1.1.2 → 1.1.5

package/README.md

@@ -1489,7 +1489,7 @@ for await (const chunk of client.streamDownload({
 | 提供商 | Provider | 模型示例 | 模型类 |
 |--------|----------|----------|--------|
 | OpenAI | `aihubmix` | `gpt-4o-mini`, `gpt-4`, `o1-preview` | `ChatOpenAI` |
-| Google Gemini | `gemini` 或 `aihubmix` | `gemini-2.5-flash-image`, `gemini-1.5-pro` | `ChatGoogleGenerativeAI` |
+| Google Gemini | `gemini` 或 `aihubmix` | `gemini-2.5-flash-image`, `gemini-3-pro-image-preview` | `ChatGoogleGenerativeAI` |
 | Anthropic Claude | `aihubmix` | `claude-3-sonnet-20240229`, `claude-3-opus-20240229` | `ChatAnthropic` |
 | Doubao | `doubao` 或 `aihubmix` | `doubao-pro-4k`, `doubao-seedream-4-5-251128` | `ChatOpenAI` |
 
@@ -1505,7 +1505,6 @@ for await (const chunk of client.streamDownload({
 - **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` - 速度快，成本低

package/dist/__tests__/example.test.js

@@ -882,7 +882,7 @@ describe("Langchain SDK Tests", () => {
         const openai = new index_1.ChatOpenAI({
             modelName: "gpt-4",
             temperature: 0.7,
-            provider: "aihubmix",
+            provider: "api2img",
         });
         let fullText = "";
         for await (const chunk of openai.stream([
@@ -891,14 +891,14 @@ describe("Langchain SDK Tests", () => {
             fullText += extractTextFromChunk(chunk);
         }
         expect(fullText.length).toBeGreaterThan(0);
-        console.log("✅ ChatOpenAI 流式测试成功");
+        console.log("✅ ChatOpenAI 流式测试成功 " + fullText);
         console.log(`完整回复长度: ${fullText.length} 字符`);
     }, 30000);
     test("ChatOpenAI - batch 方法", async () => {
         const openai = new index_1.ChatOpenAI({
             modelName: "gpt-4",
             temperature: 0.7,
-            provider: "aihubmix",
+            provider: "api2img",
         });
         const responses = await openai.batch([
             [new index_1.HumanMessage("什么是人工智能？")],

package/dist/base.d.ts

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

package/dist/base.js

@@ -130,21 +130,72 @@ class BaseChatModel {
         }
         const reader = response.body.getReader();
         const decoder = new TextDecoder();
+        let buffer = "";
         try {
             while (true) {
                 const { done, value } = await reader.read();
                 if (done)
                     break;
-                let data = decoder.decode(value, { stream: true });
-                try {
-                    const parsed = JSON.parse(data);
-                    if (config_1.sdkConfig.getDebug()) {
-                        (0, log_1.debugLog)("📦 Stream Chunk:", parsed);
+                buffer += decoder.decode(value, { stream: true });
+                // 按行分割处理 SSE 格式数据
+                const lines = buffer.split("\n");
+                // 保留最后一个可能不完整的行
+                buffer = lines.pop() || "";
+                for (const line of lines) {
+                    const trimmedLine = line.trim();
+                    // 跳过空行
+                    if (!trimmedLine)
+                        continue;
+                    // 检查结束标志
+                    if (trimmedLine === "data: [DONE]") {
+                        return;
+                    }
+                    // 处理 SSE 格式 "data: {...}"
+                    if (trimmedLine.startsWith("data: ")) {
+                        const jsonStr = trimmedLine.slice(6); // 去掉 "data: " 前缀
+                        try {
+                            const parsed = JSON.parse(jsonStr);
+                            if (config_1.sdkConfig.getDebug()) {
+                                (0, log_1.debugLog)("📦 Stream Chunk:", parsed);
+                            }
+                            yield new messages_1.AIMessageChunk(parsed);
+                        }
+                        catch (e) {
+                            if (config_1.sdkConfig.getDebug()) {
+                                (0, log_1.debugLog)("⚠️ Failed to parse SSE data:", jsonStr, e);
+                            }
+                        }
+                    }
+                    else {
+                        // 尝试直接解析（兼容非 SSE 格式）
+                        try {
+                            const parsed = JSON.parse(trimmedLine);
+                            if (config_1.sdkConfig.getDebug()) {
+                                (0, log_1.debugLog)("📦 Stream Chunk (non-SSE):", parsed);
+                            }
+                            yield new messages_1.AIMessageChunk(parsed);
+                        }
+                        catch {
+                            // 忽略无法解析的行
+                        }
                     }
-                    yield new messages_1.AIMessageChunk(parsed);
                 }
-                catch {
-                    // 忽略解析错误
+            }
+            // 处理缓冲区中剩余的数据
+            if (buffer.trim()) {
+                const trimmedBuffer = buffer.trim();
+                if (trimmedBuffer.startsWith("data: ") && trimmedBuffer !== "data: [DONE]") {
+                    const jsonStr = trimmedBuffer.slice(6);
+                    try {
+                        const parsed = JSON.parse(jsonStr);
+                        if (config_1.sdkConfig.getDebug()) {
+                            (0, log_1.debugLog)("📦 Stream Chunk (final):", parsed);
+                        }
+                        yield new messages_1.AIMessageChunk(parsed);
+                    }
+                    catch {
+                        // 忽略解析错误
+                    }
                 }
             }
         }

package/dist/config.d.ts

@@ -9,7 +9,7 @@
  *
  * 注意: {VERSION} 占位符会在构建时被替换为实际版本号
  */
-export declare const SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.2";
+export declare const SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.5";
 /**
  * 版本兼容性错误
  */
@@ -24,8 +24,8 @@ declare class SDKConfig {
     private _pluginId;
     private _versionCompatible;
     private _versionCheckPromise;
-    readonly sdkSignature = "AI_WORLD_SDK_V:1.1.2";
-    readonly sdkVersion = "1.1.2";
+    readonly sdkSignature = "AI_WORLD_SDK_V:1.1.5";
+    readonly sdkVersion = "1.1.5";
     constructor();
     /**
      * Set global base URL

package/dist/config.js

@@ -7,7 +7,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.sdkConfig = exports.VersionCompatibilityError = exports.SDK_SIGNATURE = void 0;
 // SDK 版本号（构建时自动从 package.json 更新）
 // 此版本号会在运行 npm run build 时自动从 package.json 读取并更新
-const SDK_VERSION = "1.1.2";
+const SDK_VERSION = "1.1.5";
 /**
  * SDK 特征码 - 用于在构建后的 JS 文件中识别 SDK 版本
  * 格式: AI_WORLD_SDK_V:版本号
@@ -15,7 +15,7 @@ const SDK_VERSION = "1.1.2";
  *
  * 注意: {VERSION} 占位符会在构建时被替换为实际版本号
  */
-exports.SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.2";
+exports.SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.5";
 /**
  * 版本兼容性错误
  */

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

@@ -2,7 +2,8 @@
  * Gemini Image Generation Client
  * Google Gemini 图像生成客户端
  */
-export type GeminiImageGenerationProvider = "aihubmix" | "api2img" | "gemini";
+import type { AIModelProvider } from "./provider_config";
+export type GeminiImageGenerationProvider = Extract<AIModelProvider, "aihubmix" | "shubiaobiao" | "api2img" | "gemini">;
 export interface GeminiImageGenerationConfig {
     provider?: GeminiImageGenerationProvider;
     baseUrl?: string;

package/dist/index.d.ts

@@ -12,7 +12,8 @@ import { OpenAIVideoGenerationClient, type OpenAIVideoGenerationConfig, type Ope
 import { DownloadClient, type DownloadConfig, type DownloadOptions, type StreamDownloadOptions } from "./download";
 import { AuthClient, getCurrentUserInfo, type AuthConfig, type UserInfo } from "./auth";
 export { BaseMessage, HumanMessage, AIMessage, SystemMessage, AIMessageChunk, type MessageContent, type AIMessageChunkData, } from "./messages";
-export { BaseChatModel, type BaseChatModelParams, type AIModelProvider, type ToolDefinition, type BindOptions, } from "./base";
+export { BaseChatModel, type BaseChatModelParams, type ToolDefinition, type BindOptions, } from "./base";
+export type { AIModelProvider } from "./provider_config";
 export { ChatOpenAI } from "./chat_models/openai";
 export { ChatGoogleGenerativeAI } from "./chat_models/google";
 export { ChatAnthropic } from "./chat_models/anthropic";

package/dist/index.js

@@ -24,6 +24,7 @@ const auth_1 = require("./auth");
 Object.defineProperty(exports, "AuthClient", { enumerable: true, get: function () { return auth_1.AuthClient; } });
 Object.defineProperty(exports, "getCurrentUserInfo", { enumerable: true, get: function () { return auth_1.getCurrentUserInfo; } });
 const config_1 = require("./config");
+const provider_config_1 = require("./provider_config");
 // Re-export types and classes
 var messages_1 = require("./messages");
 Object.defineProperty(exports, "HumanMessage", { enumerable: true, get: function () { return messages_1.HumanMessage; } });
@@ -58,20 +59,18 @@ function createChatModel(model, config) {
             ...config.headers,
         },
     };
-    switch (config.provider) {
-        case "doubao":
-        case "aihubmix":
-        case "api2img":
-            return new openai_1.ChatOpenAI({
-                modelName: model,
-                ...finalConfig,
-            });
-        case "gemini":
-            return new google_1.ChatGoogleGenerativeAI({
-                modelName: model,
-                ...finalConfig,
-            });
-        default:
-            throw new Error(`Unsupported provider: ${config.provider}`);
+    const providerConfig = (0, provider_config_1.getProviderConfig)(config.provider);
+    if (!providerConfig) {
+        throw new Error(`Unsupported provider: ${config.provider}`);
     }
+    if (providerConfig.modelType === "gemini") {
+        return new google_1.ChatGoogleGenerativeAI({
+            modelName: model,
+            ...finalConfig,
+        });
+    }
+    return new openai_1.ChatOpenAI({
+        modelName: model,
+        ...finalConfig,
+    });
 }

package/dist/openai_video_generation.d.ts

@@ -10,7 +10,7 @@ export interface OpenAIVideoGenerationConfig {
 export interface OpenAIVideoGenerationRequest {
     prompt: string;
     model?: string;
-    provider?: "aihubmix" | "api2img";
+    provider?: "aihubmix" | "shubiaobiao" | "api2img";
     seconds?: "4" | "8" | "12";
     size?: "720x1280" | "1280x720" | "1024x1792" | "1792x1024";
     input_reference?: string;
@@ -107,7 +107,7 @@ export declare class OpenAIVideoGenerationClient {
      * }
      * ```
      */
-    getTask(taskId: string, provider: "aihubmix" | "api2img"): Promise<OpenAIVideoTaskResponse>;
+    getTask(taskId: string, provider: "aihubmix" | "shubiaobiao" | "api2img"): Promise<OpenAIVideoTaskResponse>;
     /**
      * 轮询等待任务完成
      * Poll and wait for task completion
@@ -125,7 +125,7 @@ export declare class OpenAIVideoGenerationClient {
      * console.log('Video URL:', result.url);
      * ```
      */
-    waitForCompletion(taskId: string, provider?: "aihubmix" | "api2img", options?: {
+    waitForCompletion(taskId: string, provider?: "aihubmix" | "shubiaobiao" | "api2img", options?: {
         maxAttempts?: number;
         interval?: number;
         onProgress?: (status: OpenAIVideoTaskResponse) => void;
@@ -184,5 +184,5 @@ export declare class OpenAIVideoGenerationClient {
      * const thumbnail = await client.downloadVideo('vid_abc123', 'aihubmix', 'thumbnail');
      * ```
      */
-    downloadVideo(taskId: string, provider?: "aihubmix" | "api2img", variant?: "video" | "thumbnail"): Promise<Blob>;
+    downloadVideo(taskId: string, provider?: "aihubmix" | "shubiaobiao" | "api2img", variant?: "video" | "thumbnail"): Promise<Blob>;
 }

package/dist/provider_config.d.ts

@@ -0,0 +1,8 @@
+export type AIModelProvider = "aihubmix" | "api2img" | "doubao" | "gemini" | "shubiaobiao";
+export type ProviderModelType = "openai" | "gemini";
+export interface ProviderConfig {
+    provider: AIModelProvider;
+    modelType: ProviderModelType;
+}
+export declare const PROVIDER_CONFIGS: Record<AIModelProvider, ProviderConfig>;
+export declare function getProviderConfig(provider: AIModelProvider): ProviderConfig;

package/dist/provider_config.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PROVIDER_CONFIGS = void 0;
+exports.getProviderConfig = getProviderConfig;
+exports.PROVIDER_CONFIGS = {
+    aihubmix: { provider: "aihubmix", modelType: "openai" },
+    shubiaobiao: { provider: "shubiaobiao", modelType: "openai" },
+    api2img: { provider: "api2img", modelType: "openai" },
+    doubao: { provider: "doubao", modelType: "openai" },
+    gemini: { provider: "gemini", modelType: "gemini" },
+};
+function getProviderConfig(provider) {
+    return exports.PROVIDER_CONFIGS[provider];
+}

package/package.json

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