mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-18 18:34:27 -04:00
Leonid Ganeline
c7ca350cd3
In LangChain, all module classes are enumerated in the `__init__.py` file of the correspondent module. But some classes were missed and were not included in the module `__init__.py` This PR: - added the missed classes to the module `__init__.py` files - `__init__.py:__all_` variable value (a list of the class names) was sorted - `langchain.tools.sql_database.tool.QueryCheckerTool` was renamed into the `QuerySQLCheckerTool` because it conflicted with `langchain.tools.spark_sql.tool.QueryCheckerTool` - changes to `pyproject.toml`: - added `pgvector` to `pyproject.toml:extended_testing` - added `pandas` to `pyproject.toml:[tool.poetry.group.test.dependencies]` - commented out the `streamlit` from `collbacks/__init__.py`, It is because now the `streamlit` requires Python >=3.7, !=3.9.7 - fixed duplicate names in `tools` - fixed correspondent ut-s #### Who can review? @hwchase17 @dev2049
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from langchain.memory.buffer import (
|
|
ConversationBufferMemory,
|
|
ConversationStringBufferMemory,
|
|
)
|
|
from langchain.memory.buffer_window import ConversationBufferWindowMemory
|
|
from langchain.memory.chat_message_histories import (
|
|
CassandraChatMessageHistory,
|
|
ChatMessageHistory,
|
|
CosmosDBChatMessageHistory,
|
|
DynamoDBChatMessageHistory,
|
|
FileChatMessageHistory,
|
|
MomentoChatMessageHistory,
|
|
MongoDBChatMessageHistory,
|
|
PostgresChatMessageHistory,
|
|
RedisChatMessageHistory,
|
|
SQLChatMessageHistory,
|
|
ZepChatMessageHistory,
|
|
)
|
|
from langchain.memory.combined import CombinedMemory
|
|
from langchain.memory.entity import (
|
|
ConversationEntityMemory,
|
|
InMemoryEntityStore,
|
|
RedisEntityStore,
|
|
SQLiteEntityStore,
|
|
)
|
|
from langchain.memory.kg import ConversationKGMemory
|
|
from langchain.memory.motorhead_memory import MotorheadMemory
|
|
from langchain.memory.readonly import ReadOnlySharedMemory
|
|
from langchain.memory.simple import SimpleMemory
|
|
from langchain.memory.summary import ConversationSummaryMemory
|
|
from langchain.memory.summary_buffer import ConversationSummaryBufferMemory
|
|
from langchain.memory.token_buffer import ConversationTokenBufferMemory
|
|
from langchain.memory.vectorstore import VectorStoreRetrieverMemory
|
|
|
|
__all__ = [
|
|
"CassandraChatMessageHistory",
|
|
"ChatMessageHistory",
|
|
"CombinedMemory",
|
|
"ConversationBufferMemory",
|
|
"ConversationBufferWindowMemory",
|
|
"ConversationEntityMemory",
|
|
"ConversationKGMemory",
|
|
"ConversationStringBufferMemory",
|
|
"ConversationSummaryBufferMemory",
|
|
"ConversationSummaryMemory",
|
|
"ConversationTokenBufferMemory",
|
|
"CosmosDBChatMessageHistory",
|
|
"DynamoDBChatMessageHistory",
|
|
"FileChatMessageHistory",
|
|
"InMemoryEntityStore",
|
|
"MomentoChatMessageHistory",
|
|
"MongoDBChatMessageHistory",
|
|
"MotorheadMemory",
|
|
"PostgresChatMessageHistory",
|
|
"ReadOnlySharedMemory",
|
|
"RedisChatMessageHistory",
|
|
"RedisEntityStore",
|
|
"SQLChatMessageHistory",
|
|
"SQLiteEntityStore",
|
|
"SimpleMemory",
|
|
"VectorStoreRetrieverMemory",
|
|
"ZepChatMessageHistory",
|
|
]
|