@thanh01.pmt/curriculum-kit 1.4.26 → 1.4.28

package/dist/ai/index.cjs

@@ -177,14 +177,19 @@ function isModelAllowed(model) {
   });
 }
 function getDesignatedFallbackChain() {
-  const rawProviders = resolveApiKey("FALLBACK_AI_PROVIDER") || resolveApiKey("FALLBACK_AI_PROVIDERS") || resolveApiKey("DEFAULT_AI_PROVIDER") || "nvidia,openrouter";
-  const rawModels = resolveApiKey("FALLBACK_AI_MODEL") || resolveApiKey("FALLBACK_AI_MODELS") || resolveApiKey("DEFAULT_AI_MODEL") || "nvidia/nemotron-3-ultra-550b-a55b:free,@preset/coding-free,deepseek/deepseek-chat:free";
+  const rawProviders = resolveApiKey("FALLBACK_AI_PROVIDER") || resolveApiKey("FALLBACK_AI_PROVIDERS") || resolveApiKey("DEFAULT_AI_PROVIDER") || "9router,nvidia,openrouter";
+  const rawModels = resolveApiKey("FALLBACK_AI_MODEL") || resolveApiKey("FALLBACK_AI_MODELS") || resolveApiKey("DEFAULT_AI_MODEL") || "fast-reasoning,deep-reasoning,mid-range,free-tier";
   const providerList = rawProviders.split(",").map((p) => p.trim().toLowerCase()).filter((p) => p.length > 0 && isProviderEnabled(p));
   const modelList = rawModels.split(",").map((m) => m.trim()).filter((m) => m.length > 0 && isModelAllowed(m));
   const chain = [];
   for (const provider of providerList) {
     let providerModels = [];
-    if (provider === "nvidia") {
+    if (provider === "9router" || provider === "ninerouter") {
+      providerModels = modelList.filter((m) => ["fast-reasoning", "deep-reasoning", "mid-range", "free-tier"].includes(m));
+      if (providerModels.length === 0) {
+        providerModels = ["fast-reasoning", "mid-range", "free-tier"];
+      }
+    } else if (provider === "nvidia") {
       providerModels = modelList.filter((m) => m.startsWith("nvidia/") || m.startsWith("meta/") || m.startsWith("mistralai/"));
       if (providerModels.length === 0) {
         providerModels = ["nvidia/nemotron-3-ultra-550b-a55b:free", "nvidia/nemotron-3.5-lightning-30b-a3b"];
@@ -223,7 +228,7 @@ function getDesignatedFallbackChain() {
 function getDesignatedFallbackConfig() {
   const chain = getDesignatedFallbackChain();
   if (chain.length > 0) return chain[0];
-  return { provider: "nvidia", model: "nvidia/nemotron-3-ultra-550b-a55b:free" };
+  return { provider: "9router", model: "fast-reasoning" };
 }
 function extractThoughtAndContent(rawText) {
   let text = rawText || "";
@@ -467,8 +472,51 @@ Guidelines:
     }))
   ];
   const providerErrors = {};
-  const requestedModel = options.model || "@preset/coding-free";
+  const requestedModel = options.model || "fast-reasoning";
   const isNvidiaPreferred = requestedModel.startsWith("nvidia/") || Boolean(nvidiaKey && !openrouterKey) && isProviderEnabled("nvidia");
+  const ninerouterKey = resolveApiKey("NINEROUTER_API_KEY") || resolveApiKey("NINE_ROUTER_API_KEY");
+  if (ninerouterKey && isProviderEnabled("9router")) {
+    const rawUrl = resolveApiKey("NINEROUTER_BASE_URL") || resolveApiKey("NINE_ROUTER_BASE_URL") || "https://ai-router.orchable.app/v1";
+    const ninerouterBaseUrl = rawUrl.replace(/^["']|["']$/g, "").replace(/\/$/, "");
+    const candidateModels = [
+      options.model || "fast-reasoning",
+      "fast-reasoning",
+      "mid-range",
+      "free-tier"
+    ].filter((m) => isModelAllowed(m));
+    const unique9RouterModels = Array.from(new Set(candidateModels));
+    for (const model of unique9RouterModels) {
+      try {
+        const idle = createStreamAbortController({ idleMs: 45e3, totalMs: 3e5 });
+        const res = await fetch(`${ninerouterBaseUrl}/chat/completions`, {
+          method: "POST",
+          headers: {
+            "Authorization": `Bearer ${ninerouterKey}`,
+            "Content-Type": "application/json"
+          },
+          body: JSON.stringify({
+            model,
+            messages: formattedMessages,
+            max_tokens: maxTokens,
+            temperature,
+            stream: true,
+            stream_options: { include_usage: true }
+          }),
+          signal: idle.signal
+        });
+        if (res.ok) {
+          const text = await processOpenAISSEStream(res, onChunk, idle);
+          if (text && text.trim()) return text;
+        } else {
+          const errText = await res.text().catch(() => "");
+          providerErrors[`9router_${model}`] = `HTTP ${res.status}: ${errText.slice(0, 120)}`;
+        }
+      } catch (e) {
+        providerErrors[`9router_${model}`] = e?.message || "Network error";
+        console.warn(`[9Router Stream] Failover for ${model}:`, e?.message || e);
+      }
+    }
+  }
   if (nvidiaKey && isProviderEnabled("nvidia")) {
     const candidateNemotronModels = [
       requestedModel.startsWith("nvidia/") ? requestedModel : "nvidia/nemotron-3-ultra-550b-a55b:free",
@@ -678,7 +726,41 @@ Guidelines:
     );
     for (const target of fallbackChain) {
       const { provider, model } = target;
-      if (provider === "nvidia" && nvidiaKey && isProviderEnabled("nvidia")) {
+      if ((provider === "9router" || provider === "ninerouter") && ninerouterKey && isProviderEnabled("9router")) {
+        try {
+          const rawUrl = resolveApiKey("NINEROUTER_BASE_URL") || resolveApiKey("NINE_ROUTER_BASE_URL") || "https://ai-router.orchable.app/v1";
+          const fbBaseUrl = rawUrl.replace(/["']/g, "").replace(/\/$/, "");
+          const idle = createStreamAbortController({ idleMs: 45e3, totalMs: 3e5 });
+          const res = await fetch(`${fbBaseUrl}/chat/completions`, {
+            method: "POST",
+            headers: {
+              "Authorization": `Bearer ${ninerouterKey}`,
+              "Content-Type": "application/json"
+            },
+            body: JSON.stringify({
+              model,
+              messages: formattedMessages,
+              max_tokens: maxTokens,
+              temperature,
+              stream: true,
+              stream_options: { include_usage: true }
+            }),
+            signal: idle.signal
+          });
+          if (res.ok) {
+            const text = await processOpenAISSEStream(res, onChunk, idle);
+            if (text && text.trim()) {
+              console.log(`[streamRunner] \u2705 Designated fallback 9Router (${model}) succeeded!`);
+              return text;
+            }
+          } else {
+            const errText = await res.text().catch(() => "");
+            providerErrors[`designated_fallback_9router_${model}`] = `HTTP ${res.status}: ${errText.slice(0, 120)}`;
+          }
+        } catch (e) {
+          providerErrors[`designated_fallback_9router_${model}`] = e?.message || "Network error";
+        }
+      } else if (provider === "nvidia" && nvidiaKey && isProviderEnabled("nvidia")) {
         try {
           const nvdIdle = options.idleTimeoutMs ?? 18e4;
           const nvdTotal = options.timeoutMs ?? 6e5;
@@ -835,6 +917,8 @@ async function runCurriculumAIInference(messages, projectContext, options = {},
 
 // src/ai/provider-factory.ts
 var DEFAULT_MODELS = {
+  "9router": "fast-reasoning",
+  ninerouter: "fast-reasoning",
   nvidia: "nvidia/nemotron-3-ultra-550b-a55b:free",
   openrouter: "@preset/coding-free",
   alibaba: "qwen3.7-plus",
@@ -866,6 +950,15 @@ function getAIModel(options = {}) {
     console.warn(`[provider-factory] Model "${resolvedModelName}" is not permitted by ALLOWED_AI_MODELS. Using designated model "${designated.model}".`);
     resolvedModelName = designated.model;
   }
+  if (provider === "9router" || provider === "ninerouter") {
+    const apiKey = options.apiKey || resolveApiKey("NINEROUTER_API_KEY") || resolveApiKey("NINE_ROUTER_API_KEY") || "";
+    const rawUrl = options.baseURL || resolveApiKey("NINEROUTER_BASE_URL") || resolveApiKey("NINE_ROUTER_BASE_URL") || "https://ai-router.orchable.app/v1";
+    const ninerouter = openai.createOpenAI({
+      apiKey,
+      baseURL: rawUrl.replace(/^["']|["']$/g, "").trim()
+    });
+    return ninerouter.chat(resolvedModelName || "fast-reasoning");
+  }
   if (provider === "nvidia") {
     const apiKey = options.apiKey || process.env.NVIDIA_API_KEY || "";
     const nvidia = openai.createOpenAI({
@@ -934,6 +1027,9 @@ function getAIModel(options = {}) {
   return fallback.chat(resolvedModelName);
 }
 function detectDefaultProvider() {
+  if ((process.env.NINEROUTER_API_KEY || resolveApiKey("NINEROUTER_API_KEY") || resolveApiKey("NINE_ROUTER_API_KEY")) && isProviderEnabled("9router")) {
+    return "9router";
+  }
   if (process.env.NVIDIA_API_KEY && isProviderEnabled("nvidia")) {
     return "nvidia";
   }
@@ -5130,6 +5226,7 @@ exports.isModelAllowed = isModelAllowed;
 exports.isProviderEnabled = isProviderEnabled;
 exports.packagerTools = packagerTools;
 exports.researcherTools = researcherTools;
+exports.resolveApiKey = resolveApiKey;
 exports.reviewerTools = reviewerTools;
 exports.runCurriculumAIInference = runCurriculumAIInference;
 exports.streamCurriculumAIInference = streamCurriculumAIInference;