foliko 1.1.93 → 2.0.1

package/src/core/provider-registry.js

@@ -1,159 +0,0 @@
-/**
- * ProviderRegistry - Dynamic LLM Provider Registration
- * 允许插件动态注册 LLM Provider
- */
-
-const { createOpenAICompatible } = require('ai');
-const { logger } = require('../utils/logger');
-
-class ProviderRegistry {
-  constructor() {
-    this._providers = new Map();
-    this._models = new Map();
-  }
-
-  /**
-   * 注册 Provider
-   * @param {string} name - Provider 名称 (如 'openai', 'deepseek')
-   * @param {Object} config - Provider 配置
-   * @param {string} config.baseURL - API base URL
-   * @param {string} [config.apiKey] - API key
-   * @param {string} [config.name] - 显示名称
-   * @param {string} [config.defaultModel] - 默认模型
-   * @param {Object} [config.headers] - 自定义 headers
-   */
-  registerProvider(name, config) {
-    this._providers.set(name, {
-      name: config.name || name,
-      baseURL: config.baseURL,
-      apiKey: config.apiKey || '',
-      defaultModel: config.defaultModel || '',
-      headers: config.headers || {},
-    });
-
-    if (config.defaultModel) {
-      this.registerModel(config.defaultModel, name, {
-        label: config.defaultModel,
-        description: `Default model for ${name}`,
-      });
-    }
-
-    logger.debug(`[ProviderRegistry] Registered provider: ${name}`);
-    return this;
-  }
-
-  /**
-   * 注册模型
-   * @param {string} modelId - 模型 ID
-   * @param {string} provider - Provider 名称
-   * @param {Object} [options] - 模型选项
-   */
-  registerModel(modelId, provider, options = {}) {
-    this._models.set(modelId, { provider, ...options });
-    return this;
-  }
-
-  /**
-   * 获取 Provider 配置
-   */
-  getProvider(name) {
-    return this._providers.get(name);
-  }
-
-  /**
-   * 获取模型配置
-   */
-  getModel(modelId) {
-    return this._models.get(modelId);
-  }
-
-  /**
-   * 获取所有已注册的 Provider 名称
-   */
-  getProviderNames() {
-    return Array.from(this._providers.keys());
-  }
-
-  /**
-   * 检查 Provider 是否已注册
-   */
-  hasProvider(name) {
-    return this._providers.has(name);
-  }
-
-  /**
-   * 创建 AI 客户端
-   * @param {string} providerName - Provider 名称
-   * @param {string} model - 模型 ID
-   * @param {string} [apiKey] - API key (可选，会使用注册时提供的)
-   * @param {string} [baseURL] - Base URL (可选，会使用注册时提供的)
-   */
-  createClient(providerName, model, apiKey, baseURL) {
-    const provider = this._providers.get(providerName);
-    if (!provider) {
-      throw new Error(`Provider '${providerName}' not registered. Use framework.registerProvider() first.`);
-    }
-
-    const finalApiKey = apiKey || provider.apiKey;
-    const finalBaseURL = baseURL || provider.baseURL;
-    const finalModel = model || provider.defaultModel;
-
-    if (!finalApiKey) {
-      throw new Error(`No API key provided for provider '${providerName}'`);
-    }
-
-    const client = createOpenAICompatible({
-      baseURL: finalBaseURL,
-      apiKey: finalApiKey,
-    });
-
-    return client(finalModel);
-  }
-
-  /**
-   * 移除 Provider
-   */
-  unregisterProvider(name) {
-    this._providers.delete(name);
-    // 同时移除该 provider 的所有模型
-    for (const [modelId, modelConfig] of this._models) {
-      if (modelConfig.provider === name) {
-        this._models.delete(modelId);
-      }
-    }
-    return this;
-  }
-}
-
-// 单例
-const _registry = new ProviderRegistry();
-
-/**
- * 获取 Provider Registry 实例
- */
-function getProviderRegistry() {
-  return _registry;
-}
-
-/**
- * 注册默认的 Providers
- */
-function registerDefaultProviders() {
-  const { DEFAULT_PROVIDERS } = require('./provider');
-
-  for (const [key, config] of Object.entries(DEFAULT_PROVIDERS)) {
-    _registry.registerProvider(key, {
-      name: config.name,
-      baseURL: config.baseURL,
-      defaultModel: config.defaultModel,
-    });
-  }
-}
-
-// 初始化时注册默认 providers
-registerDefaultProviders();
-
-module.exports = {
-  ProviderRegistry,
-  getProviderRegistry,
-};

