diff --git a/libs/vertexai/langchain_google_vertexai/chat_models.py b/libs/vertexai/langchain_google_vertexai/chat_models.py index 1c62d76c7..4233b43a4 100644 --- a/libs/vertexai/langchain_google_vertexai/chat_models.py +++ b/libs/vertexai/langchain_google_vertexai/chat_models.py @@ -395,7 +395,7 @@ def _generate( response = chat.send_message(question.content, **msg_params) generations = [ ChatGeneration( - message=AIMessage(content=r.text), + message=AIMessage(content=r.text), # type: ignore generation_info=get_generation_info(r, self._is_gemini_model), ) for r in response.candidates