File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed
generative_ai/batch_predict Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -23,11 +23,11 @@ def batch_predict_gemini_createjob(
23
23
) -> BatchPredictionJob :
24
24
"""Perform batch text prediction using a Gemini AI model.
25
25
Args:
26
- input_uri (str): URI of the input file in BigQuery table or Google Cloud Storage.
27
- Example: "gs://[BUCKET]/[DATASET].jsonl" OR "bq://[PROJECT].[DATASET].[TABLE]"
26
+ input_uri (str): URI of the input file in Google Cloud Storage.
27
+ Example: "gs://[BUCKET]/[DATASET].jsonl"
28
28
29
- output_uri (str): URI of the output folder, in BigQuery table or Google Cloud Storage.
30
- Example: "gs://[BUCKET]/[OUTPUT].jsonl" OR "bq://[PROJECT].[DATASET].[TABLE]"
29
+ output_uri (str): URI of the output folder in Google Cloud Storage.
30
+ Example: "gs://[BUCKET]/[OUTPUT].jsonl"
31
31
Returns:
32
32
batch_prediction_job: The batch prediction job object containing details of the job.
33
33
"""
Original file line number Diff line number Diff line change 21
21
22
22
import batch_code_predict
23
23
import batch_text_predict
24
- import gemini_batch_predict
24
+ import gemini_batch_predict_gcs
25
25
26
26
27
27
INPUT_BUCKET = "cloud-samples-data"
@@ -74,10 +74,10 @@ def test_batch_code_predict(output_folder: pytest.fixture()) -> None:
74
74
assert OUTPUT_PATH in job .output_info .gcs_output_directory
75
75
76
76
77
- def test_batch_gemini_predict (output_folder : pytest .fixture ()) -> None :
77
+ def test_batch_gemini_predict_gcs (output_folder : pytest .fixture ()) -> None :
78
78
input_uri = f"gs://{ INPUT_BUCKET } /batch/prompt_for_batch_gemini_predict.jsonl"
79
79
job = _main_test (
80
- test_func = lambda : gemini_batch_predict .batch_predict_gemini_createjob (
80
+ test_func = lambda : gemini_batch_predict_gcs .batch_predict_gemini_createjob (
81
81
input_uri , output_folder
82
82
)
83
83
)
You can’t perform that action at this time.
0 commit comments