WiN, 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_ERRORa/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/instructor/core/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 ZddZddZeddZddZy )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 "rc~t|tstSt}|jj D]1\}}|j|j |j 3|jj D]1\}}|j|j |j 3|S)aO Combine two Hooks instances into a new one. This creates a new Hooks instance that contains all handlers from both the current instance and the other instance. Handlers are combined by appending the other's handlers after the current instance's handlers. Args: other: Another Hooks instance to combine with this one. Returns: A new Hooks instance containing all handlers from both instances. Example: >>> hooks1 = Hooks() >>> hooks2 = Hooks() >>> hooks1.on("completion:kwargs", lambda **kw: print("Hook 1")) >>> hooks2.on("completion:kwargs", lambda **kw: print("Hook 2")) >>> combined = hooks1 + hooks2 >>> combined.emit_completion_arguments() # Prints both "Hook 1" and "Hook 2" rKr>NotImplementedrAitemsextendcopy)r#othercombinedrGhandlerss r__add__z Hooks.__add__s,%'! !7$(>>#7#7#9 B Ix   y ) 0 0 A B$)??#8#8#: B Ix   y ) 0 0 A Brct|tstS|jj D]1\}}|j|j |j 3|S)a Add another Hooks instance to this one in-place. This modifies the current instance by adding all handlers from the other instance. The other instance's handlers are appended after the current instance's handlers for each event type. Args: other: Another Hooks instance to add to this one. Returns: This Hooks instance (for method chaining). Example: >>> hooks1 = Hooks() >>> hooks2 = Hooks() >>> hooks1.on("completion:kwargs", lambda **kw: print("Hook 1")) >>> hooks2.on("completion:kwargs", lambda **kw: print("Hook 2")) >>> hooks1 += hooks2 >>> hooks1.emit_completion_arguments() # Prints both "Hook 1" and "Hook 2" rj)r#rorGrqs r__iadd__zHooks.__iadd__ s[,%'! !$)??#8#8#: > Ix NN9 % , ,X]]_ = > rcr|}|D]*}t||stdt|||z },|S)a Combine multiple Hooks instances into a new one. This class method creates a new Hooks instance that contains all handlers from all provided instances. Handlers are combined in the order of the provided instances. Args: *hooks_instances: Variable number of Hooks instances to combine. Returns: A new Hooks instance containing all handlers from all instances. Example: >>> hooks1 = Hooks() >>> hooks2 = Hooks() >>> hooks3 = Hooks() >>> hooks1.on("completion:kwargs", lambda **kw: print("Hook 1")) >>> hooks2.on("completion:kwargs", lambda **kw: print("Hook 2")) >>> hooks3.on("completion:kwargs", lambda **kw: print("Hook 3")) >>> combined = Hooks.combine(hooks1, hooks2, hooks3) >>> combined.emit_completion_arguments() # Prints all three hooks zExpected Hooks instance, got )rK TypeErrortype)clshooks_instancesrphooks_instances rcombinez Hooks.combine)sN25- 'Nnc2"?^@T?U VWW  &H ' rct}|jjD]1\}}|j|j|j 3|S)av Create a deep copy of this Hooks instance. Returns: A new Hooks instance with all the same handlers. Example: >>> original = Hooks() >>> original.on("completion:kwargs", lambda **kw: print("Hook")) >>> copy = original.copy() >>> copy.emit_completion_arguments() # Prints "Hook" )r>rArlrmrn)r# new_hooksrGrqs rrnz Hooks.copyKsSG #'>>#7#7#9 C Ix    * 1 1(--/ B Cr)r)r*)rG HookNameTyperH HandlerTyper)r*)rGr~r)r )rGr r$rr%rr)r*r(r2r8r!)rGzHookNameType | Noner)r*)ror>r)r>)ryr>r)r>)r)r>)rrrr,rBrIrDrXrZr\r^r`rbrerhrrrt classmethodr{rnrrrr>r>BsU222  2<(&?:4;/222  2**.#&# #"#J>Brr>) __future__renumr collectionsrtypingrrrr r rRrTr r rr.r4r;r~rr>rrrrs"#99 CL t ?h? 33 6X6 66      YYr