From 48ec2d2fa8a8ed8128aba789a9ed6196767129a3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 2 Mar 2024 15:18:58 -0300 Subject: [PATCH] Fix empty records bug and return empty string if state is not found --- src/backend/langflow/components/utilities/RecordsAsText.py | 2 ++ src/backend/langflow/graph/graph/state_manager.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/components/utilities/RecordsAsText.py b/src/backend/langflow/components/utilities/RecordsAsText.py index debf3eed2..18bf8be8c 100644 --- a/src/backend/langflow/components/utilities/RecordsAsText.py +++ b/src/backend/langflow/components/utilities/RecordsAsText.py @@ -25,6 +25,8 @@ class RecordsAsTextComponent(CustomComponent): records: list[Record], template: str = "Text: {text}\nData: {data}", ) -> Text: + if not records: + return "" if isinstance(records, Record): records = [records] diff --git a/src/backend/langflow/graph/graph/state_manager.py b/src/backend/langflow/graph/graph/state_manager.py index 46477d220..3fcbb68a3 100644 --- a/src/backend/langflow/graph/graph/state_manager.py +++ b/src/backend/langflow/graph/graph/state_manager.py @@ -27,7 +27,7 @@ class GraphStateManager: def get_state(self, key): with self.lock: - return self.states.get(key, None) + return self.states.get(key, "") def subscribe(self, key, observer: Callable): with self.lock: