@zhin.js/ai 0.0.1 → 1.0.0

package/lib/providers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAChG,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,YAAY,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAChD,YAAY,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACtD,YAAY,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC"}

package/lib/providers/index.js

@@ -0,0 +1,9 @@
+/**
+ * @zhin.js/ai - Providers Index
+ * 导出所有 Provider
+ */
+export { BaseProvider } from './base.js';
+export { OpenAIProvider, DeepSeekProvider, MoonshotProvider, ZhipuProvider } from './openai.js';
+export { AnthropicProvider } from './anthropic.js';
+export { OllamaProvider } from './ollama.js';
+//# sourceMappingURL=index.js.map

package/lib/providers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAChG,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}

package/lib/providers/ollama.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @zhin.js/ai - Ollama Provider
+ * 支持本地 Ollama 模型
+ */
+import { BaseProvider } from './base.js';
+import type { ProviderConfig, ChatCompletionRequest, ChatCompletionResponse, ChatCompletionChunk } from '../types.js';
+export interface OllamaConfig extends ProviderConfig {
+    host?: string;
+    models?: string[];
+}
+export declare class OllamaProvider extends BaseProvider {
+    name: string;
+    models: string[];
+    private host;
+    constructor(config?: OllamaConfig);
+    chat(request: ChatCompletionRequest): Promise<ChatCompletionResponse>;
+    chatStream(request: ChatCompletionRequest): AsyncIterable<ChatCompletionChunk>;
+    listModels(): Promise<string[]>;
+    /**
+     * 拉取模型
+     */
+    pullModel(model: string): Promise<void>;
+    healthCheck(): Promise<boolean>;
+}
+//# sourceMappingURL=ollama.d.ts.map

package/lib/providers/ollama.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,KAAK,EACV,cAAc,EACd,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,aAAa,CAAC;AAIrB,MAAM,WAAW,YAAa,SAAQ,cAAc;IAClD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB;AA2DD,qBAAa,cAAe,SAAQ,YAAY;IAC9C,IAAI,SAAY;IAChB,MAAM,EAAE,MAAM,EAAE,CAAC;IAEjB,OAAO,CAAC,IAAI,CAAS;gBAET,MAAM,GAAE,YAAiB;IAoB/B,IAAI,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAsEpE,UAAU,CAAC,OAAO,EAAE,qBAAqB,GAAG,aAAa,CAAC,mBAAmB,CAAC;IA6E/E,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAWrC;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOvC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;CAQtC"}

package/lib/providers/ollama.js

