ai-world-sdk 1.0.0

package/dist/messages.js

@@ -0,0 +1,73 @@
+"use strict";
+/**
+ * Message types for Langchain SDK
+ * Similar to LangChain.js message interface
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIMessageChunk = exports.SystemMessage = exports.AIMessage = exports.HumanMessage = void 0;
+class HumanMessage {
+    constructor(content) {
+        this.role = "user";
+        this.content = content;
+    }
+    toJSON() {
+        return {
+            role: this.role,
+            content: this.content,
+        };
+    }
+}
+exports.HumanMessage = HumanMessage;
+class AIMessage {
+    constructor(content) {
+        this.role = "assistant";
+        this.content = content;
+    }
+    toJSON() {
+        return {
+            role: this.role,
+            content: this.content,
+        };
+    }
+}
+exports.AIMessage = AIMessage;
+class SystemMessage {
+    constructor(content) {
+        this.role = "system";
+        this.content = content;
+    }
+    toJSON() {
+        return {
+            role: this.role,
+            content: this.content,
+        };
+    }
+}
+exports.SystemMessage = SystemMessage;
+class AIMessageChunk {
+    constructor(data) {
+        this.role = "assistant";
+        this.type = "AIMessageChunk";
+        if (typeof data === "string" || Array.isArray(data)) {
+            // 兼容直接传入 content 的情况
+            this.content = data;
+        }
+        else {
+            // 从序列化的对象创建
+            this.content = data.content || "";
+            this.id = data.id;
+            this.tool_call_chunks = data.tool_call_chunks;
+            this.chunk_position = data.chunk_position;
+            this.response_metadata = data.response_metadata;
+            this.usage_metadata = data.usage_metadata;
+            this.additional_kwargs = data.additional_kwargs;
+        }
+    }
+    toJSON() {
+        return {
+            role: this.role,
+            content: this.content,
+        };
+    }
+}
+exports.AIMessageChunk = AIMessageChunk;

package/dist/video_generation.d.ts

@@ -0,0 +1,91 @@
+/**
+ * Video Generation Client
+ * 视频生成客户端
+ */
+export interface VideoGenerationConfig {
+    baseUrl?: string;
+    token?: string;
+    headers?: Record<string, string>;
+}
+export interface VideoGenerationContentText {
+    type: "text";
+    text: string;
+}
+export interface VideoGenerationContentImage {
+    type: "image_url";
+    image_url: {
+        url: string;
+    };
+    role: string;
+}
+export type VideoGenerationContent = VideoGenerationContentText | VideoGenerationContentImage;
+export interface VideoGenerationRequest {
+    prompt?: string;
+    image_url?: string;
+    model?: string;
+    content?: VideoGenerationContent[];
+    callback_url?: string;
+    return_last_frame?: boolean;
+    service_tier?: string;
+    execution_expires_after?: number;
+    duration?: number;
+    aspect_ratio?: "16:9" | "9:16" | "1:1";
+    resolution?: string;
+    user?: string;
+}
+export interface ContentGenerationTaskID {
+    id: string;
+}
+export interface ContentGenerationError {
+    message: string;
+    code: string;
+}
+export interface ContentGenerationContent {
+    video_url?: string;
+    last_frame_url?: string;
+    file_url?: string;
+}
+export interface Usage {
+    completion_tokens: number;
+}
+export interface ContentGenerationTask {
+    id: string;
+    model: string;
+    status: "queued" | "running" | "succeeded" | "failed" | "cancelled";
+    error?: ContentGenerationError;
+    content?: ContentGenerationContent;
+    usage?: Usage;
+    subdivisionlevel?: string;
+    fileformat?: string;
+    frames?: number;
+    framespersecond?: number;
+    created_at: number;
+    updated_at?: number;
+    seed?: number;
+    revised_prompt?: string;
+    service_tier?: string;
+    execution_expires_after?: number;
+}
+export declare class VideoGenerationClient {
+    private baseUrl;
+    private headers;
+    constructor(config?: VideoGenerationConfig);
+    /**
+     * Create a video generation task
+     * 创建视频生成任务
+     */
+    create(request: VideoGenerationRequest): Promise<ContentGenerationTaskID>;
+    /**
+     * Get video generation task status
+     * 查询视频生成任务状态
+     */
+    get(taskId: string): Promise<ContentGenerationTask>;
+    /**
+     * Poll video generation task until completion
+     * 轮询视频生成任务直到完成
+     */
+    poll(taskId: string, options?: {
+        interval?: number;
+        timeout?: number;
+    }): Promise<ContentGenerationTask>;
+}

