feat(genai): add python samples for text generation

Author: Valeriy-Burlaka

genai/text_generation/requirements-test.txt

@@ -0,0 +1,2 @@
+google-api-core==2.24.0
+pytest==8.2.0

genai/text_generation/requirements.txt

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

genai/text_generation/test_text_generation.py

@@ -0,0 +1,44 @@
+# 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.
+
+import textgen_with_txt
+import textgen_with_txt_img
+import textgen_with_txt_stream
+import textgen_chat_with_txt
+import textgen_chat_with_txt_stream
+
+
+def test_textgen_with_txt() -> None:
+    response = textgen_with_txt.generate_content()
+    assert response
+
+
+def test_textgen_with_txt_img() -> None:
+    response = textgen_with_txt_img.generate_content()
+    assert response
+
+
+def test_textgen_with_txt_stream() -> None:
+    response = textgen_with_txt_stream.generate_content()
+    assert response
+
+
+def test_textgen_chat_with_txt() -> None:
+    response = textgen_chat_with_txt.generate_content()
+    assert response
+
+
+def test_textgen_chat_with_txt_stream() -> None:
+    response = textgen_chat_with_txt_stream.generate_content()
+    assert response

genai/text_generation/textgen_chat_with_txt.py

@@ -0,0 +1,45 @@
+# 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_textgen_chat_with_txt]
+    from google import genai
+    from google.genai.types import Content, Part
+
+    client = genai.Client()
+    chat = client.chats.create(
+        model="gemini-2.0-flash-001",
+        history=[
+            Content(
+                parts=[Part(text="Hello")],
+                role="user"
+            ),
+            Content(
+                parts=[Part(text="Great to meet you. What would you like to know?")],
+                role="model"
+            )
+        ]
+    )
+    response = chat.send_message("tell me a story")
+    print(response.text)
+    # Example response:
+    # Okay, here's a story for you:
+    # ...
+    # [END googlegenaisdk_textgen_chat_with_txt]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/text_generation/textgen_chat_with_txt_stream.py

@@ -0,0 +1,37 @@
+# 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_textgen_chat_with_txt_stream]
+    from google import genai
+
+    client = genai.Client()
+    chat = client.chats.create(model="gemini-2.0-flash-001")
+    response_text = ""
+
+    for chunk in chat.send_message_stream("Why is the sky blue?"):
+        print(chunk.text)
+        response_text += chunk.text
+    # Example response:
+    # The
+    #  sky appears blue due to a phenomenon called **Rayleigh scattering**. Here's
+    #  a breakdown of why:
+    # ...
+    # [END googlegenaisdk_textgen_chat_with_txt_stream]
+    return response_text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/text_generation/textgen_with_txt.py

@@ -0,0 +1,36 @@
+# 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_textgen_with_txt]
+    from google import genai
+
+    client = genai.Client()
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-001",
+        contents="How does AI work?"
+    )
+    print(response.text)
+    # Example response:
+    # Okay, let's break down how AI works. It's a broad field, so I'll focus on the ...
+    #
+    # Here's a simplified overview:
+    # ...
+    # [END googlegenaisdk_textgen_with_txt]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/text_generation/textgen_with_txt_img.py

@@ -0,0 +1,40 @@
+# 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_textgen_with_txt_img]
+    from google import genai
+    from google.genai.types import Part
+
+    client = genai.Client()
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-001",
+        contents=[
+            "What is shown in this image?",
+            Part.from_uri(
+                file_uri="gs://cloud-samples-data/generative-ai/image/scones.jpg",
+                mime_type="image/jpeg"
+            )
+        ]
+    )
+    print(response.text)
+    # Example response:
+    # The image shows a flat lay of blueberry scones arranged on parchment paper. There are ...
+    # [END googlegenaisdk_textgen_with_txt_img]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/text_generation/textgen_with_txt_stream.py

@@ -0,0 +1,39 @@
+# 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_textgen_with_txt_stream]
+    from google import genai
+
+    client = genai.Client()
+    response_text = ""
+
+    for chunk in client.models.generate_content_stream(
+        model="gemini-2.0-flash-001",
+        contents="Why is the sky blue?"
+    ):
+        print(chunk.text)
+        response_text += chunk.text
+    # Example response:
+    # The
+    #  sky appears blue due to a phenomenon called **Rayleigh scattering**. Here's
+    #  a breakdown of why:
+    # ...
+    # [END googlegenaisdk_textgen_with_txt_stream]
+    return response_text
+
+
+if __name__ == "__main__":
+    generate_content()