[FEATURE] Add Mistral Vibe support

apps/server/scripts/acp-mock-agent.ts

@@ -21,6 +21,14 @@ const emitAskQuestion = process.env.T3_ACP_EMIT_ASK_QUESTION === "1";
 const emitXAiAskUserQuestion = process.env.T3_ACP_EMIT_XAI_ASK_USER_QUESTION === "1";
 const emitXAiPromptCompleteThenHang = process.env.T3_ACP_EMIT_XAI_PROMPT_COMPLETE_THEN_HANG === "1";
 const emitForeignSessionUpdates = process.env.T3_ACP_EMIT_FOREIGN_SESSION_UPDATES === "1";
+const useVibeConfig = process.env.T3_ACP_VIBE_CONFIG === "1";
+const vibeTrustStatus = process.env.T3_ACP_VIBE_TRUST_STATUS ?? "trusted";
+// Decisions Vibe currently has pending. Empty means no trust prompt is
+// outstanding, and `_trust/decision` is rejected the way the real agent does.
+const vibeTrustDecisions = (process.env.T3_ACP_VIBE_TRUST_DECISIONS ?? "")
+  .split(",")
+  .map((decision) => decision.trim())
+  .filter((decision) => decision.length > 0);
 const hangPromptForever = process.env.T3_ACP_HANG_PROMPT_FOREVER === "1";
 const hangFirstPromptForever = process.env.T3_ACP_HANG_FIRST_PROMPT_FOREVER === "1";
 const emitLateUpdateAfterCancel = process.env.T3_ACP_EMIT_LATE_UPDATE_AFTER_CANCEL === "1";
@@ -40,22 +48,39 @@ const failSetConfigOption = process.env.T3_ACP_FAIL_SET_CONFIG_OPTION === "1";
 const exitOnSetConfigOption = process.env.T3_ACP_EXIT_ON_SET_CONFIG_OPTION === "1";
 const promptResponseText = process.env.T3_ACP_PROMPT_RESPONSE_TEXT;
 const promptDelayMs = Number(process.env.T3_ACP_PROMPT_DELAY_MS ?? "0");
+const setConfigOptionDelayMs = Number(process.env.T3_ACP_SET_CONFIG_OPTION_DELAY_MS ?? "0");
 const permissionOptionIds = {
   allowOnce: process.env.T3_ACP_ALLOW_ONCE_OPTION_ID ?? "allow-once",
   allowAlways: process.env.T3_ACP_ALLOW_ALWAYS_OPTION_ID ?? "allow-always",
   rejectOnce: process.env.T3_ACP_REJECT_ONCE_OPTION_ID ?? "reject-once",
 };
 const sessionId = "mock-session-1";
 
-let currentModeId = "ask";
-let currentModelId = "default";
+let currentModeId = useVibeConfig ? "default" : "ask";
+let currentModelId = useVibeConfig ? "mistral-medium-3.5" : "default";
 let parameterizedModelPicker = false;
 let currentReasoning = "medium";
 let currentContext = "272k";
 let currentFast = false;
 let promptCount = 0;
 let overlappingFirstPromptId: string | undefined;
 const cancelledSessions = new Set<string>();
