@finos/legend-query-builder 4.18.0 → 4.18.2

package/lib/stores/workflows/dataProduct/DataProductQueryBuilderState.d.ts

@@ -13,13 +13,14 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { type GraphManagerState, type Class, DataProduct, type V1_DataProductArtifact, ModelAccessPointGroup, NativeModelExecutionContext, type NativeModelAccess, type Mapping, type DataProductElementScope, PackageableRuntime, type LambdaFunction, type QueryExecutionContext, type RawLambda } from '@finos/legend-graph';
+import { type GraphManagerState, type Class, DataProduct, type V1_DataProductArtifact, ModelAccessPointGroup, NativeModelExecutionContext, type NativeModelAccess, type Mapping, type DataProductElementScope, PackageableRuntime, type LambdaFunction, type MappingModelCoverageAnalysisResult, type QueryExecutionContext, LakehouseAccessPoint, type RawLambda, type Accessor, type RelationTypeMetadata, DataProductAccessor, type PureModel } from '@finos/legend-graph';
 import { QueryBuilderState } from '../../QueryBuilderState.js';
 import { type GenericLegendApplicationStore } from '@finos/legend-application';
 import type { QueryBuilderActionConfig, QueryBuilderWorkflowState } from '../../query-workflow/QueryBuilderWorkFlowState.js';
 import type { QueryBuilderConfig } from '../../../graph-manager/QueryBuilderConfig.js';
 import { ActionState, type GeneratorFn } from '@finos/legend-shared';
 import { DepotEntityWithOrigin, type QueryableSourceInfo } from '@finos/legend-storage';
