jddlmZddlmZmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZGd d ee Zd S)) annotations)ABCabstractmethod)AnyDictOptionalUnion)logging)AutoRegisterABCMetaClass) BasePredictor HPIConfigPaddlePredictorOption)uses_server_backendceZdZdZdZdddddddd"fdZedZed#dZ d$dZ d%dZ d Z d!Z xZS)& BasePipelinezBase class for all pipelines. This class serves as a foundation for creating various pipelines. It includes common attributes and methods that are shared among all pipeline implementations. TNF)deviceengine engine_config pp_optionuse_hpip hpi_configr Optional[str]rrOptional[Dict[str, Any]]rOptional[PaddlePredictorOption]rboolr*Optional[Union[Dict[str, Any], HPIConfig]]returnNonec t||_||_||_||_||_||_dS)a Initializes the class with specified parameters. Args: device (Optional[str], optional): The device to use for prediction. Defaults to `None`. engine (Optional[str], optional): Inference engine. Defaults to `None`. engine_config (Optional[Dict[str, Any]], optional): Engine-specific config. Defaults to `None`. pp_option (Optional[PaddlePredictorOption], optional): Paddle predictor options. Defaults to `None`. use_hpip (bool, optional): Whether to use HPIP. Defaults to `False`. hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional): HPIP configuration. Defaults to `None`. N)super__init__rrrrrr) selfrrrrrrkwargs __class__s p/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddlex/inference/pipelines/base.pyr#zBasePipeline.__init__$sJ0   *"  $c  td)z Declaration of an abstract method. Subclasses are expected to provide a concrete implementation of `predict`. Args: input: The input data to predict. **kwargs: Additional keyword arguments. z2The method `predict` has not been implemented yet.)NotImplementedErrorr$inputr%s r'predictzBasePipeline.predictDs""VWWWr(configDict[str, Any]inherited_engine allow_genaituple[Optional[str], bool]cd|vr|dd}||dufSd|v}|r%t|ddrd}|rdS|dfS)aResolve the effective child engine. Returns a tuple of `(engine, suppress_inherited_engine_defaults)`. Same-level `engine` has the highest priority. If a child omits `engine` but specifies another engine selector such as `use_hpip`, or a `genai_config` that targets a remote server backend, that selector should beat the inherited parent `engine` and fall back to local auto-resolution instead of reusing the parent engine defaults. rNr genai_configT)NTF)getr)r.r0r1 child_enginehas_local_engine_selectors r'_resolve_child_enginez"BasePipeline._resolve_child_engineOs v  !::h55L!55 5$.&$8!  -.vzz.$/O/OPP -(, % $ :&&r(rrc zd|vrt|d|dd}|||jd\}}|dd}|j/|s-t |j}|r|||}|d|j}|dd} |j]| pi} t|jd r|j d n"t|jt r|jni} i| | } d d l m } tjd |d||f|j|jnd} | d|d||j|||dd| || |ddd |S)zK Create a model instance based on the given configuration. model_config_error model_dirNTr1rrr model_dump exclude_noner )create_predictorzCreating model: %s model_name batch_sizer4) rAr;rrrrBrrrr4) ValueErrorr5r8rrdictupdaterrhasattrr= isinstancemodelsr@r inforcopyr) r$r.r%r; model_engine"suppress_inherited_engine_defaultsmodel_engine_configmergedrrbaser@rs r' create_modelzBasePipeline.create_modeljs 6 ) )V$89:: :JJ{D11 ;?;U;U DKT)r.rrrrrrrD)rErVr8rr5rrrHr=rIrFrrrrL) r$r.r%rVpipeline_enginerNrrrQs r'rVzBasePipeline.create_pipelines] #f , ,V$;<== =%%%%%%  & &vt{ & N N <;::j$-88ZZ d33 ? &#)rJ4?L99T***===)3DOT)J)JRdooPR  0D/J/J  ;":R@R040Jt~**,,,PT!      r(cdS)NrD)r$s r'closezBasePipeline.closes r(c |j|fi|S)aB Calls the `predict` method with the given input and keyword arguments. Args: input: The input data to be predicted. **kwargs: Additional keyword arguments to be passed to the `predict` method. Returns: The prediction result from the `predict` method. )r-r+s r'__call__zBasePipeline.__call__st|E,,V,,,r()rrrrrrrrrrrrrr )r.r/r0rr1rrr2)r.rrr)r.rrrS)__name__ __module__ __qualname____doc___BasePipeline__is_baser#rr- staticmethodr8rRrVrZr\ __classcell__)r&s@r'rrsI !% $2659AE%%%%%%%%@XX^X'''\'41 1 1 1 f# # # # J    - - - - - - -r(r) metaclassN) __future__rabcrrtypingrrrr utilsr utils.subclass_registerr rJrrrmodels.common.genairrrDr(r'rks#"""""########------------??????DDDDDDDDDD555555v-v-v-v-v-3":v-v-v-v-v-v-r(