WIP vertex tool calls

Author: aabmass

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

@@ -23,7 +23,7 @@
 from __future__ import annotations
 
 from dataclasses import asdict, dataclass
-from typing import Literal
+from typing import Literal, Sequence
 
 from opentelemetry._events import Event
 from opentelemetry.semconv._incubating.attributes import gen_ai_attributes
@@ -104,18 +104,31 @@ class ChoiceMessage:
     role: str = "assistant"
 
 
+@dataclass
+class ChoiceToolCall:
+    """The tool_calls field for a gen_ai.choice event"""
+
+    @dataclass
+    class Function:
+        name: str
+        arguments: AnyValue = None
+
+    function: Function
+    id: str
+    type: Literal["function"]
+
+
 FinishReason = Literal[
     "content_filter", "error", "length", "stop", "tool_calls"
 ]
 
 
-# TODO add tool calls
-# https://github.com/open-telemetry/opentelemetry-python-contrib/issues/3216
 def choice_event(
     *,
     finish_reason: FinishReason | str,
     index: int,
     message: ChoiceMessage,
+    tool_calls: Sequence[ChoiceToolCall] | None = None,
 ) -> Event:
     """Creates a choice event, which describes the Gen AI response message.
     https://github.com/open-telemetry/semantic-conventions/blob/v1.28.0/docs/gen-ai/gen-ai-events.md#event-gen_aichoice
@@ -125,6 +138,8 @@ def choice_event(
         "index": index,
         "message": asdict(message),
     }
+    if tool_calls:
+        body["tool_calls"] = [asdict(tool_call) for tool_call in tool_calls]
 
     return Event(
         name="gen_ai.choice",

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

@@ -242,6 +242,8 @@ def response_to_events(
                     parts=candidate.content.parts,
                 ),
             ),
+            # TODO: tool calls
+            tool_calls=None,
         )