ai-world-sdk 1.0.2 → 1.0.5

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",
         });
     }
 }

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

@@ -0,0 +1,36 @@
+/**
+ * Doubao Image Generation Client
+ * 豆包图像生成客户端
+ */
+export interface DoubaoImageGenerationConfig {
+    baseUrl?: string;
+    headers?: Record<string, string>;
+}
+export interface DoubaoImageGenerationRequest {
+    model?: string;
+    prompt: string;
+    size?: string;
+    watermark?: boolean;
+    n?: number;
+    quality?: "standard" | "hd";
+    response_format?: "url" | "b64_json";
+    style?: "vivid" | "natural";
+    user?: string;
+}
+export interface DoubaoImageData {
+    url?: string;
+    b64_json?: string;
+}
+export interface DoubaoImageGenerationResponse {
+    created: number;
+    data: DoubaoImageData[];
+}
+export declare class DoubaoImageGenerationClient {
+    private headers;
+    constructor(config?: DoubaoImageGenerationConfig);
+    /**
+     * Generate images
+     * 生成图像
+     */
+    generate(request: DoubaoImageGenerationRequest): Promise<DoubaoImageGenerationResponse>;
+}

package/dist/{image_generation.js → doubao-image-generation.js}

@@ -1,13 +1,13 @@
 "use strict";
 /**
- * Image Generation Client
- * 图像生成客户端
+ * Doubao Image Generation Client
+ * 豆包图像生成客户端
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImageGenerationClient = void 0;
+exports.DoubaoImageGenerationClient = void 0;
 const config_1 = require("./config");
 const log_1 = require("./log");
-class ImageGenerationClient {
+class DoubaoImageGenerationClient {
     constructor(config) {
         // 合并全局 headers 和配置 headers
         const globalHeaders = config_1.sdkConfig.getHeaders();
@@ -46,7 +46,7 @@ class ImageGenerationClient {
         if (request.user) {
             requestBody.user = request.user;
         }
-        const url = `${config_1.sdkConfig.getServerUrl()}/api/image-proxy/generate`;
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/doubao-image-proxy/generate`;
         (0, log_1.logRequest)("POST", url, this.headers, requestBody);
         const response = await fetch(url, {
             method: "POST",
@@ -56,11 +56,11 @@ class ImageGenerationClient {
         if (!response.ok) {
             const errorText = await response.text();
             (0, log_1.logResponse)(response.status, response.statusText, response.headers, errorText);
-            throw new Error(`Image generation API error: ${response.status} ${errorText}`);
+            throw new Error(`Doubao image generation API error: ${response.status} ${errorText}`);
         }
         const data = (await response.json());
         (0, log_1.logResponse)(response.status, response.statusText, response.headers, data);
         return data;
     }
 }
-exports.ImageGenerationClient = ImageGenerationClient;
+exports.DoubaoImageGenerationClient = DoubaoImageGenerationClient;

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

@@ -0,0 +1,36 @@
+/**
+ * Gemini Image Generation Client
+ * Google Gemini 图像生成客户端
+ */
+export interface GeminiImageGenerationConfig {
+    baseUrl?: string;
+    headers?: Record<string, string>;
+}
+export interface GeminiImageGenerationRequest {
+    prompt: string;
+    model?: string;
+    number_of_images?: number;
+    aspect_ratio?: "1:1" | "9:16" | "16:9" | "4:3" | "3:4";
+    temperature?: number;
+    max_output_tokens?: number;
+    user?: string;
+}
+export interface GeminiImageData {
+    url?: string;
+    b64_json?: string;
+    text?: string;
+}
+export interface GeminiImageGenerationResponse {
+    created: number;
+    data: GeminiImageData[];
+    text?: string;
+}
+export declare class GeminiImageGenerationClient {
+    private headers;
+    constructor(config?: GeminiImageGenerationConfig);
+    /**
+     * Generate images
+     * 生成图像
+     */
+    generate(request: GeminiImageGenerationRequest): Promise<GeminiImageGenerationResponse>;
+}

package/dist/gemini-image-generation.js

