Skip to content

Commit 47be102

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
chore: GenAI SDK client - Update the Android prompt optimization image uri to the new location.
PiperOrigin-RevId: 844955733
1 parent 8a00d43 commit 47be102

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vertexai/_genai/prompt_optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -429,7 +429,7 @@ def optimize(
429429

430430
_OPTIMIZER_METHOD_TO_CONTAINER_URI = {
431431
types.PromptOptimizerMethod.VAPO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_v1_0",
432-
types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/android-apo:preview_v1_0",
432+
types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_android_v1_0",
433433
}
434434
container_uri = _OPTIMIZER_METHOD_TO_CONTAINER_URI.get(method)
435435
if not container_uri:
@@ -809,7 +809,7 @@ async def optimize(
809809

810810
_OPTIMIZER_METHOD_TO_CONTAINER_URI = {
811811
types.PromptOptimizerMethod.VAPO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_v1_0",
812-
types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/android-apo:preview_v1_0",
812+
types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_android_v1_0",
813813
}
814814
container_uri = _OPTIMIZER_METHOD_TO_CONTAINER_URI.get(method)
815815
if not container_uri:

0 commit comments

Comments
 (0)