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

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}"