levix-bot 3.3.0-rc.2 → 4.0.0

package/src/services/aiRouter.cjs

@@ -0,0 +1,137 @@
+// file: src/services/aiRouter.cjs
+// Modular AI Provider Registry and Router.
+//
+// Instead of hardcoding conditionals for each provider across command handlers
+// and service loops, providers register an adapter implementing the BaseAIProvider
+// interface. The router dispatches based on the active `ai_provider` setting.
+
+const settings = require("../config/settings.cjs");
+const { resolveCapabilities } = require("./modelRegistry.cjs");
+
+class BaseAIProvider {
+  constructor({
+    id,
+    label,
+    keySetting,
+    modelSetting,
+    baseUrlSetting,
+    defaultBaseUrl = "",
+    supportsFiles = false,
+  }) {
+    this.id = id;
+    this.label = label;
+    this.keySetting = keySetting;
+    this.modelSetting = modelSetting;
+    this.baseUrlSetting = baseUrlSetting;
+    this.defaultBaseUrl = defaultBaseUrl;
+    this.supportsFiles = supportsFiles;
+  }
+
+  isConfigured() {
+    return Boolean(this.getKey());
+  }
+
+  getKey() {
+    return settings.get(this.keySetting) || "";
+  }
+
+  getModel() {
+    return settings.get(this.modelSetting) || "";
+  }
+
+  getBaseUrl() {
+    if (!this.baseUrlSetting) return this.defaultBaseUrl;
+    return settings.get(this.baseUrlSetting) || this.defaultBaseUrl;
+  }
+
+  supportsMedia(mime = "") {
+    const caps = detectModelCapabilities(this.id);
+    if (typeof mime === "string" && mime.startsWith("image/")) {
+      return Boolean(settings.get("ai_vision_enabled") && caps.supportsVision);
+    }
+    if (typeof mime === "string" && mime.startsWith("audio/")) {
+      return Boolean(settings.get("ai_stt_enabled") && caps.supportsAudioStt);
+    }
+    if (this.supportsFiles) return true;
+    return false;
+  }
+
+  async prepareMedia(parts, mediaMessage, mimeOverride, _context = {}) {
+    const mime = mimeOverride || mediaMessage?.mimetype || "ملف";
+    parts.push({
+      text: `[تم إرفاق وسائط (${mime}) — المزود الحالي لا يدعم هذا النوع من الوسائط]`,
+    });
+    return null;
+  }
+
+  async runTurn(_options) {
+    throw new Error(`runTurn() not implemented for provider: ${this.id}`);
+  }
+}
+
+const REGISTRY = new Map();
+
+function registerProvider(id, providerInstance) {
+  if (!id || typeof id !== "string") {
+    throw new Error("Provider id must be a non-empty string");
+  }
+  REGISTRY.set(id.toLowerCase(), providerInstance);
+}
+
+function getProvider(id = settings.get("ai_provider")) {
+  const key = String(id || "gemini").toLowerCase();
+  const provider = REGISTRY.get(key);
+  if (!provider) {
+    throw new Error(`مزود غير معروف: ${id}`);
+  }
+  return provider;
+}
+
+function hasProvider(id) {
+  return REGISTRY.has(String(id || "").toLowerCase());
+}
+
+function listProviders() {
+  return Array.from(REGISTRY.values());
+}
+
+function activeProviderKeySetting(provider = settings.get("ai_provider")) {
+  const p = getProvider(provider);
+  return p.keySetting;
+}
+
+function detectModelCapabilities(provider = settings.get("ai_provider"), model = null) {
+  const p = provider || "gemini";
+  const m = String(
+    model ||
+      (p === "gemini"
+        ? settings.get("gemini_model")
+        : p === "openai"
+          ? settings.get("openai_model")
+          : settings.get("anthropic_model")) ||
+      "",
+  );
+
+  const resolved = resolveCapabilities(p, m);
+  const caps = resolved.capabilities;
+
+  return {
+    provider: p,
+    model: m,
+    capabilities: caps,
+    capabilitySource: resolved.capabilitySource,
+    supportsVision: Boolean(caps.vision),
+    supportsAudioStt: Boolean(caps.stt),
+    visionRecommended: Boolean(caps.vision) && !settings.get("ai_vision_enabled"),
+  };
+}
+
+module.exports = {
+  BaseAIProvider,
+  registerProvider,
+  getProvider,
+  hasProvider,
+  listProviders,
+  activeProviderKeySetting,
+  detectModelCapabilities,
+};