jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor import statements and update field info in MemoryFrontendNode class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 12:36:49 -03:00
commit ef357198b9
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/backend/langflow/template/frontend_node/memories.py
View file

@ -4,7 +4,9 @@ from langchain_community.chat_message_histories.mongodb import (
DEFAULT_COLLECTION_NAME, DEFAULT_COLLECTION_NAME,
DEFAULT_DBNAME, DEFAULT_DBNAME,
) )
from langchain_community.chat_message_histories.postgres import DEFAULT_CONNECTION_STRING from langchain_community.chat_message_histories.postgres import (
DEFAULT_CONNECTION_STRING,
)
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
@ -13,7 +15,9 @@ from langflow.template.template.base import Template
class MemoryFrontendNode(FrontendNode): class MemoryFrontendNode(FrontendNode):
#! Needs testing
pinned: bool = True
def add_extra_fields(self) -> None: def add_extra_fields(self) -> None:
# chat history should have another way to add common field? # chat history should have another way to add common field?
# prevent adding incorect field in ChatMessageHistory # prevent adding incorect field in ChatMessageHistory
@ -77,7 +81,9 @@ class MemoryFrontendNode(FrontendNode):
field.show = True field.show = True
field.advanced = False field.advanced = False
field.value = "" field.value = ""
field.info = INPUT_KEY_INFO if field.name == "input_key" else OUTPUT_KEY_INFO field.info = (
INPUT_KEY_INFO if field.name == "input_key" else OUTPUT_KEY_INFO
)
if field.name == "memory_key": if field.name == "memory_key":
field.value = "chat_history" field.value = "chat_history"