Add StoreMessages component to store messages
This commit is contained in:
parent
65a88cbcbd
commit
2033fd65ef
1 changed files with 26 additions and 0 deletions
26
src/backend/langflow/components/io/StoreMessages.py
Normal file
26
src/backend/langflow/components/io/StoreMessages.py
Normal file
|
|
@ -0,0 +1,26 @@
|
|||
from typing import List
|
||||
|
||||
from langflow import CustomComponent
|
||||
from langflow.memory import add_messages
|
||||
from langflow.schema import Record
|
||||
|
||||
|
||||
class StoreMessages(CustomComponent):
|
||||
display_name = "Store Messages"
|
||||
description = "Used to store messages."
|
||||
|
||||
def build_config(self):
|
||||
return {
|
||||
"records": {
|
||||
"display_name": "Records",
|
||||
"info": "The list of records to store. Each record should contain the keys 'sender', 'sender_name', and 'session_id'.",
|
||||
},
|
||||
}
|
||||
|
||||
def build(
|
||||
self,
|
||||
records: List[Record],
|
||||
) -> List[Record]:
|
||||
self.status = records
|
||||
records = add_messages(records)
|
||||
return records
|
||||
Loading…
Add table
Add a link
Reference in a new issue