Add DocumentToRecordComponent to convert documents to records
This commit is contained in:
parent
afe0d42dd0
commit
4100266aa6
1 changed files with 21 additions and 0 deletions
|
|
@ -0,0 +1,21 @@
|
||||||
|
from git import List
|
||||||
|
from langchain_core.documents import Document
|
||||||
|
|
||||||
|
from langflow import CustomComponent
|
||||||
|
from langflow.schema import Record
|
||||||
|
|
||||||
|
|
||||||
|
class DocumentToRecordComponent(CustomComponent):
|
||||||
|
display_name = "Documents to Records"
|
||||||
|
description = "Convert documents to records."
|
||||||
|
|
||||||
|
field_config = {
|
||||||
|
"documents": {"display_name": "Documents"},
|
||||||
|
}
|
||||||
|
|
||||||
|
def build(self, documents: List[Document]) -> List[Record]:
|
||||||
|
if isinstance(documents, Document):
|
||||||
|
documents = [documents]
|
||||||
|
records = [Record.from_document(document) for document in documents]
|
||||||
|
self.status = records
|
||||||
|
return records
|
||||||
Loading…
Add table
Add a link
Reference in a new issue