github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 15:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Re-add __add__ functionality for messages (revert #8245) (#8489)

Browse Source 
This PR reverts #8245, so `__add__` is defined on base messages.

Resolves issue: https://github.com/langchain-ai/langchain/issues/8472
This commit is contained in:
Eugene Yurtsev 2023-08-02 10:51:44 -04:00 committed by GitHub
parent f31047a394
commit 1ec0b18379
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libs/langchain/langchain/schema/messages.py
View File

@ -1,12 +1,15 @@
from __future__ import annotations from __future__ import annotations
from abc import abstractmethod from abc import abstractmethod
from typing import Any, Dict, List, Sequence from typing import TYPE_CHECKING, Any, Dict, List, Sequence
from pydantic import Field from pydantic import Field
from langchain.load.serializable import Serializable from langchain.load.serializable import Serializable
if TYPE_CHECKING:
from langchain.prompts.chat import ChatPromptTemplate
def get_buffer_string( def get_buffer_string(
messages: Sequence[BaseMessage], human_prefix: str = "Human", ai_prefix: str = "AI" messages: Sequence[BaseMessage], human_prefix: str = "Human", ai_prefix: str = "AI"
@ -77,6 +80,12 @@ class BaseMessage(Serializable):
"""Whether this class is LangChain serializable.""" """Whether this class is LangChain serializable."""
return True return True
def __add__(self, other: Any) -> ChatPromptTemplate:
from langchain.prompts.chat import ChatPromptTemplate
prompt = ChatPromptTemplate(messages=[self])
return prompt + other
class BaseMessageChunk(BaseMessage): class BaseMessageChunk(BaseMessage):
def _merge_kwargs_dict( def _merge_kwargs_dict(
@ -102,7 +111,7 @@ class BaseMessageChunk(BaseMessage):
) )
return merged return merged
def __add__(self, other: Any) -> BaseMessageChunk: def __add__(self, other: Any) -> BaseMessageChunk: # type: ignore
if isinstance(other, BaseMessageChunk): if isinstance(other, BaseMessageChunk):
# If both are (subclasses of) BaseMessageChunk, # If both are (subclasses of) BaseMessageChunk,
# concat into a single BaseMessageChunk # concat into a single BaseMessageChunk