vertexai: add finish message when available to the response object fr…

…om gemini … (#595)
langchain-ai · Nov 14, 2024 · f199127 · f199127
1 parent 28eafae
commit f199127
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/vertexai/langchain_google_vertexai/_utils.py b/libs/vertexai/langchain_google_vertexai/_utils.py
@@ -178,6 +178,9 @@ def get_generation_info(
             "finish_reason": (
                 candidate.finish_reason.name if candidate.finish_reason else None
             ),
+            "finish_message": (
+                candidate.finish_message if candidate.finish_message else None
+            ),
         }
         if hasattr(candidate, "avg_logprobs") and candidate.avg_logprobs is not None:
             if (