Add tuned_models samples

google/generativeai/types/__init__.py

@@ -21,6 +21,7 @@
 from google.generativeai.types.generation_types import *
 from google.generativeai.types.helper_types import *
 from google.generativeai.types.model_types import *
+from google.generativeai.types.permission_types import *
 from google.generativeai.types.safety_types import *
 from google.generativeai.types.text_types import *
 

google/generativeai/types/permission_types.py

@@ -28,6 +28,7 @@
 from google.generativeai.utils import flatten_update_paths
 from google.generativeai import string_utils
 
+__all__ = ["Permission", "Permissions"]
 
 GranteeType = protos.Permission.GranteeType
 Role = protos.Permission.Role
@@ -89,7 +90,7 @@ def valid_id(name: str) -> bool:
 
 
 @string_utils.prettyprint
-@dataclasses.dataclass
+@dataclasses.dataclass(init=False)
 class Permission:
     """
     A permission to access a resource.
@@ -100,6 +101,24 @@ class Permission:
     grantee_type: Optional[GranteeType]
     email_address: Optional[str] = None
 
+    def __init__(
+        self,
+        name: str,
+        role: RoleOptions,
+        grantee_type: Optional[GranteeTypeOptions] = None,
+        email_address: Optional[str] = None,
+    ):
+        self.name = name
+        if role is None:
+            self.role = None
+        else:
+            self.role = to_role(role)
+        if grantee_type is None:
+            self.grantee_type = None
+        else:
+            self.grantee_type = to_grantee_type(grantee_type)
+        self.email_address = email_address
+
     def delete(
         self,
         client: glm.PermissionServiceClient | None = None,
@@ -279,6 +298,12 @@ def _make_create_permission_request(
                 f"Invalid operation: An 'email_address' must be provided when 'grantee_type' is not set to 'EVERYONE'. Currently, 'grantee_type' is set to '{grantee_type}' and 'email_address' is '{email_address if email_address else 'not provided'}'."
             )
 
+        if email_address and grantee_type is None:
+            if email_address.endswith("googlegroups.com"):
+                grantee_type = GranteeType.GROUP
+            else:
+                grantee_type = GranteeType.USER
+
         permission = protos.Permission(
             role=role,
             grantee_type=grantee_type,
@@ -367,6 +392,9 @@ def list(
             permission = type(permission).to_dict(permission)
             yield Permission(**permission)
 
+    def __iter__(self):
+        return self.list()
+
     async def list_async(
         self,
         page_size: Optional[int] = None,
@@ -385,6 +413,35 @@ async def list_async(
             permission = type(permission).to_dict(permission)
             yield Permission(**permission)
 
+    async def __aiter__(self):
+        return self.list_async()
+
+    @classmethod
+    def get(cls, name: str) -> Permission:
+        """
+        Get information about a specific permission.
+
+        Args:
+            name: The name of the permission to get.
+
+        Returns:
+            Requested permission as an instance of `Permission`.
+        """
+        return Permission.get(name)
+
+    @classmethod
+    async def get_async(cls, name: str) -> Permission:
+        """
+        Get information about a specific permission.
+
+        Args:
+            name: The name of the permission to get.
+
+        Returns:
+            Requested permission as an instance of `Permission`.
+        """
+        return await Permission.get_async(name)
+
     def transfer_ownership(
         self,
         email_address: str,

samples/count_tokens.py

@@ -20,9 +20,6 @@
 media = pathlib.Path(__file__).parents[1] / "third_party"
 
 
-
-
-
 class UnitTests(absltest.TestCase):
     def test_tokens_text_only(self):
         # [START tokens_text_only]
@@ -84,8 +81,10 @@ def test_tokens_cached_content(self):
     def test_tokens_system_instruction(self):
         # [START tokens_system_instruction]
         document = genai.upload_file(path=media / "a11.txt")
-        model = genai.GenerativeModel("models/gemini-1.5-flash-001",
-                                      system_instruction="You are an expert analyzing transcripts. Give a summary of this document.")
+        model = genai.GenerativeModel(
+            "models/gemini-1.5-flash-001",
+            system_instruction="You are an expert analyzing transcripts. Give a summary of this document.",
+        )
         print(model.count_tokens(document))
         # [END tokens_system_instruction]
 
@@ -95,25 +94,27 @@ def add(a: float, b: float):
             """returns a + b."""
             return a + b
 
-
         def subtract(a: float, b: float):
             """returns a - b."""
             return a - b
 
-
         def multiply(a: float, b: float):
             """returns a * b."""
             return a * b
 
-
         def divide(a: float, b: float):
             """returns a / b."""
             return a / b
-
-        model = genai.GenerativeModel("models/gemini-1.5-flash-001",
-                                      tools=[add, subtract, multiply, divide])
-
-        print(model.count_tokens("I have 57 cats, each owns 44 mittens, how many mittens is that in total?"))
+
+        model = genai.GenerativeModel(
+            "models/gemini-1.5-flash-001", tools=[add, subtract, multiply, divide]
+        )
+
+        print(
+            model.count_tokens(
+                "I have 57 cats, each owns 44 mittens, how many mittens is that in total?"
+            )
+        )
         # [END tokens_tools]
 
 

samples/increment_tuning_data.json

@@ -0,0 +1,62 @@
+[
+    {
+        "text_input": "1",
+        "output": "2"
+    },
+    {
+        "text_input": "3",
+        "output": "4"
+    },
+    {
+        "text_input": "-3",
+        "output": "-2"
+    },
+    {
+        "text_input": "twenty two",
+        "output": "twenty three"
+    },
+    {
+        "text_input": "two hundred",
+        "output": "two hundred one"
+    },
+    {
+        "text_input": "ninety nine",
+        "output": "one hundred"
+    },
+    {
+        "text_input": "8",
+        "output": "9"
+    },
+    {
+        "text_input": "-98",
+        "output": "-97"
+    },
+    {
+        "text_input": "1,000",
+        "output": "1,001"
+    },
+    {
+        "text_input": "10,100,000",
+        "output": "10,100,001"
+    },
+    {
+        "text_input": "thirteen",
+        "output": "fourteen"
+    },
+    {
+        "text_input": "eighty",
+        "output": "eighty one"
+    },
+    {
+        "text_input": "one",
+        "output": "two"
+    },
+    {
+        "text_input": "three",
+        "output": "four"
+    },
+    {
+        "text_input": "seven",
+        "output": "eight"
+    }
+]

samples/text_generation.py

@@ -41,6 +41,7 @@ def test_text_gen_text_only_prompt_streaming(self):
     def test_text_gen_multimodal_one_image_prompt(self):
         # [START text_gen_multimodal_one_image_prompt]
         import PIL
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         organ = PIL.Image.open(media / "organ.jpg")
         response = model.generate_content(["Tell me about this instrument", organ])
@@ -50,6 +51,7 @@ def test_text_gen_multimodal_one_image_prompt(self):
     def test_text_gen_multimodal_one_image_prompt_streaming(self):
         # [START text_gen_multimodal_one_image_prompt_streaming]
         import PIL
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         organ = PIL.Image.open(media / "organ.jpg")
         response = model.generate_content(["Tell me about this instrument", organ], stream=True)
@@ -61,6 +63,7 @@ def test_text_gen_multimodal_one_image_prompt_streaming(self):
     def test_text_gen_multimodal_multi_image_prompt(self):
         # [START text_gen_multimodal_multi_image_prompt]
         import PIL
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         organ = PIL.Image.open(media / "organ.jpg")
         cajun_instrument = PIL.Image.open(media / "Cajun_instruments.jpg")
@@ -73,6 +76,7 @@ def test_text_gen_multimodal_multi_image_prompt(self):
     def test_text_gen_multimodal_multi_image_prompt_streaming(self):
         # [START text_gen_multimodal_multi_image_prompt_streaming]
         import PIL
+
         model = genai.GenerativeModel("gemini-1.5-flash")
         organ = PIL.Image.open(media / "organ.jpg")
         cajun_instrument = PIL.Image.open(media / "Cajun_instruments.jpg")