-
Notifications
You must be signed in to change notification settings - Fork 256
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
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