add deprecation warning to TextEmbeddingGenerator model, espeically gemini-1.0-X and gemini-1.5-X

Author: shuoweil

bigframes/ml/llm.py

@@ -452,7 +452,7 @@ def to_gbq(self, model_name: str, replace: bool = False) -> PaLM2TextGenerator:
 
 
 @typing_extensions.deprecated(
-    "PaLM2TextEmbeddingGenerator has been deprecated. Use TextEmbeddingGenerator(https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.TextEmbeddingGenerator) instead. ",
+    "PaLM2TextEmbeddingGenerator has been deprecated, textembedding-gecko are going to be deprecated. Use TextEmbeddingGenerator(https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.TextEmbeddingGenerator) instead. ",
     category=exceptions.ApiDeprecationWarning,
 )
 @log_adapter.class_logger
@@ -918,10 +918,17 @@ def to_gbq(
         return new_model.session.read_gbq_model(model_name)
 
 
+@typing_extensions.deprecated(
+    "gemini-1.0-X and gemini-1.5-X are going to be deprecated. Use gemini-2.0-X (https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead. ",
+    category=exceptions.ApiDeprecationWarning,
+)
 @log_adapter.class_logger
 class GeminiTextGenerator(base.RetriableRemotePredictor):
     """Gemini text generator LLM model.
 
+    .. note::
+        gemini-1.0-X and gemini-1.5-X are going to be deprecated. Use GeminiTextGenerator(https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead.
+
     Args:
         model_name (str, Default to "gemini-pro"):
             The model for natural language tasks. Accepted values are "gemini-pro", "gemini-1.5-pro-preview-0514", "gemini-1.5-flash-preview-0514", "gemini-1.5-pro-001", "gemini-1.5-pro-002", "gemini-1.5-flash-001", "gemini-1.5-flash-002" and "gemini-2.0-flash-exp". Default to "gemini-pro".

tests/system/small/ml/test_llm.py

@@ -1022,6 +1022,21 @@ def test_palm2_text_embedding_deprecated():
             pass
 
 
+@pytest.mark.parametrize(
+    "model_name",
+    (
+        "gemini-1.5-pro-001",
+        "gemini-1.5-pro-002",
+        "gemini-1.5-flash-001",
+        "gemini-1.5-flash-002",
+        "gemini-1.0-pro",
+    ),
+)
+def test_gemini_text_generator_deprecated(model_name):
+    with pytest.warns(exceptions.ApiDeprecationWarning):
+        llm.GeminiTextGenerator(model_name=model_name)
+
+
 @pytest.mark.parametrize(
     "model_name",
     (