@weisiren000/oiiai 0.1.2 → 0.1.3

package/dist/index.d.ts

@@ -6,20 +6,56 @@ interface ChatMessage {
     content: string;
 }
 /**
- * 思考/推理配置
- * - effort: OpenAI o1/o3, Grok 使用
- * - maxTokens: Anthropic Claude, Gemini, Qwen 使用
+ * 统一的思考努力程度
+ * - off: 关闭思考
+ * - low: 快速思考，适合简单问题
+ * - medium: 平衡模式（默认）
+ * - high: 深度思考，适合复杂问题
+ */
+type ReasoningEffort = 'off' | 'low' | 'medium' | 'high';
+/**
+ * 统一的思考/推理配置
+ *
+ * 只需设置 effort，库会自动转换为各 Provider 需要的格式：
+ * - OpenRouter: effort 参数
+ * - Gemini: reasoning_effort 参数
+ * - Groq: reasoning_effort 参数
+ * - ModelScope: enable_thinking 参数
+ *
+ * @example
+ * ```ts
+ * // 简单用法 - 只设置 effort
+ * { effort: 'high' }
+ *
+ * // 高级用法 - 精细控制
+ * { effort: 'high', budgetTokens: 8000 }
+ * ```
  */
 interface ReasoningConfig {
-    /** 思考努力程度 */
-    effort?: 'high' | 'medium' | 'low';
-    /** 思考最大 token 数 */
-    maxTokens?: number;
-    /** 是否在响应中排除思考内容 */
+    /**
+     * 思考努力程度（推荐使用）
+     * - 'off': 关闭思考
+     * - 'low': 快速思考
+     * - 'medium': 平衡模式
+     * - 'high': 深度思考
+     */
+    effort?: ReasoningEffort;
+    /**
+     * 思考 token 预算（可选，精细控制）
+     * 设置后会覆盖 effort 对应的默认值
+     * 仅部分 Provider 支持（OpenRouter）
+     */
+    budgetTokens?: number;
+    /**
+     * 是否在响应中排除思考内容
+     * 仅部分 Provider 支持（OpenRouter）
+     */
     exclude?: boolean;
-    /** 是否启用思考 */
-    enabled?: boolean;
 }
