9|hdZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z mZmZddlmZddlmZGdd e ZGd d eeZGd d eZy)zBase classes for chain routing.) annotations)ABC)AnyDictListMapping NamedTupleOptional)AsyncCallbackManagerForChainRunCallbackManagerForChainRun Callbacks) ConfigDict)Chainc"eZdZUded<ded<y)Routez Optional[str] destinationDict[str, Any] next_inputsN)__name__ __module__ __qualname____annotations__[/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain/chains/router/base.pyrrsrrcBeZdZdZeddZdddZ d ddZy) RouterChainzHChain that outputs the name of a destination chain and the inputs to it.c ddgS)Nrrrselfs r output_keyszRouterChain.output_keyss }--rNc:|||}t|d|dS)z Route inputs to a destination chain. Args: inputs: inputs to the chain callbacks: callbacks to use for the chain Returns: a Route object  callbacksrr)rr inputsr$results rroutezRouterChain.routes'f 2VM*F=,ABBrchK|j||d{}t|d|dS7w)Nr#rr)acallrr%s raroutezRouterChain.aroute,s9zz&Iz>>VM*F=,ABB?s 202returnz List[str]N)r&rr$r r-r)rrr__doc__propertyr!r(r+rrrrrsDR .. C>BC$C1:C CrrceZdZUdZded< ded< ded< dZd ed < ed d ZeddZ eddZ d ddZ d ddZ y)MultiRouteChainzIUse a single chain to route an input to one of multiple candidate chains.r router_chainzMapping[str, Chain]destination_chainsr default_chainFbool silent_errorsTforbid)arbitrary_types_allowedextrac.|jjS)zWWill be whatever keys the router chain prompt expects. :meta private: )r3 input_keysrs rr<zMultiRouteChain.input_keysEs   +++rcgS)z=Will always return text key. :meta private: rrs rr!zMultiRouteChain.output_keysMs  rNcz|xstj}|j}|jj ||}|j t |jdzt |jz|j|js|j|j|S|j|jvr+|j|j|j|S|jr|j|j|Std|jdNr#z: )verbosez)Received invalid destination chain name '')r get_noop_manager get_childr3r(on_textstrrrr@r5r4r7 ValueErrorr r& run_manager _run_managerr$r(s r_callzMultiRouteChain._callUs* #S&@&Q&Q&S  **, !!'')'D !! "T )C0A0A,B BDLL    %%e&7&79%M M   $"9"9 9=4**5+<+<=!!Y   %%e&7&79%M M;ErTsP%"AA  ' J C%C8VeVr