Skip to content

Commit 24ff152

Browse files
Valeriy-Burlakacode-review-assist[bot]
andauthored
Update genai/batch_prediction/submit_with_gcs.py
Co-authored-by: code-review-assist[bot] <182814678+code-review-assist[bot]@users.noreply.github.com>
1 parent 37142a7 commit 24ff152

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

genai/batch_prediction/submit_with_gcs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ def generate_content(output_uri: str) -> str:
2626
# See the documentation: https://googleapis.github.io/python-genai/genai.html#genai.batches.Batches.create
2727
job = client.batches.create(
2828
model="gemini-1.5-pro-002",
29-
src="gs://cloud-samples-data/batch/prompt_for_batch_gemini_predict.jsonl",
29+
27: model="gemini-1.5-pro-002",
30+
28: src="gs://cloud-samples-data/batch/prompt_for_batch_gemini_predict.jsonl", # Source link: https://storage.cloud.google.com/cloud-samples-data/batch/prompt_for_batch_gemini_predict.jsonl
3031
config={
3132
"dest": output_uri
3233
}

0 commit comments

Comments
 (0)