Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enable serialization for anthropic (#6049)

Browse Source
This commit is contained in:
Nuno Campos 2023-06-13 06:39:10 +01:00 committed by GitHub
parent 6ac5d80286
commit a9b3b2e327
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/chat_models/anthropic.py
View File

@ -44,6 +44,10 @@ class ChatAnthropic(BaseChatModel, _AnthropicCommon):
"""Return type of chat model.""" """Return type of chat model."""
return "anthropic-chat" return "anthropic-chat"
@property
def lc_serializable(self) -> bool:
return True
def _convert_one_message_to_text(self, message: BaseMessage) -> str: def _convert_one_message_to_text(self, message: BaseMessage) -> str:
if isinstance(message, ChatMessage): if isinstance(message, ChatMessage):
message_text = f"\n\n{message.role.capitalize()}: {message.content}" message_text = f"\n\n{message.role.capitalize()}: {message.content}"