File tree Expand file tree Collapse file tree 2 files changed +17
-1
lines changed
models/cassettes/test_model_names Expand file tree Collapse file tree 2 files changed +17
-1
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ interactions:
15
15
response :
16
16
headers :
17
17
content-length :
18
- - ' 545 '
18
+ - ' 550 '
19
19
content-security-policy :
20
20
- default-src 'none'; frame-ancestors 'none'
21
21
content-type :
@@ -46,6 +46,7 @@ interactions:
46
46
- text-to-text
47
47
- model_id : claude-4-sonnet
48
48
regions :
49
+ - eu
49
50
- us
50
51
type :
51
52
- text-to-text
Original file line number Diff line number Diff line change @@ -126,3 +126,18 @@ def test_vercel_with_http_client():
126
126
provider = VercelProvider (api_key = 'test-key' , http_client = http_client )
127
127
assert provider .client .api_key == 'test-key'
128
128
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
You can’t perform that action at this time.
0 commit comments