Skip to content

Commit bbe1baf

Browse files
Add test coverage for VercelProvider model naming
1 parent 54c09ea commit bbe1baf

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

tests/models/cassettes/test_model_names/test_known_model_names.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ interactions:
1515
response:
1616
headers:
1717
content-length:
18-
- '545'
18+
- '550'
1919
content-security-policy:
2020
- default-src 'none'; frame-ancestors 'none'
2121
content-type:
@@ -46,6 +46,7 @@ interactions:
4646
- text-to-text
4747
- model_id: claude-4-sonnet
4848
regions:
49+
- eu
4950
- us
5051
type:
5152
- text-to-text

tests/providers/test_vercel.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,3 +126,18 @@ def test_vercel_with_http_client():
126126
provider = VercelProvider(api_key='test-key', http_client=http_client)
127127
assert provider.client.api_key == 'test-key'
128128
assert str(provider.client.base_url) == 'https://ai-gateway.vercel.sh/v1/'
129+
130+
131+
def test_vercel_provider_invalid_model_name():
132+
provider = VercelProvider(api_key='api-key')
133+
134+
with pytest.raises(UserError, match="Model name must be in 'provider/model' format"):
135+
provider.model_profile('invalid-model-name')
136+
137+
138+
def test_vercel_provider_unknown_provider():
139+
provider = VercelProvider(api_key='api-key')
140+
141+
profile = provider.model_profile('unknown/gpt-4')
142+
assert profile is not None
143+
assert profile.json_schema_transformer == OpenAIJsonSchemaTransformer

0 commit comments

Comments
 (0)