diff --git a/google/generativeai/client.py b/google/generativeai/client.py index c9c5c8c5b..4e029ad43 100644 --- a/google/generativeai/client.py +++ b/google/generativeai/client.py @@ -12,6 +12,8 @@ from io import IOBase import google.ai.generativelanguage as glm +import google.ai.generativelanguage_v1alpha as glm_v1alpha + import google.generativeai.protos as protos from google.auth import credentials as ga_credentials @@ -226,6 +228,10 @@ def make_client(self, name): cls = FileServiceClient elif name == "file_async": cls = FileServiceAsyncClient + elif name == "generative_alpha": + cls = glm_alpha.GenerativeServiceClient + elif name == "generative_alpha_async": + cls = glm_alpha.GenerativeServiceAsyncClient elif name.endswith("_async"): name = name.split("_")[0] cls = getattr(glm, name.title() + "ServiceAsyncClient") @@ -355,7 +361,6 @@ def get_default_file_client() -> glm.FilesServiceClient: def get_default_file_async_client() -> glm.FilesServiceAsyncClient: return _client_manager.get_default_client("file_async") - def get_default_generative_client() -> glm.GenerativeServiceClient: return _client_manager.get_default_client("generative") @@ -363,6 +368,12 @@ def get_default_generative_client() -> glm.GenerativeServiceClient: def get_default_generative_async_client() -> glm.GenerativeServiceAsyncClient: return _client_manager.get_default_client("generative_async") +def get_default_generative_client() -> glm.GenerativeServiceClient: + return _client_manager.get_default_client("generative_alpha") + +def get_default_generative_async_client() -> glm.GenerativeServiceAsyncClient: + return _client_manager.get_default_client("generative_alpha_async") + def get_default_operations_client() -> operations_v1.OperationsClient: return _client_manager.get_default_client("operations") diff --git a/setup.py b/setup.py index c61393765..0e8a7fc79 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ def get_version(): release_status = "Development Status :: 5 - Production/Stable" dependencies = [ - "google-ai-generativelanguage==0.6.10", + "google-ai-generativelanguage@https://storage.googleapis.com/generativeai-downloads/preview/ai-generativelanguage-speech-py.tar.gz", "google-api-core", "google-api-python-client", "google-auth>=2.15.0", # 2.15 adds API key auth support