57ea047acf
Signed-off-by: Mike_Qiu <qiudayu.qdy@antgroup.com> Signed-off-by: Yueqian Lin <linyueqian@outlook.com> Signed-off-by: Mike Qiu <qdy220091330@gmail.com> Co-authored-by: Mike_Qiu <qiudayu.qdy@antgroup.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: Yueqian Lin <linyueqian@outlook.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
25 lines
787 B
Python
25 lines
787 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
|
|
from vllm.reasoning import ReasoningParserManager
|
|
|
|
# Step-Audio specific reasoning parser.
|
|
# Registered at import time so it is available before the engine
|
|
# resolves ``--reasoning-parser step_audio``.
|
|
|
|
_OMNI_REASONING_PARSERS = {
|
|
"step_audio": (
|
|
"vllm_omni.reasoning.step_audio_reasoning_parser",
|
|
"StepAudioReasoningParser",
|
|
),
|
|
}
|
|
|
|
|
|
def register_omni_reasoning_parsers():
|
|
"""Register vllm-omni reasoning parsers with vLLM's ReasoningParserManager."""
|
|
for name, (module_path, class_name) in _OMNI_REASONING_PARSERS.items():
|
|
ReasoningParserManager.register_lazy_module(name, module_path, class_name)
|
|
|
|
|
|
register_omni_reasoning_parsers()
|