forked from phoenix/litellm-mirror
(test) embedding hidden params
This commit is contained in:
parent
7f37d7e44f
commit
c1686b264f
1 changed files with 7 additions and 2 deletions
|
@ -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():
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue