diff --git a/libs/community/langchain_community/chat_models/yandex.py b/libs/community/langchain_community/chat_models/yandex.py index 02aed416..faa788fb 100644 --- a/libs/community/langchain_community/chat_models/yandex.py +++ b/libs/community/langchain_community/chat_models/yandex.py @@ -92,7 +92,7 @@ def _generate( """ text = completion_with_retry(self, messages=messages) text = text if stop is None else enforce_stop_tokens(text, stop) - message = AIMessage(content=text) + message = AIMessage(content=text, additional_kwargs={}) return ChatResult(generations=[ChatGeneration(message=message)]) async def _agenerate( @@ -117,7 +117,7 @@ async def _agenerate( """ text = await acompletion_with_retry(self, messages=messages) text = text if stop is None else enforce_stop_tokens(text, stop) - message = AIMessage(content=text) + message = AIMessage(content=text, additional_kwargs={}) return ChatResult(generations=[ChatGeneration(message=message)])