fix(generative_ai): fix typos in function calling sample (GoogleCloudPlatform#12679)

Valeriy-Burlaka · msampathkumar · web-flow · commit 1485c420f6b2 · 2024-10-09T14:00:43.000+02:00
* Fix typos in the function declaration 'description'; wrap long line
* Code formatting in generative_ai/function_calling/parallel_function_calling_example.py

Co-authored-by: Sampath Kumar &lt;sam1990kumar@gmail.com&gt;
diff --git a/generative_ai/context_caching/list_context_caches.py b/generative_ai/context_caching/list_context_caches.py
@@ -38,7 +38,6 @@ def list_content_caches() -> list[str]:
         # Cached content 'example-cache' for model '.../gemini-1.5-pro-001'
         # Last updated at: 2024-09-16T12:41:09.998635Z
         # Expires at: 2024-09-16T13:41:09.989729Z
-
     # [END generativeaionvertexai_context_caching_list]
     return [cached_content.name for cached_content in cache_list]
 
diff --git a/generative_ai/embeddings/batch_example.py b/generative_ai/embeddings/batch_example.py
@@ -26,6 +26,7 @@ def embed_text_batch() -> BatchPredictionJob:
     """
     # [START generativeaionvertexai_embedding_batch]
     import vertexai
+
     from vertexai.preview import language_models
 
     # TODO(developer): Update & uncomment line below
@@ -52,9 +53,7 @@ def embed_text_batch() -> BatchPredictionJob:
     # BatchPredictionJob 2024-09-10 15:47:51.336391
     # projects/1234567890/locations/us-central1/batchPredictionJobs/123456789012345
     # JobState.JOB_STATE_SUCCEEDED
-
     # [END generativeaionvertexai_embedding_batch]
-
     return batch_prediction_job
 
 
diff --git a/generative_ai/embeddings/generate_embeddings_with_lower_dimension.py b/generative_ai/embeddings/generate_embeddings_with_lower_dimension.py
@@ -55,9 +55,7 @@ def generate_embeddings_with_lower_dimension() -> MultiModalEmbeddingResponse:
     # Example response:
     # Image Embedding: [0.0622573346, -0.0406507477, 0.0260440577, ...]
     # Text Embedding: [0.27469793, -0.146258667, 0.0222803634, ...]
-
     # [END generativeaionvertexai_embeddings_specify_lower_dimension]
-
     return embeddings
 
 
diff --git a/generative_ai/evaluation/get_rouge_score.py b/generative_ai/evaluation/get_rouge_score.py
@@ -95,7 +95,6 @@ def get_rouge_score() -> EvalResult:
     # 0  Summarize the following text:\n\n\n    ...         0.659794       0.484211    ...
     # 1  Summarize the following text:\n\n\n    ...         0.704762       0.524272    ...
     # ...
-
     # [END generativeaionvertexai_evaluation_get_rouge_score]
     return result
 
diff --git a/generative_ai/evaluation/pairwise_summarization_quality.py b/generative_ai/evaluation/pairwise_summarization_quality.py
@@ -98,7 +98,6 @@ def evaluate_output() -> EvalResult:
     #
     # Winner: CANDIDATE
     # Explanation: Both responses adhere to the prompt's constraints, are grounded in the provided text, and ... However, Response B ...
-
     # [END generativeaionvertexai_evaluation_pairwise_summarization_quality]
     return result
 
diff --git a/generative_ai/function_calling/advanced_example.py b/generative_ai/function_calling/advanced_example.py
@@ -22,6 +22,7 @@
 def generate_function_call_advanced() -> GenerationResponse:
     # [START generativeaionvertexai_gemini_function_calling_advanced]
     import vertexai
+
     from vertexai.preview.generative_models import (
         FunctionDeclaration,
         GenerativeModel,
diff --git a/generative_ai/function_calling/basic_example.py b/generative_ai/function_calling/basic_example.py
@@ -22,6 +22,7 @@
 def generate_function_call() -> GenerationResponse:
     # [START generativeaionvertexai_gemini_function_calling]
     import vertexai
+
     from vertexai.generative_models import (
         Content,
         FunctionDeclaration,
diff --git a/generative_ai/function_calling/chat_example.py b/generative_ai/function_calling/chat_example.py
@@ -21,6 +21,7 @@
 def generate_function_call_chat() -> ChatSession:
     # [START generativeaionvertexai_gemini_function_calling_chat]
     import vertexai
+
     from vertexai.generative_models import (
         FunctionDeclaration,
         GenerationConfig,
diff --git a/generative_ai/function_calling/parallel_function_calling_example.py b/generative_ai/function_calling/parallel_function_calling_example.py
@@ -21,6 +21,7 @@
 def parallel_function_calling_example() -> ChatSession:
     # [START generativeaionvertexai_function_calling_generate_parallel_calls]
     import vertexai
+
     from vertexai.generative_models import (
         FunctionDeclaration,
         GenerativeModel,
@@ -44,7 +45,9 @@ def parallel_function_calling_example() -> ChatSession:
             "properties": {
                 "location": {
                     "type": "string",
-                    "description": "The location to whih to get the weather. Can be a city name, a city name and state, or a zip code. Examples: 'San Francisco', 'San Francisco, CA', '95616', etc.",
+                    "description": "The location for which to get the weather. \
+                      It can be a city name, a city name and state, or a zip code. \
+                      Examples: 'San Francisco', 'San Francisco, CA', '95616', etc.",
                 },
             },
         },
@@ -67,8 +70,8 @@ def mock_weather_api_service(location: str) -> str:
     )
 
     # Start a chat session
-    chat = model.start_chat()
-    response = chat.send_message("Get weather details in New Delhi and San Francisco?")
+    chat_session = model.start_chat()
+    response = chat_session.send_message("Get weather details in New Delhi and San Francisco?")
 
     function_calls = response.candidates[0].function_calls
     print("Suggested finction calls:\n", function_calls)
@@ -86,7 +89,7 @@ def mock_weather_api_service(location: str) -> str:
                 )
 
         # Return the API response to Gemini
-        response = chat.send_message(
+        response = chat_session.send_message(
             [
                 Part.from_function_response(
                     name="get_current_weather",
@@ -102,7 +105,6 @@ def mock_weather_api_service(location: str) -> str:
         print(response.text)
         # Example response:
         # The current weather in New Delhi is 35°C. The current weather in San Francisco is 25°C.
-
         # [END generativeaionvertexai_function_calling_generate_parallel_calls]
         return response
 

Original file line number	Diff line number	Diff line change
`@@ -98,7 +98,6 @@ def evaluate_output() -> EvalResult:`
`98`	`98`	`#`
`99`	`99`	`# Winner: CANDIDATE`
`100`	`100`	`# Explanation: Both responses adhere to the prompt's constraints, are grounded in the provided text, and ... However, Response B ...`
`101`		`-`
`102`	`101`	`# [END generativeaionvertexai_evaluation_pairwise_summarization_quality]`
`103`	`102`	`return result`
`104`	`103`