-
Notifications
You must be signed in to change notification settings - Fork 324
Centralize Vertex AI API Version Configuration #913
Copy link
Copy link
Open
Labels
enhancementNew feature or requestNew feature or requestgood first issueGood for newcomersGood for newcomerspythonPull requests that update python codePull requests that update python code
Description
Description:
Currently, the Vertex AI API version (e.g., v1beta1) is either hardcoded or relying on the SDK default across multiple model files (models/veo.py, models/image_models.py, etc.).
We should centralize this configuration to ensure consistency and make upgrades easier for users that want to explicitly set a version (v1beta1, v1, etc.).
Proposed Changes:
-
Update
config/default.py:
Add a new configuration variable to theDefaultclass:VERTEX_API_VERSION: str = os.environ.get("VERTEX_API_VERSION", "v1beta1")
-
Refactor Model Clients:
Update allgenai.Clientinstantiations to use this config value.client = genai.Client( # ... http_options={'api_version': config.VERTEX_API_VERSION} )
Affected Files:
config/default.pymodels/image_models.pymodels/veo.pymodels/vto.pymodels/upscale.pymodels/character_consistency.pymodels/model_setup.py
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or requestgood first issueGood for newcomersGood for newcomerspythonPull requests that update python codePull requests that update python code