@finos/legend-application-studio 28.21.27 → 28.21.29

package/src/components/editor/editor-group/service-editor/ServiceEditor.tsx

@@ -37,7 +37,13 @@ import {
   PanelContentLists,
   SparkleIcon,
 } from '@finos/legend-art';
-import { debounce, prettyCONSTName } from '@finos/legend-shared';
+import {
+  assertErrorThrown,
+  debounce,
+  HttpStatus,
+  NetworkClientError,
+  prettyCONSTName,
+} from '@finos/legend-shared';
 import { ServiceExecutionEditor } from './ServiceExecutionEditor.js';
 import { LEGEND_STUDIO_TEST_ID } from '../../../../__lib__/LegendStudioTesting.js';
 import { ServiceRegistrationEditor } from './ServiceRegistrationEditor.js';
@@ -70,11 +76,11 @@ import {
 import { LEGEND_STUDIO_APPLICATION_NAVIGATION_CONTEXT_KEY } from '../../../../__lib__/LegendStudioApplicationNavigationContext.js';
 import { ServiceTestableEditor } from './testable/ServiceTestableEditor.js';
 import { flowResult } from 'mobx';
-import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../../../__lib__/LegendStudioDocumentation.js';
-import { DocumentationLink } from '@finos/legend-lego/application';
 import { ServicePostValidationsEditor } from './ServicePostValidationEditor.js';
 import type { DSL_Service_LegendStudioApplicationPlugin_Extension } from '../../../../stores/extensions/DSL_Service_LegendStudioApplicationPlugin_Extension.js';
 import { LegendStudioTelemetryHelper } from '../../../../__lib__/LegendStudioTelemetryHelper.js';
+import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../../../__lib__/LegendStudioDocumentation.js';
+import { DocumentationLink } from '@finos/legend-lego/application';
 
 type UserOption = { label: string; value: string };
 
@@ -278,6 +284,26 @@ const ServiceGeneralEditor = observer(() => {
         ]);
       const suggestion = await aiDocSuggester(serviceGrammar, legendAIUrl);
       setAIDocSuggestion(suggestion);
+    } catch (error) {
+      assertErrorThrown(error);
+      LegendStudioTelemetryHelper.logEvent_ServiceLegendAISuggestFailure(
+        applicationStore.telemetryService,
+        service.path,
+        error.message,
+      );
+      if (
+        error instanceof NetworkClientError &&
+        (error.response.status === HttpStatus.UNAUTHORIZED ||
+          error.response.status === HttpStatus.FORBIDDEN)
+      ) {
+        const docEntry = applicationStore.documentationService.getDocEntry(
+          LEGEND_STUDIO_DOCUMENTATION_KEY.LEGENDAI_HOW_TO_GET_ENTITLEMENTS,
+        );
+        if (docEntry?.url) {
+          error.message = `${error.message}. Please check how to get entitlements: ${docEntry.url}`;
+        }
+      }
+      throw error;
     } finally {
       setIsSuggestingWithAI(false);
     }

package/src/components/editor/editor-group/testable/LakehouseTestableEditor.tsx

@@ -55,7 +55,7 @@ import {
 } from '@finos/legend-art';
 import type { AtomicTest, TestSuite } from '@finos/legend-graph';
 import type { ActionState, GeneratorFn } from '@finos/legend-shared';
