ai-world-sdk 1.0.3 → 1.0.6

package/dist/__tests__/example.test.js

@@ -62,8 +62,9 @@ describe("Langchain SDK Tests", () => {
     });
     test("ChatGoogleGenerativeAI - 非流式调用", async () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
-            modelName: "gemini-2.0-flash-exp-image-generation",
+            modelName: "gemini-2.5-flash-image",
             temperature: 0.7,
+            provider: "gemini",
         });
         const response = await gemini.invoke([
             new index_1.HumanMessage("你好，请用一句话介绍人工智能。"),
@@ -78,6 +79,7 @@ describe("Langchain SDK Tests", () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
             modelName: "gemini-2.0-flash-exp-image-generation",
             temperature: 0.7,
+            provider: "gemini",
         });
         let fullText = "";
         for await (const chunk of gemini.stream([
@@ -95,6 +97,7 @@ describe("Langchain SDK Tests", () => {
     test("createChatModel 工厂函数", async () => {
         const model = (0, index_1.createChatModel)("gemini-2.0-flash-exp-image-generation", {
             temperature: 0.7,
+            provider: "gemini",
         });
         const response = await model.invoke([
             new index_1.SystemMessage("You are a helpful assistant."),
@@ -110,6 +113,7 @@ describe("Langchain SDK Tests", () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
             modelName: "gemini-2.0-flash-exp-image-generation",
             temperature: 0.5,
+            provider: "gemini",
         });
         // 使用 bind 方法绑定新的参数
         const boundModel = gemini.bind({ temperature: 0.9 });
@@ -125,6 +129,7 @@ describe("Langchain SDK Tests", () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
             modelName: "gemini-2.0-flash-exp-image-generation",
             temperature: 0.7,
+            provider: "gemini",
         });
         // 批量调用
         const responses = await gemini.batch([
@@ -142,13 +147,26 @@ describe("Langchain SDK Tests", () => {
         console.log("✅ batch 方法测试成功");
         console.log(`批量响应数量: ${responses.length}`);
     }, 30000);
+    // 添加openai模型测试
+    test("ChatOpenAI - 基础测试", async () => {
+        const openai = new index_1.ChatOpenAI({
+            modelName: "gpt-4o-mini",
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        const response = await openai.invoke([
+            new index_1.HumanMessage("你好，请介绍一下你自己"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ ChatOpenAI 基础测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
     test("ChatGoogleGenerativeAI - aihubmix.com no stream", async () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
-            modelName: "gemini-2.0-flash-exp-image-generation",
+            modelName: "gemini-2.5-flash-image",
             temperature: 0.7,
-            headers: {
-                "X-Base-Url": "https://aihubmix.com",
-            },
+            provider: "aihubmix",
         });
         const response = await gemini.invoke([
             new index_1.HumanMessage("aihubmix.com是做什么的"),
@@ -161,11 +179,9 @@ describe("Langchain SDK Tests", () => {
     }, 30000);
     test("ChatGoogleGenerativeAI - aihubmix.com stream", async () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
-            modelName: "gemini-2.0-flash-exp-image-generation",
+            modelName: "gemini-2.5-flash-image",
             temperature: 0.7,
-            headers: {
-                "X-Base-Url": "https://aihubmix.com",
-            },
+            provider: "aihubmix",
         });
         let fullText = "";
         for await (const chunk of gemini.stream([
@@ -184,6 +200,7 @@ describe("Langchain SDK Tests", () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
             modelName: "gemini-3-pro-image-preview",
             temperature: 0.7,
+            provider: "gemini",
         });
         const response = await gemini.invoke([
             new index_1.HumanMessage("请生成一张展示人工智能概念的图片"),
@@ -215,43 +232,7 @@ describe("Langchain SDK Tests", () => {
         const gemini = new index_1.ChatGoogleGenerativeAI({
             modelName: "gemini-3-pro-image-preview",
             temperature: 0.7,
-            headers: {
-                "X-Base-Url": "https://aihubmix.com",
-            },
-        });
-        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("ChatGoogleGenerativeAI - seedream 图像生成", async () => {
-        const gemini = new index_1.ChatGoogleGenerativeAI({
-            modelName: "doubao-seedream-4-5-251128",
-            temperature: 0.7,
-            headers: {
-                "X-Base-Url": "https://ark.cn-beijing.volces.com/api/v3/images/generations",
-            },
+            provider: "aihubmix",
         });
         const response = await gemini.invoke([
             new index_1.HumanMessage("请生成一张展示人工智能概念的图片"),
@@ -447,4 +428,462 @@ describe("Langchain SDK Tests", () => {
         console.log("✅ 使用 content 参数创建任务成功");
         console.log("任务 ID:", task.id);
     }, 30000);
+    // ========== 新增测试用例 ==========
+    test("ChatAnthropic - 基础测试", async () => {
+        const claude = new index_1.ChatAnthropic({
+            modelName: "claude-3-sonnet-20240229",
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        const response = await claude.invoke([
+            new index_1.HumanMessage("你好，请用一句话介绍你自己"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        expect(typeof response.content).toBe("string");
+        console.log("✅ ChatAnthropic 基础测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("ChatAnthropic - 流式调用", async () => {
+        const claude = new index_1.ChatAnthropic({
+            modelName: "claude-3-sonnet-20240229",
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        let fullText = "";
+        for await (const chunk of claude.stream([
+            new index_1.HumanMessage("请用一句话介绍人工智能，然后解释它的重要性。"),
+        ])) {
+            fullText += extractTextFromChunk(chunk);
+        }
+        expect(fullText.length).toBeGreaterThan(0);
+        console.log("✅ ChatAnthropic 流式测试成功");
+        console.log(`完整回复长度: ${fullText.length} 字符`);
+    }, 30000);
+    test("createChatModel - Claude 模型", async () => {
+        const model = (0, index_1.createChatModel)("claude-3-sonnet-20240229", {
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        expect(model).toBeInstanceOf(index_1.ChatAnthropic);
+        const response = await model.invoke([
+            new index_1.HumanMessage("What is the capital of France?"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ createChatModel Claude 测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("createChatModel - Doubao 模型", async () => {
+        const model = (0, index_1.createChatModel)("doubao-seedream-4-5-251128", {
+            temperature: 0.7,
+            provider: "doubao",
+        });
+        expect(model).toBeInstanceOf(index_1.ChatOpenAI);
+        const response = await model.invoke([
+            new index_1.HumanMessage("你好"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ createChatModel Doubao 测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("createChatModel - GPT 模型", async () => {
+        const model = (0, index_1.createChatModel)("gpt-4o-mini", {
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        expect(model).toBeInstanceOf(index_1.ChatOpenAI);
+        const response = await model.invoke([
+            new index_1.HumanMessage("Hello"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ createChatModel GPT 测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("BaseChatModel - getModelName 方法", () => {
+        const gemini = new index_1.ChatGoogleGenerativeAI({
+            modelName: "gemini-2.0-flash-exp-image-generation",
+            temperature: 0.7,
+            provider: "gemini",
+        });
+        expect(gemini.getModelName()).toBe("gemini-2.0-flash-exp-image-generation");
+        console.log("✅ getModelName 方法测试成功");
+    });
+    test("BaseChatModel - bindTools 方法", async () => {
+        const gemini = new index_1.ChatGoogleGenerativeAI({
+            modelName: "gemini-2.0-flash-exp-image-generation",
+            temperature: 0.7,
+            provider: "gemini",
+        });
+        const tools = [
+            {
+                type: "function",
+                function: {
+                    name: "get_weather",
+                    description: "Get the current weather in a location",
+                    parameters: {
+                        type: "object",
+                        properties: {
+                            location: {
+                                type: "string",
+                                description: "The city and state, e.g. San Francisco, CA",
+                            },
+                        },
+                        required: ["location"],
+                    },
+                },
+            },
+        ];
+        const modelWithTools = gemini.bindTools(tools);
+        const response = await modelWithTools.invoke([
+            new index_1.HumanMessage("What is the weather in Paris?"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ bindTools 方法测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("BaseChatModel - 多模态输入（文本+图像）", async () => {
+        const gemini = new index_1.ChatGoogleGenerativeAI({
+            modelName: "gemini-2.0-flash-exp-image-generation",
+            temperature: 0.7,
+            provider: "gemini",
+        });
+        const response = await gemini.invoke([
+            new index_1.HumanMessage([
+                { type: "text", text: "Describe this image" },
+                {
+                    type: "image_url",
+                    image_url: "data:image/jpeg;base64,/9j/4AAQSkZJRg...",
+                },
+            ]),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ 多模态输入测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("消息类型 - SystemMessage", () => {
+        const message = new index_1.SystemMessage("You are a helpful assistant.");
+        expect(message.role).toBe("system");
+        expect(message.content).toBe("You are a helpful assistant.");
+        console.log("✅ SystemMessage 测试成功");
+    });
+    test("消息类型 - AIMessage", () => {
+        const message = new index_1.AIMessage("Hello! How can I help you?");
+        expect(message.role).toBe("assistant");
+        expect(message.content).toBe("Hello! How can I help you?");
+        console.log("✅ AIMessage 测试成功");
+    });
+    test("消息类型 - HumanMessage with array content", () => {
+        const message = new index_1.HumanMessage([
+            { type: "text", text: "Hello" },
+            { type: "image_url", image_url: "https://example.com/image.jpg" },
+        ]);
+        expect(message.role).toBe("user");
+        expect(Array.isArray(message.content)).toBe(true);
+        console.log("✅ HumanMessage with array content 测试成功");
+    });
+    test("GeminiImageGenerationClient - 基础图像生成", async () => {
+        const imageClient = new index_1.GeminiImageGenerationClient({});
+        const result = await imageClient.generate({
+            prompt: "A beautiful sunset over the ocean with vibrant colors",
+            model: "gemini-2.0-flash-exp-image-generation",
+            number_of_images: 1,
+            aspect_ratio: "16:9",
+        });
+        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 基础测试成功");
+        console.log(`生成图像数量: ${result.data.length}`);
+        console.log("图像 URL:", result.data[0]?.url || "Base64 编码");
+        if (result.text) {
+            console.log("图像描述:", result.text);
+        }
+    }, 120000);
+    test("GeminiImageGenerationClient - 多图像生成", async () => {
+        const imageClient = new index_1.GeminiImageGenerationClient({});
+        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",
+        });
+        expect(result).toBeDefined();
+        expect(result.data).toBeDefined();
+        expect(Array.isArray(result.data)).toBe(true);
+        expect(result.data.length).toBe(2);
+        console.log("✅ GeminiImageGenerationClient 多图像生成测试成功");
+        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} 测试成功`);
+        }
+    }, 180000);
+    test("DoubaoImageGenerationClient - quality 参数测试", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
+        const qualities = ["standard", "hd"];
+        for (const quality of qualities) {
+            const result = await imageClient.generate({
+                prompt: "A beautiful sunset",
+                model: "doubao-seedream-4-5-251128",
+                size: "2K",
+                quality: quality,
+                n: 1,
+            });
+            expect(result).toBeDefined();
+            expect(result.data).toBeDefined();
+            expect(result.data.length).toBeGreaterThan(0);
+            console.log(`✅ quality ${quality} 测试成功`);
+        }
+    }, 180000);
+    test("DoubaoImageGenerationClient - style 参数测试", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
+        const styles = ["vivid", "natural"];
+        for (const style of styles) {
+            const result = await imageClient.generate({
+                prompt: "A beautiful landscape",
+                model: "doubao-seedream-4-5-251128",
+                size: "2K",
+                style: style,
+                n: 1,
+            });
+            expect(result).toBeDefined();
+            expect(result.data).toBeDefined();
+            expect(result.data.length).toBeGreaterThan(0);
+            console.log(`✅ style ${style} 测试成功`);
+        }
+    }, 180000);
+    test("DoubaoImageGenerationClient - response_format 参数测试", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
+        // 测试 URL 格式
+        const urlResult = await imageClient.generate({
+            prompt: "A beautiful sunset",
+            model: "doubao-seedream-4-5-251128",
+            size: "2K",
+            response_format: "url",
+            n: 1,
+        });
+        expect(urlResult).toBeDefined();
+        expect(urlResult.data[0]?.url).toBeDefined();
+        console.log("✅ response_format url 测试成功");
+        // 测试 b64_json 格式
+        const b64Result = await imageClient.generate({
+            prompt: "A beautiful sunset",
+            model: "doubao-seedream-4-5-251128",
+            size: "2K",
+            response_format: "b64_json",
+            n: 1,
+        });
+        expect(b64Result).toBeDefined();
+        expect(b64Result.data[0]?.b64_json).toBeDefined();
+        console.log("✅ response_format b64_json 测试成功");
+    }, 180000);
+    test("DoubaoImageGenerationClient - watermark 参数测试", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
+        const result = await imageClient.generate({
+            prompt: "A beautiful landscape",
+            model: "doubao-seedream-4-5-251128",
+            size: "2K",
+            watermark: false,
+            n: 1,
+        });
+        expect(result).toBeDefined();
+        expect(result.data).toBeDefined();
+        expect(result.data.length).toBeGreaterThan(0);
+        console.log("✅ watermark 参数测试成功");
+    }, 120000);
+    test("sdkConfig - 全局配置测试", () => {
+        const originalBaseUrl = index_1.sdkConfig.getServerUrl();
+        const originalToken = index_1.sdkConfig.getToken();
+        const originalHeaders = index_1.sdkConfig.getHeaders();
+        const originalDebug = index_1.sdkConfig.getDebug();
+        // 测试 setBaseUrl 和 getServerUrl
+        index_1.sdkConfig.setBaseUrl("http://test.example.com");
+        expect(index_1.sdkConfig.getServerUrl()).toBe("http://test.example.com");
+        // 测试 setToken 和 getToken
+        index_1.sdkConfig.setToken("test-token-123");
+        expect(index_1.sdkConfig.getToken()).toBe("test-token-123");
+        // 测试 setHeaders 和 getHeaders
+        index_1.sdkConfig.setHeaders({ "X-Custom-Header": "test-value" });
+        const headers = index_1.sdkConfig.getHeaders();
+        expect(headers["X-Custom-Header"]).toBe("test-value");
+        // 测试 setDebug 和 getDebug
+        index_1.sdkConfig.setDebug(true);
+        expect(index_1.sdkConfig.getDebug()).toBe(true);
+        index_1.sdkConfig.setDebug(false);
+        expect(index_1.sdkConfig.getDebug()).toBe(false);
+        // 测试 reset
+        index_1.sdkConfig.reset();
+        expect(index_1.sdkConfig.getServerUrl()).toBeNull();
+        expect(index_1.sdkConfig.getToken()).toBeNull();
+        expect(Object.keys(index_1.sdkConfig.getHeaders()).length).toBe(0);
+        expect(index_1.sdkConfig.getDebug()).toBe(false);
+        // 恢复原始配置
+        if (originalBaseUrl)
+            index_1.sdkConfig.setBaseUrl(originalBaseUrl);
+        if (originalToken)
+            index_1.sdkConfig.setToken(originalToken);
+        index_1.sdkConfig.setHeaders(originalHeaders);
+        index_1.sdkConfig.setDebug(originalDebug);
+        console.log("✅ sdkConfig 全局配置测试成功");
+    });
+    test("BaseChatModel - bind 方法链式调用", async () => {
+        const gemini = new index_1.ChatGoogleGenerativeAI({
+            modelName: "gemini-2.0-flash-exp-image-generation",
+            temperature: 0.5,
+            provider: "gemini",
+        });
+        // 链式绑定多个参数
+        const boundModel = gemini
+            .bind({ temperature: 0.9 })
+            .bind({ maxTokens: 1000 });
+        const response = await boundModel.invoke([
+            new index_1.HumanMessage("用一句话介绍人工智能。"),
+        ]);
+        expect(response).toBeDefined();
+        expect(response.content).toBeDefined();
+        console.log("✅ bind 方法链式调用测试成功");
+        console.log("AI:", response.content);
+    }, 30000);
+    test("ChatOpenAI - 流式调用", async () => {
+        const openai = new index_1.ChatOpenAI({
+            modelName: "gpt-4",
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        let fullText = "";
+        for await (const chunk of openai.stream([
+            new index_1.HumanMessage("请用一句话介绍人工智能。"),
+        ])) {
+            fullText += extractTextFromChunk(chunk);
+        }
+        expect(fullText.length).toBeGreaterThan(0);
+        console.log("✅ ChatOpenAI 流式测试成功");
+        console.log(`完整回复长度: ${fullText.length} 字符`);
+    }, 30000);
+    test("ChatOpenAI - batch 方法", async () => {
+        const openai = new index_1.ChatOpenAI({
+            modelName: "gpt-4",
+            temperature: 0.7,
+            provider: "aihubmix",
+        });
+        const responses = await openai.batch([
+            [new index_1.HumanMessage("什么是人工智能？")],
+            [new index_1.HumanMessage("什么是机器学习？")],
+        ]);
+        expect(responses).toBeDefined();
+        expect(Array.isArray(responses)).toBe(true);
+        expect(responses.length).toBe(2);
+        responses.forEach((response) => {
+            expect(response).toBeDefined();
+            expect(response.content).toBeDefined();
+        });
+        console.log("✅ ChatOpenAI batch 方法测试成功");
+    }, 30000);
+    test("VideoGenerationClient - 不同宽高比测试", async () => {
+        const videoClient = new index_1.VideoGenerationClient({});
+        const aspectRatios = ["16:9", "9:16", "1:1"];
+        for (const aspectRatio of aspectRatios) {
+            const task = await videoClient.create({
+                prompt: "A beautiful landscape",
+                model: "doubao-seedance-1-0-pro-fast-251015",
+                duration: 3,
+                aspect_ratio: aspectRatio,
+            });
+            expect(task).toBeDefined();
+            expect(task.id).toBeDefined();
+            console.log(`✅ 宽高比 ${aspectRatio} 测试成功，任务 ID: ${task.id}`);
+        }
+    }, 60000);
+    test("VideoGenerationClient - 不同时长测试", async () => {
+        const videoClient = new index_1.VideoGenerationClient({});
+        const durations = [1, 3, 5];
+        for (const duration of durations) {
+            const task = await videoClient.create({
+                prompt: "A short video clip",
+                model: "doubao-seedance-1-0-pro-fast-251015",
+                duration: duration,
+            });
+            expect(task).toBeDefined();
+            expect(task.id).toBeDefined();
+            console.log(`✅ 时长 ${duration} 秒测试成功，任务 ID: ${task.id}`);
+        }
+    }, 60000);
+    test("VideoGenerationClient - return_last_frame 参数测试", async () => {
+        const videoClient = new index_1.VideoGenerationClient({});
+        const task = await videoClient.create({
+            prompt: "A beautiful sunset",
+            model: "doubao-seedance-1-0-pro-fast-251015",
+            duration: 3,
+            return_last_frame: true,
+        });
+        expect(task).toBeDefined();
+        expect(task.id).toBeDefined();
+        // 查询任务，检查是否有 last_frame_url
+        const taskStatus = await videoClient.get(task.id);
+        if (taskStatus.status === "succeeded" && taskStatus.content?.last_frame_url) {
+            expect(taskStatus.content.last_frame_url).toBeDefined();
+            console.log("✅ return_last_frame 测试成功，最后一帧 URL:", taskStatus.content.last_frame_url);
+        }
+        else {
+            console.log("✅ return_last_frame 参数已设置，任务 ID:", task.id);
+        }
+    }, 30000);
+    test("AIMessageChunk - 构造函数测试", () => {
+        // 测试从字符串创建
+        const chunk1 = new index_1.AIMessageChunk("Hello");
+        expect(chunk1.content).toBe("Hello");
+        expect(chunk1.role).toBe("assistant");
+        // 测试从数组创建
+        const chunk2 = new index_1.AIMessageChunk([
+            { type: "text", text: "Hello" },
+        ]);
+        expect(Array.isArray(chunk2.content)).toBe(true);
+        // 测试从对象创建
+        const chunk3 = new index_1.AIMessageChunk({
+            content: "Hello",
+            id: "chunk-123",
+            response_metadata: { token_count: 5 },
+        });
+        expect(chunk3.content).toBe("Hello");
+        expect(chunk3.id).toBe("chunk-123");
+        expect(chunk3.response_metadata).toBeDefined();
+        console.log("✅ AIMessageChunk 构造函数测试成功");
+    });
+    test("createChatModel - 不支持的模型应该抛出错误", () => {
+        expect(() => {
+            (0, index_1.createChatModel)("unsupported-model", {
+                temperature: 0.7,
+                provider: "aihubmix",
+            });
+        }).toThrow("Unsupported model");
+        console.log("✅ createChatModel 错误处理测试成功");
+    });
 });

package/dist/base.d.ts

@@ -3,14 +3,15 @@
  * Similar to LangChain.js BaseChatModel
  */
 import { BaseMessage, AIMessage, AIMessageChunk } from "./messages";
+export type AIModelProvider = "aihubmix" | "doubao" | "gemini";
 export interface BaseChatModelParams {
+    provider: AIModelProvider;
     baseUrl?: string;
     headers?: Record<string, string>;
     temperature?: number;
     maxTokens?: number;
     topP?: number;
     modelName?: string;
-    provider?: string;
     apiKey?: string;
 }
 export interface ChatResult {
@@ -44,7 +45,7 @@ export declare abstract class BaseChatModel {
     protected topP?: number;
     protected modelName: string;
     protected boundOptions?: BindOptions;
-    protected provider?: string;
+    protected provider: string;
     protected apiKey?: string;
     constructor(config: {
         baseUrl?: string;
@@ -53,7 +54,7 @@ export declare abstract class BaseChatModel {
         maxTokens?: number;
         topP?: number;
         modelName: string;
-        provider?: string;
+        provider: AIModelProvider;
         apiKey?: string;
     });
     /**

package/dist/chat_models/google.js

@@ -10,7 +10,7 @@ class ChatGoogleGenerativeAI extends base_1.BaseChatModel {
     constructor(config) {
         super({
             ...config,
-            modelName: config.modelName || "gemini-1.5-pro",
+            modelName: config.modelName || "gemini-3-pro-image-preview",
         });
     }
 }

package/dist/chat_models/openai.js

@@ -10,7 +10,7 @@ class ChatOpenAI extends base_1.BaseChatModel {
     constructor(config) {
         super({
             ...config,
-            modelName: config.modelName || "gpt-3.5-turbo",
+            modelName: config.modelName || "gpt-4o-mini",
         });
     }
 }