{h@ ddlmZddlmZddlmZddlmZmZm Z m Z m Z ddl Z ddl Z e dZGddeZGd d e ZGd d e ZGd de ZGdde Ze eedfZe eeeefZGddZy)) annotations)Enum) defaultdict)AnyLiteralTypeVarProtocolUnionNTc eZdZdZdZdZdZdZy)HookNamecompletion:kwargscompletion:responsecompletion:errorcompletion:last_attempt parse:errorN)__name__ __module__ __qualname__COMPLETION_KWARGSCOMPLETION_RESPONSECOMPLETION_ERRORCOMPLETION_LAST_ATTEMPT PARSE_ERRORQ/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/instructor/hooks.pyr r s+/)7Krr ceZdZdZddZy)CompletionKwargsHandlerz(Protocol for completion kwargs handlers.cyNrselfargskwargss r__call__z CompletionKwargsHandler.__call__rNr$rr%rreturnNonerrr__doc__r&rrrrrs2>rrceZdZdZddZy)CompletionResponseHandlerz*Protocol for completion response handlers.cyr!rr#responses rr&z"CompletionResponseHandler.__call__r'rNr1rr)r*r+rrrr.r.s42rr.ceZdZdZddZy)CompletionErrorHandlerz8Protocol for completion error and last attempt handlers.cyr!rr#errors rr&zCompletionErrorHandler.__call__$r'rNr7 Exceptionr)r*r+rrrr4r4!sB5rr4ceZdZdZddZy)ParseErrorHandlerz"Protocol for parse error handlers.cyr!rr6s rr&zParseErrorHandler.__call__*r'rNr8r+rrrr;r;'s,5rr;)rrrrrceZdZdZddZ ddZddZddZddZddZ ddZ dd Z dd Z dd Z d dd Zy )Hooksz Hooks class for handling and emitting events related to completion processes. This class provides a mechanism to register event handlers and emit events for various stages of the completion process. c,tt|_y)zInitialize the hooks container.N)rlist _handlers)r#s r__init__zHooks.__init__JsCNtCTrcb|j|}|j|j|y)a  Register an event handler for a specific event. This method allows you to attach a handler function to a specific event. When the event is emitted, all registered handlers for that event will be called. Args: hook_name: The event to listen for. This can be either a HookName enum value or a string representation of the event name. handler: The function to be called when the event is emitted. Raises: ValueError: If the hook_name is not a valid HookName enum or string representation. Example: >>> def on_completion_kwargs(*args: Any, **kwargs: Any) -> None: ... print(f"Completion kwargs: {args}, {kwargs}") >>> hooks = Hooks() >>> hooks.on(HookName.COMPLETION_KWARGS, on_completion_kwargs) >>> hooks.emit_completion_arguments(model="gpt-3.5-turbo", temperature=0.7) Completion kwargs: (), {'model': 'gpt-3.5-turbo', 'temperature': 0.7} N) get_hook_namerAappendr# hook_namehandlers ronzHooks.onNs+6&&y1  y!((1rc~t|tr t|S|S#t$r}td||d}~wwxYw)aM Convert a string hook name to its corresponding enum value. Args: hook_name: Either a HookName enum value or string representation. Returns: The corresponding HookName enum value. Raises: ValueError: If the string doesn't match any HookName enum value. zInvalid hook name: N) isinstancestrr ValueError)r#rGerrs rrDzHooks.get_hook_namelsP i % M ** M #6yk!BCL Ms  <7<c|j|D] } ||i| y#t$r>tj}t j d|j d|dYUwxYw)z Generic method to emit events for any hook type. Args: hook_name: The hook to emit *args: Positional arguments to pass to handlers **kwargs: Keyword arguments to pass to handlers z Error in z handler: ) stacklevelN)rAr9 traceback format_excwarningswarnvalue)r#rGr$r%rHerror_tracebacks remitz Hooks.emitsr~~i0 G ((  "+"6"6"8  0 O;LM  sAA&%A&cL|jtjg|i|y)z Emit a completion arguments event. Args: *args: Positional arguments to pass to handlers **kwargs: Keyword arguments to pass to handlers N)rXr rr"s remit_completion_argumentszHooks.emit_completion_argumentss"  (,,>t>v>rcD|jtj|y)z Emit a completion response event. Args: response: The completion response to pass to handlers N)rXr rr0s remit_completion_responsezHooks.emit_completion_responses (..9rcD|jtj|y)zt Emit a completion error event. Args: error: The exception to pass to handlers N)rXr rr6s remit_completion_errorzHooks.emit_completion_errors (++U3rcD|jtj|y)z{ Emit a completion last attempt event. Args: error: The exception to pass to handlers N)rXr rr6s remit_completion_last_attemptz"Hooks.emit_completion_last_attempts (22E:rcD|jtj|y)zo Emit a parse error event. Args: error: The exception to pass to handlers N)rXr rr6s remit_parse_errorzHooks.emit_parse_errors (&&.rc|j|}||jvrN||j|vr<|j|j||j|s|j|=yyyy)z Remove a specific handler from an event. Args: hook_name: The name of the hook. handler: The handler to remove. N)rDrAremoverFs roffz Hooks.offso&&y1  &$..33y)009~~i0y114 'rNc|.|j|}|jj|dy|jjy)z Clear handlers for a specific event or all events. Args: hook_name: The name of the event to clear handlers for. If None, all handlers are cleared. N)rDrApopclear)r#rGs rrhz Hooks.clears>  **95I NN  y$ / NN "r)r)r*)rG HookNameTyperH HandlerTyper)r*)rGrir)r )rGr r$rr%rr)r*r(r2r8r!)rGzHookNameType | Noner)r*)rrrr,rBrIrDrXrZr\r^r`rbrerhrrrr>r>BsU222  2<(&?:4;/222  2**.#&# #rr>) __future__renumr collectionsrtypingrrrr r rRrTr r rr.r4r;rirjr>rrrros"#99 CL t ?h? 33 6X6 66      a#a#r