From ced412e1c12586a032eef937b97f195b9542c6d3 Mon Sep 17 00:00:00 2001 From: Pandazki Date: Wed, 15 Mar 2023 23:00:26 +0800 Subject: [PATCH] fix: correct a small mistake in SimpleChatModel. (#1685) --- langchain/chat_models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/chat_models/base.py b/langchain/chat_models/base.py index fc8daba6d27..ffb7c03ce84 100644 --- a/langchain/chat_models/base.py +++ b/langchain/chat_models/base.py @@ -127,7 +127,7 @@ class SimpleChatModel(BaseChatModel): self, messages: List[BaseMessage], stop: Optional[List[str]] = None ) -> ChatResult: output_str = self._call(messages, stop=stop) - message = AIMessage(text=output_str) + message = AIMessage(content=output_str) generation = ChatGeneration(message=message) return ChatResult(generations=[generation])