refactor(prompt): use Union in cast

mszenfeld · Dec 23, 2024 · d0cc7df · d0cc7df
1 parent d2fd3d4
commit d0cc7df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pydantic_ai_slim/pydantic_ai/prompt.py b/pydantic_ai_slim/pydantic_ai/prompt.py
@@ -77,7 +77,7 @@ def _normalize_type(content: Any) -> dict[str, Any] | list[Any] | BasicType:
     elif isinstance(content, Iterable) and not isinstance(content, (str, dict)):
         return list(cast(Iterable[Any], content))
     elif isinstance(content, (str, int, float, bool, dict)):
-        return cast(BasicType | dict[str, Any], content)
+        return cast(Union[BasicType, dict[str, Any]], content)
     else:
         raise TypeError(f'Unsupported content type: {type(content)}')