.hm2RddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZddlmZmZddlmZmZmZddlmZdd lmZdd lmZeGd d ZeGd dZe dZd$dZej dZ! d%d&dZ" d'd(d#Z#dS))) annotationsN) dataclass)AnyCallableLiteralget_args get_originget_type_hints) DocstringDocstringSectionKind) BaseModelField create_model) UserError)RunContextWrapper)ensure_strict_json_schemacxeZdZUdZded< ded< ded< ded < d ed < d Zd ed< dZd ed< ddZdS) 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_schemadatar return tuple[list[Any], dict[str, Any]]cg}i}d}t|jjD]\}\}}|jr|dkrt ||d}|j|jkr||pgd}Q|j|j kr| |piy|j|j |j fvr|s| ||||<|||<||fS)z Converts validated data from the Pydantic model into (args, kwargs), suitable for calling the original function. FrNT) enumerater parametersitemsrgetattrkindVAR_POSITIONALextend VAR_KEYWORDupdatePOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDappend) selfr positional_args keyword_argsseen_var_positionalidxrparamvalues S/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/function_schema.py to_call_argszFuncSchema.to_call_args(s( &(') ##,DN,E,K,K,M,M"N"N + + C$! cQhhD$--EzU111&&u{333&*##u000##EKR0000 5u7RSSS*/#**51111).L&&&+ T"" ,,N)r r r!r")__name__ __module__ __qualname____doc____annotations__rrr8r9r7rrsIII#*****E&&&&Y    (M_#####= - - - - - -r9rc6eZdZUdZded< ded< ded<dS) FuncDocumentationzHContains metadata about a python function, extracted from its docstring.rrrrzdict[str, str] | Noneparam_descriptionsN)r:r;r<r=r>r?r9r7rArAKs?RR III3F----QQr9rA)googlenumpysphinxdocrr!DocstringStylecdddd}gd}|D]2}tj||tjr|dxxdz cc<3gd}|D]2}tj||tjr|dxxdz cc<3gd}|D]2}tj||tjr|d xxdz cc<3t|}|dkrd Sgd}|D]}|||kr|cSd S) Nr)rErDrC)z ^:param\sz^:type\sz ^:return:z^:rtype:rEr)z^Parameters\s*\n\s*-{3,}z^Returns\s*\n\s*-{3,}z^Yields\s*\n\s*-{3,}rD)z^(Args|Arguments):z ^(Returns):z ^(Raises):rC)research MULTILINEmaxvalues) rFscoressphinx_patternspatternnumpy_patternsgoogle_patterns max_scorestylesstyles r7_detect_docstring_stylerV\sj34qA(N(NFMLLO""" 9Wc2< 0 0 " 8    !   N "!! 9Wc2< 0 0 ! 7OOOq OOOMLLO""" 9Wc2< 0 0 " 8    !   FMMOO$$IA~~x$A#@#@F %=I % %LLL & 8r9c#Ktjd}|}|tj dV||dS#||wxYw)Ngriffe)logging getLoggergetEffectiveLevelsetLevelERROR)loggerprevious_levels r7_suppress_griffe_loggingr`s{ x ( (F--//N OOGM"""( '''''''''s A&&A=funcCallable[..., Any]rUDocstringStyle | Nonec|j}tj|}|st|ddSt 5t |d|pt |}|}dddn #1swxYwYtd|Dd}d|D}t|j||pdS)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)rrrBr)linenoparserc3NK|] }|jtjk|jV!dSN)r(r textr6).0sections r7 z.generate_func_documentation..s4ZZ7 @T@Y0Y0Y0Y0Y0Y0YZZr9cfi|].}|jtjk|jD]}|j|j/Sr?)r(r r%r6rr)rjrkr5s r7 z/generate_func_documentation..sM***  </: : :] ; :   E% : : : :r9) r:inspectgetdocrAr`r rVparsenext)rarUrrF docstringparsedrrBs r7generate_func_documentationrus9 =D .  C W dQUVVVV ! # ###c!E4Y=TUX=Y=YZZZ ""################ZZfZZZ\`K*****  ]-5   s6BBBTdocstring_style name_overriderdescription_overrideuse_docstring_inforrc |rt||}|jpi}nd}i}|r |p|jn|j}t j|} t |} t| j } d} g} | r| d\}}| ||j }|tj kr5t|p|}|turd} n/| ||fn| ||f| ddD]x\}}| ||j }|tj kr1t|p|}|turt!d|j| ||fyi}| D]\}}| ||j }|j}|tj krt$}| |d}|j|jkrt|t*urXt-|}t/|dkr#|dt0urt|d}n tt$}n t|}|t3t|f||<|j|jkrt|t6urXt-|}t/|dkrt6|d|df}n.t6t8t$f}nt6t8|f}|t3t6|f||<|tj kr|t3d | f||<|t3|| f||<t;|d fd t<i|}|}|rtA|}tC||r |p|j"nd||| | |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 providing correct JSON input. Returns: A `FuncSchema` object containing the function's name, description, parameter descriptions, and other metadata. NFrTrz@RunContextWrapper param found at non-first position in function )default_factoryr.)r)defaultr_args__base__)rrrrrrr)#rurBrr:rorr listr%r&get annotation_emptyr rr/rr}rr(r)tuplerlenEllipsisrr+dictrrr model_json_schemarrr)rarvrwrxryrdoc_info param_descs func_namesig type_hintsparamsrfiltered_params first_name first_paramannoriginrr5fieldsr}field_description args_of_tuple dict_args dynamic_model json_schemas r7function_schemars5@.t_EE17R  2:M . I  D ! !C%%J #.&&(( ) )FMO >"() KnnZ)?@@ '. __+F*** $ && K'@AAAA  " "J #< = = =abbz . . ennT5#344 '. __+F***( (( e}---- F&>> ennT5#344- '. C(OOD$77 :- - -#%'' ( }%%**}Q/?8/K/K}Q/0CCs)CC3id8IJJJF4LL Z5, , ,#$&&$SMM y>>Q&&y|Yq\9:CCsCx.CC38nd8IJJJF4LL'.((#+<=== t '7HIII t !I!4!4!4SSySFSSM 1133K=/ <<   DLV(@H,@RV+&#-   r9)rFrr!rGrh)rarbrUrcr!rA)NNNTT)rarbrvrcrwrrxrryrrrr!r)$ __future__r contextlibrorYrI dataclassesrtypingrrrrr r rXr r pydanticr rr exceptionsr run_contextr strict_schemarrrArGrVcontextmanagerr`rurr?r9r7rs"""""" !!!!!!OOOOOOOOOOOOOOOO222222223333333333!!!!!!******444444 5-5-5-5-5-5-5- 5-p RRRRRRR R45 %%%%P (((>B'''''X.2 $'+## ^^^^^^^r9