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])