feat: Update lUpdate GeminiTextGenerator Tuning and Support score() m…

…ethod in Gemini-pro-1.5 \n Bug: b/381936588 and b/344891364

bigframes/ml/llm.py

@@ -892,6 +892,7 @@ def fit(
         Returns:
             GeminiTextGenerator: Fitted estimator.
         """
+        # Support gemini-1.5 and gemini-pro
         supported_models = ["gemini-pro", "gemini-1.5-pro-002", "gemini-1.5-flash-002"]
         if self.model_name not in supported_models:
             raise NotImplementedError(

tests/system/load/test_llm.py

@@ -76,7 +76,6 @@ def test_llm_gemini_configure_fit(
         ],
         index=3,
     )
-    # TODO(ashleyxu b/335492787): After bqml rolled out version control: save, load, check parameters to ensure configuration was kept
 
 
 @pytest.mark.flaky(retries=2)

tests/system/small/ml/test_llm.py

@@ -413,6 +413,7 @@ def test_llm_palm_score_params(llm_fine_tune_df_default_index):
     )
 
 
+# test score() function for "gemini-pro" and "gemini-1.5" model
 @pytest.mark.parametrize(
     "model_name",
     (
@@ -421,7 +422,7 @@ def test_llm_palm_score_params(llm_fine_tune_df_default_index):
         "gemini-1.5-flash-002",
     ),
 )
-def test_llm_gemini_pro_score(model_name, llm_fine_tune_df_default_index):
+def test_llm_gemini_score(llm_fine_tune_df_default_index, model_name):
     model = llm.GeminiTextGenerator(model_name=model_name)
 
     # Check score to ensure the model was fitted