o
پiN @ sz d dl Z d dlZd dlZd dlmZmZmZmZ d dlm Z d dl
m Z d dl m
Z
mZmZ eeZG dd de ZdS ) N)AnyDictListTuple)Tool)BaseFormatDetector)StreamingParseResult ToolCallItem _GetInfoFuncc
sv e Zd ZdZ fddZdedefddZdedee de
fd d
Z d ed ede fd
dZ
de dee fddZd edee de fddZdededee fddZdedee de
fddZd edee dee fd d Zd1d efd!d"Zdedee deeee f fd#d$Zd%edee dee fd&d'Zd(ed)ed*edee de f
d+d,Zdefd-d.Zde fd/d0Z Z S )2MinimaxM2Detectora
Detector for MiniMax M2 models.
Assumes function call format:
value1
value2
c sz t d| _d| _d| _d| _tdtj| _ tdtj| _
tdtj| _ d| _ d| _
i | _i | _d | _d | _d S )
Nzzz|| s zEMinimaxM2Detector._convert_param_value_with_types..)r8 nonenil) r: rH r; rI r9 rG r= r< r5 rD r% )r5 rD r% )r: rH )r; rI )r9 rG )true1yesonT)false0nooffF)r= r< ) rS rH
ValueError TypeErrorrI stripjsonloadsJSONDecodeError)r r0 rR normalized_types
type_priorityr1 val lower_valr# r# r$ r2 sV $ z1MinimaxM2Detector._convert_param_value_with_types
param_name param_configc C s0 ||vrdgS || }t |tsdgS | |S )a<
Get parameter types from parameter configuration.
Handles anyOf, oneOf, allOf, and direct type definitions.
Args:
param_name: The name of the parameter
param_config: The properties dict from the tool schema
Returns:
List of type strings
r5 )rA rB rJ )r rj rk param_schemar# r# r$ _get_param_types_from_config s
z.MinimaxM2Detector._get_param_types_from_confignew_textc
C s | j |7 _ d}g }t| ds| || _ | js*| j| j vr*|| j 7 }d| _ n0| jsn| j | j}|dkrB|| j 7 }d| _ n|| j d | 7 }| j |d | _ d| _d| _d| _i | _ i | _
| j t | jd | _ q| jst
d| j }|r|d }|| jv r|| _d| _| jdkrd| _t | j| jkr| ji t | j| jkst | j| jkr| jd t | j| jks|i d | j| j< |t| j|dd
| j | d | _ qtd | | || j 7 }d| _ nfne| jrY| | j |}|| | j| j v rX| j | j} | j| j }
|
r>|
d } |
d
} | | kr>|t| jd d
d
|
d
| j| j< | j | t | j d | _ | d | jd7 _qnqt ||dS )Nr
_tool_indicesTFz]+)\"> r name arguments
tool_indexrs
parameterszInvalid function name: {}r* ) r hasattr_get_tool_indicesro r r findr r r r lenr searchgrouprb current_tool_idprev_tool_call_arrappendstreamed_args_for_toolr endloggerwarning_reset_streaming_state _parse_and_stream_parametersextendr countr )
r rn r) r. r, sfunction_match
function_nameparameter_callsend_poscurrent_streamed open_braces close_bracesr# r# r$ parse_streaming_increment s
vz+MinimaxM2Detector.parse_streaming_increment
text_to_parsec C s g }t td|tj}i }|D ]}|d }|d}| | j|||||< q|| jkr| j | j
} | jspg }
| D ]\} } |
t
j| dd dt
j| dd q>dd|
}
| t| j
d |
d
|
| j | j
< nHt| t| j }|rg }|D ]} || } | t
j| dd dt
j| dd qdd| }
| t| j
d |
d
| |
| j | j
< || _|| j| j
d < |S ) a
Parse complete parameter blocks from text and return any tool call items to emit.
This method:
1. Finds all complete blocks
2. Parses them into a dictionary
3. Compares with current parameters and generates diff if needed
4. Updates internal state
Args:
text_to_parse: The text to search for parameter blocks
Returns:
List of ToolCallItem objects to emit (may be empty)
z-]+)\">(.*?)rq F) ensure_asciiz: rx z, Nru rt )rF r finditerr r rb _parse_parameterr r r r itemsr rc dumpsjoinr rC keysr )r r r) r,
param_matches
new_paramsmatchrj param_valueprevious_args_jsonr keyr0
json_fragmentnew_keyscontinuation_partsr# r# r$ r e sj
z.MinimaxM2Detector._parse_and_stream_parametersFstill_in_tool_callc C s( || _ d| _d| _i | _i | _d| _dS )z,Reset streaming state for the next tool callFr N)r r r r r current_tool_name_sent)r r r# r# r$ r s
z(MinimaxM2Detector._reset_streaming_statec C s g }g }d} | | j|}|dkr |||d n:|||| | | j|}|dkr:|||d n |||t| j }|t| j }|| || qd||fS )Nr Trp r )r| r r r r} r _parse_blockr ) r r% r) normal_partsr, cursorr eblockr# r# r$ r- s$
zMinimaxM2Detector._extractr c C s& g }| j |D ]}|d r|d n|d }d|vrq|d}|d | }||d d }i } | j|D ]8}
|
d rB|
d n|
d } d| vrKq8| d} | d | }
| | d d dd}| ||
||| |
< q8|| d}z || || W q t
y t d| Y qw |S )Nr rq z">r
rr z invalid tool call for %s dropped)
r findallindexrb r lstriprstripr r parse_base_json Exceptionr r )r r r) resmtxtidxfnamebodyparamspmptxtpidxpnamepvalrawr# r# r$ r s0
z MinimaxM2Detector._parse_blockr r r c C sd i }|D ]!}|j j|kr%|j jd ur%|j j}t|tr%d|v r%|d } nq| ||}| ||S )N
properties)functionrs rw rA rB rm r2 ) r r r r r) rk toolrw r1 r# r# r$ r s z"MinimaxM2Detector._parse_parameterc C s dS )NFr# r r# r# r$ supports_structural_tag z)MinimaxM2Detector.supports_structural_tagc C s t r' )NotImplementedErrorr r# r# r$ structure_info r z MinimaxM2Detector.structure_info)F) __name__
__module__ __qualname____doc__r rD rG r( r r r r/ r r3 rF rJ r2 rB rm r r r r r r- r r r r
r
__classcell__r# r# r! r$ r sh <
L
\&
r )rc loggingr typingr r r r &sglang.srt.entrypoints.openai.protocolr -sglang.srt.function_call.base_format_detectorr #sglang.srt.function_call.core_typesr r r
getLoggerr r r r# r# r# r$ s