test: add gemini-2.0-flash-lite-011 test (#1729)

Author: shuoweil

bigframes/ml/llm.py

@@ -529,7 +529,8 @@ def fit(
         y: utils.ArrayType,
     ) -> GeminiTextGenerator:
         """Fine tune GeminiTextGenerator model. Only support "gemini-1.5-pro-002",
-           and "gemini-1.5-flash-002" models for now.
+           "gemini-1.5-flash-002", "gemini-2.0-flash-001",
+           and "gemini-2.0-flash-lite-001"models for now.
 
         .. note::
 
@@ -549,7 +550,7 @@ def fit(
         """
         if self.model_name not in _GEMINI_FINE_TUNE_SCORE_ENDPOINTS:
             msg = exceptions.format_message(
-                "fit() only supports gemini-1.5-pro-002, or gemini-1.5-flash-002 model."
+                "fit() only supports gemini-1.5-pro-002, gemini-1.5-flash-002, gemini-2.0-flash-001, or gemini-2.0-flash-lite-001 model."
             )
             warnings.warn(msg)
 

tests/system/load/test_llm.py

@@ -44,6 +44,7 @@ def llm_remote_text_df(session, llm_remote_text_pandas_df):
         "gemini-1.5-pro-002",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-001",
+        "gemini-2.0-flash-lite-001",
     ),
 )
 def test_llm_gemini_configure_fit(