ai-world-sdk 1.0.7 → 1.0.9

package/README.md

@@ -29,7 +29,7 @@ npm install ai-world-sdk
 import { sdkConfig } from 'ai-world-sdk';
 
 // 设置全局配置（只需一次）
-sdkConfig.setBaseUrl('http://localhost:8000');
+sdkConfig.setBaseUrl('your-base-url');
 sdkConfig.setToken('your-jwt-token');
 sdkConfig.setDebug(true); // 可选：启用调试模式
 
@@ -233,6 +233,48 @@ const result2 = await client.generate({
   temperature: 0.7,                             // 可选: 0.0-2.0
   max_output_tokens: 1000,                      // 可选
 });
+
+// 使用 aihubmix provider（通过 aihubmix 代理服务）
+const aihubmixClient = new GeminiImageGenerationClient({
+  provider: 'aihubmix',  // 使用 aihubmix 代理
+});
+
+const result3 = await aihubmixClient.generate({
+  prompt: 'A beautiful landscape with mountains',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '16:9',
+  image_size: '1K',
+  number_of_images: 1,
+  response_modalities: ['IMAGE'],
+});
+
+// 图像编辑（使用文本提示编辑图片）
+const editResult = await client.edit({
+  prompt: 'Add a small wizard hat on the cat\'s head',
+  image: 'data:image/png;base64,iVBORw0KGgo...', // base64 编码的图片数据或 data URL
+  model: 'gemini-2.5-flash-image',
+  aspect_ratio: '1:1',
+  response_modalities: ['IMAGE'],
+});
+
+// 多轮图片修改（迭代式优化图片）
+// 第一轮：创建初始图片
+const firstResponse = await client.chat({
+  message: 'Create a vibrant infographic about photosynthesis',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '16:9',
+  response_modalities: ['TEXT', 'IMAGE'],
+});
+
+// 后续轮次：修改图片（使用返回的 chat_id）
+const secondResponse = await client.chat({
+  chat_id: firstResponse.chat_id,
+  message: 'Update this infographic to be in Spanish',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '16:9',
+  image_size: '2K',
+  response_modalities: ['TEXT', 'IMAGE'],
+});
 ```
 
 **参数说明：**
@@ -250,6 +292,11 @@ const result2 = await client.generate({
 
 **支持的宽高比：** `1:1`, `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9`
 
+**Provider 说明：**
+
+- `provider: 'gemini'`（默认）- 直接使用 Google Gemini 官方 API
+- `provider: 'aihubmix'` - 通过 aihubmix 代理服务调用，适合需要统一管理多个 API 的场景
+
 **模型对比：**
 
 | 模型 | 分辨率 | 特点 | 适用场景 |
@@ -257,6 +304,17 @@ const result2 = await client.generate({
 | `gemini-2.5-flash-image` | 1024px | 快速、高效、成本低 | 日常使用、批量生成 |
 | `gemini-3-pro-image-preview` | 1K/2K/4K | 专业级、高分辨率、高级功能 | 专业设计、高分辨率需求 |
 
+**图像编辑和多轮修改：**
+
+- **`edit()`** - 图像编辑：使用文本提示编辑图片，支持添加、移除或修改元素
+  - 需要提供输入图片（base64 编码或 data URL）
+  - 适用于单次编辑操作
+
+- **`chat()`** - 多轮图片修改：通过对话迭代式优化图片
+  - 首次调用创建新的聊天会话，返回 `chat_id`
+  - 后续调用使用 `chat_id` 继续对话
+  - 推荐使用 `gemini-3-pro-image-preview` 模型进行多轮编辑
+
 ### 视频生成
 
 #### VideoGenerationClient
@@ -290,7 +348,7 @@ const result = await client.poll(task.id, {
 import { sdkConfig } from 'ai-world-sdk';
 
 // 设置配置
-sdkConfig.setBaseUrl('http://localhost:8000');
+sdkConfig.setBaseUrl('your-base-url');
 sdkConfig.setToken('your-jwt-token');
 sdkConfig.setHeaders({ 'X-Custom-Header': 'value' });
 sdkConfig.setDebug(true);
@@ -345,11 +403,26 @@ const gptModel = new ChatOpenAI({
 
 // Gemini 模型（通过 aihubmix）
 const geminiModel = new ChatGoogleGenerativeAI({
-  modelName: 'gemini-2.5-flash-image',
+  modelName: 'gemini-2.5-flash',
   temperature: 0.7,
   provider: 'aihubmix',
 });
 
+// Gemini 图像生成（通过 aihubmix）
+import { GeminiImageGenerationClient } from 'ai-world-sdk';
+
+const geminiImageClient = new GeminiImageGenerationClient({
+  provider: 'aihubmix',
+});
+
+const imageResult = await geminiImageClient.generate({
+  prompt: 'A beautiful landscape',
+  model: 'gemini-3-pro-image-preview',
+  aspect_ratio: '16:9',
+  image_size: '1K',
+  response_modalities: ['IMAGE'],
+});
+
 // Claude 模型（通过 aihubmix）
 const claudeModel = new ChatAnthropic({
   modelName: 'claude-3-sonnet-20240229',

package/dist/__tests__/example.test.js

@@ -228,38 +228,6 @@ describe("Langchain SDK Tests", () => {
             console.log("响应内容:", content);
         }
     }, 60000);
-    test("ChatGoogleGenerativeAI - aihubmix.com 图像生成", async () => {
-        const gemini = new index_1.ChatGoogleGenerativeAI({
-            modelName: "gemini-3-pro-image-preview",
-            temperature: 0.7,
-            provider: "aihubmix",
-        });
-        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("DoubaoImageGenerationClient - 基础图像生成", async () => {
         const imageClient = new index_1.DoubaoImageGenerationClient({});
         const result = await imageClient.generate({
@@ -592,7 +560,7 @@ describe("Langchain SDK Tests", () => {
             prompt: 'A beautiful sunset over the ocean',
             model: 'gemini-3-pro-image-preview',
             aspect_ratio: '16:9',
-            image_size: '2K', // 仅适用于 gemini-3-pro-image-preview
+            image_size: '1K', // 仅适用于 gemini-3-pro-image-preview
             number_of_images: 1,
             response_modalities: ['IMAGE'], // 仅返回图片
         });
@@ -613,41 +581,121 @@ describe("Langchain SDK Tests", () => {
             console.log("图像描述:", result.text);
         }
     }, 120000);
-    test("GeminiImageGenerationClient - 多图像生成", async () => {
-        const imageClient = new index_1.GeminiImageGenerationClient({});
+    test("GeminiImageGenerationClient - 使用 aihubmix provider", async () => {
+        const imageClient = new index_1.GeminiImageGenerationClient({
+            provider: "aihubmix",
+        });
         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",
+            prompt: 'A beautiful sunset over the ocean',
+            model: 'gemini-3-pro-image-preview',
+            aspect_ratio: '16:9',
+            image_size: '1K',
+            number_of_images: 1,
+            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).toBe(2);
-        console.log("✅ GeminiImageGenerationClient 多图像生成测试成功");
+        expect(result.data.length).toBeGreaterThan(0);
+        result.data.forEach((item) => {
+            expect(item).toBeDefined();
+            expect(item.url || item.b64_json).toBeDefined();
+        });
+        console.log("✅ GeminiImageGenerationClient (aihubmix 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 () => {
+    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} 测试成功`);
+        // 首先生成一张基础图片
+        const generateResult = await imageClient.generate({
+            prompt: 'A simple red apple on a white background',
+            model: 'gemini-2.5-flash-image',
+            aspect_ratio: '1:1',
+            response_modalities: ['IMAGE'],
+        });
+        expect(generateResult).toBeDefined();
+        expect(generateResult.data).toBeDefined();
+        expect(generateResult.data.length).toBeGreaterThan(0);
+        // 获取第一张图片的 base64 数据
+        const firstImage = generateResult.data[0];
+        expect(firstImage).toBeDefined();
+        // 提取 base64 数据（优先使用 b64_json，否则从 text 中提取）
+        let base64Image;
+        if (firstImage.b64_json) {
+            base64Image = firstImage.b64_json;
         }
