Restore ModelNameOptions for backwards compatibility. (google-gemini#87)

markmcd · Oct 30, 2023 · 6cb0d53 · 6cb0d53
1 parent f6e425a
commit 6cb0d53
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):