✨ Support union output types.

shroominic · Jan 22, 2024 · da14b32 · da14b32
1 parent 26bee8a
commit da14b32
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/funcchain/schema/signature.py b/src/funcchain/schema/signature.py
@@ -1,3 +1,4 @@
+from types import UnionType
 from typing import Any, Generic, TypeVar
 
 from langchain_core.messages import BaseMessage
@@ -27,7 +28,7 @@ class Signature(BaseModel, Generic[T]):
     # -> e.g. Callbacks adds custom callbacks
     # -> e.g. SystemMessage adds a system message
 
-    output_type: type[T]
+    output_type: type[T] | UnionType
     """ Type to parse the output into. """
 
     # todo: is history really needed? maybe this could be a background optimization