fixes after review

langchain-ai · Feb 20, 2024 · 825d019 · 825d019
1 parent 072559d
commit 825d019
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/libs/vertexai/langchain_google_vertexai/chat_models.py b/libs/vertexai/langchain_google_vertexai/chat_models.py
@@ -397,14 +397,14 @@ def _generate(
             )
             generations = [
                 ChatGeneration(
-                    message=_parse_response_candidate(c),
+                    message=_parse_response_candidate(candidate),
                     generation_info=get_generation_info(
-                        c,
+                        candidate,
                         self._is_gemini_model,
                         usage_metadata=response.to_dict().get("usage_metadata"),
                     ),
                 )
-                for c in response.candidates
+                for candidate in response.candidates
             ]
         else:
             question = _get_question(messages)