o پi@s&ddlmZmZddlmZddlmZmZmZm Z m Z m Z m Z ddl Z ddlZddlmZddlmZmZdeded efd d Zd ed ed e eeffddZd ed efddZdeed efddZded efddZdeeefd e efddZdeede ee dfd e efddZdS))JSONDecodeError JSONDecoder) WHITESPACE)AnyDictListLiteralOptionalTupleUnionN)Allow)Tool ToolChoices1s2returncCsLd}tt|t|}td|D]}||||kr!|||7}q|S|S)Nr)minlenrange)rrprefix min_lengthirR/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/function_call/utils.py_find_common_prefix sr input_strflagsc Csz t||t|fWSttfyE}z,t|dt|}d|vs%d|vr@t|d }t ||\}}||fWYd}~Sd}~ww)a Parse incomplete or partial JSON strings commonly encountered during streaming. Args: input_str (str): The potentially incomplete JSON string to parse. flags (Allow): Bitwise flags controlling what types of partial data are allowed. Common flags include: - Allow.STR: Allow partial strings (e.g., '"hello wo' -> 'hello wo') - Allow.OBJ: Allow partial objects (e.g., '{"key":' -> {'key': None}) - Allow.ARR: Allow partial arrays (e.g., '[1, 2,' -> [1, 2]) - Allow.ALL: Allow all types of partial data Returns: Tuple[Any, int]: A tuple containing: - parsed_object: The Python object parsed from the JSON - consumed_length: Number of characters consumed from input_str msgz Extra datazpop from empty listrN) partial_json_parserloadsrr IndexErrorgetattrstrrmatchendr raw_decode)rrerstartobjr%rrr_partial_json_loadssr*cCs&zt|WdStyYdSw)NTF)orjsonr r)rrrr_is_complete_json4s   r,toolscCsni}|D]0}|jjdur q|jjdi}|D]\}}||vr/|||kr/td|d|||<qq|S)a Get consolidated $defs from all tools, validating for conflicts. Args: tools: List of tools to process Returns: Dictionary of consolidated $defs from all tools Raises: ValueError: If conflicting $defs are found N$defszTool definition 'z/' has multiple schemas, which is not supported.)function parametersgetitems ValueError)r-all_defstooldefsdef_name def_schemarrr_get_tool_schema_defs<s     r9r5cCs6d|jjgd|jjr|jjndiddddgdS) Nstring)typeenumobject)r; properties)namer0r?r0)r>required)r/r?r0)r5rrr_get_tool_schemaZs rAschemac Cs.t|tsdSd|vr,|d}t|tr|St|tr,|r,dd|D}|r*|dSdSd|vs4d|vrm|dp=|d}g}t|trm|D] }t|}|rT||qG|rmtt|d krc|dSd|vridS|dSd |vrt|d tr|d s~dSt}|d D]M}|dur| d qt|t r| d qt|t r| d qt|t r| dqt|tr| dqt|tr| dqt|tr| dqt|d kr| SdSd|vrt|dtr|d}|D]}t|}|r|dkr|SqdSd|vrdSd|vrdSdS)a' Infer the primary type of a parameter from JSON Schema. Supports complex JSON Schema structures including: - Direct type field (including type arrays) - anyOf/oneOf: parameter can be any of multiple types - enum: parameter must be one of enum values - allOf: parameter must satisfy all type definitions - properties: inferred as object type - items: inferred as array type Args: schema: JSON Schema definition Returns: Inferred type ('string', 'number', 'object', 'array', etc.) or None Nr;cSsg|]}|dkr|qS)nullr).0trrr sz/infer_type_from_json_schema..rr:anyOfoneOfr<rCbooleanintegernumberarrayr=allOfr>r2) isinstancedictr#listr1infer_type_from_json_schemaappendrsetaddboolintfloatpop) rB type_valuenon_null_typesschemastypes sub_schema inferred_type enum_typesvaluerrrrRhs|                     rR tool_choicer@cCst|tr!|jj}|D]}|jj|krdddt|dSq dS|dkr?ddddd|Dd d }t|}|r=||d <|SdS) z Get the JSON schema constraint for the specified tool choice. Args: tool_choice: The tool choice specification Returns: JSON schema dict, or None if no valid tools found rMrI)r;minItemsmaxItemsr2Nr@r=cSsg|]}t|qSr)rA)rDr5rrrrFsz.get_json_schema_constraint..)r;rG)r;rcr2r.)rOrr/r?rAr9)r-rbfn_namer5 json_schemajson_schema_defsrrrget_json_schema_constraints.    rh)jsonrr json.decoderrtypingrrrrr r r r+r partial_json_parser.core.optionsr &sglang.srt.entrypoints.openai.protocolr rr#rrWr*rVr,rPr9rArRrhrrrrs( $  g