refactor test: pass the output_uri as a parameter, rather than environ

Author: glasnt

generative_ai/embeddings/batch_example.py

@@ -16,10 +16,9 @@
 from google.cloud.aiplatform import BatchPredictionJob
 
 PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
-OUTPUT_URI = os.getenv("GCS_OUTPUT_URI")
 
 
-def embed_text_batch() -> BatchPredictionJob:
+def embed_text_batch(OUTPUT_URI) -> BatchPredictionJob:
     """Example of how to generate embeddings from text using batch processing.
 
     Read more: https://cloud.google.com/vertex-ai/generative-ai/docs/embeddings/batch-prediction-genai-embeddings

generative_ai/embeddings/test_embeddings_examples.py

@@ -35,8 +35,7 @@
 
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=10)
 def test_embed_text_batch() -> None:
-    os.environ["GCS_OUTPUT_URI"] = "gs://python-docs-samples-tests/"
-    batch_prediction_job = batch_example.embed_text_batch()
+    batch_prediction_job = batch_example.embed_text_batch("gs://python-docs-samples-tests/")
     assert batch_prediction_job