mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-09 22:31:41 +00:00
test_embedding_caching_azure
This commit is contained in:
parent
2ba34d22bd
commit
973d4b71e7
1 changed files with 2 additions and 2 deletions
|
|
@ -527,7 +527,7 @@ def test_embedding_caching_azure():
|
|||
print(api_key)
|
||||
print(api_base)
|
||||
embedding1 = embedding(
|
||||
model="text-embedding-ada-002",
|
||||
model="azure/text-embedding-ada-002",
|
||||
input=["good morning from litellm", "this is another item"],
|
||||
api_key=api_key,
|
||||
api_base=api_base,
|
||||
|
|
@ -540,7 +540,7 @@ def test_embedding_caching_azure():
|
|||
time.sleep(1)
|
||||
start_time = time.time()
|
||||
embedding2 = embedding(
|
||||
model="text-embedding-ada-002",
|
||||
model="azure/text-embedding-ada-002",
|
||||
input=["good morning from litellm", "this is another item"],
|
||||
api_key=api_key,
|
||||
api_base=api_base,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue