feat(genai): Add additional Controlled Generation Samples

- CtrlGen with Enum Class
- CtrlGen with Resp Schema

Added requirements.txt

Author: holtskinner

genai/controlled_generation/ctrlgen_with_enum_class_schema.py

@@ -0,0 +1,47 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+def generate_content() -> str:
+    # [START googlegenaisdk_ctrlgen_with_enum_class_schema]
+    from google import genai
+    import enum
+
+    class InstrumentClass(enum.Enum):
+        PERCUSSION = "Percussion"
+        STRING = "String"
+        WOODWIND = "Woodwind"
+        BRASS = "Brass"
+        KEYBOARD = "Keyboard"
+
+    client = genai.Client()
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-001",
+        contents="What type of instrument is a guitar?",
+        config={
+            "response_mime_type": "text/x.enum",
+            "response_schema": InstrumentClass,
+        },
+    )
+
+    print(response.text)
+    # Example output:
+    # String
+
+    # [END googlegenaisdk_ctrlgen_with_enum_class_schema]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/controlled_generation/ctrlgen_with_resp_schema.py

@@ -0,0 +1,69 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+def generate_content() -> str:
+    # [START googlegenaisdk_ctrlgen_with_resp_schema]
+    from google import genai
+
+    response_schema = {
+        "type": "ARRAY",
+        "items": {
+            "type": "OBJECT",
+            "properties": {
+                "recipe_name": {"type": "STRING"},
+                "ingredients": {"type": "ARRAY", "items": {"type": "STRING"}},
+            },
+            "required": ["recipe_name", "ingredients"],
+        },
+    }
+
+    prompt = """
+        List a few popular cookie recipes.
+    """
+
+    client = genai.Client()
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-001",
+        contents=prompt,
+        config={
+            "response_mime_type": "application/json",
+            "response_schema": response_schema,
+        },
+    )
+
+    print(response.text)
+    # Example output:
+    # [
+    #     {
+    #         "ingredients": [
+    #             "2 1/4 cups all-purpose flour",
+    #             "1 teaspoon baking soda",
+    #             "1 teaspoon salt",
+    #             "1 cup (2 sticks) unsalted butter, softened",
+    #             "3/4 cup granulated sugar",
+    #             "3/4 cup packed brown sugar",
+    #             "1 teaspoon vanilla extract",
+    #             "2 large eggs",
+    #             "2 cups chocolate chips",
+    #         ],
+    #         "recipe_name": "Chocolate Chip Cookies",
+    #     }
+    # ]
+    # [END googlegenaisdk_ctrlgen_with_resp_schema]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/controlled_generation/requirements.txt

@@ -0,0 +1 @@
+google-genai==1.1.0

genai/controlled_generation/test_controlled_generation_samples.py

@@ -20,8 +20,10 @@
 
 import ctrlgen_with_class_schema
 import ctrlgen_with_enum_schema
+import ctrlgen_with_enum_class_schema
 import ctrlgen_with_nested_class_schema
 import ctrlgen_with_nullable_schema
+import ctrlgen_with_resp_schema
 
 os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
@@ -37,9 +39,17 @@ def test_ctrlgen_with_enum_schema() -> None:
     assert ctrlgen_with_enum_schema.generate_content()
 
 
+def test_ctrlgen_with_enum_class_schema() -> None:
+    assert ctrlgen_with_enum_class_schema.generate_content()
+
+
 def test_ctrlgen_with_nested_class_schema() -> None:
     assert ctrlgen_with_nested_class_schema.generate_content()
 
 
 def test_ctrlgen_with_nullable_schema() -> None:
     assert ctrlgen_with_nullable_schema.generate_content()
+
+
+def test_ctrlgen_with_resp_schema() -> None:
+    assert ctrlgen_with_resp_schema.generate_content()