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

🚀 feat(constants.py): import BaseChatMemory from langchain.memory.chat_memory module to add support for chat memory in custom interfaces

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-09 21:49:20 -03:00
commit 1dcdc02e4f
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/interface/custom/constants.py
View file

@ -9,7 +9,7 @@ from langchain.tools import Tool
from langchain.vectorstores.base import VectorStore from langchain.vectorstores.base import VectorStore
from langchain.schema import BaseOutputParser from langchain.schema import BaseOutputParser
from langchain.schema.memory import BaseMemory from langchain.schema.memory import BaseMemory
from langchain.memory.chat_memory import BaseChatMemory
LANGCHAIN_BASE_TYPES = { LANGCHAIN_BASE_TYPES = {
"Chain": Chain, "Chain": Chain,
@ -24,6 +24,7 @@ LANGCHAIN_BASE_TYPES = {
"BaseRetriever": BaseRetriever, "BaseRetriever": BaseRetriever,
"BaseOutputParser": BaseOutputParser, "BaseOutputParser": BaseOutputParser,
"BaseMemory": BaseMemory, "BaseMemory": BaseMemory,
"BaseChatMemory": BaseChatMemory,
} }
# Langchain base types plus Python base types # Langchain base types plus Python base types