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

test fix - azure deprecated azure ai mistral

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-15 21:08:48 -07:00
parent 9b77559ccf
commit 65f8015221
1 changed files with 0 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/local_testing/test_completion_cost.py
View file

@ -864,29 +864,6 @@ def test_vertex_ai_embedding_completion_cost(caplog):
# assert False # assert False
def test_completion_azure_ai():
try:
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
litellm.set_verbose = True
response = litellm.completion(
model="azure_ai/Mistral-large-nmefg",
messages=[{"content": "what llm are you", "role": "user"}],
max_tokens=15,
num_retries=3,
api_base=os.getenv("AZURE_AI_MISTRAL_API_BASE"),
api_key=os.getenv("AZURE_AI_MISTRAL_API_KEY"),
)
print(response)
assert "response_cost" in response._hidden_params
assert isinstance(response._hidden_params["response_cost"], float)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
@pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_cost_hidden_params(sync_mode): async def test_completion_cost_hidden_params(sync_mode):