diff --git a/libs/partners/anthropic/langchain_anthropic/chat_models.py b/libs/partners/anthropic/langchain_anthropic/chat_models.py index 2aab9c3a062..330bcb4d6e3 100644 --- a/libs/partners/anthropic/langchain_anthropic/chat_models.py +++ b/libs/partners/anthropic/langchain_anthropic/chat_models.py @@ -263,7 +263,11 @@ class ChatAnthropic(BaseChatModel): await run_manager.on_llm_new_token(text, chunk=chunk) yield chunk - def _format_output(self, data: Any) -> ChatResult: + def _format_output( + self, + data: Any, + **kwargs: Any, + ) -> ChatResult: return ChatResult( generations=[ ChatGeneration(message=AIMessage(content=data.content[0].text)) @@ -285,7 +289,7 @@ class ChatAnthropic(BaseChatModel): return generate_from_stream(stream_iter) params = self._format_params(messages=messages, stop=stop, **kwargs) data = self._client.messages.create(**params) - return self._format_output(data) + return self._format_output(data, **kwargs) async def _agenerate( self, @@ -301,7 +305,7 @@ class ChatAnthropic(BaseChatModel): return await agenerate_from_stream(stream_iter) params = self._format_params(messages=messages, stop=stop, **kwargs) data = await self._async_client.messages.create(**params) - return self._format_output(data) + return self._format_output(data, **kwargs) @deprecated(since="0.1.0", removal="0.2.0", alternative="ChatAnthropic")