o yq i )@sddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZedZhd Zgd ZGd d d eZd dZddZeZdFddZeZeddZddZedFddZddddZddZdd Z ed!d"Z!ddd#d$d%Z"dd&d'd(Z#d)d*Z$d+d,Z%d-d.Z&d/d0Z'd1d2Z(d3d4Z)d5d6Z*d7d8Z+dGd:d;Z,dGdd?Z.d@dAZ/dBdCZ0dDdEZ1dS)HN) defaultdict) lru_cachewraps) Requirement)Version)loggingzG(?:;|and)*[ \t]*extra[ \t]*==[ \t]*['\"]([a-z0-9]+(?:-[a-z0-9]+)*)['\"]>pluginsbaseall)zfastdeploy-serverz vllm-serverz sglang-serverc@s eZdZdS)DependencyErrorN)__name__ __module__ __qualname__rrY/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddlex/utils/deps.pyr #sr cCs@t|}|r|d|d|||dfSd|fS)Nr)_EXTRA_PATTERNsearchgroupstartend)dep_specmrrr'_get_extra_name_and_remove_extra_marker's *rcCstjd}i}|dgD] }|tvrtt||<qtjdD]*}t|\}}|durK|tvrK|}t |}||vsAJ||||j  |q!|S)NpaddlexzProvides-Extra) importlibmetadataget_all_COLLECTIVE_EXTRA_NAMESrlistrequiresrrstriprnameappend)rextrasr"r extra_namereqrrr _get_extras0s   r'FcCsjtt}tjdD](}t|\}}|r|dus"|s2|dus"|dkr2|}t|}||j |q |S)Nrr ) rrrrr rr!rr"r#) required_only dep_specsrr%r&rrr_get_base_dep_specsDs  r*T)r(cCs(ztj|WStjjyYdSwN)rrversionZPackageNotFoundError)deprrrget_dep_versionUs r.cCs|dvr |r td|dkrtjdduS|dkr"tjdduS|dkr.tjdduS|d kr:tjd duSt|}|durDd S|rc|tvrNtd t|D]}t|t|jvr`d SqRdSd S) N) paddlepaddlepaddle-custom-device ultra-infer fastdeployzx`check_version` is not allowed to be `True` for `paddlepaddle`, `paddle-custom-device`, `ultra-infer`, and `fastdeploy`.r/Zpaddler0Zpaddle_custom_devicer1Z ultra_inferr2FzHCurrently, `check_version=True` is supported only for base dependencies.T) ValueErrorrutil find_specr.BASE_DEP_SPECSrr specifier)r-Z check_versionr,rrrris_dep_available\s8 r8)obj_namecGsRdd|D}t|dkr'|durd|d}nd}|dd|7}t|dS) NcSsg|]}t|s|qSrr8.0r-rrr ~z require_deps..r`z#` is not ready for use, because theZThez+ following dependencies are not available:  )lenjoinr )r9depsZunavailable_depsmsgrrr require_deps}s rEcfdd}|S)Ncs"tfdd}t_|S)Ncs tjdji|i|SNr9rE_deps_r )argskwargsfuncrr_wrappersz7function_requires_deps.._deco.._wrapper)rsetrI)rMrNrCrLr_decos z%function_requires_deps.._decorrCrQrrPrfunction_requires_depss  rScrF)Ncs~tjfdd}t_tddD]}t|dr(j|jqdjvr2jnfdd}||_S) Ncs(tjdji|g|Ri|SrGrHselfrJrKcls old_init_funcrrrNsz4class_requires_deps.._deco.._wrapperrrI__init__cst|j|i|Sr+)superrZrT)rWrr_forwardsz4class_requires_deps.._deco.._forward) rrZrOrIinspectgetmrohasattrupdate__dict__)rWrNZbase_clsr\rPrVrrQs    z"class_requires_deps.._decorrRrrPrclass_requires_depss rbcCsBddt|D}t|rdStdddtt||DdS)NcSsg|]}t|qSrr:r;rrrr=sz&is_extra_available..Tz(These dependencies are not available: %scSsg|]\}}|s|qSrr)r<dfrrrr=r>F)EXTRASr rdebugzip)extraflagsrrris_extra_availablesrjr9altcCslt|s |durt|rdS|durd|d}nd}|d|d|d7}|dur2|d|d7}t|) Nr?z#` requires additional dependencies.z%Additional dependencies are required.z, To install them, run `pip install "paddlex[ul]=="` if you’re installing `paddlex` from an index, or `pip install -e "/path/to/PaddleX[u-]"` if you’re installing `paddlex` locally.z) Alternatively, you can install `paddlex[z ]` instead.)rjr )rhr9rlrDrrr require_extrasrm)rlcsfdd}|S)NcsXt|jfdd}|j|jttr'tdks#Jd||_|S)Ncs$td|g|Ri|S)Nrk)rmrT)rlrhrX pipeline_namerrrNsz8pipeline_requires_extra.._deco.._wrapperrr)rrZentities isinstancerrA)Z pipeline_clsrNrlrh)rXrnrrQs z&pipeline_requires_extra.._decor)rhrlrQrrqrpipeline_requires_extrasrrcCtdS)Nr1r:rrrris_hpip_availablertcCtstddS)NzSThe high-performance inference plugin is not available. Please install it properly.)rtr rrrr require_hpip rwcCrsNZservingrjrrrris_serving_plugin_availablerur{cCrv)Nz@The serving plugin is not available. Please install it properly.)r{r rrrrrequire_serving_pluginrxr|cC"g}tdD]}||7}q|Sryrevaluesr)itemrrrget_serving_dep_specs rcCrsNZ paddle2onnxr:rrrris_paddle2onnx_plugin_availablerurcCrv)NzDThe Paddle2ONNX plugin is not available. Please install it properly.)rr rrrrrequire_paddle2onnx_pluginrxrcCr}rr~rrrrget_paddle2onnx_dep_specsrranycCs|dkr|tvrtd||dkr!tD] }t|rdSqdSd|vr)tdStd|rCddlm}|rAtd o@td SdSdS) NrzUnknown backend type: TFr2genai-r)is_cuda_availableZxformersz flash-attn) _SUPPORTED_GENAI_ENGINE_BACKENDSr3 is_genai_engine_plugin_availabler8rjenvr)backendberrrrr s  rcCs8t|s|dkr d}ndt|d}t|ddS)Nrz$The generative AI engine plugins arezThe generative AI z engine plugin isz+ not available. Please install it properly.)rrepr RuntimeError)rprefixrrrrequire_genai_engine_plugins rcCrs)Nz genai-clientrzrrrr is_genai_client_plugin_available'rurcCrv)NzMThe generative AI client plugin is not available. Please install it properly.)rrrrrrrequire_genai_client_plugin+rxrcCs*d}||vr td||dkrdSt)N)gpuzUnsupported device type: rzfastdeploy-gpu == 2.3.0rc0)r3AssertionError)Z device_typeZSUPPORTED_DEVICE_TYPESrrrget_genai_fastdeploy_spec2s rcCs`|dkr|tvrtd|d|vrtt|dg}td|D]}||7}q'|S)NclientzInvalid type: r2z is not supportedr)rr3rrer)typer)rrrrget_genai_dep_specs<s r)F)r)2importlib.metadatarimportlib.utilr]re collectionsr functoolsrrZpackaging.requirementsrZpackaging.versionrrcompilerrr Exceptionr rr'rer*r6ZREQUIRED_DEP_SPECSr.r8rErSrbrjrmrrrtrwr{r|rrrrrrrrrrrrrrsZ