package/src/core/provider.js

@@ -1,156 +0,0 @@
-/**
- * AI Provider 工厂
- * 支持多种 AI 提供商
- */
-
-const { createOpenAICompatible } = require('@ai-sdk/openai-compatible');
-const { createOpenAI } = require('@ai-sdk/openai');
-const { createAnthropic } = require('@ai-sdk/anthropic');
-
-/**
- * 默认提供商配置
- */
-const DEFAULT_PROVIDERS = {
-  openai: {
-    name: 'OpenAI',
-    baseURL: 'https://api.openai.com/v1',
-    defaultModel: 'gpt-4o',
-  },
-  ollama: {
-    name: 'Ollama',
-    baseURL: 'http://localhost:11434/v1',
-    defaultModel: 'llama3',
-  },
-  lmstudio: {
-    name: 'LM Studio',
-    baseURL: 'http://localhost:1234/v1',
-    defaultModel: 'local-model',
-  },
-  deepseek: {
-    name: 'DeepSeek',
-    baseURL: 'https://api.deepseek.com/v1',
-    defaultModel: 'deepseek-chat',
-  },
-  anthropic: {
-    name: 'Anthropic',
-    baseURL: 'https://api.anthropic.com/v1',
-    defaultModel: 'claude-sonnet-4-20250514',
-  },
-  minimax: {
-    name: 'MiniMax',
-    baseURL: 'https://api.minimaxi.com/v1',
-    defaultModel: 'MiniMax-M2.7',
-  },
-};
-
-/**
- * 需要禁用 temperature 的模型（thinking mode 模型）
- */
-const THINKING_MODELS = [
-  'deepseek-v4-pro',
-  'deepseek-v4-flash',
-  'deepseek-reasoner',
-];
-
-/**
- * 检查是否是 thinking mode 模型
- * @param {string} model - 模型名称
- * @returns {boolean}
- */
-function isThinkingModel(model) {
-  if (!model) return false;
-  const lowerModel = model.toLowerCase();
-  return THINKING_MODELS.some(
-    (tm) => lowerModel.includes(tm) || lowerModel === tm
-  );
-}
-
-/**
- * 创建 AI 客户端
- * @param {Object} config - 配置
- * @param {string} config.provider - 提供商名称
- * @param {string} config.model - 模型名称
- * @param {string} config.apiKey - API 密钥
- * @param {string} [config.baseURL] - 自定义 API 地址
- */
-function createAI(config) {
-  const { provider, model, apiKey, baseURL } = config;
-  const providerName = (provider || 'deepseek').toLowerCase();
-  const isThinking = isThinkingModel(model);
-
-  // 检查是否是预定义提供商
-  if (DEFAULT_PROVIDERS[providerName]) {
-    const providerConfig = DEFAULT_PROVIDERS[providerName];
-
-    // 为 DeepSeek thinking mode 模型添加特殊配置
-    const streamOptions = {
-      includeUsage: true,
-    };
-
-    // DeepSeek 模型添加 reasoning_content 输出
-    if (providerName === 'deepseek') {
-      streamOptions.includeReasoningContent = true;
-    }
-
-    return createOpenAICompatible({
-      name: providerConfig.name,
-      baseURL: baseURL || providerConfig.baseURL,
-      apiKey: apiKey || 'dummy-key',
-      headers:
-        provider === 'anthropic'
-          ? {
-              'x-api-key': apiKey,
-              'anthropic-version': '2023-06-01',
-            }
-          : undefined,
-      models: {
-        default: {
-          id: model || 'deepseek-chat',
-          streamOptions,
-        },
-      },
-    });
-  }
-
-  // 自定义提供商
-  return createOpenAICompatible({
-    name: providerName || 'Custom',
-    baseURL: baseURL,
-    apiKey: apiKey || 'dummy-key',
-    models: {
-      default: {
-        id: model || 'gpt-4',
-      },
-    },
-  });
-}
-
-/**
- * 获取可用提供商列表
- */
-function getAvailableProviders() {
-  return Object.entries(DEFAULT_PROVIDERS).map(([key, value]) => ({
-    id: key,
-    name: value.name,
-    baseURL: value.baseURL,
-  }));
-}
-
-/**
- * 创建模型实例
- * @param {string} model - 模型名称
- * @param {Object} provider - Provider 实例
- */
-function createModel(model, provider) {
-  return provider(model);
-}
-
-module.exports = {
-  createAI,
-  createModel,
-  getAvailableProviders,
-  DEFAULT_PROVIDERS,
-  isThinkingModel,
-  getProviderRegistry: () => require('./provider-registry').getProviderRegistry(),
-  registerProvider: (name, config) => require('./provider-registry').getProviderRegistry().registerProvider(name, config),
-};

