mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
Add test for gpt-5-search-api config
This commit is contained in:
parent
15d8367bd0
commit
fedd58462b
1 changed files with 23 additions and 0 deletions
23
tests/test_model_config.py
Normal file
23
tests/test_model_config.py
Normal file
|
|
@ -0,0 +1,23 @@
|
|||
import pytest
|
||||
import json
|
||||
|
||||
def test_gpt_5_search_api_config():
|
||||
"""Test that gpt-5-search-api model is properly configured"""
|
||||
with open('model_prices_and_context_window.json', 'r') as f:
|
||||
config = json.load(f)
|
||||
|
||||
model = config.get('gpt-5-search-api')
|
||||
|
||||
# Check model exists
|
||||
assert model is not None, "gpt-5-search-api not found in config"
|
||||
|
||||
# Check required fields
|
||||
assert model['litellm_provider'] == 'openai'
|
||||
assert model['input_cost_per_token'] == 1.25e-06
|
||||
assert model['output_cost_per_token'] == 1e-05
|
||||
assert model['max_input_tokens'] == 400000
|
||||
assert model['max_output_tokens'] == 128000
|
||||
assert model['mode'] == 'chat'
|
||||
assert model['supports_function_calling'] == True
|
||||
|
||||
print("✅ gpt-5-search-api config is valid!")
|
||||
Loading…
Add table
Reference in a new issue