Skip to content

Commit d49ea3d

Browse files
committed
refactor: simplify request handler logic
Signed-off-by: googs1025 <[email protected]>
1 parent 6071e98 commit d49ea3d

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

vllm/entrypoints/openai/run_batch.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -365,8 +365,8 @@ async def main(args):
365365

366366
# Determine the type of request and run it.
367367
if request.url == "/v1/chat/completions":
368-
chat_handler_fn = (None if openai_serving_chat is None else
369-
openai_serving_chat.create_chat_completion)
368+
chat_handler_fn = openai_serving_chat.create_chat_completion if \
369+
openai_serving_chat is not None else None
370370
if chat_handler_fn is None:
371371
response_futures.append(
372372
make_async_error_request_output(
@@ -380,8 +380,8 @@ async def main(args):
380380
run_request(chat_handler_fn, request, tracker))
381381
tracker.submitted()
382382
elif request.url == "/v1/embeddings":
383-
embed_handler_fn = (None if openai_serving_embedding is None else
384-
openai_serving_embedding.create_embedding)
383+
embed_handler_fn = openai_serving_embedding.create_embedding if \
384+
openai_serving_embedding is not None else None
385385
if embed_handler_fn is None:
386386
response_futures.append(
387387
make_async_error_request_output(
@@ -394,8 +394,8 @@ async def main(args):
394394
run_request(embed_handler_fn, request, tracker))
395395
tracker.submitted()
396396
elif request.url == "/v1/score":
397-
score_handler_fn = (None if openai_serving_scores is None else
398-
openai_serving_scores.create_score)
397+
score_handler_fn = openai_serving_scores.create_score if \
398+
openai_serving_scores is not None else None
399399
if score_handler_fn is None:
400400
response_futures.append(
401401
make_async_error_request_output(

0 commit comments

Comments
 (0)