fix: DIA-666: [FE] Semantic search polish

src/components/Filters/types/String.js

@@ -1,7 +1,6 @@
 import { observer } from "mobx-react";
 import React from "react";
 import { FilterInput } from "../FilterInput";
-import { VariantSelect } from "./List";
 
 const BaseInput = observer(({ value, onChange, placeholder }) => {
   return (
@@ -46,10 +45,4 @@ export const StringFilter = [
     valueType: "single",
     input: (props) => <BaseInput {...props} />,
   },
-  {
-    key: "similar_to",
-    label: "similar to",
-    valueType: "list",
-    input: (props) => <VariantSelect {...props} />,
-  },
 ];

src/stores/DataStores/tasks.js

@@ -7,6 +7,7 @@ import { DynamicModel, registerModel } from "../DynamicModel";
 import { CustomJSON } from "../types";
 import { FF_DEV_2536, FF_LOPS_E_3, isFF } from "../../utils/feature-flags";
 
+const SIMILARITY_UPPER_LIMIT_PRECISION = 1000;
 const fileAttributes = types.model({
   "certainty": types.optional(types.maybeNull(types.number), 0),
   "distance": types.optional(types.maybeNull(types.number), 0),
@@ -218,17 +219,19 @@ export const create = (columns) => {
       },
 
       postProcessData(data) {
-        const { total_annotations, total_predictions } = data;
+        const { total_annotations, total_predictions, similarity_score_upper_limit } = data;
 
         if (total_annotations !== null)
           self.totalAnnotations = total_annotations;
         if (total_predictions !== null)
           self.totalPredictions = total_predictions;
+        if (similarity_score_upper_limit !== null)
+          self.similarityUpperLimit = (Math.ceil(similarity_score_upper_limit * SIMILARITY_UPPER_LIMIT_PRECISION) / SIMILARITY_UPPER_LIMIT_PRECISION);
       },
 
     }))
     .preProcessSnapshot((snapshot) => {
-      const { total_annotations, total_predictions, ...sn } = snapshot;
+      const { total_annotations, total_predictions, similarity_score_upper_limit, ...sn } = snapshot;
 
       return {
         ...sn,
@@ -239,6 +242,7 @@ export const create = (columns) => {
         })),
         totalAnnotations: total_annotations,
         totalPredictions: total_predictions,
+        similarityUpperLimit: similarity_score_upper_limit,
       };
     });
 };

src/stores/Tabs/tab.js

@@ -17,8 +17,6 @@ import { History } from '../../utils/history';
 import { FF_DEV_1470, FF_LOPS_12, isFF } from "../../utils/feature-flags";
 import { CustomJSON, StringOrNumberID, ThresholdType } from "../types";
 
-const DEFAULT_THRESHOLD = { min: 0, max: 10 };
-
 export const Tab = types
   .model("View", {
     id: StringOrNumberID,
@@ -53,7 +51,7 @@ export const Tab = types
     editable: true,
     deletable: true,
     semantic_search: types.optional(types.array(CustomJSON), []),
-    threshold: types.optional(ThresholdType, DEFAULT_THRESHOLD),
+    threshold: types.optional(types.maybeNull(ThresholdType), null),
   })
   .volatile(() => {
     const defaultWidth = getComputedStyle(document.body).getPropertyValue("--menu-sidebar-width").replace("px", "").trim();
@@ -218,7 +216,7 @@ export const Tab = types
         columnsDisplayType: self.columnsDisplayType.toPOJO(),
         gridWidth: self.gridWidth,
         semantic_search: self.semantic_search?.toJSON() ?? [],
-        threshold: self.threshold?.toJSON() ?? DEFAULT_THRESHOLD,
+        threshold: self.threshold?.toJSON(),
       };
 
       if (self.saved || apiVersion === 1) {
@@ -305,16 +303,24 @@ export const Tab = types
       self.selected = ids;
     },
 
-    setSemanticSearch(semanticSearchList) {
+    setSemanticSearch(semanticSearchList, save = true) {
       self.semantic_search = semanticSearchList ?? [];
-      return self.save();
+      if (self.semantic_search.length === 0) {
+        self.threshold = null;
+      }
+      return save && self.save();
     },
 
     setSemanticSearchThreshold(min, max) {
       self.threshold = { min, max };
       return self.save();
     },
 
+    clearSemanticSearchThreshold(save = true) {
+      self.threshold = null;
+      return save && self.save();
+    },
+
     selectAll() {
       self.selected.toggleSelectedAll();
     },
@@ -381,6 +387,8 @@ export const Tab = types
       } else if (isFF(FF_LOPS_12) && self.root.SDK.type === 'labelops') {
         yield self.dataStore.reload({ query: self.query, interaction });
       }
+
+      getRoot(self).SDK.invoke("tabReloaded", self);
     }),
 
     deleteFilter(filter) {