fix: add null safety for finish_reason and gen_ai.tool.definitions attribute

Author: shuwpan

instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/patch.py

@@ -14,6 +14,7 @@
 
 from __future__ import annotations
 
+import json
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -168,6 +169,14 @@ def _build_invocation(
             request_attributes[GenAIAttributes.GEN_AI_OUTPUT_TYPE]
         )
 
+    if capture_content and params.tools:
+        from google.protobuf import json_format as _jf
+
+        tool_defs = [_jf.MessageToDict(t._pb) for t in params.tools]  # type: ignore[union-attr]
+        invocation.attributes[GenAIAttributes.GEN_AI_TOOL_DEFINITIONS] = (
+            json.dumps(tool_defs)
+        )
+
     return invocation
 
 

instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/utils.py

@@ -209,8 +209,11 @@ def convert_content_to_message_parts(
 
 def _map_finish_reason(
     finish_reason: content.Candidate.FinishReason
-    | content_v1beta1.Candidate.FinishReason,
+    | content_v1beta1.Candidate.FinishReason
+    | None,
 ) -> FinishReason | str:
+    if finish_reason is None:
+        return "error"
     EnumType = type(finish_reason)  # pylint: disable=invalid-name
     if (
         finish_reason is EnumType.FINISH_REASON_UNSPECIFIED