feat: from vertexai.batch_prediction import BatchPredictionJob (#12736)

* feat: from vertexai.batch_prediction import BatchPredictionJob

* feat: from vertexai.batch_prediction import BatchPredictionJob

Author: lee1premium-2024

generative_ai/batch_predict/gemini_batch_predict_gcs.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import os
 
-from vertexai.preview.batch_prediction import BatchPredictionJob
+from vertexai.batch_prediction import BatchPredictionJob
 
 PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
 
@@ -36,7 +36,7 @@ def batch_predict_gemini_createjob(
     import time
     import vertexai
 
-    from vertexai.preview.batch_prediction import BatchPredictionJob
+    from vertexai.batch_prediction import BatchPredictionJob
 
     # TODO(developer): Update and un-comment below lines
     # input_uri ="gs://[BUCKET]/[OUTPUT].jsonl" # Example
@@ -77,7 +77,6 @@ def batch_predict_gemini_createjob(
     # https://storage.googleapis.com/cloud-samples-data/batch/prompt_for_batch_gemini_predict.jsonl
 
     return batch_prediction_job
-
     # [END generativeaionvertexai_batch_predict_gemini_createjob]
 
 

generative_ai/batch_predict/requirements.txt

@@ -3,7 +3,7 @@ pandas==2.0.3; python_version == '3.8'
 pandas==2.1.4; python_version > '3.8'
 pillow==10.3.0; python_version < '3.8'
 pillow==10.3.0; python_version >= '3.8'
-google-cloud-aiplatform[all]==1.69.0
+google-cloud-aiplatform[all]==1.71.0
 sentencepiece==0.2.0
 google-auth==2.29.0
 anthropic[vertex]==0.28.0