phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): bug fixes

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-11 19:02:17 +05:30
parent 0440168915
commit f8d1720690
3 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_caching.py
View file

@ -290,6 +290,7 @@ async def test_embedding_caching_azure_individual_items():
embedding_val_2 = await aembedding( embedding_val_2 = await aembedding(
model="azure/azure-embedding-model", input=embedding_2, caching=True model="azure/azure-embedding-model", input=embedding_2, caching=True
) )
print(f"embedding_val_2: {embedding_val_2}")
if ( if (
embedding_val_2["data"][0]["embedding"] embedding_val_2["data"][0]["embedding"]
!= embedding_val_1["data"][0]["embedding"] != embedding_val_1["data"][0]["embedding"]

2
litellm/tests/test_custom_callback_input.py
View file

@ -673,7 +673,7 @@ async def test_async_embedding_bedrock():
response = await litellm.aembedding( response = await litellm.aembedding(
model="bedrock/cohere.embed-multilingual-v3", model="bedrock/cohere.embed-multilingual-v3",
input=["good morning from litellm"], input=["good morning from litellm"],
aws_region_name="os.environ/AWS_REGION_NAME_2", aws_region_name="us-east-1",
) )
await asyncio.sleep(1) await asyncio.sleep(1)
print(f"customHandler_success.errors: {customHandler_success.errors}") print(f"customHandler_success.errors: {customHandler_success.errors}")

4
litellm/utils.py
View file

@ -2341,7 +2341,7 @@ def client(original_function):
for val in non_null_list: for val in non_null_list:
idx, cr = val # (idx, cr) tuple idx, cr = val # (idx, cr) tuple
if cr is not None: if cr is not None:
final_embedding_cached_response.data[idx] = val final_embedding_cached_response.data[idx] = cr
if len(remaining_list) == 0: if len(remaining_list) == 0:
# LOG SUCCESS # LOG SUCCESS
@ -2485,9 +2485,9 @@ def client(original_function):
for item in final_embedding_cached_response.data: for item in final_embedding_cached_response.data:
if item is None: if item is None:
final_data_list.append(result.data[idx]) final_data_list.append(result.data[idx])
idx += 1
else: else:
final_data_list.append(item) final_data_list.append(item)
idx += 1
final_embedding_cached_response.data = final_data_list final_embedding_cached_response.data = final_data_list
return final_embedding_cached_response return final_embedding_cached_response