Refactor SharedState build method to accept optional record parameter

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-02 00:47:04 -03:00
commit 5ebcdf5d52

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,8 +20,9 @@ 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 record:
if append: if append:
self.append_state(name, record) self.append_state(name, record)
else: else: