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

Revert "changes"

Browse source 
This reverts commit fa88bc9632.
This commit is contained in:
Utkash Dubey 2025-03-10 14:46:21 -07:00
parent 1f17daf52c
commit c7a04140e9
19 changed files with 141 additions and 191 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/llm_translation/test_openai_o1.py
View file

@ -29,7 +29,8 @@ async def test_o1_handle_system_role(model):
from openai import AsyncOpenAI
from litellm.utils import supports_system_messages
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
litellm.set_verbose = True
@ -82,7 +83,8 @@ async def test_o1_handle_tool_calling_optional_params(
from litellm.utils import ProviderConfigManager
from litellm.types.utils import LlmProviders
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
config = ProviderConfigManager.get_provider_chat_config(
model=model, provider=LlmProviders.OPENAI
@ -188,7 +190,8 @@ class TestOpenAIO3(BaseOSeriesModelsTest, BaseLLMChatTest):
def test_o1_supports_vision():
"""Test that o1 supports vision"""
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
for k, v in litellm.model_cost.items():
if k.startswith("o1") and v.get("litellm_provider") == "openai":
assert v.get("supports_vision") is True, f"{k} does not support vision"