Commit graph

1 commit

Author SHA1 Message Date
Jonathan Chang
4bae7df72d
Add support for vertex_ai/ model name prefix format
This commit adds support for using the vertex_ai/ prefix with models in the format:
vertex_ai/claude-3-7-sonnet@20250219

The implementation:
1. Adds explicit handling for the vertex_ai/ prefix in get_llm_provider()
2. Correctly sets the custom_llm_provider to 'vertex_ai'
3. Returns the base model name without the prefix
4. Includes tests to verify the functionality

This allows for proper provider detection and cost calculation with Vertex AI hosted models.
2025-04-21 14:06:34 +08:00