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