feat: UTC-443: Add per-dimension agreements to DM (#9031)

Co-authored-by: hakan458 <hakan@heartex.com>
Co-authored-by: niklub <niklub@users.noreply.github.com>
Co-authored-by: ricardoantoniocm <ricardoantoniocm@users.noreply.github.com>
Co-authored-by: Ricardo Cabral <ricardo.cabralm@gmail.com>

Author: 4 people

label_studio/data_manager/managers.py

@@ -804,6 +804,22 @@ def annotate_queryset(
         queryset, fields_for_evaluation=None, all_fields=False, excluded_fields_for_evaluation=None, request=None
     ):
         annotations_map = get_annotations_map()
+        # If we have dynamic control-tag level agreement columns, inject into annotation map
+        # without mutating the global map
+        if flag_set('fflag_utc_428_consensus_control_tag_agreement', user='auto'):
+            inject_path = getattr(settings, 'GET_DYNAMIC_DM_ANNOTATIONS', None)
+            if inject_path:
+                overlay_func = load_func(inject_path)
+                # Expect a dict of {field_name: function that annotates the queryset}
+                overlay_map = overlay_func(request=request, project=getattr(queryset.first(), 'project', None)) or {}
+                if isinstance(overlay_map, dict) and overlay_map:
+                    # Ensure dynamic fields are evaluated even if not explicitly selected
+                    if fields_for_evaluation is None:
+                        fields_for_evaluation = list(overlay_map.keys())
+                    else:
+                        fields_for_evaluation = list(set(fields_for_evaluation) | set(overlay_map.keys()))
+                    # Merge overlay with base map for this call only
+                    annotations_map = {**annotations_map, **overlay_map}
 
         if fields_for_evaluation is None:
             fields_for_evaluation = []

web/libs/datamanager/src/components/CellViews/Agreement/Agreement.jsx

@@ -29,8 +29,20 @@ export const Agreement = (cell) => {
   const { value, original: task } = cell;
   const sdk = useSDK();
   const [content, setContent] = useState(null);
-  const isAgreementPopoverEnabled =
-    window.APP_SETTINGS.billing?.enterprise && ff.isActive(FF_AVERAGE_AGREEMENT_SCORE_POPOVER);
+  const basePopoverEnabled = window.APP_SETTINGS.billing?.enterprise && ff.isActive(FF_AVERAGE_AGREEMENT_SCORE_POPOVER);
+
+  const colId =
+    (cell && cell.column && typeof cell.column.id === "string" && cell.column.id) ||
+    (cell &&
+      cell.column &&
+      cell.column.original &&
+      typeof cell.column.original.alias === "string" &&
+      cell.column.original.alias) ||
+    "";
+
+  const colPath = String(colId).split(":").pop() || "";
+  const isDimensionAgreementColumn = colPath.startsWith("dimension_agreement__");
+  const isAgreementPopoverEnabled = basePopoverEnabled && !isDimensionAgreementColumn;
 
   const handleClick = isAgreementPopoverEnabled
     ? (e) => {

web/libs/datamanager/src/components/CellViews/index.js

@@ -23,5 +23,11 @@ export function normalizeCellAlias(alias) {
   // remove trailing separators to make `toStudlyCaps` safe
   const safeAlias = alias.replace(/[-_\s]+$/g, "");
 
+  // Treat dimension agreement columns like the built-in agreement column
+  // so they use the same percentage formatting and coloring.
+  if (safeAlias === "agreement" || safeAlias.startsWith("dimension_agreement__")) {
+    return "Agreement";
+  }
+
   return toStudlyCaps(safeAlias);
 }

web/libs/datamanager/src/components/MainView/DataView/Table.jsx

@@ -131,6 +131,22 @@ export const DataView = injector(
             {original?.readableType ?? parent.title}
           </Tag>,
         );
+      } else if (typeof original?.alias === "string" && original.alias.startsWith("dimension_agreement__")) {
+        // Show a short tag for per-dimension agreement columns (root columns, no parent)
+        children.push(
+          <Tag
+            key="column-type"
+            color="blue"
+            style={{
+              fontWeight: "500",
+              fontSize: 14,
+              cursor: "pointer",
+              padding: "0 6px",
+            }}
+          >
+            {original.readableType}
+          </Tag>,
+        );
       }
 
       if (help && decoration?.help !== false) {

web/libs/datamanager/src/stores/Tabs/tab_column.jsx

@@ -187,6 +187,12 @@ export const TabColumn = types
     },
 
     get readableType() {
+      // Show a friendly tag for per-dimension agreement columns
+      if (typeof self.alias === "string") {
+        if (self.alias.startsWith("dimension_agreement__")) {
+          return "agreement";
+        }
+      }
       return ViewColumnTypeShort(self.currentType);
     },
 

web/libs/ui/src/lib/select/select.module.scss

@@ -12,6 +12,8 @@
   text-overflow: ellipsis;
   height: var(--select-trigger-height);
   width: 100%;
+  background: var(--color-neutral-background);
+  transition: background 150ms ease-out, border-color 150ms ease-out;
 
   /* body/medium */
   font-size: var(--select-trigger-font-size);