@kevisual/ai 0.0.18 → 0.0.20

package/dist/ai-provider-browser.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 };