Enable serialization for anthropic (#6049)

This commit is contained in:
Nuno Campos authored and GitHub committed 2023-06-12 22:39:10 -07:00
1 parent 6ac5d80286
commit a9b3b2e327
1 file changed
+4
+4
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}"