@cosmicstack/mercury-agent 1.1.1 → 1.1.3

package/dist/index.js

@@ -80,7 +80,7 @@ function getDefaultConfig() {
       ollamaCloud: {
         name: "ollamaCloud",
         apiKey: getEnv("OLLAMA_CLOUD_API_KEY", ""),
-        baseUrl: getEnv("OLLAMA_CLOUD_BASE_URL", "https://ollama.com/api"),
+        baseUrl: getEnv("OLLAMA_CLOUD_BASE_URL", "https://ollama.com/v1"),
         model: getEnv("OLLAMA_CLOUD_MODEL", "gpt-oss:120b"),
         enabled: getEnvBool("OLLAMA_CLOUD_ENABLED", true)
       },
@@ -90,6 +90,20 @@ function getDefaultConfig() {
         baseUrl: getEnv("OLLAMA_LOCAL_BASE_URL", "http://127.0.0.1:11434/api"),
         model: getEnv("OLLAMA_LOCAL_MODEL", "gpt-oss:20b"),
         enabled: getEnvBool("OLLAMA_LOCAL_ENABLED", false)
+      },
+      mimo: {
+        name: "mimo",
+        apiKey: getEnv("MIMO_API_KEY", ""),
+        baseUrl: getEnv("MIMO_BASE_URL", "https://api.xiaomimimo.com/v1"),
+        model: getEnv("MIMO_MODEL", "mimo-v2.5-pro"),
+        enabled: getEnvBool("MIMO_ENABLED", true)
+      },
+      mimoTokenPlan: {
+        name: "mimoTokenPlan",
+        apiKey: getEnv("MIMO_TOKEN_PLAN_API_KEY", ""),
+        baseUrl: getEnv("MIMO_TOKEN_PLAN_BASE_URL", "https://token-plan-cn.xiaomimimo.com/v1"),
+        model: getEnv("MIMO_TOKEN_PLAN_MODEL", "mimo-v2.5-pro"),
+        enabled: getEnvBool("MIMO_TOKEN_PLAN_ENABLED", false)
       }
     },
     channels: {
@@ -131,7 +145,9 @@ function loadConfig() {
     const raw = readFileSync(CONFIG_PATH, "utf-8");
     const fileConfig = parseYaml(raw);
     const defaults = getDefaultConfig();
-    return migrateLegacyTelegramAccess(deepMerge(defaults, fileConfig));
+    return migrateLegacyOllamaCloudBaseUrl(
+      migrateLegacyTelegramAccess(deepMerge(defaults, fileConfig))
+    );
   }
   return migrateLegacyTelegramAccess(getDefaultConfig());
 }
