@kevisual/ai 0.0.4

package/dist/ai-provider.d.ts

@@ -0,0 +1,230 @@
+import * as openai_resources_index_mjs from 'openai/resources/index.mjs';
+import OpenAI, { OpenAI as OpenAI$1 } from 'openai';
+
+type ChatMessage = OpenAI.Chat.Completions.ChatCompletionMessageParam;
+type ChatMessageOptions = Partial<OpenAI.Chat.Completions.ChatCompletionCreateParams>;
+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(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatMessageComplete>;
+}
+interface BaseChatUsageInterface {
+    /**
+     * 提示词令牌
+     */
+    prompt_tokens: number;
+    /**
+     *  总令牌
+     */
+    total_tokens: number;
+    /**
+     * 完成令牌
+     */
+    completion_tokens: number;
+}
+type ChatStream = AsyncGenerator<ChatMessageComplete, void, unknown>;
+
+type BaseChatOptions<T = Record<string, any>> = {
+    /**
+     * 默认baseURL
+     */
+    baseURL: string;
+    /**
+     * 默认模型
+     */
+    model?: string;
+    /**
+     * 默认apiKey
+     */
+    apiKey: string;
+    /**
+     * 是否在浏览器中使用
+     */
+    isBrowser?: boolean;
+    /**
+     * 是否流式输出， 默认 false
+     */
+    stream?: boolean;
+} & T;
+declare class BaseChat implements BaseChatInterface, BaseChatUsageInterface {
+    /**
+     * 默认baseURL
+     */
+    baseURL: string;
+    /**
+     * 默认模型
+     */
+    model: string;
+    /**
+     * 默认apiKey
+     */
+    apiKey: string;
+    /**
+     * 是否在浏览器中使用
+     */
+    isBrowser: boolean;
+    /**
+     * openai实例
+     */
+    openai: OpenAI$1;
+    prompt_tokens: number;
+    total_tokens: number;
+    completion_tokens: number;
+    constructor(options: BaseChatOptions);
+    /**
+     * 聊天
+     */
+    chat(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatMessageComplete>;
+    chatStream(messages: ChatMessage[], options?: ChatMessageOptions): Promise<ChatStream>;
+    /**
+     * 测试
+     */
+    test(): Promise<OpenAI$1.Chat.Completions.ChatCompletion>;
+    /**
+     * 获取聊天使用情况
+     * @returns
+     */
+    getChatUsage(): {
+        prompt_tokens: number;
+        total_tokens: number;
+        completion_tokens: number;
+    };
+    getHeaders(headers?: Record<string, string>): {
+        'Content-Type': string;
+        Authorization: string;
+    };
+    /**
+     * 生成embedding 内部
+     * @param text
+     * @returns
+     */
+    generateEmbeddingCore(text: string | string[], options?: EmbeddingMessage): Promise<EmbeddingMessageComplete>;
+}
+
+/**
+ *  for await (const chunk of chatStream) {
+ *   console.log(chunk);
+ *  }
+ * @param chatStream
+ */
+declare const readStream: (chatStream: ChatStream) => Promise<void>;
+
+type OllamaOptions$1 = Partial<BaseChatOptions>;
+type OllamaModel = {
+    name: string;
+    model: string;
+    modified_at: string;
+    size: number;
+    digest: string;
+    details: {
+        parent_model: string;
+        format: string;
+        family: string;
+        families: string[];
+        parameter_size: string;
+        quantization_level: string;
+    };
+};
+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
+     */
+    listModels(): Promise<{
+        models: OllamaModel[];
+    }>;
+    listRunModels(): Promise<{
+        models: OllamaModel[];
+    }>;
+}
+
+type SiliconFlowOptions = Partial<BaseChatOptions>;
+type SiliconFlowUsageData = {
+    id: string;
+    name: string;
+    image: string;
+    email: string;
+    isAdmin: boolean;
+    balance: string;
+    status: 'normal' | 'suspended' | 'expired' | string;
+    introduce: string;
+    role: string;
+    chargeBalance: string;
+    totalBalance: string;
+    category: string;
+};
+type SiliconFlowUsageResponse = {
+    code: number;
+    message: string;
+    status: boolean;
+    data: SiliconFlowUsageData;
+};
+declare class SiliconFlow extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: SiliconFlowOptions);
+    getUsageInfo(): Promise<SiliconFlowUsageResponse>;
+    chat(messages: OpenAI$1.Chat.Completions.ChatCompletionMessageParam[], options?: Partial<OpenAI$1.Chat.Completions.ChatCompletionCreateParams>): Promise<OpenAI$1.Chat.Completions.ChatCompletion>;
+}
+
+type OllamaOptions = BaseChatOptions;
+/**
+ * 自定义模型
+ */
+declare class Custom extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: OllamaOptions);
+}
+
+type VolcesOptions = Partial<BaseChatOptions>;
+declare class Volces extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: VolcesOptions);
+}
+
+type DeepSeekOptions = Partial<BaseChatOptions>;
+declare class DeepSeek extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: DeepSeekOptions);
+}
+
+type ModelScopeOptions = Partial<BaseChatOptions>;
+declare class ModelScope extends BaseChat {
+    static BASE_URL: string;
+    constructor(options: ModelScopeOptions);
+}
+
+declare const OllamaProvider: typeof Ollama;
+declare const SiliconFlowProvider: typeof SiliconFlow;
+declare const CustomProvider: typeof Custom;
+declare const VolcesProvider: typeof Volces;
+declare const DeepSeekProvider: typeof DeepSeek;
+declare const ModelScopeProvider: typeof ModelScope;
+declare const ProviderMap: {
+    Ollama: typeof Ollama;
+    SiliconFlow: typeof SiliconFlow;
+    Custom: typeof Custom;
+    Volces: typeof Volces;
+    DeepSeek: typeof DeepSeek;
+    ModelScope: typeof ModelScope;
+    BaseChat: typeof BaseChat;
+};
+type ProviderManagerConfig = {
+    provider: string;
+    model: string;
+    apiKey: string;
+    baseURL?: string;
+};
+declare class ProviderManager {
+    provider: BaseChat;
+    constructor(config: ProviderManagerConfig);
+    static createProvider(config: ProviderManagerConfig): Promise<BaseChat>;
+    chat(messages: ChatMessage[]): Promise<openai_resources_index_mjs.ChatCompletion>;
+}
+
+export { BaseChat, CustomProvider, DeepSeekProvider, ModelScopeProvider, OllamaProvider, ProviderManager, ProviderMap, SiliconFlowProvider, VolcesProvider, readStream };
+export type { BaseChatInterface, BaseChatOptions, BaseChatUsageInterface, ChatMessage, ChatMessageComplete, ChatMessageOptions, ChatMessageStream, ChatStream, EmbeddingMessage, EmbeddingMessageComplete };