Allow non-dict output

2026-09-10 03:37:51 +02:00 · 2023-08-19 10:35:04 +01:00
parent 0d39132602
commit d1a518fdfa
1 changed files with 5 additions and 1 deletions
@@ -202,7 +202,11 @@ class PubSub(Serializable, Runnable[Any, Any], ABC):
                    run_manager.on_chain_error(exceptions[0])
                    raise exceptions[0]
                else:
-                    run_manager.on_chain_end(final_output)
+                    run_manager.on_chain_end(
+                        final_output
+                        if isinstance(final_output, dict)
+                        else {"output": final_output}
+                    )


 PubSub.update_forward_refs()