Support disabling sort checks

api/requirements.py

@@ -71,7 +71,8 @@ async def categorize_reqs(params: ReqGroupingRequest, prompt_env: Environment =
         # check for remaining requirements assigned to none of the categories
         unassigned_ids = valid_ids_universe - valid_assigned_ids
 
-        if len(unassigned_ids) == 0:
+
+        if len(unassigned_ids) == 0 or params.disable_sort_checks:
             categories.extend(output.categories)
             break
         else:

schemas.py

@@ -118,6 +118,7 @@ class ReqGroupingRequest(BaseModel):
     requirements: list[RequirementInfo]
     max_n_categories: Optional[int] = Field(
         default=None, description="Max number of categories to construct. Defaults to None")
+    disable_sort_checks: bool = Field(default=False, description="Disable sort checks when grouping requirements")
 
 
 class ReqGroupingResponse(BaseModel):

static/index.html

@@ -240,30 +240,55 @@
         <div id="solutions-tab-contents" class="hidden pt-10">
             <!--Header de catégorisation des requirements-->
             <div class="w-full mx-auto mt-4 p-4 bg-base-100 rounded-box shadow-lg border border-base-200">
-                <!-- Séléction du nb de catégories / mode auto-->
-                <div class="flex flex-wrap justify-begin items-center gap-6">
-                    <!-- Number input -->
-                    <div class="form-control flex flex-col items-center justify-center">
-                        <label class="label" for="category-count">
-                            <span class="label-text text-base-content"># Categories</span>
-                        </label>
-                        <input id="category-count" type="number" class="input input-bordered w-24" min="1" value="3" />
-                    </div>
+                <div class="flex flex-wrap">
+                    <!-- First Group: Aligned to the start (left) -->
+                    <div class="flex items-center gap-6">
+                        <!-- Number input -->
+                        <div class="form-control flex flex-col items-center justify-center">
+                            <div class="tooltip" data-tip="Number of categories for sorting requirements">
+                                <label class="label" for="category-count">
+                                    <span class="label-text text-base-content"># Categories</span>
+                                </label>
+                            </div>
+                            <input id="category-count" type="number" class="input input-bordered w-24" min="1" value="3"
+                                disabled="">
+                        </div>
+
+                        <!-- Auto detect toggle -->
+                        <div class="form-control flex flex-col items-center justify-center">
+                            <div class="tooltip"
+                                data-tip="Whether to use the user provided value for sorting or determine automatically the number of categories">
+                                <label class="label">
+                                    <span class="label-text text-base-content">Number of categories</span>
+                                </label>
+                            </div>
+                            <input type="checkbox" class="toggle toggle-primary" id="auto-detect-toggle" checked="true">
+                            <div class="text-xs mt-1 text-base-content">Manual | Auto detect</div>
+                        </div>
 
-                    <!-- Auto detect toggle -->
-                    <div class="form-control flex flex-col items-center justify-center">
-                        <label class="label">
-                            <span class="label-text text-base-content">Number of categories</span>
-                        </label>
-                        <input type="checkbox" class="toggle toggle-primary" id="auto-detect-toggle" checked="true" />
-                        <div class="text-xs mt-1 text-base-content">Manual | Auto detect</div>
+                        <!-- Action Button -->
+                        <div class="tooltip" data-tip="Categorize the selected requirements">
+                            <button id="categorize-requirements-btn" class="btn btn-primary btn-l self-center">
+                                Categorize
+                            </button>
+                        </div>
                     </div>
 
-                    <!-- Action Button -->
-                    <button id="categorize-requirements-btn" class="btn btn-primary btn-l self-center">
-                        Categorize
-                    </button>
+                    <!-- Second Group: Pushed to the end (right) -->
+                    <div class="flex items-center gap-6 ml-auto"> <!-- Added ml-auto here -->
+                        <div class="form-control flex flex-col items-center justify-center">
+                            <div class="tooltip"
+                                data-tip="Disable the sorting checks. Use if models fails repeatedly at sorting requirements">
+                                <label class="label">
+                                    <span class="label-text text-base-content">Disable sort checks</span>
+                                </label>
+                            </div>
+                            <input type="checkbox" class="toggle toggle-error" id="disable-sort-checks">
+                            <div class="text-xs mt-1 text-base-content">Checks on | Checks off</div>
+                        </div>
+                    </div>
                 </div>
+
             </div>
             <div id="categorized-requirements-container pt-10" class="mb-6">
                 <div class="flex mb-4 pt-10">

static/js/app.js

@@ -420,6 +420,8 @@ async function categorizeRequirements(max_categories) {
         return;
     }
 
+    const disable_sort_checks = document.getElementById('disable-sort-checks').checked;
+
     showLoadingOverlay('Categorizing requirements...');
     toggleElementsEnabled(['categorize-requirements-btn'], false);
 
@@ -427,7 +429,7 @@ async function categorizeRequirements(max_categories) {
         const response = await fetch('/requirements/categorize_requirements', {
             method: 'POST',
             headers: { 'Content-Type': 'application/json' },
-            body: JSON.stringify({ requirements: formattedRequirements, "max_n_categories": max_categories })
+            body: JSON.stringify({ requirements: formattedRequirements, "max_n_categories": max_categories, "disable_sort_checks": disable_sort_checks })
         });
 
         const data = await response.json();
@@ -441,7 +443,7 @@ async function categorizeRequirements(max_categories) {
 
     } catch (error) {
         console.error('Error while categorizing requirements:', error);
-        alert('Error while categorizing requirements');
+        alert(`Error while categorizing requirements: ${e}. If model fails repeatedly, try disabling sort checks.`);
     } finally {
         hideLoadingOverlay();
         toggleElementsEnabled(['categorize-requirements-btn'], true);