@finos/legend-graph 32.3.22 → 32.3.24

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_DataProductSerializationHelper.ts

@@ -48,6 +48,11 @@ import {
   V1_Expertise,
   V1_FunctionAccessPoint,
   V1_DataProductOperationalMetadata,
+  V1_NativeModelAccess,
+  type V1_SampleQuery,
+  V1_PackageableElementSampleQuery,
+  V1_InLineSampleQuery,
+  V1_NativeModelExecutionContext,
 } from '../../../model/packageableElements/dataProduct/V1_DataProduct.js';
 import {
   UnsupportedOperationError,
@@ -87,6 +92,11 @@ export enum V1_AccessPointGrouptype {
   MODEL_ACCESS_POINT_GROUP = 'modelAccessPointGroup',
 }
 
+export enum V1_SampleQueryType {
+  IN_LINE_SAMPLE_QUERY = 'inLineSampleQuery',
+  PACKAGEABLE_ELEMENT_SAMPLE_QUERY = 'packageableElementSampleQuery',
+}
+
 export enum V1_DataProductIconType {
   LIBRARY_ICON = 'libraryIcon',
   EMBEDDED_IMAGE_ICON = 'embeddedImageIcon',
@@ -199,6 +209,85 @@ export const V1_ModelAccessPointGroupModelSchema = createModelSchema(
   },
 );
 
+export const V1_PackageableElementSampleQueryModelSchema = createModelSchema(
+  V1_PackageableElementSampleQuery,
+  {
+    _type: usingConstantValueSchema(
+      V1_SampleQueryType.PACKAGEABLE_ELEMENT_SAMPLE_QUERY,
+    ),
+    description: optional(primitive()),
+    executionContextKey: optional(primitive()),
+    id: primitive(),
+    query: usingModelSchema(V1_packageableElementPointerModelSchema),
+    title: primitive(),
+  },
+);
+
+export const V1_InLineSampleQueryModelSchema = createModelSchema(
+  V1_InLineSampleQuery,
+  {
+    _type: usingConstantValueSchema(V1_SampleQueryType.IN_LINE_SAMPLE_QUERY),
+    description: optional(primitive()),
+    executionContextKey: optional(primitive()),
+    id: primitive(),
+    query: usingModelSchema(V1_rawLambdaModelSchema),
+    title: primitive(),
+  },
+);
+
+const V1_serializeSampleQuery = (
+  protocol: V1_SampleQuery,
+): PlainObject<V1_SampleQuery> => {
+  if (protocol instanceof V1_InLineSampleQuery) {
+    return serialize(V1_InLineSampleQueryModelSchema, protocol);
+  } else if (protocol instanceof V1_PackageableElementSampleQuery) {
+    return serialize(V1_PackageableElementSampleQueryModelSchema, protocol);
+  }
+  throw new UnsupportedOperationError(
+    `Can't serialize data product sample query`,
+    protocol,
+  );
+};
+
+export const V1_deserializeSampleQuery = (
+  json: PlainObject<V1_SampleQuery>,
+): V1_SampleQuery => {
+  switch (json._type) {
+    case V1_SampleQueryType.IN_LINE_SAMPLE_QUERY:
+      return deserialize(V1_InLineSampleQueryModelSchema, json);
+    case V1_SampleQueryType.PACKAGEABLE_ELEMENT_SAMPLE_QUERY:
+    default:
+      return deserialize(V1_PackageableElementSampleQueryModelSchema, json);
+  }
+};
+
+export const V1_NativeModelExecutionContextSchema = createModelSchema(
+  V1_NativeModelExecutionContext,
+  {
+    key: primitive(),
+    mapping: usingModelSchema(V1_packageableElementPointerModelSchema),
+    runtime: optionalCustomUsingModelSchema(
+      V1_packageableElementPointerModelSchema,
+    ),
+  },
+);
+
+export const V1_NativeModelAccessModelSchema = createModelSchema(
+  V1_NativeModelAccess,
+  {
+    diagrams: optionalCustomListWithSchema(V1_dataProductDiagramModelSchema),
+    defaultExecutionContext: primitive(),
+    featuredElements: optionalCustomListWithSchema(V1_ElementScopeModelSchema),
+    nativeModelExecutionContexts: customListWithSchema(
+      V1_NativeModelExecutionContextSchema,
+    ),
+    sampleQueries: optionalCustomList(
+      (val: V1_SampleQuery) => V1_serializeSampleQuery(val),
+      (val) => V1_deserializeSampleQuery(val),
+    ),
+  },
+);
+
 export const V1_DefaultAccessPointGroupModelSchema = createModelSchema(
   V1_AccessPointGroup,
   {
@@ -379,6 +468,9 @@ export const V1_dataProductModelSchema = (
       V1_deserializeDataProductIcon,
     ),
     name: primitive(),
+    nativeModelAccess: optionalCustomUsingModelSchema(
+      V1_NativeModelAccessModelSchema,
+    ),
     operationalMetadata: optionalCustomUsingModelSchema(
       V1_DataProductOperationalMetadataModelSchema,
     ),

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_EntitlementSerializationHelper.ts

@@ -249,6 +249,7 @@ export const V1_dataContractModelSchema = (
       (val) => V1_deserializeOrganizationalScope(val, plugins),
     ),
     createdBy: primitive(),
+    createdAt: primitive(),
   });
 
 export const V1_dataContractSubscriptionsModelSchema = (
@@ -276,6 +277,7 @@ export const V1_liteDataContractModelSchema = (
       (val) => V1_serializeOrganizationalScope(val, plugins),
       (val) => V1_deserializeOrganizationalScope(val, plugins),
     ),
+    createdAt: primitive(),
     createdBy: primitive(),
     resourceId: primitive(),
     resourceType: primitive(),
@@ -323,6 +325,7 @@ export const V1_liteDataContractWithUserStatusModelSchema = (
     pendingTaskWithAssignees: optional(
       usingModelSchema(V1_pendingTaskWithAssigneesModelSchema),
     ),
+    user: primitive(),
   });
 
 export const V1_contractUserEventPrivilegeManagerPayloadModelSchema =

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_ServiceSerializationHelper.ts

@@ -383,4 +383,5 @@ export const V1_serviceModelSchema = (
     postValidations: customListWithSchema(V1_servicePostValidationModelSchema, {
       INTERNAL__forceReturnEmptyInTest: true,
     }),
+    mcpServer: optional(primitive()),
   });

package/src/index.ts

@@ -399,6 +399,14 @@ export {
   V1_DataProductArtifact,
   V1_AccessPointImplementation,
   V1_AccessPointGroupInfo,
+  V1_SampleQueryInfo,
+  V1_InLineSampleQueryInfo,
+  V1_PackageableElementSampleQueryInfo,
+  V1_NativeModelAccessInfo,
+  V1_DiagramInfo,
+  V1_EnumerationDocumentationEntry,
+  V1_AssociationDocumentationEntry,
+  V1_ClassDocumentationEntry,
 } from './graph-manager/protocol/pure/v1/lakehouse/deploy/V1_DataProductArtifact.js';
 export {
   V1_IngestEnvironmentType,
@@ -770,6 +778,11 @@ export {
   LakehouseTargetEnv,
   SupportInfo,
   ModelAccessPointGroup,
+  InLineSampleQuery,
+  NativeModelAccess,
+  NativeModelExecutionContext,
+  PackageableElementSampleQuery,
+  SampleQuery,
   DataProductRuntimeInfo,
   UnknownDataProductIcon,
   DataProductElementScope,