@kevisual/ai 0.0.12 → 0.0.13

package/dist/ai-provider.d.ts

@@ -1,7 +1,8 @@
 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 { Permission as Permission$1 } from '@kevisual/permission';
+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> & {
@@ -34,6 +35,77 @@ interface BaseChatUsageInterface {
 }
 type ChatStream = AsyncGenerator<ChatMessageComplete, void, unknown>;
 
+declare class AIUtils {
+    /**
+     * 从 Markdown 代码块中提取 JSON
+     * @param str 包含 JSON 的字符串
+     * @returns 解析后的对象或 null
+     */
+    extractJsonFromMarkdown(str: string): any | null;
+    /**
+     * 从 Markdown 代码块中提取代码
+     * @param str Markdown 字符串
+     * @param language 语言类型，不指定则返回所有代码块
+     * @returns 提取的代码字符串或数组
+     */
+    extractCodeFromMarkdown(str: string, language?: string): string | string[] | null;
+    /**
+     * 清理 AI 响应中的多余空白和格式
+     * @param str 原始字符串
+     * @returns 清理后的字符串
+     */
+    cleanResponse(str: string): string;
+    /**
+     * 从 AI 响应中提取标签
+     * @param str 响应字符串
+     * @returns 标签数组
+     */
+    extractTags(str: string): string[];
+    /**
+     * 从文本中提取 URL
+     * @param str 文本字符串
+     * @returns URL 数组
+     */
+    extractUrls(str: string): string[];
+    /**
+     * 分割长文本为指定 token 数量的块
+     * @param text 原始文本
+     * @param maxTokens 每块最大 token 数（粗略估算：1 token ≈ 4 字符）
+     * @returns 文本块数组
+     */
+    chunkText(text: string, maxTokens?: number): string[];
+    /**
+     * 移除 AI 响应中的思考过程（thinking 标签）
+     * @param str 响应字符串
+     * @returns 清理后的字符串
+     */
+    removeThinkingTags(str: string): string;
+    /**
+     * 转义特殊字符用于 AI 提示词
+     * @param str 原始字符串
+     * @returns 转义后的字符串
+     */
+    escapeForPrompt(str: string): string;
+    /**
+     * 统计文本的大致 token 数量
+     * @param text 文本
+     * @returns 估算的 token 数量
+     */
+    estimateTokens(text: string): number;
+    /**
+     * 从响应中提取结构化数据（key: value 格式）
+     * @param str 响应字符串
+     * @returns 键值对对象
+     */
+    extractKeyValuePairs(str: string): Record<string, string>;
+    /**
+     * 验证 AI 响应是否完整（检查截断）
+     * @param str 响应字符串
+     * @returns 是否完整
+     */
+    isResponseComplete(str: string): boolean;
+}
+
 type BaseChatOptions<T = Record<string, any>> = {
     /**
      * 默认baseURL
@@ -56,7 +128,6 @@ type BaseChatOptions<T = Record<string, any>> = {
      */
     stream?: boolean;
 } & T;
-declare const getIsBrowser: () => boolean;
 declare class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
     /**
      * 默认baseURL
@@ -70,18 +141,11 @@ declare class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
      * 默认apiKey
      */
     apiKey: string;
-    /**
-     * 是否在浏览器中使用
-     */
-    isBrowser: boolean;
-    /**
-     * openai实例
-     */
-    openai: OpenAI$1;
     prompt_tokens: number;
     total_tokens: number;
     completion_tokens: number;
     responseText: string;
+    utils: AIUtils;
     constructor(options: BaseChatOptions);
     post(url?: string, opts?: {
         headers?: Record<string, string>;
@@ -217,6 +281,18 @@ declare class BailianChat extends BaseChat {
     constructor(options: BailianOptions);
 }
 
+type ZhipuOptions = Partial<BaseChatOptions>;
+declare class Zhipu extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: ZhipuOptions);
+}
+
+type KimiOptions = Partial<BaseChatOptions>;
+declare class Kimi extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: KimiOptions);
+}
+
 declare const OllamaProvider: typeof Ollama;
 declare const SiliconFlowProvider: typeof SiliconFlow;
 declare const CustomProvider: typeof Custom;
@@ -224,6 +300,8 @@ declare const VolcesProvider: typeof Volces;
 declare const DeepSeekProvider: typeof DeepSeek;
 declare const ModelScopeProvider: typeof ModelScope;
 declare const BailianProvider: typeof BailianChat;
+declare const ZhipuProvider: typeof Zhipu;
+declare const KimiProvider: typeof Kimi;
 declare const ChatProviderMap: {
     Ollama: typeof Ollama;
     SiliconFlow: typeof SiliconFlow;
@@ -233,6 +311,8 @@ declare const ChatProviderMap: {
     ModelScope: typeof ModelScope;
     BaseChat: typeof BaseChat;
     Bailian: typeof BailianChat;
+    Zhipu: typeof Zhipu;
+    Kimi: typeof Kimi;
 };
 type ProviderManagerConfig = {
     provider: string;
@@ -352,7 +432,7 @@ type AIConfig = {
     description?: string;
     models: AIModel[];
     secretKeys: SecretKey[];
-    permission?: Permission$1;
+    permission?: Permission;
     filter?: {
         objectKey: string;
         type: 'array' | 'object';
@@ -437,7 +517,7 @@ declare class AIConfigParser {
         title?: string;
         description?: string;
         models?: AIModel[];
-        permission?: Permission;
+        permission?: _kevisual_permission.Permission;
         filter?: {
             objectKey: string;
             type: "array" | "object";
@@ -447,5 +527,5 @@ declare class AIConfigParser {
     };
 }
 
-export { AIConfigParser, BailianProvider, BaseChat, ChatProviderMap, CustomProvider, DeepSeekProvider, KnowledgeBase, ModelScopeProvider, OllamaProvider, ProviderManager, SiliconFlowKnowledge, SiliconFlowProvider, VolcesProvider, decryptAES, encryptAES, getIsBrowser, readStream };
+export { AIConfigParser, BailianProvider, BaseChat, ChatProviderMap, CustomProvider, DeepSeekProvider, KimiProvider, KnowledgeBase, ModelScopeProvider, OllamaProvider, ProviderManager, SiliconFlowKnowledge, SiliconFlowProvider, VolcesProvider, ZhipuProvider, decryptAES, encryptAES, readStream };
 export type { AIConfig, AIModel, BaseChatInterface, BaseChatOptions, BaseChatUsageInterface, ChatMessage, ChatMessageComplete, ChatMessageOptions, ChatMessageStream, ChatStream, EmbeddingMessage, EmbeddingMessageComplete, GetProviderOpts, KnowledgeOptions, ProviderResult, RerankOptions, SecretKey };

package/dist/ai-provider.js

@@ -1235,31 +1235,131 @@ var require_enc_utf8 = __commonJS((exports, module) => {
   });
 });
 
-// src/provider/core/chat.ts
-var getIsBrowser = () => {
-  try {
-    return typeof window !== "undefined" && typeof window.document !== "undefined";
-  } catch (e) {
-    return false;
+// src/provider/core/utils/index.ts
+class AIUtils {
+  extractJsonFromMarkdown(str) {
+    const jsonRegex = /```json\s*([\s\S]*?)\s*```/;
+    const match = str.match(jsonRegex);
+    let jsonStr = match && match[1] ? match[1] : str;
+    try {
+      return JSON.parse(jsonStr);
+    } catch {
+      return null;
+    }
   }
-};
+  extractCodeFromMarkdown(str, language) {
+    if (language) {
+      const regex2 = new RegExp(`\`\`\`${language}\\s*([\\s\\S]*?)\\s*\`\`\``, "g");
+      const matches2 = str.match(regex2);
+      if (!matches2)
+        return null;
+      return matches2.map((m) => m.replace(new RegExp(`\`\`\`${language}\\s*|\\s*\`\`\``, "g"), "").trim());
+    }
+    const regex = /```[\w]*\s*([\s\S]*?)\s*```/g;
+    const matches = [...str.matchAll(regex)];
+    if (matches.length === 0)
+      return null;
+    return matches.map((m) => m[1].trim());
+  }
+  cleanResponse(str) {
+    return str.trim().replace(/\n{3,}/g, `
+
+`).replace(/[ \t]+$/gm, "");
+  }
+  extractTags(str) {
+    const tagPatterns = [
+      /#(\w+)/g,
+      /\[(\w+)\]/g,
+      /tags?:\s*\[([^\]]+)\]/gi
+    ];
+    const tags = new Set;
+    for (const pattern of tagPatterns) {
+      const matches = str.matchAll(pattern);
+      for (const match of matches) {
+        if (match[1]) {
+          const extracted = match[1].split(/[,;]/).map((t) => t.trim()).filter(Boolean);
+          extracted.forEach((tag) => tags.add(tag));
+        }
+      }
+    }
+    return Array.from(tags);
+  }
+  extractUrls(str) {
+    const urlRegex = /(https?:\/\/[^\s]+)/g;
+    const matches = str.match(urlRegex);
+    return matches || [];
+  }
+  chunkText(text, maxTokens = 1000) {
+    const chunkSize = maxTokens * 4;
+    const chunks = [];
+    const paragraphs = text.split(/\n\n+/);
+    let currentChunk = "";
+    for (const paragraph of paragraphs) {
+      if ((currentChunk + paragraph).length > chunkSize && currentChunk) {
+        chunks.push(currentChunk.trim());
+        currentChunk = paragraph;
+      } else {
+        currentChunk += (currentChunk ? `
+
+` : "") + paragraph;
+      }
+    }
+    if (currentChunk) {
+      chunks.push(currentChunk.trim());
+    }
+    return chunks;
+  }
+  removeThinkingTags(str) {
+    return str.replace(/<thinking>[\s\S]*?<\/thinking>/gi, "").replace(/\[thinking\][\s\S]*?\[\/thinking\]/gi, "").trim();
+  }
+  escapeForPrompt(str) {
+    return str.replace(/\\/g, "\\\\").replace(/`/g, "\\`").replace(/\$/g, "\\$");
+  }
+  estimateTokens(text) {
+    const chineseChars = (text.match(/[\u4e00-\u9fa5]/g) || []).length;
+    const otherChars = text.length - chineseChars;
+    return Math.ceil(chineseChars / 1.5 + otherChars / 4);
+  }
+  extractKeyValuePairs(str) {
+    const result = {};
+    const lines = str.split(`
+`);
+    for (const line of lines) {
+      const match = line.match(/^([^:：]+)[：:]\s*(.+)$/);
+      if (match) {
+        const key = match[1].trim();
+        const value = match[2].trim();
+        result[key] = value;
+      }
+    }
+    return result;
+  }
+  isResponseComplete(str) {
+    const incompleteSigns = [
+      /```[\w]*\s*[\s\S]*?(?<!```)$/,
+      /\{[\s\S]*(?<!\})$/,
+      /\[[\s\S]*(?<!\])$/,
+      /\.{3,}$/
+    ];
+    return !incompleteSigns.some((pattern) => pattern.test(str.trim()));
+  }
+}
 
+// src/provider/core/chat.ts
 class BaseChat {
   baseURL;
   model;
   apiKey;
-  isBrowser;
-  openai;
   prompt_tokens;
   total_tokens;
   completion_tokens;
   responseText;
+  utils;
   constructor(options) {
     this.baseURL = options.baseURL;
     this.model = options.model;
     this.apiKey = options.apiKey;
-    const DEFAULT_IS_BROWSER = getIsBrowser();
-    this.isBrowser = options.isBrowser ?? DEFAULT_IS_BROWSER;
+    this.utils = new AIUtils;
   }
   post(url = "", opts = {}) {
     let _url = url.startsWith("http") ? url : this.baseURL + url;
@@ -1478,6 +1578,24 @@ class BailianChat extends BaseChat {
   }
 }
 
+// src/provider/chat-adapter/zhipu.ts
+class Zhipu extends BaseChat {
+  static BASE_URL = "https://open.bigmodel.cn/api/paas/v4/";
+  constructor(options) {
+    const baseURL = options.baseURL || Zhipu.BASE_URL;
+    super({ ...options, baseURL });
+  }
+}
+
+// src/provider/chat-adapter/kimi.ts
+class Kimi extends BaseChat {
+  static BASE_URL = "https://api.moonshot.cn/v1/";
+  constructor(options) {
+    const baseURL = options.baseURL || Kimi.BASE_URL;
+    super({ ...options, baseURL });
+  }
+}
+
 // src/provider/chat.ts
 var OllamaProvider = Ollama;
 var SiliconFlowProvider = SiliconFlow;
@@ -1486,6 +1604,8 @@ var VolcesProvider = Volces;
 var DeepSeekProvider = DeepSeek;
 var ModelScopeProvider = ModelScope;
 var BailianProvider = BailianChat;
+var ZhipuProvider = Zhipu;
+var KimiProvider = Kimi;
 var ChatProviderMap = {
   Ollama: OllamaProvider,
   SiliconFlow: SiliconFlowProvider,
@@ -1494,7 +1614,9 @@ var ChatProviderMap = {
   DeepSeek: DeepSeekProvider,
   ModelScope: ModelScopeProvider,
   BaseChat,
-  Bailian: BailianProvider
+  Bailian: BailianProvider,
+  Zhipu: ZhipuProvider,
+  Kimi: KimiProvider
 };
 
 class ProviderManager {
@@ -1703,9 +1825,9 @@ class AIConfigParser {
 }
 export {
   readStream,
-  getIsBrowser,
   encryptAES,
   decryptAES,
+  ZhipuProvider,
   VolcesProvider,
   SiliconFlowProvider,
   SiliconFlowKnowledge,
@@ -1713,6 +1835,7 @@ export {
   OllamaProvider,
   ModelScopeProvider,
   KnowledgeBase,
+  KimiProvider,
   DeepSeekProvider,
   CustomProvider,
   ChatProviderMap,

package/package.json

@@ -1,12 +1,11 @@
 {
   "name": "@kevisual/ai",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "description": "AI Center Services",
   "main": "index.js",
   "basename": "/root/ai-center-services",
   "app": {
     "entry": "dist/app.mjs",
-    "key": "ai-center-services",
     "type": "system-app"
   },
   "files": [
@@ -28,7 +27,7 @@
   ],
   "author": "abearxiong <xiongxiao@xiongxiao.me> (https://www.xiongxiao.me)",
   "license": "MIT",
-  "packageManager": "pnpm@10.23.0",
+  "packageManager": "pnpm@10.24.0",
   "type": "module",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
@@ -58,14 +57,14 @@
     "@kevisual/mark": "0.0.7",
     "@kevisual/router": "0.0.33",
     "@kevisual/types": "^0.0.10",
-    "@kevisual/use-config": "^1.0.19",
+    "@kevisual/use-config": "^1.0.21",
     "@types/bun": "^1.3.3",
     "@types/crypto-js": "^4.2.2",
     "@types/formidable": "^3.4.6",
     "@types/lodash-es": "^4.17.12",
     "@types/node": "^24.10.1",
     "@vitejs/plugin-basic-ssl": "^2.1.0",
-    "cookie": "^1.0.2",
+    "cookie": "^1.1.1",
     "cross-env": "^10.1.0",
     "crypto-js": "^4.2.0",
     "dayjs": "^1.11.19",
@@ -75,17 +74,18 @@
     "json5": "^2.2.3",
     "lodash-es": "^4.17.21",
     "openai": "6.9.1",
-    "pm2": "^6.0.13",
+    "pm2": "^6.0.14",
     "rimraf": "^6.1.2",
     "rollup": "^4.53.3",
-    "rollup-plugin-dts": "^6.2.3",
+    "rollup-plugin-dts": "^6.3.0",
     "sequelize": "^6.37.7",
     "tape": "^5.9.0",
     "tiktoken": "^1.0.22",
     "typescript": "^5.9.3",
-    "vite": "^7.2.4"
+    "vite": "^7.2.6"
   },
   "dependencies": {
-    "@kevisual/logger": "^0.0.4"
+    "@kevisual/logger": "^0.0.4",
+    "@kevisual/permission": "^0.0.3"
   }
 }

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

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

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

@@ -0,0 +1,10 @@
+import { BaseChat, BaseChatOptions } from '../core/chat.ts';
+
+export type ZhipuOptions = Partial<BaseChatOptions>;
+export class Zhipu extends BaseChat {
+  static BASE_URL = 'https://open.bigmodel.cn/api/paas/v4/';
+  constructor(options: ZhipuOptions) {
+    const baseURL = options.baseURL || Zhipu.BASE_URL;
+    super({ ...(options as BaseChatOptions), baseURL: baseURL });
+  }
+}

package/src/provider/chat.ts

@@ -8,6 +8,8 @@ import { Volces } from './chat-adapter/volces.ts';
 import { DeepSeek } from './chat-adapter/deepseek.ts';
 import { ModelScope } from './chat-adapter/model-scope.ts';
 import { BailianChat } from './chat-adapter/dashscope.ts';
+import { Zhipu } from './chat-adapter/zhipu.ts';
+import { Kimi } from './chat-adapter/kimi.ts';
 
 import { ChatMessage } from './core/type.ts';
 
@@ -18,6 +20,8 @@ export const VolcesProvider = Volces;
 export const DeepSeekProvider = DeepSeek;
 export const ModelScopeProvider = ModelScope;
 export const BailianProvider = BailianChat;
+export const ZhipuProvider = Zhipu;
+export const KimiProvider = Kimi;
 
 export const ChatProviderMap = {
   Ollama: OllamaProvider,
@@ -28,6 +32,8 @@ export const ChatProviderMap = {
   ModelScope: ModelScopeProvider,
   BaseChat: BaseChat,
   Bailian: BailianProvider,
+  Zhipu: ZhipuProvider,
+  Kimi: KimiProvider,
 };
 
 type ProviderManagerConfig = {

package/src/provider/core/chat.ts

@@ -9,6 +9,7 @@ import type {
   EmbeddingMessage,
   EmbeddingMessageComplete,
 } from './type.ts';
+import { AIUtils } from './utils/index.ts';
 
 export type BaseChatOptions<T = Record<string, any>> = {
   /**
@@ -32,14 +33,7 @@ export type BaseChatOptions<T = Record<string, any>> = {
    */
   stream?: boolean;
 } & T;
-export const getIsBrowser = () => {
-  try {
-    // 检查是否存在window对象
-    return typeof window !== 'undefined' && typeof window.document !== 'undefined';
-  } catch (e) {
-    return false;
-  }
-};
+
 export class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
   /**
    * 默认baseURL
@@ -53,32 +47,16 @@ export class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
    * 默认apiKey
    */
   apiKey: string;
-  /**
-   * 是否在浏览器中使用
-   */
-  isBrowser: boolean;
-  /**
-   * openai实例
-   */
-  openai: OpenAI;
-
   prompt_tokens: number;
   total_tokens: number;
   completion_tokens: number;
   responseText: string;
-
+  utils: AIUtils;
   constructor(options: BaseChatOptions) {
     this.baseURL = options.baseURL;
     this.model = options.model;
     this.apiKey = options.apiKey;
-    // @ts-ignore
-    const DEFAULT_IS_BROWSER = getIsBrowser();
-    this.isBrowser = options.isBrowser ?? DEFAULT_IS_BROWSER;
-    // this.openai = new OpenAI({
-    //   apiKey: this.apiKey,
-    //   baseURL: this.baseURL,
-    //   dangerouslyAllowBrowser: options?.dangerouslyAllowBrowser ?? this.isBrowser,
-    // });
+    this.utils = new AIUtils();
   }
   post(url = '', opts: { headers?: Record<string, string>, data?: any } = {}) {
     let _url = url.startsWith('http') ? url : this.baseURL + url;