From d1a518fdfa6232b5457a3e44f3c7c38e80e79926 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Sat, 19 Aug 2023 10:35:04 +0100 Subject: [PATCH] Allow non-dict output --- permchain/pubsub.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/permchain/pubsub.py b/permchain/pubsub.py index d52a8bc8b..e495a9bb3 100644 --- a/permchain/pubsub.py +++ b/permchain/pubsub.py @@ -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()