phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(o_series_handler.py): handle async calls

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-11 21:22:13 -07:00
parent 3d292fc95e
commit 317ba18f4f
2 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/llm_translation/base_llm_unit_tests.py
View file

@ -868,10 +868,13 @@ class BaseLLMChatTest(ABC):
except Exception as e:
pytest.fail(f"Error occurred: {e}")
@pytest.mark.flaky(retries=3, delay=1)
@pytest.mark.asyncio
async def test_completion_cost(self):
from litellm import completion_cost
litellm._turn_on_debug()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")