@finos/legend-graph 29.0.4 → 30.0.0

package/lib/graph-manager/protocol/pure/v1/V1_PureGraphManager.js

@@ -15,7 +15,7 @@
  */
 import { GRAPH_MANAGER_EVENT } from '../../../../__lib__/GraphManagerEvent.js';
 import { CORE_PURE_PATH, PackageableElementPointerType, } from '../../../../graph/MetaModelConst.js';
-import { ActionState, TracerService, LogEvent, getClass, guaranteeNonNullable, UnsupportedOperationError, assertErrorThrown, promisify, StopWatch, isNonNullable, addUniqueEntry, uuid, deleteEntry, uniq, IllegalStateError, filterByType, isString, } from '@finos/legend-shared';
+import { ActionState, TracerService, LogEvent, getClass, guaranteeNonNullable, UnsupportedOperationError, assertErrorThrown, promisify, StopWatch, isNonNullable, addUniqueEntry, uuid, deleteEntry, IllegalStateError, filterByType, isString, } from '@finos/legend-shared';
 import { AbstractPureGraphManager, } from '../../../../graph-manager/AbstractPureGraphManager.js';
 import { PureModel, } from '../../../../graph/PureModel.js';
 import { RawLambda } from '../../../../graph/metamodel/pure/rawValueSpecification/RawLambda.js';
@@ -63,7 +63,7 @@ import { V1_PureModelContextComposite } from './model/context/V1_PureModelContex
 import { V1_LegendSDLC } from './model/context/V1_SDLC.js';
 import { V1_Protocol } from './model/V1_Protocol.js';
 import { V1_GraphBuilderExtensions } from './transformation/pureGraph/to/V1_GraphBuilderExtensions.js';
-import { V1_DatabaseBuilderConfig, V1_DatabaseBuilderInput, V1_DatabasePattern, V1_setupDatabaseBuilderInputSerialization, V1_TargetDatabase, } from './engine/generation/V1_DatabaseBuilderInput.js';
+import { V1_DatabaseBuilderConfig, V1_DatabaseBuilderInput, V1_DatabasePattern, V1_TargetDatabase, } from './engine/generation/V1_DatabaseBuilderInput.js';
 import { V1_transformRelationalDatabaseConnection } from './transformation/pureGraph/from/V1_ConnectionTransformer.js';
 import { V1_FlatData } from './model/packageableElements/store/flatData/model/V1_FlatData.js';
 import { V1_Database } from './model/packageableElements/store/relational/model/V1_Database.js';
