diff --git a/libs/community/langchain_community/chat_models/litellm.py b/libs/community/langchain_community/chat_models/litellm.py index c92c8f38e..3ba87a11f 100644 --- a/libs/community/langchain_community/chat_models/litellm.py +++ b/libs/community/langchain_community/chat_models/litellm.py @@ -432,7 +432,7 @@ def _create_chat_result(self, response: Mapping[str, Any]) -> ChatResult: message.usage_metadata = _create_usage_metadata(token_usage) gen = ChatGeneration( message=message, - generation_info=dict(finish_reason=res.get("finish_reason")), + generation_info=dict(finish_reason=res.get("finish_reason"), logprobs=res.get("logprobs")), ) generations.append(gen) set_model_value = self.model