Add tuned_models samples (#409)

* Add tuned_models samples

Change-Id: I4b85a0179d2e7e4c112ed6250a7e602de57787c4

* format

Change-Id: I19e37673d0451fbb243f5041abf40a961dd03a2f

* fix tests and debug

Change-Id: I8a71ea707f1ec9229d7ea1aee8b8fac25835c296

* format

Change-Id: I0514cff86d043c088aa70e2f10f0815326ed3d09

* revert -q

Change-Id: I4f2dfabd14777182f01038d608c2c0b65f43d02f

Author: MarkDaoust

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/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"
+    }
+]