From c790e6b67c7391a4269be42d98bb113947e9d49b Mon Sep 17 00:00:00 2001
From: Shilpa Kancharla <snkancharla@google.com>
Date: Sat, 29 Jun 2024 17:32:39 -0400
Subject: [PATCH] Add no json schema sample

---
 samples/chat.py                  |  9 ++++++---
 samples/controlled_generation.py | 17 ++++++++++++++---
 2 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/samples/chat.py b/samples/chat.py
index 5958979cc..5089450d9 100644
--- a/samples/chat.py
+++ b/samples/chat.py
@@ -31,7 +31,7 @@ def test_chat(self):
             ]
         )
         response = chat.send_message("I have 2 dogs in my house.")
-        print(response.text) 
+        print(response.text)
         response = chat.send_message("How many paws are in my house?")
         print(response.text)
         # [END chat]
@@ -62,11 +62,14 @@ def test_chat_streaming_with_images(self):
         model = genai.GenerativeModel("gemini-1.5-flash")
         chat = model.start_chat()
 
-        response = chat.send_message("Hello, I'm interested in learning about musical instruments. Can I show you one?", stream=True)
+        response = chat.send_message(
+            "Hello, I'm interested in learning about musical instruments. Can I show you one?",
+            stream=True,
+        )
         for chunk in response:
             print(chunk.text)  # Yes.
             print("_" * 80)
-        
+
         organ = genai.upload_file(media / "organ.jpg")
         response = chat.send_message(
             ["What family of intruments does this instrument belong to?", organ], stream=True
diff --git a/samples/controlled_generation.py b/samples/controlled_generation.py
index b34f8372a..e46b1a912 100644
--- a/samples/controlled_generation.py
+++ b/samples/controlled_generation.py
@@ -16,8 +16,8 @@
 
 
 class UnitTests(absltest.TestCase):
-    def test_controlled_generation(self):
-        # [START controlled_generation]
+    def test_json_controlled_generation(self):
+        # [START json_controlled_generation]
         import typing_extensions as typing
 
         class Recipe(typing.TypedDict):
@@ -31,7 +31,18 @@ class Recipe(typing.TypedDict):
             ),
         )
         print(result)
-        # [END controlled_generation]
+        # [END json_controlled_generation]
+
+    def test_json_no_schema(self):
+        # [START json_no_schema]
+        model = genai.GenerativeModel("gemini-1.5-pro-latest")
+        prompt = """List a few popular cookie recipes using this JSON schema:
+
+        Recipe = {'recipe_name': str}
+        Return: list[Recipe]"""
+        result = model.generate_content(prompt)
+        print(result)
+        # [END json_no_schema]
 
 
 if __name__ == "__main__":