ai-world-sdk 1.0.0

package/dist/base.js

@@ -0,0 +1,181 @@
+"use strict";
+/**
+ * Base chat model class
+ * Similar to LangChain.js BaseChatModel
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseChatModel = void 0;
+const messages_1 = require("./messages");
+const config_1 = require("./config");
+class BaseChatModel {
+    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}`;
+        }
+        // 如果提供了 provider，添加到请求头
+        if (config.provider) {
+            this.headers["X-Provider"] = config.provider;
+            this.provider = config.provider;
+        }
+        // 如果提供了 apiKey，添加到请求头（用于指定后端使用的环境变量名）
+        if (config.apiKey) {
+            this.headers["X-Api-Key-Env"] = config.apiKey;
+            this.apiKey = config.apiKey;
+        }
+        this.temperature = config.temperature ?? 0.7;
+        this.maxTokens = config.maxTokens;
+        this.topP = config.topP;
+        this.modelName = config.modelName;
+    }
+    /**
+     * Invoke the model with messages (non-streaming)
+     * 遵循标准 LangChain API
+     */
+    async invoke(messages) {
+        const options = this._mergeOptions({});
+        const response = await fetch(`${this.baseUrl}/api/langchain-proxy/invoke`, {
+            method: "POST",
+            headers: this.headers,
+            body: JSON.stringify({
+                messages: messages.map((msg) => msg.toJSON()),
+                config: {
+                    temperature: options.temperature ?? this.temperature,
+                    max_tokens: options.maxTokens ?? this.maxTokens,
+                    top_p: options.topP ?? this.topP,
+                    tools: options.tools,
+                    tool_choice: options.toolChoice,
+                },
+                model: this.modelName,
+                provider: this.provider,
+                api_key_env: this.apiKey,
+            }),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Langchain API error: ${response.status} ${errorText}`);
+        }
+        // 返回标准 AIMessage 格式（从 message_to_dict 序列化）
+        const data = (await response.json());
+        // 从标准 AIMessage 格式创建 AIMessage 对象
+        const content = data.content || "";
+        return new messages_1.AIMessage(content);
+    }
+    /**
+     * Stream the model with messages
+     * 遵循标准 LangChain API，返回 AIMessageChunk
+     */
+    async *stream(messages) {
+        const options = this._mergeOptions({});
+        const requestBody = {
+            messages: messages.map((msg) => msg.toJSON()),
+            config: {
+                temperature: options.temperature ?? this.temperature,
+                max_tokens: options.maxTokens ?? this.maxTokens,
+                top_p: options.topP ?? this.topP,
+                tools: options.tools,
+                tool_choice: options.toolChoice,
+            },
+            model: this.modelName,
+            provider: this.provider,
+            api_key_env: this.apiKey,
+        };
+        const response = await fetch(`${this.baseUrl}/api/langchain-proxy/stream`, {
+            method: "POST",
+            headers: {
+                ...this.headers,
+                Accept: "text/event-stream",
+            },
+            body: JSON.stringify(requestBody),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Langchain API error: ${response.status} ${errorText}`);
+        }
+        const decoder = new TextDecoder();
+        for await (const chunk of response.body?.values() || []) {
+            try {
+                yield new messages_1.AIMessageChunk(JSON.parse(decoder.decode(chunk)));
+            }
+            catch {
+                // 忽略解析错误
+            }
+        }
+    }
+    /**
+     * Batch invoke multiple message sets
+     * 遵循标准 LangChain API
+     */
+    async batch(messagesList) {
+        const response = await fetch(`${this.baseUrl}/api/langchain-proxy/batch`, {
+            method: "POST",
+            headers: this.headers,
+            body: JSON.stringify({
+                messages_list: messagesList.map((msgs) => msgs.map((msg) => msg.toJSON())),
+                config: {
+                    temperature: this.temperature,
+                    max_tokens: this.maxTokens,
+                    top_p: this.topP,
+                },
+                model: this.modelName,
+                provider: this.provider,
+                api_key_env: this.apiKey,
+            }),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Langchain API error: ${response.status} ${errorText}`);
+        }
+        // 返回标准 AIMessage 格式列表（从 message_to_dict 序列化）
+        const results = (await response.json());
+        return results.map((result) => new messages_1.AIMessage(result.content || ""));
+    }
+    /**
+     * Bind options to the model (temperature, maxTokens, tools, etc.)
+     * Returns a new model instance with bound options
+     * Similar to LangChain.js bind method
+     */
+    bind(options) {
+        const newModel = Object.create(Object.getPrototypeOf(this));
+        Object.assign(newModel, this);
+        newModel.boundOptions = { ...this.boundOptions, ...options };
+        return newModel;
+    }
+    /**
+     * Bind tools to the model
+     * Similar to LangChain.js bindTools method
+     */
+    bindTools(tools) {
+        return this.bind({ tools });
+    }
+    /**
+     * Merge bound options with provided options
+     */
+    _mergeOptions(options) {
+        return {
+            ...this.boundOptions,
+            ...options,
+        };
+    }
+    /**
+     * Get the current model name
+     */
+    getModelName() {
+        return this.modelName;
+    }
+}
+exports.BaseChatModel = BaseChatModel;

package/dist/chat_models/anthropic.d.ts

@@ -0,0 +1,11 @@
+/**
+ * ChatAnthropic - Anthropic Claude chat model
+ * Similar to LangChain.js ChatAnthropic
+ */
+import { BaseChatModel, BaseChatModelParams } from "../base";
+export interface ChatAnthropicConfig extends BaseChatModelParams {
+    modelName: string;
+}
+export declare class ChatAnthropic extends BaseChatModel {
+    constructor(config: ChatAnthropicConfig);
+}

package/dist/chat_models/anthropic.js

@@ -0,0 +1,17 @@
+"use strict";
+/**
+ * ChatAnthropic - Anthropic Claude chat model
+ * Similar to LangChain.js ChatAnthropic
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatAnthropic = void 0;
+const base_1 = require("../base");
+class ChatAnthropic extends base_1.BaseChatModel {
+    constructor(config) {
+        super({
+            ...config,
+            modelName: config.modelName || "claude-3-sonnet-20240229",
+        });
+    }
+}
+exports.ChatAnthropic = ChatAnthropic;

package/dist/chat_models/google.d.ts

@@ -0,0 +1,11 @@
+/**
+ * ChatGoogleGenerativeAI - Google Gemini chat model
+ * Similar to LangChain.js ChatGoogleGenerativeAI
+ */
+import { BaseChatModel, BaseChatModelParams } from "../base";
+export interface ChatGoogleGenerativeAIConfig extends BaseChatModelParams {
+    modelName: string;
+}
+export declare class ChatGoogleGenerativeAI extends BaseChatModel {
+    constructor(config: ChatGoogleGenerativeAIConfig);
+}

package/dist/chat_models/google.js

@@ -0,0 +1,17 @@
+"use strict";
+/**
+ * ChatGoogleGenerativeAI - Google Gemini chat model
+ * Similar to LangChain.js ChatGoogleGenerativeAI
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatGoogleGenerativeAI = void 0;
+const base_1 = require("../base");
+class ChatGoogleGenerativeAI extends base_1.BaseChatModel {
+    constructor(config) {
+        super({
+            ...config,
+            modelName: config.modelName || "gemini-1.5-pro",
+        });
+    }
+}
+exports.ChatGoogleGenerativeAI = ChatGoogleGenerativeAI;

package/dist/chat_models/openai.d.ts

@@ -0,0 +1,11 @@
+/**
+ * ChatOpenAI - OpenAI chat model
+ * Similar to LangChain.js ChatOpenAI
+ */
+import { BaseChatModel, BaseChatModelParams } from "../base";
+export interface ChatOpenAIConfig extends BaseChatModelParams {
+    modelName: string;
+}
+export declare class ChatOpenAI extends BaseChatModel {
+    constructor(config: ChatOpenAIConfig);
+}

package/dist/chat_models/openai.js

@@ -0,0 +1,17 @@
+"use strict";
+/**
+ * ChatOpenAI - OpenAI chat model
+ * Similar to LangChain.js ChatOpenAI
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatOpenAI = void 0;
+const base_1 = require("../base");
+class ChatOpenAI extends base_1.BaseChatModel {
+    constructor(config) {
+        super({
+            ...config,
+            modelName: config.modelName || "gpt-3.5-turbo",
+        });
+    }
+}
+exports.ChatOpenAI = ChatOpenAI;

package/dist/config.d.ts

@@ -0,0 +1,46 @@
+/**
+ * Global SDK Configuration
+ * 全局 SDK 配置
+ */
+declare class SDKConfig {
+    private _baseUrl;
+    private _token;
+    private _headers;
+    /**
+     * Set global base URL
+     * 设置全局 base URL
+     */
+    setBaseUrl(baseUrl: string): void;
+    /**
+     * Get global base URL
+     * 获取全局 base URL
+     */
+    getBaseUrl(): string | null;
+    /**
+     * Set global token
+     * 设置全局 token
+     */
+    setToken(token: string): void;
+    /**
+     * Get global token
+     * 获取全局 token
+     */
+    getToken(): string | null;
+    /**
+     * Set global headers
+     * 设置全局 headers
+     */
+    setHeaders(headers: Record<string, string>): void;
+    /**
+     * Get global headers
+     * 获取全局 headers
+     */
+    getHeaders(): Record<string, string>;
+    /**
+     * Reset all global configuration
+     * 重置所有全局配置
+     */
+    reset(): void;
+}
+export declare const sdkConfig: SDKConfig;
+export {};

package/dist/config.js

@@ -0,0 +1,67 @@
+"use strict";
+/**
+ * Global SDK Configuration
+ * 全局 SDK 配置
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sdkConfig = void 0;
+class SDKConfig {
+    constructor() {
+        this._baseUrl = null;
+        this._token = null;
+        this._headers = {};
+    }
+    /**
+     * Set global base URL
+     * 设置全局 base URL
+     */
+    setBaseUrl(baseUrl) {
+        this._baseUrl = baseUrl.replace(/\/$/, "");
+    }
+    /**
+     * Get global base URL
+     * 获取全局 base URL
+     */
+    getBaseUrl() {
+        return this._baseUrl;
+    }
+    /**
+     * Set global token
+     * 设置全局 token
+     */
+    setToken(token) {
+        this._token = token;
+    }
+    /**
+     * Get global token
+     * 获取全局 token
+     */
+    getToken() {
+        return this._token;
+    }
+    /**
+     * Set global headers
+     * 设置全局 headers
+     */
+    setHeaders(headers) {
+        this._headers = { ...this._headers, ...headers };
+    }
+    /**
+     * Get global headers
+     * 获取全局 headers
+     */
+    getHeaders() {
+        return { ...this._headers };
+    }
+    /**
+     * Reset all global configuration
+     * 重置所有全局配置
+     */
+    reset() {
+        this._baseUrl = null;
+        this._token = null;
+        this._headers = {};
+    }
+}
+// 导出单例实例
+exports.sdkConfig = new SDKConfig();

package/dist/image_generation.d.ts

@@ -0,0 +1,38 @@
+/**
+ * Image Generation Client
+ * 图像生成客户端
+ */
+export interface ImageGenerationConfig {
+    baseUrl?: string;
+    token?: 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 baseUrl;
+    private headers;
+    constructor(config?: ImageGenerationConfig);
+    /**
+     * Generate images
+     * 生成图像
+     */
+    generate(request: ImageGenerationRequest): Promise<ImageGenerationResponse>;
+}

package/dist/image_generation.js

@@ -0,0 +1,70 @@
+"use strict";
+/**
+ * Image Generation Client
+ * 图像生成客户端
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ImageGenerationClient = void 0;
+const config_1 = require("./config");
+class ImageGenerationClient {
+    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}`;
+        }
+    }
+    /**
+     * Generate images
+     * 生成图像
+     */
+    async generate(request) {
+        const requestBody = {
+            model: request.model || "doubao-seedream-4-5-251128",
+            prompt: request.prompt,
+            size: request.size || "2K",
+            n: request.n || 1,
+        };
+        // 添加可选参数
+        if (request.watermark !== undefined) {
+            requestBody.watermark = request.watermark;
+        }
+        if (request.quality) {
+            requestBody.quality = request.quality;
+        }
+        if (request.response_format) {
+            requestBody.response_format = request.response_format;
+        }
+        if (request.style) {
+            requestBody.style = request.style;
+        }
+        if (request.user) {
+            requestBody.user = request.user;
+        }
+        const response = await fetch(`${this.baseUrl}/api/image-proxy/generate`, {
+            method: "POST",
+            headers: this.headers,
+            body: JSON.stringify(requestBody),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Image generation API error: ${response.status} ${errorText}`);
+        }
+        const data = (await response.json());
+        return data;
+    }
+}
+exports.ImageGenerationClient = ImageGenerationClient;

package/dist/index.d.ts

@@ -0,0 +1,27 @@
+/**
+ * AI World SDK
+ * TypeScript SDK for AI World Platform
+ * Includes Langchain-compatible chat models and image generation
+ * @see https://github.com/langchain-ai/langchainjs
+ */
+import { BaseChatModel, BaseChatModelParams } from "./base";
+import { ImageGenerationClient, type ImageGenerationConfig, type ImageGenerationRequest, type ImageGenerationResponse } from "./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";
+export { BaseChatModel, type BaseChatModelParams, type ToolDefinition, type BindOptions, } from "./base";
+export { ChatOpenAI } from "./chat_models/openai";
+export { ChatGoogleGenerativeAI } from "./chat_models/google";
+export { ChatAnthropic } from "./chat_models/anthropic";
+export interface LangchainClientConfig {
+    baseUrl?: string;
+    token?: string;
+    headers?: Record<string, string>;
+}
+export { ImageGenerationClient, type ImageGenerationConfig, type ImageGenerationRequest, type ImageGenerationResponse, };
+export { VideoGenerationClient, type VideoGenerationConfig, type VideoGenerationRequest, type ContentGenerationTaskID, type ContentGenerationTask, };
+export { sdkConfig };
+/**
+ * Create a chat model instance based on model name
+ */
+export declare function createChatModel(model: string, config: LangchainClientConfig & BaseChatModelParams): BaseChatModel;

package/dist/index.js

@@ -0,0 +1,77 @@
+"use strict";
+/**
+ * AI World SDK
+ * TypeScript SDK for AI World Platform
+ * Includes Langchain-compatible chat models and image generation
+ * @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.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 video_generation_1 = require("./video_generation");
+Object.defineProperty(exports, "VideoGenerationClient", { enumerable: true, get: function () { return video_generation_1.VideoGenerationClient; } });
+const config_1 = require("./config");
+Object.defineProperty(exports, "sdkConfig", { enumerable: true, get: function () { return config_1.sdkConfig; } });
+// Re-export types and classes
+var messages_1 = require("./messages");
+Object.defineProperty(exports, "HumanMessage", { enumerable: true, get: function () { return messages_1.HumanMessage; } });
+Object.defineProperty(exports, "AIMessage", { enumerable: true, get: function () { return messages_1.AIMessage; } });
+Object.defineProperty(exports, "SystemMessage", { enumerable: true, get: function () { return messages_1.SystemMessage; } });
+Object.defineProperty(exports, "AIMessageChunk", { enumerable: true, get: function () { return messages_1.AIMessageChunk; } });
+var base_1 = require("./base");
+Object.defineProperty(exports, "BaseChatModel", { enumerable: true, get: function () { return base_1.BaseChatModel; } });
+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; } });
+/**
+ * Create a chat model instance based on model name
+ */
+function createChatModel(model, config) {
+    // 如果没有提供 baseUrl，使用全局配置
+    const finalConfig = {
+        ...config,
+        baseUrl: config.baseUrl || config_1.sdkConfig.getBaseUrl() || undefined,
+        token: config.token || config_1.sdkConfig.getToken() || undefined,
+        headers: {
+            ...config_1.sdkConfig.getHeaders(),
+            ...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-*`);
+    }
+}

package/dist/messages.d.ts

@@ -0,0 +1,71 @@
+/**
+ * Message types for Langchain SDK
+ * Similar to LangChain.js message interface
+ */
+export interface MessageContent {
+    type: "text" | "image_url";
+    text?: string;
+    image_url?: string;
+}
+export interface BaseMessage {
+    content: string | MessageContent[];
+    role?: "user" | "assistant" | "system";
+    toJSON(): {
+        role: string;
+        content: string | MessageContent[];
+    };
+}
+export declare class HumanMessage implements BaseMessage {
+    content: string | MessageContent[];
+    role: "user";
+    constructor(content: string | MessageContent[]);
+    toJSON(): {
+        role: string;
+        content: string | MessageContent[];
+    };
+}
+export declare class AIMessage implements BaseMessage {
+    content: string | MessageContent[];
+    role: "assistant";
+    constructor(content: string | MessageContent[]);
+    toJSON(): {
+        role: string;
+        content: string | MessageContent[];
+    };
+}
+export declare class SystemMessage implements BaseMessage {
+    content: string | MessageContent[];
+    role: "system";
+    constructor(content: string | MessageContent[]);
+    toJSON(): {
+        role: string;
+        content: string | MessageContent[];
+    };
+}
+export interface AIMessageChunkData {
+    content?: string | MessageContent[];
+    type?: "AIMessageChunk" | "ai";
+    id?: string;
+    tool_call_chunks?: any[];
+    chunk_position?: "last" | null;
+    response_metadata?: Record<string, any>;
+    usage_metadata?: Record<string, any>;
+    additional_kwargs?: Record<string, any>;
+    [key: string]: any;
+}
+export declare class AIMessageChunk implements BaseMessage {
+    content: string | MessageContent[];
+    role: "assistant";
+    type: "AIMessageChunk";
+    id?: string;
+    tool_call_chunks?: any[];
+    chunk_position?: "last" | null;
+    response_metadata?: Record<string, any>;
+    usage_metadata?: Record<string, any>;
+    additional_kwargs?: Record<string, any>;
+    constructor(data: AIMessageChunkData | string | MessageContent[]);
+    toJSON(): {
+        role: string;
+        content: string | MessageContent[];
+    };
+}