qi(dZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZddlZddlmZerddlmZGdde ZGdd ZGd d eZy) auDirect function wrapper utilities for LLM function calling. This module provides utilities for wrapping "direct" functions that handle LLM function calls. Direct functions have their metadata automatically extracted from function signatures and docstrings, allowing them to be used without accompanying configurations (as FunctionSchemas or in provider-specific formats). N) TYPE_CHECKINGAnyCallableDictListMappingProtocolSetTupleUnionget_args get_originget_type_hints)FunctionSchema)FunctionCallParamsc$eZdZdZdddeddfdZy)DirectFunctiona5Protocol for a "direct" function that handles LLM function calls. "Direct" functions' metadata is automatically extracted from their function signature and docstrings, allowing them to be used without accompanying function configurations (as FunctionSchemas or in provider-specific formats). paramsrkwargsreturnNc Kyw)zExecute the direct function. Args: params: Function call parameters from the LLM service. **kwargs: Additional keyword arguments passed to the function. N)selfrrs Z/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/adapters/schemas/direct_function.py__call__zDirectFunction.__call__2s  s)__name__ __module__ __qualname____doc__rrrrrr*s" %9 S T r rc eZdZdZdefdZedefdZededdfdZ de fdZ d Z d ed e ejdeeeefe effd Zd edeeeffdZy)BaseDirectFunctionWrapperzBase class for a wrapper around a DirectFunction. Provides functionality to: - extract metadata from the function signature and docstring - use that metadata to generate a corresponding FunctionSchema functionch|jj|||_|jy)zInitialize the direct function wrapper. Args: function: The function to wrap and extract metadata from. N) __class__validate_functionr#_initialize_metadata)rr#s r__init__z"BaseDirectFunctionWrapper.__init__Es) ((2   !!#r rctd)aGet the name of the special first function parameter. The special first parameter is ignored by metadata extraction as it's not relevant to the LLM (e.g., 'params' for FunctionCallParams). Returns: The name of the special first parameter. z8Subclasses must define the special first parameter name.)NotImplementedErrorclss rspecial_first_param_namez2BaseDirectFunctionWrapper.special_first_param_nameOs""\]]r Nctj|std|jdt tj |j j}|j}t|dk(rtd|jd|d|dd}||k7rtd|jd|dy) a%Validate that the function meets direct function requirements. Args: function: The function to validate. Raises: Exception: If function doesn't meet requirements (not async, missing parameters, incorrect first parameter name). zDirect function z must be asyncrz# must have at least one parameter ()z first parameter must be named ''N) inspectiscoroutinefunction Exceptionrlist signature parametersitemsr-len)r,r#rr-first_param_names rr&z+BaseDirectFunctionWrapper.validate_function[s**84.x/@/@.APQ Qg''1<<BBDE#&#?#?#A v;! "8#4#4"55XYqXrrst "!9Q< 7 7"8#4#4"55UVnUoopq  8r cpt|j|j|j|jS)zConvert the wrapped function to a FunctionSchema. Returns: A FunctionSchema instance with extracted metadata. )name description propertiesrequired)rr;r<r=r>)rs rto_function_schemaz,BaseDirectFunctionWrapper.to_function_schemats/ ((]]   r cJ|jj|_tjt j |j}|jxsdj|_|j|j|j\|_ |_ y)z:Initialize metadata from function signature and docstring.N) r#rr;docstring_parserparser1getdocr<strip_get_parameters_as_jsonschemarr=r>)r docstrings rr'z.BaseDirectFunctionWrapper._initialize_metadatas|MM** %**7>>$--+HI &117R>>@*.)K)K MM9++* &r funcdocstring_paramsctj|}t|}i}g}|jj D]\}}|dk(r |t t |jk(} | r0|j|} |j| ||<|jtjjur|j||D](} | j|k(s| jxsd||d<*||fS)aGet function parameters as a dictionary of JSON schemas and a list of required parameters. Ignore the first parameter, as it's expected to be the "special" one. Args: func: Function to get parameters from. docstring_params: List of parameters extracted from the function's docstring. Returns: A tuple containing: - A dictionary mapping each function parameter to its JSON schema - A list of required parameter names rrAr<)r1r5rr6r7nextiterget_typehint_to_jsonschemadefault Parameteremptyappendarg_namer<) rrHrIsighintsr=r>r;paramis_first_param type_hint doc_params rrFz7BaseDirectFunctionWrapper._get_parameters_as_jsonschemas"%t$ >>//1 RKD%v~"T$s~~*>%??N $I $;;IFJt }} 1 1 7 77%. R %%-6?6K6K6QrJt$]3 R- R48##r rXcP|iS|tdurddiS|turddiS|turddiS|turddiS|turddiS|t us|t urddiS|tus|turdd iSt|}t|}|tus|tjur!d |Dcgc]}|j|c}iS|tttt t"t$fvr|rd |j|d d S|t t fvr%t'|d k(rd|j|ddSt)|drhi}g}t+|dd}t-|j/D]-\}} |j| ||<|s|j1|/d|d} |r|| d<| SiScc}w)zConvert a Python type hint to a JSON Schema. Args: type_hint: A Python type hint Returns: A dictionary representing the JSON Schema NtypenullstringintegernumberbooleanobjectarrayanyOfr)r[r7)r[additionalProperties__annotations__ __total__T)r[r=r>)r[strintfloatbooldictrr4rrr r types UnionTyperNtupler setr r8hasattrgetattrrr7rR) rrXoriginargsargr=r>all_fields_required field_name field_typeschemas rrNz1BaseDirectFunctionWrapper._typehint_to_jsonschemas  I T "F# #  H% % # I& & % H% % $ I& & $ )t"3H% % $ )t"3G$ $I& " U?f74PCd::3?PQ Q dD%S9 9d#d.J.J4PQ7.ST T dD\ !c$i1n$d>Z>Z[_`a[b>cd d 9/ 0JH #*)[$"G *8*C*I*I*K 0& J)-)E)Ej)Q :&&OOJ/ 0 'jAF%-z"M EQs#F#)rrrrrr( classmethodrir-r&rr?r'rrBDocstringParamr rrrFrNrr rr"r"<s$$ ^ ^ ^d0  N   "0$0$045E5T5T0U0$ tCH~tCy( )0$dDDc3hDr r"c@eZdZdZedefdZdeeefddfdZ y) DirectFunctionWrapperzWrapper around a DirectFunction for LLM function calling. This class: - Extracts metadata from the function signature and docstring - Generates a corresponding FunctionSchema - Helps with function invocation rcy)zGet the special first parameter name for direct functions. Returns: The string "params" which is expected as the first parameter. rrr+s rr-z.DirectFunctionWrapper.special_first_param_namesr rurrcFK|jdd|i|d{S7w)aInvoke the wrapped function with the provided arguments. Args: args: Arguments to pass to the function. params: Function call parameters from the LLM service. Returns: The result of the function call. rNr)r#)rrurs rinvokezDirectFunctionWrapper.invokes'#T]]9&9D9999s !!N) rrrrr{rir-rrrrr rr~r~ s? :c!2 :rsZ  C? X $LL^:5:r