@'h2NdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlmZddlmZmZmZmZmZddlZddlmZdd lmZmZmZdd lmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlZ>m?Z?ddl@mAZAddlBmCZCddlDmEZEddlFmGZGddlHmIZIddlJmKZKddlLmMZMddlNmOZOejeQZRGdd e=ZSGd!d"e=ZTGd#d$e+eeefZUGd%d&e+eeVeefZWGd'd(eSZXGd)d*eTZYed+eCd,-Gd.d/eSZZed+eCd,-Gd0d1eSZ[Gd2d3e9Z\eVeeeefZ]eeXeYfZ^Gd4d5eKZ_y)6zEChain that takes in an input and produces an action and action input.) annotationsN)abstractmethod) AsyncIteratorIteratorSequence)Path)AnyCallableOptionalUnioncast) deprecated) AgentAction AgentFinish AgentStep)AsyncCallbackManagerForChainRunAsyncCallbackManagerForToolRunBaseCallbackManagerCallbackManagerForChainRunCallbackManagerForToolRun Callbacks)OutputParserException)BaseLanguageModel) BaseMessage)BaseOutputParser)BasePromptTemplate)FewShotPromptTemplate)PromptTemplate)RunnableRunnableConfig ensure_config) AddableDict)BaseTool)get_color_mapping) BaseModel ConfigDictmodel_validator)Self)AGENT_DEPRECATION_WARNING)AgentExecutorIterator) AgentType) InvalidTool)Chain)LLMChain)asyncio_timeoutceZdZdZed dZddZe d ddZe d ddZ eed dZ ddZ e d ddZ edd Zdfd Zdd Zdd ZxZS)BaseSingleActionAgentzBase Single Action Agent class.cdgSReturn values of the agent.outputselfs ]/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langchain/agents/agent.py return_valuesz#BaseSingleActionAgent.return_values< zcyNr6r7s r9get_allowed_toolsz'BaseSingleActionAgent.get_allowed_toolsAsr<c y)0Given input, decided what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Action specifying what tool to use. Nr6r8intermediate_steps callbackskwargss r9planzBaseSingleActionAgent.planDr<c Kyw)6Async given input, decided what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Action specifying what tool to use. Nr6rBs r9aplanzBaseSingleActionAgent.aplanWcy7Return the input keys. :meta private: Nr6r7s r9 input_keysz BaseSingleActionAgent.input_keysjrGr<c F|dk(rtddidStd|d)Return response when agent has been stopped due to max iterations. Args: early_stopping_method: Method to use for early stopping. intermediate_steps: Steps the LLM has taken to date, along with observations. **kwargs: User inputs. Returns: AgentFinish: Agent finish object. Raises: ValueError: If `early_stopping_method` is not supported. forcer53Agent stopped due to iteration limit or time limit.'Got unsupported early_stopping_method ``r ValueErrorr8early_stopping_methodrCrEs r9return_stopped_responsez-BaseSingleActionAgent.return_stopped_responsersC( !G +PQSU 9:O9PPQR r<c t)a.Construct an agent from an LLM and tools. Args: llm: Language model to use. tools: Tools to use. callback_manager: Callback manager to use. kwargs: Additional arguments. Returns: BaseSingleActionAgent: Agent object. NotImplementedError)clsllmtoolscallback_managerrEs r9from_llm_and_toolsz(BaseSingleActionAgent.from_llm_and_toolss &"!r<ctz#Return Identifier of an agent type.r^r7s r9 _agent_typez!BaseSingleActionAgent._agent_type "!r<c t|} |j}t |t rt |j|d<|S|||d<|S#t$rd}Y@wxYw)z{Return dictionary representation of agent. Returns: Dict: Dictionary representation of agent. N_type)super model_dumprgr_ isinstancer+strvalue)r8rE_dictrj __class__s r9dictzBaseSingleActionAgent.dictss "$ $$E eY ' -E'N  "E'N # E s A AAct|tr t|}n|}|j}|j dd|j }d|vrt d|d|jdk(r.t|d5}tj||d d d d y |jjd r.t|d5}tj||d d d d y t|d#1swYy xYw#1swYy xYw)zSave the agent. Args: file_path: Path to file to save the agent to. Example: .. code-block:: python # If working with agent executor agent.agent.save(file_path="path/agent.yaml") Tparentsexist_okrjAgent z does not support saving.jsonwindentNz.yamlz.ymlFdefault_flow_style must be json or yaml)rmrnrparentmkdirrrr_suffixopenjsondumpendswithyamlrY)r8 file_path save_pathdirectory_path agent_dictfs r9savezBaseSingleActionAgent.saves  i %YI!I"))TD9YY[ * $%tf4L&MN N   w &i% 3 *a2 3 3    & &'8 9i% C *aEB C C {*?@A A  3 3 C C<C5D5C>D ciS#Return logging kwargs for tool run.r6r7s r9tool_run_logging_kwargsz-BaseSingleActionAgent.tool_run_logging_kwargss r<return list[str]rOptional[list[str]]r>rClist[tuple[AgentAction, str]]rDrrEr rUnion[AgentAction, AgentFinish]r[rnrCrrEr rr) rarrbSequence[BaseTool]rcOptional[BaseCallbackManager]rEr rr1rrnrEr r builtins.dictrzUnion[Path, str]rNonerr)__name__ __module__ __qualname____doc__propertyr:r?rrFrJrPr\ classmethodrdrgrrrr __classcell__rqs@r9r1r19sP)  $ 9    )   $ $ 9    )   $  ":   < ;? " """8 "  "  ""("""!BFr<r1ceZdZdZed dZd dZe d ddZe d ddZ eed dZ ddZ eddZ dfd Z dd Zdd ZxZS)BaseMultiActionAgentzBase Multi Action Agent class.cdgSr3r6r7s r9r:z"BaseMultiActionAgent.return_valuesr;r<cy)z]Get allowed tools. Returns: Optional[List[str]]: Allowed tools. Nr6r7s r9r?z&BaseMultiActionAgent.get_allowed_toolss r<c y)a5Given input, decided what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with the observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Actions specifying what tool to use. Nr6rBs r9rFzBaseMultiActionAgent.planrGr<c Kyw)a;Async given input, decided what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with the observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Actions specifying what tool to use. Nr6rBs r9rJzBaseMultiActionAgent.aplanrKrLcyrNr6r7s r9rPzBaseMultiActionAgent.input_keysrGr<c F|dk(rtddidStd|d)rRrSr5z$Agent stopped due to max iterations.rUrVrWrXrZs r9r\z,BaseMultiActionAgent.return_stopped_response!s<( !G +*PQSUV V9:O9PPQR r<ctrfr^r7s r9rgz BaseMultiActionAgent._agent_type=rhr<c vt|} t|j|d<|S#t$rY|SwxYw)*Return dictionary representation of agent.rj)rkrlrnrgr_r8rErprqs r9rrzBaseMultiActionAgent.dictBsI"$  !1!12E'N #    s + 88ct|tr t|}n|}|j}d|vrt d|d|j }|j dd|jdk(r.t|d5}tj||d d d d y |jjd r.t|d5}tj||d d d d y t|d#1swYy xYw#1swYy xYw)aSave the agent. Args: file_path: Path to file to save the agent to. Raises: NotImplementedError: If agent does not support saving. ValueError: If file_path is not json or yaml. Example: .. code-block:: python # If working with agent executor agent.agent.save(file_path="path/agent.yaml") rjrwz does not support saving.Trtrxryrzr{Nr}Fr~r)rmrnrrrr_rrrrrrrrrY)r8rrrrrs r9rzBaseMultiActionAgent.saveKs " i %YI!IYY[ * $%tf4M&NO O"))TD9   w &i% 3 *a2 3 3    & &'8 9i% C *aEB C C {*?@A A  3 3 C CrciSrr6r7s r9rz,BaseMultiActionAgent.tool_run_logging_kwargsrs  r<rrr>rCrrDrrEr r%Union[list[AgentAction], AgentFinish]rrrrr)rrrrrr:r?rrFrJrPr\rgrrrrrrs@r9rrs(  $ 9    /   $ $ 9    /   $  ":   8""%BNr<rc"eZdZdZeddZy)AgentOutputParserz=Base class for parsing agent output into agent action/finish.cy)z$Parse text into agent action/finish.Nr6r8texts r9parsezAgentOutputParser.parse{rGr<N)rrnrrrrrrrrr6r<r9rrxsG33r<rc"eZdZdZeddZy)MultiActionAgentOutputParserzBase class for parsing agent output into agent actions/finish. This is used for agents that can return multiple actions. cy)zParse text into agent actions/finish. Args: text: Text to parse. Returns: Union[List[AgentAction], AgentFinish]: List of agent actions or agent finish. Nr6rs r9rz"MultiActionAgentOutputParser.parserGr<N)rrnrrrr6r<r9rrs     r<rceZdZUdZded< gZded<gZded<dZded < ed Z e dd Z e dd Z d ddZ d ddZy ) RunnableAgentAgent powered by Runnables.z/Runnable[dict, Union[AgentAction, AgentFinish]]runnablerinput_keys_argreturn_keys_argTboolstream_runnablearbitrary_types_allowedc|jSr4rr7s r9r:zRunnableAgent.return_values###r<c|jS)zReturn the input keys.rr7s r9rPzRunnableAgent.input_keyss"""r<Nc i|d|i}d}|jr0|jj|d|iD] }||}||z }|S|jj|d|i}|SaPBased on past history and current inputs, decide what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with the observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Action specifying what tool to use. rCNrDconfigrrstreaminvoker8rCrDrEinputs final_outputchunks r9rFzRunnableAgent.plans"JFI35GHI   --fk9=U-V *'#(L E)L  * ==// Y?W/XLr<cKi|d|i}d}|jr3|jj|d|i23d{}||}||z }|jj|d|id{}|S786|S7 waRAsync based on past history and current inputs, decide what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Action specifying what tool to use. rCNrDrrrastreamainvokers r9rJzRunnableAgent.aplans(JFI35GHI    $}}44 Y7 5  * *e '#(L E)L!%!6!6 Y7"7"L * 26A<A7A5A7.A<.A:/A<5A77A<rr>rrrrr__annotations__rrrr& model_configrr:rPrFrJr6r<r9rrs%==/ "NI"!#OY# OT  $L$$## $"9"" " ) "N $(9(( (   (r<rceZdZUdZded< gZded<gZded<dZded < ed Z e dd Z e dd Z d ddZ d ddZy )RunnableMultiActionAgentrz5Runnable[dict, Union[list[AgentAction], AgentFinish]]rrrrTrrrc|jSrrr7s r9r:z&RunnableMultiActionAgent.return_valuesrr<c|jS)QReturn the input keys. Returns: List of input keys. rr7s r9rPz#RunnableMultiActionAgent.input_keyss"""r<Nc i|d|i}d}|jr0|jj|d|iD] }||}||z }|S|jj|d|i}|Srrrs r9rFzRunnableMultiActionAgent.plan$s(JFI35GHI   --fk9=U-V *'#(L E)L  * ==// Y?W/XLr<cKi|d|i}d}|jr3|jj|d|i23d{}||}||z }|jj|d|id{}|S786|S7 wrrrs r9rJzRunnableMultiActionAgent.aplanKs(JFI35GHI    $}}44 Y7 5  * *e '#(L E)L!%!6!6 Y7"7"L *  rrr>rrr6r<r9rrs%CC0 "NI"!#OY# OT  $L$$## $%9%% %   %T $)9)) )   )r<rz0.1.0z1.0)messageremovalceZdZUdZded< ded< ded< ed dZdfd Z d dd Z d dd Z dd Z xZ S)LLMSingleActionAgentz$Base class for single action agents.r. llm_chainr output_parserrstopcZtt|jjdhz S)rrClistsetrrPr7s r9rPzLLMSingleActionAgent.input_keyss'C1126J5KKLLr<c *t|}|d=|Srrrkrrrs r9rrzLLMSingleActionAgent.dict  / " r<c |jjd||j|d|}|jj |S)a4Given input, decided what to do. Args: intermediate_steps: Steps the LLM has taken to date, along with the observations. callbacks: Callbacks to run. **kwargs: User inputs. Returns: Action specifying what tool to use. rCrrDr6)rrunrrrr8rCrDrEr5s r9rFzLLMSingleActionAgent.plansO"$## 1    !!''//r<cK|jjd||j|d|d{}|jj |S7w)rIrNr6)rarunrrrrs r9rJzLLMSingleActionAgent.aplans]"+t~~** 1     !!''//  s.AA Ac^dt|jdk(rddS|jddS)rrUr llm_prefixobservation_prefix)lenrr7s r9rz,LLMSingleActionAgent.tool_run_logging_kwargss8(+DII!(;"  AE1  r<rrr>rr) rrrrrrrPrrrFrJrrrs@r9rrws /$$$) O% MM $0900 0 ) 08 $0900 0 ) 02 r<rceZdZUdZded< ded< dZded< dfd Zdd Zed d Z d!d Z ed d Z d"dZ d# d$dZ d# d$dZ d%dZed dZedd&dZeed'dZeed'dZeed(dZed)dZeed*dZe d+ d,dZ d-dZd.dZxZS)/AgentzAgent that calls the language model and deciding the action. This is driven by a LLMChain. The prompt in the LLMChain MUST include a variable called "agent_scratchpad" where the agent can put its intermediary work. r.rrrNr allowed_toolsc *t|}|d=|Srrrs r9rrz Agent.dictrr<c|jS)zGet allowed tools.)rr7s r9r?zAgent.get_allowed_toolss!!!r<cdgSr3r6r7s r9r:zAgent.return_valuesr;r<ctd)zsFix the text. Args: text: Text to fix. Returns: str: Fixed text. z(fix_text not implemented for this agent.rYrs r9 _fix_textzAgent._fix_textsCDDr<cvd|jjd|jjgS)N z )r rstripr7s r9_stopz Agent._stopsB((//12 34**1134 5  r<c~d}|D]5\}}||jz }|d|j|d|jz }7|S)zJConstruct the scratchpad that lets the agent continue its thought process.rUr)logr r )r8rCthoughtsaction observations r9_construct_scratchpadzAgent._construct_scratchpads[#5 W FK  "H "T445k]"T__DUV VH Wr<c |j|fi|}|jjdd|i|}|jj |S)rArDr6)get_full_inputsrpredictrr)r8rCrDrE full_inputs full_outputs r9rFz Agent.plansQ"+d**+=HH ,dnn,,PyPKP !!'' 44r<cK|j|fi|}|jjdd|i|d{}|jj |d{}|S7)7w)rIrDNr6)r rapredictraparse)r8rCrDrEr"r# agent_outputs r9rJz Agent.aplan#sj"+d**+=HH 3DNN33WiW;WW !//66{CC XCs!5A%A!"A%A#A%#A%c R|j|}||jd}i||}|S)a1Create the full inputs for the LLMChain from intermediate steps. Args: intermediate_steps: Steps the LLM has taken to date, along with observations. **kwargs: User inputs. Returns: Dict[str, Any]: Full inputs for the LLMChain. agent_scratchpadr)rr)r8rCrEr new_inputsr"s r9r zAgent.get_full_inputs9s8--.@A*2DJJG ..:. r<cZtt|jjdhz S)rOr*rr7s r9rPzAgent.input_keysKs' C1126H5IIJJr<aftermodecx|jj}d|jvrtj d|jj dt |tr|xjdz c_|St |tr|xjdz c_ |Stdt||S)aFValidate that prompt matches format. Args: values: Values to validate. Returns: Dict: Validated values. Raises: ValueError: If `agent_scratchpad` is not in prompt.input_variables and prompt is not a FewShotPromptTemplate or a PromptTemplate. r*zl`agent_scratchpad` should be a variable in prompt.input_variables. Did not find it, so adding it at the end.z {agent_scratchpad}zGot unexpected prompt type ) rpromptinput_variablesloggerwarningappendrmrtemplaterrrYtype)r8r1s r9validate_promptzAgent.validate_promptSs&& V%;%; ; NN=   " " ) )*< =&.1#99  F$9: !77  !#>tF|n!MNN r<cy)z&Prefix to append the observation with.Nr6r7s r9r zAgent.observation_prefixprGr<cy)z#Prefix to append the LLM call with.Nr6r7s r9r zAgent.llm_prefixurGr<cy)zCreate a prompt for this class. Args: tools: Tools to use. Returns: BasePromptTemplate: Prompt template. Nr6r`rbs r9 create_promptzAgent.create_promptzrGr<cy)zgValidate that appropriate tools are passed in. Args: tools: Tools to use. Nr6r<s r9_validate_toolszAgent._validate_toolss r<c y)z)Get default output parser for this class.Nr6)r`rEs r9_get_default_output_parserz Agent._get_default_output_parserrGr<c |j|t||j||}|Dcgc]}|j}}|xs|j } |d||| d|Scc}w)aOConstruct an agent from an LLM and tools. Args: llm: Language model to use. tools: Tools to use. callback_manager: Callback manager to use. output_parser: Output parser to use. kwargs: Additional arguments. Returns: Agent: Agent object. )rar1rc)rrrr6)r?r.r=namerA) r`rarbrcrrErtool tool_names_output_parsers r9rdzAgent.from_llm_and_toolss* E"$$U+- -22Ddii2 2&J#*H*H*J $(    3sA(c |dk(rtddidS|dk(rd}|D]5\}}||jz }|d|j|d|jz }7|dz }||jd}i||}|j j d i|} |jj| } t| tr| Std| i| Std |) aReturn response when agent has been stopped due to max iterations. Args: early_stopping_method: Method to use for early stopping. intermediate_steps: Steps the LLM has taken to date, along with observations. **kwargs: User inputs. Returns: AgentFinish: Agent finish object. Raises: ValueError: If `early_stopping_method` is not in ['force', 'generate']. rSr5rTrUgeneraterzB I now need to return a final answer based on the previous steps:r)zBearly_stopping_method should be one of `force` or `generate`, got r6) rrr r rrr!rrrmrY) r8r[rCrErrrr+r"r# parsed_outputs r9r\zAgent.return_stopped_responses( !G +PQSU #j 0H'9 # FJJ&001+b@QR  V H/7 KJ2V2z2K0$..00?;?K ..44[AM-5$$#Hk#:KHH,-/ r<c4|j|jdS)rrrr7s r9rzAgent.tool_run_logging_kwargss//"&"9"9  r<rrr)rrnrrn)rCrrzUnion[str, list[BaseMessage]]r>r)rCrrEr rzbuiltins.dict[str, Any]rr(r)rbrrr)rbrrr)rEr rr)NN) rarrbrrcrrzOptional[AgentOutputParser]rEr rr rr)rrrrrrrrr?rr:rrrrFrJr rPr'r8rr r rr=r?rArdr\rrrs@r9r r s/ $$$))-M&-F " E  "? & $5955 5 ) 50 $9  ) ,"?KN $KK'"#85522    88 ;?59 ! ! "! 8 ! 3 !  !  ! ! F5"5:5 5  5n r<r c^eZdZUdZdZded< dZded< d d dZ d d d Zy) ExceptionToolz!Tool that just returns the query. _ExceptionrnrCzException tool descriptionNc|Sr>r6r8query run_managers r9_runzExceptionTool._runs  r<cK|Swr>r6rQs r9_arunzExceptionTool._aruns  sr>)rRrnrSz#Optional[CallbackManagerForToolRun]rrn)rRrnrSz(Optional[AsyncCallbackManagerForToolRun]rrn) rrrrrCrrOrTrVr6r<r9rMrMsn+D#'K'" <@9 AE> r<rMceZdZUdZded< ded< dZded< d Zd ed < d Zd ed< dZded< dZ ded< dZ ded< e d5 d6dZ e dd7dZe de d8dZed9dZd:dZd:dZ d5ddd  d;d!Zedd%Z d5 d?d&Z d5 d@d'Z dAd(Z d5 dBd)Z d5 dCd*Z d5 dDd+Z d5 dEd,Z d5 dFd-Z d5 dGd.Z! d5 dHd/Z" d5 dId0Z# dJd1Z$ dKd2Z% d5 dLd3Z& d5 dMd4Z'y )N AgentExecutorzAgent that is using tools.C'Dd 'D&EQH *G(Es B 'B beforec<|jd}|rpt|tr` |j}|tt t tfk(}|jdd}|rt|||d<|St|||d<|S#t$r }d}Yd}~Hd}~wwxYw)zConvert runnable to agent if passed in. Args: values: Values to validate. Returns: Dict: Validated values. rZFNrT)rr) getrmr OutputTyper rrr Exceptionpoprr)r`valuesrZ output_type multi_action_rs r9validate_runnable_agentz%AgentExecutor.validate_runnable_agentus 7# Zx0 T#..  +eD4E{4R.SS $jj):DAO":"O#w #0"O#w  %$  %s B BBBct|jtrtt|jS|jS)aCType cast self.agent. If the `agent` attribute is a Runnable, it will be converted one of RunnableAgentType in the validate_runnable_agent root_validator. To support instantiating with a Runnable, here we explicitly cast the type to reflect the changes made in the root_validator. )rmrZrr RunnableAgentTyper7s r9 _action_agentzAgentExecutor._action_agents. djj( +)4::6 6:: r<ctd)zRaise error - saving not supported for Agent Executors. Args: file_path: Path to save to. Raises: ValueError: Saving not supported for agent executors. zpSaving not supported for agent executors. If you are trying to save the agent, please use the `.save_agent(...)`rr8rs r9rzAgentExecutor.saves !  r<c8|jj|S)zZSave the underlying agent. Args: file_path: Path to save to. )rtrrvs r9 save_agentzAgentExecutor.save_agents !!&&y11r<)include_run_infoasync_c6t||||j|S)agEnables iteration over steps taken to reach final output. Args: inputs: Inputs to the agent. callbacks: Callbacks to run. include_run_info: Whether to include run info. async_: Whether to run async. (Ignored) Returns: AgentExecutorIterator: Agent executor iterator object. )tagsry)r*r|)r8rrDryrzs r9iterzAgentExecutor.iters$&%   -   r<c.|jjS)rO)rtrPr7s r9rPzAgentExecutor.input_keyss !!,,,r<cz|jr|jjdgzS|jjS)z@Return the singular output key. :meta private: rC)r[rtr:r7s r9 output_keyszAgentExecutor.output_keyss:  ) )%%337K6LL L%%33 3r<c\|jDcic]}|j|c}|Scc}w)zLookup tool by name. Args: name: Name of tool. Returns: BaseTool: Tool object. )rbrC)r8rCrDs r9 lookup_toolzAgentExecutor.lookup_tools)-1JJ7D 47==7s)ct|j||jk\ry|j||jk\ryy)NFT)r]r^)r8 iterations time_elapseds r9_should_continuezAgentExecutor._should_continues>    *zT=P=P/P  # # / 7 77r<c|r|j|d|j|j}|jr||d<|SNgreen)colorverboserCon_agent_finishrr:r[r8r5rCrSrs r9_returnzAgentExecutor._returnsE   ' 'gt|| ' T++  ) )1CL- .r<cK|r&|j|d|jd{|j}|jr||d<|S7#wrrrs r9_areturnzAgentExecutor._areturn s_ --gt||.  ++  ) )1CL- .  s$A A $A ct|dtrt|dk(sJ|dS|Dcgc]+}t|ts|j|j f-c}Scc}w)Nr`)rmrr rrr)r8rmas r9_consume_next_stepz AgentExecutor._consume_next_stepsc fRj+ .v;!# ##": 4:./Z9=U1==) s A A c p|j|j|||||Dcgc]}|c}Scc}wr>)r_iter_next_stepr8name_to_tool_map color_mappingrrCrSrs r9_take_next_stepzAgentExecutor._take_next_step%sN&&--$!&      s 3c#|K |j|}|jj|fd|r|jndi|}t |t,r|yt |tr|g} n|} | D]} | | D]} |j/||| |y#t$r}t |j tr|j }nd}|rtdt|t|} t |j tr:|jr+t|j} t|j} n\d} nYt |j tr |j } n2t|j r|j |} n tdtd| | }|r|j|d |jj!} t#j$|j&f|j(d|r|jndd | } t+|| Yd}~yd}~wwxYww) Take a single step in the thought-action-observation loop. Override this to take control of how the agent makes and acts on choices. rDNFAn output parsing error occurred. In order to pass this error back to the agent and have it try again, pass `handle_parsing_errors=True` to the AgentExecutor. This is the error: Invalid or incomplete response.Got unexpected type of `handle_parsing_errors`rNrrrrrDrr)_prepare_intermediate_stepsrtrF get_childrrmr_rrYrn send_to_llmr llm_outputcallableron_agent_actionrrMr tool_inputrrr_perform_agent_action)r8rrrrCrSr5e raise_errorrrtool_run_kwargsactions agent_actions r9rzAgentExecutor._iter_next_step:s9. !%!A!ABT!U -T'',,"5@+//1dFZ fk *L  fk *hGG# L  # L,, -{  g%% $44d;"&"<"<< #  *+.a&3 q6D$44d;=="%amm"4Kq||,D"BKD66<"88 $445"88;  !QRR {DAF++F'+B"00HHJO--/--!! 5@+//1d  " K6{C C K% s1H<ABAH< H9$F H4/H<4H99H<cf|r|j|d|j|vr||j}|j}||j}|jj }|rd|d<|j |j f|j||r|jndd|} ny|jj }tj |jt|jdf|jd|r|jndd|} t|| S)NrrrUr rrequested_tool_nameavailable_tool_namesr) rrD return_directrtrrrrrr,rkeysr r8rrrrSrDrrrrs r9rz#AgentExecutor._perform_agent_actions;   ' ' G ' D    0 0#L$5$56D ..M!,"3"34E"00HHJO02 -"$(('' 5@+//1d  " K#00HHJO++-+++7+<+<,01A1F1F1H,I   5@+//1d " K +FFr<c K|j|j|||||2cgc3d{}| 76c}Scc}wwr>)r_aiter_next_steprs r9_atake_next_stepzAgentExecutor._atake_next_steps]&& $44$!&         s&#A=5 35 =5 = Ac K |j|}|jj|fd|r|jndi|d{}t |t*r|yt |tr|g} n|} | D]} |  t-j.| D cgc]} |j1||| |c} d{}|D]}| y7#t$r}t |j tr|j }nd}|rtdt|t|} t |j tr:|jr+t|j} t|j} n\d} nYt |j tr |j } n2t|j r|j |} n tdtd| | }|jj} t!j"|j$f|j&d|r|jndd| d{7} t)|| Yd}~yd}~wwxYwcc} w7w) rrDNFrrrrNrr)rrtrJrrrmr_rrYrnrrrrrrrMrrrrrasynciogather_aperform_agent_action)r8rrrrCrSr5rrrrrrrresultrs r9rzAgentExecutor._aiter_next_stepsp, !%!A!ABT!U 44--33"5@+//1dFV fk *L  fk *hGG# L  ~~ %, !++$m\;   EK E %# $44d;"&"<"<< #  *+.a&3 q6D$44d;=="%amm"4Kq||,D"BKD66<"88 $445"88;  !QRR {DAF"00HHJO 4  4 4!!! 5@+//1d ! " !K6{C C G# h sgI.AC C  CA I.I&4I.;I+<I. C I#E*IIII.I## I.cK|r&|j||jdd{|j|vr||j}|j}||j}|jj }|rd|d<|j |jf|j||r|jndd|d{} n|jj }tj |jt|jdf|jd|r|jndd|d{} t|| S7277w)Nr)rrrUr rrr) rrrDrrtrrrrr,rrrrs r9rz$AgentExecutor._aperform_agent_actionsq --dll'.      0 0#L$5$56D ..M!,"3"34E"00HHJO02 - ) ''! 5@+//1d ! " !K#00HHJO 2 2 2+7+<+<,01A1F1F1H,I !  5@+//1d !" ! K +FF?  s5$EEBEEBEEEEEc|jDcic]}|j|}}t|jDcgc]}|jc}ddg}g}d}d}tj} |j ||r|j |||||} t | tr|j| ||S|j| t| dk(r,| d} |j| } | |j| ||S|dz }tj| z }|j ||r|jj|j|fi|} |j| ||Scc}wcc}w)z(Run text through and get agent response.rredexcluded_colorsrrSr)rbrCr$timerrrmrrextendr _get_tool_returnrtr\r[r8rrSrDrrrCrr start_timenext_step_outputnext_step_action tool_returnr5s r9_callzAgentExecutor._callAs9= CDIItOCC)#':: .4TYY .%@P =?  YY[ ##J =#33 "' 4  *K8||$&8k$  % %&6 7#$)#3A#6 "334DE *<<#%7[( !OJ99;3L/##J =0<##;;  & &(: >D ||F$6K|PPMD /s E2E7cK|jDcic]}|j|}}t|jDcgc]}|jc}dg}g}d}d}tj} t |j 4d{|j ||r|j|||||d{} t| tr-|j| ||d{cdddd{S|j| t| dk(rE| d} |j| } | -|j| ||d{cdddd{S|dz }tj| z }|j ||r|jj|j |fi|} |j| ||d{cdddd{Scc}wcc}w7d7777777/7!#1d{7swYyxYw#t"t$j"f$rH|jj|j |fi|} |j| ||d{7cYSwxYww)z.Async run text through and get agent response.rrrrNrr)rbrCr$rr/r^rrrmrrrr rrtr\r[ TimeoutErrorrrs r9_acallzAgentExecutor._acallps9= CDIItOCC)#':: .4TYY .  =?  YY[ * &t'>'>?! ! ++J E-1-B-B(%*$/ .C.($""2K@%)]],.(3&3& ! ! !  '--.>?+,1+;A+>(&*&;&;?! ! ! ! Dg223 ?T''??**,>BHF* '   s6I<G%I<G* $I<2H G/ H,H;G2<+H'G5(H+ H7G88H<I<=AH G; H HG=H I<!.H?HG?H HH H$ I</H2H5H8H;H=H?HHH H HHI<HAI90I31I96I<8I99I<cD|\}}|jDcic]}|j|}}d}t|jjdkDr|jjd}|j |vr'||j j rt||idSycc}w)z&Check if the tool is a returning tool.r5rrUN)rbrCr rtr:rDrr)r8rrrrDrreturn_value_keys r9rzAgentExecutor._get_tool_returns%5! k8< CDIItOCC# t!!// 01 4#11??B     0 0 1 12@@"%{3DsBct|jtr|jdkDr||j dSt|jr|j|S|S)Nr)rmraintr)r8rCs r9rz)AgentExecutor._prepare_intermediate_stepss_ t33S 9,,q0%t'C'C&C&EF F d22 3//0BC C% %r<c +Kt|}t|||jdf|jd|jd|jd|jddd|}|Ed{y7w) aEnables streaming over steps taken to reach final output. Args: input: Input to the agent. config: Config to use. kwargs: Additional arguments. Yields: AddableDict: Addable dictionary. rDr|metadatarun_namerun_idTr|rrr yield_actionsNr!r*ri)r8inputrrEiterators r9rzAgentExecutor.streams} v&(   JJ{ #  F#ZZ +ZZ +::h'    sA0A:2A83A:c Kt|}t|||jdf|jd|jd|jd|jddd|}|23d{}|7 6yw) a Async enables streaming over steps taken to reach final output. Args: input: Input to the agent. config: Config to use. kwargs: Additional arguments. Yields: AddableDict: Addable dictionary. rDr|rrrTrNr)r8rrrErsteps r9rzAgentExecutor.astreams"v&(   JJ{ #  F#ZZ +ZZ +::h'    #  $J (s*A-B/B3A>4B7B>BBr>) rZrYrbrrDrrEr rrXrK)rmrrrr )rz2Union[BaseSingleActionAgent, BaseMultiActionAgent]r) rr rDrryrrzrrr*r)rCrnrr#)rrrfloatrr)r5rrCrrS$Optional[CallbackManagerForChainRun]rdict[str, Any])r5rrCrrS)Optional[AsyncCallbackManagerForChainRun]rr)rmNextStepOutputr1Union[AgentFinish, list[tuple[AgentAction, str]]]) rdict[str, BaseTool]rdict[str, str]rrrCrrSrrr) rrrrrrrCrrSrrz4Iterator[Union[AgentFinish, AgentAction, AgentStep]]) rrrrrrrSrrr) rrrrrrrCrrSrrr) rrrrrrrCrrSrrz9AsyncIterator[Union[AgentFinish, AgentAction, AgentStep]]) rrrrrrrSrrr)rrrSrrr)rrrSrrr)rztuple[AgentAction, str]rzOptional[AgentFinish])rCrrr)rUnion[dict[str, Any], Any]rOptional[RunnableConfig]rEr rzIterator[AddableDict])rrrrrEr rzAsyncIterator[AddableDict])(rrrrrr[r]r^r[r_rarrcr'rfrqrrtrrxr}rPrrrrrrrrrrrrrrrrrrr6r<r9rXrXs$ GG3 -&+t+3$&NM&9+/.")3( S   $  K "        2'"#.(#$:    2 $ "'          6--44 > =A  ! :   "BF  ! ?    $  : "=A  - &   :  :   ; 6=A K-K&K K : K : K >Kd=A %G-%G&%G" %G : %G  %GZBF  - &   :  ?   ; 6BF S-S&S S : S ? S CStBF 'G-'G&'G" 'G ? 'G  'GX=A-Q-Q:-Q  -QdBF<<?<  <| 7 $ &"? & & & ,0))   B,0))  $ r<rX)`r __future__rrbuiltinsrloggingrabcrcollections.abcrrrpathlibrtypingr r r r r rlangchain_core._apirlangchain_core.agentsrrrlangchain_core.callbacksrrrrrrlangchain_core.exceptionsrlangchain_core.language_modelsrlangchain_core.messagesrlangchain_core.output_parsersrlangchain_core.promptsrlangchain_core.prompts.few_shotrlangchain_core.prompts.promptrlangchain_core.runnablesrr r!langchain_core.runnables.utilsr"langchain_core.toolsr#langchain_core.utils.inputr$pydanticr%r&r'typing_extensionsr(langchain._api.deprecationr)langchain.agents.agent_iteratorr*langchain.agents.agent_typesr+langchain.agents.toolsr,langchain.chains.baser-langchain.chains.llmr.langchain.utilities.asyncior/ getLoggerrr3r1rrrrrrrr rMrrsrXr6r<r9r sK"  == *EE<</:5A8LL6)8;;"@A2.')7   8 $gIgTR9Rj3({K/G)HI3 U4 ,k9:; *j)jZr3rj  %  P 0P  P f  %  a !a  a H H.eKi?@A-)AABv Ev r<