U hv@sdZddlmZddlZddlZddlZddlZddlmZddl 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/m0Z0ddl1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9ddl:m;Z;ddlm?Z?ddl@mAZAddlBmCZCddlDmEZEddlFmGZGddlHmIZIddlJmKZKeLeMZNGddde4ZOGddde4ZPGd d!d!e,eeefZQGd"d#d#e,eeeefZRGd$d%d%eOZSGd&d'd'ePZTed(d)d*d+Gd,d-d-eOZUed(d)d*d+Gd.d/d/eOZVGd0d1d1e=ZWeeeeefZXeeSeTfZYGd2d3d3eGZZdS)4zEChain 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) BaseModelroot_validator)RunnableRunnableConfig ensure_config) AddableDict)BaseTool)get_color_mapping)AgentExecutorIterator) AgentType) InvalidTool)Chain)LLMChain)asyncio_timeoutcseZdZdZeddddZddddZed-d d d d dddZed.d d d d dddZ eeddddZ dd d ddddZ e d/dddd ddddZ eddd d!Zd d"d#fd$d% Zd&d'd(d)d*Zd"dd+d,ZZS)0BaseSingleActionAgentzBase Single Action Agent class. List[str]returncCsdgSReturn values of the agent.outputselfr6r6:/tmp/pip-unpacked-wheel-bo69hh5q/langchain/agents/agent.py return_values>sz#BaseSingleActionAgent.return_valuesOptional[List[str]]cCsdSNr6r7r6r6r9get_allowed_toolsCsz'BaseSingleActionAgent.get_allowed_toolsNList[Tuple[AgentAction, str]]rrUnion[AgentAction, AgentFinish]intermediate_steps callbackskwargsr2cKsdS)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. Nr6r8rArBrCr6r6r9planFszBaseSingleActionAgent.plancsdS)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. Nr6rEr6r6r9aplanYszBaseSingleActionAgent.aplancCsdS7Return the input keys. :meta private: Nr6r7r6r6r9 input_keyslsz BaseSingleActionAgent.input_keysstrrearly_stopping_methodrArCr2cKs*|dkrtddidStd|ddS)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 ``Nr ValueErrorr8rNrArCr6r6r9return_stopped_responsets z-BaseSingleActionAgent.return_stopped_responserSequence[BaseTool]Optional[BaseCallbackManager])llmtoolscallback_managerrCr2cKstdS)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. NNotImplementedError)clsr[r\r]rCr6r6r9from_llm_and_toolssz(BaseSingleActionAgent.from_llm_and_toolscCstdSz#Return Identifier of an agent type.Nr^r7r6r6r9 _agent_typesz!BaseSingleActionAgent._agent_typerrCr2c s\t}z |j}Wntk r,d}YnXt|trHt|j|d<n|dk rX||d<|S)z{Return dictionary representation of agent. Returns: Dict: Dictionary representation of agent. N_type)superdictrcr_ isinstancer*rLvalue)r8rC_dictre __class__r6r9rgs    zBaseSingleActionAgent.dictUnion[Path, str]None file_pathr2c Cst|trt|}n|}|j}|jddd|}d|krLtd|d|jdkr~t|d}t j ||dd W5QRXnB|j d rt|d}t j ||d d W5QRXnt |d dS)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_okreAgent z does not support saving.jsonwindentz.yamlz.ymlFZdefault_flow_style must be json or yamlN)rhrLrparentmkdirrgr_suffixopenjsondumpendswithyamlrV)r8rp save_pathdirectory_path agent_dictfr6r6r9saves      zBaseSingleActionAgent.savecCsiS#Return logging kwargs for tool run.r6r7r6r6r9tool_run_logging_kwargssz-BaseSingleActionAgent.tool_run_logging_kwargs)N)N)N)__name__ __module__ __qualname____doc__propertyr:r=rrFrHrKrX classmethodrarcrgrr __classcell__r6r6rkr9r/;s,#r/cseZdZdZeddddZddddZed'd d d d dddZed(d d d d dddZ eeddddZ dd d ddddZ eddddZ d ddfdd Z d d!d"d#d$Zddd%d&ZZS))BaseMultiActionAgentzBase Multi Action Agent class.r0r1cCsdgSr3r6r7r6r6r9r:sz"BaseMultiActionAgent.return_valuesr;cCsdS)z]Get allowed tools. Returns: Optional[List[str]]: Allowed tools. Nr6r7r6r6r9r=sz&BaseMultiActionAgent.get_allowed_toolsNr>rr%Union[List[AgentAction], AgentFinish]r@cKsdS)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. Nr6rEr6r6r9rFszBaseMultiActionAgent.plancsdS)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. Nr6rEr6r6r9rHszBaseMultiActionAgent.aplancCsdSrIr6r7r6r6r9rKszBaseMultiActionAgent.input_keysrLrrMcKs*|dkrtddidStd|ddS)rOrPr5z$Agent stopped due to max iterations.rRrSrTNrUrWr6r6r9rX#s  z,BaseMultiActionAgent.return_stopped_responsecCstdSrbr^r7r6r6r9rc?sz BaseMultiActionAgent._agent_typerrdc s6t}zt|j|d<Wntk r0YnX|S)*Return dictionary representation of agent.re)rfrgrLrcr_r8rCrjrkr6r9rgDs  zBaseMultiActionAgent.dictrmrnroc Cst|trt|}n|}|}d|kr8td|d|j}|jddd|jdkr~t|d}t j ||dd W5QRXnB|j d rt|d}t j ||d d W5QRXnt |d dS)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") rertz does not support saving.TrqrurvrwrxrzFr{r|N)rhrLrrgr_r}r~rrrrrrrV)r8rprrrrr6r6r9rMs      zBaseMultiActionAgent.savecCsiSrr6r7r6r6r9rtsz,BaseMultiActionAgent.tool_run_logging_kwargs)N)N)rrrrrr:r=rrFrHrKrXrcrgrrrr6r6rkr9rs& 'rc@s$eZdZdZedddddZdS)AgentOutputParserz=Base class for parsing agent output into agent action/finish.rLr?textr2cCsdS)z$Parse text into agent action/finish.Nr6r8rr6r6r9parse}szAgentOutputParser.parseNrrrrrrr6r6r6r9rzsrc@s$eZdZdZedddddZdS)MultiActionAgentOutputParserzBase class for parsing agent output into agent actions/finish. This is used for agents that can return multiple actions. rLrrcCsdS)zParse text into agent actions/finish. Args: text: Text to parse. Returns: Union[List[AgentAction], AgentFinish]: List of agent actions or agent finish. Nr6rr6r6r9rsz"MultiActionAgentOutputParser.parseNrr6r6r6r9rsrc@seZdZUdZded<gZded<gZded<dZded <Gd d d Ze dd d dZ e dd ddZ ddddddddZ ddddddddZ dS) RunnableAgentAgent powered by Runnables.z/Runnable[dict, Union[AgentAction, AgentFinish]]runnabler0input_keys_argreturn_keys_argTboolstream_runnablec@seZdZdZdS)zRunnableAgent.ConfigTNrrrZarbitrary_types_allowedr6r6r6r9Configsrr1cCs|jSr4rr7r6r6r9r:szRunnableAgent.return_valuescCs|jS)zReturn the input keys.rr7r6r6r9rKszRunnableAgent.input_keysNr>rrr?r@cKs`|d|i}d}|jrH|jj|d|idD]}|dkr<|}q*||7}q*n|jj|d|id}|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. rANrBconfigrrstreamZinvoker8rArBrCinputs final_outputchunkr6r6r9rFs  zRunnableAgent.plancsp|d|i}d}|jrR|jj|d|id2z"3dHW}|dkrD|}q*||7}q*6n|jj|d|idIdH}|SaRAsync 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. rANrBrrrastreamZainvokerr6r6r9rHs  zRunnableAgent.aplan)N)Nrrrr__annotations__rrrrrr:rKrFrHr6r6r6r9rs    'rc@seZdZUdZded<gZded<gZded<dZded <Gd d d Ze dd d dZ e dd ddZ ddddddddZ ddddddddZ dS)RunnableMultiActionAgentrz5Runnable[dict, Union[List[AgentAction], AgentFinish]]rr0rrTrrc@seZdZdZdS)zRunnableMultiActionAgent.ConfigTNrr6r6r6r9rsrr1cCs|jSrrr7r6r6r9r:sz&RunnableMultiActionAgent.return_valuescCs|jS)QReturn the input keys. Returns: List of input keys. rr7r6r6r9rKsz#RunnableMultiActionAgent.input_keysNr>rrrr@cKs`|d|i}d}|jrH|jj|d|idD]}|dkr<|}q*||7}q*n|jj|d|id}|Srrrr6r6r9rF$s  zRunnableMultiActionAgent.plancsp|d|i}d}|jrR|jj|d|id2z"3dHW}|dkrD|}q*||7}q*6n|jj|d|idIdH}|Srrrr6r6r9rHKs  zRunnableMultiActionAgent.aplan)N)Nrr6r6r6r9rs     *rz0.1.0zpUse new agent constructor methods like create_react_agent, create_json_agent, create_structured_chat_agent, etc.z1.0)messageZremovalcseZdZUdZded<ded<ded<eddd d Zd d d fdd Zdddd ddddZdddd ddddZ d dddZ Z S)LLMSingleActionAgentz$Base class for single action agents.r- llm_chainr output_parserr0stopr1cCstt|jjdhS)rrAlistsetrrKr7r6r6r9rKszLLMSingleActionAgent.input_keysrrrdc st}|d=|Srrrfrgrrkr6r9rgs zLLMSingleActionAgent.dictNr>rr?r@cKs(|jjf||j|d|}|j|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. rArrB)rrunrrrr8rArBrCr5r6r6r9rFszLLMSingleActionAgent.plancs.|jjf||j|d|IdH}|j|S)rGrN)rarunrrrrr6r6r9rHs zLLMSingleActionAgent.aplancCs"dt|jdkrdn|jddS)rrRr llm_prefixobservation_prefix)lenrr7r6r6r9rsz,LLMSingleActionAgent.tool_run_logging_kwargs)N)N) rrrrrrrKrgrFrHrrr6r6rkr9rws  rc seZdZUdZded<ded<dZded<d d d fd d ZddddZeddddZ dddddZ eddddZ dddddZ dLddd d d!d"d#Z dMddd d d!d$d%Zdd d&d'd(d)Zeddd*d+Zed,d-d.d d d/d0d1Zeeddd2d3Zeeddd4d5Zeed6d7d8d9d:Zed6d;d8dd?ZedNd@d6dAdBd ddCdDdEZddd dFdGdHdIZd ddJdKZZS)OAgentzAgent 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_toolsrrrdc st}|d=|Srrrrkr6r9rgs z Agent.dictr1cCs|jS)zGet allowed tools.)rr7r6r6r9r=szAgent.get_allowed_toolsr0cCsdgSr3r6r7r6r6r9r:szAgent.return_valuesrLrcCs tddS)zsFix the text. Args: text: Text to fix. Returns: str: Fixed text. z(fix_text not implemented for this agent.NrVrr6r6r9 _fix_texts zAgent._fix_textcCs d|jd|jgS)N z )rrstripr7r6r6r9_stopsz Agent._stopr>zUnion[str, List[BaseMessage]]rAr2cCs<d}|D].\}}||j7}|d|j|d|j7}q|S)zJConstruct the scratchpad that lets the agent continue its thought process.rRr)logrr)r8rAthoughtsaction observationr6r6r9_construct_scratchpad s   zAgent._construct_scratchpadrr?r@cKs0|j|f|}|jjfd|i|}|j|S)rDrB)get_full_inputsrpredictrr)r8rArBrC full_inputs full_outputr6r6r9rFsz Agent.plancs@|j|f|}|jjfd|i|IdH}|j|IdH}|S)rGrBN)rrZapredictrZaparse)r8rArBrCrrZ agent_outputr6r6r9rH)sz Agent.aplanDict[str, Any])rArCr2cKs"||}||jd}||}|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)r8rArCr new_inputsrr6r6r9r?s  zAgent.get_full_inputscCstt|jjdhS)rJrrr7r6r6r9rKQszAgent.input_keysFTpreZskip_on_failurevaluesr2cCst|dj}d|jkrptd|jdt|trD|jd7_n,t|tr^|j d7_ nt dt ||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. rrzl`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 ) promptZinput_variablesloggerwarningappendrhr templaterrrVtype)r`rrr6r6r9validate_promptYs     zAgent.validate_promptcCsdS)z&Prefix to append the observation with.Nr6r7r6r6r9rvszAgent.observation_prefixcCsdS)z#Prefix to append the LLM call with.Nr6r7r6r6r9r{szAgent.llm_prefixrYr)r\r2cCsdS)zCreate a prompt for this class. Args: tools: Tools to use. Returns: BasePromptTemplate: Prompt template. Nr6r`r\r6r6r9 create_promptszAgent.create_promptrncCsdS)zgValidate that appropriate tools are passed in. Args: tools: Tools to use. Nr6rr6r6r9_validate_toolsszAgent._validate_toolscKsdS)z)Get default output parser for this class.Nr6)r`rCr6r6r9_get_default_output_parsersz Agent._get_default_output_parserrrZzOptional[AgentOutputParser])r[r\r]rrCr2c KsN||t||||d}dd|D}|p6|}|f|||d|S)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. )r[rr]cSsg|] }|jqSr6name.0toolr6r6r9 sz,Agent.from_llm_and_tools..)rrr)rr-rr) r`r[r\r]rrCrZ tool_namesZ_output_parserr6r6r9ras  zAgent.from_llm_and_toolsrrMc Ks|dkrtddidS|dkrd}|D].\}}||j7}|d|j|d|j7}q&|d7}||jd}||}|jjf|} |j| } t | tr| Std| i| Snt d |d S) 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']. rPr5rQrRgeneraterzB I now need to return a final answer based on the previous steps:rzBearly_stopping_method should be one of `force` or `generate`, got N) rrrrrrrrrrhrV) r8rNrArCrrrrrrZ parsed_outputr6r6r9rXs2     zAgent.return_stopped_responsecCs|j|jdS)rrrr7r6r6r9rszAgent.tool_run_logging_kwargs)N)N)NN)rrrrrrrgr=rr:rrrrFrHrrKr"rrrrrrrrrarXrrr6r6rkr9rsR       #7rc@sReZdZUdZdZded<dZded<ddddd d d Zddd dd d dZdS) ExceptionToolz!Tool that just returns the query. _ExceptionrLrzException tool descriptionNz#Optional[CallbackManagerForToolRun])query run_managerr2cCs|Sr<r6r8rrr6r6r9_runszExceptionTool._runz(Optional[AsyncCallbackManagerForToolRun]cs|Sr<r6rr6r6r9_arun szExceptionTool._arun)N)N) rrrrrrrrrr6r6r6r9rs    rc@seZdZUdZded<ded<dZded<d Zd ed <d Zd ed<dZded<dZ ded<dZ ded<e d|ddddddddZ e ddddddd d!Ze dd"dddd#d$Zed%d&d'd(Zd)d*d+d,d-Zd)d*d+d.d/Zd}ddd0ddddd1d2d3d4Zed5d&d6d7Zed5d&d8d9Zdd:d;dd?dd@dAdBZd~dCdDdEdFdGdHdIZddCdDdJdFdGdKdLZdMdNddOdPZddQdRdRdSdEdNdTdUdVZddQdRdRdSdEdWdTdXdYZddQdRdZdEd[d\d]d^ZddQdRdRdSdJdNdTd_d`ZddQdRdRdSdJdadTdbdcZ ddQdRdZdJd[d\dddeZ!ddRdEdFdfdgdhZ"ddRdJdRdfdidjZ#dkdldmdndoZ$dSdSdpdqdrZ%ddsdtddudvdwdxZ&ddsdtddydvdzd{Z'd S) AgentExecutorzAgent that is using tools.z.zAllowed tools (z!) different than provided tools (cSsg|] }|jqSr6rrr6r6r9rvs))r=rrV)r`rrr\rr6r6r9validate_toolscs zAgentExecutor.validate_tools)rc Cs|d}|rt|trz |j}Wn&tk rH}zd}W5d}~XYnX|ttttfk}| dd}|rt ||d|d<nt ||d|d<|S)zConvert runnable to agent if passed in. Args: values: Values to validate. Returns: Dict: Validated values. rFNrT)rr) getrhr#Z OutputType Exceptionrr rrpoprr)r`rr output_type_Z multi_actionrr6r6r9validate_runnable_agent{s$     z%AgentExecutor.validate_runnable_agentz2Union[BaseSingleActionAgent, BaseMultiActionAgent]r1cCs"t|jtrtt|jS|jSdS)aGType cast self.agent. The .agent attribute type includes Runnable, but is converted to 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. N)rhrr#rRunnableAgentTyper7r6r6r9 _action_agents  zAgentExecutor._action_agentrmrnrocCs tddS)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(...)`Nrr8rpr6r6r9rs zAgentExecutor.savecCs |j|S)zZSave the underlying agent. Args: file_path: Path to save to. )rrrr6r6r9 save_agentszAgentExecutor.save_agent)include_run_infoasync_r))rrBrrr2cCst||||j|dS)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)r8rrBrrr6r6r9iterszAgentExecutor.iterr0cCs|jjS)rJ)rrKr7r6r6r9rKszAgentExecutor.input_keyscCs |jr|jjdgS|jjSdS)z@Return the singular output key. :meta private: rAN)rrr:r7r6r6r9 output_keysszAgentExecutor.output_keysr')rr2cCsdd|jD|S)zLookup tool by name. Args: name: Name of tool. Returns: BaseTool: Tool object. cSsi|] }|j|qSr6rrr6r6r9 sz-AgentExecutor.lookup_tool..)r\)r8rr6r6r9 lookup_tools zAgentExecutor.lookup_toolintfloat) iterations time_elapsedr2cCs4|jdk r||jkrdS|jdk r0||jkr0dSdS)NFT)rr)r8rr r6r6r9_should_continueszAgentExecutor._should_continuerrz$Optional[CallbackManagerForChainRun]r)r5rArr2cCs.|r|j|d|jd|j}|jr*||d<|SNgreen)colorverboserAZon_agent_finishr%r:rr8r5rArrr6r6r9_returns zAgentExecutor._returnz)Optional[AsyncCallbackManagerForChainRun]cs4|r|j|d|jdIdH|j}|jr0||d<|Sr"r&r'r6r6r9_areturns zAgentExecutor._areturnNextStepOutputz1Union[AgentFinish, List[Tuple[AgentAction, str]]]cCs8t|dtr&t|dkst|dSdd|DSdS)NrcSs"g|]}t|tr|j|jfqSr6)rhrrrrar6r6r9r&s z4AgentExecutor._consume_next_step..)rhrrAssertionError)r8rr6r6r9_consume_next_steps z AgentExecutor._consume_next_stepzDict[str, BaseTool]zDict[str, str]r>)name_to_tool_map color_mappingrrArr2c Cs"|dd||||||DS)NcSsg|]}|qSr6r6r,r6r6r9r3sz1AgentExecutor._take_next_step..)r/_iter_next_stepr8r0r1rrArr6r6r9_take_next_step*szAgentExecutor._take_next_stepz4Iterator[Union[AgentFinish, AgentAction, AgentStep]]c csz4||}|jj|fd|r$|ndi|}Wn*tk r^}zt|jtr`|j }nd}|rztdt |t |} t|jtr|j rt |j } t |j } qd} n2t|jt r|j} nt |jr||} ntdtd| | }|r|j|dd |j} tj|jf|jd|r,|ndd | } t|| d VWYdSd}~XYnXt|trv|VdSt|tr|g} n|} | D] } | Vq| D]} |||| |VqdS) Take a single step in the thought-action-observation loop. Override this to take control of how the agent makes and acts on choices. rBNFAn 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`rr#r$r%r$rBrr)_prepare_intermediate_stepsrrF get_childrrhrrrVrL send_to_llmr llm_outputcallableron_agent_actionrrr tool_inputr%rr_perform_agent_action)r8r0r1rrArr5e raise_errorrrtool_run_kwargsactions agent_actionr6r6r9r2?st                 zAgentExecutor._iter_next_steprr)r0r1rHrr2c Cs|r|j|dd|j|krx||j}|j}||j}|j}|rLd|d<|j|jf|j||rh|ndd|} nD|j}t j|jt | df|jd|r|ndd|} t || dS)Nr#r9rRrr:Zrequested_tool_nameZavailable_tool_namesr;) rAr return_directrrrrBr%r=r+rkeysr r8r0r1rHrrrJr$rFrr6r6r9rCs>       z#AgentExecutor._perform_agent_actionc s(|dd||||||2IdHS)Ncsg|z3dHW}|q6Sr<r6r,r6r6r9rsz2AgentExecutor._atake_next_step..)r/_aiter_next_stepr3r6r6r9_atake_next_steps zAgentExecutor._atake_next_stepz9AsyncIterator[Union[AgentFinish, AgentAction, AgentStep]]c sz:|}jj|fdr$ndi|IdH}Wntk rT}ztjtrdj }nd}|r~tdt |t |} tjtr|j rt |j } t |j } qd} n2tjt rȈj} nt jrވ|} ntdtd| | }j} tj|jfjdrndd| IdH} t|| d VWYdSd}~XYnXt|trl|VdSt|tr|g} n|} | D] } | Vqtjfd d | DIdH}|D] }|VqdS) r5rBNFr6r7r8rr:r;csg|]}|qSr6)_aperform_agent_action)rrHr1r0rr8r6r9rsz2AgentExecutor._aiter_next_step..)r<rrHr=rrhrrrVrLr>rr?r@rrrrrBr%rrasyncioZgather)r8r0r1rrArr5rDrErrrFrGrHresultrr6rPr9rMsp                  zAgentExecutor._aiter_next_stepc s|r|j||jddIdH|j|kr||j}|j}||j}|j}|rVd|d<|j|jf|j||rr|ndd|IdH} nJ|j}t j|jt | df|jd|r|ndd|IdH} t || dS)Nr#)r%r$rRrr:rIr;) rAr%rrJrrrrBr=r+rrKrrLr6r6r9rOsF        z$AgentExecutor._aperform_agent_action)rrr2c Csdd|jD}tdd|jDddgd}g}d}d }t}|||r|j|||||d } t| trz|j| ||d S|| t | d kr| d} | | } | d k r|j| ||d S|d 7}t|}q@|j j |j |f|} |j| ||d S) z(Run text through and get agent response.cSsi|] }|j|qSr6rrr6r6r9rMsz'AgentExecutor._call..cSsg|] }|jqSr6rrr6r6r9rPsz'AgentExecutor._call..r#ZredZexcluded_colorsrrr+N)r\r(timer!r4rhrr(extendr_get_tool_returnrrXrN r8rrr0r1rArr  start_timenext_step_outputZnext_step_actionZ tool_returnr5r6r6r9_callFsT     zAgentExecutor._callc sdd|jD}tdd|jDdgd}g}d}d}t}z4t|j4Id H|||r(|j|||||d Id H} t| tr|j | ||d Id HW5QId HRWS| | t | d kr| d} | | } | d k r|j | ||d Id HW5QId HRWS|d 7}t|}qX|j j|j|f|} |j | ||d Id HW5QId HRWSQId HRXWnFttjfk r|j j|j|f|} |j | ||d Id HYSXd S) z.Async run text through and get agent response.cSsi|] }|j|qSr6rrr6r6r9r|sz(AgentExecutor._acall..cSsg|] }|jqSr6rrr6r6r9rsz(AgentExecutor._acall..r#rSrrTNrUr+)r\r(rVr.rr!rNrhrr)rWrrXrrXrN TimeoutErrorrQrYr6r6r9_acallusx       0zAgentExecutor._acallzTuple[AgentAction, str]zOptional[AgentFinish])r[r2cCs`|\}}dd|jD}d}t|jjdkr8|jjd}|j|kr\||jjr\t||idSdS)z&Check if the tool is a returning tool.cSsi|] }|j|qSr6rrr6r6r9rsz2AgentExecutor._get_tool_return..r5rrRN)r\rrr:rrJr)r8r[rHrr0Zreturn_value_keyr6r6r9rXs   zAgentExecutor._get_tool_returnrcCsBt|jtr&|jdkr&||j dSt|jr:||S|SdS)Nr)rhrrr@)r8rAr6r6r9r<s   z)AgentExecutor._prepare_intermediate_stepszUnion[Dict[str, Any], Any]zOptional[RunnableConfig]zIterator[AddableDict])inputrrCr2cksZt|}t|||df|d|d|d|ddd|}|D] }|VqJdS) 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. rBrmetadatarun_namerun_idTrr`rarbZ yield_actionsNr%r)r r8r_rrCiteratorstepr6r6r9rs   zAgentExecutor.streamzAsyncIterator[AddableDict]cKsdt|}t|||df|d|d|d|ddd|}|2z3dHW}|VqJ6dS) 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. rBrr`rarbTrcNrdrer6r6r9rs   zAgentExecutor.astream)N)N)N)N)N)N)N)N)N)N)N)N)N)N)(rrrrrrrrrNrrrr r"r rrrrrrrKrrr!r(r)r/r4r2rCrNrMrOr\r^rXr<rrr6r6r6r9rsx           R-Z,2>"r)[r __future__rrQrloggingrVabcrpathlibrtypingrrrrr r r r r rrrZlangchain_core._apirZlangchain_core.agentsrrrZlangchain_core.callbacksrrrrrrZlangchain_core.exceptionsrZlangchain_core.language_modelsrZlangchain_core.messagesrZlangchain_core.output_parsersrZlangchain_core.promptsrZlangchain_core.prompts.few_shotrZlangchain_core.prompts.promptr Zlangchain_core.pydantic_v1r!r"Zlangchain_core.runnablesr#r$r%Zlangchain_core.runnables.utilsr&Zlangchain_core.toolsr'Zlangchain_core.utils.inputr(Zlangchain.agents.agent_iteratorr)Zlangchain.agents.agent_typesr*Zlangchain.agents.toolsr+Zlangchain.chains.baser,Zlangchain.chains.llmr-Zlangchain.utilities.asyncior. getLoggerrrr/rrrrrrrrr*rrr6r6r6r9sv   4                   + ltS&