o
پi:(
@ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZ d dl m
Z
d dl m Z d dl
mZ d dlmZmZ eeZdeded ee
d
efd d Zd
ededed ee
d
ef
ddZG dd deZdS ) N)AnyDictList)Tool)envs)BaseFormatDetector)StreamingParseResult _GetInfoFunc func_name
param_nametoolsreturnc C sF |D ] }|j j| kr |j jdi }||v r || dd S qdS )z$Get parameter type from tool schema.
propertiestypestring)functionname
parametersget)r
r r toolprops r Z/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/function_call/mimo_detector.py_get_param_type s r param_valuec
C s t | } | dkr
dS t|||}|dv r| S |ds7|ds7|ds7|ds7|ds7|d rRzt| W S ttfyQ t d
| || | Y S w |d s\|d rzt
| }|t| d
krl|W S t|W S ttfy t d| || | Y S w |dv r| } | dvrt d| || | dkS |dv s|ds|drzt | } | W S t j
ttfy t d| || Y nw zt| } W | S tttfy t d| || Y | S w )z
Convert parameter value based on its type in the schema.
Adapted from vllm-project/vllm (vllm/entrypoints/openai/tool_parsers/qwen3coder_tool_parser.py)
nullN)r strtextvarcharcharenumintintegeruintlongshortunsignedz[Parsed value '%s' of parameter '%s' is not an integer in tool '%s', degenerating to string.numfloatr zXParsed value '%s' of parameter '%s' is not a float in tool '%s', degenerating to string.)booleanboolbinary)truefalsezmParsed value '%s' of parameter '%s' is not a boolean (`true` or `false`) in tool '%s', degenerating to false.r, )objectarrayarrdictlistzvParsed value '%s' of parameter '%s' cannot be parsed with json.loads in tool '%s', will try other methods to parse it.z}Parsed value '%s' of parameter '%s' cannot be converted via Python `ast.literal_eval()` in tool '%s', degenerating to string.)htmlunescapelowerr
startswithr!
ValueError TypeErrorloggerwarningr( jsonloadsJSONDecodeErrorast literal_eval SyntaxError)r r r
r
param_typefloat_param_valuer r r _convert_param_value( s
rC c s e Zd ZdZ fddZdedefddZdedee de
fd d
Z d edee de
fd d
Z dedee de
eef fddZdefddZdefddZ ZS ) MiMoDetectorz
Detector for MiMo function call format.
Format:
pwd && ls
c sJ t d| _d| _tdtj| _tdtj| _tdtj| _ d S )Nz z z (.*?)z"]+)>(.*?)z$]+)>(.*?))
super__init__ bot_token eot_tokenrecompileDOTALLtool_call_regex
func_regex param_regexself __class__r r rF s
zMiMoDetector.__init__r r
c C s
| j |v S N)rG )rP r r r r
has_tool_call s
zMiMoDetector.has_tool_callr c C s | | j}|dkrt|g dS |d| }| |}g }|}| j|D ]A}|d} | | |}
|
rb|
d} | |vrYt
d| t j sY||||
7 }|
}q%|| |
| |
}q%t||dS )z#Parse complete text for tool calls.) normal_textcallsN r zUnknown function: )findrG r _get_tool_indicesrL finditergroup_parse_tool_callr r9 r: r SGLANG_FORWARD_UNKNOWN_TOOLSendextendparse_base_json) rP r r idxrV tool_indicesrW last_endmatchtool_call_bodyparsedr
r r r detect_and_parse s*
z MiMoDetector.detect_and_parsenew_textc C s | j |7 _ | j }|| j}|dkr&| jdkr tddS d| _ t|dS || j|}|dkrC|d| }||d | _ t|dS | |d|t| j |}|jr| jdkrdd| _g | _ dg| _
t| j | jkrz| j i t| j | jkslt| j
| jkr| j
d t| j
| jks|jd }|j |j
rt|j
ni d| j | j< |j
| j
| j< | j|_| jd7 _||t| j d | _ |S )zK
Streaming parsing: buffer until complete tool call block.
rU r )rV N)r argumentsrX )_bufferrY rG current_tool_idr rH rh lenrW prev_tool_call_arrstreamed_args_for_toolappendr r r; r<
tool_index) rP ri r current_textstartr_ rV resultcallr r r parse_streaming_increment sD
z&MiMoDetector.parse_streaming_incrementrf c
C sv | j |}|s
dS |d }|d}i }| j|D ]}|d }|d} t| |||||< q ||dS )z
Parse content inside ....
Structure:
tool_call_body contains: ...params...
NrX )r r )rM searchr\ striprN r[ rC )
rP rf r
func_matchr
func_bodyparams param_matchr r r r r r] s
z MiMoDetector._parse_tool_callc C s dS )NFr rO r r r supports_structural_tag z$MiMoDetector.supports_structural_tagc C s t rS )NotImplementedErrorrO r r r structure_info r zMiMoDetector.structure_info)__name__
__module__ __qualname____doc__rF r r* rT r r r rh rw r r r] r r r
__classcell__r r rQ r rD s*
!
7
rD )r> r3 r; loggingrI typingr r r &sglang.srt.entrypoints.openai.protocolr sglang.srt.environr -sglang.srt.function_call.base_format_detectorr #sglang.srt.function_call.core_typesr r getLoggerr r9 r r rC rD r r r r s0
a