Merge pull request #12156 from RomneyDa/add-mercury-2

Author: sestinj

core/llm/toolSupport.ts

@@ -270,6 +270,10 @@ export const PROVIDER_TOOL_SUPPORT: Record<string, (model: string) => boolean> =
 
       return false;
     },
+    inception: (model) => {
+      const lower = model.toLowerCase();
+      return lower.startsWith("mercury-2");
+    },
     deepseek: (model) => {
       // https://api-docs.deepseek.com/quick_start/pricing
       // https://api-docs.deepseek.com/guides/function_calling

gui/src/pages/AddNewModel/configs/models.ts

@@ -3325,4 +3325,20 @@ export const models: { [key: string]: ModelPackage } = {
     icon: "mimo.png",
     isOpenSource: false,
   },
+
+  // Inception Labs models
+  mercury2: {
+    title: "Mercury 2",
+    description:
+      "Inception Labs' fastest reasoning LLM with 128k context, tool calling, and structured outputs.",
+    refUrl: "https://docs.inceptionlabs.ai/",
+    params: {
+      title: "Mercury 2",
+      model: "mercury-2",
+      contextLength: 128_000,
+    },
+    providerOptions: ["inception"],
+    icon: "inception.png",
+    isOpenSource: false,
+  },
 };

gui/src/pages/AddNewModel/configs/providers.ts

@@ -590,6 +590,28 @@ Select the \`GPT-4o\` model below to complete your provider configuration, but n
     ],
     apiKeyUrl: "https://platform.minimax.io",
   },
+  inception: {
+    title: "Inception Labs",
+    provider: "inception",
+    icon: "inception.png",
+    description:
+      "Inception Labs provides Mercury 2, a fast diffusion model with 128k context and tool calling.",
+    longDescription:
+      "To get started with Inception Labs, obtain an API key from the [Inception Labs platform](https://platform.inceptionlabs.ai/). Mercury 2 is OpenAI-compatible and supports chat, tool calling, and structured outputs.",
+    tags: [ModelProviderTags.RequiresApiKey],
+    collectInputFor: [
+      {
+        inputType: "text",
+        key: "apiKey",
+        label: "API Key",
+        placeholder: "Enter your Inception Labs API key",
+        required: true,
+      },
+      ...completionParamsInputsConfigs,
+    ],
+    packages: [models.mercury2],
+    apiKeyUrl: "https://platform.inceptionlabs.ai/",
+  },
   deepseek: {
     title: "DeepSeek",
     provider: "deepseek",

packages/llm-info/src/index.ts

@@ -4,6 +4,7 @@ import { Bedrock } from "./providers/bedrock.js";
 import { Cohere } from "./providers/cohere.js";
 import { CometAPI } from "./providers/cometapi.js";
 import { Gemini } from "./providers/gemini.js";
+import { Inception } from "./providers/inception.js";
 import { MiniMax } from "./providers/minimax.js";
 import { Mistral } from "./providers/mistral.js";
 import { Ollama } from "./providers/ollama.js";
@@ -26,6 +27,7 @@ export const allModelProviders: ModelProvider[] = [
   Bedrock,
   Cohere,
   CometAPI,
+  Inception,
   MiniMax,
   xAI,
   zAI,

packages/llm-info/src/providers/inception.ts

@@ -0,0 +1,32 @@
+import { ModelProvider } from "../types.js";
+
+export const Inception: ModelProvider = {
+  models: [
+    {
+      model: "mercury-2",
+      displayName: "Mercury 2",
+      contextLength: 128000,
+      description:
+        "Inception Labs' fastest reasoning diffusion model and their most powerful model, with tool calling and structured outputs support.",
+      regex: /mercury-2/i,
+      recommendedFor: ["chat"],
+    },
+    {
+      model: "mercury-edit-2",
+      displayName: "Mercury Edit 2",
+      contextLength: 32000,
+      description:
+        "Inception Labs' code editing model for autocomplete, apply edit, and next edit suggestions.",
+      regex: /mercury-edit-2/i,
+      recommendedFor: ["autocomplete"],
+    },
+    {
+      model: "mercury-coder-small",
+      displayName: "Mercury Coder Small",
+      contextLength: 32000,
+      regex: /mercury-coder-small/i,
+    },
+  ],
+  id: "inception",
+  displayName: "Inception Labs",
+};