scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-24 00:52:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 91

test(llms): annotate local_model_cost_map fixtures in phase 6 tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yuneng 2026-09-20 07:48:07 +00:00
parent e1d2789d29
commit 4048062e53
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/unit/llms/chatgpt/responses/test_chatgpt_responses_transformation.py
View file

@ -5,6 +5,7 @@ Source: litellm/llms/chatgpt/responses/transformation.py
"""
import json
from collections.abc import Generator
from unittest.mock import MagicMock, patch
import httpx
@ -20,7 +21,7 @@ from litellm.utils import ProviderConfigManager
@pytest.fixture
def local_model_cost_map(monkeypatch):
def local_model_cost_map(monkeypatch: pytest.MonkeyPatch) -> Generator[None, None, None]:
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
monkeypatch.setattr(litellm, "model_cost", litellm.get_model_cost_map(url=""))
litellm.get_model_info.cache_clear()

3
tests/unit/llms/deepseek/test_deepseek_cost_calculator.py
View file

@ -1,3 +1,4 @@
from collections.abc import Generator
from datetime import datetime, timezone
from typing import Final
@ -9,7 +10,7 @@ from litellm.types.utils import ModelResponse, PromptTokensDetailsWrapper, Usage
@pytest.fixture
def local_model_cost_map(monkeypatch):
def local_model_cost_map(monkeypatch: pytest.MonkeyPatch) -> Generator[None, None, None]:
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
monkeypatch.setattr(litellm, "model_cost", litellm.get_model_cost_map(url=""))
litellm.get_model_info.cache_clear()