@@ -175,6 +191,9 @@ function isProviderConfigured(provider) {
   if (provider.name === "ollamaLocal") {
     return provider.baseUrl.length > 0 && provider.model.length > 0;
   }
+  if (provider.name === "ollamaCloud") {
+    return provider.apiKey.length > 0 && provider.baseUrl.length > 0;
+  }
   return provider.apiKey.length > 0;
 }
 function getTelegramApprovedUsers(config) {
@@ -316,6 +335,13 @@ function migrateLegacyTelegramAccess(config) {
   delete telegram.pairedUsername;
   return config;
 }
+function migrateLegacyOllamaCloudBaseUrl(config) {
+  if (config.providers.ollamaCloud.baseUrl === "https://ollama.com/api") {
+    config.providers.ollamaCloud.baseUrl = "https://ollama.com/v1";
+    saveConfig(config);
+  }
+  return config;
+}
 
 // src/utils/logger.ts
 import pino from "pino";
@@ -1517,7 +1543,7 @@ var OpenAICompatProvider = class extends BaseProvider {
   model;
   client;
   modelInstance;
-  constructor(config) {
+  constructor(config, { useChatApi } = {}) {
     super(config);
     this.name = config.name;
     this.model = config.model;
@@ -1525,7 +1551,7 @@ var OpenAICompatProvider = class extends BaseProvider {
       apiKey: config.apiKey,
       baseURL: config.baseUrl
     });
-    this.modelInstance = this.client(config.model);
+    this.modelInstance = useChatApi ? this.client.chat(config.model) : this.client(config.model);
   }
   async generateText(prompt, systemPrompt) {
     const result = await generateText({
@@ -1672,6 +1698,36 @@ var OllamaProvider = class extends BaseProvider {
     if (this.name === "ollamaLocal") {
       return this.config.baseUrl.length > 0 && this.config.model.length > 0;
     }
+    return this.config.apiKey.length > 0 && this.config.baseUrl.length > 0;
+  }
+  getModelInstance() {
+    return this.modelInstance;
+  }
+};
+
+// src/providers/mimo.ts
+import { createOpenAI as createOpenAI2 } from "@ai-sdk/openai";
+var MiMoProvider = class extends BaseProvider {
+  name;
+  model;
+  modelInstance;
+  constructor(config) {
+    super(config);
+    this.name = config.name;
+    this.model = config.model;
+    const client = createOpenAI2({
+      apiKey: config.apiKey,
+      baseURL: config.baseUrl
+    });
+    this.modelInstance = client.chat(config.model);
+  }
+  async generateText(_prompt, _systemPrompt) {
+    throw new Error("Use getModelInstance() with the AI SDK agent loop");
+  }
+  async *streamText(_prompt, _systemPrompt) {
+    throw new Error("Use getModelInstance() with the AI SDK agent loop");
+  }
+  isAvailable() {
     return this.config.apiKey.length > 0;
   }
   getModelInstance() {
@@ -1692,7 +1748,9 @@ var ProviderRegistry = class {
       config.providers.anthropic,
       config.providers.grok,
       config.providers.ollamaCloud,
-      config.providers.ollamaLocal
+      config.providers.ollamaLocal,
+      config.providers.mimo,
+      config.providers.mimoTokenPlan
     ];
     for (const pc of entries) {
       if (!isProviderConfigured(pc)) continue;
@@ -1702,8 +1760,12 @@ var ProviderRegistry = class {
           provider = new AnthropicProvider(pc);
         } else if (pc.name === "deepseek") {
           provider = new DeepSeekProvider(pc);
-        } else if (pc.name === "ollamaCloud" || pc.name === "ollamaLocal") {
+        } else if (pc.name === "ollamaLocal") {
           provider = new OllamaProvider(pc);
+        } else if (pc.name === "ollamaCloud") {
+          provider = new OpenAICompatProvider(pc, { useChatApi: true });
+        } else if (pc.name === "mimo" || pc.name === "mimoTokenPlan") {
+          provider = new MiMoProvider(pc);
         } else {
           provider = new OpenAICompatProvider(pc);
         }
@@ -8100,6 +8162,14 @@ var OLLAMA_LOCAL_PREFERRED_MODELS = [
   "gpt-oss:20b",
   "gpt-oss:120b"
 ];
+var MIMO_PREFERRED_MODELS = [
+  "mimo-v2.5-pro",
+  "mimo-v2.5",
+  "mimo-v2-pro",
+  "mimo-v2-omni",
+  "mimo-v2-flash"
+];
+var MIMO_TOKEN_PLAN_PREFERRED_MODELS = MIMO_PREFERRED_MODELS;
 var ProviderModelFetchError = class extends Error {
   constructor(message) {
     super(message);
@@ -8154,7 +8224,9 @@ function chooseRecommendedModel(provider, models, currentModel) {
     anthropic: ANTHROPIC_PREFERRED_MODELS,
     grok: GROK_PREFERRED_MODELS,
     ollamaCloud: OLLAMA_CLOUD_PREFERRED_MODELS,
-    ollamaLocal: OLLAMA_LOCAL_PREFERRED_MODELS
+    ollamaLocal: OLLAMA_LOCAL_PREFERRED_MODELS,
+    mimo: MIMO_PREFERRED_MODELS,
+    mimoTokenPlan: MIMO_TOKEN_PLAN_PREFERRED_MODELS
   };
   for (const candidate of preferredByProvider[provider]) {
     if (models.includes(candidate)) {
@@ -8178,7 +8250,9 @@ function buildModelCatalog(provider, models, currentModel) {
     anthropic: ANTHROPIC_PREFERRED_MODELS,
     grok: GROK_PREFERRED_MODELS,
     ollamaCloud: OLLAMA_CLOUD_PREFERRED_MODELS,
-    ollamaLocal: OLLAMA_LOCAL_PREFERRED_MODELS
+    ollamaLocal: OLLAMA_LOCAL_PREFERRED_MODELS,
+    mimo: MIMO_PREFERRED_MODELS,
+    mimoTokenPlan: MIMO_TOKEN_PLAN_PREFERRED_MODELS
   };
   const withoutRecommended = filtered.filter((model) => model !== recommendedModel);
   const prioritized = prioritizeModels(withoutRecommended, preferredByProvider[provider]);
@@ -8232,17 +8306,59 @@ async function fetchGrokModels(config) {
   const ids = (data.data ?? []).filter((model) => model.output_modalities?.includes("text") || model.output_modalities == null).map((model) => model.id?.trim() ?? "").filter((id) => id.startsWith("grok-"));
   return buildModelCatalog("grok", ids, config.model);
 }
-async function fetchOllamaModels(provider, config) {
-  const headers = config.apiKey ? { Authorization: `Bearer ${config.apiKey}` } : void 0;
+async function fetchOllamaCloudModels(config) {
   const data = await fetchJson(
-    `${trimTrailingSlash(config.baseUrl)}/tags`,
+    `${trimTrailingSlash(config.baseUrl)}/models`,
     {
-      headers
+      headers: {
+        Authorization: `Bearer ${config.apiKey}`
+      }
     },
-    provider === "ollamaCloud" ? "Mercury could not fetch models for this Ollama Cloud key. Please re-enter it." : "Mercury could not fetch models from this Ollama Local server. Please check the base URL and try again."
+    "Mercury could not fetch models for this Ollama Cloud key. Please re-enter it."
+  );
+  const ids = (data.data ?? []).map((model) => model.id?.trim() ?? "").filter(Boolean);
+  return buildModelCatalog("ollamaCloud", ids, config.model);
+}
+async function fetchOllamaLocalModels(config) {
+  const data = await fetchJson(
+    `${trimTrailingSlash(config.baseUrl)}/tags`,
+    {},
+    "Mercury could not fetch models from this Ollama Local server. Please check the base URL and try again."
   );
   const ids = (data.models ?? []).map((model) => model.model?.trim() || model.name?.trim() || "").filter(Boolean);
-  return buildModelCatalog(provider, ids, config.model);
+  return buildModelCatalog("ollamaLocal", ids, config.model);
+}
+async function fetchMiMoModels(config) {
+  const data = await fetchJson(
+    `${trimTrailingSlash(config.baseUrl)}/models`,
+    {
+      headers: {
+        Authorization: `Bearer ${config.apiKey}`
+      }
+    },
+    "Mercury could not fetch models for this MiMo key. Please re-enter it."
+  );
+  const ids = (data.data ?? []).map((model) => model.id?.trim() ?? "").filter((id) => {
+    const lower = id.toLowerCase();
+    return lower.startsWith("mimo-") && !lower.includes("tts");
+  });
+  return buildModelCatalog("mimo", ids, config.model);
+}
+async function fetchMiMoTokenPlanModels(config) {
+  const data = await fetchJson(
+    `${trimTrailingSlash(config.baseUrl)}/models`,
+    {
+      headers: {
+        Authorization: `Bearer ${config.apiKey}`
+      }
+    },
+    "Mercury could not fetch models for this MiMo Token Plan key. Please re-enter it."
+  );
+  const ids = (data.data ?? []).map((model) => model.id?.trim() ?? "").filter((id) => {
+    const lower = id.toLowerCase();
+    return lower.startsWith("mimo-") && !lower.includes("tts");
+  });
+  return buildModelCatalog("mimoTokenPlan", ids, config.model);
 }
 async function fetchProviderModelCatalog(provider, config) {
   if (provider === "anthropic") {
@@ -8251,8 +8367,17 @@ async function fetchProviderModelCatalog(provider, config) {
   if (provider === "grok") {
     return fetchGrokModels(config);
   }
-  if (provider === "ollamaCloud" || provider === "ollamaLocal") {
-    return fetchOllamaModels(provider, config);
+  if (provider === "ollamaCloud") {
+    return fetchOllamaCloudModels(config);
+  }
+  if (provider === "ollamaLocal") {
+    return fetchOllamaLocalModels(config);
+  }
+  if (provider === "mimo") {
+    return fetchMiMoModels(config);
+  }
+  if (provider === "mimoTokenPlan") {
+    return fetchMiMoTokenPlanModels(config);
   }
   return fetchOpenAICompatModels(provider, config);
 }
@@ -8311,7 +8436,9 @@ var PROVIDER_OPTIONS = [
   { key: "anthropic", label: "Anthropic" },
   { key: "grok", label: "Grok (xAI)" },
   { key: "ollamaCloud", label: "Ollama Cloud" },
-  { key: "ollamaLocal", label: "Ollama Local" }
+  { key: "ollamaLocal", label: "Ollama Local" },
+  { key: "mimo", label: "MiMo (Xiaomi)" },
+  { key: "mimoTokenPlan", label: "MiMo Token Plan (Xiaomi)" }
 ];
 function getConfiguredProviderNames(config) {
   return PROVIDER_OPTIONS.map((option) => option.key).filter((key) => isProviderConfigured(config.providers[key]));
@@ -8412,6 +8539,12 @@ function validateApiKey(provider, value) {
   if (provider === "ollamaCloud") {
     return looksLikeToken(value) ? null : "Ollama Cloud keys must look like a real API token: long, no spaces, and not plain text.";
   }
+  if (provider === "mimo") {
+    return /^sk-[A-Za-z0-9_-]{16,}$/i.test(value) ? null : "MiMo keys must start with `sk-`.";
+  }
+  if (provider === "mimoTokenPlan") {
+    return /^tp-[A-Za-z0-9_-]{16,}$/i.test(value) ? null : "MiMo Token Plan keys must start with `tp-`.";
+  }
   return null;
 }
 function validateBaseUrl(value) {
@@ -8760,6 +8893,38 @@ async function configure(existingConfig) {
           config.providers.ollamaLocal.model = result.model;
           config.providers.ollamaLocal.enabled = true;
         }
+        continue;
+      }
+      if (provider === "mimo") {
+        const mask = isReconfig && config.providers.mimo.apiKey ? ` [${maskKey(config.providers.mimo.apiKey)}]` : "";
+        const result = await promptApiKeyWithModelSelection(
+          config,
+          "mimo",
+          "MiMo",
+          chalk7.white(`  MiMo API key${mask}${isReconfig ? "" : " (Enter to skip)"}: `),
+          isReconfig
+        );
+        if (!result.skipped && result.apiKey && result.model) {
+          config.providers.mimo.apiKey = result.apiKey;
+          config.providers.mimo.model = result.model;
+          config.providers.mimo.enabled = true;
+        }
+        continue;
+      }
+      if (provider === "mimoTokenPlan") {
+        const mask = isReconfig && config.providers.mimoTokenPlan.apiKey ? ` [${maskKey(config.providers.mimoTokenPlan.apiKey)}]` : "";
+        const result = await promptApiKeyWithModelSelection(
+          config,
+          "mimoTokenPlan",
+          "MiMo Token Plan",
+          chalk7.white(`  MiMo Token Plan API key${mask}${isReconfig ? "" : " (Enter to skip)"}: `),
+          isReconfig
+        );
+        if (!result.skipped && result.apiKey && result.model) {
+          config.providers.mimoTokenPlan.apiKey = result.apiKey;
+          config.providers.mimoTokenPlan.model = result.model;
+          config.providers.mimoTokenPlan.enabled = true;
+        }
       }
     }
     const configuredProviders = getConfiguredProviderNames(config);