9|hJdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlZddlmZddlmZmZmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>m?Z?m@Z@ddlAmBZBddlCmDZDddlEmFZFddlGmHZHddlImJZJddlKmLZLddlMmNZNddlOmPZPejeRZSGdde>ZTGd d!e>ZUGd"d#e,eeefZVGd$d%e,eeeefZWGd&d'eTZXGd(d)eUZYed*eDd+,Gd-d.eTZZed*eDd+,Gd/d0eTZ[Gd1d2e:Z\eeeeefZ]eeXeYfZ^Gd3d4eLZ_y)5zEChain that takes in an input and produces an action and action input.) annotationsN)abstractmethod)Path) Any AsyncIteratorCallableDictIteratorListOptionalSequenceTupleUnioncast) 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 U/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain/agents/agent.py return_valuesz#BaseSingleActionAgent.return_values@ zcyNr9r:s r<get_allowed_toolsz'BaseSingleActionAgent.get_allowed_toolsEsr?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. Nr9r;intermediate_steps callbackskwargss r<planzBaseSingleActionAgent.planHr?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. Nr9rEs r<aplanzBaseSingleActionAgent.aplan[cy7Return the input keys. :meta private: Nr9r:s r< input_keysz BaseSingleActionAgent.input_keysnrJr?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. forcer83Agent stopped due to iteration limit or time limit.'Got unsupported early_stopping_method ``r ValueErrorr;early_stopping_methodrFrHs r<return_stopped_responsez-BaseSingleActionAgent.return_stopped_responsevsC( !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_managerrHs r<from_llm_and_toolsz(BaseSingleActionAgent.from_llm_and_toolss &"!r?ctz#Return Identifier of an agent type.rar:s r< _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_dumprjrb isinstancer.strvalue)r;rH_dictrm __class__s r<dictzBaseSingleActionAgent.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_okrmAgent z does not support saving.jsonwindentNz.yamlz.ymlFdefault_flow_style must be json or yaml)rprqrparentmkdirrurbsuffixopenjsondumpendswithyamlr\)r; file_path save_pathdirectory_path agent_dictfs r<savezBaseSingleActionAgent.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.r9r:s r<tool_run_logging_kwargsz-BaseSingleActionAgent.tool_run_logging_kwargss r?return List[str]rOptional[List[str]]rArFList[Tuple[AgentAction, str]]rGrrHrrUnion[AgentAction, AgentFinish]r^rqrFrrHrrr) rdrreSequence[BaseTool]rfOptional[BaseCallbackManager]rHrrr4rrqrHrrr rzUnion[Path, str]rNonerr )__name__ __module__ __qualname____doc__propertyr=rBrrIrMrSr_ classmethodrgrjrurr __classcell__rts@r<r4r4=sP)  $ 9    )   $ $ 9    )   $  ":   < ;? " """8 "  "  ""("""!BFr?r4ceZdZdZed dZd dZe d ddZe d ddZ eed dZ ddZ eddZ dfd Z dd Zdd ZxZS)BaseMultiActionAgentzBase Multi Action Agent class.cdgSr6r9r:s r<r=z"BaseMultiActionAgent.return_valuesr>r?cy)z]Get allowed tools. Returns: Optional[List[str]]: Allowed tools. Nr9r:s r<rBz&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. Nr9rEs r<rIzBaseMultiActionAgent.planrJr?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. Nr9rEs r<rMzBaseMultiActionAgent.aplan rNrOcyrQr9r:s r<rSzBaseMultiActionAgent.input_keysrJr?c F|dk(rtddidStd|d)rUrVr8z$Agent stopped due to max iterations.rXrYrZr[r]s r<r_z,BaseMultiActionAgent.return_stopped_response%s<( !G +*PQSUV V9:O9PPQR r?ctrirar:s r<rjz BaseMultiActionAgent._agent_typeArkr?c vt|} t|j|d<|S#t$rY|SwxYw)*Return dictionary representation of agent.rm)rnrorqrjrbr;rHrsrts r<ruzBaseMultiActionAgent.dictFsI"$  !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") rmrzz does not support saving.Trwr{r|r}r~NrFrr)rprqrrurbrrrrrrrrr\)r;rrrrrs r<rzBaseMultiActionAgent.saveOs " i %YI!IYY[ * $%tf4M&NO O"))TD9   w &i% 3 *a2 3 3    & &'8 9i% C *aEB C C {*?@A A  3 3 C CrciSrr9r:s r<rz,BaseMultiActionAgent.tool_run_logging_kwargsvs  r?rrrArFrrGrrHrr%Union[List[AgentAction], AgentFinish]rrrrr)rrrrrr=rBrrIrMrSr_rjrurrrrs@r<rrs(  $ 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.Nr9r;texts r<parsezAgentOutputParser.parserJr?N)rrqrrrrrrrrr9r?r<rr|sG33r?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. Nr9rs r<rz"MultiActionAgentOutputParser.parserJr?N)rrqrrrr9r?r<rrs     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|jSr7rr:s r<r=zRunnableAgent.return_values###r?c|jS)zReturn the input keys.rr:s r<rSzRunnableAgent.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. rFNrGconfigrrstreaminvoker;rFrGrHinputs final_outputchunks r<rIzRunnableAgent.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. rFNrGrrrastreamainvokers r<rMzRunnableAgent.aplans(JFI35GHI    $}}44 Y7 5  * *e '#(L E)L!%!6!6 Y7"7"L * 26A<A7A5A7.A<.A:/A<5A77A<rrArrrrr__annotations__rrrr) model_configrr=rSrIrMr9r?r<rrs%==/ "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|jSrrr:s r<r=z&RunnableMultiActionAgent.return_valuesrr?c|jS)QReturn the input keys. Returns: List of input keys. rr:s r<rSz#RunnableMultiActionAgent.input_keyss"""r?Nc i|d|i}d}|jr0|jj|d|iD] }||}||z }|S|jj|d|i}|Srrrs r<rIzRunnableMultiActionAgent.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 r<rMzRunnableMultiActionAgent.aplanOs(JFI35GHI    $}}44 Y7 5  * *e '#(L E)L!%!6!6 Y7"7"L *  rrrArrr9r?r<rrs%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.r1 llm_chainr output_parserrstopcZtt|jjdhz S)rrFlistsetrrSr:s r<rSzLLMSingleActionAgent.input_keyss'C1126J5KKLLr?c *t|}|d=|Srrrnrurs r<ruzLLMSingleActionAgent.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. rFrrGr9)rrunrrrr;rFrGrHr8s r<rIzLLMSingleActionAgent.plansO"$## 1    !!''//r?cK|jjd||j|d|d{}|jj |S7w)rLrNr9)rarunrrrrs r<rMzLLMSingleActionAgent.aplans]"+t~~** 1     !!''//  s.AA Ac^dt|jdk(rddS|jddS)rrXr llm_prefixobservation_prefix)lenrr:s r<rz,LLMSingleActionAgent.tool_run_logging_kwargss8(+DII!(;"  AE1  r?rrrArr) rrrrrrrSrurIrMrrrs@r<rr{s /$$$) 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. r1rrrNr allowed_toolsc *t|}|d=|Srrrs r<ruz Agent.dictrr?c|jS)zGet allowed tools.)rr:s r<rBzAgent.get_allowed_toolss!!!r?cdgSr6r9r:s r<r=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.r\rs r< _fix_textzAgent._fix_textsCDDr?cvd|jjd|jjgS)N z )r rstripr:s r<_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.rXr)logr r )r;rFthoughtsaction observations r<_construct_scratchpadzAgent._construct_scratchpads[#5 W FK  "H "T445k]"T__DUV VH Wr?c |j|fi|}|jjdd|i|}|jj |S)rDrGr9)get_full_inputsrpredictrr)r;rFrGrH full_inputs full_outputs r<rIz Agent.plansQ"+d**+=HH ,dnn,,PyPKP !!'' 44r?cK|j|fi|}|jjdd|i|d{}|jj |d{}|S7)7w)rLrGNr9)r"rapredictraparse)r;rFrGrHr$r% agent_outputs r<rMz 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)r r)r;rFrHr new_inputsr$s r<r"zAgent.get_full_inputs=s8--.@A*2DJJG ..:. r?cZtt|jjdhz S)rRr,rr:s r<rSzAgent.input_keysOs' 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_variablesloggerwarningappendrpr!templater rr\type)r;r3s r<validate_promptzAgent.validate_promptWs&& V%;%; ; NN=   " " ) )*< =&.1#99  F$9: !77  !#>tF|n!MNN r?cy)z&Prefix to append the observation with.Nr9r:s r<r zAgent.observation_prefixtrJr?cy)z#Prefix to append the LLM call with.Nr9r:s r<r zAgent.llm_prefixyrJr?cy)zCreate a prompt for this class. Args: tools: Tools to use. Returns: BasePromptTemplate: Prompt template. Nr9rcres r< create_promptzAgent.create_prompt~rJr?cy)zgValidate that appropriate tools are passed in. Args: tools: Tools to use. Nr9r>s r<_validate_toolszAgent._validate_toolss r?c y)z)Get default output parser for this class.Nr9)rcrHs r<_get_default_output_parserz Agent._get_default_output_parserrJr?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. )rdr3rf)rrrr9)rAr1r?namerC) rcrdrerfrrHrtool tool_names_output_parsers r<rgzAgent.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']. rVr8rWrXgeneraterzB 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 r9) rrr r rrr#rrrpr\) r;r^rFrHrrrr-r$r% parsed_outputs r<r_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)rr r r:s r<rzAgent.tool_run_logging_kwargss//"&"9"9  r?rrr)rrqrrq)rFrrzUnion[str, List[BaseMessage]]rAr)rFrrHrrDict[str, Any]rr+r)rerrr)rerrr)rHrrr)NN) rdrrerrfrrzOptional[AgentOutputParser]rHrrrrr)rrrrrrrurBrr=rrr rIrMr"rSr*r:rr r rr?rArCrgr_rrrs@r<rrs/ $$$))-M&-F " E  "? & $5955 5 ) 50 $9  ) ,"?KN $KK'"#85522    88 ;?59 ! ! "! 8 ! 3 !  !  ! ! F5"5:5 5  5n r?rc^eZdZUdZdZded< dZded< d d dZ d d d Zy) ExceptionToolz!Tool that just returns the query. _ExceptionrqrEzException tool descriptionNc|SrAr9r;query run_managers r<_runzExceptionTool._runs  r?cK|SwrAr9rTs r<_arunzExceptionTool._arun s  srA)rUrqrVz#Optional[CallbackManagerForToolRun]rrq)rUrqrVz(Optional[AsyncCallbackManagerForToolRun]rrq) rrrrrErrRrWrYr9r?r<rPrPsn+D#'K'" <@9 AE> r?rPceZdZUdZded< 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. r]FNrT)rr) getrpr" OutputTyperr rr Exceptionpoprr)rcvaluesr] output_type multi_action_rs r<validate_runnable_agentz%AgentExecutor.validate_runnable_agentys 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. )rpr]r"rRunnableAgentTyper:s r< _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(...)`rr;rs r<rzAgentExecutor.saves !  r?c8|jj|S)zZSave the underlying agent. Args: file_path: Path to save to. )rwrrys r< 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. )tagsr|)r-r)r;rrGr|r}s r<iterzAgentExecutor.iters$&%   -   r?c.|jjS)rR)rwrSr:s r<rSzAgentExecutor.input_keyss !!,,,r?cz|jr|jjdgzS|jjS)z@Return the singular output key. :meta private: rF)r^rwr=r:s r< 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. )rerE)r;rErFs r< lookup_toolzAgentExecutor.lookup_tools)-1JJ7D 47==7s)ct|j||jk\ry|j||jk\ryy)NFT)r`ra)r; iterations time_elapseds r<_should_continuezAgentExecutor._should_continues>    *zT=P=P/P  # # / 7 77r?c|r|j|d|j|j}|jr||d<|SNgreen)colorverboserFon_agent_finishrr=r^r;r8rFrVrs r<_returnzAgentExecutor._returnsE   ' 'gt|| ' T++  ) )1CL- .r?cK|r&|j|d|jd{|j}|jr||d<|S7#wrrrs r<_areturnzAgentExecutor._areturns_ --gt||.  ++  ) )1CL- .  s$A A $A ct|dtrt|dk(sJ|dS|Dcgc]+}t|ts|j|j f-c}Scc}w)Nrc)rprr rrr)r;rpas r<_consume_next_stepz AgentExecutor._consume_next_stepsc fRj+ .v;!# ##": 4:./Z9=U1==) s A A c p|j|j|||||Dcgc]}|c}Scc}wrA)r_iter_next_stepr;name_to_tool_map color_mappingrrFrVrs r<_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. rGNFAn 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`rQrrrrrGrr)_prepare_intermediate_stepsrwrI get_childrrprbrr\rq send_to_llmr llm_outputcallableron_agent_actionrrPr tool_inputrrr_perform_agent_action)r;rrrrFrVr8e raise_errorrrtool_run_kwargsactions agent_actions r<rzAgentExecutor._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)NrrrXr rrequested_tool_nameavailable_tool_namesr) rrF return_directrwrrrrrr/rkeysr r;rrrrVrFrrrrs r<rz#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}wwrA)r_aiter_next_steprs r<_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) rrGNFrrrrQrr)rrwrMrrrprbrr\rqrrrrrrrPrrrrrasynciogather_aperform_agent_action)r;rrrrFrVr8rrrrrrrresultrs r<rzAgentExecutor._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)rrrXr rrr) rrrFrrwrrrrr/rrrrs r<rz$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_colorsrrVr)rerEr'timerrrprrextendr _get_tool_returnrwr_r^r;rrVrFrrrFrr start_timenext_step_outputnext_step_action tool_returnr8s r<_callzAgentExecutor._callEs9= 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)rerEr'rr2rarrrprrrr rrwr_r^ TimeoutErrorrrs r<_acallzAgentExecutor._acallts9= 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.r8rrXN)rerEr rwr=rFrr)r;rrrrFrreturn_value_keys r<rzAgentExecutor._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)rprdintr)r;rFs r<rz)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|}|D]}|yw) 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. rGrmetadatarun_namerun_idTrrrr yield_actionsNr$r-rlr;inputrrHiteratorsteps r<rzAgentExecutor.streams v&(   JJ{ #  F#ZZ +ZZ +::h'     DJ sA7A9c 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. rGrrrrTrNrrs r<rzAgentExecutor.astreams"v&(   JJ{ #  F#ZZ +ZZ +::h'    #  $J (s*A-B/B3A>4B7B>BBrA) r]r\rerrGrrHrrr[rN)rpr rr)rz2Union[BaseSingleActionAgent, BaseMultiActionAgent]r) rrrGrr|rr}rrr-r)rErqrr&)rrrfloatrr)r8rrFrrV$Optional[CallbackManagerForChainRun]rrM)r8rrFrrV)Optional[AsyncCallbackManagerForChainRun]rrM)rpNextStepOutputr1Union[AgentFinish, List[Tuple[AgentAction, str]]]) rDict[str, BaseTool]rDict[str, str]rrrFrrVrrr) rrrrrrrFrrVrrz4Iterator[Union[AgentFinish, AgentAction, AgentStep]]) rrrrrrrVrrr) rrrrrrrFrrVrrr) rrrrrrrFrrVrrz9AsyncIterator[Union[AgentFinish, AgentAction, AgentStep]]) rrrrrrrVrrr)rrrVrrrM)rrrVrrr)rzTuple[AgentAction, str]rzOptional[AgentFinish])rFrrr)rUnion[Dict[str, Any], Any]rOptional[RunnableConfig]rHrrzIterator[AddableDict])rrrrrHrrzAsyncIterator[AddableDict])(rrrrrr^r`rar^rbrdrrfr*rirtrrwrr{rrSrrrrrrrrrrrrrrrrrrr9r?r<r[r[s$ 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))   D,0))  $ r?r[)`r __future__rrrloggingrabcrpathlibrtypingrrrr r r r r rrrrlangchain_core._apirlangchain_core.agentsrrrlangchain_core.callbacksrrrrrrlangchain_core.exceptionsrlangchain_core.language_modelsrlangchain_core.messagesrlangchain_core.output_parsersrlangchain_core.promptsrlangchain_core.prompts.few_shotr langchain_core.prompts.promptr!langchain_core.runnablesr"r#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.baser0langchain.chains.llmr1langchain.utilities.asyncior2 getLoggerrr5r4rrrrrrrrPrrvr[r9r?r<r 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-)AABw Ew r?