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

fix /v1/batches POST

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-26 18:06:00 -07:00
parent c0717133a9
commit 159a880dcc
1 changed files with 22 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

28
litellm/proxy/proxy_server.py
View file

@ -4808,10 +4808,18 @@ async def create_batch(
""" """
global proxy_logging_obj global proxy_logging_obj
data: Dict = {} data: Dict = {}
try: try:
# Use orjson to parse JSON data, orjson speeds up requests significantly body = await request.body()
form_data = await request.form() body_str = body.decode()
data = {key: value for key, value in form_data.items() if key != "file"} try:
data = ast.literal_eval(body_str)
except:
data = json.loads(body_str)
verbose_proxy_logger.debug(
"Request received by LiteLLM:\n{}".format(json.dumps(data, indent=4)),
)
# Include original request and headers in the data # Include original request and headers in the data
data = await add_litellm_data_to_request( data = await add_litellm_data_to_request(
@ -4915,10 +4923,18 @@ async def retrieve_batch(
""" """
global proxy_logging_obj global proxy_logging_obj
data: Dict = {} data: Dict = {}
data = {}
try: try:
# Use orjson to parse JSON data, orjson speeds up requests significantly body = await request.body()
form_data = await request.form() body_str = body.decode()
data = {key: value for key, value in form_data.items() if key != "file"} try:
data = ast.literal_eval(body_str)
except:
data = json.loads(body_str)
verbose_proxy_logger.debug(
"Request received by LiteLLM:\n{}".format(json.dumps(data, indent=4)),
)
# Include original request and headers in the data # Include original request and headers in the data
data = await add_litellm_data_to_request( data = await add_litellm_data_to_request(