package/dist/video_generation.js

@@ -0,0 +1,88 @@
+"use strict";
+/**
+ * Video Generation Client
+ * 视频生成客户端
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VideoGenerationClient = void 0;
+const config_1 = require("./config");
+class VideoGenerationClient {
+    constructor(config) {
+        // 使用提供的 baseUrl，否则使用全局配置
+        const baseUrl = config?.baseUrl || config_1.sdkConfig.getBaseUrl();
+        if (!baseUrl) {
+            throw new Error("baseUrl is required. Either provide it in config or set it globally using sdkConfig.setBaseUrl()");
+        }
+        this.baseUrl = baseUrl.replace(/\/$/, "");
+        // 合并全局 headers 和配置 headers
+        const globalHeaders = config_1.sdkConfig.getHeaders();
+        this.headers = {
+            "Content-Type": "application/json",
+            ...globalHeaders,
+            ...config?.headers,
+        };
+        // 使用提供的 token，否则使用全局 token
+        const token = config?.token || config_1.sdkConfig.getToken();
+        if (token) {
+            this.headers["Authorization"] = `Bearer ${token}`;
+        }
+    }
+    /**
+     * Create a video generation task
+     * 创建视频生成任务
+     */
+    async create(request) {
+        const response = await fetch(`${this.baseUrl}/api/video-proxy/generate`, {
+            method: "POST",
+            headers: this.headers,
+            body: JSON.stringify(request),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Video generation API error: ${response.status} ${errorText}`);
+        }
+        const data = (await response.json());
+        return data;
+    }
+    /**
+     * Get video generation task status
+     * 查询视频生成任务状态
+     */
+    async get(taskId) {
+        const response = await fetch(`${this.baseUrl}/api/video-proxy/${taskId}`, {
+            method: "GET",
+            headers: this.headers,
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Get video task API error: ${response.status} ${errorText}`);
+        }
+        const data = (await response.json());
+        return data;
+    }
+    /**
+     * Poll video generation task until completion
+     * 轮询视频生成任务直到完成
+     */
+    async poll(taskId, options) {
+        const interval = options?.interval || 5000;
+        const timeout = options?.timeout || 300000;
+        const startTime = Date.now();
+        while (true) {
+            const task = await this.get(taskId);
+            // 检查是否完成或失败
+            if (task.status === "succeeded" ||
+                task.status === "failed" ||
+                task.status === "cancelled") {
+                return task;
+            }
+            // 检查超时
+            if (Date.now() - startTime > timeout) {
+                throw new Error(`Video generation task polling timeout after ${timeout}ms`);
+            }
+            // 等待后继续轮询
+            await new Promise((resolve) => setTimeout(resolve, interval));
+        }
+    }
+}
+exports.VideoGenerationClient = VideoGenerationClient;

package/package.json

@@ -0,0 +1,56 @@
+{
+  "name": "ai-world-sdk",
+  "version": "1.0.0",
+  "description": "TypeScript SDK for AI World Platform - Chat Models, Image Generation, and Video Generation",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "npm run build",
+    "test": "jest",
+    "test:watch": "jest --watch",
+    "test:non-stream": "jest -t 'ChatGoogleGenerativeAI - 非流式调用'",
+    "test:stream": "jest -t 'ChatGoogleGenerativeAI - 流式调用'",
+    "test:factory": "jest -t '工厂函数'",
+    "test:bind": "jest -t 'bind 方法'",
+    "test:batch": "jest -t 'batch 方法'",
+    "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 图像生成'",
+    "test:seedream-image": "jest -t 'seedream 图像生成'",
+    "test:image-generation": "jest -t 'ImageGenerationClient - 基础图像生成'",
+    "test:video-generation": "jest -t 'VideoGenerationClient - 轮询视频生成任务'"
+  },
+  "keywords": [
+    "ai-world",
+    "ai",
+    "llm",
+    "langchain",
+    "image-generation",
+    "video-generation",
+    "sdk",
+    "typescript",
+    "openai",
+    "gemini",
+    "claude",
+    "doubao"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "dotenv": "^16.0.0"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "@types/jest": "^29.5.0",
+    "@types/dotenv": "^8.2.0",
+    "typescript": "^5.0.0",
+    "ts-node": "^10.9.2",
+    "jest": "^29.5.0",
+    "ts-jest": "^29.1.0"
+  },
+  "files": [
+    "dist"
+  ]
+}
+