fix: add _source in rest object of internal component (Azure#28523)

Author: elliotzh

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

@@ -191,6 +191,7 @@ def _from_rest_object_to_init_params(cls, obj: ComponentVersionData) -> Dict:
 
     def _to_rest_object(self) -> ComponentVersionData:
         component = convert_ordered_dict_to_dict(self._to_dict())
+        component["_source"] = self._source
 
         properties = ComponentVersionDetails(
             component_spec=component,

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

@@ -99,6 +99,7 @@ def test_component_load(
                 json.dump(loaded_dict, f, indent=2)
         with open(json_path, "r") as f:
             expected_dict = json.load(f)
+            expected_dict["_source"] = "REMOTE.WORKSPACE.COMPONENT"
 
             # TODO: check if loaded environment is expected to be an ordered dict
             assert pydash.omit(loaded_dict, *omit_fields) == pydash.omit(expected_dict, *omit_fields)

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

@@ -81,6 +81,7 @@ def test_load_from_registered_internal_command_component_rest_obj(self):
             "tags": {"category": "Component Tutorial", "contact": "amldesigner@microsoft.com"},
             "type": "CommandComponent",
             "$schema": "https://componentsdk.azureedge.net/jsonschema/CommandComponent.json",
+            "_source": "CLASS",
             "version": "0.0.1",
             "display_name": "0.0.1",
             "is_deterministic": True,
@@ -139,6 +140,7 @@ def test_load_from_registered_internal_scope_component_rest_obj(self):
             "tags": {"org": "bing", "project": "relevance"},
             "type": "ScopeComponent",
             "$schema": "https://componentsdk.azureedge.net/jsonschema/CommandComponent.json",
+            "_source": "CLASS",
             "version": "85b54741.0bf9.4734.a5bb.0e469c7bf792",
             "display_name": "Convert Text to StructureStream",
             "is_deterministic": True,
@@ -218,8 +220,11 @@ def test_component_serialization(self, yaml_path):
             expected_dict["code"] = parse_local_path(expected_dict["code"], entity.base_path)
 
         assert entity._to_dict() == expected_dict
+
+        expected_rest_object = copy.deepcopy(expected_dict)
+        expected_rest_object["_source"] = "YAML.COMPONENT"
         rest_obj = entity._to_rest_object()
-        assert rest_obj.properties.component_spec == expected_dict
+        assert rest_obj.properties.component_spec == expected_rest_object
 
         # inherit input type map from Component._from_rest_object
         for input_port in expected_dict.get("inputs", {}).values():