Add Samples for Vertex AI RAG (GoogleCloudPlatform#11460)

holtskinner · gcf-owl-bot[bot] · web-flow · commit 28bb55e56240 · 2024-04-22T11:35:12.000-05:00
* Add Rag Sample * Fix import - init rag_test * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * Change import order * Update vertexai library version * Move rag tags * Add real tests for RAG * Change import for rag * Add lint ignore * Addressed comments and changed structure to follow new guidelines * Attempts at fixing tests * Add fixes for tests - One change is due to b/335864815 * Fixes to Async import function * Add comment to init vertex ai once per session * Update quickstart * Update pytest * Move general imports to fit with indentation * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * Add TODOs and Skip Tests --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
diff --git a/generative_ai/imagen/edit_image_inpainting_insert_mask.py b/generative_ai/imagen/edit_image_inpainting_insert_mask.py
@@ -27,7 +27,6 @@ def edit_image_inpainting_insert_mask(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_inpainting_insert_mask]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_inpainting_insert_mask_mode.py b/generative_ai/imagen/edit_image_inpainting_insert_mask_mode.py
@@ -28,7 +28,6 @@ def edit_image_inpainting_insert_mask_mode(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_inpainting_insert_mask_mode]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_inpainting_remove_mask.py b/generative_ai/imagen/edit_image_inpainting_remove_mask.py
@@ -26,7 +26,6 @@ def edit_image_inpainting_remove_mask(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_inpainting_remove_mask]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_inpainting_remove_mask_mode.py b/generative_ai/imagen/edit_image_inpainting_remove_mask_mode.py
@@ -28,7 +28,6 @@ def edit_image_inpainting_remove_mask_mode(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_inpainting_remove_mask_mode]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_mask.py b/generative_ai/imagen/edit_image_mask.py
@@ -26,7 +26,6 @@ def edit_image_mask(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_mask]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_mask_free.py b/generative_ai/imagen/edit_image_mask_free.py
@@ -22,7 +22,6 @@
 def edit_image_mask_free(
     project_id: str, input_file: str, output_file: str, prompt: str
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_mask_free]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_outpainting_mask.py b/generative_ai/imagen/edit_image_outpainting_mask.py
@@ -27,7 +27,6 @@ def edit_image_outpainting_mask(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_outpainting_mask]
 
     import vertexai
diff --git a/generative_ai/imagen/edit_image_product_image.py b/generative_ai/imagen/edit_image_product_image.py
@@ -25,7 +25,6 @@ def edit_image_product_image(
     output_file: str,
     prompt: str,
 ) -> vision_models.ImageGenerationResponse:
-
     # [START generativeaionvertexai_imagen_edit_image_product_image]
 
     import vertexai
diff --git a/generative_ai/imagen/get_short_form_image_captions.py b/generative_ai/imagen/get_short_form_image_captions.py
@@ -17,7 +17,6 @@
 
 
 def get_short_form_image_captions(project_id: str, input_file: str) -> list:
-
     # [START generativeaionvertexai_imagen_get_short_form_image_captions]
 
     import vertexai
diff --git a/generative_ai/imagen/get_short_form_image_responses.py b/generative_ai/imagen/get_short_form_image_responses.py
@@ -20,7 +20,6 @@
 def get_short_form_image_responses(
     project_id: str, input_file: str, question: str
 ) -> list:
-
     # [START generativeaionvertexai_imagen_get_short_form_image_responses]
 
     import vertexai
diff --git a/generative_ai/rag.py b/generative_ai/rag.py
diff --git a/generative_ai/rag_test.py b/generative_ai/rag_test.py
diff --git a/generative_ai/requirements-test.txt b/generative_ai/requirements-test.txt
diff --git a/generative_ai/requirements.txt b/generative_ai/requirements.txt