package/src/core/request-context.js

@@ -1,98 +0,0 @@
-/**
- * RequestContext - Request 级别的上下文封装
- *
- * 职责：
- * 1. 追踪请求链路（requestId, traceId）
- * 2. 管理请求级超时
- * 3. 传递请求元数据
- */
-
-const { uuid } = require('../utils/id');
-
-class RequestContext {
-  /**
-   * @param {Object} options - 配置选项
-   * @param {string} [options.requestId] - 请求 ID，默认自动生成
-   * @param {string} [options.traceId] - 追踪 ID，默认自动生成
-   * @param {string} [options.parentSpanId] - 父 span ID
-   * @param {number} [options.timeout] - 超时时间（毫秒），默认 120000
-   * @param {boolean} [options.isStream] - 是否为流式请求
-   * @param {string} [options.userId] - 用户标识
-   */
-  constructor(options = {}) {
-    this.requestId = options.requestId || `req_${uuid()}`;
-    this.traceId = options.traceId || `trace_${uuid()}`;
-    this.traceId = options.traceId || generateId();
-    this.parentSpanId = options.parentSpanId || null;
-
-    this.timeout = options.timeout || 120000; // 默认 2 分钟
-    this.startTime = options.startTime || Date.now();
-
-    this.isStream = options.isStream || false;
-    this.userId = options.userId || null;
-
-    // 请求级状态
-    this.cancelled = false;
-    this.abortController = null;
-
-    // 元数据
-    this.metadata = options.metadata || {};
-  }
-
-  /**
-   * 创建 AbortController（延迟初始化）
-   * @returns {AbortController}
-   */
-  getAbortController() {
-    if (!this.abortController) {
-      this.abortController = new AbortController();
-    }
-    return this.abortController;
-  }
-
-  /**
-   * 检查是否超时
-   * @returns {boolean}
-   */
-  isTimeout() {
-    return Date.now() - this.startTime > this.timeout;
-  }
-
-  /**
-   * 取消请求
-   */
-  cancel() {
-    this.cancelled = true;
-    if (this.abortController) {
-      this.abortController.abort();
-    }
-  }
-
-  /**
-   * 获取请求耗时（毫秒）
-   * @returns {number}
-   */
-  getElapsed() {
-    return Date.now() - this.startTime;
-  }
-
-  /**
-   * 获取请求摘要
-   * @returns {Object}
-   */
-  toJSON() {
-    return {
-      requestId: this.requestId,
-      traceId: this.traceId,
-      parentSpanId: this.parentSpanId,
-      timeout: this.timeout,
-      startTime: this.startTime,
-      elapsed: this.getElapsed(),
-      isStream: this.isStream,
-      userId: this.userId,
-      cancelled: this.cancelled,
-    };
-  }
-}
-
-module.exports = { RequestContext };