test: add Gemini 2.0 tests (#1213)

* test: add Gemini 2.0 tests

* preview warnings tests

Author: GarrettWu

tests/system/small/ml/test_llm.py

@@ -267,6 +267,7 @@ def test_text_embedding_generator_multi_cols_predict_success(
         "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
+        "gemini-2.0-flash-exp",
     ),
 )
 def test_create_load_gemini_text_generator_model(
@@ -297,6 +298,7 @@ def test_create_load_gemini_text_generator_model(
         "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
+        "gemini-2.0-flash-exp",
     ),
 )
 @pytest.mark.flaky(retries=2)
@@ -322,6 +324,7 @@ def test_gemini_text_generator_predict_default_params_success(
         "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
+        "gemini-2.0-flash-exp",
     ),
 )
 @pytest.mark.flaky(retries=2)
@@ -349,6 +352,7 @@ def test_gemini_text_generator_predict_with_params_success(
         "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
+        "gemini-2.0-flash-exp",
     ),
 )
 @pytest.mark.flaky(retries=2)
@@ -469,3 +473,16 @@ def test_palm2_text_embedding_deprecated():
             llm.PaLM2TextEmbeddingGenerator()
         except (Exception):
             pass
+
+
+@pytest.mark.parametrize(
+    "model_name",
+    (
+        "gemini-1.5-pro-preview-0514",
+        "gemini-1.5-flash-preview-0514",
+        "gemini-2.0-flash-exp",
+    ),
+)
+def test_gemini_preview_model_warnings(model_name):
+    with pytest.warns(exceptions.PreviewWarning):
+        llm.GeminiTextGenerator(model_name=model_name)