Fix response_schema sample. (#498)

MarkDaoust · web-flow · commit 36123289097d · 2024-08-07T06:48:57.000-07:00
Change-Id: Id06f8556aef0a9b20204992b34c1d4bbc1437ef9
diff --git a/samples/controlled_generation.py b/samples/controlled_generation.py
@@ -27,7 +27,7 @@ class Recipe(typing.TypedDict):
         result = model.generate_content(
             "List a few popular cookie recipes.",
             generation_config=genai.GenerationConfig(
-                response_mime_type="application/json", response_schema=list([Recipe])
+                response_mime_type="application/json", response_schema=list[Recipe]
             ),
         )
         print(result)

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ class Recipe(typing.TypedDict):`
`27`	`27`	`result = model.generate_content(`
`28`	`28`	`"List a few popular cookie recipes.",`
`29`	`29`	`generation_config=genai.GenerationConfig(`
`30`		`- response_mime_type="application/json", response_schema=list([Recipe])`
	`30`	`+ response_mime_type="application/json", response_schema=list[Recipe]`
`31`	`31`	`),`
`32`	`32`	`)`
`33`	`33`	`print(result)`