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

(test) embedding hidden params

Browse source
This commit is contained in:
ishaan-jaff 2024-01-12 17:35:33 -08:00
parent 7f37d7e44f
commit c1686b264f
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_embedding.py
View file

@ -177,11 +177,16 @@ def test_cohere_embedding3():
input=["good morning from litellm", "this is another item"], input=["good morning from litellm", "this is another item"],
) )
print(f"response:", response) print(f"response:", response)
custom_llm_provider = response._hidden_params["custom_llm_provider"]
assert custom_llm_provider == "cohere"
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_cohere_embedding3() test_cohere_embedding3()
def test_bedrock_embedding_titan(): def test_bedrock_embedding_titan():
@ -226,7 +231,7 @@ def test_bedrock_embedding_titan():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_bedrock_embedding_titan() # test_bedrock_embedding_titan()
def test_bedrock_embedding_cohere(): def test_bedrock_embedding_cohere():