-import { useState } from 'react';
+import { useState, type ReactNode } from 'react';
 import type { TestableTestEditorState } from '../../../../stores/editor/editor-state/element-editor-state/testable/TestableEditorState.js';
 import {
   TESTABLE_RESULT,
@@ -341,8 +341,14 @@ export const LakehouseTestDataEditor = observer(
 // ─── Test editor: assertion viewer ────────────────────────────────────────────
 
 export const LakehouseTestEditor = observer(
-  (props: { testState: TestableTestEditorState; isReadOnly: boolean }) => {
-    const { testState } = props;
+  (props: {
+    testState: TestableTestEditorState;
+    isReadOnly: boolean;
+    renderTestStateExtension?: (
+      testState: TestableTestEditorState,
+    ) => ReactNode;
+  }) => {
+    const { testState, renderTestStateExtension } = props;
     const selectedAssertion = testState.selectedAsertionState;
 
     return (
@@ -353,6 +359,7 @@ export const LakehouseTestEditor = observer(
           </div>
         </div>
         <div className="panel">
+          {renderTestStateExtension?.(testState)}
           {selectedAssertion && (
             <TestAssertionEditor testAssertionState={selectedAssertion} />
           )}
@@ -449,8 +456,12 @@ export const LakehouseTestsEditor = observer(
     suiteState: LakehouseSuiteStateForEditor;
     testableState: LakehouseTestableStateForEditor;
     isReadOnly: boolean;
+    renderTestStateExtension?: (
+      testState: TestableTestEditorState,
+    ) => ReactNode;
   }) => {
-    const { suiteState, testableState, isReadOnly } = props;
+    const { suiteState, testableState, isReadOnly, renderTestStateExtension } =
+      props;
     const selectedTest = suiteState.selectTestState;
 
     return (
@@ -513,6 +524,9 @@ export const LakehouseTestsEditor = observer(
                 <LakehouseTestEditor
                   testState={selectedTest}
                   isReadOnly={isReadOnly}
+                  {...(renderTestStateExtension
+                    ? { renderTestStateExtension }
+                    : {})}
                 />
               ) : (
                 <BlankPanelPlaceholder
@@ -535,8 +549,12 @@ export const LakehouseTestSuiteEditor = observer(
     suiteState: LakehouseSuiteStateForEditor;
     testableState: LakehouseTestableStateForEditor;
     isReadOnly: boolean;
+    renderTestStateExtension?: (
+      testState: TestableTestEditorState,
+    ) => ReactNode;
   }) => {
-    const { suiteState, testableState, isReadOnly } = props;
+    const { suiteState, testableState, isReadOnly, renderTestStateExtension } =
+      props;
 
     return (
       <div className="service-test-suite-editor">
@@ -555,6 +573,9 @@ export const LakehouseTestSuiteEditor = observer(
               suiteState={suiteState}
               testableState={testableState}
               isReadOnly={isReadOnly}
+              {...(renderTestStateExtension
+                ? { renderTestStateExtension }
+                : {})}
             />
           </ResizablePanel>
         </ResizablePanelGroup>

package/src/index.ts

@@ -65,8 +65,11 @@ export * from './stores/graph-modifier/RawValueSpecificationGraphModifierHelper.
 export * from './stores/extensions/DSL_Mapping_LegendStudioApplicationPlugin_Extension.js';
 export * from './stores/extensions/DSL_Service_LegendStudioApplicationPlugin_Extension.js';
 export * from './stores/extensions/DSL_DataProduct_LegendStudioApplicationPlugin_Extension.js';
+export * from './stores/extensions/DSL_DataSpace_LegendStudioApplicationPlugin_Extension.js';
 export * from './stores/extensions/DSL_Data_LegendStudioApplicationPlugin_Extension.js';
 export { DataProductEditorState } from './stores/editor/editor-state/element-editor-state/dataProduct/DataProductEditorState.js';
+export { LEGEND_STUDIO_DOCUMENTATION_KEY } from './__lib__/LegendStudioDocumentation.js';
+export { LegendStudioTelemetryHelper } from './__lib__/LegendStudioTelemetryHelper.js';
 
 // components
 export { queryClass } from './components/editor/editor-group/uml-editor/ClassQueryBuilder.js';

package/src/stores/editor/editor-state/element-editor-state/dataProduct/testable/DataProductTestableState.ts

@@ -93,7 +93,7 @@ export type { LakehouseElementTestDataState as DataProductElementTestDataState }
  * Returns the lambda for an access point (handles both LakehouseAccessPoint
  * and FunctionAccessPoint).
  */
-const getAccessPointLambda = (
+export const getAccessPointLambda = (
   accessPoint: AccessPoint,
 ): RawLambda | undefined =>
   accessPoint instanceof LakehouseAccessPoint

package/src/stores/extensions/DSL_DataSpace_LegendStudioApplicationPlugin_Extension.ts

@@ -0,0 +1,54 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import type { DSL_LegendStudioApplicationPlugin_Extension } from '../LegendStudioApplicationPlugin.js';
+
+// ── Response type ─────────────────────────────────────────────────────────────
+
+export type DataSpaceDocResponse = {
+  description: string;
+  confidence: number;
+};
+
+// ── Request type ──────────────────────────────────────────────────────────────
+
+export type DataSpaceDocRequest = {
+  /**
+   * Raw Pure grammar text containing the data space definitions.
+   */
+  definitions: string;
+  /**
+   * Fully-qualified name of the data space (e.g. dataSpace::MY_SPACE).
+   */
+  data_space_name: string;
+  /**
+   * LLM model override (uses server default if omitted).
+   */
+  model?: string;
+};
+
+// ── Plugin extension interface ────────────────────────────────────────────────
+
+export interface DSL_DataSpace_LegendStudioApplicationPlugin_Extension
+  extends DSL_LegendStudioApplicationPlugin_Extension {
+  /**
+   * Generate AI-suggested description for a data space.
+   */
+  getExtraDataSpaceDocumentationAISuggester?(
+    request: DataSpaceDocRequest,
+    legendAIUrl: string,
+  ): Promise<DataSpaceDocResponse>;
+}

package/tsconfig.json

@@ -200,6 +200,7 @@
     "./src/stores/editor/utils/TestableUtils.ts",
     "./src/stores/editor/utils/TreeUtils.ts",
     "./src/stores/extensions/DSL_DataProduct_LegendStudioApplicationPlugin_Extension.ts",
+    "./src/stores/extensions/DSL_DataSpace_LegendStudioApplicationPlugin_Extension.ts",
     "./src/stores/extensions/DSL_Data_LegendStudioApplicationPlugin_Extension.ts",
     "./src/stores/extensions/DSL_Generation_LegendStudioApplicationPlugin_Extension.ts",
     "./src/stores/extensions/DSL_Mapping_LegendStudioApplicationPlugin_Extension.ts",