ڎj dZddlZddlmZmZmZddlmZddlm Z m Z m Z de de fdZ d edefd Z dd ed edede fdZdS)zThe tool module utils.N)AnyDictCallable)parse)Field create_model ConfigDictschemareturncRd|vr|dd|vrA|dD]&}t|trt |'d|vr0t|dtrt |dd|vr0t|dtrt |dd|vr\t|dtrA|dD]&}t|trt |'|S)zLRemove the title field from the JSON schema to avoid misleading the LLM.title propertiesitemsadditionalPropertiesz$defs)popvalues isinstancedict_remove_title_field)r prop def_schemas DC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\tool\_utils.pyrr sD& 7v<(//11 * *D$%% *#D)))&Zw>>F7O,,,''J%& --' F#9:;;; &Zw>> /0022 0 0J*d++ 0#J/// M docstringct|pd}g}|j||j|j||jd|S)zExtract the function description from the docstring. Args: docstring (`str`): The docstring to extract the function description from. Returns: `str`: The extracted function description. N )rshort_descriptionappendlong_descriptionjoin)rparsed_docstring descriptionss r_extract_func_descriptionr$.soY_"--L)5,>???(4,=>>> 99\ " ""rF tool_funcinclude_var_positionalinclude_var_keywordc t|jpd}d|jD}i}tj|jD]\}}|dvr |jtjj kr|s(|j tjj krtttfntt|j ft|d|||d|j|j urin|jf||<|jtjjkr|s|j tjj krt&tnt&|j t|d|||d|j|j urgn|jf||<z|j tjj krtn|j t||d|j|j urdn|jf||<t) d d t+d i|}|} t/| | S)aExtract input schema from the tool function's docstring Args: tool_func (`Callable`): The tool function to extract the JSON schema from. include_var_positional (`bool`): Whether to include variable positional arguments in the JSON schema. include_var_keyword (`bool`): Whether to include variable keyword arguments in the JSON schema. Returns: `dict`: The extracted input JSON schema. rc(i|]}|j|jS)arg_name description).0_s r z)_extract_input_schema..YsLLLa AMLLLr)selfclsz**N)r,default*._StructuredOutputDynamicClass __config__T)arbitrary_types_allowed)r4)r__doc__paramsinspect signature parametersrkind Parameter VAR_KEYWORD annotationemptyrstrrrgetr2VAR_POSITIONALlistrr model_json_schemar) r%r&r'rparams_docstringfieldsnameparam base_modelparams_json_schemas r_extract_input_schemarLDs(i'-2..ILL9;KLLLF(33>DDFF77 e ? " "  :*6 6 6& #w'8'>>>S#X#u//0 0 4 4#T (,,T488!! } 33B F4LLZ7,; ; ;) #w'8'>>>S %*+ 0 4 4"D (,,T488!! } 33B F4LL$#w'8'>>>% 0 4 4T4 @ @} 33 C  F4LL'd;;; J $5577*+++ r)FF)r7r9typingrrrdocstring_parserrpydanticrrr rrrAr$boolrLr*rrrQs&&&&&&&&&&""""""4444444444!!!!!!H######0$) %\\\ \\ \\\\\\r