feat(router.py): support region routing for bedrock, vertex ai, watsonx

This commit is contained in:
Krrish Dholakia 2024-05-11 11:04:00 -07:00
parent ebc927f1c8
commit 6714854bb7
7 changed files with 187 additions and 12 deletions

View file

@ -198,6 +198,23 @@ class VertexAIConfig:
optional_params[mapped_params[param]] = value
return optional_params
def get_eu_regions(self) -> List[str]:
"""
Source: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions
"""
return [
"europe-central2",
"europe-north1",
"europe-southwest1",
"europe-west1",
"europe-west2",
"europe-west3",
"europe-west4",
"europe-west6",
"europe-west8",
"europe-west9",
]
import asyncio