1503 - Support new DataEntityType - VECTOR_STORE and new DatasetField… (

#1507) Co-authored-by: Andrey Nenashev <[email protected]>
opendatadiscovery · Nov 13, 2023 · ed33451 · ed33451
1 parent 7a49ed0
commit ed33451
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 3 deletions.
diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
@@ -2,7 +2,7 @@
 spring-webflux = '6.0.9'
 reactor-extra = '3.5.1'
 micrometer-registry-prometheus = '1.9.0'
-ingestion-contract-server = '0.1.27'
+ingestion-contract-server = '0.1.28'
 oddrn-generator-java = '0.1.20'
 odd-integration-manifests = '0.0.4'
 apache-collections = '4.4'

diff --git a/odd-platform-api/src/main/java/org/opendatadiscovery/oddplatform/dto/DataEntityClassDto.java b/odd-platform-api/src/main/java/org/opendatadiscovery/oddplatform/dto/DataEntityClassDto.java
@@ -32,11 +32,12 @@
 import static org.opendatadiscovery.oddplatform.dto.DataEntityTypeDto.ML_MODEL_INSTANCE;
 import static org.opendatadiscovery.oddplatform.dto.DataEntityTypeDto.ML_MODEL_TRAINING;
 import static org.opendatadiscovery.oddplatform.dto.DataEntityTypeDto.TABLE;
+import static org.opendatadiscovery.oddplatform.dto.DataEntityTypeDto.VECTOR_STORE;
 import static org.opendatadiscovery.oddplatform.dto.DataEntityTypeDto.VIEW;
 
 @Getter
 public enum DataEntityClassDto {
-    DATA_SET(1, Set.of(TABLE, FILE, FEATURE_GROUP, KAFKA_TOPIC, VIEW, GRAPH_NODE)),
+    DATA_SET(1, Set.of(TABLE, FILE, FEATURE_GROUP, KAFKA_TOPIC, VIEW, GRAPH_NODE, VECTOR_STORE)),
     DATA_TRANSFORMER(2, Set.of(JOB, ML_MODEL_TRAINING, ML_MODEL_INSTANCE, VIEW, MICROSERVICE)),
     DATA_TRANSFORMER_RUN(3, JOB_RUN),
     DATA_QUALITY_TEST(4, JOB),

diff --git a/odd-platform-api/src/main/java/org/opendatadiscovery/oddplatform/dto/DataEntityTypeDto.java b/odd-platform-api/src/main/java/org/opendatadiscovery/oddplatform/dto/DataEntityTypeDto.java
@@ -31,7 +31,8 @@ public enum DataEntityTypeDto {
     DATABASE_SERVICE(19),
     API_SERVICE(20),
     KAFKA_SERVICE(21),
-    DOMAIN(22);
+    DOMAIN(22),
+    VECTOR_STORE(23);
 
     private static final Map<Integer, DataEntityTypeDto> MAP = Arrays
         .stream(DataEntityTypeDto.values())

diff --git a/odd-platform-specification/components.yaml b/odd-platform-specification/components.yaml
@@ -737,6 +737,7 @@ components:
             - API_SERVICE
             - KAFKA_SERVICE
             - DOMAIN
+            - VECTOR_STORE
       required:
         - id
         - name
@@ -1748,6 +1749,7 @@ components:
             - TYPE_DURATION
             - TYPE_REFERENCE
             - TYPE_UNKNOWN
+            - TYPE_VECTOR
         logical_type:
           type: string
         is_nullable:

diff --git a/odd-platform-ui/src/redux/interfaces/datasetStructure.ts b/odd-platform-ui/src/redux/interfaces/datasetStructure.ts
@@ -39,6 +39,7 @@ export const DatasetTypeLabelMap: Record<
   TYPE_DURATION: { short: 'Dur', plural: 'durations' },
   TYPE_UNKNOWN: { short: 'Unk', plural: 'unknowns' },
   TYPE_REFERENCE: { short: 'Ref', plural: 'references' },
+  TYPE_VECTOR: { short: 'Vec', plural: 'vectors' },
 };
 
 export type DataSetFormattedStatsKeys = keyof Required<DataSetFormattedStats>;

diff --git a/odd-platform-ui/src/theme/palette.ts b/odd-platform-ui/src/theme/palette.ts
@@ -234,6 +234,7 @@ export const palette = createPalette({
     TYPE_DURATION: { border: colors.blue40 },
     TYPE_UNKNOWN: { border: colors.black10 },
     TYPE_REFERENCE: { border: colors.purple40 },
+    TYPE_VECTOR: { border: colors.orange40 },
   },
   datasetFieldKey: {
     primary: { background: colors.black30, color: colors.white, border: colors.blue30 },