+export declare const buildDataProductAccessor: (relationMetadata: RelationTypeMetadata, dataProduct: DataProduct, accessPoint: LakehouseAccessPoint, graph: PureModel) => DataProductAccessor;
 export type DataProductOption = {
     label: string;
     value: DepotEntityWithOrigin;
@@ -34,13 +35,19 @@ export type ModelAccessPointGroupOption = {
     value: ModelAccessPointGroup;
 };
 export declare const buildModelAccessPointGroupOption: (value: ModelAccessPointGroup) => ModelAccessPointGroupOption;
+export type ExecutionIdOption = {
+    label: string;
+    tag: string;
+    value: NativeModelExecutionContext | ModelAccessPointGroup | LakehouseAccessPoint;
+};
 export declare abstract class DataProductExecutionState<T> {
     readonly queryBuilderState: DataProductQueryBuilderState;
     exectionValue: T;
     constructor(executionState: T, queryBuilderState: DataProductQueryBuilderState);
     abstract get label(): string;
-    abstract get mapping(): Mapping;
+    abstract get mapping(): Mapping | undefined;
     abstract get featuredElements(): DataProductElementScope[] | undefined;
+    get showRuntimeOptions(): boolean;
     get selectedOption(): {
         label: string;
         value: T;
@@ -65,38 +72,60 @@ export declare class ModelAccessPointDataProductExecutionState extends DataProdu
     get showRuntimeOptions(): boolean;
     get compatibleRuntimes(): PackageableRuntime[];
 }
+export declare class LakehouseDataProductExecutionState extends DataProductExecutionState<LakehouseAccessPoint> {
+    selectedRuntime: PackageableRuntime | undefined;
+    adhocRuntime: boolean;
+    constructor(executionState: LakehouseAccessPoint, queryBuilderState: DataProductQueryBuilderState);
+    changeSelectedRuntime(val: PackageableRuntime): void;
+    get label(): string;
+    get mapping(): Mapping | undefined;
+    get featuredElements(): DataProductElementScope[] | undefined;
+    get compatibleRuntimes(): PackageableRuntime[];
+    get showRuntimeOptions(): boolean;
+}
 export declare class DataProductQueryBuilderState extends QueryBuilderState {
     readonly onClassChange?: ((val: Class) => void) | undefined;
     readonly onDataProductChange?: (val: DepotEntityWithOrigin) => Promise<void>;
-    readonly onExecutionContextChange?: ((val: NativeModelExecutionContext) => void) | undefined;
     loadDataProductModelState: ActionState;
     dataProduct: DataProduct;
     dataProductArtifact: V1_DataProductArtifact | undefined;
-    executionState: DataProductExecutionState<NativeModelExecutionContext | ModelAccessPointGroup>;
+    executionState: DataProductExecutionState<NativeModelExecutionContext | ModelAccessPointGroup | LakehouseAccessPoint>;
     entities: DepotEntityWithOrigin[] | undefined;
+    mappingToMappingCoverageResult?: Map<string, MappingModelCoverageAnalysisResult>;
     prioritizeEntityFunc?: ((val: DepotEntityWithOrigin) => boolean) | undefined;
     TEMPORARY__setupPanelContentRenderer: () => React.ReactNode;
-    constructor(applicationStore: GenericLegendApplicationStore, graphManagerState: GraphManagerState, workflow: QueryBuilderWorkflowState, dataProduct: DataProduct, artifact: V1_DataProductArtifact | undefined, actionConfig: QueryBuilderActionConfig, executionState: NativeModelExecutionContext | ModelAccessPointGroup, prioritizeEntityFunc: ((val: DepotEntityWithOrigin) => boolean) | undefined, onDataProductChange: (val: DepotEntityWithOrigin) => Promise<void>, onExecutionContextChange?: ((val: NativeModelExecutionContext) => void) | undefined, onClassChange?: ((val: Class) => void) | undefined, config?: QueryBuilderConfig | undefined, sourceInfo?: QueryableSourceInfo | undefined);
+    constructor(applicationStore: GenericLegendApplicationStore, graphManagerState: GraphManagerState, workflow: QueryBuilderWorkflowState, dataProduct: DataProduct, artifact: V1_DataProductArtifact | undefined, actionConfig: QueryBuilderActionConfig, executionState: NativeModelExecutionContext | ModelAccessPointGroup | LakehouseAccessPoint, prioritizeEntityFunc: ((val: DepotEntityWithOrigin) => boolean) | undefined, onDataProductChange: (val: DepotEntityWithOrigin) => Promise<void>, onClassChange?: ((val: Class) => void) | undefined, config?: QueryBuilderConfig | undefined, sourceInfo?: QueryableSourceInfo | undefined);
+    prepareAccessForExecution(): Promise<void>;
     get isProductLinkable(): boolean;
     get isNativeMode(): boolean;
     get isModelAccessPointGroupMode(): boolean;
+    get isLakehouseMode(): boolean;
     get showExecutionContextOptions(): boolean;
     get showModelAccessPointGroupSelector(): boolean;
+    get hasNativeModelAccess(): boolean;
+    get hasBothAccessModes(): boolean;
+    get showContextSelector(): boolean;
+    get allContextOptions(): ExecutionIdOption[];
+    get selectedContextOption(): ExecutionIdOption | undefined;
+    get executionIdOptions(): ExecutionIdOption[];
+    get selectedExecutionIdOption(): ExecutionIdOption | undefined;
+    get showExecutionIdSelector(): boolean;
+    changeExecutionId(option: ExecutionIdOption): Promise<void>;
     get selectedExecOption(): {
         label: string;
         value: NativeModelExecutionContext;
     } | undefined;
+    get requiresMappingForExecution(): boolean;
     get selectedModelAccessPointGroupOption(): ModelAccessPointGroupOption | undefined;
     get usableClasses(): Class[];
-    changeNativeExecutionContext(val: NativeModelExecutionContext): void;
-    changeModelAccessPointGroupValue(val: ModelAccessPointGroup): void;
     buildQueryForPersistence(): RawLambda;
     getQueryExecutionContext(): QueryExecutionContext;
     handleDataProductChange(val: DepotEntityWithOrigin): void;
-    initWithDataProduct(dataProduct: DataProduct, preResolvedState?: NativeModelExecutionContext | ModelAccessPointGroup): void;
+    initWithDataProduct(dataProduct: DataProduct, accessor: Accessor | undefined, preResolvedState?: NativeModelExecutionContext | ModelAccessPointGroup | LakehouseAccessPoint): void;
     copyDataProductLinkToClipBoard(): void;
     protected getElementType(): typeof DataProduct;
-    setExecutionState(val: NativeModelExecutionContext | ModelAccessPointGroup): void;
+    setExecutionState(val: NativeModelExecutionContext | ModelAccessPointGroup | LakehouseAccessPoint): void;
+    changeExecutionState(val: NativeModelExecutionContext | ModelAccessPointGroup | LakehouseAccessPoint): Promise<void>;
     get modelAccessPointGroups(): ModelAccessPointGroup[];
     get hasModelAccessPointGroups(): boolean;
     get modelAccessPointGroupOptions(): ModelAccessPointGroupOption[];

package/lib/stores/workflows/dataProduct/DataProductQueryBuilderState.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DataProductQueryBuilderState.d.ts","sourceRoot":"","sources":["../../../../src/stores/workflows/dataProduct/DataProductQueryBuilderState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,KAAK,EACV,WAAW,EACX,KAAK,sBAAsB,EAC3B,qBAAqB,EACrB,2BAA2B,EAE3B,KAAK,iBAAiB,EACtB,KAAK,OAAO,EACZ,KAAK,uBAAuB,EAC5B,kBAAkB,EAGlB,KAAK,cAAc,EASnB,KAAK,qBAAqB,EAG1B,KAAK,SAAS,EAEf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAE/D,OAAO,EAAE,KAAK,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAC/E,OAAO,KAAK,EACV,wBAAwB,EACxB,yBAAyB,EAC1B,MAAM,mDAAmD,CAAC;AAC3D,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAC;AAEvF,OAAO,EACL,WAAW,EAIX,KAAK,WAAW,EACjB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EACL,qBAAqB,EACrB,KAAK,mBAAmB,EACzB,MAAM,uBAAuB,CAAC;AAK/B,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,qBAAqB,CAAC;CAC9B,CAAC;AAEF,eAAO,MAAM,sBAAsB,GACjC,OAAO,qBAAqB,KAC3B,iBAGD,CAAC;AAEH,eAAO,MAAM,gBAAgB,GAC3B,KAAK,2BAA2B,KAC/B;IACD,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,2BAA2B,CAAC;CAMpC,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,qBAAqB,CAAC;CAC9B,CAAC;AAEF,eAAO,MAAM,gCAAgC,GAC3C,OAAO,qBAAqB,KAC3B,2BAGD,CAAC;AAEH,8BAAsB,yBAAyB,CAAC,CAAC;IAC/C,QAAQ,CAAC,iBAAiB,EAAE,4BAA4B,CAAC;IACzD,aAAa,EAAE,CAAC,CAAC;gBAGf,cAAc,EAAE,CAAC,EACjB,iBAAiB,EAAE,4BAA4B;IAMjD,QAAQ,KAAK,KAAK,IAAI,MAAM,CAAC;IAE7B,QAAQ,KAAK,OAAO,IAAI,OAAO,CAAC;IAEhC,QAAQ,KAAK,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAAC;IAEvE,IAAI,cAAc,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,CAAC,CAAA;KAAE,CAKhD;CACF;AAED,qBAAa,oCAAqC,SAAQ,yBAAyB,CAAC,2BAA2B,CAAC;IAC9G,iBAAiB,EAAE,iBAAiB,CAAC;gBAGnC,cAAc,EAAE,2BAA2B,EAC3C,iBAAiB,EAAE,4BAA4B;IAMjD,IAAa,KAAK,IAAI,MAAM,CAE3B;IAED,IAAI,OAAO,IAAI,OAAO,CAErB;IAED,IAAI,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAE5D;CACF;AAED,qBAAa,yCAA0C,SAAQ,yBAAyB,CAAC,qBAAqB,CAAC;IAC7G,eAAe,EAAE,kBAAkB,GAAG,SAAS,CAAC;IAChD,YAAY,UAAS;gBAGnB,cAAc,EAAE,qBAAqB,EACrC,iBAAiB,EAAE,4BAA4B;IAcjD,qBAAqB,CAAC,GAAG,EAAE,kBAAkB,GAAG,IAAI;IAKpD,gBAAgB,IAAI,yCAAyC;IAK7D,IAAa,KAAK,IAAI,MAAM,CAE3B;IAED,IAAI,OAAO,IAAI,OAAO,CAErB;IAED,IAAI,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAE5D;IAED,IAAI,kBAAkB,IAAI,OAAO,CAEhC;IAED,IAAI,kBAAkB,IAAI,kBAAkB,EAAE,CAI7C;CACF;AAED,qBAAa,4BAA6B,SAAQ,iBAAiB;IACjE,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IAC5D,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7E,QAAQ,CAAC,wBAAwB,CAAC,EAC9B,CAAC,CAAC,GAAG,EAAE,2BAA2B,KAAK,IAAI,CAAC,GAC5C,SAAS,CAAC;IAEd,yBAAyB,cAAwB;IACjD,WAAW,EAAE,WAAW,CAAC;IACzB,mBAAmB,EAAE,sBAAsB,GAAG,SAAS,CAAC;IACxD,cAAc,EAAE,yBAAyB,CACvC,2BAA2B,GAAG,qBAAqB,CACpD,CAAC;IACF,QAAQ,EAAE,qBAAqB,EAAE,GAAG,SAAS,CAAC;IAE9C,oBAAoB,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,GAAG,SAAS,CAAC;IAEpE,oCAAoC,QAAO,KAAK,CAAC,SAAS,CACZ;gBAGrD,gBAAgB,EAAE,6BAA6B,EAC/C,iBAAiB,EAAE,iBAAiB,EACpC,QAAQ,EAAE,yBAAyB,EACnC,WAAW,EAAE,WAAW,EACxB,QAAQ,EAAE,sBAAsB,GAAG,SAAS,EAC5C,YAAY,EAAE,wBAAwB,EACtC,cAAc,EAAE,2BAA2B,GAAG,qBAAqB,EACnE,oBAAoB,EAAE,CAAC,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,GAAG,SAAS,EAC3E,mBAAmB,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,IAAI,CAAC,EAClE,wBAAwB,CAAC,EACrB,CAAC,CAAC,GAAG,EAAE,2BAA2B,KAAK,IAAI,CAAC,GAC5C,SAAS,EACb,aAAa,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,EAClD,MAAM,CAAC,EAAE,kBAAkB,GAAG,SAAS,EACvC,UAAU,CAAC,EAAE,mBAAmB,GAAG,SAAS;IA0C9C,IAAI,iBAAiB,IAAI,OAAO,CAE/B;IAED,IAAI,YAAY,IAAI,OAAO,CAE1B;IAED,IAAI,2BAA2B,IAAI,OAAO,CAIzC;IAED,IAAI,2BAA2B,IAAI,OAAO,CAEzC;IAED,IAAI,iCAAiC,IAAI,OAAO,CAK/C;IAED,IAAI,kBAAkB,IAClB;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,2BAA2B,CAAA;KAAE,GACrD,SAAS,CAIZ;IAED,IAAI,mCAAmC,IACnC,2BAA2B,GAC3B,SAAS,CAKZ;IAED,IAAI,aAAa,IAAI,KAAK,EAAE,CAU3B;IAED,4BAA4B,CAAC,GAAG,EAAE,2BAA2B,GAAG,IAAI;IAUpE,gCAAgC,CAAC,GAAG,EAAE,qBAAqB,GAAG,IAAI;IAazD,wBAAwB,IAAI,SAAS;IAgBrC,wBAAwB,IAAI,qBAAqB;IAiB1D,uBAAuB,CAAC,GAAG,EAAE,qBAAqB,GAAG,IAAI;IAkCzD,mBAAmB,CACjB,WAAW,EAAE,WAAW,EACxB,gBAAgB,CAAC,EAAE,2BAA2B,GAAG,qBAAqB,GACrE,IAAI;IA6CP,8BAA8B,IAAI,IAAI;IAQtC,SAAS,CAAC,cAAc,IAAI,OAAO,WAAW;IAI9C,iBAAiB,CACf,GAAG,EAAE,2BAA2B,GAAG,qBAAqB,GACvD,IAAI;IAOP,IAAI,sBAAsB,IAAI,qBAAqB,EAAE,CAIpD;IAED,IAAI,yBAAyB,IAAI,OAAO,CAEvC;IAED,IAAI,4BAA4B,IAAI,2BAA2B,EAAE,CAIhE;IAED,IAAI,aAAa,IAAI,OAAO,GAAG,SAAS,CAEvC;IAED,IAAI,sBAAsB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAElE;IAEQ,+BAA+B,CACtC,cAAc,EAAE,cAAc,GAC7B,cAAc;IAsChB,YAAY,IAAI,WAAW,CAAC,IAAI,CAAC;IAQlC,SAAS,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,GAAG,qBAAqB;IASvE,IAAI,kBAAkB,IAAI,iBAAiB,EAAE,CAe5C;IAED,IAAI,yBAAyB,IAAI,iBAAiB,CAUjD;IAED,IAAI,WAAW,IAAI,OAAO,CAQzB;IAGD,IAAI,WAAW,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,2BAA2B,CAAA;KAAE,EAAE,CAMzE;IAEc,+BAA+B,CAC5C,sBAAsB,CAAC,EAAE,OAAO,GAAG,SAAS,GAC3C,OAAO,CAAC,IAAI,CAAC;CAkBjB"}
+{"version":3,"file":"DataProductQueryBuilderState.d.ts","sourceRoot":"","sources":["../../../../src/stores/workflows/dataProduct/DataProductQueryBuilderState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,KAAK,EACV,WAAW,EACX,KAAK,sBAAsB,EAC3B,qBAAqB,EACrB,2BAA2B,EAE3B,KAAK,iBAAiB,EACtB,KAAK,OAAO,EACZ,KAAK,uBAAuB,EAC5B,kBAAkB,EAGlB,KAAK,cAAc,EAInB,KAAK,kCAAkC,EAMvC,KAAK,qBAAqB,EAI1B,oBAAoB,EACpB,KAAK,SAAS,EAEd,KAAK,QAAQ,EACb,KAAK,oBAAoB,EACzB,mBAAmB,EAMnB,KAAK,SAAS,EAGf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAE/D,OAAO,EAAE,KAAK,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAC/E,OAAO,KAAK,EACV,wBAAwB,EACxB,yBAAyB,EAC1B,MAAM,mDAAmD,CAAC;AAC3D,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAC;AAEvF,OAAO,EACL,WAAW,EAIX,KAAK,WAAW,EACjB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EACL,qBAAqB,EACrB,KAAK,mBAAmB,EAEzB,MAAM,uBAAuB,CAAC;AAK/B,eAAO,MAAM,wBAAwB,GACnC,kBAAkB,oBAAoB,EACtC,aAAa,WAAW,EACxB,aAAa,oBAAoB,EACjC,OAAO,SAAS,KACf,mBAsBF,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,qBAAqB,CAAC;CAC9B,CAAC;AAEF,eAAO,MAAM,sBAAsB,GACjC,OAAO,qBAAqB,KAC3B,iBAGD,CAAC;AAEH,eAAO,MAAM,gBAAgB,GAC3B,KAAK,2BAA2B,KAC/B;IACD,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,2BAA2B,CAAC;CAMpC,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,qBAAqB,CAAC;CAC9B,CAAC;AAEF,eAAO,MAAM,gCAAgC,GAC3C,OAAO,qBAAqB,KAC3B,2BAGD,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EACD,2BAA2B,GAC3B,qBAAqB,GACrB,oBAAoB,CAAC;CAC1B,CAAC;AAEF,8BAAsB,yBAAyB,CAAC,CAAC;IAC/C,QAAQ,CAAC,iBAAiB,EAAE,4BAA4B,CAAC;IACzD,aAAa,EAAE,CAAC,CAAC;gBAGf,cAAc,EAAE,CAAC,EACjB,iBAAiB,EAAE,4BAA4B;IAMjD,QAAQ,KAAK,KAAK,IAAI,MAAM,CAAC;IAE7B,QAAQ,KAAK,OAAO,IAAI,OAAO,GAAG,SAAS,CAAC;IAE5C,QAAQ,KAAK,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAAC;IAEvE,IAAI,kBAAkB,IAAI,OAAO,CAEhC;IAED,IAAI,cAAc,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,CAAC,CAAA;KAAE,CAKhD;CACF;AAED,qBAAa,oCAAqC,SAAQ,yBAAyB,CAAC,2BAA2B,CAAC;IAC9G,iBAAiB,EAAE,iBAAiB,CAAC;gBAGnC,cAAc,EAAE,2BAA2B,EAC3C,iBAAiB,EAAE,4BAA4B;IAMjD,IAAa,KAAK,IAAI,MAAM,CAE3B;IAED,IAAI,OAAO,IAAI,OAAO,CAErB;IAED,IAAI,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAE5D;CACF;AAED,qBAAa,yCAA0C,SAAQ,yBAAyB,CAAC,qBAAqB,CAAC;IAC7G,eAAe,EAAE,kBAAkB,GAAG,SAAS,CAAC;IAChD,YAAY,UAAS;gBAGnB,cAAc,EAAE,qBAAqB,EACrC,iBAAiB,EAAE,4BAA4B;IAcjD,qBAAqB,CAAC,GAAG,EAAE,kBAAkB,GAAG,IAAI;IAKpD,gBAAgB,IAAI,yCAAyC;IAK7D,IAAa,KAAK,IAAI,MAAM,CAE3B;IAED,IAAI,OAAO,IAAI,OAAO,CAErB;IAED,IAAI,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAE5D;IAED,IAAa,kBAAkB,IAAI,OAAO,CAEzC;IAED,IAAI,kBAAkB,IAAI,kBAAkB,EAAE,CAI7C;CACF;AAED,qBAAa,kCAAmC,SAAQ,yBAAyB,CAAC,oBAAoB,CAAC;IACrG,eAAe,EAAE,kBAAkB,GAAG,SAAS,CAAC;IAChD,YAAY,UAAS;gBAEnB,cAAc,EAAE,oBAAoB,EACpC,iBAAiB,EAAE,4BAA4B;IAOjD,qBAAqB,CAAC,GAAG,EAAE,kBAAkB,GAAG,IAAI;IAKpD,IAAa,KAAK,IAAI,MAAM,CAE3B;IAED,IAAI,OAAO,IAAI,OAAO,GAAG,SAAS,CAEjC;IAED,IAAI,gBAAgB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAE5D;IAED,IAAI,kBAAkB,IAAI,kBAAkB,EAAE,CAI7C;IAED,IAAa,kBAAkB,IAAI,OAAO,CAEzC;CACF;AAED,qBAAa,4BAA6B,SAAQ,iBAAiB;IACjE,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IAC5D,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAE7E,yBAAyB,cAAwB;IACjD,WAAW,EAAE,WAAW,CAAC;IACzB,mBAAmB,EAAE,sBAAsB,GAAG,SAAS,CAAC;IACxD,cAAc,EAAE,yBAAyB,CACvC,2BAA2B,GAAG,qBAAqB,GAAG,oBAAoB,CAC3E,CAAC;IACF,QAAQ,EAAE,qBAAqB,EAAE,GAAG,SAAS,CAAC;IAC9C,8BAA8B,CAAC,EAAE,GAAG,CAClC,MAAM,EACN,kCAAkC,CACnC,CAAC;IAEF,oBAAoB,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,GAAG,SAAS,CAAC;IAEpE,oCAAoC,QAAO,KAAK,CAAC,SAAS,CACZ;gBAGrD,gBAAgB,EAAE,6BAA6B,EAC/C,iBAAiB,EAAE,iBAAiB,EACpC,QAAQ,EAAE,yBAAyB,EACnC,WAAW,EAAE,WAAW,EACxB,QAAQ,EAAE,sBAAsB,GAAG,SAAS,EAC5C,YAAY,EAAE,wBAAwB,EACtC,cAAc,EACV,2BAA2B,GAC3B,qBAAqB,GACrB,oBAAoB,EACxB,oBAAoB,EAAE,CAAC,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,GAAG,SAAS,EAC3E,mBAAmB,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,OAAO,CAAC,IAAI,CAAC,EAClE,aAAa,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,EAClD,MAAM,CAAC,EAAE,kBAAkB,GAAG,SAAS,EACvC,UAAU,CAAC,EAAE,mBAAmB,GAAG,SAAS;IAuDxC,yBAAyB,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBhD,IAAI,iBAAiB,IAAI,OAAO,CAE/B;IAED,IAAI,YAAY,IAAI,OAAO,CAE1B;IAED,IAAI,2BAA2B,IAAI,OAAO,CAIzC;IAED,IAAI,eAAe,IAAI,OAAO,CAE7B;IAED,IAAI,2BAA2B,IAAI,OAAO,CAEzC;IAED,IAAI,iCAAiC,IAAI,OAAO,CAK/C;IACD,IAAI,oBAAoB,IAAI,OAAO,CAElC;IAED,IAAI,kBAAkB,IAAI,OAAO,CAEhC;IAID,IAAI,mBAAmB,IAAI,OAAO,CAEjC;IAED,IAAI,iBAAiB,IAAI,iBAAiB,EAAE,CAE3C;IAED,IAAI,qBAAqB,IAAI,iBAAiB,GAAG,SAAS,CAEzD;IAED,IAAI,kBAAkB,IAAI,iBAAiB,EAAE,CA2B5C;IAED,IAAI,yBAAyB,IAAI,iBAAiB,GAAG,SAAS,CAsB7D;IAED,IAAI,uBAAuB,IAAI,OAAO,CAErC;IAEK,iBAAiB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBjE,IAAI,kBAAkB,IAClB;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,2BAA2B,CAAA;KAAE,GACrD,SAAS,CAIZ;IAED,IAAa,2BAA2B,IAAI,OAAO,CAKlD;IAED,IAAI,mCAAmC,IACnC,2BAA2B,GAC3B,SAAS,CAKZ;IAED,IAAI,aAAa,IAAI,KAAK,EAAE,CAU3B;IAEQ,wBAAwB,IAAI,SAAS;IAgBrC,wBAAwB,IAAI,qBAAqB;IAwB1D,uBAAuB,CAAC,GAAG,EAAE,qBAAqB,GAAG,IAAI;IAkCzD,mBAAmB,CACjB,WAAW,EAAE,WAAW,EACxB,QAAQ,EAAE,QAAQ,GAAG,SAAS,EAC9B,gBAAgB,CAAC,EACb,2BAA2B,GAC3B,qBAAqB,GACrB,oBAAoB,GACvB,IAAI;IAsEP,8BAA8B,IAAI,IAAI;IAQtC,SAAS,CAAC,cAAc,IAAI,OAAO,WAAW;IAI9C,iBAAiB,CACf,GAAG,EACC,2BAA2B,GAC3B,qBAAqB,GACrB,oBAAoB,GACvB,IAAI;IASD,oBAAoB,CACxB,GAAG,EACC,2BAA2B,GAC3B,qBAAqB,GACrB,oBAAoB,GACvB,OAAO,CAAC,IAAI,CAAC;IAkBhB,IAAI,sBAAsB,IAAI,qBAAqB,EAAE,CAIpD;IAED,IAAI,yBAAyB,IAAI,OAAO,CAEvC;IAED,IAAI,4BAA4B,IAAI,2BAA2B,EAAE,CAIhE;IAED,IAAI,aAAa,IAAI,OAAO,GAAG,SAAS,CAEvC;IAED,IAAI,sBAAsB,IAAI,uBAAuB,EAAE,GAAG,SAAS,CAElE;IAEQ,+BAA+B,CACtC,cAAc,EAAE,cAAc,GAC7B,cAAc;IAsChB,YAAY,IAAI,WAAW,CAAC,IAAI,CAAC;IAQlC,SAAS,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,GAAG,qBAAqB;IASvE,IAAI,kBAAkB,IAAI,iBAAiB,EAAE,CAe5C;IAED,IAAI,yBAAyB,IAAI,iBAAiB,CAUjD;IAED,IAAI,WAAW,IAAI,OAAO,CAYzB;IAGD,IAAI,WAAW,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,2BAA2B,CAAA;KAAE,EAAE,CAMzE;IAEc,+BAA+B,CAC5C,sBAAsB,CAAC,EAAE,OAAO,GAAG,SAAS,GAC3C,OAAO,CAAC,IAAI,CAAC;CAsFjB"}

package/lib/stores/workflows/dataProduct/DataProductQueryBuilderState.js

@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { DataProduct, ModelAccessPointGroup, NativeModelExecutionContext, CORE_PURE_PATH, PackageableRuntime, resolveUsableDataProductClasses, LakehouseRuntime, SimpleFunctionExpression, InstanceValue, Multiplicity, extractElementNameFromPath, SUPPORTED_FUNCTIONS, PackageableElementExplicitReference, resolveDataProductExecutionState, RuntimePointer, QueryDataProductNativeExecutionContext, QueryDataProductModelAccessExecutionContext, buildRawLambdaFromLambdaFunction, } from '@finos/legend-graph';
+import { DataProduct, ModelAccessPointGroup, NativeModelExecutionContext, CORE_PURE_PATH, PackageableRuntime, resolveUsableDataProductClasses, LakehouseRuntime, SimpleFunctionExpression, InstanceValue, Multiplicity, extractElementNameFromPath, SUPPORTED_FUNCTIONS, PackageableElementExplicitReference, resolveDataProductExecutionState, RuntimePointer, QueryDataProductNativeExecutionContext, QueryDataProductModelAccessExecutionContext, QueryDataProductLakehouseExecutionContext, LakehouseAccessPoint, buildRawLambdaFromLambdaFunction, DataProductAccessor, RelationType, RelationColumn, GenericType, GenericTypeExplicitReference, findLakehouseAccessPointGroup, DataProductAccessType, LegendSDLC, } from '@finos/legend-graph';
 import { QueryBuilderState } from '../../QueryBuilderState.js';
 import {} from '@finos/legend-application';
 import { renderDataProductQueryBuilderSetupPanelContent } from '../../../components/workflows/DataProductQueryBuilder.js';
@@ -23,6 +23,12 @@ import { DepotEntityWithOrigin, } from '@finos/legend-storage';
 import { compareLabelFn } from '@finos/legend-art';
 import { QueryBuilderEmbeddedFromExecutionContextState } from '../../QueryBuilderExecutionContextState.js';
 import { buildLambdaFunction } from '../../QueryBuilderValueSpecificationBuilder.js';
+export const buildDataProductAccessor = (relationMetadata, dataProduct, accessPoint, graph) => {
+    const relationType = new RelationType(accessPoint.title ?? accessPoint.id);
+    relationType.columns = relationMetadata.columns.map((col) => new RelationColumn(col.name, GenericTypeExplicitReference.create(new GenericType(graph.getType(col.type)))));
+    const groupResult = findLakehouseAccessPointGroup(dataProduct, accessPoint.id);
+    return new DataProductAccessor(dataProduct.path, groupResult?.group.id, accessPoint.id, relationType, dataProduct);
+};
 export const buildDataProductOption = (value) => ({
     label: value.name,
     value,
@@ -44,6 +50,9 @@ export class DataProductExecutionState {
         this.queryBuilderState = queryBuilderState;
         this.exectionValue = executionState;
     }
+    get showRuntimeOptions() {
+        return false;
+    }
     get selectedOption() {
         return {
             label: this.label,
@@ -106,18 +115,46 @@ export class ModelAccessPointDataProductExecutionState extends DataProductExecut
         return this.queryBuilderState.graphManagerState.usableRuntimes.filter((runtime) => runtime.runtimeValue instanceof LakehouseRuntime);
     }
 }
+export class LakehouseDataProductExecutionState extends DataProductExecutionState {
+    selectedRuntime;
+    adhocRuntime = false;
+    constructor(executionState, queryBuilderState) {
+        super(executionState, queryBuilderState);
+        makeObservable(this, {});
+        this.selectedRuntime = this.compatibleRuntimes[0];
+    }
+    changeSelectedRuntime(val) {
+        this.selectedRuntime = val;
+        this.queryBuilderState.changeRuntime(val);
+    }
+    get label() {
+        return this.exectionValue.title ?? this.exectionValue.id;
+    }
+    get mapping() {
+        return undefined;
+    }
+    get featuredElements() {
+        return undefined;
+    }
+    get compatibleRuntimes() {
+        return this.queryBuilderState.graphManagerState.usableRuntimes.filter((runtime) => runtime.runtimeValue instanceof LakehouseRuntime);
+    }
+    get showRuntimeOptions() {
+        return this.compatibleRuntimes.length > 1;
+    }
+}
 export class DataProductQueryBuilderState extends QueryBuilderState {
     onClassChange;
     onDataProductChange;
-    onExecutionContextChange;
     loadDataProductModelState = ActionState.create();
     dataProduct;
     dataProductArtifact;
     executionState;
     entities;
+    mappingToMappingCoverageResult;
     prioritizeEntityFunc;
     TEMPORARY__setupPanelContentRenderer = () => renderDataProductQueryBuilderSetupPanelContent(this);
-    constructor(applicationStore, graphManagerState, workflow, dataProduct, artifact, actionConfig, executionState, prioritizeEntityFunc, onDataProductChange, onExecutionContextChange, onClassChange, config, sourceInfo) {
+    constructor(applicationStore, graphManagerState, workflow, dataProduct, artifact, actionConfig, executionState, prioritizeEntityFunc, onDataProductChange, onClassChange, config, sourceInfo) {
         super(applicationStore, graphManagerState, workflow, config, sourceInfo);
         makeObservable(this, {
             dataProduct: observable,
@@ -129,8 +166,17 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
             isProductLinkable: computed,
             isNativeMode: computed,
             isModelAccessPointGroupMode: computed,
+            isLakehouseMode: computed,
             showExecutionContextOptions: computed,
             showModelAccessPointGroupSelector: computed,
+            showExecutionIdSelector: computed,
+            executionIdOptions: computed,
+            selectedExecutionIdOption: computed,
+            hasNativeModelAccess: computed,
+            hasBothAccessModes: computed,
+            showContextSelector: computed,
+            allContextOptions: computed,
+            selectedContextOption: computed,
             selectedExecOption: computed,
             selectedModelAccessPointGroupOption: computed,
             usableClasses: computed,
@@ -151,11 +197,28 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
         this.executionState =
             executionState instanceof NativeModelExecutionContext
                 ? new NativeModelDataProductExecutionState(executionState, this)
-                : new ModelAccessPointDataProductExecutionState(executionState, this);
+                : executionState instanceof LakehouseAccessPoint
+                    ? new LakehouseDataProductExecutionState(executionState, this)
+                    : new ModelAccessPointDataProductExecutionState(executionState, this);
         this.prioritizeEntityFunc = prioritizeEntityFunc;
         this.onDataProductChange = onDataProductChange;
-        this.onExecutionContextChange = onExecutionContextChange;
         this.onClassChange = onClassChange;
+        // force from.
+        this.executionContextState =
+            new QueryBuilderEmbeddedFromExecutionContextState(this);
+    }
+    async prepareAccessForExecution() {
+        if (this.executionState instanceof NativeModelDataProductExecutionState) {
+            const runtime = this.executionState.exectionValue.runtime;
+            if (runtime) {
+                this.changeRuntime(new RuntimePointer(runtime));
+            }
+        }
+        else if (this.executionState instanceof
+            ModelAccessPointDataProductExecutionState &&
+            this.executionState.selectedRuntime) {
+            this.changeRuntime(new RuntimePointer(PackageableElementExplicitReference.create(this.executionState.selectedRuntime)));
+        }
     }
     get isProductLinkable() {
         return false;
@@ -166,6 +229,9 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
     get isModelAccessPointGroupMode() {
         return (this.executionState instanceof ModelAccessPointDataProductExecutionState);
     }
+    get isLakehouseMode() {
+        return this.executionState instanceof LakehouseDataProductExecutionState;
+    }
     get showExecutionContextOptions() {
         return this.isNativeMode && this.execOptions.length > 1;
     }
@@ -173,11 +239,100 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
         return (this.isModelAccessPointGroupMode &&
             this.modelAccessPointGroupOptions.length > 1);
     }
+    get hasNativeModelAccess() {
+        return this.dataProduct.nativeModelAccess !== undefined;
+    }
+    get hasBothAccessModes() {
+        return this.hasModelAccessPointGroups && this.hasNativeModelAccess;
+    }
+    // showContextSelector / allContextOptions / selectedContextOption are aliases
+    // over the upstream ExecutionIdOption API so existing UI code keeps working.
+    get showContextSelector() {
+        return this.showExecutionIdSelector;
+    }
+    get allContextOptions() {
+        return this.executionIdOptions;
+    }
+    get selectedContextOption() {
+        return this.selectedExecutionIdOption;
+    }
+    get executionIdOptions() {
+        const nativeOptions = (this.dataProduct.nativeModelAccess?.nativeModelExecutionContexts ?? []).map((ctx) => ({
+            label: ctx.key,
+            tag: 'NATIVE',
+            value: ctx,
+        }));
+        const modelOptions = this.modelAccessPointGroups.map((group) => ({
+            label: group.title ?? group.id,
+            tag: 'MODEL',
+            value: group,
+        }));
+        const lakehouseOptions = this.dataProduct.accessPointGroups
+            .flatMap((group) => group.accessPoints)
+            .filter(filterByType(LakehouseAccessPoint))
+            .map((ap) => ({
+            label: ap.title ?? ap.id,
+            tag: 'LAKEHOUSE',
+            value: ap,
+        }));
+        return [...modelOptions, ...lakehouseOptions, ...nativeOptions].sort(compareLabelFn);
+    }
+    get selectedExecutionIdOption() {
+        const state = this.executionState;
+        if (state instanceof NativeModelDataProductExecutionState) {
+            return {
+                label: state.exectionValue.key,
+                tag: 'NATIVE',
+                value: state.exectionValue,
+            };
+        }
+        else if (state instanceof ModelAccessPointDataProductExecutionState) {
+            return {
+                label: state.exectionValue.title ?? state.exectionValue.id,
+                tag: 'MODEL',
+                value: state.exectionValue,
+            };
+        }
+        else if (state instanceof LakehouseDataProductExecutionState) {
+            return {
+                label: state.exectionValue.title ?? state.exectionValue.id,
+                tag: 'LAKEHOUSE',
+                value: state.exectionValue,
+            };
+        }
+        return undefined;
+    }
+    get showExecutionIdSelector() {
+        return this.executionIdOptions.length > 1;
+    }
+    async changeExecutionId(option) {
+        const val = option.value;
+        if (val === this.executionState.exectionValue) {
+            return;
+        }
+        const switchingToModel = val instanceof ModelAccessPointGroup;
+        const switchingToNative = val instanceof NativeModelExecutionContext;
+        const wasModeSwitch = (switchingToModel && this.isNativeMode) ||
+            (switchingToNative && this.isModelAccessPointGroupMode);
+        if (wasModeSwitch) {
+            this.changeHistoryState.querySnapshotBuffer = [];
+            this.changeHistoryState.pointer = -1;
+            this.changeHistoryState.setCurrentQuery(undefined);
+        }
+        await this.changeExecutionState(val);
+        await this.propagateExecutionContextChange();
+    }
     get selectedExecOption() {
         return this.executionState instanceof NativeModelDataProductExecutionState
             ? buildExecOptions(this.executionState.exectionValue)
             : undefined;
     }
+    get requiresMappingForExecution() {
+        if (this.executionState instanceof LakehouseDataProductExecutionState) {
+            return false;
+        }
+        return true;
+    }
     get selectedModelAccessPointGroupOption() {
         return this.executionState instanceof
             ModelAccessPointDataProductExecutionState
@@ -190,23 +345,6 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
             ? resolveUsableDataProductClasses(this.activeFeaturedElements, activeMapping, this.graphManagerState, this.explorerState.mappingModelCoverageAnalysisResult)
             : [];
     }
-    changeNativeExecutionContext(val) {
-        if (this.isNativeMode && val === this.executionState.exectionValue) {
-            return;
-        }
-        this.setExecutionState(val);
-        this.propagateExecutionContextChange()
-            .then(() => this.onExecutionContextChange?.(val))
-            .catch(this.applicationStore.alertUnhandledError);
-    }
-    changeModelAccessPointGroupValue(val) {
-        if (this.isModelAccessPointGroupMode &&
-            val === this.executionState.exectionValue) {
-            return;
-        }
-        this.setExecutionState(val);
-        this.propagateExecutionContextChange().catch(this.applicationStore.alertUnhandledError);
-    }
     buildQueryForPersistence() {
         if (!this.isQuerySupported) {
             return this.buildQuery();
@@ -232,6 +370,12 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
             execContext.accessPointGroupId = this.executionState.exectionValue.id;
             return execContext;
         }
+        else if (this.executionState instanceof LakehouseDataProductExecutionState) {
+            const execContext = new QueryDataProductLakehouseExecutionContext();
+            execContext.dataProductPath = this.dataProduct.path;
+            execContext.accessPointId = this.executionState.exectionValue.id;
+            return execContext;
+        }
         return super.getQueryExecutionContext();
     }
     handleDataProductChange(val) {
@@ -240,7 +384,7 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
             const dataProduct = this.graphManagerState.graph.getOwnNullableDataProduct(val.path) ??
                 this.graphManagerState.graph.generationModel.getOwnNullableDataProduct(val.path);
             if (dataProduct) {
-                this.initWithDataProduct(dataProduct);
+                this.initWithDataProduct(dataProduct, undefined, undefined);
                 this.loadDataProductModelState.pass();
             }
             else if (this.onDataProductChange) {
@@ -261,15 +405,15 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
             this.loadDataProductModelState.fail();
         }
     }
-    initWithDataProduct(dataProduct, preResolvedState) {
+    initWithDataProduct(dataProduct, accessor, preResolvedState) {
         try {
             const execValue = preResolvedState ?? resolveDataProductExecutionState(dataProduct);
             this.dataProduct = dataProduct;
-            this.executionState =
-                execValue instanceof NativeModelExecutionContext
-                    ? new NativeModelDataProductExecutionState(execValue, this)
-                    : new ModelAccessPointDataProductExecutionState(execValue, this);
-            this.changeMapping(this.executionState.mapping);
+            this.setExecutionState(execValue);
+            const mapping = this.executionState.mapping;
+            if (mapping) {
+                this.changeMapping(mapping);
+            }
             if (this.executionState instanceof NativeModelDataProductExecutionState) {
                 const runtime = guaranteeNonNullable(this.executionState.exectionValue.runtime, 'runtime unable to be resolved');
                 this.changeRuntime(new RuntimePointer(runtime));
@@ -279,13 +423,27 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
                 this.executionState.selectedRuntime instanceof PackageableRuntime) {
                 this.changeRuntime(this.executionState.selectedRuntime);
             }
-            const compatibleClasses = resolveUsableDataProductClasses(this.activeFeaturedElements, this.executionState.mapping, this.graphManagerState, undefined);
-            // if there is no chosen class or the chosen one is not compatible
-            // with the mapping then pick a compatible class if possible
-            if (!this.sourceClass || !compatibleClasses.includes(this.sourceClass)) {
-                const possibleNewClass = compatibleClasses[0];
-                if (possibleNewClass) {
-                    this.changeSourceElement(possibleNewClass);
+            else if (this.executionState instanceof LakehouseDataProductExecutionState &&
+                accessor &&
+                this.executionState.selectedRuntime instanceof PackageableRuntime) {
+                this.setSourceElement(accessor);
+                this.changeRuntime(new RuntimePointer(PackageableElementExplicitReference.create(this.executionState.selectedRuntime)));
+            }
+            if (mapping) {
+                const coverageResult = this.mappingToMappingCoverageResult?.get(mapping.path);
+                if (coverageResult) {
+                    this.explorerState.mappingModelCoverageAnalysisResult =
+                        coverageResult;
+                }
+                const compatibleClasses = resolveUsableDataProductClasses(this.activeFeaturedElements, mapping, this.graphManagerState, coverageResult);
+                // if there is no chosen class or the chosen one is not compatible
+                // with the mapping then pick a compatible class if possible
+                if (!this.sourceClass ||
+                    !compatibleClasses.includes(this.sourceClass)) {
+                    const possibleNewClass = compatibleClasses[0];
+                    if (possibleNewClass) {
+                        this.changeSourceElement(possibleNewClass);
+                    }
                 }
             }
         }
@@ -306,7 +464,17 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
         this.executionState =
             val instanceof NativeModelExecutionContext
                 ? new NativeModelDataProductExecutionState(val, this)
-                : new ModelAccessPointDataProductExecutionState(val, this);
+                : val instanceof LakehouseAccessPoint
+                    ? new LakehouseDataProductExecutionState(val, this)
+                    : new ModelAccessPointDataProductExecutionState(val, this);
+    }
+    async changeExecutionState(val) {
+        this.setExecutionState(val);
+        if (val instanceof LakehouseAccessPoint) {
+            const relationMetadata = await this.graphManagerState.graphManager.getLambdaRelationType(val.func, this.graphManagerState.graph);
+            const accessor = buildDataProductAccessor(relationMetadata, this.dataProduct, val, this.graphManagerState.graph);
+            this.setSourceElement(accessor);
+        }
     }
     get modelAccessPointGroups() {
         return this.dataProduct.accessPointGroups.filter(filterByType(ModelAccessPointGroup));
@@ -380,7 +548,9 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
     get isSupported() {
         return (this.dataProduct.nativeModelAccess !== undefined ||
             // contains model access point group
-            this.dataProduct.accessPointGroups.filter(filterByType(ModelAccessPointGroup)).length > 0);
+            this.dataProduct.accessPointGroups.filter(filterByType(ModelAccessPointGroup)).length > 0 ||
+            // contains lakehouse access point
+            this.dataProduct.accessPointGroups.some((group) => group.accessPoints.some((ap) => ap instanceof LakehouseAccessPoint)));
     }
     // includes model access point group if more than one group
     get execOptions() {
@@ -389,15 +559,63 @@ export class DataProductQueryBuilderState extends QueryBuilderState {
     async propagateExecutionContextChange(requireReBuildingGraph) {
         const currentMapping = this.executionContextState.mapping;
         const newMapping = this.activeMapping;
-        if (newMapping && newMapping !== currentMapping) {
-            this.changeMapping(newMapping, {
-                keepQueryContent: true,
-            });
-            const classes = resolveUsableDataProductClasses(this.activeFeaturedElements, newMapping, this.graphManagerState, undefined);
-            if (this.sourceClass && !classes.includes(this.sourceClass)) {
-                this.setSourceElement(classes[0]);
+        if (!newMapping || newMapping === currentMapping) {
+            return;
+        }
+        const coverageResult = this.mappingToMappingCoverageResult?.get(newMapping.path);
+        if (coverageResult && this.dataProductArtifact) {
+            const origin = this.graphManagerState.graph.origin;
+            if (origin instanceof LegendSDLC) {
+                const newGraph = this.graphManagerState.createNewGraph();
+                const projectInfo = {
+                    groupId: origin.groupId,
+                    artifactId: origin.artifactId,
+                    versionId: origin.versionId,
+                };
+                let accessPointId;
+                let dataProductAccessType;
+                if (this.executionState instanceof NativeModelDataProductExecutionState) {
+                    accessPointId = this.executionState.exectionValue.key;
+                    dataProductAccessType = DataProductAccessType.NATIVE;
+                }
+                else {
+                    accessPointId = this.executionState.exectionValue.id;
+                    dataProductAccessType = DataProductAccessType.MODEL;
+                }
+                const analysisResult = await this.graphManagerState.graphManager.buildDataProductAnalysis(this.dataProductArtifact, this.dataProduct.path, newGraph, accessPointId, dataProductAccessType, projectInfo);
+                this.graphManagerState.graph = newGraph;
+                this.dataProduct = newGraph.getDataProduct(this.dataProduct.path);
+                this.setExecutionState(analysisResult.targetExecState);
+                if (analysisResult.dataProductAnalysis.mappingToMappingCoverageResult) {
+                    this.mappingToMappingCoverageResult =
+                        analysisResult.dataProductAnalysis.mappingToMappingCoverageResult;
+                }
+                const newCoverageResult = this.mappingToMappingCoverageResult?.get(newMapping.path);
+                if (newCoverageResult) {
+                    this.explorerState.mappingModelCoverageAnalysisResult =
+                        newCoverageResult;
+                }
+            }
+            else {
+                this.explorerState.mappingModelCoverageAnalysisResult = coverageResult;
+            }
+        }
+        else if (coverageResult) {
+            this.explorerState.mappingModelCoverageAnalysisResult = coverageResult;
+        }
+        await this.prepareAccessForExecution();
+        this.changeMapping(newMapping, {
+            keepQueryContent: true,
+        });
+        const classes = resolveUsableDataProductClasses(this.activeFeaturedElements, newMapping, this.graphManagerState, this.explorerState.mappingModelCoverageAnalysisResult);
+        if (!this.sourceClass ||
+            (!classes.includes(this.sourceClass) && classes.length)) {
+            const possibleNewClass = classes[0];
+            if (possibleNewClass) {
+                this.changeSourceElement(possibleNewClass);
             }
         }
+        this.explorerState.refreshTreeData();
     }
 }
 //# sourceMappingURL=DataProductQueryBuilderState.js.map