From dafaf6f016cc872fa4015d5ae20f27d89b8a3fb3 Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Wed, 7 Aug 2024 06:46:51 -0700 Subject: [PATCH] Fix response_schema sample. Change-Id: Id06f8556aef0a9b20204992b34c1d4bbc1437ef9 --- samples/controlled_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samples/controlled_generation.py b/samples/controlled_generation.py index e46b1a912..b0c269bb7 100644 --- 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)