.Ui:ddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZmZddlmZmZddlmZmZmZddlmZdd lmZdd lmZdd lmZdd lm Z eGd dZ!eGddZ"e dZ#ddZ$ejJdZ& d ddZ'ddZ(ddZ) d ddZ*y)) annotationsN) dataclass) AnnotatedAnyCallableLiteralget_args get_originget_type_hints) DocstringDocstringSectionKind) BaseModelField create_model) FieldInfo) UserError)RunContextWrapper)ensure_strict_json_schema) ToolContextcveZdZUdZded< ded< ded< ded < d ed < d Zd ed< dZd ed< ddZy) FuncSchemazg Captures the schema for a python function, in preparation for sending it to an LLM as a tool. strname str | None descriptionztype[BaseModel]params_pydantic_modelzdict[str, Any]params_json_schemazinspect.Signature signatureFbool takes_contextTstrict_json_schemac g}i}d}t|jjjD]\}\}}|jr|dk(rt ||d}|j |jk(r|j|xsgd}Y|j |jk(r|j|xsi|j |j|jfvr|s|j||||<|||<||fS)z Converts validated data from the Pydantic model into (args, kwargs), suitable for calling the original function. FrNT) enumerater parametersitemsr!getattrkindVAR_POSITIONALextend VAR_KEYWORDupdatePOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDappend) selfdatapositional_args keyword_argsseen_var_positionalidxrparamvalues r/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/function_schema.py to_call_argszFuncSchema.to_call_args*s &(') ##,DNN,E,E,K,K,M"N + C$!!cQhD$-EzzU111&&u{3&*#u000##EKR0 5 5u7R7RSS*#**51).L&&+ T"+ +, ,,N)r1rreturnz tuple[list[Any], dict[str, Any]])__name__ __module__ __qualname____doc____annotations__r!r"r9r:r8rrsW I#***E&&Y  (M4_##= -r:rc4eZdZUdZded< ded< ded<y) FuncDocumentationzHContains metadata about a Python function, extracted from its docstring.rrrrzdict[str, str] | Noneparam_descriptionsN)r<r=r>r?r@rAr:r8rCrCMsR I3F--Qr:rC)googlenumpysphinxcdddd}gd}|D]5}tj||tjs)|dxxdz cc<7gd}|D]5}tj||tjs)|dxxdz cc<7gd}|D]5}tj||tjs)|d xxdz cc<7t|j }|dk(ry gd}|D]}|||k(s |cSy ) Nr)rGrFrE)z ^:param\sz^:type\sz ^:return:z^:rtype:rGr)z^Parameters\s*\n\s*-{3,}z^Returns\s*\n\s*-{3,}z^Yields\s*\n\s*-{3,}rF)z^(Args|Arguments):z ^(Returns):z ^(Raises):rE)research MULTILINEmaxvalues) docscoressphinx_patternspatternnumpy_patternsgoogle_patterns max_scorestylesstyles r8_detect_docstring_stylerW^s34qA(NFMO"" 99Wc2<< 0 8  ! " N "! 99Wc2<< 0 7Oq O! MO"" 99Wc2<< 0 8  ! "FMMO$IA~$AF %=I %L r:c#Ktjd}|j}|jtj d|j|y#|j|wxYww)Ngriffe)logging getLoggergetEffectiveLevelsetLevelERROR)loggerprevious_levels r8_suppress_griffe_loggingrasV  x (F--/N OOGMM"( ''sAA4A A4A11A4c |j}tj|}|st|ddSt 5t |d|xs t |}|j}dddtdDd}|Dcic]G}|jtjk(r(|jD]}|j|jI} }}t|j|| xsdS#1swYxYwcc}}w)a Extracts metadata from a function docstring, in preparation for sending it to an LLM as a tool. Args: func: The function to extract documentation from. style: The style of the docstring to use for parsing. If not provided, we will attempt to auto-detect the style. Returns: A FuncDocumentation object containing the function's name, description, and parameter descriptions. N)rrrDr)linenoparserc3pK|].}|jtjk(s!|j0ywN)r(r textr7).0sections r8 z.generate_func_documentation..s&Z7 @T@Y@Y0YZs"66)r<inspectgetdocrCrar rWparsenextr(r r%r7rr) funcrVrrN docstringparsedrrir6rDs r8generate_func_documentationrrs ==D .. C  dQUVV ! ##c!E4Y=TUX=YZ "##ZfZ\`K *  <</:: :]] *   E%%%***  ]]-5 ##*s,C3A C?3C<cd}|}t|tur3t|}|s ||fS|d}g||dd}t|tur3||fS)zIReturns the underlying annotation and any metadata from typing.Annotated.rArrN)r rr ) annotationmetadataannargss r8_strip_annotatedrxso!#H C S/Y &}  =1g)X)QR) S/Y & =r:c:|D]}t|ts|cSy)zIExtracts a human readable description from Annotated metadata if present.N) isinstancer)ruitems r8"_extract_description_from_metadatar|s' dC K r:c |r&t||}t|jxsi}nd}i}t|d}i} i} |j D]1\} } | dk(r t | \} }| | | <t |}|-|| | <3| j D]\} }|j| ||xs|r |jn |j}tj|}t|jj }d}g}|r|d\}}| j||j}|tj k7r6t#|xs|}|t$us|t&urd}n'|j)||fn|j)||f|ddD]~\} }| j| |j}|tj k7r7t#|xs|}|t$us|t&urt+d|j|j)| |fi}|D]\} }| j| |j}|j,}|tj k(rt.}|j| d}|j0|j2k(rpt#|t4ur?t7|}t9|d k(r|dt:ur t|d}ntt.}n t|}|t=t| f|| <|j0|j>k(rwt#|tur?t7|}t9|d k(rt|d|df}n#tt@t.f}ntt@|f}|t=t| f|| <s|tj k(r|t=d | f|| <tC|tDr,|tEjF||xs |jH f|| <|t=|| f|| <tK|dfdtLi|}|jO} |r tQ| } tS||xs|r |jHnd|| |||S)a Given a Python function, extracts a `FuncSchema` from it, capturing the name, description, parameter descriptions, and other metadata. Args: func: The function to extract the schema from. docstring_style: The style of the docstring to use for parsing. If not provided, we will attempt to auto-detect the style. name_override: If provided, use this name instead of the function's `__name__`. description_override: If provided, use this description instead of the one derived from the docstring. use_docstring_info: If True, uses the docstring to generate the description and parameter descriptions. strict_json_schema: Whether the JSON schema is in strict mode. If True, we'll ensure that the schema adheres to the "strict" standard the OpenAI API expects. We **strongly** recommend setting this to True, as it increases the likelihood of the LLM producing correct JSON input. Returns: A `FuncSchema` object containing the function's name, description, parameter descriptions, and other metadata. NT)include_extrasr;FrrzLRunContextWrapper/ToolContext param found at non-first position in function )default_factoryr.)r)defaultr_args__base__)rrrrrr!r")*rrdictrDr r&rxr| setdefaultrr<rkrlistr%getrt_emptyr rrr/rrrr(r)tupler lenEllipsisrr+rrzrmerge_field_infosrrrmodel_json_schemarr)!rodocstring_style name_overridedescription_overrideuse_docstring_infor"doc_info param_descstype_hints_with_extras type_hintsannotated_param_descsrrt stripped_annrur func_namesigparamsr!filtered_params first_name first_paramrvoriginr6fieldsrfield_description args_of_tuple dict_args dynamic_model json_schemas! r8function_schemars@.t_E866<"=  +DF!#J,.288: 6j 8  !1*!= h' 48B  "*5 !$ ' 6388:2kt[12O8(--I   D !C #..&&( )FMO "() KnnZ)?)?@ '.. _+F**f .C $ && K'@A  " "J #< =abz . ennT5#3#34 '.. _+F**f .C ( e}- . F&F ennT5#3#34-- '.. C(OOD$7 ::-- -#%' ( }%*}Q/?8/K}Q/0Cs)C3id8IJF4L ZZ5,, ,#$&$SM y>Q&y|Yq\9:CsCx.C38nd8IJF4L'..(#+<= t GY///->-U'BUBU t '7HI t GFR!I;e!4SySFSM 113K/ <   (XXX-A-ASW+&#-  r:)rNrr;DocstringStylerf)roCallable[..., Any]rVDocstringStyle | Noner;rC)rtrr;ztuple[Any, tuple[Any, ...]])ruztuple[Any, ...]r;r)NNNTT)rorrrrrrrrr r"r r;r)+ __future__r contextlibrkrZrI dataclassesrtypingrrrrr r r rYr r pydanticrrrpydantic.fieldsr exceptionsr run_contextr strict_schemar tool_contextrrrCrrWcontextmanagerrarrrxr|rrAr:r8rs)" !ZZZ233%!*4% 5-5- 5-p RR R45 %P ((>B' '%:''T  .2 $'+## y y*yy% y  y  yyr: