Skip to content

Add Samples for Vertex AI RAG #11460

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
kweinmeister merged 22 commits into from
Apr 22, 2024
Merged

Add Samples for Vertex AI RAG #11460

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
6ae4f26
Add Rag Sample
holtskinner Apr 2, 2024
c8ab44a
Fix import - init rag_test
holtskinner Apr 3, 2024
24eed1b
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Apr 16, 2024
e0206ea
Merge branch 'main' into rag
holtskinner Apr 16, 2024
cfdba65
Change import order
holtskinner Apr 16, 2024
dd6c022
Update vertexai library version
holtskinner Apr 16, 2024
e604214
Move rag tags
holtskinner Apr 16, 2024
f5a2396
Merge branch 'rag' of https://github.com/GoogleCloudPlatform/python-d…
holtskinner Apr 16, 2024
188efd6
Add real tests for RAG
holtskinner Apr 17, 2024
acbc287
Change import for rag
holtskinner Apr 17, 2024
f39d931
Add lint ignore
holtskinner Apr 17, 2024
d2656e4
Addressed comments and changed structure to follow new guidelines
holtskinner Apr 17, 2024
bb19ce0
Attempts at fixing tests
holtskinner Apr 17, 2024
1ead635
Add fixes for tests
holtskinner Apr 19, 2024
d0b6b2f
Fixes to Async import function
holtskinner Apr 19, 2024
2e62629
Add comment to init vertex ai once per session
holtskinner Apr 19, 2024
665bafb
Update quickstart
holtskinner Apr 19, 2024
c1203b5
Update pytest
holtskinner Apr 19, 2024
427cb41
Move general imports to fit with indentation
holtskinner Apr 19, 2024
113f190
Merge branch 'main' into rag
holtskinner Apr 22, 2024
a3d631a
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Apr 22, 2024
6e5d5db
Add TODOs and Skip Tests
holtskinner Apr 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_inpainting_insert_mask.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_inpainting_insert_mask_mode.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_inpainting_remove_mask.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_inpainting_remove_mask_mode.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_mask.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@ def edit_image_mask(
output_file: str,
prompt: str,
) -> vision_models.ImageGenerationResponse:

# [START generativeaionvertexai_imagen_edit_image_mask]

import vertexai
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_mask_free.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_outpainting_mask.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/edit_image_product_image.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/get_short_form_image_captions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion generative_ai/imagen/get_short_form_image_responses.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Loading