+/**
+ * effort 到 token 预算的默认映射
+ */
+declare const EFFORT_TOKEN_MAP: Record<ReasoningEffort, number>;
 interface ChatOptions {
     /** 模型 ID */
     model: string;
@@ -130,6 +166,54 @@ declare abstract class BaseProvider implements AIProvider {
     askWithSystem(model: string, systemPrompt: string, userMessage: string, options?: Omit<ChatOptions, 'model' | 'messages'>): Promise<string>;
 }
 
+/**
+ * 统一的 Provider 工厂
+ * 让所有 Provider 使用方式一致
+ */
+
+/**
+ * 支持的 Provider 类型
+ */
+type ProviderType = 'openrouter' | 'gemini' | 'groq' | 'huggingface' | 'modelscope' | 'deepseek';
+/**
+ * 统一的 Provider 配置
+ */
+interface ProviderConfig {
+    /** Provider 类型 */
+    provider: ProviderType;
+    /** API Key */
+    apiKey: string;
+    /** 自定义 API 地址（可选） */
+    baseUrl?: string;
+}
+/**
+ * 创建 AI Provider 实例
+ *
+ * @example
+ * ```ts
+ * // 所有 Provider 使用相同方式创建
+ * const openrouter = createProvider({ provider: 'openrouter', apiKey: 'xxx' });
+ * const gemini = createProvider({ provider: 'gemini', apiKey: 'xxx' });
+ * const groq = createProvider({ provider: 'groq', apiKey: 'xxx' });
+ *
+ * // 之后使用方式完全一致
+ * const answer = await openrouter.ask('gpt-4o', '你好');
+ * const answer = await gemini.ask('gemini-2.5-flash', '你好');
+ * ```
+ */
+declare function createProvider(config: ProviderConfig): AIProvider;
+/**
+ * 快捷创建函数
+ */
+declare const ai: {
+    openrouter: (apiKey: string, baseUrl?: string) => AIProvider;
+    gemini: (apiKey: string, baseUrl?: string) => AIProvider;
+    groq: (apiKey: string, baseUrl?: string) => AIProvider;
+    huggingface: (apiKey: string, baseUrl?: string) => AIProvider;
+    modelscope: (apiKey: string, baseUrl?: string) => AIProvider;
+    deepseek: (apiKey: string, baseUrl?: string) => AIProvider;
+};
+
 /**
  * OpenRouter Provider 实现
  */
@@ -172,4 +256,129 @@ declare class OpenRouterProvider extends BaseProvider {
     listModels(): Promise<OpenRouterModelInfo[]>;
 }
 
-export { type AIProvider, BaseProvider, type ChatMessage, type ChatOptions, type ChatResult, type ModelInfo, type ModelPricing, type OpenRouterModelInfo, OpenRouterProvider, type ReasoningConfig, type StreamChunk, type TokenUsage };
+/**
+ * ModelScope Provider 实现
+ * 使用 OpenAI 兼容 API
+ */
+
+/**
+ * ModelScope Provider 配置
+ */
+interface ModelScopeConfig {
+    apiKey: string;
+    baseUrl?: string;
+}
+/**
+ * ModelScope Provider
+ * 兼容 OpenAI API 格式，支持 DeepSeek 等模型
+ */
+declare class ModelScopeProvider extends BaseProvider {
+    readonly name = "modelscope";
+    private apiKey;
+    private baseUrl;
+    constructor(config: ModelScopeConfig | string);
+    /**
+     * 发送聊天请求（非流式）
+     */
+    chat(options: ChatOptions): Promise<ChatResult>;
+    /**
+     * 发送流式聊天请求
+     */
+    chatStream(options: ChatOptions): AsyncGenerator<StreamChunk, void, unknown>;
+}
+
+/**
+ * HuggingFace Provider 实现
+ * 使用 HuggingFace Router API (OpenAI 兼容)
+ */
+
+/**
+ * HuggingFace Provider 配置
+ */
+interface HuggingFaceConfig {
+    apiKey: string;
+    baseUrl?: string;
+}
+/**
+ * HuggingFace Provider
+ * 使用 HuggingFace Router，兼容 OpenAI API 格式
+ */
+declare class HuggingFaceProvider extends BaseProvider {
+    readonly name = "huggingface";
+    private apiKey;
+    private baseUrl;
+    constructor(config: HuggingFaceConfig | string);
+    /**
+     * 发送聊天请求（非流式）
+     */
+    chat(options: ChatOptions): Promise<ChatResult>;
+    /**
+     * 发送流式聊天请求
+     */
+    chatStream(options: ChatOptions): AsyncGenerator<StreamChunk, void, unknown>;
+}
+
+/**
+ * Groq Provider 实现
+ * 使用 Groq API (OpenAI 兼容)
+ */
+
+/**
+ * Groq Provider 配置
+ */
+interface GroqConfig {
+    apiKey: string;
+    baseUrl?: string;
+}
+/**
+ * Groq Provider
+ * 兼容 OpenAI API 格式，支持 reasoning_effort 参数
+ */
+declare class GroqProvider extends BaseProvider {
+    readonly name = "groq";
+    private apiKey;
+    private baseUrl;
+    constructor(config: GroqConfig | string);
+    /**
+     * 发送聊天请求（非流式）
+     */
+    chat(options: ChatOptions): Promise<ChatResult>;
+    /**
+     * 发送流式聊天请求
+     */
+    chatStream(options: ChatOptions): AsyncGenerator<StreamChunk, void, unknown>;
+}
+
+/**
+ * Gemini Provider 实现
+ * 使用 Google Gemini API (OpenAI 兼容格式)
+ * https://ai.google.dev/gemini-api/docs/openai
+ */
+
+/**
+ * Gemini Provider 配置
+ */
+interface GeminiConfig {
+    apiKey: string;
+    baseUrl?: string;
+}
+/**
+ * Gemini Provider
+ * 使用 Google Gemini API，兼容 OpenAI 格式
+ */
+declare class GeminiProvider extends BaseProvider {
+    readonly name = "gemini";
+    private apiKey;
+    private baseUrl;
+    constructor(config: GeminiConfig | string);
+    /**
+     * 发送聊天请求（非流式）
+     */
+    chat(options: ChatOptions): Promise<ChatResult>;
+    /**
+     * 发送流式聊天请求
+     */
+    chatStream(options: ChatOptions): AsyncGenerator<StreamChunk, void, unknown>;
+}
+
+export { type AIProvider, BaseProvider, type ChatMessage, type ChatOptions, type ChatResult, EFFORT_TOKEN_MAP, GeminiProvider, GroqProvider, HuggingFaceProvider, type ModelInfo, type ModelPricing, ModelScopeProvider, type OpenRouterModelInfo, OpenRouterProvider, type ProviderConfig, type ProviderType, type ReasoningConfig, type ReasoningEffort, type StreamChunk, type TokenUsage, ai, createProvider };