🔄 Update Union parser instantiation

shroominic · Jan 22, 2024 · 26bee8a · 26bee8a
1 parent 6bf5896
commit 26bee8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/funcchain/parser/selector.py b/src/funcchain/parser/selector.py
@@ -18,8 +18,8 @@ def parser_for(
     Get the parser from the type annotation of the parent caller function.
     """
     if isinstance(output_type, types.UnionType) or getattr(output_type, "__origin__", None) is Union:
-        output_type = output_type.__args__[0]  # type: ignore
-        return RetryJsonPydanticUnionParser(pydantic_objects=output_type.__args__)  # type: ignore  # TODO: fix this
+        # output_type = output_type.__args__[0]  # type: ignore
+        return RetryJsonPydanticUnionParser()  # type: ignore  # TODO: fix this
     if output_type is str:
         return StrOutputParser()
     if output_type is bool: