f3fi dZddlmZddlZddlmZmZer6ddlmZddl m Z ddl m Z ddl mZdd lmZmZdd lmZdd lmZdd lmZmZmZej4eZGd dZGddZGddZGddZ GddZ!GddZ"Gddeee ee!e"Z#Gdde#Z$Gdde!Z%e&e#e%zdzZ'y)$Base callback handler for LangChain.) annotationsN) TYPE_CHECKINGAny)Sequence)UUID)RetryCallState)Self) AgentAction AgentFinish)Document) BaseMessage)ChatGenerationChunkGenerationChunk LLMResultcPeZdZdZdd ddZdd ddZy)RetrieverManagerMixinzMixin for Retriever callbacks.N parent_run_idc y)a%Run when Retriever errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nselferrorrun_idrkwargss [/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_core/callbacks/base.pyon_retriever_errorz(RetrieverManagerMixin.on_retriever_errorc y)a/Run when Retriever ends running. Args: documents: The documents retrieved. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)r documentsrrrs ron_retriever_endz&RetrieverManagerMixin.on_retriever_end+rr r BaseExceptionrrr UUID | Nonerrreturnr) r"Sequence[Document]rrrr&rrr'r)__name__ __module__ __qualname____doc__rr#rr rrrsy(&*     #      ,&*  %   #      r rcveZdZdZddd ddZdd d dZdd d dZy) LLMManagerMixinzMixin for LLM callbacks.N)chunkrc y)aRun on new output token. Only available when streaming is enabled. For both chat models and non-chat models (legacy LLMs). Args: token: The new token. chunk: The new generated chunk, containing content and other information. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)rtokenr/rrrs ron_llm_new_tokenz LLMManagerMixin.on_llm_new_token@rr rc y)a1Run when LLM ends running. Args: response: The response which was generated. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)rresponserrrs r on_llm_endzLLMManagerMixin.on_llm_endUrr c y)aRun when LLM errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nrrs r on_llm_errorzLLMManagerMixin.on_llm_errorfrr ) r1strr/,GenerationChunk | ChatGenerationChunk | Nonerrrr&rrr'r) r4rrrrr&rrr'rr$)r)r*r+r,r2r5r7rr rr.r.=s" ?C%)   <    #     4&*     #      ,&*     #      r r.ceZdZdZdd ddZdd d dZdd d dZdd d dZy) ChainManagerMixinzMixin for chain callbacks.Nrc y)a*Run when chain ends running. Args: outputs: The outputs of the chain. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)routputsrrrs r on_chain_endzChainManagerMixin.on_chain_end{rr c y)a!Run when chain errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nrrs ron_chain_errorz ChainManagerMixin.on_chain_errorrr c y)aRun on agent action. Args: action: The agent action. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)ractionrrrs ron_agent_actionz!ChainManagerMixin.on_agent_actionrr c y)aRun on the agent end. Args: finish: The agent finish. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)rfinishrrrs ron_agent_finishz!ChainManagerMixin.on_agent_finishrr ) r=dict[str, Any]rrrr&rrr'rr$) rBr rrrr&rrr'r) rEr rrrr&rrr'r)r)r*r+r,r>r@rCrFrr rr;r;xs$&*     #      ,&*     #      ,&*     #      ,&*     #      r r;cPeZdZdZdd ddZdd ddZy)ToolManagerMixinzMixin for tool callbacks.Nrc y)a*Run when the tool ends running. Args: output: The output of the tool. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)routputrrrs r on_tool_endzToolManagerMixin.on_tool_endrr c y)a Run when tool errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nrrs r on_tool_errorzToolManagerMixin.on_tool_errorrr ) rKrrrrr&rrr'rr$)r)r*r+r,rLrNrr rrIrIsy#&*     #      ,&*     #      r rIceZdZdZdddd d dZdddd d dZdddd d dZdddd d dZddddd dd Zy)CallbackManagerMixinzMixin for callback manager.Nrtagsmetadatac y)a]Run when LLM starts running. !!! warning This method is called for non-chat models (regular LLMs). If you're implementing a handler for a chat model, you should use `on_chat_model_start` instead. Args: serialized: The serialized LLM. prompts: The prompts. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. **kwargs: Additional keyword arguments. Nrr serializedpromptsrrrRrSrs r on_llm_startz!CallbackManagerMixin.on_llm_startrr c J|jjd}t|aMRun when a chat model starts running. !!! warning This method is called for chat models. If you're implementing a handler for a non-chat model, you should use `on_llm_start` instead. Args: serialized: The serialized chat model. messages: The messages. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. **kwargs: Additional keyword arguments. z) does not implement `on_chat_model_start` __class__r)NotImplementedError rrVmessagesrrrRrSrmsgs ron_chat_model_startz(CallbackManagerMixin.on_chat_model_starts'8(())RS!#&&r c y)aRun when the Retriever starts running. Args: serialized: The serialized Retriever. query: The query. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. **kwargs: Additional keyword arguments. NrrrVqueryrrrRrSrs ron_retriever_startz'CallbackManagerMixin.on_retriever_start$rr c yaRun when a chain starts running. Args: serialized: The serialized chain. inputs: The inputs. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. **kwargs: Additional keyword arguments. NrrrVinputsrrrRrSrs ron_chain_startz#CallbackManagerMixin.on_chain_start;rr rrRrSric y)aRun when the tool starts running. Args: serialized: The serialized chain. input_str: The input string. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. inputs: The inputs. **kwargs: Additional keyword arguments. Nr rrV input_strrrrRrSrirs r on_tool_startz"CallbackManagerMixin.on_tool_startRrr )rVrGrW list[str]rrrr&rRlist[str] | NonerSdict[str, Any] | Nonerrr'rrVrGr_zlist[list[BaseMessage]]rrrr&rRrqrSrrrrr'r)rVrGrdr8rrrr&rRrqrSrrrrr'r)rVrGrirGrrrr&rRrqrSrrrrr'r)rVrGrnr8rrrr&rRrqrSrrrirrrrr'r) r)r*r+r,rXrarerjrorr rrPrPs%&*!%*. "    #   (    D&*!%*.'"'*'  ' # ''('' 'J&*!%*. "    #   (    :&*!%*. "    #   (    :&*!%*.(, "    #   ( &    r rPczeZdZdZdd ddZdd d dZddd d dZy) RunManagerMixinzMixin for run manager.Nrc y)aRun on an arbitrary text. Args: text: The text. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nr)rtextrrrs ron_textzRunManagerMixin.on_textorr c yaRun on a retry event. Args: retry_state: The retry state. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. **kwargs: Additional keyword arguments. Nrr retry_staterrrs ron_retryzRunManagerMixin.on_retryrr rRrSc y)aOverride to define a handler for a custom event. Args: name: The name of the custom event. data: The data for the custom event. Format will match the format specified by the user. run_id: The ID of the run. tags: The tags associated with the custom event (includes inherited tags). metadata: The metadata associated with the custom event (includes inherited metadata). NrrnamedatarrRrSrs ron_custom_eventzRunManagerMixin.on_custom_eventrr ) rwr8rrrr&rrr'r r|r rrrr&rrr'r)rr8rrrrrRrqrSrrrrr'r)r)r*r+r,rxr}rrr rruruls &*     #      ,&*  #   #      ."&*.       (    r ruceZdZUdZdZded< dZded< eddZeddZ eddZ edd Z edd Z edd Z edd Zy )BaseCallbackHandlerrFbool raise_error run_inlinecy)z Whether to ignore LLM callbacks.Frrs r ignore_llmzBaseCallbackHandler.ignore_llmr cy)z"Whether to ignore retry callbacks.Frrs r ignore_retryz BaseCallbackHandler.ignore_retryrr cy)z"Whether to ignore chain callbacks.Frrs r ignore_chainz BaseCallbackHandler.ignore_chainrr cy)z"Whether to ignore agent callbacks.Frrs r ignore_agentz BaseCallbackHandler.ignore_agentrr cy)z&Whether to ignore retriever callbacks.Frrs rignore_retrieverz$BaseCallbackHandler.ignore_retrieverrr cy)z'Whether to ignore chat model callbacks.Frrs rignore_chat_modelz%BaseCallbackHandler.ignore_chat_modelrr cy)zIgnore custom event.Frrs rignore_custom_eventz'BaseCallbackHandler.ignore_custom_eventrr Nr'r)r)r*r+r,r__annotations__rpropertyrrrrrrrrr rrrs/K;J- r rceZdZdZdddd ddZdddd ddZdddd ddZddd dd Zddd d d Zdddd d!d Z ddd d"d Z ddd d d Z ddddd d#dZ ddd d$dZ ddd d dZddd d%dZdd d&dZddd d'dZddd d(dZdddd d)dZddd d*dZddd d dZddd d+dZy),AsyncCallbackHandlerz%Async callback handler for LangChain.NrQc Kyw)acRun when the model starts running. !!! warning This method is called for non-chat models (regular LLMs). If you're implementing a handler for a chat model, you should use `on_chat_model_start` instead. Args: serialized: The serialized LLM. prompts: The prompts. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. **kwargs: Additional keyword arguments. NrrUs rrXz!AsyncCallbackHandler.on_llm_startcRK|jjd}t|wrZr[r^s rraz(AsyncCallbackHandler.on_chat_model_starts+8(())RS!#&&s%')r/rrRc Kyw)aRun on new output token. Only available when streaming is enabled. For both chat models and non-chat models (legacy LLMs). Args: token: The new token. chunk: The new generated chunk, containing content and other information. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rr1r/rrrRrs rr2z%AsyncCallbackHandler.on_llm_new_tokenrr)rrRc Kyw)aSRun when the model ends running. Args: response: The response which was generated. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rr4rrrRrs rr5zAsyncCallbackHandler.on_llm_end2rrc Kyw)aRun when LLM errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. - response (LLMResult): The response which was generated before the error occurred. NrrrrrrRrs rr7z!AsyncCallbackHandler.on_llm_errorErrc Kywrgrrhs rrjz#AsyncCallbackHandler.on_chain_startZrrc Kyw)aHRun when a chain ends running. Args: outputs: The outputs of the chain. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rr=rrrRrs rr>z!AsyncCallbackHandler.on_chain_endqrrc Kyw)a=Run when chain errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nrrs rr@z#AsyncCallbackHandler.on_chain_errorrrrkc Kyw)aRun when the tool starts running. Args: serialized: The serialized tool. input_str: The input string. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. inputs: The inputs. **kwargs: Additional keyword arguments. Nrrms rroz"AsyncCallbackHandler.on_tool_startrrc Kyw)aFRun when the tool ends running. Args: output: The output of the tool. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rrKrrrRrs rrLz AsyncCallbackHandler.on_tool_endrrc Kyw)a<Run when tool errors. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nrrs rrNz"AsyncCallbackHandler.on_tool_errorrrc Kyw)a0Run on an arbitrary text. Args: text: The text. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rrwrrrRrs rrxzAsyncCallbackHandler.on_textrrrc Kywrzrr{s rr}zAsyncCallbackHandler.on_retryrrc Kyw)a5Run on agent action. Args: action: The agent action. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rrBrrrRrs rrCz$AsyncCallbackHandler.on_agent_actionrrc Kyw)a6Run on the agent end. Args: finish: The agent finish. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rrErrrRrs rrFz$AsyncCallbackHandler.on_agent_finish rrc Kyw)aRun on the retriever start. Args: serialized: The serialized retriever. query: The query. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. metadata: The metadata. **kwargs: Additional keyword arguments. Nrrcs rrez'AsyncCallbackHandler.on_retriever_start rrc Kyw)aDRun on the retriever end. Args: documents: The documents retrieved. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nr)rr"rrrRrs rr#z%AsyncCallbackHandler.on_retriever_end7rrc Kyw)a>Run on retriever error. Args: error: The error that occurred. run_id: The run ID. This is the ID of the current run. parent_run_id: The parent run ID. This is the ID of the parent run. tags: The tags. **kwargs: Additional keyword arguments. Nrrs rrz'AsyncCallbackHandler.on_retriever_errorJrrr~c Kyw)aOverride to define a handler for custom events. Args: name: The name of the custom event. data: The data for the custom event. Format will match the format specified by the user. run_id: The ID of the run. tags: The tags associated with the custom event (includes inherited tags). metadata: The metadata associated with the custom event (includes inherited metadata). Nrrs rrz$AsyncCallbackHandler.on_custom_event]rr)rVrGrWrprrrr&rRrqrSrrrrr'Noners)r1r8r/r9rrrr&rRrqrrr'r) r4rrrrr&rRrqrrr'r) rr%rrrr&rRrqrrr'r)rVrGrirGrrrr&rRrqrSrrrrr'r) r=rGrrrr&rRrqrrr'r)rVrGrnr8rrrr&rRrqrSrrrirrrrr'r) rKrrrrr&rRrqrrr'r) rwr8rrrr&rRrqrrr'rr) rBr rrrr&rRrqrrr'r) rEr rrrr&rRrqrrr'r)rVrGrdr8rrrr&rRrqrSrrrrr'r) r"r(rrrr&rRrqrrr'r)rr8rrrrrRrqrSrrrrr'r)r)r*r+r,rXrar2r5r7rjr>r@rorLrNrxr}rCrFrer#rrrr rrrs/&*!%*. "    #   (    D&*!%*.'"'*'  ' # ''('' 'F?C%)!%  <    #      8&*!%     #       0&*!%     #       6&*!%*. "    #   (    8&*!%     #       0&*!%     #       2&*!%*.(, "    #   ( &    <&*!%     #       0&*!%     #       0&*!%     #       0&*  #   #      ,&*!%     #       0&*!%     #       2&*!%*. "    #   (    8&*!%  %   #       0&*!%     #       2"&*.       (    r rceZdZdZ dddddd ddZddZddZeddZ d ddZ dd Z d dd Z d dd Z d dd Z dd Z d ddZddZy)BaseCallbackManagerz$Base callback manager for LangChain.N)rRinheritable_tagsrSinheritable_metadatac||_|xsg|_||_|xsg|_|xsg|_|xsi|_|xsi|_y)arInitialize callback manager. Args: handlers: The handlers. inheritable_handlers: The inheritable handlers. parent_run_id: The parent run ID. tags: The tags. inheritable_tags: The inheritable tags. metadata: The metadata. inheritable_metadata: The inheritable metadata. Nhandlersinheritable_handlersrrRrrSr)rrrrrRrrSrs r__init__zBaseCallbackManager.__init__xsU,4< &B !+8JB 0 6B B $8$>B!r c f|j|jj|jj|j|j j|j j|jj|jjS)z&Return a copy of the callback manager.r) r\rcopyrrrRrrSrrs rrzBaseCallbackManager.copys~~]]'')!%!:!:!?!?!A,,!!22779]]'')!%!:!:!?!?!A  r c 6tt|jt|jz}tt|jt|jz}|j |j xs |j ||tt|j |j ztt|j|jzi|j|ji|j|jS)aMerge the callback manager with another callback manager. May be overwritten in subclasses. Primarily used internally within merge_configs. Returns: The merged callback manager of the same type as the current object. Example: Merging two callback managers. ```python from langchain_core.callbacks.manager import ( CallbackManager, trace_as_chain_group, ) from langchain_core.callbacks.stdout import StdOutCallbackHandler manager = CallbackManager(handlers=[StdOutCallbackHandler()], tags=["tag2"]) with trace_as_chain_group("My Group Name", tags=["tag1"]) as group_manager: merged_manager = group_manager.merge(manager) print(merged_manager.handlers) # [ # , # , # ] print(merged_manager.tags) # ['tag2', 'tag1'] ``` )rrrrRrrSr) listsetrrr\rrRrrSr)rothercombined_handlerscombined_inheritables rmergezBaseCallbackManager.mergesB!T]]!3c%..6I!IJ# )) *S1K1K-L L ~~,,C0C0C&!5c$))ejj012!#d&;&;e>T>T&T"UV--.."++",,"  r cy)z&Whether the callback manager is async.Frrs ris_asynczBaseCallbackManager.is_asyncrr c||jvr|jj||r+||jvr|jj|yyy)zAdd a handler to the callback manager. Args: handler: The handler to add. inherit: Whether to inherit the handler. N)rappendrrhandlerinherits r add_handlerzBaseCallbackManager.add_handlersN $-- ' MM  ) wd&?&??  % % , ,W 5@7r c||jvr|jj|||jvr|jj|yy)zoRemove a handler from the callback manager. Args: handler: The handler to remove. N)rremover)rrs rremove_handlerz"BaseCallbackManager.remove_handlersI dmm # MM  ) d// /  % % , ,W 5 0r cTg|_g|_|D]}|j||y)zSet handlers as the only handlers on the callback manager. Args: handlers: The handlers to set. inherit: Whether to inherit the handlers. rN)rrr)rrrrs r set_handlersz BaseCallbackManager.set_handlerss5 $&! 7G   Wg  6 7r c,|j|g|y)zSet handler as the only handler on the callback manager. Args: handler: The handler to set. inherit: Whether to inherit the handler. rN)rrs r set_handlerzBaseCallbackManager.set_handler s 7)W5r c|D]#}||jvs|j|g%|jj||r|jj|yy)zAdd tags to the callback manager. Args: tags: The tags to add. inherit: Whether to inherit the tags. N)rR remove_tagsextendr)rrRrtags radd_tagszBaseCallbackManager.add_tagss_ (Cdii  #' (    ! ! ( ( . r c|D]U}||jvr|jj|||jvs;|jj|Wy)zdRemove tags from the callback manager. Args: tags: The tags to remove. N)rRrr)rrRrs rrzBaseCallbackManager.remove_tags)sS  2Cdii   %d+++%%,,S1  2r cv|jj||r|jj|yy)zAdd metadata to the callback manager. Args: metadata: The metadata to add. inherit: Whether to inherit the metadata. N)rSupdater)rrSrs r add_metadataz BaseCallbackManager.add_metadata5s2 X&   % % , ,X 6 r c|D]:}|jj|d|jj|d<y)zhRemove metadata from the callback manager. Args: keys: The keys to remove. N)rSpopr)rkeyskeys rremove_metadataz#BaseCallbackManager.remove_metadataDs=  5C MM  c4 (  % % ) )#t 4 5r )NN)rlist[BaseCallbackHandler]rz list[BaseCallbackHandler] | Nonerr&rRrqrrqrSrrrrrr'r)r'r )rrr'r r)T)rrrrr'r)rrr'r)rrrrr'r)rRrprrr'r)rRrpr'r)rSrGrrr'r)rrpr'r)r)r*r+r,rrrrrrrrrrrrrrr rrrusb. BF%) ? "&-1*.6:?+???# ?  ?+?(?4? ?@  4 l6$66  6 67+77  7& 6$ 6 6  6 ///  /$ 2 7  7 7  75r r)(r, __future__rloggingtypingrrcollections.abcruuidrtenacityr typing_extensionsr langchain_core.agentsr r langchain_core.documentsr langchain_core.messagesrlangchain_core.outputsrrr getLoggerr)_LOGGERrr.r;rIrPrurrrr Callbacksrr rrs*"%('&>13VV '  H %# # L8 8 vE E P# # LC C L: : z1 1hU .U p W5.W5t $ %(; ;d B r