From 5ebcdf5d527c76437364dfd8ba00b954aa00df0b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 2 Mar 2024 00:47:04 -0300 Subject: [PATCH] Refactor SharedState build method to accept optional record parameter --- .../langflow/components/utilities/SharedState.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/src/backend/langflow/components/utilities/SharedState.py b/src/backend/langflow/components/utilities/SharedState.py index 845252dab..c3f0c0634 100644 --- a/src/backend/langflow/components/utilities/SharedState.py +++ b/src/backend/langflow/components/utilities/SharedState.py @@ -1,4 +1,4 @@ -from typing import Union +from typing import Optional from langflow import CustomComponent from langflow.field_typing import Text @@ -20,12 +20,13 @@ class SharedState(CustomComponent): } def build( - self, name: str, record: Union[Text, Record], append: bool = False + self, name: str, record: Optional[Record] = None, append: bool = False ) -> Record: - if append: - self.append_state(name, record) - else: - self.update_state(name, record) + if record: + if append: + self.append_state(name, record) + else: + self.update_state(name, record) state = self.get_state(name) if state and not isinstance(state, Record):