)`i0TdZddlmZmZmZmZmZGddZeZdS)z.Handler registry for managing Custom handlers.)AnyCallableDictListOptionalceZdZdZddZdededefddfdZd ededefddfd Z d ededefddfd Z dede edeffd Z d ede edeffd Z d ede edeffdZd edefdZd edefdZdedefdZdeddfdZd eddfdZd eddfdZdeefdZdeefdZdeefdZddZdS)HandlerRegistryzDGeneral registry for managing handlers by name with proper priority.returnNc0i|_i|_i|_dS)N) _handlers_framework_default_handlers_decorator_handlersselfs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/common/handler/registry.py__init__zHandlerRegistry.__init__ s8:JL(BD   namehandler.c||j|<dS)zSet a handler by name.Nr )rrrs r set_handlerzHandlerRegistry.set_handlers&tr handler_typec||j|<dS)zSet a framework default handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') handler: The handler function to register as framework default Nr rrrs rset_framework_defaultz%HandlerRegistry.set_framework_defaults:A(666rc||j|<dS)zSet a decorator handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') handler: The handler function to register as decorator handler Nrrs rset_decorator_handlerz%HandlerRegistry.set_decorator_handlers29 ...rc6|j|S)zGet a handler by name.)r getrrs r get_handlerzHandlerRegistry.get_handler(s~!!$'''rc6|j|S)zGet a framework default handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: The framework default handler function, or None if not found )r r"rrs rget_framework_defaultz%HandlerRegistry.get_framework_default,s/33LAAArc6|j|S)zGet a decorator handler. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: The decorator handler function, or None if not found )rr"r&s rget_decorator_handlerz%HandlerRegistry.get_decorator_handler7s'++L999rc||jvS)zCheck if a framework default handler is registered. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: True if the framework default handler is registered, False otherwise rr&s rhas_framework_defaultz%HandlerRegistry.has_framework_defaultBst???rc||jvS)zCheck if a decorator handler is registered. Args: handler_type: The type of handler (e.g., 'ping', 'invoke') Returns: True if the decorator handler is registered, False otherwise rr&s rhas_decorator_handlerz%HandlerRegistry.has_decorator_handlerMst777rc||jvS)z)Check if a handler is registered by name.rr#s r has_handlerzHandlerRegistry.has_handlerXst~%%rc<|j|ddS)zRemove a handler by name.N)r popr#s rremove_handlerzHandlerRegistry.remove_handler\s  4&&&&&rc<|j|ddS)z+Remove a framework default handler by type.N)r r1r&s rremove_framework_defaultz(HandlerRegistry.remove_framework_default`s! (,,\4@@@@@rc<|j|ddS)z#Remove a decorator handler by type.N)rr1r&s rremove_decorator_handlerz(HandlerRegistry.remove_decorator_handlerds!  $$\488888rcNt|jS)z"List all registered handler names.)listr keysrs r list_handlerszHandlerRegistry.list_handlershsDN''))***rcNt|jS)z4List all registered framework default handler types.)r8r r9rs rlist_framework_defaultsz'HandlerRegistry.list_framework_defaultslsD499;;<<r@rrr r sNNEEEE ''hsCx.@'T'''' A A*238*< A  A A A A 9 9*238*< 9  9 9 9 9((#s(1C(D(((( B# B(8CQTHCU:V B B B B :# :(8CQTHCU:V : : : : @# @$ @ @ @ @ 8# 8$ 8 8 8 8&&&&&&'3'4''''ASATAAAA9S9T9999+tCy++++=c====5c5555))))))rr N) rDtypingrrrrrr handler_registryrGrrrJs|4466666666666666r)r)r)r)r)r)r)r)l#?$$r