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

(test) aembedding includes custom_llm_provider

Browse source
This commit is contained in:
ishaan-jaff 2024-01-12 17:52:14 -08:00
parent 39f724d9f3
commit 99dcce1e0f
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_embedding.py
View file

@ -319,6 +319,12 @@ def test_aembedding_azure():
input=["good morning from litellm", "this is another item"],
)
print(response)
print(
"hidden params - custom_llm_provider",
response._hidden_params["custom_llm_provider"],
)
assert response._hidden_params["custom_llm_provider"] == "azure"
except Exception as e:
pytest.fail(f"Error occurred: {e}")