format

Change-Id: I8f6f9389f1cae0a7c934217968d4e2e20bb9590e

Author: MarkDaoust

samples/controlled_generation.py

@@ -90,14 +90,12 @@ class Recipe(TypedDict):
             recipe_name: str
             grade: Grade
 
-
         model = genai.GenerativeModel("gemini-1.5-pro-latest")
 
         result = model.generate_content(
             "List about 10 cookie recipes, grade them based on popularity",
             generation_config=genai.GenerationConfig(
-                response_mime_type="application/json",
-                response_schema=list[Recipe]
+                response_mime_type="application/json", response_schema=list[Recipe]
             ),
         )
         print(result)  # [{"grade": "a+", "recipe_name": "Chocolate Chip Cookies"}, ...]
@@ -121,7 +119,6 @@ def test_json_enum_raw(self):
         print(result)  # "Keyboard"
         # [END json_enum_raw]
 
-
     def test_x_enum(self):
         # [START x_enum]
         import enum
@@ -163,5 +160,6 @@ def test_x_enum_raw(self):
         print(result)  # "Keyboard"
         # [END x_enum_raw]
 
+
 if __name__ == "__main__":
     absltest.main()