Fix linting

langchain-ai · Feb 13, 2024 · afe9f42 · afe9f42
1 parent 3f2e0c4
commit afe9f42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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