github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-11 13:55:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: fix chat history circular import (#23182)

Browse Source
This commit is contained in:
Bagatur 2024-06-19 09:08:36 -07:00 committed by GitHub
parent 883e90d06e
commit 677408bfc9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/core/langchain_core/chat_history.py
View File

@ -26,7 +26,6 @@ from langchain_core.messages import (
get_buffer_string, get_buffer_string,
) )
from langchain_core.pydantic_v1 import BaseModel, Field from langchain_core.pydantic_v1 import BaseModel, Field
from langchain_core.runnables import run_in_executor
class BaseChatMessageHistory(ABC): class BaseChatMessageHistory(ABC):
@ -102,6 +101,8 @@ class BaseChatMessageHistory(ABC):
In general, fetching messages may involve IO to the underlying In general, fetching messages may involve IO to the underlying
persistence layer. persistence layer.
""" """
from langchain_core.runnables.config import run_in_executor
return await run_in_executor(None, lambda: self.messages) return await run_in_executor(None, lambda: self.messages)
def add_user_message(self, message: Union[HumanMessage, str]) -> None: def add_user_message(self, message: Union[HumanMessage, str]) -> None:
@ -172,6 +173,8 @@ class BaseChatMessageHistory(ABC):
Args: Args:
messages: A list of BaseMessage objects to store. messages: A list of BaseMessage objects to store.
""" """
from langchain_core.runnables.config import run_in_executor
await run_in_executor(None, self.add_messages, messages) await run_in_executor(None, self.add_messages, messages)
@abstractmethod @abstractmethod
@ -180,6 +183,8 @@ class BaseChatMessageHistory(ABC):
async def aclear(self) -> None: async def aclear(self) -> None:
"""Remove all messages from the store""" """Remove all messages from the store"""
from langchain_core.runnables.config import run_in_executor
await run_in_executor(None, self.clear) await run_in_executor(None, self.clear)
def __str__(self) -> str: def __str__(self) -> str:

2
libs/core/langchain_core/runnables/__init__.py
View File

@ -36,6 +36,7 @@ from langchain_core.runnables.config import (
run_in_executor, run_in_executor,
) )
from langchain_core.runnables.fallbacks import RunnableWithFallbacks from langchain_core.runnables.fallbacks import RunnableWithFallbacks
from langchain_core.runnables.history import RunnableWithMessageHistory
from langchain_core.runnables.passthrough import ( from langchain_core.runnables.passthrough import (
RunnableAssign, RunnableAssign,
RunnablePassthrough, RunnablePassthrough,
@ -78,6 +79,7 @@ __all__ = [
"RunnablePick", "RunnablePick",
"RunnableSequence", "RunnableSequence",
"RunnableWithFallbacks", "RunnableWithFallbacks",
"RunnableWithMessageHistory",
"get_config_list", "get_config_list",
"aadd", "aadd",
"add", "add",

1
libs/core/tests/unit_tests/runnables/test_imports.py
View File

@ -26,6 +26,7 @@ EXPECTED_ALL = [
"RunnablePick", "RunnablePick",
"RunnableSequence", "RunnableSequence",
"RunnableWithFallbacks", "RunnableWithFallbacks",
"RunnableWithMessageHistory",
"get_config_list", "get_config_list",
"aadd", "aadd",
"add", "add",