+const pendingCancellationResolvers = new Map<string, () => void>();
+
+function awaitCancellation(cancelledSessionId: string): Effect.Effect<void> {
+  return Effect.callback<void>((resume) => {
+    const resolve = () => resume(Effect.void);
+    pendingCancellationResolvers.set(cancelledSessionId, resolve);
+    if (cancelledSessions.has(cancelledSessionId)) {
+      resolve();
+    }
+    return Effect.sync(() => {
+      if (pendingCancellationResolvers.get(cancelledSessionId) === resolve) {
+        pendingCancellationResolvers.delete(cancelledSessionId);
+      }
+    });
+  });
+}
 
 function promptIdFromRequestMeta(
   request: Pick<AcpSchema.PromptRequest, "_meta">,
@@ -94,6 +119,47 @@ process.once("exit", (code) => {
 });
 
 function configOptions(): ReadonlyArray<AcpSchema.SessionConfigOption> {
+  if (useVibeConfig) {
+    return [
+      {
+        id: "mode",
+        name: "Mode",
+        category: "mode",
+        type: "select",
+        currentValue: currentModeId,
+        options: availableModes.map((mode) => ({
+          value: mode.id,
+          name: mode.name,
+          ...(mode.description ? { description: mode.description } : {}),
+        })),
+      },
+      {
+        id: "model",
+        name: "Model",
+        category: "model",
+        type: "select",
+        currentValue: currentModelId,
+        options: [
+          { value: "mistral-medium-3.5", name: "Mistral Medium 3.5" },
+          { value: "devstral-small", name: "Devstral Small" },
+          { value: "local", name: "Local" },
+        ],
+      },
+      {
+        id: "thinking",
+        name: "Thinking",
+        category: "model_config",
+        type: "select",
+        currentValue: "high",
+        options: [
+          { value: "off", name: "Off" },
+          { value: "low", name: "Low" },
+          { value: "high", name: "High" },
+        ],
+      },
+    ];
+  }
+
   if (parameterizedModelPicker) {
     const baseOptions: Array<AcpSchema.SessionConfigOption> = [
       {
@@ -253,23 +319,30 @@ function availableModels(): ReadonlyArray<{
   }));
 }
 
-const availableModes: ReadonlyArray<AcpSchema.SessionMode> = [
-  {
-    id: "ask",
-    name: "Ask",
-    description: "Request permission before making any changes",
-  },
-  {
-    id: "architect",
-    name: "Architect",
-    description: "Design and plan software systems without implementation",
-  },
-  {
-    id: "code",
-    name: "Code",
-    description: "Write and modify code with full tool access",
-  },
-];
+const availableModes: ReadonlyArray<AcpSchema.SessionMode> = useVibeConfig
+  ? [
+      { id: "default", name: "Default" },
+      { id: "accept-edits", name: "Accept edits" },
+      { id: "auto-approve", name: "Auto approve" },
+      { id: "plan", name: "Plan" },
+    ]
+  : [
+      {
+        id: "ask",
+        name: "Ask",
+        description: "Request permission before making any changes",
+      },
+      {
+        id: "architect",
+        name: "Architect",
+        description: "Design and plan software systems without implementation",
+      },
+      {
+        id: "code",
+        name: "Code",
+        description: "Write and modify code with full tool access",
+      },
+    ];
 
 function modeState(): AcpSchema.SessionModeState {
   return {
@@ -282,14 +355,20 @@ const grokAcpModels: ReadonlyArray<AcpSchema.ModelInfo> = [
   { modelId: "grok-build", name: "Grok Build" },
   { modelId: "grok-mock-alt", name: "Grok Mock Alt" },
 ];
+const vibeAcpModels: ReadonlyArray<AcpSchema.ModelInfo> = [
+  { modelId: "mistral-medium-3.5", name: "Mistral Medium 3.5" },
+  { modelId: "devstral-small", name: "Devstral Small" },
+  { modelId: "local", name: "Local" },
+];
 
 function modelState(): AcpSchema.SessionModelState {
-  const modelId = grokAcpModels.some((model) => model.modelId === currentModelId)
+  const availableModels = useVibeConfig ? vibeAcpModels : grokAcpModels;
+  const modelId = availableModels.some((model) => model.modelId === currentModelId)
     ? currentModelId
-    : "grok-build";
+    : availableModels[0]!.modelId;
   return {
     currentModelId: modelId,
-    availableModels: grokAcpModels,
+    availableModels,
   };
 }
 
@@ -382,7 +461,8 @@ const program = Effect.gen(function* () {
 
   yield* agent.handleSetSessionModel((request) =>
     Effect.gen(function* () {
-      if (!grokAcpModels.some((model) => model.modelId === request.modelId)) {
+      const availableModels = useVibeConfig ? vibeAcpModels : grokAcpModels;
+      if (!availableModels.some((model) => model.modelId === request.modelId)) {
         return yield* AcpError.AcpRequestError.invalidParams(
           `Unknown mock model id: ${request.modelId}`,
           {
@@ -403,6 +483,9 @@ const program = Effect.gen(function* () {
           process.exit(7);
         });
       }
+      if (Number.isFinite(setConfigOptionDelayMs) && setConfigOptionDelayMs > 0) {
+        yield* Effect.sleep(`${setConfigOptionDelayMs} millis`);
+      }
       if (failSetConfigOption) {
         return yield* AcpError.AcpRequestError.invalidParams(
           "Mock invalid params for session/set_config_option",
@@ -449,6 +532,7 @@ const program = Effect.gen(function* () {
           });
         });
       }
+      pendingCancellationResolvers.get(cancelledSessionId)?.();
     }),
   );
 
@@ -519,7 +603,9 @@ const program = Effect.gen(function* () {
       }
 
       if (hangPromptForever || (hangFirstPromptForever && promptCount === 1)) {
-        return yield* Effect.never;
+        yield* awaitCancellation(requestedSessionId);
+        cancelledSessions.delete(requestedSessionId);
+        return { stopReason: "cancelled" };
       }
 
       if (emitXAiPromptCompleteThenHang) {
@@ -878,6 +964,35 @@ const program = Effect.gen(function* () {
   );
 
   yield* agent.handleUnknownExtRequest((method, params) => {
+    if (method === "_trust/status") {
+      return Effect.succeed({
+        trust_status: vibeTrustStatus,
+        details:
+          vibeTrustDecisions.length > 0
+            ? {
+                cwd: process.cwd(),
+                repoRoot: null,
+                ignoredFiles: [],
+                availableDecisions: vibeTrustDecisions,
+              }
+            : null,
+      });
+    }
+    if (method === "_trust/decision") {
+      const requested =
+        typeof params === "object" && params !== null && "decision" in params
+          ? params.decision
+          : undefined;
+      // Matches the real agent: a decision can only be answered while one is
+      // pending, otherwise `-32602 No workspace trust decision is available`.
+      if (typeof requested !== "string" || !vibeTrustDecisions.includes(requested)) {
+        return Effect.fail(
+          AcpError.AcpRequestError.invalidParams("No workspace trust decision is available."),
+        );
+      }
+      return Effect.succeed({ trust_status: "trusted", details: null });
+    }
+
     if (method === "cursor/list_available_models") {
       return Effect.succeed({
         models: availableModels(),