@@ -0,0 +1,236 @@
+/**
+ * @zhin.js/ai - Ollama Provider
+ * 支持本地 Ollama 模型
+ */
+import { Logger } from '@zhin.js/core';
+import { BaseProvider } from './base.js';
+const logger = new Logger(null, 'Ollama');
+/**
+ * 转换消息格式
+ */
+function toOllamaMessages(messages) {
+    return messages.map(msg => {
+        let content;
+        const images = [];
+        if (typeof msg.content === 'string') {
+            content = msg.content;
+        }
+        else {
+            content = msg.content
+                .filter(p => p.type === 'text')
+                .map(p => p.text)
+                .join('');
+            for (const part of msg.content) {
+                if (part.type === 'image_url') {
+                    const url = part.image_url.url;
+                    if (url.startsWith('data:')) {
+                        // 提取 base64 数据
+                        const base64 = url.split(',')[1];
+                        if (base64)
+                            images.push(base64);
+                    }
+                }
+            }
+        }
+        const result = {
+            role: msg.role === 'tool' ? 'user' : msg.role,
+            content,
+        };
+        if (images.length > 0) {
+            result.images = images;
+        }
+        return result;
+    });
+}
+/**
+ * 转换工具定义
+ */
+function toOllamaTools(tools) {
+    if (!tools?.length)
+        return undefined;
+    return tools.map(tool => ({
+        type: 'function',
+        function: {
+            name: tool.function.name,
+            description: tool.function.description,
+            parameters: tool.function.parameters,
+        },
+    }));
+}
+export class OllamaProvider extends BaseProvider {
+    name = 'ollama';
+    models;
+    host;
+    constructor(config = {}) {
+        super(config);
+        this.host = config.host || config.baseUrl || 'http://localhost:11434';
+        // 使用配置中的模型列表，如果没有则使用默认列表
+        this.models = config.models?.length ? config.models : [
+            'llama3.3',
+            'llama3.2',
+            'llama3.1',
+            'qwen2.5',
+            'qwen2.5-coder',
+            'deepseek-r1',
+            'deepseek-v3',
+            'mistral',
+            'mixtral',
+            'phi4',
+            'gemma2',
+        ];
+        logger.debug(`初始化完成, host: ${this.host}`);
+    }
+    async chat(request) {
+        const messages = toOllamaMessages(request.messages);
+        logger.debug(`请求 ${request.model}, 消息: ${messages.length}`);
+        const ollamaRequest = {
+            model: request.model,
+            messages,
+            stream: false,
+            options: {},
+        };
+        if (request.temperature !== undefined) {
+            ollamaRequest.options.temperature = request.temperature;
+        }
+        if (request.top_p !== undefined) {
+            ollamaRequest.options.top_p = request.top_p;
+        }
+        if (request.max_tokens !== undefined) {
+            ollamaRequest.options.num_predict = request.max_tokens;
+        }
+        const tools = toOllamaTools(request.tools);
+        if (tools) {
+            ollamaRequest.tools = tools;
+        }
+        const startTime = Date.now();
+        const response = await this.fetch(`${this.host}/api/chat`, {
+            method: 'POST',
+            json: ollamaRequest,
+        });
+        logger.debug(`响应耗时: ${Date.now() - startTime}ms, 工具调用: ${response.message?.tool_calls?.length || 0}`);
+        // 转换响应格式
+        const toolCalls = response.message?.tool_calls?.map((tc, i) => ({
+            id: `call_${i}`,
+            type: 'function',
+            function: {
+                name: tc.function.name,
+                arguments: JSON.stringify(tc.function.arguments),
+            },
+        }));
+        return {
+            id: `ollama-${Date.now()}`,
+            object: 'chat.completion',
+            created: Date.now(),
+            model: response.model,
+            choices: [{
+                    index: 0,
+                    message: {
+                        role: 'assistant',
+                        content: response.message?.content || '',
+                        tool_calls: toolCalls?.length ? toolCalls : undefined,
+                    },
+                    finish_reason: toolCalls?.length ? 'tool_calls' : 'stop',
+                }],
+            usage: {
+                prompt_tokens: response.prompt_eval_count || 0,
+                completion_tokens: response.eval_count || 0,
+                total_tokens: (response.prompt_eval_count || 0) + (response.eval_count || 0),
+            },
+        };
+    }
+    async *chatStream(request) {
+        const messages = toOllamaMessages(request.messages);
+        const ollamaRequest = {
+            model: request.model,
+            messages,
+            stream: true,
+            options: {},
+        };
+        if (request.temperature !== undefined) {
+            ollamaRequest.options.temperature = request.temperature;
+        }
+        if (request.top_p !== undefined) {
+            ollamaRequest.options.top_p = request.top_p;
+        }
+        const response = await globalThis.fetch(`${this.host}/api/chat`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(ollamaRequest),
+        });
+        if (!response.ok) {
+            const error = await response.text();
+            throw new Error(`Ollama API Error (${response.status}): ${error}`);
+        }
+        if (!response.body) {
+            throw new Error('Response body is empty');
+        }
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = '';
+        const id = `ollama-${Date.now()}`;
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split('\n');
+            buffer = lines.pop() || '';
+            for (const line of lines) {
+                if (!line.trim())
+                    continue;
+                try {
+                    const data = JSON.parse(line);
+                    yield {
+                        id,
+                        object: 'chat.completion.chunk',
+                        created: Date.now(),
+                        model: data.model || request.model,
+                        choices: [{
+                                index: 0,
+                                delta: data.done
+                                    ? {}
+                                    : { content: data.message?.content || '' },
+                                finish_reason: data.done ? 'stop' : null,
+                            }],
+                        usage: data.done ? {
+                            prompt_tokens: data.prompt_eval_count || 0,
+                            completion_tokens: data.eval_count || 0,
+                            total_tokens: (data.prompt_eval_count || 0) + (data.eval_count || 0),
+                        } : undefined,
+                    };
+                }
+                catch {
+                    // 忽略解析错误
+                }
+            }
+        }
+    }
+    async listModels() {
+        try {
+            const response = await this.fetch(`${this.host}/api/tags`);
+            return response.models.map(m => m.name);
+        }
+        catch {
+            return this.models;
+        }
+    }
+    /**
+     * 拉取模型
+     */
+    async pullModel(model) {
+        await this.fetch(`${this.host}/api/pull`, {
+            method: 'POST',
+            json: { name: model },
+        });
+    }
+    async healthCheck() {
+        try {
+            await globalThis.fetch(`${this.host}/api/tags`);
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/lib/providers/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAUzC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AAO1C;;GAEG;AACH,SAAS,gBAAgB,CAAC,QAAuB;IAC/C,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACxB,IAAI,OAAe,CAAC;QACpB,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,GAAG,CAAC,OAAO;iBAClB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAoC,CAAC,IAAI,CAAC;iBACpD,IAAI,CAAC,EAAE,CAAC,CAAC;YAEZ,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;oBAC/B,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;wBAC5B,eAAe;wBACf,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;wBACjC,IAAI,MAAM;4BAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAClC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAQ;YAClB,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI;YAC7C,OAAO;SACR,CAAC;QAEF,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;QACzB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,KAAwB;IAC7C,IAAI,CAAC,KAAK,EAAE,MAAM;QAAE,OAAO,SAAS,CAAC;IAErC,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxB,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI;YACxB,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW;YACtC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU;SACrC;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC9C,IAAI,GAAG,QAAQ,CAAC;IAChB,MAAM,CAAW;IAET,IAAI,CAAS;IAErB,YAAY,SAAuB,EAAE;QACnC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC;QACtE,yBAAyB;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YACpD,UAAU;YACV,UAAU;YACV,UAAU;YACV,SAAS;YACT,eAAe;YACf,aAAa;YACb,aAAa;YACb,SAAS;YACT,SAAS;YACT,MAAM;YACN,QAAQ;SACT,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,gBAAgB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA8B;QACvC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEpD,MAAM,CAAC,KAAK,CAAC,MAAM,OAAO,CAAC,KAAK,SAAS,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAE5D,MAAM,aAAa,GAAQ;YACzB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ;YACR,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACtC,aAAa,CAAC,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QAC1D,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAChC,aAAa,CAAC,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC9C,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACrC,aAAa,CAAC,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;QACzD,CAAC;QAED,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,KAAK,EAAE,CAAC;YACV,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC;QAC9B,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAM,GAAG,IAAI,CAAC,IAAI,WAAW,EAAE;YAC9D,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,aAAa;SACpB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,aAAa,QAAQ,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;QAEtG,SAAS;QACT,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC,EAAO,EAAE,CAAS,EAAE,EAAE,CAAC,CAAC;YAC3E,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gBACtB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;aACjD;SACF,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,EAAE,EAAE,UAAU,IAAI,CAAC,GAAG,EAAE,EAAE;YAC1B,MAAM,EAAE,iBAAiB;YACzB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;YACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,OAAO,EAAE,CAAC;oBACR,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;wBACxC,UAAU,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;qBACtD;oBACD,aAAa,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM;iBACzD,CAAC;YACF,KAAK,EAAE;gBACL,aAAa,EAAE,QAAQ,CAAC,iBAAiB,IAAI,CAAC;gBAC9C,iBAAiB,EAAE,QAAQ,CAAC,UAAU,IAAI,CAAC;gBAC3C,YAAY,EAAE,CAAC,QAAQ,CAAC,iBAAiB,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,CAAC;aAC7E;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAA8B;QAC9C,MAAM,QAAQ,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEpD,MAAM,aAAa,GAAQ;YACzB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ;YACR,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACtC,aAAa,CAAC,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QAC1D,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAChC,aAAa,CAAC,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC9C,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,WAAW,EAAE;YAC/D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;SACpC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,KAAK,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,MAAM,EAAE,GAAG,UAAU,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QAElC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;oBAAE,SAAS;gBAE3B,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAE9B,MAAM;wBACJ,EAAE;wBACF,MAAM,EAAE,uBAAuB;wBAC/B,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;wBACnB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK;wBAClC,OAAO,EAAE,CAAC;gCACR,KAAK,EAAE,CAAC;gCACR,KAAK,EAAE,IAAI,CAAC,IAAI;oCACd,CAAC,CAAC,EAAE;oCACJ,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,EAAE;gCAC5C,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;6BACzC,CAAC;wBACF,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;4BACjB,aAAa,EAAE,IAAI,CAAC,iBAAiB,IAAI,CAAC;4BAC1C,iBAAiB,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;4BACvC,YAAY,EAAE,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;yBACrE,CAAC,CAAC,CAAC,SAAS;qBACd,CAAC;gBACJ,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAC/B,GAAG,IAAI,CAAC,IAAI,WAAW,CACxB,CAAC;YACF,OAAO,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC,MAAM,CAAC;QACrB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,KAAa;QAC3B,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,WAAW,EAAE;YACxC,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,WAAW,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}

package/lib/providers/openai.d.ts

@@ -0,0 +1,46 @@
+/**
+ * @zhin.js/ai - OpenAI Provider
+ * 支持 OpenAI API 及兼容接口（DeepSeek、Moonshot 等）
+ */
+import { BaseProvider } from './base.js';
+import type { ProviderConfig, ChatCompletionRequest, ChatCompletionResponse, ChatCompletionChunk } from '../types.js';
+export interface OpenAIConfig extends ProviderConfig {
+    organization?: string;
+}
+export declare class OpenAIProvider extends BaseProvider {
+    name: string;
+    models: string[];
+    private baseUrl;
+    constructor(config?: OpenAIConfig);
+    chat(request: ChatCompletionRequest): Promise<ChatCompletionResponse>;
+    chatStream(request: ChatCompletionRequest): AsyncIterable<ChatCompletionChunk>;
+    listModels(): Promise<string[]>;
+}
+/**
+ * DeepSeek Provider（基于 OpenAI 兼容接口）
+ */
+export declare class DeepSeekProvider extends OpenAIProvider {
+    name: string;
+    models: string[];
+    constructor(config?: ProviderConfig);
+    listModels(): Promise<string[]>;
+}
+/**
+ * Moonshot Provider（基于 OpenAI 兼容接口）
+ */
+export declare class MoonshotProvider extends OpenAIProvider {
+    name: string;
+    models: string[];
+    constructor(config?: ProviderConfig);
+    listModels(): Promise<string[]>;
+}
+/**
+ * 智谱 AI Provider（基于 OpenAI 兼容接口）
+ */
+export declare class ZhipuProvider extends OpenAIProvider {
+    name: string;
+    models: string[];
+    constructor(config?: ProviderConfig);
+    listModels(): Promise<string[]>;
+}
+//# sourceMappingURL=openai.d.ts.map

package/lib/providers/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,KAAK,EACV,cAAc,EACd,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACpB,MAAM,aAAa,CAAC;AAErB,MAAM,WAAW,YAAa,SAAQ,cAAc;IAClD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,cAAe,SAAQ,YAAY;IAC9C,IAAI,SAAY;IAChB,MAAM,WAUJ;IAEF,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,GAAE,YAAiB;IAY/B,IAAI,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAapE,UAAU,CAAC,OAAO,EAAE,qBAAqB,GAAG,aAAa,CAAC,mBAAmB,CAAC;IAuB/E,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CActC;AAED;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,cAAc;IAClD,IAAI,SAAc;IAClB,MAAM,WAIJ;gBAEU,MAAM,GAAE,cAAmB;IAOjC,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAGtC;AAED;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,cAAc;IAClD,IAAI,SAAc;IAClB,MAAM,WAIJ;gBAEU,MAAM,GAAE,cAAmB;IAOjC,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAGtC;AAED;;GAEG;AACH,qBAAa,aAAc,SAAQ,cAAc;IAC/C,IAAI,SAAW;IACf,MAAM,WAMJ;gBAEU,MAAM,GAAE,cAAmB;IAOjC,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAGtC"}

package/lib/providers/openai.js

@@ -0,0 +1,132 @@
+/**
+ * @zhin.js/ai - OpenAI Provider
+ * 支持 OpenAI API 及兼容接口（DeepSeek、Moonshot 等）
+ */
+import { BaseProvider } from './base.js';
+export class OpenAIProvider extends BaseProvider {
+    name = 'openai';
+    models = [
+        'gpt-4o',
+        'gpt-4o-mini',
+        'gpt-4-turbo',
+        'gpt-4',
+        'gpt-3.5-turbo',
+        'o1',
+        'o1-mini',
+        'o1-preview',
+        'o3-mini',
+    ];
+    baseUrl;
+    constructor(config = {}) {
+        super(config);
+        this.baseUrl = config.baseUrl || 'https://api.openai.com/v1';
+        if (config.organization) {
+            this.config.headers = {
+                ...this.config.headers,
+                'OpenAI-Organization': config.organization,
+            };
+        }
+    }
+    async chat(request) {
+        return this.fetch(`${this.baseUrl}/chat/completions`, {
+            method: 'POST',
+            json: {
+                ...request,
+                stream: false,
+            },
+        });
+    }
+    async *chatStream(request) {
+        const stream = this.fetchStream(`${this.baseUrl}/chat/completions`, {
+            method: 'POST',
+            json: {
+                ...request,
+                stream: true,
+                stream_options: { include_usage: true },
+            },
+        });
+        for await (const data of stream) {
+            try {
+                const chunk = JSON.parse(data);
+                yield chunk;
+            }
+            catch {
+                // 忽略解析错误的行
+            }
+        }
+    }
+    async listModels() {
+        try {
+            const response = await this.fetch(`${this.baseUrl}/models`);
+            return response.data
+                .map((m) => m.id)
+                .filter((id) => id.includes('gpt') || id.includes('o1') || id.includes('o3'));
+        }
+        catch {
+            return this.models;
+        }
+    }
+}
+/**
+ * DeepSeek Provider（基于 OpenAI 兼容接口）
+ */
+export class DeepSeekProvider extends OpenAIProvider {
+    name = 'deepseek';
+    models = [
+        'deepseek-chat',
+        'deepseek-coder',
+        'deepseek-reasoner',
+    ];
+    constructor(config = {}) {
+        super({
+            ...config,
+            baseUrl: config.baseUrl || 'https://api.deepseek.com/v1',
+        });
+    }
+    async listModels() {
+        return this.models;
+    }
+}
+/**
+ * Moonshot Provider（基于 OpenAI 兼容接口）
+ */
+export class MoonshotProvider extends OpenAIProvider {
+    name = 'moonshot';
+    models = [
+        'moonshot-v1-8k',
+        'moonshot-v1-32k',
+        'moonshot-v1-128k',
+    ];
+    constructor(config = {}) {
+        super({
+            ...config,
+            baseUrl: config.baseUrl || 'https://api.moonshot.cn/v1',
+        });
+    }
+    async listModels() {
+        return this.models;
+    }
+}
+/**
+ * 智谱 AI Provider（基于 OpenAI 兼容接口）
+ */
+export class ZhipuProvider extends OpenAIProvider {
+    name = 'zhipu';
+    models = [
+        'glm-4-plus',
+        'glm-4',
+        'glm-4-air',
+        'glm-4-flash',
+        'glm-4v-plus',
+    ];
+    constructor(config = {}) {
+        super({
+            ...config,
+            baseUrl: config.baseUrl || 'https://open.bigmodel.cn/api/paas/v4',
+        });
+    }
+    async listModels() {
+        return this.models;
+    }
+}
+//# sourceMappingURL=openai.js.map

package/lib/providers/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAYzC,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC9C,IAAI,GAAG,QAAQ,CAAC;IAChB,MAAM,GAAG;QACP,QAAQ;QACR,aAAa;QACb,aAAa;QACb,OAAO;QACP,eAAe;QACf,IAAI;QACJ,SAAS;QACT,YAAY;QACZ,SAAS;KACV,CAAC;IAEM,OAAO,CAAS;IAExB,YAAY,SAAuB,EAAE;QACnC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,2BAA2B,CAAC;QAE7D,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG;gBACpB,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO;gBACtB,qBAAqB,EAAE,MAAM,CAAC,YAAY;aAC3C,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA8B;QACvC,OAAO,IAAI,CAAC,KAAK,CACf,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAClC;YACE,MAAM,EAAE,MAAM;YACd,IAAI,EAAE;gBACJ,GAAG,OAAO;gBACV,MAAM,EAAE,KAAK;aACd;SACF,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAA8B;QAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAC7B,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAClC;YACE,MAAM,EAAE,MAAM;YACd,IAAI,EAAE;gBACJ,GAAG,OAAO;gBACV,MAAM,EAAE,IAAI;gBACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;aACxC;SACF,CACF,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAChC,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAwB,CAAC;gBACtD,MAAM,KAAK,CAAC;YACd,CAAC;YAAC,MAAM,CAAC;gBACP,WAAW;YACb,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QAKd,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAY,GAAG,IAAI,CAAC,OAAO,SAAS,CAAC,CAAC;YACvE,OAAO,QAAQ,CAAC,IAAI;iBACjB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChB,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAClF,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC,MAAM,CAAC;QACrB,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,cAAc;IAClD,IAAI,GAAG,UAAU,CAAC;IAClB,MAAM,GAAG;QACP,eAAe;QACf,gBAAgB;QAChB,mBAAmB;KACpB,CAAC;IAEF,YAAY,SAAyB,EAAE;QACrC,KAAK,CAAC;YACJ,GAAG,MAAM;YACT,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,6BAA6B;SACzD,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,cAAc;IAClD,IAAI,GAAG,UAAU,CAAC;IAClB,MAAM,GAAG;QACP,gBAAgB;QAChB,iBAAiB;QACjB,kBAAkB;KACnB,CAAC;IAEF,YAAY,SAAyB,EAAE;QACrC,KAAK,CAAC;YACJ,GAAG,MAAM;YACT,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,4BAA4B;SACxD,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,aAAc,SAAQ,cAAc;IAC/C,IAAI,GAAG,OAAO,CAAC;IACf,MAAM,GAAG;QACP,YAAY;QACZ,OAAO;QACP,WAAW;QACX,aAAa;QACb,aAAa;KACd,CAAC;IAEF,YAAY,SAAyB,EAAE;QACrC,KAAK,CAAC;YACJ,GAAG,MAAM;YACT,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,sCAAsC;SAClE,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF"}

package/lib/session.d.ts

@@ -0,0 +1,186 @@
+/**
+ * @zhin.js/ai - Session Manager
+ * 会话管理器，支持上下文记忆和数据库持久化
+ *
+ * 特性：
+ * - 数据库持久化存储（使用 Zhin 的数据库服务）
+ * - 内存缓存加速读取
+ * - 自动过期清理
+ * - 更长的上下文记忆能力
+ */
+import type { ChatMessage, SessionConfig, Session } from './types.js';
+/**
+ * 数据库模型定义
+ */
+export declare const AI_SESSION_MODEL: {
+    session_id: {
+        type: "text";
+        nullable: boolean;
+    };
+    messages: {
+        type: "json";
+        default: never[];
+    };
+    config: {
+        type: "json";
+        default: {};
+    };
+    created_at: {
+        type: "integer";
+        default: number;
+    };
+    updated_at: {
+        type: "integer";
+        default: number;
+    };
+};
+/**
+ * 会话管理器接口
+ */
+export interface ISessionManager {
+    get(sessionId: string, config?: SessionConfig): Session | Promise<Session>;
+    has(sessionId: string): boolean | Promise<boolean>;
+    addMessage(sessionId: string, message: ChatMessage): void | Promise<void>;
+    getMessages(sessionId: string): ChatMessage[] | Promise<ChatMessage[]>;
+    setSystemPrompt(sessionId: string, prompt: string): void | Promise<void>;
+    clear(sessionId: string): boolean | Promise<boolean>;
+    reset(sessionId: string): void | Promise<void>;
+    listSessions(): string[] | Promise<string[]>;
+    getStats(): {
+        total: number;
+        active: number;
+        expired: number;
+    } | Promise<{
+        total: number;
+        active: number;
+        expired: number;
+    }>;
+    cleanup(): number | Promise<number>;
+    dispose(): void | Promise<void>;
+}
+/**
+ * 内存会话管理器（回退方案）
+ */
+export declare class MemorySessionManager implements ISessionManager {
+    private sessions;
+    private config;
+    private cleanupTimer?;
+    constructor(config?: {
+        maxHistory?: number;
+        expireMs?: number;
+    });
+    get(sessionId: string, config?: SessionConfig): Session;
+    has(sessionId: string): boolean;
+    addMessage(sessionId: string, message: ChatMessage): void;
+    private trimMessages;
+    getMessages(sessionId: string): ChatMessage[];
+    setSystemPrompt(sessionId: string, prompt: string): void;
+    clear(sessionId: string): boolean;
+    reset(sessionId: string): void;
+    listSessions(): string[];
+    getStats(): {
+        total: number;
+        active: number;
+        expired: number;
+    };
+    cleanup(): number;
+    dispose(): void;
+}
+/**
+ * 数据库会话管理器
+ * 使用 Zhin 的数据库服务进行持久化存储
+ */
+export declare class DatabaseSessionManager implements ISessionManager {
+    private cache;
+    private config;
+    private cleanupTimer?;
+    private saveQueue;
+    private saveTimer?;
+    private model;
+    constructor(model: any, config?: {
+        maxHistory?: number;
+        expireMs?: number;
+    });
+    /**
+     * 从数据库加载会话
+     */
+    private loadSession;
+    /**
+     * 保存会话到数据库（防抖）
+     */
+    private schedulesSave;
+    /**
+     * 批量保存队列中的会话
+     */
+    private flushSaveQueue;
+    get(sessionId: string, config?: SessionConfig): Promise<Session>;
+    has(sessionId: string): Promise<boolean>;
+    addMessage(sessionId: string, message: ChatMessage): Promise<void>;
+    private trimMessages;
+    getMessages(sessionId: string): Promise<ChatMessage[]>;
+    setSystemPrompt(sessionId: string, prompt: string): Promise<void>;
+    clear(sessionId: string): Promise<boolean>;
+    reset(sessionId: string): Promise<void>;
+    listSessions(): Promise<string[]>;
+    getStats(): Promise<{
+        total: number;
+        active: number;
+        expired: number;
+    }>;
+    cleanup(): Promise<number>;
+    dispose(): Promise<void>;
+}
+/**
+ * 会话管理器包装器
+ * 支持同步和异步接口的统一使用
+ */
+export declare class SessionManager implements ISessionManager {
+    private manager;
+    constructor(manager: ISessionManager);
+    /**
+     * 生成会话 ID
+     */
+    static generateId(platform: string, userId: string, channelId?: string): string;
+    get(sessionId: string, config?: SessionConfig): Session | Promise<Session>;
+    has(sessionId: string): boolean | Promise<boolean>;
+    addMessage(sessionId: string, message: ChatMessage): void | Promise<void>;
+    getMessages(sessionId: string): ChatMessage[] | Promise<ChatMessage[]>;
+    setSystemPrompt(sessionId: string, prompt: string): void | Promise<void>;
+    clear(sessionId: string): boolean | Promise<boolean>;
+    reset(sessionId: string): void | Promise<void>;
+    listSessions(): string[] | Promise<string[]>;
+    getStats(): {
+        total: number;
+        active: number;
+        expired: number;
+    } | Promise<{
+        total: number;
+        active: number;
+        expired: number;
+    }>;
+    cleanup(): number | Promise<number>;
+    dispose(): void | Promise<void>;
+}
+/**
+ * 创建内存会话管理器（回退方案）
+ */
+export declare function createMemorySessionManager(config?: {
+    maxHistory?: number;
+    expireMs?: number;
+}): SessionManager;
+/**
+ * 创建数据库会话管理器
+ */
+export declare function createDatabaseSessionManager(model: any, config?: {
+    maxHistory?: number;
+    expireMs?: number;
+}): SessionManager;
+/**
+ * 创建会话管理器（向后兼容）
+ * @deprecated 使用 createMemorySessionManager 或 createDatabaseSessionManager
+ */
+export declare function createSessionManager(config?: {
+    maxHistory?: number;
+    expireMs?: number;
+}): SessionManager;
+//# sourceMappingURL=session.d.ts.map