g3fio2dZddlmZmZmZddlmZmZmZddl m Z ddl m Z ddl mZmZddlmZmZmZmZGdd eZGd d eZed ZGd deZGddeZGddeZGddeZGddeZeezezZGddeZGddeZGddeZ GddeeefZ!y) Human in the loop middleware.)AnyLiteralProtocol) AIMessageToolCall ToolMessage)Runtime) interrupt) NotRequired TypedDict)AgentMiddleware AgentStateContextTStateTc2eZdZUdZeed< eeefed<y)Actionz*Represents an action with a name and args.nameargsN)__name__ __module__ __qualname____doc__str__annotations__dictrk/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain/agents/middleware/human_in_the_loop.pyrr s4 IM sCx.SrrcDeZdZUdZeed< eeefed< eeed<y) ActionRequestz@Represents an action request with a name, args, and description.rr descriptionN) rrrrrrrrr rrrr!r!s,J I1 sCx.SS!!7rr!approveeditrejectcJeZdZUdZeed< eeed< ee ee fed<y) ReviewConfigz$Policy for reviewing a HITL request. action_nameallowed_decisions args_schemaN) rrrrrrlist DecisionTyper rrrrrr(r('s2.GL)):T#s(^,,Trr(c4eZdZUdZeeed< eeed<y) HITLRequestzIRequest for human feedback on a sequence of actions requested by a model.action_requestsreview_configsN)rrrrr,r!rr(rrrr/r/4s!S-((3&&8rr/c"eZdZUdZeded<y)ApproveDecisionz*Response when a human approves the action.r$typeN)rrrrrrrrrr3r3>s4 ) @rr3c.eZdZUdZeded< eed<y) EditDecisionz'Response when a human edits the action.r%r4 edited_actionN)rrrrrrrrrrr6r6Es1 &/=rr6c4eZdZUdZeded< eeed<y)RejectDecisionz)Response when a human rejects the action.r&r4messageN)rrrrrrr rrrrr9r9Rs!3 ( ?  Orr9c"eZdZUdZeeed<y) HITLResponsez#Response payload for a HITLRequest. decisionsN)rrrrr,Decisionrrrrr<r<_s-H~*rr<c.eZdZdZdededeedefdZ y)_DescriptionFactoryz6Callable that generates a description for a tool call. tool_callstateruntimereturncy)z'Generate a description for a tool call.Nr)selfrArBrCs r__call__z_DescriptionFactory.__call__is rN) rrrrrrr rrrGrrrr@r@fs-@ ( : PXHY ^a rr@cVeZdZUdZeeed< eee zed< ee ee fed<y)InterruptOnConfigzConfiguration for an action requiring human in the loop. This is the configuration format used in the `HumanInTheLoopMiddleware.__init__` method. r*r"r+N) rrrrr,r-rr rr@rrrrrrIrInsF L))9S#6677!DT#s(^,,TrrIc eZdZdZdddeeeezfdeddffdZd e d ed e d e e de eeff d Zded e d ede e dzedzffdZd e d e e deeefdzfdZd e d e e deeefdzfdZxZS)HumanInTheLoopMiddlewarerz Tool execution requires approval)description_prefix interrupt_onrLrDNct|i}|jD]C\}}t|tr|dust gd||<-|j ds?|||<E||_||_y)aInitialize the human in the loop middleware. Args: interrupt_on: Mapping of tool name to allowed actions. If a tool doesn't have an entry, it's auto-approved by default. * `True` indicates all decisions are allowed: approve, edit, and reject. * `False` indicates that the tool is auto-approved. * `InterruptOnConfig` indicates the specific decisions allowed for this tool. The `InterruptOnConfig` can include a `description` field (`str` or `Callable`) for custom formatting of the interrupt description. description_prefix: The prefix to use when constructing action requests. This is used to provide context about the tool call and the action being requested. Not used if a tool has a `description` in its `InterruptOnConfig`. Tr#)r*r*N) super__init__items isinstanceboolrIgetrMrL)rFrMrLresolved_configs tool_name tool_config __class__s rrPz!HumanInTheLoopMiddleware.__init__s6 9;&2&8&8&: : "I{+t,$&2C*G3$Y/!45.9 + :-"4rrAconfigrBrCc|d}|d}|jd}t|r ||||}n||}n|jd|d|}t|||} t ||d} | | fS) z9Create an ActionRequest and ReviewConfig for a tool call.rrr"z Tool: z Args: )rrr"r*)r)r*)rTcallablerLr!r() rFrArYrBrCrV tool_argsdescription_valuer"action_request review_configs r_create_action_and_configz2HumanInTheLoopMiddleware._create_action_and_configsf% f% #JJ}5 % &+IugFK  *+K!445Z {(S\R]^K'# %!$%89 },,rdecisionc f|d}|ddk(rd|vr|dfS|ddk(r#d|vr|d}td|d|d |d  dfS|dd k(r>d |vr:|jd xsd|dd|d }t||d|d d}||fSd|d|jdd|dd|d }t|)zUProcess a single decision and return the revised tool call and optional tool message.r*r4r$Nr%r7rArrid)r4rrrcr&r:z!User rejected the tool call for `z ` with id error)contentr tool_call_idstatuszUnexpected human decision: z. Decision type 'z' is not allowed for tool 'z'. Expected one of z# based on the tool's configuration.)rrTr ValueError) rFrarArYr*r7re tool_messagemsgs r_process_decisionz*HumanInTheLoopMiddleware._process_decisionsH##67 F y (Y:K-Kd? " F v %&4E*E$_5M$&v.&v.     F x 'H8I,Ill9-3If4E3FjQZ[_Q`Pab 'v&&t_ L l* *)(4&ll6234((1&(9':;011T V orc^|d}|sytdt|Dd}|r |jsyg}g}g}t|jD]p\}} |jj | dx} '|j | | ||\} } |j| |j| |j|r|syt||} t| d}t|x}t|x}k7rd|d|d }t|g}g}d }t|jD]t\}} ||vrZ|j| d} ||}|d z }|j|| | \}}||j||sR|j|d|j| v||_d|g|iS) zETrigger interrupt flows for relevant tool calls after an `AIMessage`.messagesNc3BK|]}t|ts|yw)N)rRr).0rjs r z7HumanInTheLoopMiddleware.after_model..s\CCQZA[C\sr)r0r1r=zNumber of human decisions (z/) does not match number of hanging tool calls (z).r) nextreversed tool_calls enumeraterMrTr`appendr/r lenrhrk)rFrBrCrm last_ai_msgr0r1interrupt_indicesidxrArYr^r_ hitl_requestr= decisions_leninterrupt_countrjrevised_tool_callsartificial_tool_messages decision_idxrarevised_tool_callris r after_modelz$HumanInTheLoopMiddleware.after_models$\8H+=\^bc +"8"802-/')' (>(>? .NC++// &0ABBO040N0Nvug1-  &&~6%%m4!((- .#+) l+K8 !^ +MCHYDZ1Z [-m_=11@0AE S/ !.068  ' (>(>? 5NC''**9V+<=$\2! 262H2Hi3/!<%0&--.?@,33LA#)))4! 5&"4 [D+CDEErc.K|j||Sw)zKAsync trigger interrupt flows for relevant tool calls after an `AIMessage`.)r)rFrBrCs r aafter_modelz%HumanInTheLoopMiddleware.aafter_modelasw//s)rrrrrrrSrIrPrrr rtupler!r(r`r>r rkrrr __classcell__)rXs@rrKrKs*' #E &53'8 889&5 &5  &5P"-"-""- "- " "- }l* + "-H(((" ( x$ d 22 3 (TGFGFgh6GGFDQTVYQYN]aLaGFR00*1(*;0 c3h$ 0rrKN)"rtypingrrrlangchain_core.messagesrrr langgraph.runtimer langgraph.typesr typing_extensionsr r !langchain.agents.middleware.typesrrrrrr!r-r(r/r3r6r9r>r<r@rIrKrrrrs#))DD%%4[[TYT 8I 823  U9 U9)9AiA 9 PYP \ )N :+9+ ( .U .UbF0vx/?@F0r