Update Unions

keertk · Jun 2, 2023 · 4d2e710 · 4d2e710
1 parent 3d93a3e
commit 4d2e710
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/google/generativeai/client.py b/google/generativeai/client.py
@@ -36,15 +36,15 @@
 def configure(
     *,
     api_key: str | None = None,
-    credentials: Union[ga_credentials.Credentials, dict, None] = None,
+    credentials: ga_credentials.Credentials | dict | None = None,
     # The user can pass a string to choose `rest` or `grpc` or 'grpc_asyncio'.
     # See `_transport_registry` in `DiscussServiceClientMeta`.
     # Since the transport classes align with the client classes it wouldn't make
     # sense to accept a `Transport` object here even though the client classes can.
     # We could accept a dict since all the `Transport` classes take the same args,
     # but that seems rare. Users that need it can just switch to the low level API.
     transport: Union[str, None] = None,
-    client_options: Union[client_options_lib.ClientOptions, dict, None] = None,
+    client_options: client_options_lib.ClientOptions | dict | None = None,
     client_info: gapic_v1.client_info.ClientInfo | None = None,
 ):
     """Captures default client configuration.

diff --git a/google/generativeai/notebook/flag_def.py b/google/generativeai/notebook/flag_def.py
@@ -42,7 +42,7 @@
 
 # These are the intermediate types that argparse.ArgumentParser.parse_args()
 # will pass command line arguments into.
-_PARSETYPES = Union[str, int, float]
+_PARSETYPES = str | int | float
 # These are the final result types that the intermediate parsed values will be
 # converted into. It is a superset of _PARSETYPES because we support converting
 # the parsed type into a more precise type, e.g. from str to Enum.

diff --git a/google/generativeai/text.py b/google/generativeai/text.py
@@ -27,7 +27,7 @@
 from google.generativeai.types import safety_types
 
 
-def _make_text_prompt(prompt: Union[str, dict[str, str]]) -> glm.TextPrompt:
+def _make_text_prompt(prompt: str | dict[str, str]) -> glm.TextPrompt:
     if isinstance(prompt, str):
         return glm.TextPrompt(text=prompt)
     elif isinstance(prompt, dict):

diff --git a/google/generativeai/types/discuss_types.py b/google/generativeai/types/discuss_types.py
@@ -45,7 +45,7 @@ class MessageDict(TypedDict):
     citation_metadata: citation_types.CitationMetadataDict | None
 
 
-MessageOptions = Union[str, MessageDict, glm.Message]
+MessageOptions = str | MessageDict | glm.Message
 MESSAGE_OPTIONS = (str, dict, glm.Message)
 
 MessagesOptions = Union[