Move imports into samples (#638)

* Move imports into samples Change-Id: I86550a09c6555a36679dad517ed4239f7ded90ff * format Change-Id: I0c72841eb2ebad0075d0024624a42fe36fef883f
google-gemini · Nov 26, 2024 · 4c00257 · 4c00257
1 parent a04fcd1
commit 4c00257
Show file tree

Hide file tree

Showing 14 changed files with 145 additions and 22 deletions.
diff --git a/samples/cache.py b/samples/cache.py
@@ -14,7 +14,6 @@
 # limitations under the License.
 from absl.testing import absltest
 
-import google.generativeai as genai
 
 import pathlib
 
@@ -24,6 +23,8 @@
 class UnitTests(absltest.TestCase):
     def test_cache_create(self):
         # [START cache_create]
+        import google.generativeai as genai
+
         document = genai.upload_file(path=media / "a11.txt")
         model_name = "gemini-1.5-flash-001"
         cache = genai.caching.CachedContent.create(
@@ -41,6 +42,8 @@ def test_cache_create(self):
 
     def test_cache_create_from_name(self):
         # [START cache_create_from_name]
+        import google.generativeai as genai
+
         document = genai.upload_file(path=media / "a11.txt")
         model_name = "gemini-1.5-flash-001"
         cache = genai.caching.CachedContent.create(
@@ -60,6 +63,8 @@ def test_cache_create_from_name(self):
 
     def test_cache_create_from_chat(self):
         # [START cache_create_from_chat]
+        import google.generativeai as genai
+
         model_name = "gemini-1.5-flash-001"
         system_instruction = "You are an expert analyzing transcripts."
 
@@ -92,6 +97,8 @@ def test_cache_create_from_chat(self):
 
     def test_cache_delete(self):
         # [START cache_delete]
+        import google.generativeai as genai
+
         document = genai.upload_file(path=media / "a11.txt")
         model_name = "gemini-1.5-flash-001"
         cache = genai.caching.CachedContent.create(
@@ -104,6 +111,8 @@ def test_cache_delete(self):
 
     def test_cache_get(self):
         # [START cache_get]
+        import google.generativeai as genai
+
         document = genai.upload_file(path=media / "a11.txt")
         model_name = "gemini-1.5-flash-001"
         cache = genai.caching.CachedContent.create(
@@ -117,6 +126,8 @@ def test_cache_get(self):
 
     def test_cache_list(self):
         # [START cache_list]
+        import google.generativeai as genai
+
         document = genai.upload_file(path=media / "a11.txt")
         model_name = "gemini-1.5-flash-001"
         cache = genai.caching.CachedContent.create(
@@ -132,6 +143,8 @@ def test_cache_list(self):
 
     def test_cache_update(self):
         # [START cache_update]
+        import google.generativeai as genai
+
         import datetime
 
         document = genai.upload_file(path=media / "a11.txt")

diff --git a/samples/chat.py b/samples/chat.py
@@ -14,7 +14,6 @@
 # limitations under the License.
 from absl.testing import absltest
 
-import google.generativeai as genai
 import pathlib
 
 media = pathlib.Path(__file__).parents[1] / "third_party"
@@ -23,6 +22,8 @@
 class UnitTests(absltest.TestCase):
     def test_chat(self):
         # [START chat]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         chat = model.start_chat(
             history=[
@@ -38,6 +39,8 @@ def test_chat(self):
 
     def test_chat_streaming(self):
         # [START chat_streaming]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         chat = model.start_chat(
             history=[
@@ -59,6 +62,8 @@ def test_chat_streaming(self):
 
     def test_chat_streaming_with_images(self):
         # [START chat_streaming_with_images]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         chat = model.start_chat()
 

diff --git a/samples/code_execution.py b/samples/code_execution.py
@@ -14,12 +14,12 @@
 # limitations under the License.
 from absl.testing import absltest
 
-import google.generativeai as genai
-
 
 class UnitTests(absltest.TestCase):
     def test_code_execution_basic(self):
         # [START code_execution_basic]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel(model_name="gemini-1.5-flash", tools="code_execution")
         response = model.generate_content(
             (
@@ -38,6 +38,8 @@ def test_code_execution_basic(self):
         # [END code_execution_basic]
 
         # [START code_execution_basic_return]
+        import google.generativeai as genai
+
         # text: "I can help with that! To calculate the sum of the first 50 prime numbers, we\'ll need to first identify all the prime numbers up to the 50th prime number. \n\nHere is the code to find and sum the first 50 prime numbers:\n\n"
         #
         # executable_code {
@@ -92,6 +94,8 @@ def test_code_execution_basic(self):
 
     def test_code_execution_request_override(self):
         # [START code_execution_request_override]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel(model_name="gemini-1.5-flash")
         response = model.generate_content(
             (
@@ -103,6 +107,8 @@ def test_code_execution_request_override(self):
         print(response.text)
         # [END code_execution_request_override]
         # [START code_execution_request_override_return]
+        import google.generativeai as genai
+
         # ``` python
         # def is_prime(n):
         #     """
@@ -140,6 +146,8 @@ def test_code_execution_request_override(self):
 
     def test_code_execution_chat(self):
         # [START code_execution_chat]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel(model_name="gemini-1.5-flash", tools="code_execution")
         chat = model.start_chat()
         response = chat.send_message('Can you print "Hello world!"?')
@@ -152,6 +160,8 @@ def test_code_execution_chat(self):
         print(response.text)
         # [END code_execution_chat]
         # [START code_execution_chat_return]
+        import google.generativeai as genai
+
         # ``` python
         # def is_prime(n):
         #     """

diff --git a/samples/configure_model_parameters.py b/samples/configure_model_parameters.py
@@ -14,12 +14,12 @@
 # limitations under the License.
 from absl.testing import absltest
 
-import google.generativeai as genai
-
 
 class UnitTests(absltest.TestCase):
     def test_configure_model(self):
         # [START configure_model_parameters]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         response = model.generate_content(
             "Tell me a story about a magic backpack.",

diff --git a/samples/controlled_generation.py b/samples/controlled_generation.py
@@ -13,14 +13,15 @@
 from absl.testing import absltest
 import pathlib
 
-import google.generativeai as genai
 
 media = pathlib.Path(__file__).parents[1] / "third_party"
 
 
 class UnitTests(absltest.TestCase):
     def test_json_controlled_generation(self):
         # [START json_controlled_generation]
+        import google.generativeai as genai
+
         import typing_extensions as typing
 
         class Recipe(typing.TypedDict):
@@ -39,6 +40,8 @@ class Recipe(typing.TypedDict):
 
     def test_json_no_schema(self):
         # [START json_no_schema]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-pro-latest")
         prompt = """List a few popular cookie recipes in JSON format.
 
@@ -52,6 +55,8 @@ def test_json_no_schema(self):
 
     def test_json_enum(self):
         # [START json_enum]
+        import google.generativeai as genai
+
         import enum
 
         class Choice(enum.Enum):
@@ -75,6 +80,8 @@ class Choice(enum.Enum):
 
     def test_enum_in_json(self):
         # [START enum_in_json]
+        import google.generativeai as genai
+
         import enum
         from typing_extensions import TypedDict
 
@@ -103,6 +110,8 @@ class Recipe(TypedDict):
 
     def test_json_enum_raw(self):
         # [START json_enum_raw]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-pro-latest")
 
         organ = genai.upload_file(media / "organ.jpg")
@@ -121,6 +130,8 @@ def test_json_enum_raw(self):
 
     def test_x_enum(self):
         # [START x_enum]
+        import google.generativeai as genai
+
         import enum
 
         class Choice(enum.Enum):
@@ -144,6 +155,8 @@ class Choice(enum.Enum):
 
     def test_x_enum_raw(self):
         # [START x_enum_raw]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-pro-latest")
 
         organ = genai.upload_file(media / "organ.jpg")

diff --git a/samples/count_tokens.py b/samples/count_tokens.py
@@ -14,7 +14,6 @@
 # limitations under the License.
 from absl.testing import absltest
 
-import google.generativeai as genai
 import pathlib
 
 media = pathlib.Path(__file__).parents[1] / "third_party"
@@ -23,6 +22,8 @@
 class UnitTests(absltest.TestCase):
     def test_tokens_context_window(self):
         # [START tokens_context_window]
+        import google.generativeai as genai
+
         model_info = genai.get_model("models/gemini-1.5-flash")
 
         # Returns the "context window" for the model,
@@ -34,6 +35,8 @@ def test_tokens_context_window(self):
 
     def test_tokens_text_only(self):
         # [START tokens_text_only]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("models/gemini-1.5-flash")
 
         prompt = "The quick brown fox jumps over the lazy dog."
@@ -54,6 +57,8 @@ def test_tokens_text_only(self):
 
     def test_tokens_chat(self):
         # [START tokens_chat]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("models/gemini-1.5-flash")
 
         chat = model.start_chat(
@@ -86,6 +91,8 @@ def test_tokens_chat(self):
 
     def test_tokens_multimodal_image_inline(self):
         # [START tokens_multimodal_image_inline]
+        import google.generativeai as genai
+
         import PIL.Image
 
         model = genai.GenerativeModel("models/gemini-1.5-flash")
@@ -112,6 +119,8 @@ def test_tokens_multimodal_image_inline(self):
 
     def test_tokens_multimodal_image_file_api(self):
         # [START tokens_multimodal_image_file_api]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("models/gemini-1.5-flash")
 
         prompt = "Tell me about this image"
@@ -136,6 +145,8 @@ def test_tokens_multimodal_image_file_api(self):
 
     def test_tokens_multimodal_video_audio_file_api(self):
         # [START tokens_multimodal_video_audio_file_api]
+        import google.generativeai as genai
+
         import time
 
         model = genai.GenerativeModel("models/gemini-1.5-flash")
@@ -169,6 +180,8 @@ def test_tokens_multimodal_video_audio_file_api(self):
 
     def test_tokens_multimodal_pdf_file_api(self):
         # [START tokens_multimodal_pdf_file_api]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         sample_pdf = genai.upload_file(media / "test.pdf")
         token_count = model.count_tokens(["Give me a summary of this document.", sample_pdf])
@@ -180,6 +193,8 @@ def test_tokens_multimodal_pdf_file_api(self):
 
     def test_tokens_cached_content(self):
         # [START tokens_cached_content]
+        import google.generativeai as genai
+
         import time
 
         model = genai.GenerativeModel("models/gemini-1.5-flash")
@@ -220,6 +235,8 @@ def test_tokens_cached_content(self):
 
     def test_tokens_system_instruction(self):
         # [START tokens_system_instruction]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel(model_name="gemini-1.5-flash")
 
         prompt = "The quick brown fox jumps over the lazy dog."
@@ -239,6 +256,8 @@ def test_tokens_system_instruction(self):
 
     def test_tokens_tools(self):
         # [START tokens_tools]
+        import google.generativeai as genai
+
         model = genai.GenerativeModel(model_name="gemini-1.5-flash")
 
         prompt = "I have 57 cats, each owns 44 mittens, how many mittens is that in total?"

diff --git a/samples/embed.py b/samples/embed.py
@@ -15,12 +15,10 @@
 from absl.testing import absltest
 
 
-import google.generativeai as genai
-
-
 class UnitTests(absltest.TestCase):
     def test_embed_content(self):
         # [START embed_content]
+        import google.generativeai as genai
 
         text = "Hello World!"
         result = genai.embed_content(
@@ -31,6 +29,8 @@ def test_embed_content(self):
 
     def batch_embed_contents(self):
         # [START batch_embed_contents]
+        import google.generativeai as genai
+
         texts = [
             "What is the meaning of life?",
             "How much wood would a woodchuck chuck?",