From 39a2801095f0165629e14851bd97d10a84b2ea6d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 27 Feb 2024 18:07:12 -0300 Subject: [PATCH] Refactor message parameter in ContractEdge --- src/backend/langflow/graph/edge/base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/edge/base.py b/src/backend/langflow/graph/edge/base.py index 99b24e2da..6706156a0 100644 --- a/src/backend/langflow/graph/edge/base.py +++ b/src/backend/langflow/graph/edge/base.py @@ -137,14 +137,15 @@ class ContractEdge(Edge): log_transaction(self, source, target, "success") # If the target vertex is a power component we log messages if target.vertex_type == "ChatOutput" and ( - isinstance(target.params.get("message"), str) or isinstance(target.params.get("message"), dict) + isinstance(target.params.get("input_value"), str) + or isinstance(target.params.get("input_value"), dict) ): if target.params.get("message") == "": return self.result await log_message( sender=target.params.get("sender", ""), sender_name=target.params.get("sender_name", ""), - message=target.params.get("message", {}), + message=target.params.get("input_value", {}), session_id=target.params.get("session_id", ""), artifacts=target.artifacts, )