o i@shdZddlmZddlmZddlZddlmZddlm Z ddl m Z e e Z Gd d d Zd gZdS) zHRequest director using openapi-core for stateless HTTP request building.)Any)urljoinN) SchemaPath) get_logger) HTTPRoutec @seZdZdZdefddZ ddedeee fded e j fd d Z dedeee fd e eee feee feee fe ffd d Zdedeee fded efddZdS)RequestDirectorzMBuilds httpx.Request objects from HTTPRoute and arguments using openapi-core.speccCs ||_dS)z+Initialize with a parsed SchemaPath object.N)_spec)selfr r u/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/experimental/utilities/openapi/director.py__init__s zRequestDirector.__init__http://localhostroute flat_argsbase_urlreturnc Cstd|jd|jd||||\}}}}td|d|d|d|||j||}|j} |r<|nd} |rB|nd} d} d} |durXt|tt BrV|} n|} t j | || | | | d S) a Constructs a final httpx.Request object, handling all OpenAPI serialization. Args: route: HTTPRoute containing OpenAPI operation details flat_args: Flattened arguments from LLM (may include suffixed parameters) base_url: Base URL for the request Returns: httpx.Request: Properly formatted HTTP request zBuilding request for  z with args: zUnflattened - path: z , query: z , headers: z, body: N)methodurlparamsheadersjsoncontent) loggerdebugrpath_unflatten_arguments _build_urlupper isinstancedictlisthttpxRequest)r rrr path_params query_params header_paramsbodyrrrr json_bodyrr r r builds6    zRequestDirector.buildcCsDi}i}i}i}t|drm|jrm|D]W\}}|durq||jvr/td|d|jq|j|} | d} | d} | dkrE||| <q| dkrN||| <q| d krW||| <q| d kr`||| <qtd | d |qnstd i} |jD]} | j| | j <qw|D][\}}|durqd|vr| dd\}} | dvr| dkr|||<n| dkr|||<n| d kr|||<q|| vr| |} | dkr|||<q| dkr|||<q| d kr|||<q|||<qd}|r|j r|j j rt t|j j }|j j |}|ddkr|}nt|dkrt t|}n|}n|}||||fS)aF Maps flat arguments back to their OpenAPI locations using the parameter map. Args: route: HTTPRoute with parameter_map containing location mappings flat_args: Flat arguments from LLM call Returns: Tuple of (path_params, query_params, header_params, body) parameter_mapNz Argument 'z!' not found in parameter map for location openapi_namerqueryheaderr)zUnknown parameter location 'z' for z2No parameter map available, using fallback mapping__r)rr/r0typeobject)hasattrr,itemsrwarning operation_idr parametersr-namersplit request_bodycontent_schemanextitergetlenvalues)r rrr&r'r( body_propsarg_namevaluemappingr-r.param_locationsparam base_namer) content_type body_schemar r r rPs                z$RequestDirector._unflatten_arguments path_templater&cCsT|}|D]\}}d|d}||vr||t|}qt|dd|dS)aG Build URL by substituting path parameters in the template. Args: path_template: OpenAPI path template (e.g., "/users/{id}") path_params: Path parameter values base_url: Base URL to prepend Returns: Complete URL with path parameters substituted {}/)r5replacestrrrstriplstrip)r rKr&rurl_path param_name param_value placeholderr r r rs zRequestDirector._build_urlN)r)__name__ __module__ __qualname____doc__rrrr"rPrr$r%r+tuplerrr r r r rs:  9 & l r)rZtypingr urllib.parserr$jsonschema_pathrfastmcp.utilities.loggingrmodelsrrWrr__all__r r r r s      G