feat: GenAI - Added function_calls shortcut property to Candidate class.

PiperOrigin-RevId: 613984044

Author: happy-qiao

tests/system/vertexai/test_generative_models.py

@@ -280,7 +280,12 @@ def test_generate_content_function_calling(self):
             == "get_current_weather"
         )
 
-        assert response.candidates[0].content.parts[0].function_call.args["location"]
+        assert response.candidates[0].function_calls[0].args["location"]
+        assert len(response.candidates[0].function_calls) == 1
+        assert (
+            response.candidates[0].function_calls[0]
+            == response.candidates[0].content.parts[0].function_call
+        )
 
         # fake api_response data
         api_response = {
@@ -309,6 +314,7 @@ def test_generate_content_function_calling(self):
             tools=[weather_tool],
         )
         assert response
+        assert len(response.candidates[0].function_calls) == 0
 
         # Get the model summary response
         summary = response.candidates[0].content.parts[0].text

tests/unit/vertexai/test_generative_models.py

@@ -23,7 +23,9 @@
 import vertexai
 from google.cloud.aiplatform import initializer
 from vertexai import generative_models
-from vertexai.preview import generative_models as preview_generative_models
+from vertexai.preview import (
+    generative_models as preview_generative_models,
+)
 from vertexai.generative_models._generative_models import (
     prediction_service,
     gapic_prediction_service_types,
@@ -352,6 +354,10 @@ def test_chat_function_calling(self, generative_models: generative_models):
             response1.candidates[0].content.parts[0].function_call.name
             == "get_current_weather"
         )
+        assert [
+            function_call.name
+            for function_call in response1.candidates[0].function_calls
+        ] == ["get_current_weather"]
         response2 = chat.send_message(
             generative_models.Part.from_function_response(
                 name="get_current_weather",
@@ -361,6 +367,7 @@ def test_chat_function_calling(self, generative_models: generative_models):
             ),
         )
         assert response2.text == "The weather in Boston is super nice!"
+        assert len(response2.candidates[0].function_calls) == 0
 
     @mock.patch.object(
         target=prediction_service.PredictionServiceClient,

vertexai/generative_models/_generative_models.py

@@ -1217,9 +1217,7 @@ def from_retrieval(
         cls,
         retrieval: "Retrieval",
     ):
-        raw_tool = gapic_tool_types.Tool(
-            retrieval=retrieval._raw_retrieval
-        )
+        raw_tool = gapic_tool_types.Tool(retrieval=retrieval._raw_retrieval)
         return cls._from_gapic(raw_tool=raw_tool)
 
     @classmethod
@@ -1460,6 +1458,16 @@ def citation_metadata(self) -> gapic_content_types.CitationMetadata:
     def text(self) -> str:
         return self.content.text
 
+    @property
+    def function_calls(self) -> Sequence[gapic_tool_types.FunctionCall]:
+        if not self.content or not self.content.parts:
+            return []
+        return [
+            part.function_call
+            for part in self.content.parts
+            if part and part.function_call
+        ]
+
 
 class Content:
     r"""The multi-part content of a message.