scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(cost-map): type the monkeypatch fixture in cerebras and inception registry tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-09-11 19:50:27 +00:00
parent 4ffd4ecc83
commit e3130a87bc
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/llms/cerebras/test_cerebras_chat_transformation.py
View file

@ -1,3 +1,5 @@
import pytest
import litellm
from litellm.llms.cerebras.chat import CerebrasConfig
@ -62,7 +64,7 @@ def test_map_openai_params_preserves_max_retries_zero_falsy() -> None:
)
def test_qwen_3_8_27b_cost_and_tokens(monkeypatch) -> None:
def test_qwen_3_8_27b_cost_and_tokens(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
monkeypatch.setattr(litellm, "model_cost", litellm.get_model_cost_map(url=""))
model = "cerebras/qwen-3.8-27b"

3
tests/test_litellm/llms/inception/test_inception_chat_transformation.py
View file

@ -7,6 +7,7 @@ import os
from unittest import mock
import httpx
import pytest
import litellm
from litellm.llms.inception.chat.transformation import InceptionChatConfig
@ -307,7 +308,7 @@ def test_inception_completion_targets_inception_endpoint():
assert response.choices[0].message.content == "hi"
def test_inception_mercury_2_5_cost_and_tokens(monkeypatch):
def test_inception_mercury_2_5_cost_and_tokens(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
monkeypatch.setattr(litellm, "model_cost", litellm.get_model_cost_map(url=""))
model = "inception/mercury-2.5"