phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test_completion_codestral_fim_api_stream

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-16 20:02:15 -08:00
parent 842bfc4ebf
commit 401531a8c9
1 changed files with 0 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_text_completion.py
View file

@ -4141,7 +4141,6 @@ async def test_completion_codestral_fim_api(model):
print(response)
assert response.choices[0].text is not None
assert len(response.choices[0].text) > 0
# cost = litellm.completion_cost(completion_response=response)
# print("cost to make mistral completion=", cost)
@ -4188,9 +4187,6 @@ async def test_completion_codestral_fim_api_stream(model):
full_response += chunk.get("choices")[0].get("text") or ""
print("full_response", full_response)
assert len(full_response) > 2 # we at least have a few chars in response :)
# cost = litellm.completion_cost(completion_response=response)
# print("cost to make mistral completion=", cost)
# assert cost > 0.0