@finos/legend-query-builder 4.15.3 → 4.15.5

package/src/components/explorer/QueryBuilderFunctionsExplorerPanel.tsx

@@ -44,16 +44,14 @@ import {
 import {
   type QueryBuilderFunctionsExplorerTreeNodeData,
   type QueryBuilderFunctionsExplorerDragSource,
-  generateFunctionsExplorerTreeNodeData,
   getFunctionsExplorerTreeNodeChildren,
   QUERY_BUILDER_FUNCTION_DND_TYPE,
+  generateFunctionsExplorerTreeNodeDataFromFunctionAnalysisInfo,
 } from '../../stores/explorer/QueryFunctionsExplorerState.js';
 import { useDrag } from 'react-dnd';
 import {
-  ConcreteFunctionDefinition,
-  generateFunctionPrettyName,
+  type FunctionAnalysisInfo,
   getElementRootPackage,
-  Package,
   ROOT_PACKAGE_NAME,
   getMultiplicityDescription,
   CORE_PURE_PATH,
@@ -61,20 +59,22 @@ import {
 } from '@finos/legend-graph';
 import type { QueryBuilderState } from '../../stores/QueryBuilderState.js';
 import { QueryBuilderTelemetryHelper } from '../../__lib__/QueryBuilderTelemetryHelper.js';
+import { QUERY_BUILDER_TEST_ID } from '../../__lib__/QueryBuilderTesting.js';
 
 const isDependencyTreeNode = (
   node: QueryBuilderFunctionsExplorerTreeNodeData,
 ): boolean =>
-  getElementRootPackage(node.packageableElement).name ===
-  ROOT_PACKAGE_NAME.PROJECT_DEPENDENCY_ROOT;
+  node.package !== undefined &&
+  getElementRootPackage(node.package).name ===
+    ROOT_PACKAGE_NAME.PROJECT_DEPENDENCY_ROOT;
 
 const QueryBuilderFunctionInfoTooltip: React.FC<{
-  element: ConcreteFunctionDefinition;
   children: React.ReactElement;
   placement?: TooltipPlacement | undefined;
+  functionInfo: FunctionAnalysisInfo;
 }> = (props) => {
-  const { element, children, placement } = props;
-  const documentation = element.taggedValues
+  const { children, placement, functionInfo } = props;
+  const documentation = functionInfo.taggedValues
     .filter(
       (taggedValue) =>
         taggedValue.tag.ownerReference.value.path ===
@@ -83,6 +83,23 @@ const QueryBuilderFunctionInfoTooltip: React.FC<{
     )
     .map((taggedValue) => taggedValue.value);
 
+  const generateParameterCallString = (): string => {
+    if (
+      functionInfo.parameterInfoList.length &&
+      functionInfo.parameterInfoList.length > 0
+    ) {
+      return functionInfo.parameterInfoList
+        .map(
+          (param) =>
+            `${param.name}: ${
+              param.type
+            }${getMultiplicityDescription(param.multiplicity)}`,
+        )
+        .join('; ');
+    }
+    return '';
+  };
+
   return (
     <Tooltip
       arrow={true}
@@ -102,7 +119,7 @@ const QueryBuilderFunctionInfoTooltip: React.FC<{
               Function Name
             </div>
             <div className="query-builder__tooltip__item__value">
-              {element.path}
+              {functionInfo.functionPath}
             </div>
           </div>
           <div className="query-builder__tooltip__item">
@@ -110,14 +127,7 @@ const QueryBuilderFunctionInfoTooltip: React.FC<{
               Parameters
             </div>
             <div className="query-builder__tooltip__item__value">
-              {element.parameters
-                .map(
-                  (param) =>
-                    `${param.name}: ${
-                      param.type.value.name
-                    }${getMultiplicityDescription(param.multiplicity)}`,
-                )
-                .join('; ')}
+              {generateParameterCallString()}
             </div>
           </div>
           {Boolean(documentation.length) && (
@@ -135,7 +145,7 @@ const QueryBuilderFunctionInfoTooltip: React.FC<{
               Return Type
             </div>
             <div className="query-builder__tooltip__item__value">
-              {element.returnType.value.name}
+              {functionInfo.returnType}
             </div>
           </div>
         </div>
@@ -147,20 +157,10 @@ const QueryBuilderFunctionInfoTooltip: React.FC<{
 };
 
 const QueryBuilderFunctionsExplorerListEntry = observer(
-  (props: {
-    queryBuilderState: QueryBuilderState;
-    element: ConcreteFunctionDefinition;
-    rootPackageName: ROOT_PACKAGE_NAME;
-  }) => {
-    const { queryBuilderState, element, rootPackageName } = props;
-    const node = generateFunctionsExplorerTreeNodeData(
-      queryBuilderState,
-      element,
-      rootPackageName,
-    );
-    const functionPrettyName = generateFunctionPrettyName(element, {
-      fullPath: true,
-    });
+  (props: { element: FunctionAnalysisInfo }) => {
+    const { element } = props;
+    const node =
+      generateFunctionsExplorerTreeNodeDataFromFunctionAnalysisInfo(element);
     const [, dragConnector, dragPreviewConnector] = useDrag(
       () => ({
         type: QUERY_BUILDER_FUNCTION_DND_TYPE,
@@ -183,15 +183,13 @@ const QueryBuilderFunctionsExplorerListEntry = observer(
           </div>
           <div
             className="query-builder__functions-explorer__function__label"
-            title={functionPrettyName}
+            title={element.functionPrettyName}
           >
-            {functionPrettyName}
+            {element.functionPrettyName}
           </div>
         </div>
         <div className="query-builder__functions-explorer__function__actions">
-          <QueryBuilderFunctionInfoTooltip
-            element={node.packageableElement as ConcreteFunctionDefinition}
-          >
+          <QueryBuilderFunctionInfoTooltip functionInfo={element}>
             <div className="query-builder__functions-explorer__function__action query-builder__functions-explorer__function__node__info">
               <InfoCircleIcon />
             </div>
@@ -212,11 +210,11 @@ const QueryBuilderFunctionsExplorerTreeNodeContainer = observer(
     >,
   ) => {
     const { node, level, stepPaddingInRem, onNodeSelect } = props;
-    const isPackage = node.packageableElement instanceof Package;
-    const name =
-      node.packageableElement instanceof ConcreteFunctionDefinition
-        ? generateFunctionPrettyName(node.packageableElement)
-        : node.packageableElement.name;
+    const name = node.functionAnalysisInfo
+      ? node.functionAnalysisInfo.functionPrettyName.split(
+          `${node.functionAnalysisInfo.packagePath}::`,
+        )[1]
+      : (node.package?.name ?? '');
     const isExpandable = Boolean(node.childrenIds.length);
     const nodeExpandIcon = isExpandable ? (
       node.isOpen ? (
@@ -231,7 +229,7 @@ const QueryBuilderFunctionsExplorerTreeNodeContainer = observer(
     const iconPackageColor = isDependencyTreeNode(node)
       ? 'color--dependency'
       : '';
-    const nodeTypeIcon = isPackage ? (
+    const nodeTypeIcon = node.package ? (
       node.isOpen ? (
         <FolderOpenIcon className={iconPackageColor} />
       ) : (
@@ -248,10 +246,7 @@ const QueryBuilderFunctionsExplorerTreeNodeContainer = observer(
     }>(
       () => ({
         type: QUERY_BUILDER_FUNCTION_DND_TYPE,
-        item: () =>
-          node.packageableElement instanceof ConcreteFunctionDefinition
-            ? { node }
-            : {},
+        item: () => (node.functionAnalysisInfo ? { node } : {}),
       }),
       [node],
     );
@@ -289,10 +284,9 @@ const QueryBuilderFunctionsExplorerTreeNodeContainer = observer(
                 {name}
               </div>
               <div className="query-builder__functions-explorer__tree__node__actions">
-                {node.packageableElement instanceof
-                  ConcreteFunctionDefinition && (
+                {node.functionAnalysisInfo && (
                   <QueryBuilderFunctionInfoTooltip
-                    element={node.packageableElement}
+                    functionInfo={node.functionAnalysisInfo}
                   >
                     <div className="query-builder__functions-explorer__tree__node__action query-builder__functions-explorer__tree__node__info">
                       <InfoCircleIcon />
@@ -331,6 +325,7 @@ const QueryBuilderPackableElementExplorerTree = observer(
     const dependencyTreeData = queryFunctionsState.getTreeData(
       ROOT_PACKAGE_NAME.PROJECT_DEPENDENCY_ROOT,
     );
+
     const onDependencyTreeSelect = (
       node: QueryBuilderFunctionsExplorerTreeNodeData,
     ): void => {
@@ -431,6 +426,10 @@ export const QueryBuilderFunctionsExplorerPanel = observer(
       setShowDependencyFuncions(!showDependencyFuncions);
     };
 
+    useEffect(() => {
+      queryFunctionsState.initializeTreeData();
+    }, [queryFunctionsState]);
+
     useEffect(() => {
       QueryBuilderTelemetryHelper.logEvent_RenderPanelFunctionExplorer(
         queryBuilderState.applicationStore.telemetryService,
@@ -438,7 +437,10 @@ export const QueryBuilderFunctionsExplorerPanel = observer(
     }, [queryBuilderState.applicationStore]);
 
     return (
-      <div className="panel query-builder__functions-explorer">
+      <div
+        data-testid={QUERY_BUILDER_TEST_ID.QUERY_BUILDER_FUNCTIONS}
+        className="panel query-builder__functions-explorer"
+      >
         <div className="panel__header">
           <div className="panel__header__title">
             <div className="panel__header__title__label">functions</div>
@@ -494,16 +496,13 @@ export const QueryBuilderFunctionsExplorerPanel = observer(
             labelGetter={(
               item: QueryBuilderFunctionsExplorerDragSource,
             ): string =>
-              generateFunctionPrettyName(
-                item.node.packageableElement as ConcreteFunctionDefinition,
-                { fullPath: true },
-              )
+              item.node.functionAnalysisInfo?.functionPrettyName ?? ''
             }
             types={[QUERY_BUILDER_FUNCTION_DND_TYPE]}
           />
           {((showDependencyFuncions &&
-            (!queryFunctionsState.dependencyTreeData ||
-              queryFunctionsState.dependencyTreeData.rootIds.length === 0)) ||
+            queryFunctionsState.dependencyFunctionExplorerStates.length ===
+              0) ||
             !showDependencyFuncions) &&
             queryFunctionsState.treeData?.rootIds.length === 0 && (
               <BlankPanelContent>
@@ -521,8 +520,8 @@ export const QueryBuilderFunctionsExplorerPanel = observer(
                 />
               )}
               {showDependencyFuncions &&
-                queryFunctionsState.dependencyTreeData &&
-                queryFunctionsState.dependencyTreeData.rootIds.length !== 0 && (
+                queryFunctionsState.dependencyFunctionExplorerStates.length !==
+                  0 && (
                   <QueryBuilderPackableElementExplorerTree
                     queryBuilderState={queryBuilderState}
                     rootPackageName={ROOT_PACKAGE_NAME.PROJECT_DEPENDENCY_ROOT}
@@ -538,15 +537,13 @@ export const QueryBuilderFunctionsExplorerPanel = observer(
                 (funcExplorerState) => (
                   <QueryBuilderFunctionsExplorerListEntry
                     key={funcExplorerState.uuid}
-                    queryBuilderState={queryBuilderState}
-                    element={funcExplorerState.concreteFunctionDefinition}
-                    rootPackageName={ROOT_PACKAGE_NAME.MAIN}
+                    element={funcExplorerState.functionAnalysisInfo}
                   />
                 ),
               )}
               {showDependencyFuncions &&
-                queryFunctionsState.dependencyTreeData &&
-                queryFunctionsState.dependencyTreeData.rootIds.length !== 0 && (
+                queryFunctionsState.dependencyFunctionExplorerStates.length >
+                  0 && (
                   <>
                     {queryFunctionsState.treeData?.rootIds.length !== 0 && (
                       <div className="query-builder__functions-explorer__separator" />
@@ -555,11 +552,7 @@ export const QueryBuilderFunctionsExplorerPanel = observer(
                       (funcExplorerState) => (
                         <QueryBuilderFunctionsExplorerListEntry
                           key={funcExplorerState.uuid}
-                          queryBuilderState={queryBuilderState}
-                          element={funcExplorerState.concreteFunctionDefinition}
-                          rootPackageName={
-                            ROOT_PACKAGE_NAME.PROJECT_DEPENDENCY_ROOT
-                          }
+                          element={funcExplorerState.functionAnalysisInfo}
                         />
                       ),
                     )}

package/src/components/fetch-structure/QueryBuilderTDSPanel.tsx

@@ -71,8 +71,7 @@ import { QUERY_BUILDER_TEST_ID } from '../../__lib__/QueryBuilderTesting.js';
 import { flowResult } from 'mobx';
 import { useApplicationStore } from '@finos/legend-application';
 import {
-  type ConcreteFunctionDefinition,
-  generateFunctionCallString,
+  type ValueSpecification,
   LAMBDA_PIPE,
   VARIABLE_REFERENCE_TOKEN,
   AbstractPropertyExpression,
@@ -81,12 +80,12 @@ import {
   PropertyExplicitReference,
   VariableExpression,
   Multiplicity,
-  type ValueSpecification,
   PrimitiveType,
   GenericType,
   GenericTypeExplicitReference,
   observe_PrimitiveInstanceValue,
   PrimitiveInstanceValue,
+  generateFunctionCallStringFromFunctionAnalysisInfo,
 } from '@finos/legend-graph';
 import {
   type QueryBuilderFunctionsExplorerDragSource,
@@ -239,13 +238,20 @@ const QueryBuilderDerivationProjectionColumnEditor = observer(
             }`,
           );
         } else if (type === QUERY_BUILDER_FUNCTION_DND_TYPE) {
+          const functionAnalysisInfo = (
+            item as QueryBuilderFunctionsExplorerDragSource
+          ).node.functionAnalysisInfo;
+          const functionPrettyName = functionAnalysisInfo
+            ? generateFunctionCallStringFromFunctionAnalysisInfo(
+                projectionColumnState.tdsState.queryBuilderState
+                  .graphManagerState.graph,
+                functionAnalysisInfo,
+              )
+            : '';
           projectionColumnState.derivationLambdaEditorState.setLambdaString(
             `${
               projectionColumnState.derivationLambdaEditorState.lambdaString
-            }${`${generateFunctionCallString(
-              (item as QueryBuilderFunctionsExplorerDragSource).node
-                .packageableElement as ConcreteFunctionDefinition,
-            )}`}`,
+            }${functionPrettyName}`,
           );
         } else {
           projectionColumnState.derivationLambdaEditorState.setLambdaString(
@@ -1262,11 +1268,17 @@ export const QueryBuilderTDSPanel = observer(
                   { addDummyParameter: true },
                 ),
               );
+            const functionAnalysisInfo = (
+              item as QueryBuilderFunctionsExplorerDragSource
+            ).node.functionAnalysisInfo;
+            const functionPrettyName = functionAnalysisInfo
+              ? generateFunctionCallStringFromFunctionAnalysisInfo(
+                  tdsState.queryBuilderState.graphManagerState.graph,
+                  functionAnalysisInfo,
+                )
+              : '';
             derivationProjectionColumn.derivationLambdaEditorState.setLambdaString(
-              `${DEFAULT_LAMBDA_VARIABLE_NAME}${LAMBDA_PIPE}${generateFunctionCallString(
-                (item as QueryBuilderFunctionsExplorerDragSource).node
-                  .packageableElement as ConcreteFunctionDefinition,
-              )}`,
+              `${DEFAULT_LAMBDA_VARIABLE_NAME}${LAMBDA_PIPE}${functionPrettyName} `,
             );
             tdsState.addColumn(derivationProjectionColumn);
             break;

package/src/index.ts

@@ -34,6 +34,7 @@ export { QueryBuilder } from './components/QueryBuilder.js';
 export { QUERY_BUILDER_COMPONENT_ELEMENT_ID } from './components/QueryBuilderComponentElement.js';
 export {
   type QueryableSourceInfo as QuerySDLC,
+  type QueryBuilderExtraFunctionAnalysisInfo,
   QueryBuilderState,
   QUERY_BUILDER_LAMBDA_WRITER_MODE,
   INTERNAL__BasicQueryBuilderState,
@@ -126,5 +127,4 @@ export * from './stores/filter/QueryBuilderFilterValueSpecificationBuilder.js';
 export * from './stores/QueryBuilderValueSpecificationHelper.js';
 export * from './stores/filter/QueryBuilderFilterState.js';
 export * from './stores/filter/QueryBuilderFilterStateBuilder.js';
-export * from './stores/data-cube/QueryBuilderDataCubeApplicationEngine.js';
 export * from './stores/data-cube/QueryBuilderDataCubeEngine.js';

package/src/stores/QueryBuilderState.ts

@@ -56,6 +56,7 @@ import {
   type Type,
   type QueryGridConfig,
   type QueryExecutionContext,
+  type FunctionAnalysisInfo,
   GRAPH_MANAGER_EVENT,
   CompilationError,
   extractSourceInformationCoordinates,
@@ -117,6 +118,11 @@ export type QueryableClassMappingRuntimeInfo = QueryableSourceInfo & {
   runtime: string;
 };
 
+export type QueryBuilderExtraFunctionAnalysisInfo = {
+  functionInfoMap: Map<string, FunctionAnalysisInfo>;
+  dependencyFunctionInfoMap: Map<string, FunctionAnalysisInfo>;
+};
+
 export enum QUERY_BUILDER_LAMBDA_WRITER_MODE {
   STANDARD = 'STANDARD',
   TYPED_FETCH_STRUCTURE = 'TYPED_FETCH_STRUCTURE',
@@ -832,6 +838,12 @@ export abstract class QueryBuilderState implements CommandRegistrar {
     );
   }
 
+  buildFunctionAnalysisInfo():
+    | QueryBuilderExtraFunctionAnalysisInfo
+    | undefined {
+    return undefined;
+  }
+
   /**
    * This method can be used to simplify the current query builder state
    * to a basic one that can be used for testing or some special operations,

package/src/stores/data-cube/QueryBuilderDataCubeEngine.ts

@@ -33,36 +33,22 @@ import {
 import {
   _elementPtr,
   _functionName,
-  DataCubeGetBaseQueryResult,
   DataCubeEngine,
-  type CompletionItem,
-  DataCubeQuerySource,
-  type DataCubeInfrastructureInfo,
+  DataCubeSource,
   type RelationType,
   DataCubeQuery,
+  type DataCubeInitialInput,
+  type CompletionItem,
+  _function,
+  DataCubeFunction,
 } from '@finos/legend-data-cube';
-import { guaranteeType, LogService } from '@finos/legend-shared';
+import { guaranteeType, isNonNullable, LogService } from '@finos/legend-shared';
 
-export class _DataCubeQuery {
-  name: string;
-  query: string;
-  partialQuery: string;
-  source: DataCubeQuerySource;
-
-  constructor(
-    name: string,
-    query: string,
-    partialQuery: string,
-    source: DataCubeQuerySource,
-  ) {
-    this.name = name;
-    this.query = query;
-    this.partialQuery = partialQuery;
-    this.source = source;
-  }
+class QueryBuilderDataCubeSource extends DataCubeSource {
+  mapping?: string | undefined;
+  runtime!: string;
 }
 
-export class QueryBuilderDataCubeQuerySource extends DataCubeQuerySource {}
 export class QueryBuilderDataCubeEngine extends DataCubeEngine {
   readonly logService = new LogService();
   readonly graphState: GraphManagerState;
@@ -90,12 +76,12 @@ export class QueryBuilderDataCubeEngine extends DataCubeEngine {
   get sourceLabel(): string {
     return `Query Builder Report`;
   }
-  override getBaseQuery(): Promise<DataCubeGetBaseQueryResult> {
+
+  override getInitialInput(): Promise<DataCubeInitialInput> {
     return this.buildBaseQuery();
   }
 
-  async buildBaseQuery(): Promise<DataCubeGetBaseQueryResult> {
-    const timestamp = Date.now();
+  async buildBaseQuery(): Promise<DataCubeInitialInput> {
     let srcFuncExp = V1_deserializeValueSpecification(
       this.graphState.graphManager.serializeRawValueSpecification(
         this.selectInitialQuery,
@@ -121,60 +107,44 @@ export class QueryBuilderDataCubeEngine extends DataCubeEngine {
     if (this.runtimePath) {
       fromFuncExp.parameters.push(_elementPtr(this.runtimePath));
     }
-    const [relationType, queryString, fromQuerystring] = await Promise.all([
-      this.getRelationalType(this.selectInitialQuery),
-      this.graphState.graphManager.valueSpecificationToPureCode(
-        V1_serializeValueSpecification(srcFuncExp, []),
-      ),
-      this.graphState.graphManager.valueSpecificationToPureCode(
-        V1_serializeValueSpecification(fromFuncExp, []),
-      ),
-    ]);
-    const columns = relationType.columns;
-    const source = new QueryBuilderDataCubeQuerySource();
-    source.columns = columns;
+    const columns = (await this.getRelationalType(this.selectInitialQuery))
+      .columns;
+    const query = new DataCubeQuery();
+    query.query = `~[${columns.map((e) => `'${e.name}'`)}]->select()`;
+    const source = new QueryBuilderDataCubeSource();
+    source.sourceColumns = columns;
     source.mapping = this.mappingPath;
     source.runtime = this.runtimePath;
-    source.query = queryString;
-    const partialQuery = `~[${columns.map((e) => `'${e.name}'`)}]->select()`;
-    const result = new DataCubeQuery(this.sourceLabel, fromQuerystring);
-    result.partialQuery = partialQuery;
-    result.source = source;
-    const baseQueryResult = new DataCubeGetBaseQueryResult();
-    baseQueryResult.timestamp = timestamp;
-    baseQueryResult.query = result;
-    baseQueryResult.partialQuery = await this.parseQuery(partialQuery);
-    baseQueryResult.sourceQuery = srcFuncExp;
-    return baseQueryResult;
+    source.query = srcFuncExp;
+    return {
+      query,
+      source,
+    };
   }
 
   get graph(): PureModel {
     return this.graphState.graph;
   }
 
-  private buildRawLambdaFromValueSpec(query: V1_ValueSpecification): RawLambda {
+  private buildRawLambdaFromValueSpec(query: V1_Lambda): RawLambda {
     const json = guaranteeType(
       V1_deserializeRawValueSpecification(
-        V1_serializeValueSpecification(query, []),
+        V1_serializeValueSpecification(
+          query.body[0] as V1_ValueSpecification,
+          [],
+        ),
       ),
       V1_RawLambda,
     );
     return new RawLambda(json.parameters, json.body);
   }
 
-  override getInfrastructureInfo(): Promise<DataCubeInfrastructureInfo> {
-    // we return undefined as we assume the grid license is set at the application level where query builder is built
-    return Promise.resolve({
-      gridClientLicense: undefined,
-      simpleSampleDataTableName: '',
-      complexSampleDataTableName: '',
-    });
-  }
-  override async getQueryTypeahead(
+  async getQueryTypeahead(
     code: string,
-    query: V1_ValueSpecification,
-  ): Promise<CompletionItem[]> {
-    const lambda = this.buildRawLambdaFromValueSpec(query);
+    baseQuery: V1_Lambda,
+    source: DataCubeSource,
+  ) {
+    const lambda = this.buildRawLambdaFromValueSpec(baseQuery);
     const queryString =
       await this.graphState.graphManager.lambdaToPureCode(lambda);
     const offset = queryString.length;
@@ -184,13 +154,13 @@ export class QueryBuilderDataCubeEngine extends DataCubeEngine {
       this.graph,
       offset,
     );
-    return result.completions;
+    return result.completions as CompletionItem[];
   }
 
-  override async parseQuery(
+  override async parseValueSpecification(
     code: string,
     returnSourceInformation?: boolean,
-  ): Promise<V1_ValueSpecification> {
+  ) {
     return V1_deserializeValueSpecification(
       await this.graphState.graphManager.pureCodeToValueSpecification(
         code,
@@ -200,36 +170,35 @@ export class QueryBuilderDataCubeEngine extends DataCubeEngine {
     );
   }
 
-  override getQueryCode(
-    query: V1_ValueSpecification,
+  override getValueSpecificationCode(
+    value: V1_ValueSpecification,
     pretty?: boolean | undefined,
-  ): Promise<string> {
+  ) {
     return this.graphState.graphManager.valueSpecificationToPureCode(
-      V1_serializeValueSpecification(query, []),
+      V1_serializeValueSpecification(value, []),
       pretty,
     );
   }
 
-  override getQueryRelationType(
-    query: V1_ValueSpecification,
-  ): Promise<RelationType> {
-    const lambda = this.buildRawLambdaFromValueSpec(query);
-    return this.getRelationalType(lambda);
-  }
-
-  async getRelationalType(query: RawLambda): Promise<RelationType> {
-    const realtion_type =
+  private async getRelationalType(query: RawLambda): Promise<RelationType> {
+    const relationType =
       await this.graphState.graphManager.getLambdaRelationType(
         query,
         this.graph,
       );
-    return realtion_type;
+    return relationType;
+  }
+
+  override getQueryRelationType(query: V1_Lambda, source: DataCubeSource) {
+    const lambda = this.buildRawLambdaFromValueSpec(query);
+    return this.getRelationalType(lambda);
   }
 
   override async getQueryCodeRelationReturnType(
     code: string,
     baseQuery: V1_ValueSpecification,
-  ): Promise<RelationType> {
+    source: DataCubeSource,
+  ) {
     const queryString =
       await this.graphState.graphManager.valueSpecificationToPureCode(
         V1_serializeValueSpecification(baseQuery, []),
@@ -240,11 +209,7 @@ export class QueryBuilderDataCubeEngine extends DataCubeEngine {
     );
   }
 
-  override async executeQuery(query: V1_Lambda): Promise<{
-    result: TDSExecutionResult;
-    executedQuery: string;
-    executedSQL: string;
-  }> {
+  override async executeQuery(query: V1_Lambda, source: DataCubeSource) {
     const lambda = this.buildRawLambdaFromValueSpec(query);
     lambda.parameters = this._parameters;
     const [executionWithMetadata, queryString] = await Promise.all([
@@ -275,4 +240,19 @@ export class QueryBuilderDataCubeEngine extends DataCubeEngine {
       executedSQL: sqlString,
     };
   }
+
+  override buildExecutionContext(
+    source: DataCubeSource,
+  ): V1_AppliedFunction | undefined {
+    if (source instanceof QueryBuilderDataCubeSource) {
+      return _function(
+        DataCubeFunction.FROM,
+        [
+          source.mapping ? _elementPtr(source.mapping) : undefined,
+          _elementPtr(source.runtime),
+        ].filter(isNonNullable),
+      );
+    }
+    return undefined;
+  }
 }