We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a42955b commit 7efa372Copy full SHA for 7efa372
1 file changed
providers/google/tests/system/google/cloud/gen_ai/example_gen_ai_gemini_batch_api.py
@@ -77,7 +77,8 @@
77
LOCAL_FILE_NAME = "gemini_batch_requests.jsonl"
78
LOCAL_EMBEDDINGS_FILE_NAME = "gemini_batch_embeddings_requests.jsonl"
79
UPLOAD_FILE_PATH = str(Path(__file__).parent / "resources" / LOCAL_FILE_NAME)
80
-PATH_TO_SAVE_RESULTS = str(Path(__file__).parent / "resources")
+IS_COMPOSER = bool(os.environ.get("COMPOSER_ENVIRONMENT", ""))
81
+PATH_TO_SAVE_RESULTS = "gcs/data" if IS_COMPOSER else str(Path(__file__).parent / "resources")
82
UPLOAD_EMBEDDINGS_FILE_PATH = str(Path(__file__).parent / "resources" / LOCAL_EMBEDDINGS_FILE_NAME)
83
84
UPLOADED_FILE_NAME = (
0 commit comments