@@ -103,12 +103,13 @@ import { V1_EntitlementReportAnalyticsInput, V1_StoreEntitlementAnalysisInput, V
 import { LegendSDLC, GraphEntities, } from '../../../../graph/GraphDataOrigin.js';
 import { InMemoryGraphData, GraphDataWithOrigin, } from '../../../GraphData.js';
 import { DEPRECATED__validate_MappingTest } from '../../../action/validation/DSL_Mapping_ValidationHelper.js';
-import { V1_SERVICE_ELEMENT_PROTOCOL_TYPE } from './transformation/pureProtocol/serializationHelpers/V1_ServiceSerializationHelper.js';
 import { V1_INTERNAL__UnknownPackageableElement } from './model/packageableElements/V1_INTERNAL__UnknownPackageableElement.js';
 import { FunctionActivatorConfiguration } from '../../../action/functionActivator/FunctionActivatorConfiguration.js';
 import { V1_FunctionActivatorInput } from './engine/functionActivator/V1_FunctionActivatorInput.js';
 import { V1_FunctionActivator } from './model/packageableElements/function/V1_FunctionActivator.js';
 import { V1_INTERNAL__UnknownFunctionActivator } from './model/packageableElements/function/V1_INTERNAL__UnknownFunctionActivator.js';
+import { V1_RawSQLExecuteInput } from './engine/execution/V1_RawSQLExecuteInput.js';
+import { V1_INTERNAL__UnknownStore } from './model/packageableElements/store/V1_INTERNAL__UnknownStore.js';
 class V1_PureModelContextDataIndex {
     elements = [];
     nativeElements = [];
@@ -136,8 +137,10 @@ const mergePureModelContextData = (...data) => {
     const mergedData = new V1_PureModelContextData();
     for (const _data of data) {
         mergedData.elements = mergedData.elements.concat(_data.elements);
-        mergedData.INTERNAL__rawDependencyEntities =
-            mergedData.INTERNAL__rawDependencyEntities.concat(_data.INTERNAL__rawDependencyEntities);
+        const rawDependencyEntities = (mergedData.INTERNAL__rawDependencyEntities ?? []).concat(_data.INTERNAL__rawDependencyEntities ?? []);
+        mergedData.INTERNAL__rawDependencyEntities = rawDependencyEntities.length
+            ? rawDependencyEntities
+            : undefined;
         mergedData.serializer = _data.serializer ?? mergedData.serializer;
         mergedData.origin = _data.origin ?? mergedData.origin;
     }
@@ -267,6 +270,10 @@ export const V1_indexPureModelContextData = (report, data, extensions) => {
 };
 class V1_PureGraphManager extends AbstractPureGraphManager {
     elementClassifierPathMap = new Map();
+    subtypeInfo = {
+        storeSubtypes: [],
+        functionActivatorSubtypes: [],
+    };
     // Pure Client Version represent the version of the pure protocol.
     // Most Engine APIs will interrupt an undefined pure client version to mean
     // use the latest production version of the protocol i.e V20_0_0, while version
@@ -281,12 +288,6 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         this.engine = new V1_Engine({}, logService);
         // setup plugins
         this.graphBuilderExtensions = new V1_GraphBuilderExtensions(this.pluginManager.getPureProtocolProcessorPlugins());
-        // setup serialization plugins
-        V1_setupPureModelContextDataSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
-        V1_setupDatabaseSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
-        V1_setupEngineRuntimeSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
-        V1_setupLegacyRuntimeSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
-        V1_setupDatabaseBuilderInputSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
     }
     TEMPORARY__getEngineConfig() {
         return this.engine.config;
@@ -296,13 +297,31 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         this.engine
             .getEngineServerClient()
             .setTracerService(options?.tracerService ?? new TracerService());
-        await this.engine.setup(config);
+        await Promise.all([
+            this.engine.setup(config),
+            this.configureElementClassifierPathMap(config),
+            this.configureSubtypeInfoMap(config),
+        ]);
+        // setup serialization plugins
+        V1_setupPureModelContextDataSerialization(this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
+        V1_setupDatabaseSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
+        V1_setupEngineRuntimeSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
+        V1_setupLegacyRuntimeSerialization(this.pluginManager.getPureProtocolProcessorPlugins());
+    }
+    async configureElementClassifierPathMap(config) {
         const classifierPathMapEntries = config.TEMPORARY__classifierPathMapping ??
             (await this.engine.getClassifierPathMapping());
         classifierPathMapEntries.forEach((entry) => {
             this.elementClassifierPathMap.set(entry.type, entry.classifierPath);
         });
     }
+    async configureSubtypeInfoMap(config) {
+        const subtypeInfo = config.TEMPORARY__subtypeInfo ?? (await this.engine.getSubtypeInfo());
+        this.subtypeInfo.storeSubtypes = subtypeInfo.storeSubtypes;
+        this.subtypeInfo.functionActivatorSubtypes =
+            subtypeInfo.functionActivatorSubtypes;
+    }
+    // --------------------------------------------- Generic ---------------------------------------------
     getSupportedProtocolVersion() {
         return PureClientVersion.V1_0_0;
     }
@@ -366,7 +385,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             await Promise.all(Array.from(dependencyEntitiesIndex.entries()).map(([dependencyKey, entitiesWithOrigin]) => {
                 const projectModelData = new V1_PureModelContextData();
                 dependencyGraphDataIndex.set(dependencyKey, projectModelData);
-                return V1_entitiesToPureModelContextData(entitiesWithOrigin.entities, projectModelData, this.pluginManager.getPureProtocolProcessorPlugins());
+                return V1_entitiesToPureModelContextData(entitiesWithOrigin.entities, projectModelData, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
             }));
             stopWatch.record(GRAPH_MANAGER_EVENT.GRAPH_BUILDER_DESERIALIZE_ELEMENTS__SUCCESS);
             // prepare build inputs
@@ -408,7 +427,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             // deserialize
             buildState.setMessage(`Deserializing elements...`);
             const data = new V1_PureModelContextData();
-            await V1_entitiesToPureModelContextData(entities, data, this.pluginManager.getPureProtocolProcessorPlugins());
+            await V1_entitiesToPureModelContextData(entities, data, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
             stopWatch.record(GRAPH_MANAGER_EVENT.GRAPH_BUILDER_DESERIALIZE_ELEMENTS__SUCCESS);
             // prepare build inputs
             const buildInputs = [
@@ -462,7 +481,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             // deserialize
             buildState.setMessage(`Deserializing elements...`);
             const data = new V1_PureModelContextData();
-            await V1_entitiesToPureModelContextData(entities, data, this.pluginManager.getPureProtocolProcessorPlugins());
+            await V1_entitiesToPureModelContextData(entities, data, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
             stopWatch.record(GRAPH_MANAGER_EVENT.GRAPH_BUILDER_DESERIALIZE_ELEMENTS__SUCCESS);
             // prepare build inputs
             const buildInputs = [
@@ -517,7 +536,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             await Promise.all(Array.from(generatedEntities.entries()).map(([generationParentPath, entities]) => {
                 const generatedData = new V1_PureModelContextData();
                 generationGraphDataIndex.set(generationParentPath, generatedData);
-                return V1_entitiesToPureModelContextData(entities, generatedData, this.pluginManager.getPureProtocolProcessorPlugins());
+                return V1_entitiesToPureModelContextData(entities, generatedData, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
             }));
             stopWatch.record(GRAPH_MANAGER_EVENT.GRAPH_BUILDER_DESERIALIZE_ELEMENTS__SUCCESS);
             // prepare build inputs
@@ -1018,7 +1037,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
     // --------------------------------------------- Execution ---------------------------------------------
     prepareExecutionContextGraphData(graphData) {
         if (graphData instanceof InMemoryGraphData) {
-            return this.buildExecutionInputGraphData(graphData.graph);
+            return this.getFullGraphModelData(graphData.graph);
         }
         else if (graphData instanceof GraphDataWithOrigin) {
             return this.buildPureModelSDLCPointer(graphData.origin, V1_PureGraphManager.PROD_PROTOCOL_VERSION);
@@ -1027,7 +1046,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
     }
     createExecutionInput = (graph, mapping, lambda, runtime, clientVersion, parameterValues) => this.createExecutionInputWithPureModelContext(graph.origin
         ? this.buildPureModelSDLCPointer(graph.origin, undefined)
-        : this.buildExecutionInputGraphData(graph), mapping, lambda, runtime, clientVersion, new V1_ExecuteInput(), parameterValues);
+        : this.getFullGraphModelData(graph), mapping, lambda, runtime, clientVersion, new V1_ExecuteInput(), parameterValues);
     createExecutionInputWithPureModelContext = (data, mapping, lambda, runtime, clientVersion, executeInput, parameterValues) => {
         // NOTE: for execution, we usually will just assume that we send the connections embedded in the runtime value, since we don't want the user to have to create
         // packageable runtime and connection just to play with execution.
@@ -1044,42 +1063,6 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         }
         return executeInput;
     };
-    buildExecutionInputGraphData(graph) {
-        /**
-         * NOTE: to lessen network load, we might need to think of a way to only include relevant part of the pure model context data here
-         *
-         * Graph data models can be classified based on dependency hieararchy:
-         * 1. Building blocks: models that all other models depend on: e.g. domain models, connections, etc.
-         * 2. Consumers: models that depends on other models: e.g. mapping, service, etc.
-         * 3. Unrelated: models that depends on nothing and vice versa: e.g. text
-         *
-         * It would be great if we can provide a way to walk the mapping to select only relevant part, but the problem is we cannot really walk the lambda
-         * object to identify relevant classes yet. so the more economical way to to base on the classification above and the knowledge about hierarchy between
-         * models (e.g. service can use mapping, runtime, connection, store, etc.) we can roughly prune the graph model data by group. Following is an example
-         * for mapping used for execution, but this can generalized if we introduce hierarchy/ranking for model type
-         */
-        const graphData = this.getFullGraphModelData(graph);
-        const prunedGraphData = this.prunePureModelContextData(graphData, (element) => element instanceof V1_Class ||
-            element instanceof V1_Enumeration ||
-            element instanceof V1_Profile ||
-            element instanceof V1_Association ||
-            element instanceof V1_ConcreteFunctionDefinition ||
-            element instanceof V1_FunctionActivator ||
-            element instanceof V1_Measure ||
-            element instanceof V1_Store ||
-            element instanceof V1_PackageableConnection ||
-            element instanceof V1_PackageableRuntime ||
-            element instanceof V1_Mapping, undefined);
-        const extraExecutionElements = this.pluginManager
-            .getPureProtocolProcessorPlugins()
-            .flatMap((element) => element.V1_getExtraExecutionInputCollectors?.() ?? [])
-            .flatMap((getter) => getter(graph, graphData));
-        prunedGraphData.elements = uniq([
-            ...prunedGraphData.elements,
-            ...extraExecutionElements,
-        ]);
-        return prunedGraphData;
-    }
     async runQuery(lambda, mapping, runtime, graph, options, _report) {
         const report = _report ?? createGraphManagerOperationReport();
         const stopWatch = new StopWatch();
@@ -1098,7 +1081,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         const stopWatch = new StopWatch();
         const pureModelContext = graph.origin
             ? this.buildPureModelSDLCPointer(graph.origin, undefined)
-            : this.buildExecutionInputGraphData(graph);
+            : this.getFullGraphModelData(graph);
         stopWatch.record(GRAPH_MANAGER_EVENT.V1_ENGINE_OPERATION_INPUT__SUCCESS);
         await Promise.all(tests.map((t) => this.DEPRECATED__runLegacyMappingTest(t, mapping, pureModelContext, options, report)));
     }
@@ -1157,7 +1140,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         const testDataGenerationExecuteInput = new V1_TestDataGenerationExecutionInput();
         this.createExecutionInputWithPureModelContext(graph.origin
             ? this.buildPureModelSDLCPointer(graph.origin, undefined)
-            : this.buildExecutionInputGraphData(graph), mapping, lambda, runtime, V1_PureGraphManager.DEV_PROTOCOL_VERSION, testDataGenerationExecuteInput);
+            : this.getFullGraphModelData(graph), mapping, lambda, runtime, V1_PureGraphManager.DEV_PROTOCOL_VERSION, testDataGenerationExecuteInput, options?.parameterValues);
         testDataGenerationExecuteInput.parameters = parameters;
         testDataGenerationExecuteInput.hashStrings = Boolean(options?.anonymizeGeneratedData);
         stopWatch.record(GRAPH_MANAGER_EVENT.V1_ENGINE_OPERATION_INPUT__SUCCESS);
@@ -1225,7 +1208,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         input.mapping = mapping.path;
         input.model = graph.origin
             ? this.buildPureModelSDLCPointer(graph.origin, undefined)
-            : this.buildExecutionInputGraphData(graph);
+            : this.getFullGraphModelData(graph);
         return V1_buildModelCoverageAnalysisResult(await this.engine.analyzeMappingModelCoverage(input), mapping);
     }
     buildMappingModelCoverageAnalysisResult(input, mapping) {
@@ -1273,7 +1256,13 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             return pattern;
         });
         dbBuilderInput.config = config;
-        return this.pureModelContextDataToEntities(await this.engine.buildDatabase(dbBuilderInput));
+        return this.pureModelContextDataToEntities(await this.engine.buildDatabase(dbBuilderInput, this.pluginManager.getPureProtocolProcessorPlugins()));
+    }
+    async executeRawSQL(connection, sql) {
+        const input = new V1_RawSQLExecuteInput();
+        input.connection = V1_transformRelationalDatabaseConnection(connection, new V1_GraphTransformerContextBuilder(this.pluginManager.getPureProtocolProcessorPlugins()).build());
+        input.sql = sql;
+        return this.engine.executeRawSQL(input, this.pluginManager.getPureProtocolProcessorPlugins());
     }
     // --------------------------------------------- Function ---------------------------------------------
     async getAvailableFunctionActivatorConfigurations(coreModel, systemModel) {
@@ -1331,8 +1320,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         const protocol = new V1_Protocol(V1_PureGraphManager.PURE_PROTOCOL_NAME, serverServiceInfo.services.dependencies.pure);
         switch (executionMode) {
             case ServiceExecutionMode.FULL_INTERACTIVE: {
-                const data = this.createServiceRegistrationInputGraphData(graph);
-                data.elements.push(this.elementToProtocol(service));
+                const data = this.TEMPORARY__prepareGraphDataForServiceRegistration(this.getFullGraphModelData(graph), this.elementToProtocol(service));
                 data.origin = new V1_PureModelContextPointer(protocol);
                 input = data;
                 break;
@@ -1474,25 +1462,27 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         const serviceStorage = await this.engine.getServiceVersionInfo(serviceUrl, serviceId);
         await this.engine.activateServiceGeneration(serviceUrl, serviceStorage.getGenerationId());
     }
-    // NOTE: We almost should never be poking into dependency entities. However for service registration the input
-    // expects only one service in the graph data. Perhaps, the service registration API should be modified to accept
-    // service as a parameter outside the model
-    pruneServicesFromGraphForRegistration = (graphData) => this.prunePureModelContextData(graphData, (element) => !(element instanceof V1_Service), (entity) => {
-        const content = entity.content;
-        return content._type !== V1_SERVICE_ELEMENT_PROTOCOL_TYPE;
-    });
-    createServiceRegistrationInputGraphData = (graph) => {
-        const graphData = this.getFullGraphModelData(graph);
-        const prunedGraphData = this.pruneServicesFromGraphForRegistration(graphData);
-        return prunedGraphData;
+    // NOTE: This is somewhat hacky: service registration the input expects only the first service in the graph data
+    // to be the service used for registration.
+    // TODO: the service registration API should be modified to accept service as a parameter outside the model
+    TEMPORARY__prepareGraphDataForServiceRegistration = (graphData, service) => {
+        graphData.elements = graphData.elements.filter((element) => element.path !== service.path);
+        // insert the service as the first element so it gets indexed first
+        graphData.elements.unshift(service);
+        if (graphData.INTERNAL__rawDependencyEntities) {
+            graphData.INTERNAL__rawDependencyEntities =
+                graphData.INTERNAL__rawDependencyEntities.filter((entity) => entity.path !== service.path);
+        }
+        return graphData;
     };
     createBulkServiceRegistrationInput = (graph, services) => {
         const graphData = this.getFullGraphModelData(graph);
         const results = [];
         services.forEach((service) => {
-            const prunedGraphData = this.pruneServicesFromGraphForRegistration(graphData);
-            prunedGraphData.elements.push(this.elementToProtocol(service));
-            results.push({ service: service, context: prunedGraphData });
+            results.push({
+                service: service,
+                context: this.TEMPORARY__prepareGraphDataForServiceRegistration(graphData, this.elementToProtocol(service)),
+            });
         });
         return results;
     };
@@ -1519,7 +1509,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
          * reflect the renaming of function in local changes.
          */
         const TEMPORARY__entityPathIndex = new Map();
-        await V1_entitiesToPureModelContextData(entities, pureModelContextData, this.pluginManager.getPureProtocolProcessorPlugins(), TEMPORARY__entityPathIndex);
+        await V1_entitiesToPureModelContextData(entities, pureModelContextData, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo, TEMPORARY__entityPathIndex);
         await Promise.all(pureModelContextData.elements.map((element) => promisify(() => hashMap.set(TEMPORARY__entityPathIndex.get(element.path)
             ? guaranteeNonNullable(TEMPORARY__entityPathIndex.get(element.path))
             : element.path, element.hashCode))));
@@ -1533,13 +1523,6 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
         }
         return entity;
     };
-    prunePureModelContextData = (data, elementFilter, entityFilter) => {
-        const prunedGraphData = new V1_PureModelContextData();
-        prunedGraphData.elements = data.elements.filter((element) => elementFilter ? elementFilter(element) : true);
-        prunedGraphData.INTERNAL__rawDependencyEntities =
-            data.INTERNAL__rawDependencyEntities.filter((entity) => entityFilter ? entityFilter(entity) : true);
-        return prunedGraphData;
-    };
     buildPureModelSDLCPointer(origin, clientVersion) {
         if (origin instanceof LegendSDLC) {
             return new V1_PureModelContextPointer(clientVersion
@@ -1574,7 +1557,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             }
             return true;
         })
-            .map((entity) => entityProcessorFn ? entityProcessorFn(entity) : entity), data, this.pluginManager.getPureProtocolProcessorPlugins());
+            .map((entity) => entityProcessorFn ? entityProcessorFn(entity) : entity), data, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
         const mainGraphBuilderInput = [
             {
                 model: graph,
@@ -1598,7 +1581,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
                 }
                 return true;
             })
-                .map((entity) => entityProcessorFn ? entityProcessorFn(entity) : entity), projectModelData, this.pluginManager.getPureProtocolProcessorPlugins());
+                .map((entity) => entityProcessorFn ? entityProcessorFn(entity) : entity), projectModelData, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
         }));
         const dependencyGraphBuilderInput = Array.from(dependencyGraphDataIndex.entries()).map(([dependencyKey, dependencyData]) => ({
             data: V1_indexPureModelContextData(report, dependencyData, this.graphBuilderExtensions),
@@ -1628,10 +1611,13 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
             ...contextData1.elements,
             ...contextData2.elements,
         ];
-        contextData1.INTERNAL__rawDependencyEntities = [
-            ...contextData1.INTERNAL__rawDependencyEntities,
-            ...contextData2.INTERNAL__rawDependencyEntities,
+        const rawDependencyEntities = [
+            ...(contextData1.INTERNAL__rawDependencyEntities ?? []),
+            ...(contextData2.INTERNAL__rawDependencyEntities ?? []),
         ];
+        contextData1.INTERNAL__rawDependencyEntities = rawDependencyEntities.length
+            ? rawDependencyEntities
+            : undefined;
         return contextData1;
     }
     elementProtocolToEntity = (elementProtocol) => ({
@@ -1642,7 +1628,7 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
     pureModelContextDataToEntities = (graphProtocol) => graphProtocol.elements.map((element) => this.elementProtocolToEntity(element));
     async entitiesToPureModelContextData(entities) {
         const graphData = new V1_PureModelContextData();
-        await V1_entitiesToPureModelContextData(entities, graphData, this.pluginManager.getPureProtocolProcessorPlugins());
+        await V1_entitiesToPureModelContextData(entities, graphData, this.pluginManager.getPureProtocolProcessorPlugins(), this.subtypeInfo);
         return graphData;
     }
     elementToProtocol = (element, options) => V1_transformPackageableElement(element, this.pluginManager.getPureProtocolProcessorPlugins(), new V1_GraphTransformerContextBuilder(this.pluginManager.getPureProtocolProcessorPlugins())
@@ -1651,7 +1637,8 @@ class V1_PureGraphManager extends AbstractPureGraphManager {
     getElementPath = (elementProtocol) => `${elementProtocol.package}${ENTITY_PATH_DELIMITER}${elementProtocol.name}`;
     getElementClassiferPath = (protocol) => {
         if (protocol instanceof V1_INTERNAL__UnknownPackageableElement ||
-            protocol instanceof V1_INTERNAL__UnknownFunctionActivator) {
+            protocol instanceof V1_INTERNAL__UnknownFunctionActivator ||
+            protocol instanceof V1_INTERNAL__UnknownStore) {
             const _type = protocol.content._type;
             const classifierPath = isString(_type)
                 ? this.elementClassifierPathMap.get(_type)