Add new NIM models + dynamic (Recommended) model logic

openai-compatible.config.ts

@@ -6,6 +6,36 @@ const providers: OpenAICompatibleProvider[] = [
   "apiKey": process.env.BACKEND_API_KEY as string,
   "baseUrl": "https://augment17-claude-code-backend.hf.space/v1",
   "models": [
+    {
+      "apiName": "nvidia/nemotron-3-ultra-550b-a55b",
+      "uiName": "Nemotron 3 Ultra 550B (Agentic)",
+      "supportsTools": true
+    },
+    {
+      "apiName": "z-ai/glm-5.1",
+      "uiName": "GLM 5.1 (Agentic)",
+      "supportsTools": true
+    },
+    {
+      "apiName": "moonshotai/kimi-k2.6",
+      "uiName": "Kimi K2.6 (Agentic)",
+      "supportsTools": true
+    },
+    {
+      "apiName": "minimaxai/minimax-m3",
+      "uiName": "MiniMax M3 (Agentic)",
+      "supportsTools": true
+    },
+    {
+      "apiName": "stepfun-ai/step-3.7-flash",
+      "uiName": "Step 3.7 Flash (Agentic)",
+      "supportsTools": true
+    },
+    {
+      "apiName": "minimaxai/minimax-m2.7",
+      "uiName": "MiniMax M2.7 (Agentic)",
+      "supportsTools": true
+    },
     {
       "apiName": "meta/llama-3.1-70b-instruct",
       "uiName": "Llama 3.1 70B (Agentic)",

src/app/api/chat/models/route.ts

@@ -1,8 +1,57 @@
 import { customModelProvider } from "lib/ai/models";
 
 export const GET = async () => {
+  let recommendedModelId = "";
+  try {
+    const res = await fetch("https://augment17-claude-code-backend.hf.space/health", {
+      headers: { "Authorization": `Bearer ${process.env.BACKEND_API_KEY}` },
+      next: { revalidate: 60 }
+    });
+    if (res.ok) {
+      const data = await res.json();
+      recommendedModelId = data.recommended_model || "";
+    }
+  } catch (error) {
+    console.error("Failed to fetch recommended model from backend:", error);
+  }
+
+  const apiNames: Record<string, string> = {
+    "Nemotron 3 Ultra 550B (Agentic)": "nvidia/nemotron-3-ultra-550b-a55b",
+    "GLM 5.1 (Agentic)": "z-ai/glm-5.1",
+    "Kimi K2.6 (Agentic)": "moonshotai/kimi-k2.6",
+    "MiniMax M3 (Agentic)": "minimaxai/minimax-m3",
+    "Step 3.7 Flash (Agentic)": "stepfun-ai/step-3.7-flash",
+    "MiniMax M2.7 (Agentic)": "minimaxai/minimax-m2.7",
+    "Llama 3.1 70B (Agentic)": "meta/llama-3.1-70b-instruct",
+    "Llama 3.1 405B (Agentic)": "meta/llama-3.1-405b-instruct",
+    "Qwen 2.5 Coder 32B (Agentic)": "qwen/qwen2.5-coder-32b-instruct",
+    "Nemotron 70B (Agentic)": "nvidia/llama-3.1-nemotron-70b-instruct",
+    "Llama 3.3 70B (Agentic)": "meta/llama-3.3-70b-instruct",
+    "DeepSeek R1 (Chat only)": "deepseek-ai/deepseek-r1",
+    "Mistral Large 2 (Chat only)": "mistralai/mistral-large-2-instruct",
+  };
+
+  const modelsInfo = customModelProvider.modelsInfo.map(providerInfo => {
+    if (providerInfo.provider === "Claude Code") {
+      return {
+        ...providerInfo,
+        models: providerInfo.models.map(model => {
+          const apiName = apiNames[model.name];
+          if (apiName && apiName === recommendedModelId) {
+            return {
+              ...model,
+              name: `${model.name} (Recommended)`
+            };
+          }
+          return model;
+        })
+      };
+    }
+    return providerInfo;
+  });
+
   return Response.json(
-    customModelProvider.modelsInfo.sort((a, b) => {
+    modelsInfo.sort((a, b) => {
       if (a.hasAPIKey && !b.hasAPIKey) return -1;
       if (!a.hasAPIKey && b.hasAPIKey) return 1;
       return 0;