jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor SharedState build method to accept optional record parameter

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-02 00:47:04 -03:00
commit 5ebcdf5d52
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/langflow/components/utilities/SharedState.py
View file

@ -1,4 +1,4 @@
from typing import Union from typing import Optional
from langflow import CustomComponent from langflow import CustomComponent
from langflow.field_typing import Text from langflow.field_typing import Text
@ -20,12 +20,13 @@ class SharedState(CustomComponent):
} }
def build( def build(
self, name: str, record: Union[Text, Record], append: bool = False self, name: str, record: Optional[Record] = None, append: bool = False
) -> Record: ) -> Record:
if append: if record:
self.append_state(name, record) if append:
else: self.append_state(name, record)
self.update_state(name, record) else:
self.update_state(name, record)
state = self.get_state(name) state = self.get_state(name)
if state and not isinstance(state, Record): if state and not isinstance(state, Record):