ai-world-sdk 1.1.1 → 1.1.3

package/dist/__tests__/example.test.js

@@ -44,7 +44,7 @@ const fs_1 = require("fs");
 dotenv.config();
 index_1.sdkConfig.setBaseUrl("http://localhost:8000");
 index_1.sdkConfig.setToken(process.env.AUTH_TOKEN || process.env.TOKEN || "");
-// sdkConfig.setDebug(true);
+index_1.sdkConfig.setDebug(true);
 function extractTextFromChunk(chunk) {
     if (typeof chunk.content === "string") {
         return chunk.content;
@@ -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("什么是人工智能？")],
@@ -1148,4 +1148,397 @@ describe("Langchain SDK Tests", () => {
             console.log("响应内容:", content);
         }
     }, 30000);
+    // ========== OpenAI 视频生成测试 ==========
+    test("OpenAIVideoGenerationClient - 创建视频生成任务（文本）", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        const task = await openaiVideoClient.generate({
+            prompt: "A beautiful sunset over the ocean with waves crashing on the shore",
+            model: "sora-2",
+            provider: "aihubmix",
+            seconds: "4",
+            size: "1280x720",
+        });
+        expect(task).toBeDefined();
+        expect(task.id).toBeDefined();
+        expect(typeof task.id).toBe("string");
+        expect(task.status).toBeDefined();
+        expect(["queued", "in_progress", "completed", "failed"]).toContain(task.status);
+        console.log("✅ OpenAI 视频生成任务创建成功（文本）");
+        console.log("任务 ID:", task.id);
+        console.log("任务状态:", task.status);
+    }, 60000);
+    test("OpenAIVideoGenerationClient - 创建视频生成任务（图像）", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        // 使用一个小的 base64 图像
+        const base64Image = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAgAAAAJCAIAAACAMfp5AAAACXBIWXMAAAsTAAALEwEAmpwYAAAFgmlUWHRYTUw6Y29tLmFkb2JlLnhtcAAAAAAAPD94cGFja2V0IGJlZ2luPSLvu78iIGlkPSJXNU0wTXBDZWhpSHpyZVN6TlRjemtjOWQiPz4gPHg6eG1wbWV0YSB4bWxuczp4PSJhZG9iZTpuczptZXRhLyIgeDp4bXB0az0iQWRvYmUgWE1QIENvcmUgOS4xLWMwMDIgNzkuZGJhM2RhM2I1LCAyMDIzLzEyLzE1LTEwOjQyOjM3ICAgICAgICAiPiA8cmRmOlJERiB4bWxuczpyZGY9Imh0dHA6Ly93d3cudzMub3JnLzE5OTkvMDIvMjItcmRmLXN5bnRheC1ucyMiPiA8cmRmOkRlc2NyaXB0aW9uIHJkZjphYm91dD0iIiB4bWxuczp0aWZmPSJodHRwOi8vbnMuYWRvYmUuY29tL3RpZmYvMS4wLyIgeG1sbnM6ZXhpZj0iaHR0cDovL25zLmFkb2JlLmNvbS9leGlmLzEuMC8iIHhtbG5zOnhtcD0iaHR0cDovL25zLmFkb2JlLmNvbS94YXAvMS4wLyIgeG1sbnM6ZGM9Imh0dHA6Ly9wdXJsLm9yZy9kYy9lbGVtZW50cy8xLjEvIiB4bWxuczpwaG90b3Nob3A9Imh0dHA6Ly9ucy5hZG9iZS5jb20vcGhvdG9zaG9wLzEuMC8iIHhtbG5zOnhtcE1NPSJodHRwOi8vbnMuYWRvYmUuY29tL3hhcC8xLjAvbW0vIiB4bWxuczpzdEV2dD0iaHR0cDovL25zLmFkb2JlLmNvbS94YXAvMS4wL3NUeXBlL1Jlc291cmNlRXZlbnQjIiB0aWZmOk9yaWVudGF0aW9uPSIxIiBleGlmOkNvbG9yU3BhY2U9IjEiIGV4aWY6UGl4ZWxYRGltZW5zaW9uPSI4IiBleGlmOlBpeGVsWURpbWVuc2lvbj0iOSIgeG1wOkNyZWF0ZURhdGU9IjIwMjUtMTItMTlUMTI6MDY6MDMrMDg6MDAiIHhtcDpNb2RpZnlEYXRlPSIyMDI1LTEyLTE5VDEyOjA2OjM1KzA4OjAwIiB4bXA6TWV0YWRhdGFEYXRlPSIyMDI1LTEyLTE5VDEyOjA2OjM1KzA4OjAwIiBkYzpmb3JtYXQ9ImltYWdlL3BuZyIgcGhvdG9zaG9wOkNvbG9yTW9kZT0iMyIgeG1wTU06SW5zdGFuY2VJRD0ieG1wLmlpZDo5Y2IwNmRjYi04MmI0LTQ3ODctYmQ1Ny1lYWI5MzE5MzkyYzUiIHhtcE1NOkRvY3VtZW50SUQ9InhtcC5kaWQ6OWNiMDZkY2ItODJiNC00Nzg3LWJkNTctZWFiOTMxOTM5MmM1IiB4bXBNTTpPcmlnaW5hbERvY3VtZW50SUQ9InhtcC5kaWQ6OWNiMDZkY2ItODJiNC00Nzg3LWJkNTctZWFiOTMxOTM5MmM1Ij4gPHhtcE1NOkhpc3Rvcnk+IDxyZGY6U2VxPiA8cmRmOmxpIHN0RXZ0OmFjdGlvbj0ic2F2ZWQiIHN0RXZ0Omluc3RhbmNlSUQ9InhtcC5paWQ6OWNiMDZkY2ItODJiNC00Nzg3LWJkNTctZWFiOTMxOTM5MmM1IiBzdEV2dDp3aGVuPSIyMDI1LTEyLTE5VDEyOjA2OjM1KzA4OjAwIiBzdEV2dDpzb2Z0d2FyZUFnZW50PSJBZG9iZSBQaG90b3Nob3AgMjUuNiAoTWFjaW50b3NoKSIgc3RFdnQ6Y2hhbmdlZD0iLyIvPiA8L3JkZjpTZXE+IDwveG1wTU06SGlzdG9yeT4gPC9yZGY6RGVzY3JpcHRpb24+IDwvcmRmOlJERj4gPC94OnhtcG1ldGE+IDw/eHBhY2tldCBlbmQ9InIiPz6KNh+8AAAAEElEQVQIHWNgYPiPAw0PCQA/Vke5jW7wRQAAAABJRU5ErkJggg==";
+        const task = await openaiVideoClient.generate({
+            prompt: "Animate this scene with gentle movements",
+            input_reference: base64Image,
+            model: "sora-2",
+            provider: "aihubmix",
+            seconds: "4",
+        });
+        console.log("task:", task);
+        expect(task).toBeDefined();
+        expect(task.id).toBeDefined();
+        expect(typeof task.id).toBe("string");
+        console.log("✅ OpenAI 视频生成任务创建成功（图像）");
+        console.log("任务 ID:", task.id);
+    }, 60000);
+    test("OpenAIVideoGenerationClient - 查询任务状态", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        // // 先创建一个任务
+        // const createTask = await openaiVideoClient.generate({
+        //   prompt: "A serene mountain landscape",
+        //   model: "sora-2",
+        //   provider: "aihubmix",
+        //   seconds: "4",
+        // });
+        // console.log("createTask:", createTask);
+        // expect(createTask.id).toBeDefined();
+        // 查询任务状态
+        let taskId = "eyJtb2RlbCI6InNvcmEtMiIsImlkIjoidmlkZW9fNjk0YmYxYzZjNTEwODE5MDliZjBlNDc5ZDI3OGU5ODYifQchannel2361";
+        // 轮询任务（使用较短的超时时间用于测试）
+        try {
+            const result = await openaiVideoClient.waitForCompletion(taskId, "aihubmix", {
+                maxAttempts: 30, // 最多轮询 30 次
+                interval: 5000, // 每 5 秒轮询一次
+                onProgress: (status) => {
+                    console.log(`进度更新: ${status.status}`);
+                }
+            });
+            console.log("--------------------------------");
+            console.log("result:", JSON.stringify(result, null, 2));
+            console.log("--------------------------------");
+            expect(result).toBeDefined();
+            expect(["completed", "failed"]).toContain(result.status);
+            console.log("✅ OpenAI 视频任务轮询完成");
+            console.log("最终状态:", result.status);
+            if (result.status === "completed") {
+                console.log("视频 ID:", result.id);
+            }
+            else if (result.error) {
+                console.log("错误信息:", result.error);
+            }
+        }
+        catch (error) {
+            // 如果超时，这是正常的（视频生成需要时间）
+            if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                console.log("⏭️  轮询超时（正常，视频生成需要较长时间）");
+            }
+            else {
+                throw error;
+            }
+        }
+    }, 60000);
+    test("OpenAIVideoGenerationClient - 轮询等待完成", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        // 创建任务
+        const createTask = await openaiVideoClient.generate({
+            prompt: "A short video of clouds moving across the sky",
+            model: "sora-2",
+            provider: "aihubmix",
+            seconds: "4",
+        });
+        expect(createTask.id).toBeDefined();
+        console.log("任务创建:", createTask.id);
+        // 轮询任务（使用较短的超时时间用于测试）
+        try {
+            const result = await openaiVideoClient.waitForCompletion(createTask.id, "aihubmix", {
+                maxAttempts: 30, // 最多轮询 30 次
+                interval: 5000, // 每 5 秒轮询一次
+                onProgress: (status) => {
+                    console.log(`进度更新: ${status.status}`);
+                }
+            });
+            expect(result).toBeDefined();
+            expect(["completed", "failed"]).toContain(result.status);
+            console.log("✅ OpenAI 视频任务轮询完成");
+            console.log("最终状态:", result.status);
+            if (result.status === "completed") {
+                console.log("视频 ID:", result.id);
+            }
+            else if (result.error) {
+                console.log("错误信息:", result.error);
+            }
+        }
+        catch (error) {
+            // 如果超时，这是正常的（视频生成需要时间）
+            if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                console.log("⏭️  轮询超时（正常，视频生成需要较长时间）");
+            }
+            else {
+                throw error;
+            }
+        }
+    }, 180000);
+    test("OpenAIVideoGenerationClient - 一键生成并等待", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        try {
+            const result = await openaiVideoClient.generateAndWait({
+                prompt: "A futuristic city skyline",
+                model: "sora-2",
+                provider: "aihubmix",
+                seconds: "4",
+                size: "1280x720"
+            }, {
+                maxAttempts: 30,
+                interval: 5000,
+                onProgress: (status) => {
+                    console.log("状态:", status.status);
+                }
+            });
+            expect(result).toBeDefined();
+            expect(result.id).toBeDefined();
+            expect(["completed", "failed"]).toContain(result.status);
+            console.log("✅ OpenAI 一键生成并等待测试完成");
+            console.log("最终状态:", result.status);
+        }
+        catch (error) {
+            if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                console.log("⏭️  轮询超时（正常）");
+            }
+            else {
+                throw error;
+            }
+        }
+    }, 180000);
+    test("OpenAIVideoGenerationClient - 不同 provider 测试", async () => {
+        const providers = ["api2img"];
+        for (const provider of providers) {
+            const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+            const task = await openaiVideoClient.generate({
+                prompt: "A simple animation",
+                model: "sora-2",
+                provider: provider,
+                seconds: "4",
+            });
+            expect(task).toBeDefined();
+            expect(task.id).toBeDefined();
+            console.log(`✅ provider ${provider} 测试成功，任务 ID: ${task.id}`);
+        }
+    }, 120000);
+    test("OpenAIVideoGenerationClient - 不同 seconds 参数测试", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        const secondsOptions = ["4", "8"];
+        for (const seconds of secondsOptions) {
+            const task = await openaiVideoClient.generate({
+                prompt: "A video clip",
+                model: "sora-2",
+                provider: "aihubmix",
+                seconds: seconds,
+            });
+            expect(task).toBeDefined();
+            expect(task.id).toBeDefined();
+            console.log(`✅ seconds ${seconds} 测试成功，任务 ID: ${task.id}`);
+        }
+    }, 120000);
+    test("OpenAIVideoGenerationClient - 不同 size 参数测试", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        const sizeOptions = [
+            "1280x720",
+            "720x1280",
+        ];
+        for (const size of sizeOptions) {
+            const task = await openaiVideoClient.generate({
+                prompt: "A landscape video",
+                model: "sora-2",
+                provider: "aihubmix",
+                seconds: "4",
+                size: size,
+            });
+            expect(task).toBeDefined();
+            expect(task.id).toBeDefined();
+            console.log(`✅ size ${size} 测试成功，任务 ID: ${task.id}`);
+        }
+    }, 120000);
+    test("OpenAIVideoGenerationClient - 下载视频测试", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        let taskId = "eyJtb2RlbCI6InNvcmEtMiIsImlkIjoidmlkZW9fNjk0YmYxYzZjNTEwODE5MDliZjBlNDc5ZDI3OGU5ODYifQchannel2361";
+        // 等待任务完成
+        try {
+            const result = await openaiVideoClient.waitForCompletion(taskId, "aihubmix", {
+                maxAttempts: 30,
+                interval: 5000,
+            });
+            console.log("result:", result);
+            if (result.status === "completed") {
+                // 下载视频
+                const videoBlob = await openaiVideoClient.downloadVideo(result.id, "aihubmix", "video");
+                expect(videoBlob).toBeDefined();
+                expect(videoBlob instanceof Blob).toBe(true);
+                expect(videoBlob.size).toBeGreaterThan(0);
+                console.log("✅ OpenAI 视频下载测试成功");
+                console.log(`视频大小: ${(videoBlob.size / 1024 / 1024).toFixed(2)} MB`);
+            }
+            else {
+                console.log("⏭️  任务未完成，跳过下载测试");
+            }
+        }
+        catch (error) {
+            if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                console.log("⏭️  轮询超时，跳过下载测试");
+            }
+            else {
+                throw error;
+            }
+        }
+    }, 180000);
+    test("OpenAIVideoGenerationClient - 下载缩略图测试", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        // 创建任务
+        const createTask = await openaiVideoClient.generate({
+            prompt: "A simple scene for thumbnail test",
+            model: "sora-2",
+            provider: "aihubmix",
+            seconds: "4",
+        });
+        expect(createTask.id).toBeDefined();
+        console.log("创建任务:", createTask.id);
+        // 等待任务完成
+        try {
+            const result = await openaiVideoClient.waitForCompletion(createTask.id, "aihubmix", {
+                maxAttempts: 30,
+                interval: 5000,
+            });
+            if (result.status === "completed") {
+                // 下载缩略图
+                const thumbnailBlob = await openaiVideoClient.downloadVideo(result.id, "aihubmix", "thumbnail");
+                expect(thumbnailBlob).toBeDefined();
+                expect(thumbnailBlob instanceof Blob).toBe(true);
+                expect(thumbnailBlob.size).toBeGreaterThan(0);
+                console.log("✅ OpenAI 缩略图下载测试成功");
+                console.log(`缩略图大小: ${(thumbnailBlob.size / 1024).toFixed(2)} KB`);
+            }
+            else {
+                console.log("⏭️  任务未完成，跳过缩略图下载测试");
+            }
+        }
+        catch (error) {
+            if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                console.log("⏭️  轮询超时，跳过缩略图下载测试");
+            }
+            else {
+                throw error;
+            }
+        }
+    }, 180000);
+    test("OpenAIVideoGenerationClient - 不同 provider 下载测试", async () => {
+        const providers = ["api2img"];
+        for (const provider of providers) {
+            const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+            // 创建任务
+            const createTask = await openaiVideoClient.generate({
+                prompt: "A test video for download",
+                model: "sora-2",
+                provider: provider,
+                seconds: "4",
+            });
+            expect(createTask.id).toBeDefined();
+            console.log(`${provider} 创建任务:`, createTask.id);
+            try {
+                const result = await openaiVideoClient.waitForCompletion(createTask.id, provider, {
+                    maxAttempts: 30,
+                    interval: 5000,
+                });
+                if (result.status === "completed") {
+                    // 下载视频
+                    const videoBlob = await openaiVideoClient.downloadVideo(result.id, provider, "video");
+                    expect(videoBlob).toBeDefined();
+                    expect(videoBlob instanceof Blob).toBe(true);
+                    expect(videoBlob.size).toBeGreaterThan(0);
+                    console.log(`✅ provider ${provider} 下载测试成功`);
+                    console.log(`视频大小: ${(videoBlob.size / 1024 / 1024).toFixed(2)} MB`);
+                }
+                else {
+                    console.log(`⏭️  ${provider} 任务未完成，跳过下载测试`);
+                }
+            }
+            catch (error) {
+                if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                    console.log(`⏭️  ${provider} 轮询超时，跳过下载测试`);
+                }
+                else {
+                    throw error;
+                }
+            }
+        }
+    }, 360000);
+    test("OpenAIVideoGenerationClient - 直接下载已完成的视频", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        // 使用一个已知完成的视频 ID（从前面测试中获取）
+        // 注意：这需要一个真实的已完成视频 ID，此处仅作示例
+        const knownVideoId = "eyJtb2RlbCI6InNvcmEtMiIsImlkIjoidmlkZW9fNjk0YmYxYzZjNTEwODE5MDliZjBlNDc5ZDI3OGU5ODYifQchannel2361";
+        try {
+            // 先查询状态
+            const status = await openaiVideoClient.getTask(knownVideoId, "aihubmix");
+            if (status.status === "completed") {
+                // 下载视频
+                const videoBlob = await openaiVideoClient.downloadVideo(knownVideoId, "aihubmix", "video");
+                expect(videoBlob).toBeDefined();
+                expect(videoBlob instanceof Blob).toBe(true);
+                expect(videoBlob.size).toBeGreaterThan(0);
+                console.log("✅ 直接下载已完成视频测试成功");
+                console.log(`视频大小: ${(videoBlob.size / 1024 / 1024).toFixed(2)} MB`);
+            }
+            else {
+                console.log(`⏭️  视频状态: ${status.status}，跳过下载测试`);
+            }
+        }
+        catch (error) {
+            // 视频可能已过期或不存在
+            console.log("⏭️  视频不存在或已过期，跳过测试");
+        }
+    }, 60000);
+    test("OpenAIVideoGenerationClient - 完整流程：生成、等待、下载", async () => {
+        const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});
+        try {
+            console.log("1️⃣ 开始生成视频...");
+            const task = await openaiVideoClient.generate({
+                prompt: "A complete workflow test video",
+                model: "sora-2",
+                provider: "aihubmix",
+                seconds: "4",
+            });
+            expect(task.id).toBeDefined();
+            console.log(`任务创建成功: ${task.id}`);
+            console.log("2️⃣ 等待视频生成完成...");
+            const result = await openaiVideoClient.waitForCompletion(task.id, "aihubmix", {
+                maxAttempts: 30,
+                interval: 5000,
+                onProgress: (status) => {
+                    console.log(`   进度: ${status.status}`);
+                },
+            });
+            expect(result.status).toBeDefined();
+            console.log(`视频生成完成: ${result.status}`);
+            if (result.status === "completed") {
+                console.log("3️⃣ 下载视频...");
+                const videoBlob = await openaiVideoClient.downloadVideo(result.id, "aihubmix", "video");
+                expect(videoBlob).toBeDefined();
+                expect(videoBlob instanceof Blob).toBe(true);
+                expect(videoBlob.size).toBeGreaterThan(0);
+                console.log("✅ 完整流程测试成功");
+                console.log(`   - 任务 ID: ${result.id}`);
+                console.log(`   - 视频大小: ${(videoBlob.size / 1024 / 1024).toFixed(2)} MB`);
+                // 也下载缩略图
+                console.log("4️⃣ 下载缩略图...");
+                const thumbnailBlob = await openaiVideoClient.downloadVideo(result.id, "aihubmix", "thumbnail");
+                expect(thumbnailBlob).toBeDefined();
+                expect(thumbnailBlob instanceof Blob).toBe(true);
+                console.log(`   - 缩略图大小: ${(thumbnailBlob.size / 1024).toFixed(2)} KB`);
+            }
+            else {
+                console.log("⏭️  视频生成失败或未完成");
+            }
+        }
+        catch (error) {
+            if (error.message?.includes("已轮询") || error.message?.includes("仍未完成")) {
+                console.log("⏭️  轮询超时（正常，视频生成需要较长时间）");
+            }
+            else {
+                throw error;
+            }
+        }
+    }, 240000);
 });

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,73 @@ 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 });
+                console.log('buffer', buffer);
+                // 按行分割处理 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.1";
+export declare const SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.3";
 /**
  * 版本兼容性错误
  */
@@ -24,8 +24,8 @@ declare class SDKConfig {
     private _pluginId;
     private _versionCompatible;
     private _versionCheckPromise;
-    readonly sdkSignature = "AI_WORLD_SDK_V:1.1.1";
-    readonly sdkVersion = "1.1.1";
+    readonly sdkSignature = "AI_WORLD_SDK_V:1.1.3";
+    readonly sdkVersion = "1.1.3";
     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.1";
+const SDK_VERSION = "1.1.3";
 /**
  * SDK 特征码 - 用于在构建后的 JS 文件中识别 SDK 版本
  * 格式: AI_WORLD_SDK_V:版本号
@@ -15,7 +15,7 @@ const SDK_VERSION = "1.1.1";
  *
  * 注意: {VERSION} 占位符会在构建时被替换为实际版本号
  */
-exports.SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.1";
+exports.SDK_SIGNATURE = "AI_WORLD_SDK_V:1.1.3";
 /**
  * 版本兼容性错误
  */

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

@@ -8,10 +8,12 @@ import { BaseChatModel, BaseChatModelParams } from "./base";
 import { DoubaoImageGenerationClient, DoubaoImageSize, type DoubaoImageGenerationConfig, type DoubaoImageGenerationRequest, type DoubaoImageGenerationResponse } from "./doubao-image-generation";
 import { GeminiImageGenerationClient, type GeminiImageGenerationConfig, type GeminiImageGenerationRequest, type GeminiImageGenerationResponse } from "./gemini-image-generation";
 import { VideoGenerationClient, type VideoGenerationConfig, type VideoGenerationRequest, type ContentGenerationTaskID, type ContentGenerationTask } from "./video_generation";
+import { OpenAIVideoGenerationClient, type OpenAIVideoGenerationConfig, type OpenAIVideoGenerationRequest, type OpenAIVideoTaskResponse } from "./openai_video_generation";
 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";
@@ -23,6 +25,7 @@ export interface LangchainClientConfig {
 export { DoubaoImageGenerationClient, type DoubaoImageGenerationConfig, type DoubaoImageGenerationRequest, type DoubaoImageGenerationResponse, type DoubaoImageSize, };
 export { GeminiImageGenerationClient, type GeminiImageGenerationConfig, type GeminiImageGenerationRequest, type GeminiImageGenerationResponse, };
 export { VideoGenerationClient, type VideoGenerationConfig, type VideoGenerationRequest, type ContentGenerationTaskID, type ContentGenerationTask, };
+export { OpenAIVideoGenerationClient, type OpenAIVideoGenerationConfig, type OpenAIVideoGenerationRequest, type OpenAIVideoTaskResponse, };
 export { DownloadClient, type DownloadConfig, type DownloadOptions, type StreamDownloadOptions, };
 export { AuthClient, getCurrentUserInfo, type AuthConfig, type UserInfo, };
 export { sdkConfig, VersionCompatibilityError, SDK_SIGNATURE } from "./config";

package/dist/index.js

@@ -6,7 +6,7 @@
  * @see https://github.com/langchain-ai/langchainjs
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SDK_SIGNATURE = exports.VersionCompatibilityError = exports.sdkConfig = exports.getCurrentUserInfo = exports.AuthClient = exports.DownloadClient = exports.VideoGenerationClient = exports.GeminiImageGenerationClient = exports.DoubaoImageGenerationClient = exports.ChatAnthropic = exports.ChatGoogleGenerativeAI = exports.ChatOpenAI = exports.BaseChatModel = exports.AIMessageChunk = exports.SystemMessage = exports.AIMessage = exports.HumanMessage = void 0;
+exports.SDK_SIGNATURE = exports.VersionCompatibilityError = exports.sdkConfig = exports.getCurrentUserInfo = exports.AuthClient = exports.DownloadClient = exports.OpenAIVideoGenerationClient = exports.VideoGenerationClient = exports.GeminiImageGenerationClient = exports.DoubaoImageGenerationClient = exports.ChatAnthropic = exports.ChatGoogleGenerativeAI = exports.ChatOpenAI = exports.BaseChatModel = exports.AIMessageChunk = exports.SystemMessage = exports.AIMessage = exports.HumanMessage = void 0;
 exports.createChatModel = createChatModel;
 const openai_1 = require("./chat_models/openai");
 const google_1 = require("./chat_models/google");
@@ -16,12 +16,15 @@ const gemini_image_generation_1 = require("./gemini-image-generation");
 Object.defineProperty(exports, "GeminiImageGenerationClient", { enumerable: true, get: function () { return gemini_image_generation_1.GeminiImageGenerationClient; } });
 const video_generation_1 = require("./video_generation");
 Object.defineProperty(exports, "VideoGenerationClient", { enumerable: true, get: function () { return video_generation_1.VideoGenerationClient; } });
+const openai_video_generation_1 = require("./openai_video_generation");
+Object.defineProperty(exports, "OpenAIVideoGenerationClient", { enumerable: true, get: function () { return openai_video_generation_1.OpenAIVideoGenerationClient; } });
 const download_1 = require("./download");
 Object.defineProperty(exports, "DownloadClient", { enumerable: true, get: function () { return download_1.DownloadClient; } });
 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; } });
@@ -56,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,
+    });
 }