Enable serialization for anthropic (#6049)

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

View File

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