@kevisual/ai 0.0.18 → 0.0.20

package/dist/ai-provider.d.ts

@@ -1,25 +1,171 @@
-import * as openai_resources_index_mjs from 'openai/resources/index.mjs';
-import OpenAI, { OpenAI as OpenAI$1 } from 'openai';
-import * as openai_resources_embeddings_mjs from 'openai/resources/embeddings.mjs';
 import * as _kevisual_permission from '@kevisual/permission';
 import { Permission } from '@kevisual/permission';
 
-type ChatMessage = OpenAI.Chat.Completions.ChatCompletionMessageParam;
-type ChatMessageOptions = Partial<OpenAI.Chat.Completions.ChatCompletionCreateParams> & {
+type ChatMessage = {
+    role?: 'user' | 'assistant' | 'system' | 'tool';
+    content: string;
+};
+type ChatMessageOptions<T = {}> = {
+    messages?: ChatMessage[];
+    /**
+    * 模型名称
+    */
+    model?: string;
+    /**
+     * 温度参数，控制随机性 (0-2)
+     * 较高的值如0.8会使输出更随机，较低的值如0.2会使其更集中和确定
+     */
+    temperature?: number;
+    /**
+     * 核采样参数 (0-1)
+     * 与temperature类似，但使用不同的采样方法
+     */
+    top_p?: number;
+    /**
+     * 生成的最大令牌数
+     */
+    max_tokens?: number;
+    /**
+     * 停止序列
+     * 当遇到这些序列时停止生成
+     */
+    stop?: string | string[];
+    /**
+     * 频率惩罚 (-2.0 到 2.0)
+     * 正值会根据新令牌在文本中的现有频率来惩罚它们
+     */
+    frequency_penalty?: number;
+    /**
+     * 存在惩罚 (-2.0 到 2.0)
+     * 正值会根据新令牌是否出现在文本中来惩罚它们
+     */
+    presence_penalty?: number;
+    /**
+     * 流式输出
+     */
+    stream?: boolean;
     /**
      * 是否能够思考
-     * 如果会话是千文，服务器的接口，默认为 true
+     * 如果会话是千文，服务器的接口，默认为 false
      */
     enable_thinking?: boolean;
+    response_format?: 'text' | 'json' | 'xml' | 'html';
+    /**
+     * 工具调用参数
+     */
+    tool_calls?: any;
+    [key: string]: any;
+} & T;
+type Choice = {
+    finish_reason: 'stop' | 'length' | 'tool_calls' | 'content_filter' | 'function_call';
+    index: number;
+    message: {
+        role: 'assistant' | 'tool';
+        content: string;
+    };
+};
+type ChatMessageComplete = {
+    /**
+     * 聊天完成的唯一标识符
+     */
+    id: string;
+    /**
+     * 聊天完成选项列表
+     * 如果 n 大于 1，可以有多个选项
+     */
+    choices: Array<Choice>;
+    /**
+     * 聊天完成创建时的 Unix 时间戳（秒）
+     */
+    created: number;
+    /**
+     * 用于聊天完成的模型名称
+     */
+    model: string;
+    /**
+     * 对象类型，始终为 `chat.completion`
+     */
+    object: 'chat.completion';
+    /**
+     * 系统指纹
+     * 用于标识后端配置
+     */
+    system_fingerprint?: string;
+    /**
+     * 完成请求的使用统计信息
+     */
+    usage?: Usage;
+};
+/**
+ * 向量嵌入请求参数
+ */
+type EmbeddingMessage = {
+    /**
+     * 输入文本或文本数组
+     * 要生成嵌入向量的文本内容
+     */
+    input: string | string[];
+    /**
+     * 模型名称
+     * 用于生成嵌入向量的模型
+     */
+    model: string;
+    /**
+     * 编码格式
+     * 返回的嵌入向量编码格式
+     */
+    encoding_format?: 'float' | 'base64';
+    /**
+     * 维度
+     * 输出嵌入向量的维度（某些模型支持）
+     */
+    dimensions?: number;
+};
+/**
+ * 单个嵌入向量对象
+ */
+type EmbeddingObject = {
+    /**
+     * 对象类型，始终为 `embedding`
+     */
+    object: 'embedding';
+    /**
+     * 嵌入向量
+     * 浮点数数组，表示文本的向量表示
+     */
+    embedding: number[];
+    /**
+     * 索引位置
+     * 该嵌入在输入数组中的位置
+     */
+    index: number;
+};
+/**
+ * 向量嵌入响应结果
+ */
+type EmbeddingMessageComplete = {
+    /**
+     * 对象类型，始终为 `list`
+     */
+    object: 'list';
+    /**
+     * 嵌入向量数据列表
+     */
+    data: EmbeddingObject[];
+    /**
+     * 使用的模型名称
+     */
+    model: string;
+    /**
+     * 使用统计信息
+     */
+    usage: Usage;
 };
-type ChatMessageComplete = OpenAI.Chat.Completions.ChatCompletion;
-type ChatMessageStream = OpenAI.Chat.Completions.ChatCompletion;
-type EmbeddingMessage = Partial<OpenAI.Embeddings.EmbeddingCreateParams>;
-type EmbeddingMessageComplete = OpenAI.Embeddings.CreateEmbeddingResponse;
 interface BaseChatInterface {
+    chat(options: ChatMessageOptions): Promise<ChatMessageComplete>;
     chat(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatMessageComplete>;
 }
-interface BaseChatUsageInterface {
+interface Usage {
     /**
      * 提示词令牌
      */
@@ -132,7 +278,7 @@ type BaseChatOptions<T = Record<string, any>> = {
      */
     stream?: boolean;
 } & T;
-declare class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
+declare class BaseChat implements BaseChatInterface, Usage {
     /**
      * 默认baseURL
      */
@@ -161,8 +307,17 @@ declare class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
     /**
      * 聊天
      */
+    chat(options: ChatMessageOptions): Promise<ChatMessageComplete>;
     chat(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatMessageComplete>;
-    chatStream(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatStream>;
+    chatStream(options: ChatMessageOptions): AsyncGenerator<ChatMessageComplete>;
+    chatStream(messages: ChatMessage[], options?: ChatMessageOptions): AsyncGenerator<ChatMessageComplete>;
+    /**
+     * 简单提问接口
+     * @param message
+     * @param options
+     * @returns
+     */
+    question(message: string, options?: ChatMessageOptions): Promise<ChatMessageComplete>;
     /**
      * 获取聊天使用情况
      * @returns
@@ -172,6 +327,11 @@ declare class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
         total_tokens: number;
         completion_tokens: number;
     };
+    setChatUsage(usage: {
+        prompt_tokens?: number;
+        total_tokens?: number;
+        completion_tokens?: number;
+    }): void;
     getHeaders(headers?: Record<string, string>): {
         'Content-Type': string;
         Authorization: string;
@@ -211,7 +371,6 @@ type OllamaModel = {
 declare class Ollama extends BaseChat {
     static BASE_URL: string;
     constructor(options: OllamaOptions$1);
-    chat(messages: ChatMessage[], options?: ChatMessageOptions): Promise<openai_resources_index_mjs.ChatCompletion>;
     /**
      * 获取模型列表
      * @returns
@@ -249,7 +408,6 @@ declare class SiliconFlow extends BaseChat {
     static BASE_URL: string;
     constructor(options: SiliconFlowOptions);
     getUsageInfo(): Promise<SiliconFlowUsageResponse>;
-    chat(messages: OpenAI$1.Chat.Completions.ChatCompletionMessageParam[], options?: ChatMessageOptions): Promise<OpenAI$1.Chat.Completions.ChatCompletion>;
 }
 
 type OllamaOptions = BaseChatOptions;
@@ -348,7 +506,7 @@ declare class ProviderManager {
     provider: BaseChat;
     constructor(config: ProviderManagerConfig);
     static createProvider(config: ProviderManagerConfig): Promise<BaseChat>;
-    chat(messages: ChatMessage[]): Promise<openai_resources_index_mjs.ChatCompletion>;
+    chat(messages: ChatMessage[]): Promise<ChatMessageComplete>;
 }
 
 type KnowledgeOptions<T = Record<string, any>> = BaseChatOptions<{
@@ -376,7 +534,7 @@ declare class KnowledgeBase extends BaseChat {
      */
     generateEmbedding(text: string | string[]): Promise<{
         code: number;
-        data: openai_resources_embeddings_mjs.Embedding[];
+        data: EmbeddingObject[];
         message?: undefined;
     } | {
         code: any;
@@ -552,4 +710,4 @@ declare class AIConfigParser {
 }
 
 export { AIConfigParser, AIUtils, BailianChat, BailianProvider, BaseChat, ChatProviderMap, Custom, CustomProvider, DeepSeek, DeepSeekProvider, Kevisual, KevisualProvider, Kimi, KimiProvider, KnowledgeBase, ModelScope, ModelScopeProvider, Ollama, OllamaProvider, ProviderManager, SiliconFlow, SiliconFlowKnowledge, SiliconFlowProvider, Volces, VolcesProvider, Zhipu, ZhipuProvider, decryptAES, encryptAES, readStream };
-export type { AIConfig, AIModel, BaseChatInterface, BaseChatOptions, BaseChatUsageInterface, ChatMessage, ChatMessageComplete, ChatMessageOptions, ChatMessageStream, ChatStream, EmbeddingMessage, EmbeddingMessageComplete, GetProviderOpts, KnowledgeOptions, ProviderResult, RerankOptions, SecretKey };
+export type { AIConfig, AIModel, BaseChatInterface, BaseChatOptions, ChatMessage, ChatMessageComplete, ChatMessageOptions, ChatStream, EmbeddingMessage, EmbeddingMessageComplete, EmbeddingObject, GetProviderOpts, KnowledgeOptions, ProviderResult, RerankOptions, SecretKey, Usage };

package/dist/ai-provider.js

@@ -1350,10 +1350,10 @@ class BaseChat {
   baseURL;
   model;
   apiKey;
-  prompt_tokens;
-  total_tokens;
-  completion_tokens;
-  responseText;
+  prompt_tokens = 0;
+  total_tokens = 0;
+  completion_tokens = 0;
+  responseText = "";
   utils = AIUtils;
   constructor(options) {
     this.baseURL = options.baseURL;
@@ -1385,11 +1385,14 @@ class BaseChat {
       }
     }).then((res) => res.json());
   }
-  async chat(messages, options) {
+  async chat(messagesOrOptions, options) {
+    const isFirstParamOptions = !Array.isArray(messagesOrOptions);
+    const messages = isFirstParamOptions ? messagesOrOptions.messages : messagesOrOptions;
+    const opts = isFirstParamOptions ? messagesOrOptions : options || {};
     const requestBody = {
       model: this.model,
       messages,
-      ...options,
+      ...opts,
       stream: false
     };
     const response = await this.post(`${this.baseURL}/chat/completions`, { data: requestBody });
@@ -1398,20 +1401,21 @@ class BaseChat {
       throw new Error(`Chat API request failed: ${response.status} ${response.statusText} - ${errorText}`);
     }
     const res = await response.json();
-    this.prompt_tokens = res.usage?.prompt_tokens ?? 0;
-    this.total_tokens = res.usage?.total_tokens ?? 0;
-    this.completion_tokens = res.usage?.completion_tokens ?? 0;
+    this.setChatUsage(res.usage);
     this.responseText = res.choices[0]?.message?.content || "";
     return res;
   }
-  async chatStream(messages, options) {
-    if (options?.response_format) {
+  async* chatStream(messagesOrOptions, options) {
+    const isFirstParamOptions = !Array.isArray(messagesOrOptions);
+    const messages = isFirstParamOptions ? messagesOrOptions.messages : messagesOrOptions;
+    const opts = isFirstParamOptions ? messagesOrOptions : options || {};
+    if (opts.response_format) {
       throw new Error("response_format is not supported in stream mode");
     }
     const requestBody = {
       model: this.model,
       messages,
-      ...options,
+      ...opts,
       stream: true
     };
     const response = await this.post(`${this.baseURL}/chat/completions`, { data: requestBody });
@@ -1452,6 +1456,12 @@ class BaseChat {
     });
     return stream;
   }
+  question(message, options) {
+    const messages = [
+      { role: "user", content: message }
+    ];
+    return this.chat(messages, options);
+  }
   getChatUsage() {
     return {
       prompt_tokens: this.prompt_tokens,
@@ -1459,6 +1469,11 @@ class BaseChat {
       completion_tokens: this.completion_tokens
     };
   }
+  setChatUsage(usage) {
+    this.prompt_tokens = usage.prompt_tokens ?? this.prompt_tokens;
+    this.total_tokens = usage.total_tokens ?? this.total_tokens;
+    this.completion_tokens = usage.completion_tokens ?? this.completion_tokens;
+  }
   getHeaders(headers) {
     return {
       "Content-Type": "application/json",
@@ -1499,11 +1514,6 @@ class Ollama extends BaseChat {
     const baseURL = options.baseURL || Ollama.BASE_URL;
     super({ ...options, baseURL });
   }
-  async chat(messages, options) {
-    const res = await super.chat(messages, options);
-    console.log("thunk", this.getChatUsage());
-    return res;
-  }
   async listModels() {
     const _url = new URL(this.baseURL);
     const tagsURL = new URL("/api/tags", _url);
@@ -1526,10 +1536,6 @@ class SiliconFlow extends BaseChat {
   async getUsageInfo() {
     return this.get("/user/info");
   }
-  async chat(messages, options) {
-    const res = await super.chat(messages, options);
-    return res;
-  }
 }
 
 // src/provider/chat-adapter/custom.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/ai",
-  "version": "0.0.18",
+  "version": "0.0.20",
   "description": "AI Center Services",
   "main": "index.js",
   "basename": "/root/ai-center-services",
@@ -13,6 +13,13 @@
     "src",
     "types"
   ],
+  "scripts": {
+    "build": "npm run clean && bun bun.config.mjs",
+    "dev": "bun run --watch bun.config.mjs",
+    "test": "tsx  test/**/*.ts",
+    "clean": "rm -rf dist",
+    "pub": "envision pack -p -u"
+  },
   "keywords": [
     "kevisual",
     "ai",
@@ -20,6 +27,7 @@
   ],
   "author": "abearxiong <xiongxiao@xiongxiao.me> (https://www.xiongxiao.me)",
   "license": "MIT",
+  "packageManager": "pnpm@10.28.0",
   "type": "module",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
@@ -45,34 +53,27 @@
     }
   },
   "devDependencies": {
-    "@kevisual/router": "0.0.36",
+    "@kevisual/router": "0.0.52",
     "@kevisual/types": "^0.0.10",
     "@kevisual/use-config": "^1.0.21",
-    "@types/bun": "^1.3.4",
+    "@types/bun": "^1.3.5",
     "@types/crypto-js": "^4.2.2",
     "@types/formidable": "^3.4.6",
-    "@types/node": "^24.10.1",
+    "@types/node": "^25.0.5",
     "cross-env": "^10.1.0",
     "crypto-js": "^4.2.0",
     "dayjs": "^1.11.19",
     "dotenv": "^17.2.3",
     "formidable": "^3.5.4",
-    "openai": "6.10.0",
+    "openai": "6.16.0",
     "pm2": "^6.0.14",
     "rimraf": "^6.1.2",
     "typescript": "^5.9.3",
-    "vite": "^7.2.6"
+    "vite": "^7.3.1"
   },
   "dependencies": {
     "@kevisual/logger": "^0.0.4",
     "@kevisual/permission": "^0.0.3",
-    "@kevisual/query": "^0.0.31"
-  },
-  "scripts": {
-    "build": "npm run clean && bun bun.config.mjs",
-    "dev": "bun run --watch bun.config.mjs",
-    "test": "tsx  test/**/*.ts",
-    "clean": "rm -rf dist",
-    "pub": "envision pack -p -u"
+    "@kevisual/query": "^0.0.35"
   }
 }

package/src/jimeng/core.ts

@@ -1,51 +1,121 @@
-import { adapter } from '@kevisual/query/query'
-export type CoreOpts = {
-  baseURL?: string;
-  token?: string;
+import { Result } from '@kevisual/query'
+export interface JimengOptions {
+  /** API密钥，用于认证请求 */
+  apiKey: string;
+  /** API基础URL */
+  baseUrl: string;
+  /** 请求超时时间（毫秒） */
+  timeout: number;
 }
-export class Core {
-  baseURL: string = 'https://jimeng-api.kevisual.cn/v1';
-  token?: string;
-  constructor(opts: CoreOpts = {}) {
-    console.log("Core initialized");
-    if (opts.baseURL) {
-      this.baseURL = opts.baseURL;
-    }
-    if (opts.token) {
-      this.token = opts.token;
-    }
+
+export interface JimengGenerateOptions {
+  /** 图片生成提示词 */
+  prompt: string;
+  /** 使用的模型版本，默认 jimeng-4.0 */
+  model?: string;
+  /** 图片比例，默认 1:1 */
+  ratio?: string;
+  /** 图片分辨率，默认 2k */
+  resolution?: string;
+}
+
+interface JimengResponse {
+  /** 请求创建时间戳 */
+  created: number;
+  /** 生成的图片列表 */
+  data: Array<{
+    /** 图片URL */
+    url: string;
+  }>;
+}
+
+export class JimengService {
+  private apiKey: string;
+  private baseUrl: string;
+  private timeout: number;
+
+  constructor(options: JimengOptions) {
+    this.apiKey = options.apiKey;
+    this.baseUrl = options.baseUrl || 'https://jimeng-api.kevisual.cn/v1';
+    this.timeout = options.timeout;
   }
-  makeHeader() {
-    return {
-      Authorization: this.token ? `Bearer ${this.token}` : undefined,
-      'Content-Type': 'application/json'
+
+  async generateImage(options: JimengGenerateOptions): Promise<Result<JimengResponse>> {
+    const {
+      prompt,
+      model = 'jimeng-4.6',
+      ratio = '1:1',
+      resolution = '2k'
+    } = options;
+
+    try {
+      const controller = new AbortController();
+      const timeoutId = setTimeout(() => controller.abort(), this.timeout);
+
+      const response = await fetch(`${this.baseUrl}/images/generations`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${this.apiKey}`,
+        },
+        body: JSON.stringify({
+          model,
+          prompt,
+          ratio,
+          resolution,
+        }),
+        signal: controller.signal,
+      });
+
+      clearTimeout(timeoutId);
+
+      if (!response.ok) {
+        throw new Error(`jimeng API error: ${response.status} ${response.statusText}`);
+      }
+
+      const result = await response.json() as JimengResponse;
+      return { code: 200, data: result };
+    } catch (error: any) {
+      return { code: 500, message: error.message || 'Unknown error' };
     }
   }
-  generateImage({ model = 'jimeng-4.0', prompt, resolution = '2k' }: ImageOptions) {
-    const url = `${this.baseURL}/images/generations`;
-    return adapter({
-      url,
-      headers: this.makeHeader(),
-      body: {
-        model,
-        prompt,
-        resolution
+
+  async downloadImage(url: string): Promise<Uint8Array> {
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), this.timeout);
+
+    try {
+      const response = await fetch(url, {
+        signal: controller.signal,
+      });
+
+      clearTimeout(timeoutId);
+
+      if (!response.ok) {
+        throw new Error(`Failed to download image: ${response.statusText}`);
+      }
+
+      const arrayBuffer = await response.arrayBuffer();
+      return new Uint8Array(arrayBuffer);
+    } catch (error: any) {
+      clearTimeout(timeoutId);
+      if (error.name === 'AbortError') {
+        throw new Error('Image download timeout');
       }
-    });
+      throw error;
+    }
+  }
+  /** 获取图片过期时间 */
+  async getExpiredTime(url: string): Promise<{ expiredAt: number, expired: boolean }> {
+    // https://p3-dreamina-sign.byteimg.com/tos-cn-i-tb4s082cfz/c018e06ee6654dd78ccacb29eff4744e~tplv-tb4s082cfz-aigc_resize:0:0.png?lk3s=43402efa&x-expires=1767852000&x-signature=34yf37N955BP37eLaYEzKeLQn0Q%3D&format=.png
+    const urlObj = new URL(url);
+    let expires = urlObj.searchParams.get('x-expires');
+    if (!expires) {
+      expires = '0';
+    }
+    const expiredAt = parseInt(expires) * 1000;
+    const expired = Date.now() > expiredAt;
+    return { expiredAt, expired };
   }
 }
 
-export type ImageOptions = {
-  model?: string;
-  prompt: string;
-  /**
-   * 宽高比，如 "16:9", "4:3", "1:1" 等
-   */
-  ratio?: string;
-  /**
-   *  
-   * 图片分辨率，如 "1024x768", "512x512" 等
-   *  4k 2k
-   */
-  resolution?: string;
-}

package/src/provider/chat-adapter/mimo.ts

@@ -0,0 +1,10 @@
+import { BaseChat, type BaseChatOptions } from "../chat.ts";
+
+type MimoOptions = Partial<BaseChatOptions>;
+export class MimoChat extends BaseChat {
+  static BASE_URL = 'https://api.xiaomimimo.com/v1';
+  constructor(options: MimoOptions) {
+    const baseURL = options.baseURL || MimoChat.BASE_URL;
+    super({ ...(options as BaseChatOptions), baseURL: baseURL });
+  }
+}

package/src/provider/chat-adapter/ollama.ts

@@ -25,11 +25,6 @@ export class Ollama extends BaseChat {
     const baseURL = options.baseURL || Ollama.BASE_URL;
     super({ ...(options as BaseChatOptions), baseURL: baseURL });
   }
-  async chat(messages: ChatMessage[], options?: ChatMessageOptions) {
-    const res = await super.chat(messages, options);
-    console.log('thunk', this.getChatUsage());
-    return res;
-  }
   /**
    * 获取模型列表
    * @returns

package/src/provider/chat-adapter/siliconflow.ts

@@ -1,5 +1,4 @@
 import { BaseChat, BaseChatOptions } from '../core/chat.ts';
-import { OpenAI } from 'openai';
 import type { ChatMessage, ChatMessageOptions } from '../core/index.ts';
 
 export type SiliconFlowOptions = Partial<BaseChatOptions>;
@@ -33,8 +32,4 @@ export class SiliconFlow extends BaseChat {
   async getUsageInfo(): Promise<SiliconFlowUsageResponse> {
     return this.get('/user/info');
   }
-  async chat(messages: OpenAI.Chat.Completions.ChatCompletionMessageParam[], options?: ChatMessageOptions) {
-    const res = await super.chat(messages, options);
-    return res;
-  }
 }