Restore ModelNameOptions for backwards compatibility. (#87)

google-gemini · Oct 18, 2023 · 2e9f23e · 2e9f23e
1 parent 2eb5e82
commit 2e9f23e
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
@@ -30,6 +30,7 @@
 
 __all__ = [
     "Model",
+    "ModelNameOptions",
     "AnyModelNameOptions",
     "BaseModelNameOptions",
     "TunedModelNameOptions",
@@ -310,6 +311,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):