i|dZddlmZddlmZmZmZddlmZd de ede de efdZ d d e e efde de efd Zy )z%Utilities for skill schema conversion)Any) BaseModelField create_model)ParameterSchema parameters model_namereturnc&|st|tSi}|D]}t}|j}|dk(rt}nO|dk(rt}nC|dk(rt }n7|dk(rt ttf}n|dk(rtt}n |dk(rt}|j |jnd }|s|dz}i}|jr|j|d <|r|td i|f||j<|td d di|f||j<t|fd ti|S)aConvert a list of ParameterSchema to a pydantic model for structured output Args: parameters: List of parameter schemas from the skill API model_name: Name for the generated pydantic model Returns: A pydantic BaseModel class with fields matching the parameter schemas __base__stringnumberbooleanobjectarraycookieNT descriptiondefaultr ) rrstrtypefloatbooldictrlistrequiredrrname)rr fieldsparam python_type param_type is_required field_kwargss b/mnt/e/genesis-system/.venvs/browser-army/lib/python3.12/site-packages/browser_use/skills/utils.pyconvert_parameters_to_pydanticr& s.  j9 55"KU+zz*8;X;Y;Xc3h;Wc;X;#(.."<$+ t#;, !&!2!2< $e&;l&;<6%**$e&ID&IL&IJ6%**E"KJ Z>)>v>>schemac|rd|vrt|tSi}|jdi}t|jdg}|j D]T\}}|jdd}|jd}t } |dk(rt } n|dk(rt } n|dk(rt} n|d k(rt} n|d k(rtt tf} n|d k(r|jd i} | jdd} | dk(rtt } nf| dk(rtt } nS| dk(rtt} n@| d k(rtt} n-| d k(rttt tf} n tt} ||v} | s| d z} i} |r|| d<| r| tdi| f||<C| tddd i| f||<Wt|fdti|S)akConvert a JSON schema to a pydantic model Args: schema: JSON schema dictionary (OpenAPI/JSON Schema format) model_name: Name for the generated pydantic model Returns: A pydantic BaseModel class matching the schema Note: This is a simplified converter that handles basic types. For complex nested schemas, consider using datamodel-code-generator. propertiesr rrrrrintegerrrritemsNrr r) rrgetsetr,rrintrrrrr)r(r rr*required_fields field_name field_schemafield_type_strfield_descriptionr! items_schema items_typer#r$s r%convert_json_schema_to_pydanticr7Bs ,f, j9 55jjr*vzz*b12!+!1!1!31KZ##FH5."&&}5+x;!;";";!c3h; ""7B/<  2:Hs)Khu+Kis)Kit*KhtCH~&Ks)Ko-+ t#;,!2< $e&;l&;<6*$e&ID&IL&IJ6*c1Kh Z>)>v>>r'N)SkillParameters) SkillOutput)__doc__typingrpydanticrrrbrowser_use.skills.viewsrrrrr&rr7rr'r%r>si+3345?tO/D5?RU5?nrs|n}5?pJ?DcNJ?J?`den`oJ?r'