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

(feat) completion_cost - embeddings + raise Exception

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 13:11:23 +05:30
parent db50a07318
commit f681f0f2b2
3 changed files with 43 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_embedding.py
View file

@ -59,6 +59,7 @@ def test_openai_embedding():
def test_openai_azure_embedding_simple():
try:
litellm.set_verbose = True
response = embedding(
model="azure/azure-embedding-model",
input=["good morning from litellm"],
@ -70,11 +71,15 @@ def test_openai_azure_embedding_simple():
response_keys
) # assert litellm response has expected keys from OpenAI embedding response
request_cost = litellm.completion_cost(completion_response=response)
print("Calculated request cost=", request_cost)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_openai_azure_embedding_simple()
test_openai_azure_embedding_simple()
def test_openai_azure_embedding_timeouts():