Add ExtractDataFromRecord and MergeRecords components
This commit is contained in:
parent
94122cbe44
commit
13a91a2625
2 changed files with 42 additions and 0 deletions
|
|
@ -0,0 +1,17 @@
|
|||
from langflow import CustomComponent
|
||||
from langflow.schema import Record
|
||||
|
||||
|
||||
class ExtractKeyFromRecordComponent(CustomComponent):
|
||||
display_name = "Extract Key From Record"
|
||||
description = "Extracts a key from a record."
|
||||
|
||||
field_config = {
|
||||
"record": {"display_name": "Record"},
|
||||
}
|
||||
|
||||
def build(self, record: Record, key: str, silent_error: bool = True) -> dict:
|
||||
|
||||
data = getattr(record, key)
|
||||
self.status = data
|
||||
return data
|
||||
25
src/backend/langflow/components/helpers/MergeRecords.py
Normal file
25
src/backend/langflow/components/helpers/MergeRecords.py
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
from langflow import CustomComponent
|
||||
from langflow.schema import Record
|
||||
|
||||
|
||||
class MergeRecordsComponent(CustomComponent):
|
||||
display_name = "Merge Records"
|
||||
description = "Merges records."
|
||||
|
||||
field_config = {
|
||||
"records": {"display_name": "Records"},
|
||||
}
|
||||
|
||||
def build(self, records: list[Record]) -> Record:
|
||||
if not records:
|
||||
return records
|
||||
if len(records) == 1:
|
||||
return records[0]
|
||||
merged_record = None
|
||||
for record in records:
|
||||
if merged_record is None:
|
||||
merged_record = record
|
||||
else:
|
||||
merged_record += record
|
||||
self.status = merged_record
|
||||
return merged_record
|
||||
Loading…
Add table
Add a link
Reference in a new issue