fix: use origin identity check for types.UnionType in _unwrap_optional

Author: radu-mocanu

src/uipath_langchain/agent/react/json_utils.py

@@ -184,7 +184,7 @@ def _unwrap_optional(annotation: Any) -> Any:
         The unwrapped type, or the original if not Optional/Union
     """
     origin = get_origin(annotation)
-    if origin is Union or isinstance(annotation, types.UnionType):
+    if origin is Union or origin is types.UnionType:
         args = get_args(annotation)
         non_none_args = [arg for arg in args if arg is not type(None)]
         if non_none_args:

tests/agent/react/test_json_utils.py

@@ -27,6 +27,10 @@ class WithOptional(BaseModel):
     target: Optional[Target] = None
 
 
+class WithPipeUnion(BaseModel):
+    target: Target | None = None
+
+
 class WithNestedList(BaseModel):
     matrix: list[list[Target]]
 
@@ -54,6 +58,10 @@ def test_optional_field(self):
         paths = get_json_paths_by_type(WithOptional, "Target")
         assert paths == ["$.target"]
 
+    def test_pipe_union_field(self):
+        paths = get_json_paths_by_type(WithPipeUnion, "Target")
+        assert paths == ["$.target"]
+
     def test_nested_list_of_lists(self):
         paths = get_json_paths_by_type(WithNestedList, "Target")
         assert paths == ["$.matrix[*][*]"]