Bugfix: run_metadata value returns string instead of other types (#2149)

* properly handle primitive types

* extend tests

* Auto-update of E2E template

* Auto-update of NLP template

---------

Co-authored-by: GitHub Actions <[email protected]>

Author: avishniakov

src/zenml/models/v2/core/run_metadata.py

@@ -51,6 +51,11 @@ class RunMetadataRequest(WorkspaceScopedRequest):
         title="The types of the metadata to be created.",
     )
 
+    class Config:
+        """Pydantic configuration."""
+
+        smart_union = True
+
 
 # ------------------ Update Model ------------------
 
@@ -75,6 +80,11 @@ class RunMetadataResponseBody(WorkspaceScopedResponseBody):
         max_length=STR_FIELD_MAX_LENGTH,
     )
 
+    class Config:
+        """Pydantic configuration."""
+
+        smart_union = True
+
 
 class RunMetadataResponseMetadata(WorkspaceScopedResponseMetadata):
     """Response metadata for run metadata."""

tests/integration/functional/artifacts/test_utils.py

@@ -124,6 +124,17 @@ def test_log_artifact_metadata_existing(clean_client):
         artifact_name="meaning_of_life",
         artifact_version="1",
     )
+    log_artifact_metadata(
+        {
+            "float": 1.0,
+            "int": 1,
+            "str": "1.0",
+            "list_str": ["1.0", "2.0"],
+            "list_floats": [1.0, 2.0],
+        },
+        artifact_name="meaning_of_life",
+        artifact_version="1",
+    )
 
     artifact_1 = clean_client.get_artifact_version(
         "meaning_of_life", version="1"
@@ -132,6 +143,23 @@ def test_log_artifact_metadata_existing(clean_client):
     assert artifact_1.run_metadata["description"].value == "Aria is great!"
     assert "description_3" in artifact_1.run_metadata
     assert artifact_1.run_metadata["description_3"].value == "Axl is great!"
+    assert "float" in artifact_1.run_metadata
+    assert artifact_1.run_metadata["float"].value - 1.0 < 10e-6
+    assert "int" in artifact_1.run_metadata
+    assert artifact_1.run_metadata["int"].value == 1
+    assert "str" in artifact_1.run_metadata
+    assert artifact_1.run_metadata["str"].value == "1.0"
+    assert "list_str" in artifact_1.run_metadata
+    assert (
+        len(set(artifact_1.run_metadata["list_str"].value) - {"1.0", "2.0"})
+        == 0
+    )
+    assert "list_floats" in artifact_1.run_metadata
+    for each in artifact_1.run_metadata["list_floats"].value:
+        if 0.99 < each < 1.01:
+            assert each - 1.0 < 10e-6
+        else:
+            assert each - 2.0 < 10e-6
 
     artifact_2 = clean_client.get_artifact_version(
         "meaning_of_life", version="43"