"dgl_sparse/include/vscode:/vscode.git/clone" did not exist on "a03dec059e3dec13ce560745ee81acbac5500905"
Unverified Commit d7b1ce65 authored by yan97ao's avatar yan97ao Committed by GitHub
Browse files

Handle JSONDecodeError while processing request data (#5599)

parent f55933e1
...@@ -715,7 +715,10 @@ def v1_generate_response( ...@@ -715,7 +715,10 @@ def v1_generate_response(
async def v1_completions(tokenizer_manager, raw_request: Request): async def v1_completions(tokenizer_manager, raw_request: Request):
try:
request_json = await raw_request.json() request_json = await raw_request.json()
except Exception as e:
return create_error_response("Invalid request body, error: ", str(e))
all_requests = [CompletionRequest(**request_json)] all_requests = [CompletionRequest(**request_json)]
created = int(time.time()) created = int(time.time())
adapted_request, request = v1_generate_request(all_requests) adapted_request, request = v1_generate_request(all_requests)
...@@ -1379,7 +1382,10 @@ def v1_chat_generate_response( ...@@ -1379,7 +1382,10 @@ def v1_chat_generate_response(
async def v1_chat_completions( async def v1_chat_completions(
tokenizer_manager, raw_request: Request, cache_report=False tokenizer_manager, raw_request: Request, cache_report=False
): ):
try:
request_json = await raw_request.json() request_json = await raw_request.json()
except Exception as e:
return create_error_response("Invalid request body, error: ", str(e))
all_requests = [ChatCompletionRequest(**request_json)] all_requests = [ChatCompletionRequest(**request_json)]
created = int(time.time()) created = int(time.time())
adapted_request, request = v1_chat_generate_request(all_requests, tokenizer_manager) adapted_request, request = v1_chat_generate_request(all_requests, tokenizer_manager)
...@@ -1800,7 +1806,10 @@ def v1_embedding_response(ret, model_path, to_file=False): ...@@ -1800,7 +1806,10 @@ def v1_embedding_response(ret, model_path, to_file=False):
async def v1_embeddings(tokenizer_manager, raw_request: Request): async def v1_embeddings(tokenizer_manager, raw_request: Request):
try:
request_json = await raw_request.json() request_json = await raw_request.json()
except Exception as e:
return create_error_response("Invalid request body, error: ", str(e))
all_requests = [EmbeddingRequest(**request_json)] all_requests = [EmbeddingRequest(**request_json)]
adapted_request, request = v1_embedding_request(all_requests, tokenizer_manager) adapted_request, request = v1_embedding_request(all_requests, tokenizer_manager)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment