o 1 i @sddlZddlZddlmZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZmZddlmZee Ze fd ed ed efd d Zeddd ed efddZeddd efddZedd ddddeedeed efddZdS)N)wraps)CallableOptional) import_attr)DEFAULT_CONSUMER_CONCURRENCYSERVE_LOGGER_NAMETaskConsumerWrapper)TaskProcessorAdapterTaskProcessorConfig) PublicAPItask_processor_configconsumer_concurrencyreturnc Cs|j}t|tr t|}nt|r|}n tdt|jd|z||}Wnty>}z t d|jd|d}~wwt|t sQt|jdt|jz| |W|Styp}z t d|jd|d}~ww)Nz>Adapter must be either a string path or a callable class, got z: zFailed to instantiate z- must inherit from TaskProcessorAdapter, got zFailed to initialize ) adapter isinstancestrrcallable TypeErrortype__name__ Exception RuntimeErrorr Z initialize)r rr adapter_classZadapter_instanceerc/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/serve/task_consumer.py_instantiate_adapters2     ralpha)Z stabilitycCst|S)a Create a TaskProcessorAdapter instance from the provided configuration and call .initialize(). This function supports two ways to specify an adapter: 1. String path: A fully qualified module path to an adapter class Example: "ray.serve.task_processor.CeleryTaskProcessorAdapter" 2. Class reference: A direct reference to an adapter class Example: CeleryTaskProcessorAdapter Args: task_processor_config: Configuration object containing adapter specification. Returns: An initialized TaskProcessorAdapter instance ready for use. Raises: ValueError: If the adapter string path is malformed or cannot be imported. TypeError: If the adapter is not a string or callable class. Example: .. code-block:: python config = TaskProcessorConfig( adapter="my.module.CustomAdapter", adapter_config={"param": "value"}, queue_name="my_queue" ) adapter = instantiate_adapter_from_config(config) )rr rrrinstantiate_adapter_from_config9s!r csfdd}|S)a Decorator to mark a class as a TaskConsumer. Args: task_processor_config: Configuration for the task processor (required) Note: This decorator must be used with parentheses: @task_consumer(task_processor_config=config) Returns: A wrapper class that inherits from the target class and implements the task consumer functionality. Example: .. code-block:: python from ray import serve from ray.serve.task_consumer import task_consumer, task_handler @serve.deployment @task_consumer(task_processor_config=config) class MyTaskConsumer: @task_handler(name="my_task") def my_task(self, *args, **kwargs): pass csGfdddt}|S)NcsBeZdZUeed<fddZdeffdd ZfddZd S) z>task_consumer..decorator.._TaskConsumerWrapper_adaptercsj|g|Ri|dSN)__init__)selfargskwargs target_clsrrr#szGtask_consumer..decorator.._TaskConsumerWrapper.__init__rc st||_tjtjdD]\}}t|ddr*t|d|}t||}|j||qz |jt dWdSt yM}z t d|d}~ww)N) predicate_is_task_handlerF _task_namez"task consumer started successfullyzFailed to start task consumer: ) rr!inspect getmembers isfunctiongetattrZregister_task_handleZstart_consumerloggerinforerror)r$rnamemethodZ task_nameZ bound_methodrr(r rrinitialize_callables&    zRtask_consumer..decorator.._TaskConsumerWrapper.initialize_callablecs0|j|jtdr|dSdS)N__del__)r!Z stop_consumershutdownhasattrr7)r$r'rrr7s   zFtask_consumer..decorator.._TaskConsumerWrapper.__del__N) r __module__ __qualname__r __annotations__r#intr6r7rr5rr_TaskConsumerWrapper}s  r>r)r(r>rr'r decorator|s$z task_consumer..decoratorr)r r?rrr task_consumer]s 'r@r3_funcr3csHdurttr stdfdd}|dur"||S|S)a Decorator to mark a method as a task handler. Optionally specify a task name. Default is the method name. Arguments: _func: The function to decorate. name: The name of the task. Default is the method name. Returns: A wrapper function that is marked as a task handler. Example: .. code-block:: python from ray import serve from ray.serve.task_consumer import task_consumer, task_handler @serve.deployment @task_consumer(task_processor_config=config) class MyTaskConsumer: @task_handler(name="my_task") def my_task(self, *args, **kwargs): pass Nz*Task name must be a non-empty string, got cs<tstfdd}d|_pj|_|Std)Ncs|i|Sr"r)r%r&frrwrappersz0task_handler..decorator..wrapperTz)Async task handlers are not supported yet)r,iscoroutinefunctionrr*rr+NotImplementedError)rDrErArCrr?s  ztask_handler..decorator)rrstrip ValueError)rBr3r?rrAr task_handlers   rJr")r,logging functoolsrtypingrrZray._common.utilsrZray.serve._private.constantsrrZ ray.serve._private.task_consumerr Zray.serve.schemar r Zray.util.annotationsr getLoggerr0r=rr r@rrJrrrrsJ      $ #H