[InternalComponent] Add is_link_mode to internal component output (Azure#27717)

* fix link mode

* fix link mode

* test: add test for internal output attrs

* refactor: make environment_variables a property

Co-authored-by: Xingzhi Zhang <zhangxingzhi@microsoft.com>

Author: lalala123123

sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities/_input_outputs.py

@@ -106,8 +106,9 @@ def _from_base(cls, _input: Union[Input, Dict]) -> Optional["InternalInput"]:
 
 
 class InternalOutput(Output):
-    def __init__(self, *, datastore_mode=None, **kwargs):
+    def __init__(self, *, datastore_mode=None, is_link_mode=None, **kwargs):
         self.datastore_mode = datastore_mode
+        self.is_link_mode = is_link_mode
         super().__init__(**kwargs)
 
     @classmethod

sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities/command.py

@@ -30,7 +30,7 @@ def __init__(self, **kwargs):
         self._resources = kwargs.pop("resources", JobResourceConfiguration())
         self._compute = kwargs.pop("compute", None)
         self._environment = kwargs.pop("environment", None)
-        self.environment_variables = kwargs.pop("environment_variables", None)
+        self._environment_variables = kwargs.pop("environment_variables", None)
         self._limits = kwargs.pop("limits", CommandJobLimits())
         self._init = False
 
@@ -54,6 +54,16 @@ def environment(self, value: str):
         """Set the environment definition for the command."""
         self._environment = value
 
+    @property
+    def environment_variables(self) -> Dict[str, str]:
+        """Get the environment variables for the command."""
+        return self._environment_variables
+
+    @environment_variables.setter
+    def environment_variables(self, value: Dict[str, str]):
+        """Set the environment variables for the command."""
+        self._environment_variables = value
+
     @property
     def limits(self) -> CommandJobLimits:
         return self._limits

sdk/ml/azure-ai-ml/tests/internal/unittests/test_component.py

@@ -581,9 +581,18 @@ def test_component_input_with_attrs(self) -> None:
                 # yes will be converted to true in YAML 1.2, users may use "yes" as a workaround
                 "param_string_with_yes_value": {"default": "True", "type": "string"},
                 "param_string_with_quote_yes_value": {"default": "yes", "type": "string"},
+            },
+            "outputs": {
+                'output_data_path': {
+                    'datastore_mode': 'mount',
+                    'description': 'Path to the data',
+                    'is_link_mode': True,
+                    'type': 'path'
+                }
             }
         }
         assert component._to_rest_object().properties.component_spec["inputs"] == expected_inputs["inputs"]
+        assert component._to_rest_object().properties.component_spec["outputs"] == expected_inputs["outputs"]
         assert component._validate().passed is True, repr(component._validate())
 
         regenerated_component = Component._from_rest_object(component._to_rest_object())

sdk/ml/azure-ai-ml/tests/test_configs/internal/component_with_input_outputs/component_spec_with_attrs.yaml

@@ -46,6 +46,13 @@ inputs:
     type: Enum
     enum: [minimal, reuse, expiry, policies]
 
+outputs:
+  output_data_path:
+    type: path
+    description: Path to the data
+    is_link_mode: True
+    datastore_mode: mount
+
 environment: azureml:AzureEnv:1
 
 scope: