Skip to content

Commit 7efa372

Browse files
CrowiantAnton Nitochkin
andauthored
Update gen ai batch api system test (#64964)
Co-authored-by: Anton Nitochkin <nitochkin@google.com>
1 parent a42955b commit 7efa372

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

providers/google/tests/system/google/cloud/gen_ai/example_gen_ai_gemini_batch_api.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,8 @@
7777
LOCAL_FILE_NAME = "gemini_batch_requests.jsonl"
7878
LOCAL_EMBEDDINGS_FILE_NAME = "gemini_batch_embeddings_requests.jsonl"
7979
UPLOAD_FILE_PATH = str(Path(__file__).parent / "resources" / LOCAL_FILE_NAME)
80-
PATH_TO_SAVE_RESULTS = str(Path(__file__).parent / "resources")
80+
IS_COMPOSER = bool(os.environ.get("COMPOSER_ENVIRONMENT", ""))
81+
PATH_TO_SAVE_RESULTS = "gcs/data" if IS_COMPOSER else str(Path(__file__).parent / "resources")
8182
UPLOAD_EMBEDDINGS_FILE_PATH = str(Path(__file__).parent / "resources" / LOCAL_EMBEDDINGS_FILE_NAME)
8283

8384
UPLOADED_FILE_NAME = (

0 commit comments

Comments
 (0)