[Fix]HTTP Stream raise exception (#11904)
This commit is contained in:
@@ -996,7 +996,11 @@ class TokenizerManager(TokenizerCommunicatorMixin):
|
|||||||
finish_reason.get("type") == "abort"
|
finish_reason.get("type") == "abort"
|
||||||
and finish_reason.get("status_code") == HTTPStatus.BAD_REQUEST
|
and finish_reason.get("status_code") == HTTPStatus.BAD_REQUEST
|
||||||
):
|
):
|
||||||
|
if not obj.stream:
|
||||||
raise ValueError(finish_reason["message"])
|
raise ValueError(finish_reason["message"])
|
||||||
|
else:
|
||||||
|
yield out
|
||||||
|
break
|
||||||
|
|
||||||
if finish_reason.get("type") == "abort" and finish_reason.get(
|
if finish_reason.get("type") == "abort" and finish_reason.get(
|
||||||
"status_code"
|
"status_code"
|
||||||
@@ -1013,11 +1017,14 @@ class TokenizerManager(TokenizerCommunicatorMixin):
|
|||||||
# Mark ongoing LoRA request as finished.
|
# Mark ongoing LoRA request as finished.
|
||||||
if self.server_args.enable_lora and state.obj.lora_path:
|
if self.server_args.enable_lora and state.obj.lora_path:
|
||||||
await self.lora_registry.release(state.obj.lora_id)
|
await self.lora_registry.release(state.obj.lora_id)
|
||||||
|
if not obj.stream:
|
||||||
raise fastapi.HTTPException(
|
raise fastapi.HTTPException(
|
||||||
status_code=finish_reason["status_code"],
|
status_code=finish_reason["status_code"],
|
||||||
detail=finish_reason["message"],
|
detail=finish_reason["message"],
|
||||||
)
|
)
|
||||||
|
else:
|
||||||
|
yield out
|
||||||
|
break
|
||||||
yield out
|
yield out
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user