ai-world-sdk 1.0.3 → 1.0.6

package/README.md

@@ -5,174 +5,72 @@ TypeScript SDK for AI World Platform - 一个功能完整的 AI 应用开发 SDK
 [![npm version](https://img.shields.io/npm/v/ai-world-sdk)](https://www.npmjs.com/package/ai-world-sdk)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
-## 简介
+## 特性
 
-`ai-world-sdk` 是一个功能完整的 TypeScript SDK，用于与 AI World 平台交互。它提供了：
-
-- **聊天模型**: 兼容 LangChain.js 接口的聊天模型（OpenAI、Google Gemini、Anthropic Claude、Doubao）
-- **图像生成**: 支持豆包 Seedream 模型的图像生成 API
-- **视频生成**: 支持豆包 Seedance 模型的视频生成 API
-- **全局配置**: 支持全局 baseUrl 和 token 配置，简化客户端初始化
-
-设计灵感来自 [LangChain.js](https://github.com/langchain-ai/langchainjs) 的接口风格。
+- 🤖 **聊天模型**: 兼容 LangChain.js 接口（OpenAI、Gemini、Claude、Doubao）
+- 🎨 **图像生成**: 支持豆包 Seedream 和 Google Gemini
+- 🎬 **视频生成**: 支持豆包 Seedance
+- ⚙️ **全局配置**: 自动从浏览器环境获取配置，简化初始化
+- 🐛 **调试模式**: 支持详细的请求/响应日志
 
 ## 安装
 
 ```bash
 npm install ai-world-sdk
-# 或
-yarn add ai-world-sdk
-# 或
-pnpm add ai-world-sdk
+# 或 yarn add ai-world-sdk
+# 或 pnpm add ai-world-sdk
 ```
 
 ## 快速开始
 
-### 1. 全局配置（推荐）
+### 1. 配置（推荐）
 
 ```typescript
-import { sdkConfig, ChatGoogleGenerativeAI, ImageGenerationClient, VideoGenerationClient } from 'ai-world-sdk';
+import { sdkConfig } from 'ai-world-sdk';
 
-// 设置全局配置（只需设置一次）
+// 设置全局配置（只需一次）
 sdkConfig.setBaseUrl('http://localhost:8000');
 sdkConfig.setToken('your-jwt-token');
+sdkConfig.setDebug(true); // 可选：启用调试模式
 
-// 现在可以创建客户端而不需要提供 baseUrl 和 token
-const chatModel = new ChatGoogleGenerativeAI({
-  modelName: 'gemini-2.0-flash-exp-image-generation',
-  temperature: 0.7,
-});
-
-const imageClient = new ImageGenerationClient({});
-const videoClient = new VideoGenerationClient({});
+// 浏览器环境会自动从 window.location.origin 和 cookie 获取配置
 ```
 
-### 2. 基础聊天模型使用
+### 2. 聊天模型
 
 ```typescript
-import {
-  ChatGoogleGenerativeAI,
-  HumanMessage,
-  SystemMessage,
-  AIMessageChunk,
-} from 'ai-world-sdk';
-
-// 初始化聊天模型
-const model = new ChatGoogleGenerativeAI({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-jwt-token',
-  modelName: 'gemini-2.0-flash-exp-image-generation',
+import { ChatGoogleGenerativeAI, ChatOpenAI, ChatAnthropic, HumanMessage, SystemMessage } from 'ai-world-sdk';
+
+// Gemini 模型（使用 gemini provider）
+const geminiModel = new ChatGoogleGenerativeAI({
+  modelName: 'gemini-2.5-flash-image',
+  temperature: 0.7,
+  provider: 'gemini', // 或 'aihubmix'
+});
+
+// GPT 模型（使用 aihubmix provider）
+const gptModel = new ChatOpenAI({
+  modelName: 'gpt-4o-mini',
   temperature: 0.7,
+  provider: 'aihubmix',
+});
+
+// Claude 模型（使用 aihubmix provider）
+const claudeModel = new ChatAnthropic({
+  modelName: 'claude-3-sonnet-20240229',
+  temperature: 0.7,
+  provider: 'aihubmix',
 });
 
 // 非流式调用
-const response = await model.invoke([
+const response = await geminiModel.invoke([
   new SystemMessage('You are a helpful assistant.'),
   new HumanMessage('Hello! What is AI?'),
 ]);
-
-console.log(response.content); // 输出: AI 是...
+console.log(response.content);
 
 // 流式调用
-for await (const chunk of model.stream([
-  new HumanMessage('Tell me a story about AI'),
-])) {
-  // chunk 是 AIMessageChunk 对象
-  if (typeof chunk.content === 'string') {
-    process.stdout.write(chunk.content);
-  }
-}
-```
-
-## 完整 API 文档
-
-### 全局配置 API
-
-#### `sdkConfig.setBaseUrl(baseUrl: string): void`
-
-设置全局 baseUrl。所有客户端实例如果没有提供 baseUrl，将使用此全局值。
-
-```typescript
-import { sdkConfig } from 'ai-world-sdk';
-
-sdkConfig.setBaseUrl('http://localhost:8000');
-```
-
-#### `sdkConfig.setToken(token: string): void`
-
-设置全局 token。所有客户端实例如果没有提供 token，将使用此全局值。
-
-```typescript
-sdkConfig.setToken('your-jwt-token');
-```
-
-#### `sdkConfig.setHeaders(headers: Record<string, string>): void`
-
-设置全局 headers。这些 headers 会与每个客户端实例的 headers 合并。
-
-```typescript
-sdkConfig.setHeaders({
-  'X-Custom-Header': 'value',
-});
-```
-
-#### `sdkConfig.getBaseUrl(): string | null`
-
-获取全局 baseUrl。
-
-#### `sdkConfig.getToken(): string | null`
-
-获取全局 token。
-
-#### `sdkConfig.getHeaders(): Record<string, string>`
-
-获取全局 headers。
-
-#### `sdkConfig.reset(): void`
-
-重置所有全局配置。
-
-### 聊天模型 API
-
-#### 支持的模型类
-
-- `ChatOpenAI` - OpenAI 兼容模型（GPT、O1、Doubao）
-- `ChatGoogleGenerativeAI` - Google Gemini 模型
-- `ChatAnthropic` - Anthropic Claude 模型
-
-#### `BaseChatModel` 接口
-
-所有聊天模型都继承自 `BaseChatModel`，提供以下方法：
-
-##### `invoke(messages: BaseMessage[]): Promise<AIMessage>`
-
-发送非流式请求。
-
-**参数:**
-- `messages`: 消息数组，可以是 `HumanMessage`、`SystemMessage`、`AIMessage`
-
-**返回:** `AIMessage` 对象，包含 `content` 字段
-
-**示例:**
-```typescript
-const response = await model.invoke([
-  new HumanMessage('What is the capital of France?'),
-]);
-console.log(response.content); // "The capital of France is Paris."
-```
-
-##### `stream(messages: BaseMessage[]): AsyncGenerator<AIMessageChunk, void, unknown>`
-
-发送流式请求。
-
-**参数:**
-- `messages`: 消息数组
-
-**返回:** 异步生成器，每次 yield 一个 `AIMessageChunk` 对象
-
-**示例:**
-```typescript
-for await (const chunk of model.stream([
+for await (const chunk of geminiModel.stream([
   new HumanMessage('Tell me a story'),
 ])) {
   if (typeof chunk.content === 'string') {
@@ -181,513 +79,387 @@ for await (const chunk of model.stream([
 }
 ```
 
-##### `batch(messagesList: BaseMessage[][]): Promise<AIMessage[]>`
-
-批量处理多个消息集。
+### 3. 图像生成
 
-**参数:**
-- `messagesList`: 消息数组的数组
-
-**返回:** `AIMessage` 数组
-
-**示例:**
 ```typescript
-const responses = await model.batch([
-  [new HumanMessage('What is AI?')],
-  [new HumanMessage('What is ML?')],
-  [new HumanMessage('What is DL?')],
-]);
+import { DoubaoImageGenerationClient, GeminiImageGenerationClient } from 'ai-world-sdk';
 
-responses.forEach((response, index) => {
-  console.log(`Response ${index + 1}:`, response.content);
+// 豆包图像生成
+const doubaoClient = new DoubaoImageGenerationClient({});
+const result = await doubaoClient.generate({
+  prompt: 'A beautiful sunset over the ocean',
+  size: '2K',
+  quality: 'hd',
+  n: 1,
 });
-```
-
-##### `bind(options: BindOptions): BaseChatModel`
-
-绑定参数到模型实例，返回新的模型实例。
-
-**参数:**
-- `options.temperature`: 温度参数
-- `options.maxTokens`: 最大 token 数
-- `options.topP`: Top-p 采样参数
-- `options.tools`: 工具定义数组
-- `options.toolChoice`: 工具选择策略
+console.log('图像 URL:', result.data[0]?.url);
 
-**示例:**
-```typescript
-const boundModel = model.bind({
-  temperature: 0.9,
-  maxTokens: 1000,
+// Gemini 图像生成
+const geminiClient = new GeminiImageGenerationClient({});
+const geminiResult = await geminiClient.generate({
+  prompt: 'A futuristic city',
+  aspect_ratio: '16:9',
 });
-
-const response = await boundModel.invoke([
-  new HumanMessage('Hello!'),
-]);
+console.log('图像 URL:', geminiResult.data[0]?.url);
 ```
 
-##### `bindTools(tools: ToolDefinition[]): BaseChatModel`
+### 4. 视频生成
 
-绑定工具到模型实例。
-
-**参数:**
-- `tools`: 工具定义数组
-
-**示例:**
 ```typescript
-import { ToolDefinition } from 'ai-world-sdk';
-
-const tools: ToolDefinition[] = [
-  {
-    type: 'function',
-    function: {
-      name: 'get_weather',
-      description: 'Get the current weather in a location',
-      parameters: {
-        type: 'object',
-        properties: {
-          location: { type: 'string' },
-        },
-        required: ['location'],
-      },
-    },
-  },
-];
-
-const modelWithTools = model.bindTools(tools);
-const response = await modelWithTools.invoke([
-  new HumanMessage('What is the weather in Paris?'),
-]);
-```
-
-##### `getModelName(): string`
-
-获取当前模型名称。
-
-#### 消息类型
-
-##### `HumanMessage`
-
-用户消息。
-
-```typescript
-import { HumanMessage } from 'ai-world-sdk';
-
-const message = new HumanMessage('Hello!');
-```
+import { VideoGenerationClient } from 'ai-world-sdk';
 
-##### `SystemMessage`
+const videoClient = new VideoGenerationClient({});
 
-系统消息，用于设置助手的行为。
+// 创建任务
+const task = await videoClient.create({
+  prompt: 'A beautiful sunset over the ocean',
+  duration: 5,
+  aspect_ratio: '16:9',
+});
 
-```typescript
-import { SystemMessage } from 'ai-world-sdk';
+// 轮询直到完成
+const result = await videoClient.poll(task.id, {
+  interval: 3000,
+  timeout: 300000,
+});
 
-const message = new SystemMessage('You are a helpful assistant.');
+if (result.status === 'succeeded') {
+  console.log('视频 URL:', result.content?.video_url);
+}
 ```
 
-##### `AIMessage`
-
-助手消息。
+## API 参考
 
-```typescript
-import { AIMessage } from 'ai-world-sdk';
+### 聊天模型
 
-const message = new AIMessage('Hello! How can I help you?');
-```
+#### 支持的模型类
 
-##### `AIMessageChunk`
+| 模型类 | 说明 | 示例模型 |
+|--------|------|----------|
+| `ChatOpenAI` | OpenAI 兼容（GPT、O1、Doubao） | `gpt-4o-mini`, `doubao-pro-4k` |
+| `ChatGoogleGenerativeAI` | Google Gemini | `gemini-2.5-flash-image` |
+| `ChatAnthropic` | Anthropic Claude | `claude-3-sonnet-20240229` |
 
-流式响应中的消息块。
+#### 核心方法
 
 ```typescript
-import { AIMessageChunk } from 'ai-world-sdk';
+// 非流式调用
+const response = await model.invoke([new HumanMessage('Hello')]);
 
-// 在流式调用中使用
-for await (const chunk of model.stream([...])) {
-  // chunk 是 AIMessageChunk 实例
+// 流式调用
+for await (const chunk of model.stream([new HumanMessage('Hello')])) {
   console.log(chunk.content);
-  console.log(chunk.id);
-  console.log(chunk.response_metadata);
 }
-```
 
-#### 多模态输入
+// 批量处理
+const responses = await model.batch([
+  [new HumanMessage('Question 1')],
+  [new HumanMessage('Question 2')],
+]);
 
-支持文本和图像混合输入：
+// 绑定参数
+const boundModel = model.bind({ temperature: 0.9, maxTokens: 1000 });
 
-```typescript
-const response = await model.invoke([
-  {
-    role: 'user',
-    content: [
-      { type: 'text', text: 'Describe this image' },
-      {
-        type: 'image_url',
-        image_url: 'data:image/jpeg;base64,/9j/4AAQSkZJRg...',
-      },
-    ],
-  },
-]);
+// 绑定工具
+const modelWithTools = model.bindTools([...tools]);
 ```
 
 #### 工厂函数
 
-使用 `createChatModel` 根据模型名称自动选择正确的模型类：
-
 ```typescript
 import { createChatModel } from 'ai-world-sdk';
 
-// 自动选择 ChatGoogleGenerativeAI
-const gemini = createChatModel('gemini-1.5-pro', {
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
+// 根据模型名称自动选择正确的类
+const model = createChatModel('gemini-2.5-flash-image', {
   temperature: 0.7,
-});
-
-// 自动选择 ChatOpenAI
-const gpt = createChatModel('gpt-4', {
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-});
-
-// 自动选择 ChatAnthropic
-const claude = createChatModel('claude-3-sonnet-20240229', {
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
+  provider: 'gemini', // 或 'aihubmix', 'doubao'
 });
 ```
 
 **支持的模型前缀:**
 - `gpt-*`, `o1-*` → `ChatOpenAI`
-- `doubao-*` → `ChatOpenAI` (Doubao 与 OpenAI 兼容)
+- `doubao-*` → `ChatOpenAI`
 - `gemini-*` → `ChatGoogleGenerativeAI`
 - `claude-*` → `ChatAnthropic`
 
-### 图像生成 API
-
-#### `ImageGenerationClient`
+**Provider 参数说明:**
+- `provider: 'aihubmix'` - 使用 aihubmix 聚合服务（推荐，支持所有模型）
+- `provider: 'gemini'` - 直接使用 Google Gemini API
+- `provider: 'doubao'` - 使用豆包服务
 
-图像生成客户端，支持 OpenAI 标准格式的图像生成。
+### 图像生成
 
-##### 初始化
+#### DoubaoImageGenerationClient
 
 ```typescript
-import { ImageGenerationClient } from 'ai-world-sdk';
+const client = new DoubaoImageGenerationClient({});
 
-const imageClient = new ImageGenerationClient({
-  baseUrl: 'http://localhost:8000', // 可选，如果设置了全局配置
-  token: 'your-jwt-token', // 可选，如果设置了全局配置
-  headers: { // 可选
-    'X-Custom-Header': 'value',
-  },
+const result = await client.generate({
+  prompt: 'A beautiful landscape',      // 必需
+  model: 'doubao-seedream-4-5-251128', // 可选，默认值
+  size: '2K',                          // 可选: 1K, 2K, 4K, 1024x1024 等
+  quality: 'hd',                       // 可选: standard, hd
+  n: 1,                                // 可选: 1-10
+  response_format: 'url',              // 可选: url, b64_json
+  style: 'vivid',                      // 可选: vivid, natural
+  watermark: false,                     // 可选
 });
 ```
 
-##### `generate(request: ImageGenerationRequest): Promise<ImageGenerationResponse>`
-
-生成图像（同步）。
-
-**请求参数 (`ImageGenerationRequest`):**
-- `prompt` (必需): 图像生成提示词
-- `model` (可选): 模型名称，默认: `doubao-seedream-4-5-251128`
-- `n` (可选): 生成图像数量，1-10，默认: 1
-- `size` (可选): 图像尺寸，支持: `256x256`, `512x512`, `1024x1024`, `1792x1024`, `1024x1792`, `1K`, `2K`, `4K`，默认: `1024x1024`
-- `quality` (可选): 图像质量，`standard` 或 `hd`，默认: `standard`
-- `response_format` (可选): 响应格式，`url` 或 `b64_json`，默认: `url`
-- `style` (可选): 图像风格，`vivid` 或 `natural`（仅部分模型支持）
-- `user` (可选): 用户标识（用于追踪和审计）
+#### GeminiImageGenerationClient
 
-**返回 (`ImageGenerationResponse`):**
-- `created`: 创建时间戳（Unix 时间戳）
-- `data`: 图像数据数组，每个元素包含:
-  - `url`: 图像 URL（如果 `response_format` 为 `url`）
-  - `b64_json`: Base64 编码的图像数据（如果 `response_format` 为 `b64_json`）
-
-**示例:**
 ```typescript
-const result = await imageClient.generate({
-  prompt: '充满活力的特写编辑肖像，模特眼神犀利，头戴雕塑感帽子',
-  model: 'doubao-seedream-4-5-251128',
-  size: '2K',
-  n: 1,
-  quality: 'hd',
-  response_format: 'url',
-});
+const client = new GeminiImageGenerationClient({});
 
-console.log('生成的图像:', result.data);
-console.log('第一张图像 URL:', result.data[0]?.url);
+const result = await client.generate({
+  prompt: 'A beautiful landscape',              // 必需
+  model: 'gemini-2.0-flash-exp-image-generation', // 可选，默认值
+  number_of_images: 1,                          // 可选
+  aspect_ratio: '16:9',                         // 可选: 1:1, 9:16, 16:9, 4:3, 3:4
+  temperature: 0.7,                             // 可选
+});
 ```
 
-### 视频生成 API
+### 视频生成
 
-#### `VideoGenerationClient`
+#### VideoGenerationClient
 
-视频生成客户端，支持豆包 Seedance 模型的视频生成。
+```typescript
+const client = new VideoGenerationClient({});
 
-##### 初始化
+// 创建任务
+const task = await client.create({
+  prompt: 'A beautiful sunset',        // 或 image_url
+  model: 'doubao-seedance-1-0-pro-fast-251015', // 可选
+  duration: 5,                         // 可选: 1-10 秒
+  aspect_ratio: '16:9',                // 可选: 16:9, 9:16, 1:1
+  resolution: '720p',                   // 可选
+  return_last_frame: true,             // 可选
+});
 
-```typescript
-import { VideoGenerationClient } from 'ai-world-sdk';
+// 查询状态
+const status = await client.get(task.id);
 
-const videoClient = new VideoGenerationClient({
-  baseUrl: 'http://localhost:8000', // 可选，如果设置了全局配置
-  token: 'your-jwt-token', // 可选，如果设置了全局配置
-  headers: { // 可选
-    'X-Custom-Header': 'value',
-  },
+// 轮询直到完成
+const result = await client.poll(task.id, {
+  interval: 3000,   // 轮询间隔（毫秒）
+  timeout: 300000,  // 超时时间（毫秒）
 });
 ```
 
-##### `create(request: VideoGenerationRequest): Promise<ContentGenerationTaskID>`
+### 全局配置
 
-创建视频生成任务。
+```typescript
+import { sdkConfig } from 'ai-world-sdk';
 
-**请求参数 (`VideoGenerationRequest`):**
-- `prompt` (可选): 文本提示词（文本生成视频）
-- `image_url` (可选): 图像 URL（图像生成视频）
-- `model` (可选): 模型名称，默认: `doubao-seedance-1-0-pro-fast-251015`
-- `content` (可选): 内容列表（如果提供，将直接使用，忽略 prompt 和 image_url）
-- `callback_url` (可选): 回调 URL（任务完成时通知）
-- `return_last_frame` (可选): 是否返回最后一帧图像
-- `service_tier` (可选): 服务层级
-- `execution_expires_after` (可选): 执行过期时间（秒）
-- `duration` (可选): 视频时长（秒），1-10，默认: 5（将添加到 prompt）
-- `aspect_ratio` (可选): 视频宽高比，`16:9`, `9:16`, `1:1`，默认: `16:9`（将添加到 prompt）
-- `resolution` (可选): 视频分辨率，例如: `720p`, `1080p`（将添加到 prompt）
-- `user` (可选): 用户标识（用于追踪和审计）
+// 设置配置
+sdkConfig.setBaseUrl('http://localhost:8000');
+sdkConfig.setToken('your-jwt-token');
+sdkConfig.setHeaders({ 'X-Custom-Header': 'value' });
+sdkConfig.setDebug(true);
 
-**注意:** `prompt` 和 `image_url` 必须提供其中之一，或者直接提供 `content`。
+// 获取配置
+const baseUrl = sdkConfig.getServerUrl();
+const token = sdkConfig.getToken();
+const headers = sdkConfig.getHeaders();
+const debug = sdkConfig.getDebug();
 
-**返回 (`ContentGenerationTaskID`):**
-- `id`: 任务 ID
+// 重置配置
+sdkConfig.reset();
+```
 
-**示例（文本生成视频）:**
-```typescript
-const task = await videoClient.create({
-  prompt: 'A beautiful sunset over the ocean with waves crashing on the shore',
-  model: 'doubao-seedance-1-0-pro-fast-251015',
-  duration: 5,
-  aspect_ratio: '16:9',
-  resolution: '720p',
-});
+**浏览器环境自动配置:**
+- `baseUrl`: 从 `window.location.origin` 自动获取
+- `token`: 从 `document.cookie` 中的 `auth_token` 自动获取
 
-console.log('任务 ID:', task.id);
-```
+### 消息类型
 
-**示例（图像生成视频）:**
 ```typescript
-const task = await videoClient.create({
-  image_url: 'https://example.com/image.jpg',
-  model: 'doubao-seedance-1-0-pro-fast-251015',
-  duration: 5,
-});
+import { HumanMessage, SystemMessage, AIMessage } from 'ai-world-sdk';
 
-console.log('任务 ID:', task.id);
-```
+// 文本消息
+const userMsg = new HumanMessage('Hello');
+const systemMsg = new SystemMessage('You are a helpful assistant.');
 
-**示例（直接提供 content）:**
-```typescript
-const task = await videoClient.create({
-  model: 'doubao-seedance-1-0-pro-fast-251015',
-  content: [
-    {
-      type: 'text',
-      text: 'A beautiful sunset --ratio 16:9 --duration 5',
-    },
-  ],
-  return_last_frame: true,
-  callback_url: 'https://example.com/callback',
-});
+// 多模态消息（文本 + 图像）
+const multimodalMsg = new HumanMessage([
+  { type: 'text', text: 'Describe this image' },
+  { type: 'image_url', image_url: 'data:image/jpeg;base64,...' },
+]);
 ```
 
-##### `get(taskId: string): Promise<ContentGenerationTask>`
+## Provider 调用示例
 
-查询视频生成任务状态。
+SDK 支持多个 provider，每个 provider 对应不同的服务提供商。以下是各 provider 的使用示例：
 
-**参数:**
-- `taskId`: 任务 ID
+### aihubmix Provider
 
-**返回 (`ContentGenerationTask`):**
-- `id`: 任务 ID
-- `model`: 使用的模型
-- `status`: 任务状态，`queued` | `running` | `succeeded` | `failed` | `cancelled`
-- `error`: 错误信息（如果失败），包含 `message` 和 `code`
-- `content`: 生成的内容，包含:
-  - `video_url`: 视频 URL（任务完成时）
-  - `last_frame_url`: 最后一帧图像 URL（如果 `return_last_frame` 为 true）
-  - `file_url`: 文件 URL
-- `usage`: 使用情况，包含 `completion_tokens`
-- `created_at`: 创建时间戳（Unix 时间戳）
-- `updated_at`: 更新时间戳
-- 其他字段: `subdivisionlevel`, `fileformat`, `frames`, `framespersecond`, `seed`, `revised_prompt`, `service_tier`, `execution_expires_after`
+`aihubmix` 是一个多模型聚合服务，支持多种模型（OpenAI、Gemini、Claude 等）。
 
-**示例:**
 ```typescript
-const task = await videoClient.get(taskId);
+import { ChatOpenAI, ChatGoogleGenerativeAI, ChatAnthropic, HumanMessage } from 'ai-world-sdk';
 
-console.log('任务状态:', task.status);
-console.log('使用的模型:', task.model);
+// OpenAI 模型（通过 aihubmix）
+const gptModel = new ChatOpenAI({
+  modelName: 'gpt-4o-mini',
+  temperature: 0.7,
+  provider: 'aihubmix',
+});
 
-if (task.status === 'succeeded' && task.content?.video_url) {
-  console.log('视频 URL:', task.content.video_url);
-} else if (task.error) {
-  console.error('错误:', task.error.message);
-}
-```
+// Gemini 模型（通过 aihubmix）
+const geminiModel = new ChatGoogleGenerativeAI({
+  modelName: 'gemini-2.5-flash-image',
+  temperature: 0.7,
+  provider: 'aihubmix',
+});
 
-##### `poll(taskId: string, options?: PollOptions): Promise<ContentGenerationTask>`
+// Claude 模型（通过 aihubmix）
+const claudeModel = new ChatAnthropic({
+  modelName: 'claude-3-sonnet-20240229',
+  temperature: 0.7,
+  provider: 'aihubmix',
+});
 
-轮询视频生成任务直到完成。
+// 使用示例
+const response = await gptModel.invoke([
+  new HumanMessage('Hello!'),
+]);
+```
 
-**参数:**
-- `taskId`: 任务 ID
-- `options.interval` (可选): 轮询间隔（毫秒），默认: 5000
-- `options.timeout` (可选): 超时时间（毫秒），默认: 300000 (5分钟)
+### gemini Provider
 
-**返回:** 完成后的 `ContentGenerationTask`
+直接使用 Google Gemini 官方 API。
 
-**示例:**
 ```typescript
-// 创建任务
-const createTask = await videoClient.create({
-  prompt: 'A short video of a cat playing',
-  model: 'doubao-seedance-1-0-pro-fast-251015',
-  duration: 3,
-});
+import { ChatGoogleGenerativeAI, HumanMessage } from 'ai-world-sdk';
 
-// 轮询直到完成
-const result = await videoClient.poll(createTask.id, {
-  interval: 2000, // 每 2 秒轮询一次
-  timeout: 60000, // 60 秒超时
+const model = new ChatGoogleGenerativeAI({
+  modelName: 'gemini-2.5-flash-image',
+  temperature: 0.7,
+  provider: 'gemini',
 });
 
-if (result.status === 'succeeded' && result.content?.video_url) {
-  console.log('视频生成成功:', result.content.video_url);
-} else if (result.error) {
-  console.error('生成失败:', result.error.message);
-}
+const response = await model.invoke([
+  new HumanMessage('What is AI?'),
+]);
 ```
 
-## 支持的模型
+### doubao Provider
 
-### 聊天模型
+使用豆包（字节跳动）的模型服务。
 
-#### OpenAI 兼容模型
-- `gpt-3.5-turbo`
-- `gpt-4`
-- `gpt-4-turbo`
-- `o1-preview`
-- `o1-mini`
-- `doubao-*` (豆包模型，与 OpenAI 兼容)
-
-#### Google Gemini
-- `gemini-pro`
-- `gemini-1.5-pro`
-- `gemini-1.5-flash`
-- `gemini-2.0-flash-exp-image-generation`
-
-#### Anthropic Claude
-- `claude-3-opus-20240229`
-- `claude-3-sonnet-20240229`
-- `claude-3-haiku-20240307`
+```typescript
+import { ChatOpenAI, HumanMessage } from 'ai-world-sdk';
 
-### 图像生成模型
+// Doubao 模型使用 ChatOpenAI（因为与 OpenAI 兼容）
+const doubaoModel = new ChatOpenAI({
+  modelName: 'doubao-pro-4k',
+  temperature: 0.7,
+  provider: 'doubao',
+});
 
-- `doubao-seedream-4-5-251128` (默认)
-- 其他 Seedream 模型版本
+const response = await doubaoModel.invoke([
+  new HumanMessage('你好！'),
+]);
+```
 
-### 视频生成模型
+### Provider 选择指南
+
+| Provider | 适用场景 | 支持的模型 |
+|----------|----------|------------|
+| `aihubmix` | 多模型聚合，统一接口 | GPT、Gemini、Claude、Doubao 等 |
+| `gemini` | 直接使用 Google Gemini API | 所有 Gemini 模型 |
+| `doubao` | 使用豆包服务 | Doubao 系列模型 |
 
-- `doubao-seedance-1-0-pro-fast-251015` (默认，推荐)
-- `doubao-seedance-1-0-pro-250528`
-- `doubao-seedance-1-0-lite-t2v-250428` (文本生成视频)
-- `doubao-seedance-1-0-lite-i2v-250428` (图像生成视频)
+**推荐使用 `aihubmix` provider**，因为它提供了统一的接口和更好的模型选择灵活性。
 
 ## 完整示例
 
-### 示例 1: 使用全局配置
+### 聊天对话
 
 ```typescript
-import {
-  sdkConfig,
-  ChatGoogleGenerativeAI,
-  ImageGenerationClient,
-  VideoGenerationClient,
-  HumanMessage,
-} from 'ai-world-sdk';
-
-// 设置全局配置
-sdkConfig.setBaseUrl('http://localhost:8000');
-sdkConfig.setToken('your-jwt-token');
+import { ChatGoogleGenerativeAI, HumanMessage, SystemMessage } from 'ai-world-sdk';
 
-// 创建客户端（不需要提供 baseUrl 和 token）
-const chatModel = new ChatGoogleGenerativeAI({
-  modelName: 'gemini-2.0-flash-exp-image-generation',
+const model = new ChatGoogleGenerativeAI({
+  modelName: 'gemini-2.5-flash-image',
   temperature: 0.7,
+  provider: 'gemini',
 });
 
-const imageClient = new ImageGenerationClient({});
-const videoClient = new VideoGenerationClient({});
-
-// 使用
-const response = await chatModel.invoke([
-  new HumanMessage('Hello!'),
+// 简单对话
+const response = await model.invoke([
+  new HumanMessage('What is AI?'),
 ]);
+console.log(response.content);
 
-const imageResult = await imageClient.generate({
-  prompt: 'A beautiful sunset',
-});
+// 带系统提示的对话
+const response2 = await model.invoke([
+  new SystemMessage('You are a helpful assistant.'),
+  new HumanMessage('Explain machine learning.'),
+]);
 ```
 
-### 示例 2: 流式聊天
+### 流式响应
 
 ```typescript
-import { ChatGoogleGenerativeAI, HumanMessage, AIMessageChunk } from 'ai-world-sdk';
-
-const model = new ChatGoogleGenerativeAI({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-  modelName: 'gemini-2.0-flash-exp-image-generation',
-});
-
 let fullText = '';
 for await (const chunk of model.stream([
   new HumanMessage('Tell me a story about AI'),
 ])) {
-  // chunk 是 AIMessageChunk 对象
   if (typeof chunk.content === 'string') {
     fullText += chunk.content;
     process.stdout.write(chunk.content);
   }
 }
-
 console.log('\n完整回复:', fullText);
 ```
 
-### 示例 3: 图像生成工作流
+### 工具调用
 
 ```typescript
-import { ImageGenerationClient } from 'ai-world-sdk';
+import { ToolDefinition } from 'ai-world-sdk';
 
-const imageClient = new ImageGenerationClient({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-});
+const tools: ToolDefinition[] = [
+  {
+    type: 'function',
+    function: {
+      name: 'get_weather',
+      description: 'Get the current weather',
+      parameters: {
+        type: 'object',
+        properties: {
+          location: { type: 'string' },
+        },
+        required: ['location'],
+      },
+    },
+  },
+];
+
+const modelWithTools = model.bindTools(tools);
+const response = await modelWithTools.invoke([
+  new HumanMessage('What is the weather in Paris?'),
+]);
+```
+
+### 图像生成工作流
+
+```typescript
+import { DoubaoImageGenerationClient } from 'ai-world-sdk';
+
+const client = new DoubaoImageGenerationClient({});
 
 // 生成单张图像
-const result = await imageClient.generate({
+const result = await client.generate({
   prompt: 'A futuristic city skyline at sunset',
-  model: 'doubao-seedream-4-5-251128',
   size: '2K',
   quality: 'hd',
-  n: 1,
 });
 
 console.log('图像 URL:', result.data[0]?.url);
 
 // 生成多张图像
-const multiResult = await imageClient.generate({
+const multiResult = await client.generate({
   prompt: 'A beautiful landscape',
   n: 3,
   size: '2K',
@@ -698,39 +470,28 @@ multiResult.data.forEach((image, index) => {
 });
 ```
 
-### 示例 4: 视频生成工作流
+### 视频生成工作流
 
 ```typescript
 import { VideoGenerationClient } from 'ai-world-sdk';
 
-const videoClient = new VideoGenerationClient({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-});
+const client = new VideoGenerationClient({});
 
-// 1. 创建视频生成任务
-const createTask = await videoClient.create({
-  prompt: 'A beautiful sunset over the ocean with waves',
-  model: 'doubao-seedance-1-0-pro-fast-251015',
+// 1. 创建任务
+const task = await client.create({
+  prompt: 'A beautiful sunset over the ocean',
   duration: 5,
   aspect_ratio: '16:9',
   resolution: '720p',
 });
 
-console.log('任务已创建，ID:', createTask.id);
-
-// 2. 查询任务状态
-const task = await videoClient.get(createTask.id);
-console.log('当前状态:', task.status);
-
-// 3. 轮询直到完成
-const result = await videoClient.poll(createTask.id, {
-  interval: 3000, // 每 3 秒查询一次
-  timeout: 300000, // 5 分钟超时
+// 2. 轮询直到完成
+const result = await client.poll(task.id, {
+  interval: 3000,
+  timeout: 300000,
 });
 
 if (result.status === 'succeeded') {
-  console.log('视频生成成功!');
   console.log('视频 URL:', result.content?.video_url);
   if (result.content?.last_frame_url) {
     console.log('最后一帧:', result.content.last_frame_url);
@@ -740,147 +501,33 @@ if (result.status === 'succeeded') {
 }
 ```
 
-### 示例 5: 批量处理
-
-```typescript
-import { ChatGoogleGenerativeAI, HumanMessage } from 'ai-world-sdk';
-
-const model = new ChatGoogleGenerativeAI({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-  modelName: 'gemini-2.0-flash-exp-image-generation',
-});
-
-// 批量处理多个问题
-const questions = [
-  'What is artificial intelligence?',
-  'What is machine learning?',
-  'What is deep learning?',
-];
-
-const responses = await model.batch(
-  questions.map(q => [new HumanMessage(q)])
-);
-
-responses.forEach((response, index) => {
-  console.log(`问题 ${index + 1}:`, questions[index]);
-  console.log(`回答:`, response.content);
-  console.log('---');
-});
-```
-
-### 示例 6: 工具调用
-
-```typescript
-import {
-  ChatGoogleGenerativeAI,
-  HumanMessage,
-  ToolDefinition,
-} from 'ai-world-sdk';
-
-const model = new ChatGoogleGenerativeAI({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-  modelName: 'gemini-2.0-flash-exp-image-generation',
-});
-
-const tools: ToolDefinition[] = [
-  {
-    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',
-          },
-          unit: {
-            type: 'string',
-            enum: ['celsius', 'fahrenheit'],
-            description: 'The unit of temperature',
-          },
-        },
-        required: ['location'],
-      },
-    },
-  },
-];
-
-const modelWithTools = model.bindTools(tools);
-
-const response = await modelWithTools.invoke([
-  new HumanMessage('What is the weather in Paris?'),
-]);
+## 支持的模型
 
-console.log('响应:', response.content);
-// 检查是否有工具调用
-if (response.tool_calls && response.tool_calls.length > 0) {
-  console.log('工具调用:', response.tool_calls);
-}
-```
+### 聊天模型
 
-## 类型定义
+| 提供商 | 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` |
+| Anthropic Claude | `aihubmix` | `claude-3-sonnet-20240229`, `claude-3-opus-20240229` | `ChatAnthropic` |
+| Doubao | `doubao` 或 `aihubmix` | `doubao-pro-4k`, `doubao-seedream-4-5-251128` | `ChatOpenAI` |
 
-### 核心类型
+**注意:** 
+- 使用 `aihubmix` provider 可以访问所有模型，推荐用于多模型场景
+- 使用特定 provider（如 `gemini`、`doubao`）会直接调用对应的官方 API
 
-```typescript
-// 消息类型
-export class HumanMessage extends BaseMessage { ... }
-export class SystemMessage extends BaseMessage { ... }
-export class AIMessage extends BaseMessage { ... }
-export class AIMessageChunk extends BaseMessage { ... }
-
-// 配置类型
-export interface BaseChatModelParams {
-  temperature?: number;
-  maxTokens?: number;
-  topP?: number;
-  modelName?: string;
-  provider?: string;
-  apiKey?: string;
-}
+### 图像生成模型
 
-export interface LangchainClientConfig {
-  baseUrl?: string;
-  token?: string;
-  headers?: Record<string, string>;
-}
+- **豆包 Seedream**: `doubao-seedream-4-5-251128` (默认)
+- **Google Gemini**: `gemini-2.0-flash-exp-image-generation` (默认)
 
-// 工具类型
-export interface ToolDefinition {
-  type: 'function';
-  function: {
-    name: string;
-    description: string;
-    parameters?: Record<string, any>;
-  };
-}
+### 视频生成模型
 
-export interface BindOptions {
-  temperature?: number;
-  maxTokens?: number;
-  topP?: number;
-  tools?: ToolDefinition[];
-  toolChoice?: 'auto' | 'none' | 'required' | { type: 'function'; function: { name: string } };
-}
-```
+- **豆包 Seedance**: `doubao-seedance-1-0-pro-fast-251015` (默认，推荐)
 
 ## 错误处理
 
-所有 API 调用都可能抛出错误。建议使用 try-catch 进行错误处理：
-
 ```typescript
-import { ChatGoogleGenerativeAI, HumanMessage } from 'ai-world-sdk';
-
-const model = new ChatGoogleGenerativeAI({
-  baseUrl: 'http://localhost:8000',
-  token: 'your-token',
-  modelName: 'gemini-2.0-flash-exp-image-generation',
-});
-
 try {
   const response = await model.invoke([
     new HumanMessage('Hello!'),
@@ -894,25 +541,6 @@ try {
 }
 ```
 
-## 参考地址
-
-### 官方文档
-
-- **AI World 平台**: [项目仓库](https://github.com/your-org/ai-world)
-- **LangChain.js**: [https://github.com/langchain-ai/langchainjs](https://github.com/langchain-ai/langchainjs)
-- **火山引擎方舟平台**: [https://www.volcengine.com/docs/82379](https://www.volcengine.com/docs/82379)
-
-### API 文档
-
-- **图像生成 API**: [https://www.volcengine.com/docs/82379/1824121](https://www.volcengine.com/docs/82379/1824121)
-- **视频生成 API**: [https://www.volcengine.com/docs/82379/1366799](https://www.volcengine.com/docs/82379/1366799)
-- **对话 API**: [https://www.volcengine.com/docs/82379/1541595](https://www.volcengine.com/docs/82379/1541595)
-
-### 模型文档
-
-- **Seedream 4.0-4.5 教程**: [https://www.volcengine.com/docs/82379/1824121?lang=zh](https://www.volcengine.com/docs/82379/1824121?lang=zh)
-- **Seedance 提示词指南**: [https://www.volcengine.com/docs/82379/1901652](https://www.volcengine.com/docs/82379/1901652)
-
 ## 开发
 
 ### 构建
@@ -921,121 +549,34 @@ try {
 npm run build
 ```
 
-这会执行以下操作：
-- 清理 `dist/` 目录
-- 编译 TypeScript 源代码到 `dist/` 目录
-- **自动生成 `.d.ts` 类型声明文件**
-
-#### 生成 TypeScript 声明文件 (.d.ts)
-
-项目已配置为自动生成类型声明文件。配置说明：
-
-1. **tsconfig.json 配置**：
-   ```json
-   {
-     "compilerOptions": {
-       "declaration": true,  // 启用声明文件生成
-       "outDir": "./dist",   // 输出目录
-       "rootDir": "./src"    // 源代码目录
-     }
-   }
-   ```
-
-2. **构建命令**：
-   ```bash
-   npm run build
-   # 等同于: rm -rf dist && tsc
-   ```
-
-3. **生成的文件**：
-   运行构建后，`dist/` 目录会包含：
-   - `.js` 文件（编译后的 JavaScript）
-   - `.d.ts` 文件（TypeScript 类型声明）
-   
-   例如：
-   ```
-   dist/
-   ├── index.js          # 编译后的代码
-   ├── index.d.ts        # 类型声明文件
-   ├── base.js
-   ├── base.d.ts
-   └── ...
-   ```
-
-4. **package.json 配置**：
-   ```json
-   {
-     "main": "dist/index.js",
-     "types": "dist/index.d.ts"  // 指定类型声明文件入口
-   }
-   ```
-
-5. **验证声明文件**：
-   ```bash
-   # 检查 dist 目录中的 .d.ts 文件
-   ls -la dist/*.d.ts
-   
-   # 或者查看特定文件的声明
-   cat dist/index.d.ts
-   ```
-
 ### 测试
 
 ```bash
-# 运行所有测试
 npm test
-
-# 运行特定测试
+# 或运行特定测试
 npm run test:stream
 npm run test:image-generation
 npm run test:video-generation
 ```
 
-### 项目结构
-
-```
-ai-world-sdk/
-├── src/
-│   ├── base.ts              # 基础聊天模型类
-│   ├── config.ts            # 全局配置管理
-│   ├── messages.ts          # 消息类型定义
-│   ├── image_generation.ts  # 图像生成客户端
-│   ├── video_generation.ts  # 视频生成客户端
-│   ├── chat_models/         # 具体模型实现
-│   │   ├── openai.ts
-│   │   ├── google.ts
-│   │   └── anthropic.ts
-│   └── index.ts             # 主入口文件
-├── __tests__/               # 测试文件
-├── package.json
-├── tsconfig.json
-└── README.md
-```
-
-## 发布
+## 参考链接
 
-### 安装
-
-```bash
-npm install ai-world-sdk
-```
+- [LangChain.js](https://github.com/langchain-ai/langchainjs) - 设计灵感来源
+- [火山引擎方舟平台](https://www.volcengine.com/docs/82379) - 图像/视频生成 API 文档
 
 ## 许可证
 
 MIT
 
-## 贡献
-
-欢迎提交 Issue 和 Pull Request！
-
 ## 更新日志
 
-### 1.0.0
+### 1.0.3
+- ✨ 新增 Google Gemini 图像生成客户端
+- ✨ 全局配置支持自动从浏览器环境获取
+- ✨ 新增调试模式
+- 🔧 改进日志输出
 
+### 1.0.0
 - ✨ 初始版本发布
-- ✨ 支持聊天模型（OpenAI、Gemini、Claude、Doubao）
-- ✨ 支持图像生成（Seedream）
-- ✨ 支持视频生成（Seedance）
-- ✨ 支持全局配置（baseUrl、token）
+- ✨ 支持聊天模型、图像生成、视频生成
 - ✨ 兼容 LangChain.js 接口风格
-- ✨ 完整的 TypeScript 类型定义