fix: correct a small mistake in SimpleChatModel. (#1685)

This commit is contained in:
Pandazki 2023-03-15 23:00:26 +08:00 committed by GitHub
parent 1279c8de39
commit ced412e1c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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