From 32c83f0c106bbed2aba93ed83bc8511c144ef4b3 Mon Sep 17 00:00:00 2001
From: Bohdan Triapitsyn <artmore@protonmail.com>
Date: Fri, 2 Jan 2026 22:03:10 +0200
Subject: [PATCH] feat(multirun): enforce maximum model selection limit and
 update UI hints

---
 packages/ui/src/components/multirun/MultiRunLauncher.tsx | 8 +++++++-
 packages/ui/src/stores/useMultiRunStore.ts               | 5 +++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/packages/ui/src/components/multirun/MultiRunLauncher.tsx b/packages/ui/src/components/multirun/MultiRunLauncher.tsx
index 4f705cdd..961ab393 100644
--- a/packages/ui/src/components/multirun/MultiRunLauncher.tsx
+++ b/packages/ui/src/components/multirun/MultiRunLauncher.tsx
@@ -31,6 +31,9 @@ import type { ModelMetadata } from '@/types';
 /** Max file size in bytes (10MB) */
 const MAX_FILE_SIZE = 10 * 1024 * 1024;
 
+/** Max number of concurrent runs */
+const MAX_MODELS = 5;
+
 /** Attached file for multi-run (simplified from sessionStore's AttachedFile) */
 interface MultiRunAttachedFile {
   id: string;
@@ -609,6 +612,9 @@ export const MultiRunLauncher: React.FC<MultiRunLauncherProps> = ({
 
 
   const handleAddModel = (model: ModelSelectionWithId) => {
+    if (selectedModels.length >= MAX_MODELS) {
+      return;
+    }
     setSelectedModels((prev) => [...prev, model]);
     clearError();
   };
@@ -942,7 +948,7 @@ export const MultiRunLauncher: React.FC<MultiRunLauncherProps> = ({
             <div className="space-y-2">
               <label className="typography-ui-label font-medium text-foreground">
                 Models <span className="text-destructive">*</span>
-                <span className="ml-1 font-normal text-muted-foreground">(select at least 2)</span>
+                <span className="ml-1 font-normal text-muted-foreground">(select at least 2, maximum 5)</span>
               </label>
               <ModelMultiSelect
                 selectedModels={selectedModels}
diff --git a/packages/ui/src/stores/useMultiRunStore.ts b/packages/ui/src/stores/useMultiRunStore.ts
index 8652fe9c..713b9b59 100644
--- a/packages/ui/src/stores/useMultiRunStore.ts
+++ b/packages/ui/src/stores/useMultiRunStore.ts
@@ -91,6 +91,11 @@ export const useMultiRunStore = create<MultiRunStore>()(
           return null;
         }
 
+        if (models.length > 5) {
+          set({ error: 'Maximum 5 models allowed' });
+          return null;
+        }
+
         set({ isLoading: true, error: null });
 
         try {