+        else if (firstImage.text && firstImage.text.startsWith('data:')) {
+            // 从 data URL 中提取 base64 部分
+            base64Image = firstImage.text.split(',')[1];
+        }
+        else {
+            throw new Error('无法获取图片的 base64 数据');
+        }
+        expect(base64Image).toBeDefined();
+        // 使用编辑功能添加元素
+        const editResult = await imageClient.edit({
+            prompt: 'Add a small green leaf on top of the apple',
+            image: base64Image, // 直接使用 base64 字符串
+            model: 'gemini-2.5-flash-image',
+            aspect_ratio: '1:1',
+            response_modalities: ['IMAGE'],
+        });
+        expect(editResult).toBeDefined();
+        expect(editResult.created).toBeDefined();
+        expect(editResult.data).toBeDefined();
+        expect(Array.isArray(editResult.data)).toBe(true);
+        expect(editResult.data.length).toBeGreaterThan(0);
+        editResult.data.forEach((item) => {
+            expect(item).toBeDefined();
+            expect(item.url || item.b64_json).toBeDefined();
+        });
+        console.log("✅ GeminiImageGenerationClient 图像编辑测试成功");
+        console.log(`编辑后图像数量: ${editResult.data.length}`);
     }, 180000);
+    test("GeminiImageGenerationClient - 多轮图片修改", async () => {
+        const imageClient = new index_1.GeminiImageGenerationClient({});
+        // 第一轮：创建初始图片
+        const firstResponse = await imageClient.chat({
+            message: 'Create a vibrant infographic that explains photosynthesis as if it were a recipe for a plant\'s favorite food. Show the "ingredients" (sunlight, water, CO2) and the "finished dish" (sugar/energy). The style should be like a page from a colorful kids\' cookbook, suitable for a 4th grader.',
+            model: 'gemini-3-pro-image-preview',
+            aspect_ratio: '16:9',
+            response_modalities: ['TEXT', 'IMAGE'],
+        });
+        expect(firstResponse).toBeDefined();
+        expect(firstResponse.chat_id).toBeDefined();
+        expect(typeof firstResponse.chat_id).toBe("string");
+        expect(firstResponse.data).toBeDefined();
+        expect(Array.isArray(firstResponse.data)).toBe(true);
+        expect(firstResponse.data.length).toBeGreaterThan(0);
+        console.log("✅ 第一轮图片生成成功");
+        console.log(`Chat ID: ${firstResponse.chat_id}`);
+        console.log(`生成图像数量: ${firstResponse.data.length}`);
+        // 第二轮：修改图片（将语言改为西班牙语）
+        const secondResponse = await imageClient.chat({
+            chat_id: firstResponse.chat_id,
+            message: 'Update this infographic to be in Spanish. Do not change any other elements of the image.',
+            model: 'gemini-3-pro-image-preview',
+            aspect_ratio: '16:9',
+            image_size: '2K',
+            response_modalities: ['TEXT', 'IMAGE'],
+        });
+        expect(secondResponse).toBeDefined();
+        expect(secondResponse.chat_id).toBe(firstResponse.chat_id); // chat_id 应该保持一致
+        expect(secondResponse.data).toBeDefined();
+        expect(Array.isArray(secondResponse.data)).toBe(true);
+        expect(secondResponse.data.length).toBeGreaterThan(0);
+        console.log("✅ 第二轮图片修改成功");
+        console.log(`修改后图像数量: ${secondResponse.data.length}`);
+        if (secondResponse.text) {
+            console.log("文本响应:", secondResponse.text);
+        }
+    }, 240000);
     test("DoubaoImageGenerationClient - quality 参数测试", async () => {
         const imageClient = new index_1.DoubaoImageGenerationClient({});
         const qualities = ["standard", "hd"];
@@ -662,7 +710,7 @@ describe("Langchain SDK Tests", () => {
             expect(result).toBeDefined();
             expect(result.data).toBeDefined();
             expect(result.data.length).toBeGreaterThan(0);
-            console.log(`✅ quality ${quality} 测试成功`);
+            console.log(`✅ quality ${quality} 测试成功`, result.data);
         }
     }, 180000);
     test("DoubaoImageGenerationClient - style 参数测试", async () => {
@@ -679,7 +727,7 @@ describe("Langchain SDK Tests", () => {
             expect(result).toBeDefined();
             expect(result.data).toBeDefined();
             expect(result.data.length).toBeGreaterThan(0);
-            console.log(`✅ style ${style} 测试成功`);
+            console.log(`✅ style ${style} 测试成功`, result.data);
         }
     }, 180000);
     test("DoubaoImageGenerationClient - response_format 参数测试", async () => {
@@ -719,7 +767,7 @@ describe("Langchain SDK Tests", () => {
         expect(result).toBeDefined();
         expect(result.data).toBeDefined();
         expect(result.data.length).toBeGreaterThan(0);
-        console.log("✅ watermark 参数测试成功");
+        console.log("✅ watermark 参数测试成功", result.data);
     }, 120000);
     test("sdkConfig - 全局配置测试", () => {
         const originalBaseUrl = index_1.sdkConfig.getServerUrl();

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

@@ -3,6 +3,7 @@
  * Google Gemini 图像生成客户端
  */
 export interface GeminiImageGenerationConfig {
+    provider?: "aihubmix" | "gemini";
     baseUrl?: string;
     headers?: Record<string, string>;
 }
@@ -27,8 +28,35 @@ export interface GeminiImageGenerationResponse {
     data: GeminiImageData[];
     text?: string;
 }
+export interface GeminiImageEditRequest {
+    prompt: string;
+    image: string;
+    model?: string;
+    provider?: "aihubmix" | "gemini";
+    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")[];
+    user?: string;
+}
+export interface GeminiImageChatRequest {
+    message: string;
+    chat_id?: string;
+    model?: string;
+    provider?: "aihubmix" | "gemini";
+    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")[];
+    user?: string;
+}
+export interface GeminiImageChatResponse {
+    chat_id: string;
+    created: number;
+    data: GeminiImageData[];
+    text?: string;
+}
 export declare class GeminiImageGenerationClient {
     private headers;
+    private provider;
     constructor(config?: GeminiImageGenerationConfig);
     /**
      * Generate images
@@ -41,4 +69,45 @@ export declare class GeminiImageGenerationClient {
      * - number_of_images: 生成图片数量（1-4）
      */
     generate(request: GeminiImageGenerationRequest): Promise<GeminiImageGenerationResponse>;
+    /**
+     * Edit images
+     * 编辑图像
+     *
+     * 使用文本提示编辑图片，支持添加、移除或修改元素
+     *
+     * 支持的参数：
+     * - image: base64 编码的图片数据，或 data URL（如 data:image/png;base64,...）
+     * - aspect_ratio: 宽高比
+     * - image_size: 图片大小（仅 gemini-3-pro-image-preview）
+     * - response_modalities: 响应模态
+     */
+    edit(request: GeminiImageEditRequest): Promise<GeminiImageGenerationResponse>;
+    /**
+     * Chat with images (multi-turn image editing)
+     * 图像多轮对话（用于多轮图片修改）
+     *
+     * 支持通过多轮对话迭代式优化图片
+     *
+     * 使用示例：
+     * ```typescript
+     * // 第一轮：创建初始图片
+     * const firstResponse = await client.chat({
+     *   message: "Create a vibrant infographic about photosynthesis"
+     * });
+     *
+     * // 后续轮次：修改图片（使用返回的 chat_id）
+     * const secondResponse = await client.chat({
+     *   chat_id: firstResponse.chat_id,
+     *   message: "Update this infographic to be in Spanish"
+     * });
+     * ```
+     *
+     * 支持的参数：
+     * - message: 当前轮次的提示词
+     * - chat_id: 聊天 ID（用于多轮对话，首次请求时不需要）
+     * - aspect_ratio: 宽高比
+     * - image_size: 图片大小（仅 gemini-3-pro-image-preview）
+     * - response_modalities: 响应模态
+     */
+    chat(request: GeminiImageChatRequest): Promise<GeminiImageChatResponse>;
 }

package/dist/gemini-image-generation.js

@@ -9,15 +9,21 @@ const config_1 = require("./config");
 const log_1 = require("./log");
 class GeminiImageGenerationClient {
     constructor(config) {
+        // 确定 provider（默认为 gemini）
+        this.provider = config?.provider || "gemini";
         // 合并全局 headers 和配置 headers
         const globalHeaders = config_1.sdkConfig.getHeaders();
         this.headers = {
             "Content-Type": "application/json",
             "Authorization": `Bearer ${config_1.sdkConfig.getToken()}`,
-            "X-Base-Url": config?.baseUrl || "",
+            "X-Provider": this.provider, // 在 header 中传递 provider，由后端确定 baseUrl
             ...globalHeaders,
             ...config?.headers,
         };
+        // 如果明确指定了 baseUrl，也添加到 header 中
+        if (config?.baseUrl) {
+            this.headers["X-Base-Url"] = config.baseUrl;
+        }
     }
     /**
      * Generate images
@@ -70,5 +76,122 @@ class GeminiImageGenerationClient {
         (0, log_1.logResponse)(response.status, response.statusText, response.headers, data);
         return data;
     }
+    /**
+     * Edit images
+     * 编辑图像
+     *
+     * 使用文本提示编辑图片，支持添加、移除或修改元素
+     *
+     * 支持的参数：
+     * - image: base64 编码的图片数据，或 data URL（如 data:image/png;base64,...）
+     * - aspect_ratio: 宽高比
+     * - image_size: 图片大小（仅 gemini-3-pro-image-preview）
+     * - response_modalities: 响应模态
+     */
+    async edit(request) {
+        const requestBody = {
+            prompt: request.prompt,
+            image: request.image,
+            model: request.model || "gemini-2.5-flash-image",
+        };
+        // 添加可选参数
+        if (request.aspect_ratio) {
+            requestBody.aspect_ratio = request.aspect_ratio;
+        }
+        if (request.image_size) {
+            requestBody.image_size = request.image_size;
+        }
+        if (request.response_modalities) {
+            requestBody.response_modalities = request.response_modalities;
+        }
+        if (request.user) {
+            requestBody.user = request.user;
+        }
+        if (request.provider) {
+            requestBody.provider = request.provider;
+        }
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/gemini-image-proxy/edit`;
+        (0, log_1.logRequest)("POST", url, this.headers, { ...requestBody, image: "[base64 data]" });
+        const response = await fetch(url, {
+            method: "POST",
+            headers: this.headers,
+            body: JSON.stringify(requestBody),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            (0, log_1.logResponse)(response.status, response.statusText, response.headers, errorText);
+            throw new Error(`Gemini image edit API error: ${response.status} ${errorText}`);
+        }
+        const data = (await response.json());
+        (0, log_1.logResponse)(response.status, response.statusText, response.headers, data);
+        return data;
+    }
+    /**
+     * Chat with images (multi-turn image editing)
+     * 图像多轮对话（用于多轮图片修改）
+     *
+     * 支持通过多轮对话迭代式优化图片
+     *
+     * 使用示例：
+     * ```typescript
+     * // 第一轮：创建初始图片
+     * const firstResponse = await client.chat({
+     *   message: "Create a vibrant infographic about photosynthesis"
+     * });
+     *
+     * // 后续轮次：修改图片（使用返回的 chat_id）
+     * const secondResponse = await client.chat({
+     *   chat_id: firstResponse.chat_id,
+     *   message: "Update this infographic to be in Spanish"
+     * });
+     * ```
+     *
+     * 支持的参数：
+     * - message: 当前轮次的提示词
+     * - chat_id: 聊天 ID（用于多轮对话，首次请求时不需要）
+     * - aspect_ratio: 宽高比
+     * - image_size: 图片大小（仅 gemini-3-pro-image-preview）
+     * - response_modalities: 响应模态
+     */
+    async chat(request) {
+        const requestBody = {
+            message: request.message,
+            model: request.model || "gemini-3-pro-image-preview",
+        };
+        // 添加可选参数
+        if (request.chat_id) {
+            requestBody.chat_id = request.chat_id;
+        }
+        if (request.aspect_ratio) {
+            requestBody.aspect_ratio = request.aspect_ratio;
+        }
+        if (request.image_size) {
+            requestBody.image_size = request.image_size;
+        }
+        if (request.response_modalities) {
+            requestBody.response_modalities = request.response_modalities;
+        }
+        if (request.user) {
+            requestBody.user = request.user;
+        }
+        if (request.provider) {
+            requestBody.provider = request.provider;
+        }
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/gemini-image-proxy/chat`;
+        (0, log_1.logRequest)("POST", url, this.headers, requestBody);
+        const response = await fetch(url, {
+            method: "POST",
+            headers: this.headers,
+            body: JSON.stringify(requestBody),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            (0, log_1.logResponse)(response.status, response.statusText, response.headers, errorText);
+            throw new Error(`Gemini image chat API error: ${response.status} ${errorText}`);
+        }
+        const data = (await response.json());
+        (0, log_1.logResponse)(response.status, response.statusText, response.headers, data);
+        return data;
+    }
 }
 exports.GeminiImageGenerationClient = GeminiImageGenerationClient;

package/dist/index.d.ts

@@ -10,7 +10,7 @@ import { GeminiImageGenerationClient, type GeminiImageGenerationConfig, type Gem
 import { VideoGenerationClient, type VideoGenerationConfig, type VideoGenerationRequest, type ContentGenerationTaskID, type ContentGenerationTask } from "./video_generation";
 import { sdkConfig } from "./config";
 export { BaseMessage, HumanMessage, AIMessage, SystemMessage, AIMessageChunk, type MessageContent, type AIMessageChunkData, } from "./messages";
-export { BaseChatModel, type BaseChatModelParams, type ToolDefinition, type BindOptions, } from "./base";
+export { BaseChatModel, type BaseChatModelParams, type AIModelProvider, type ToolDefinition, type BindOptions, } from "./base";
 export { ChatOpenAI } from "./chat_models/openai";
 export { ChatGoogleGenerativeAI } from "./chat_models/google";
 export { ChatAnthropic } from "./chat_models/anthropic";

package/package.json

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