ai-world-sdk 1.1.1 → 1.1.3

package/dist/log.js

@@ -27,7 +27,7 @@ function logRequest(method, url, headers, body) {
     debugLog("📤 HTTP Request");
     debugLog("  Method:", method);
     debugLog("  URL:", url);
-    debugLog("  Headers:", { ...headers, Authorization: headers?.Authorization ? "Bearer ***" : undefined });
+    debugLog("  Headers:", { ...headers, Authorization: headers?.Authorization });
     if (body) {
         try {
             const bodyStr = typeof body === "string" ? body : JSON.stringify(body, null, 2);

package/dist/openai_video_generation.d.ts

@@ -0,0 +1,188 @@
+/**
+ * OpenAI Video Generation Client
+ * OpenAI 视频生成客户端
+ */
+export interface OpenAIVideoGenerationConfig {
+    baseUrl?: string;
+    token?: string;
+    headers?: Record<string, string>;
+}
+export interface OpenAIVideoGenerationRequest {
+    prompt: string;
+    model?: string;
+    provider?: "aihubmix" | "shubiaobiao" | "api2img";
+    seconds?: "4" | "8" | "12";
+    size?: "720x1280" | "1280x720" | "1024x1792" | "1792x1024";
+    input_reference?: string;
+}
+export interface OpenAIVideoTaskResponse {
+    id: string;
+    object?: string;
+    status: "queued" | "in_progress" | "completed" | "failed";
+    created_at: number;
+    model?: string;
+    prompt?: string;
+    seconds?: "4" | "8" | "12";
+    size?: string;
+    url?: string;
+    error?: Record<string, any>;
+}
+/**
+ * OpenAI Video Generation Client
+ * OpenAI 视频生成客户端
+ *
+ * 使用示例:
+ * ```typescript
+ * import { OpenAIVideoGenerationClient } from 'ai-world-sdk';
+ *
+ * const client = new OpenAIVideoGenerationClient({
+ *   baseUrl: 'http://localhost:8000',
+ *   token: 'your-auth-token'
+ * });
+ *
+ * // 生成视频
+ * const task = await client.generate({
+ *   prompt: 'A beautiful sunset over the ocean',
+ *   model: 'sora-2',
+ *   provider: 'aihubmix',
+ *   seconds: '4',
+ *   size: '1280x720'
+ * });
+ *
+ * console.log('Task ID:', task.id);
+ *
+ * // 查询任务状态
+ * const status = await client.getTask(task.id);
+ * console.log('Status:', status.status);
+ * if (status.video_url) {
+ *   console.log('Video URL:', status.video_url);
+ * }
+ * ```
+ */
+export declare class OpenAIVideoGenerationClient {
+    private headers;
+    constructor(config?: OpenAIVideoGenerationConfig);
+    /**
+     * 生成视频
+     * Generate video
+     *
+     * @param request - 视频生成请求参数
+     * @returns 任务响应，包含任务 ID
+     *
+     * @example
+     * ```typescript
+     * // 文本生成视频
+     * const task = await client.generate({
+     *   prompt: 'A beautiful sunset over the ocean with waves crashing on the shore',
+     *   model: 'sora-2',
+     *   provider: 'aihubmix',
+     *   seconds: '4',
+     *   size: '1280x720'
+     * });
+     *
+     * // 图像生成视频（使用 base64）
+     * const taskWithImage = await client.generate({
+     *   prompt: 'Animate this image with gentle waves',
+     *   input_reference: 'data:image/png;base64,iVBORw0KGgo...',
+     *   model: 'sora-2',
+     *   seconds: '4'
+     * });
+     * console.log('Task ID:', task.id);
+     * ```
+     */
+    generate(request: OpenAIVideoGenerationRequest): Promise<OpenAIVideoTaskResponse>;
+    /**
+     * 查询视频任务状态
+     * Get video task status
+     *
+     * @param taskId - 任务 ID
+     * @returns 任务状态信息
+     *
+     * @example
+     * ```typescript
+     * const status = await client.getTask('vid_abc123');
+     * console.log('Status:', status.status);
+     * if (status.status === 'completed' && status.url) {
+     *   console.log('Video URL:', status.url);
+     * }
+     * ```
+     */
+    getTask(taskId: string, provider: "aihubmix" | "shubiaobiao" | "api2img"): Promise<OpenAIVideoTaskResponse>;
+    /**
+     * 轮询等待任务完成
+     * Poll and wait for task completion
+     *
+     * @param taskId - 任务 ID
+     * @param options - 轮询选项
+     * @returns 完成的任务信息
+     *
+     * @example
+     * ```typescript
+     * const result = await client.waitForCompletion('vid_abc123', {
+     *   maxAttempts: 60,
+     *   interval: 5000
+     * });
+     * console.log('Video URL:', result.url);
+     * ```
+     */
+    waitForCompletion(taskId: string, provider?: "aihubmix" | "shubiaobiao" | "api2img", options?: {
+        maxAttempts?: number;
+        interval?: number;
+        onProgress?: (status: OpenAIVideoTaskResponse) => void;
+    }): Promise<OpenAIVideoTaskResponse>;
+    /**
+     * 生成视频并等待完成
+     * Generate video and wait for completion
+     *
+     * @param request - 视频生成请求参数
+     * @param options - 轮询选项
+     * @returns 完成的任务信息，包含视频 URL
+     *
+     * @example
+     * ```typescript
+     * const result = await client.generateAndWait({
+     *   prompt: 'A beautiful sunset',
+     *   model: 'sora-2',
+     *   provider: 'aihubmix',
+     *   seconds: '4'
+     * }, {
+     *   onProgress: (status) => {
+     *     console.log('Status:', status.status);
+     *   }
+     * });
+     * console.log('Video URL:', result.video_url);
+     * ```
+     */
+    generateAndWait(request: OpenAIVideoGenerationRequest, options?: {
+        maxAttempts?: number;
+        interval?: number;
+        onProgress?: (status: OpenAIVideoTaskResponse) => void;
+    }): Promise<OpenAIVideoTaskResponse>;
+    /**
+     * 下载视频文件
+     * Download video file
+     *
+     * @param taskId - 视频任务 ID
+     * @param provider - 提供商类型 (aihubmix, api2img)
+     * @param variant - 下载类型 (video, thumbnail)，默认: video
+     * @returns Blob 对象（可用于下载或显示）
+     *
+     * @example
+     * ```typescript
+     * // 下载视频
+     * const videoBlob = await client.downloadVideo('vid_abc123', 'aihubmix', 'video');
+     *
+     * // 在浏览器中创建下载链接
+     * const url = URL.createObjectURL(videoBlob);
+     * const a = document.createElement('a');
+     * a.href = url;
+     * a.download = 'video.mp4';
+     * a.click();
+     * URL.revokeObjectURL(url);
+     *
+     * // 或下载缩略图
+     * const thumbnail = await client.downloadVideo('vid_abc123', 'aihubmix', 'thumbnail');
+     * ```
+     */
+    downloadVideo(taskId: string, provider?: "aihubmix" | "shubiaobiao" | "api2img", variant?: "video" | "thumbnail"): Promise<Blob>;
+}

package/dist/openai_video_generation.js

@@ -0,0 +1,287 @@
+"use strict";
+/**
+ * OpenAI Video Generation Client
+ * OpenAI 视频生成客户端
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIVideoGenerationClient = void 0;
+const config_1 = require("./config");
+const log_1 = require("./log");
+/**
+ * OpenAI Video Generation Client
+ * OpenAI 视频生成客户端
+ *
+ * 使用示例:
+ * ```typescript
+ * import { OpenAIVideoGenerationClient } from 'ai-world-sdk';
+ *
+ * const client = new OpenAIVideoGenerationClient({
+ *   baseUrl: 'http://localhost:8000',
+ *   token: 'your-auth-token'
+ * });
+ *
+ * // 生成视频
+ * const task = await client.generate({
+ *   prompt: 'A beautiful sunset over the ocean',
+ *   model: 'sora-2',
+ *   provider: 'aihubmix',
+ *   seconds: '4',
+ *   size: '1280x720'
+ * });
+ *
+ * console.log('Task ID:', task.id);
+ *
+ * // 查询任务状态
+ * const status = await client.getTask(task.id);
+ * console.log('Status:', status.status);
+ * if (status.video_url) {
+ *   console.log('Video URL:', status.video_url);
+ * }
+ * ```
+ */
+class OpenAIVideoGenerationClient {
+    constructor(config = {}) {
+        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 video
+     *
+     * @param request - 视频生成请求参数
+     * @returns 任务响应，包含任务 ID
+     *
+     * @example
+     * ```typescript
+     * // 文本生成视频
+     * const task = await client.generate({
+     *   prompt: 'A beautiful sunset over the ocean with waves crashing on the shore',
+     *   model: 'sora-2',
+     *   provider: 'aihubmix',
+     *   seconds: '4',
+     *   size: '1280x720'
+     * });
+     *
+     * // 图像生成视频（使用 base64）
+     * const taskWithImage = await client.generate({
+     *   prompt: 'Animate this image with gentle waves',
+     *   input_reference: 'data:image/png;base64,iVBORw0KGgo...',
+     *   model: 'sora-2',
+     *   seconds: '4'
+     * });
+     * console.log('Task ID:', task.id);
+     * ```
+     */
+    async generate(request) {
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/openai-video-proxy/generate`;
+        // 记录请求日志
+        (0, log_1.logRequest)("POST", url, this.headers, request);
+        try {
+            const response = await fetch(url, {
+                method: "POST",
+                headers: this.headers,
+                body: JSON.stringify(request),
+            });
+            // 记录响应日志
+            (0, log_1.logResponse)(response.status, response.statusText, response.headers, response.body);
+            if (!response.ok) {
+                const errorData = await response.json().catch(() => ({
+                    detail: response.statusText,
+                }));
+                // 检查版本兼容性错误
+                if (response.status === 400 &&
+                    errorData.detail?.includes("version")) {
+                    throw new config_1.VersionCompatibilityError(errorData.detail);
+                }
+                throw new Error(errorData.detail || `HTTP error! status: ${response.status}`);
+            }
+            const data = await response.json();
+            return data;
+        }
+        catch (error) {
+            if (error instanceof config_1.VersionCompatibilityError) {
+                throw error;
+            }
+            throw new Error(`OpenAI 视频生成失败: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * 查询视频任务状态
+     * Get video task status
+     *
+     * @param taskId - 任务 ID
+     * @returns 任务状态信息
+     *
+     * @example
+     * ```typescript
+     * const status = await client.getTask('vid_abc123');
+     * console.log('Status:', status.status);
+     * if (status.status === 'completed' && status.url) {
+     *   console.log('Video URL:', status.url);
+     * }
+     * ```
+     */
+    async getTask(taskId, provider) {
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/openai-video-proxy/${taskId}?provider=${provider}`;
+        // 记录请求日志
+        (0, log_1.logRequest)("GET", url, this.headers, {});
+        try {
+            const response = await fetch(url, {
+                method: "GET",
+                headers: this.headers,
+            });
+            // 记录响应日志
+            (0, log_1.logResponse)(response.status, response.statusText, response.headers, response.body);
+            if (!response.ok) {
+                const errorData = await response.json().catch(() => ({
+                    detail: response.statusText,
+                }));
+                // 检查版本兼容性错误
+                if (response.status === 400 &&
+                    errorData.detail?.includes("version")) {
+                    throw new config_1.VersionCompatibilityError(errorData.detail);
+                }
+                throw new Error(errorData.detail || `HTTP error! status: ${response.status}`);
+            }
+            const data = await response.json();
+            return data;
+        }
+        catch (error) {
+            if (error instanceof config_1.VersionCompatibilityError) {
+                throw error;
+            }
+            throw new Error(`查询任务失败: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * 轮询等待任务完成
+     * Poll and wait for task completion
+     *
+     * @param taskId - 任务 ID
+     * @param options - 轮询选项
+     * @returns 完成的任务信息
+     *
+     * @example
+     * ```typescript
+     * const result = await client.waitForCompletion('vid_abc123', {
+     *   maxAttempts: 60,
+     *   interval: 5000
+     * });
+     * console.log('Video URL:', result.url);
+     * ```
+     */
+    async waitForCompletion(taskId, provider = "aihubmix", options = {}) {
+        const { maxAttempts = 60, interval = 5000, onProgress } = options;
+        for (let attempt = 0; attempt < maxAttempts; attempt++) {
+            const task = await this.getTask(taskId, provider || "aihubmix");
+            if (onProgress) {
+                onProgress(task);
+            }
+            if (task.status === "completed") {
+                return task;
+            }
+            if (task.status === "failed") {
+                throw new Error(`任务失败: ${task.error || "未知错误"}`);
+            }
+            // 等待指定时间后继续轮询
+            await new Promise((resolve) => setTimeout(resolve, interval));
+        }
+        throw new Error(`任务超时: 已轮询 ${maxAttempts} 次，任务仍未完成`);
+    }
+    /**
+     * 生成视频并等待完成
+     * Generate video and wait for completion
+     *
+     * @param request - 视频生成请求参数
+     * @param options - 轮询选项
+     * @returns 完成的任务信息，包含视频 URL
+     *
+     * @example
+     * ```typescript
+     * const result = await client.generateAndWait({
+     *   prompt: 'A beautiful sunset',
+     *   model: 'sora-2',
+     *   provider: 'aihubmix',
+     *   seconds: '4'
+     * }, {
+     *   onProgress: (status) => {
+     *     console.log('Status:', status.status);
+     *   }
+     * });
+     * console.log('Video URL:', result.video_url);
+     * ```
+     */
+    async generateAndWait(request, options) {
+        const task = await this.generate(request);
+        return this.waitForCompletion(task.id, request.provider || "aihubmix", options);
+    }
+    /**
+     * 下载视频文件
+     * Download video file
+     *
+     * @param taskId - 视频任务 ID
+     * @param provider - 提供商类型 (aihubmix, api2img)
+     * @param variant - 下载类型 (video, thumbnail)，默认: video
+     * @returns Blob 对象（可用于下载或显示）
+     *
+     * @example
+     * ```typescript
+     * // 下载视频
+     * const videoBlob = await client.downloadVideo('vid_abc123', 'aihubmix', 'video');
+     *
+     * // 在浏览器中创建下载链接
+     * const url = URL.createObjectURL(videoBlob);
+     * const a = document.createElement('a');
+     * a.href = url;
+     * a.download = 'video.mp4';
+     * a.click();
+     * URL.revokeObjectURL(url);
+     *
+     * // 或下载缩略图
+     * const thumbnail = await client.downloadVideo('vid_abc123', 'aihubmix', 'thumbnail');
+     * ```
+     */
+    async downloadVideo(taskId, provider = "aihubmix", variant = "video") {
+        const url = `${config_1.sdkConfig.getServerUrl()}/api/openai-video-proxy/${taskId}/download?provider=${provider}&variant=${variant}`;
+        // 记录请求日志
+        (0, log_1.logRequest)("GET", url, this.headers, { taskId, provider, variant });
+        try {
+            const response = await fetch(url, {
+                method: "GET",
+                headers: this.headers,
+            });
+            if (!response.ok) {
+                const errorData = await response.json().catch(() => null);
+                // 记录错误响应日志
+                (0, log_1.logResponse)(response.status, response.statusText, response.headers, errorData);
+                // 检查版本兼容性错误
+                if (response.status === 400 &&
+                    errorData?.detail?.includes("version")) {
+                    throw new config_1.VersionCompatibilityError(errorData.detail);
+                }
+                throw new Error(errorData?.detail || `HTTP error! status: ${response.status}`);
+            }
+            // 流式下载视频（浏览器会自动处理流式传输）
+            const blob = await response.blob();
+            // 记录成功日志
+            (0, log_1.logResponse)(response.status, response.statusText, response.headers, {
+                blobSize: blob.size,
+                blobType: blob.type,
+            });
+            return blob;
+        }
+        catch (error) {
+            if (error instanceof config_1.VersionCompatibilityError) {
+                throw error;
+            }
+            throw new Error(`下载视频失败: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+}
+exports.OpenAIVideoGenerationClient = OpenAIVideoGenerationClient;

package/dist/provider_config.d.ts

@@ -0,0 +1,8 @@
+export type AIModelProvider = "aihubmix" | "api2img" | "doubao" | "gemini" | "shubiaobiao";
+export type ProviderModelType = "openai" | "gemini";
+export interface ProviderConfig {
+    provider: AIModelProvider;
+    modelType: ProviderModelType;
+}
+export declare const PROVIDER_CONFIGS: Record<AIModelProvider, ProviderConfig>;
+export declare function getProviderConfig(provider: AIModelProvider): ProviderConfig;

package/dist/provider_config.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PROVIDER_CONFIGS = void 0;
+exports.getProviderConfig = getProviderConfig;
+exports.PROVIDER_CONFIGS = {
+    aihubmix: { provider: "aihubmix", modelType: "openai" },
+    shubiaobiao: { provider: "shubiaobiao", modelType: "openai" },
+    api2img: { provider: "api2img", modelType: "openai" },
+    doubao: { provider: "doubao", modelType: "openai" },
+    gemini: { provider: "gemini", modelType: "gemini" },
+};
+function getProviderConfig(provider) {
+    return exports.PROVIDER_CONFIGS[provider];
+}

package/package.json

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