@@ -0,0 +1,62 @@
+"use strict";
+/**
+ * Gemini Image Generation Client
+ * Google Gemini 图像生成客户端
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GeminiImageGenerationClient = void 0;
+const config_1 = require("./config");
+const log_1 = require("./log");
+class GeminiImageGenerationClient {
+    constructor(config) {
+        // 合并全局 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 || "",
+            ...globalHeaders,
+            ...config?.headers,
+        };
+    }
+    /**
+     * Generate images
+     * 生成图像
+     */
+    async generate(request) {
+        const requestBody = {
+            prompt: request.prompt,
+            model: request.model || "gemini-2.0-flash-exp-image-generation",
+            number_of_images: request.number_of_images || 1,
+        };
+        // 添加可选参数
+        if (request.aspect_ratio) {
+            requestBody.aspect_ratio = request.aspect_ratio;
+        }
+        if (request.temperature !== undefined) {
+            requestBody.temperature = request.temperature;
+        }
+        if (request.max_output_tokens) {
+            requestBody.max_output_tokens = request.max_output_tokens;
+        }
+        if (request.user) {
+            requestBody.user = request.user;
+        }
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/gemini-image-proxy/generate`;
+        (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 generation 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

@@ -5,7 +5,8 @@
  * @see https://github.com/langchain-ai/langchainjs
  */
 import { BaseChatModel, BaseChatModelParams } from "./base";
-import { ImageGenerationClient, type ImageGenerationConfig, type ImageGenerationRequest, type ImageGenerationResponse } from "./image_generation";
+import { DoubaoImageGenerationClient, 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 { sdkConfig } from "./config";
 export { BaseMessage, HumanMessage, AIMessage, SystemMessage, AIMessageChunk, type MessageContent, type AIMessageChunkData, } from "./messages";
@@ -18,7 +19,9 @@ export interface LangchainClientConfig {
     token?: string;
     headers?: Record<string, string>;
 }
-export { ImageGenerationClient, type ImageGenerationConfig, type ImageGenerationRequest, type ImageGenerationResponse, };
+export { DoubaoImageGenerationClient, type DoubaoImageGenerationConfig, type DoubaoImageGenerationRequest, type DoubaoImageGenerationResponse, };
+export { GeminiImageGenerationClient, type GeminiImageGenerationConfig, type GeminiImageGenerationRequest, type GeminiImageGenerationResponse, };
+export { DoubaoImageGenerationClient as ImageGenerationClient, type DoubaoImageGenerationConfig as ImageGenerationConfig, type DoubaoImageGenerationRequest as ImageGenerationRequest, type DoubaoImageGenerationResponse as ImageGenerationResponse, };
 export { VideoGenerationClient, type VideoGenerationConfig, type VideoGenerationRequest, type ContentGenerationTaskID, type ContentGenerationTask, };
 export { sdkConfig };
 /**

package/dist/index.js

@@ -6,13 +6,15 @@
  * @see https://github.com/langchain-ai/langchainjs
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.sdkConfig = exports.VideoGenerationClient = exports.ImageGenerationClient = exports.ChatAnthropic = exports.ChatGoogleGenerativeAI = exports.ChatOpenAI = exports.BaseChatModel = exports.AIMessageChunk = exports.SystemMessage = exports.AIMessage = exports.HumanMessage = void 0;
+exports.sdkConfig = exports.VideoGenerationClient = exports.ImageGenerationClient = 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");
-const anthropic_1 = require("./chat_models/anthropic");
-const image_generation_1 = require("./image_generation");
-Object.defineProperty(exports, "ImageGenerationClient", { enumerable: true, get: function () { return image_generation_1.ImageGenerationClient; } });
+const doubao_image_generation_1 = require("./doubao-image-generation");
+Object.defineProperty(exports, "DoubaoImageGenerationClient", { enumerable: true, get: function () { return doubao_image_generation_1.DoubaoImageGenerationClient; } });
+Object.defineProperty(exports, "ImageGenerationClient", { enumerable: true, get: function () { return doubao_image_generation_1.DoubaoImageGenerationClient; } });
+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 config_1 = require("./config");
@@ -29,8 +31,8 @@ var openai_2 = require("./chat_models/openai");
 Object.defineProperty(exports, "ChatOpenAI", { enumerable: true, get: function () { return openai_2.ChatOpenAI; } });
 var google_2 = require("./chat_models/google");
 Object.defineProperty(exports, "ChatGoogleGenerativeAI", { enumerable: true, get: function () { return google_2.ChatGoogleGenerativeAI; } });
-var anthropic_2 = require("./chat_models/anthropic");
-Object.defineProperty(exports, "ChatAnthropic", { enumerable: true, get: function () { return anthropic_2.ChatAnthropic; } });
+var anthropic_1 = require("./chat_models/anthropic");
+Object.defineProperty(exports, "ChatAnthropic", { enumerable: true, get: function () { return anthropic_1.ChatAnthropic; } });
 /**
  * Create a chat model instance based on model name
  */
@@ -46,33 +48,19 @@ function createChatModel(model, config) {
             ...config.headers,
         },
     };
-    const modelLower = model.toLowerCase();
-    if (modelLower.startsWith("gpt") || modelLower.startsWith("o1")) {
-        return new openai_1.ChatOpenAI({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else if (modelLower.startsWith("doubao-")) {
-        // Doubao 与 OpenAI 兼容，使用 ChatOpenAI
-        return new openai_1.ChatOpenAI({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else if (modelLower.startsWith("gemini")) {
-        return new google_1.ChatGoogleGenerativeAI({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else if (modelLower.startsWith("claude")) {
-        return new anthropic_1.ChatAnthropic({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else {
-        throw new Error(`Unsupported model: ${model}. Supported models: gpt-*, o1-*, doubao-*, gemini-*, claude-*`);
+    switch (config.provider) {
+        case "doubao":
+        case "aihubmix":
+            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}`);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-world-sdk",
-  "version": "1.0.2",
+  "version": "1.0.5",
   "description": "TypeScript SDK for AI World Platform - Chat Models, Image Generation, and Video Generation",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -9,11 +9,12 @@
     "prepublishOnly": "npm run build",
     "test": "jest",
     "test:watch": "jest --watch",
-    "test:non-stream": "jest -t 'ChatGoogleGenerativeAI - 非流式调用'",
-    "test:stream": "jest -t 'ChatGoogleGenerativeAI - 流式调用'",
+    "test:genimi-non-stream": "jest -t 'ChatGoogleGenerativeAI - 非流式调用'",
+    "test:genimi-stream": "jest -t 'ChatGoogleGenerativeAI - 流式调用'",
     "test:factory": "jest -t '工厂函数'",
     "test:bind": "jest -t 'bind 方法'",
     "test:batch": "jest -t 'batch 方法'",
+    "test:openai": "jest -t 'ChatOpenAI - 基础测试'",
     "test:aihubmix-no-stream": "jest -t 'aihubmix.com no stream'",
     "test:aihubmix-stream": "jest -t 'aihubmix.com stream'",
     "test:aihubmix-image": "jest -t 'aihubmix.com 图像生成'",

package/dist/image_generation.d.ts

@@ -1,36 +0,0 @@
-/**
- * Image Generation Client
- * 图像生成客户端
- */
-export interface ImageGenerationConfig {
-    baseUrl?: string;
-    headers?: Record<string, string>;
-}
-export interface ImageGenerationRequest {
-    model?: string;
-    prompt: string;
-    size?: string;
-    watermark?: boolean;
-    n?: number;
-    quality?: "standard" | "hd";
-    response_format?: "url" | "b64_json";
-    style?: "vivid" | "natural";
-    user?: string;
-}
-export interface ImageData {
-    url?: string;
-    b64_json?: string;
-}
-export interface ImageGenerationResponse {
-    created: number;
-    data: ImageData[];
-}
-export declare class ImageGenerationClient {
-    private headers;
-    constructor(config?: ImageGenerationConfig);
-    /**
-     * Generate images
-     * 生成图像
-     */
-    generate(request: ImageGenerationRequest): Promise<ImageGenerationResponse>;
-}