ai-world-sdk 1.0.2 → 1.0.3

package/dist/__tests__/example.test.js

@@ -279,8 +279,8 @@ describe("Langchain SDK Tests", () => {
             console.log("响应内容:", content);
         }
     }, 60000);
-    test("ImageGenerationClient - 基础图像生成", async () => {
-        const imageClient = new index_1.ImageGenerationClient({});
+    test("DoubaoImageGenerationClient - 基础图像生成", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
         const result = await imageClient.generate({
             prompt: "充满活力的特写编辑肖像，模特眼神犀利，头戴雕塑感帽子，色彩拼接丰富，眼部焦点锐利，景深较浅，具有Vogue杂志封面的美学风格，采用中画幅拍摄，工作室灯光效果强烈。",
             model: "doubao-seedream-4-5-251128",
@@ -306,8 +306,8 @@ describe("Langchain SDK Tests", () => {
         console.log("图像 URL:", result.data[0]?.url || "Base64 编码");
         console.log("创建时间:", new Date(result.created * 1000).toISOString());
     }, 120000);
-    test("ImageGenerationClient - 多图像生成 (n=2)", async () => {
-        const imageClient = new index_1.ImageGenerationClient({});
+    test("DoubaoImageGenerationClient - 多图像生成 (n=2)", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
         const result = await imageClient.generate({
             prompt: "A beautiful sunset over the ocean with vibrant colors",
             model: "doubao-seedream-4-5-251128",
@@ -324,8 +324,8 @@ describe("Langchain SDK Tests", () => {
             console.log(`图像 ${index + 1}:`, item.url || "Base64 编码");
         });
     }, 120000);
-    test("ImageGenerationClient - 不同尺寸测试", async () => {
-        const imageClient = new index_1.ImageGenerationClient({});
+    test("DoubaoImageGenerationClient - 不同尺寸测试", async () => {
+        const imageClient = new index_1.DoubaoImageGenerationClient({});
         const sizes = ["1K", "2K", "4K"];
         for (const size of sizes) {
             const result = await imageClient.generate({

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,16 @@
  * @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");

package/package.json

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

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>;
-}