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

fix linting errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-30 12:51:39 -07:00
parent 1e1bce4594
commit 66211b42db
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/openai.py
View file

@ -2609,7 +2609,7 @@ class OpenAIBatchesAPI(BaseLLM):
limit: Optional[int] = None, limit: Optional[int] = None,
): ):
verbose_logger.debug("listing batches, after= %s, limit= %s", after, limit) verbose_logger.debug("listing batches, after= %s, limit= %s", after, limit)
response = await openai_client.batches.list(after=after, limit=limit) response = await openai_client.batches.list(after=after, limit=limit) # type: ignore
return response return response
def list_batches( def list_batches(
@ -2646,7 +2646,7 @@ class OpenAIBatchesAPI(BaseLLM):
return self.alist_batches( # type: ignore return self.alist_batches( # type: ignore
openai_client=openai_client, after=after, limit=limit openai_client=openai_client, after=after, limit=limit
) )
response = openai_client.batches.list(after=after, limit=limit) response = openai_client.batches.list(after=after, limit=limit) # type: ignore
return response return response

4
litellm/proxy/proxy_server.py
View file

@ -5052,7 +5052,9 @@ async def list_batches(
return response return response
except Exception as e: except Exception as e:
await proxy_logging_obj.post_call_failure_hook( await proxy_logging_obj.post_call_failure_hook(
user_api_key_dict=user_api_key_dict, original_exception=e, request_data=data user_api_key_dict=user_api_key_dict,
original_exception=e,
request_data={"after": after, "limit": limit},
) )
verbose_proxy_logger.error( verbose_proxy_logger.error(
"litellm.proxy.proxy_server.retrieve_batch(): Exception occured - {}".format( "litellm.proxy.proxy_server.retrieve_batch(): Exception occured - {}".format(