@kevisual/ai 0.0.11 → 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,27 +141,24 @@ 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>;
+        data?: any;
+    }): Promise<Response>;
+    get<T = any>(url?: string, opts?: {
+        headers?: Record<string, string>;
+    }): Promise<T>;
     /**
      * 聊天
      */
     chat(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatMessageComplete>;
     chatStream(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatStream>;
-    /**
-     * 测试
-     */
-    test(): Promise<OpenAI$1.Chat.Completions.ChatCompletion>;
     /**
      * 获取聊天使用情况
      * @returns
@@ -213,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;
@@ -220,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;
@@ -229,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;
@@ -295,9 +379,7 @@ declare class KnowledgeBase extends BaseChat {
 declare class SiliconFlowKnowledge extends KnowledgeBase {
     static BASE_URL: string;
     constructor(options: KnowledgeOptions);
-    rerank(data: RerankOptions): Promise<{
-        _request_id?: string | null;
-    }>;
+    rerank(data: RerankOptions): Promise<Response>;
 }
 type RerankOptions = {
     model: string;
@@ -350,7 +432,7 @@ type AIConfig = {
     description?: string;
     models: AIModel[];
     secretKeys: SecretKey[];
-    permission?: Permission$1;
+    permission?: Permission;
     filter?: {
         objectKey: string;
         type: 'array' | 'object';
@@ -435,7 +517,7 @@ declare class AIConfigParser {
         title?: string;
         description?: string;
         models?: AIModel[];
-        permission?: Permission;
+        permission?: _kevisual_permission.Permission;
         filter?: {
             objectKey: string;
             type: "array" | "object";
@@ -445,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 };