fix: Fix serialization of typing.Any when using serialize_type utility (#8853)

* Fix issue and expand tests

* Add reno

Author: sjrl

haystack/utils/type_serialization.py

@@ -35,7 +35,7 @@ def serialize_type(target: Any) -> str:
         return target
 
     # Determine if the target is a type or an instance of a typing object
-    is_type_or_typing = isinstance(target, type) or bool(get_origin(target))
+    is_type_or_typing = isinstance(target, type) or bool(get_origin(target)) or target == Any
     type_obj = target if is_type_or_typing else type(target)
     type_obj_repr = repr(type_obj)
 

releasenotes/notes/fix-serialize-any-type-ea0ca14eadb21b3d.yaml

@@ -0,0 +1,4 @@
+---
+fixes:
+  - |
+    Fixes serialization of typing.Any when using serialize_type utility

test/utils/test_type_serialization.py

@@ -1,10 +1,9 @@
 # SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
-import copy
 import sys
 import typing
-from typing import List, Dict
+from typing import Any, List, Dict, Set, Tuple
 
 import pytest
 
@@ -14,16 +13,59 @@
 
 def test_output_type_serialization():
     assert serialize_type(str) == "str"
+    assert serialize_type(int) == "int"
+    assert serialize_type(dict) == "dict"
+    assert serialize_type(float) == "float"
+    assert serialize_type(bool) == "bool"
+
+
+def test_output_type_serialization_typing():
+    assert serialize_type(Any) == "typing.Any"
+    assert serialize_type(List) == "typing.List"
+    assert serialize_type(Dict) == "typing.Dict"
+    assert serialize_type(Set) == "typing.Set"
+    assert serialize_type(Tuple) == "typing.Tuple"
+
+
+def test_output_type_serialization_typing_and_type():
+    # List
+    assert serialize_type(List[str]) == "typing.List[str]"
     assert serialize_type(List[int]) == "typing.List[int]"
+    assert serialize_type(List[dict]) == "typing.List[dict]"
+    assert serialize_type(List[float]) == "typing.List[float]"
+    assert serialize_type(List[bool]) == "typing.List[bool]"
+    # Dict
+    assert serialize_type(Dict[str, str]) == "typing.Dict[str, str]"
+    assert serialize_type(Dict[int, int]) == "typing.Dict[int, int]"
+    assert serialize_type(Dict[dict, dict]) == "typing.Dict[dict, dict]"
+    assert serialize_type(Dict[float, float]) == "typing.Dict[float, float]"
+    assert serialize_type(Dict[bool, bool]) == "typing.Dict[bool, bool]"
+    # Set
+    assert serialize_type(Set[str]) == "typing.Set[str]"
+    assert serialize_type(Set[int]) == "typing.Set[int]"
+    assert serialize_type(Set[dict]) == "typing.Set[dict]"
+    assert serialize_type(Set[float]) == "typing.Set[float]"
+    assert serialize_type(Set[bool]) == "typing.Set[bool]"
+    # Tuple
+    assert serialize_type(Tuple[str]) == "typing.Tuple[str]"
+    assert serialize_type(Tuple[int]) == "typing.Tuple[int]"
+    assert serialize_type(Tuple[dict]) == "typing.Tuple[dict]"
+    assert serialize_type(Tuple[float]) == "typing.Tuple[float]"
+    assert serialize_type(Tuple[bool]) == "typing.Tuple[bool]"
+
+
+def test_output_type_serialization_nested():
     assert serialize_type(List[Dict[str, int]]) == "typing.List[typing.Dict[str, int]]"
-    assert serialize_type(ChatMessage) == "haystack.dataclasses.chat_message.ChatMessage"
     assert serialize_type(typing.List[Dict[str, int]]) == "typing.List[typing.Dict[str, int]]"
+
+
+def test_output_type_serialization_haystack_dataclasses():
+    assert serialize_type(ChatMessage) == "haystack.dataclasses.chat_message.ChatMessage"
     assert serialize_type(List[ChatMessage]) == "typing.List[haystack.dataclasses.chat_message.ChatMessage]"
     assert (
         serialize_type(typing.Dict[int, ChatMessage])
         == "typing.Dict[int, haystack.dataclasses.chat_message.ChatMessage]"
     )
-    assert serialize_type(int) == "int"
     assert serialize_type(ChatMessage.from_user("ciao")) == "haystack.dataclasses.chat_message.ChatMessage"
 
 
@@ -41,18 +83,35 @@ def test_output_type_serialization_pep585():
 
 def test_output_type_deserialization():
     assert deserialize_type("str") == str
+    assert deserialize_type("int") == int
+    assert deserialize_type("dict") == dict
+    assert deserialize_type("float") == float
+    assert deserialize_type("bool") == bool
+
+
+def test_output_type_deserialization_typing():
+    assert deserialize_type("typing.Any") == Any
+    assert deserialize_type("typing.List") == List
+    assert deserialize_type("typing.Dict") == Dict
+    assert deserialize_type("typing.Set") == Set
+    assert deserialize_type("typing.Tuple") == Tuple
+
+
+def test_output_type_deserialization_typing_and_type():
     assert deserialize_type("typing.List[int]") == typing.List[int]
-    assert deserialize_type("typing.List[typing.Dict[str, int]]") == typing.List[Dict[str, int]]
     assert deserialize_type("typing.Dict[str, int]") == Dict[str, int]
+    assert deserialize_type("typing.List[typing.Dict[str, int]]") == typing.List[Dict[str, int]]
     assert deserialize_type("typing.Dict[str, typing.List[int]]") == Dict[str, List[int]]
     assert deserialize_type("typing.List[typing.Dict[str, typing.List[int]]]") == List[Dict[str, List[int]]]
+
+
+def test_output_type_deserialization_haystack_dataclasses():
     assert deserialize_type("typing.List[haystack.dataclasses.chat_message.ChatMessage]") == typing.List[ChatMessage]
     assert (
         deserialize_type("typing.Dict[int, haystack.dataclasses.chat_message.ChatMessage]")
         == typing.Dict[int, ChatMessage]
     )
     assert deserialize_type("haystack.dataclasses.chat_message.ChatMessage") == ChatMessage
-    assert deserialize_type("int") == int
 
 
 def test_output_type_deserialization_pep585():