fix(variable_pool): handle invalid attributes in variable lookup (#9646)

2024-10-23 10:19:33 +08:00 · 2024-10-23 10:19:33 +08:00 · d5a93a6400
commit d5a93a6400
parent 01a2513812
2 changed files with 10 additions and 6 deletions
--- a/api/tests/unit_tests/core/workflow/test_variable_pool.py
+++ b/api/tests/unit_tests/core/workflow/test_variable_pool.py
@ -33,8 +33,8 @@ def test_get_file_attribute(pool, file):
    assert result.value == file.filename

    # Test getting a non-existent attribute
-    with pytest.raises(ValueError):
-        pool.get(("node_1", "file_var", "non_existent_attr"))
+    result = pool.get(("node_1", "file_var", "non_existent_attr"))
+    assert result is None


 def test_use_long_selector(pool):