Restore ModelNameOptions for backwards compatibility.

google-gemini · Oct 17, 2023 · 1f51a37 · 1f51a37
1 parent 923d372
commit 1f51a37
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/google/generativeai/types/model_types.py b/google/generativeai/types/model_types.py
@@ -25,6 +25,7 @@
 
 __all__ = [
     "Model",
+    "ModelNameOptions",
     "AnyModelNameOptions",
     "BaseModelNameOptions",
     "TunedModelNameOptions",
@@ -232,6 +233,7 @@ class Hyperparameters:
 BaseModelNameOptions = Union[str, Model, glm.Model]
 TunedModelNameOptions = Union[str, TunedModel, glm.TunedModel]
 AnyModelNameOptions = Union[str, Model, glm.Model, TunedModel, glm.TunedModel]
+ModelNameOptions = AnyModelNameOptions
 
 
 def make_model_name(name: AnyModelNameOptions):