scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix deepinfra test

Browse source
This commit is contained in:
Ishaan Jaff 2025-09-10 19:39:23 -07:00
parent 385206c4bc
commit 258b674dbb
3 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/model_prices_and_context_window_backup.json
View file

@ -15702,7 +15702,8 @@
"cache_read_input_token_cost": 2.16e-07,
"litellm_provider": "deepinfra",
"mode": "chat",
"supports_tool_choice": true
"supports_tool_choice": true,
"supports_reasoning": true
},
"deepinfra/google/gemini-2.0-flash-001": {
"max_tokens": 1000000,

3
model_prices_and_context_window.json
View file

@ -15702,7 +15702,8 @@
"cache_read_input_token_cost": 2.16e-07,
"litellm_provider": "deepinfra",
"mode": "chat",
"supports_tool_choice": true
"supports_tool_choice": true,
"supports_reasoning": true
},
"deepinfra/google/gemini-2.0-flash-001": {
"max_tokens": 1000000,

4
tests/test_litellm/llms/deepinfra/test_deepinfra_chat_transformation.py
View file

@ -17,6 +17,10 @@ def test_deepseek_supported_openai_params():
"""
from litellm.llms.deepinfra.chat.transformation import DeepInfraConfig
# Ensure we're using the local model cost map
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
supported_openai_params = DeepInfraConfig().get_supported_openai_params(model="deepinfra/deepseek-ai/DeepSeek-V3.1")
print(supported_openai_params)
assert "reasoning_effort" in supported_openai_params