o Jjg@s.dZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZdd lmZdd lmZdd lmZmZmZdd lmZdd lm Z ddl!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(GdddeZ)e ddddGdddeZ*e dddGdddeZ+dS)zOAttempt to implement MRKL systems as described in arxiv.org/pdf/2205.00445.pdf.) annotations)AnyCallableList NamedTupleOptionalSequence) deprecated)BaseCallbackManager)BaseLanguageModel)PromptTemplate)BaseToolTool)render_text_description)Field)Agent AgentExecutorAgentOutputParser) AgentTypeMRKLOutputParser)FORMAT_INSTRUCTIONSPREFIXSUFFIX)validate_tools_single_input)LLMChainc@s*eZdZUdZded<ded<ded<dS) ChainConfigzConfiguration for a chain to use in MRKL system. Parameters: action_name: Name of the action. action: Action function to call. action_description: Description of the action. str action_nameractionaction_descriptionN)__name__ __module__ __qualname____doc____annotations__r&r&R/var/www/html/zoom/venv/lib/python3.10/site-packages/langchain/agents/mrkl/base.pyrs  rz0.1.0create_react_agentz1.0) alternativeremovalcseZdZUdZeedZded<ed'dd Z e d(d d Z e d(d dZ e d(ddZ eeeedfd)ddZeddeeedfd*d"d#Zed+fd%d& ZZS), ZeroShotAgentzcAgent for the MRKL chain. Parameters: output_parser: Output parser for the agent. )default_factoryr output_parserkwargsrreturncKstS)Nr)clsr.r&r&r'_get_default_output_parser/sz(ZeroShotAgent._get_default_output_parserrcCstjS)z Return Identifier of agent type.)rZERO_SHOT_REACT_DESCRIPTIONselfr&r&r' _agent_type3szZeroShotAgent._agent_typecCdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r&r3r&r&r'observation_prefix8z ZeroShotAgent.observation_prefixcCr6)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r&r3r&r&r' llm_prefixAr8zZeroShotAgent.llm_prefixNtoolsSequence[BaseTool]prefixsuffixformat_instructionsinput_variablesOptional[List[str]]r c CsXtt|}ddd|D}|j|d}d||||g}|r't||dSt|S)aCreate prompt in the style of the zero shot agent. Args: tools: List of tools the agent will have access to, used to format the prompt. prefix: String to put before the list of tools. Defaults to PREFIX. suffix: String to put after the list of tools. Defaults to SUFFIX. format_instructions: Instructions on how to use the tools. Defaults to FORMAT_INSTRUCTIONS input_variables: List of input variables the final prompt will expect. Defaults to None. Returns: A PromptTemplate with the template assembled from the pieces here. z, cSg|]}|jqSr&name.0toolr&r&r' cz/ZeroShotAgent.create_prompt..) tool_namesz )templater?)rlistjoinformatr from_template) r0r:r<r=r>r? tool_stringsrIrJr&r&r' create_promptJs    zZeroShotAgent.create_promptllmr callback_managerOptional[BaseCallbackManager]Optional[AgentOutputParser]rc Ks\|||j|||||d} t|| |d} dd|D} |p"|} |d| | | d| S)aConstruct an agent from an LLM and tools. Args: llm: The LLM to use as the agent LLM. tools: The tools to use. callback_manager: The callback manager to use. Defaults to None. output_parser: The output parser to use. Defaults to None. prefix: The prefix to use. Defaults to PREFIX. suffix: The suffix to use. Defaults to SUFFIX. format_instructions: The format instructions to use. Defaults to FORMAT_INSTRUCTIONS. input_variables: The input variables to use. Defaults to None. kwargs: Additional parameters to pass to the agent. )r<r=r>r?)rQpromptrRcSrAr&rBrDr&r&r'rGrHz4ZeroShotAgent.from_llm_and_tools..) llm_chain allowed_toolsr-Nr&)_validate_toolsrPrr1)r0rQr:rRr-r<r=r>r?r.rUrVrI_output_parserr&r&r'from_llm_and_toolsjs,  z ZeroShotAgent.from_llm_and_toolsNonecs`t|j|t|dkrtd|jd|D]}|jdur'td|jdqt|dS)NrzGot no tools for z%. At least one tool must be provided.z Got a tool zN without a description. For this agent, a description must always be provided.)rr!len ValueError descriptionrCsuperrX)r0r:rF __class__r&r'rXs     zZeroShotAgent._validate_tools)r.rr/r)r/r) r:r;r<rr=rr>rr?r@r/r )rQr r:r;rRrSr-rTr<rr=rr>rr?r@r.rr/r)r:r;r/r[)r!r"r#r$rrr-r% classmethodr1propertyr5r7r9rrrrPrZrX __classcell__r&r&r`r'r+%s6     0r+)r*c@seZdZdZed d d Zd S) MRKLChainz&Chain that implements the MRKL system.rQr chainsList[ChainConfig]r.rr/rcKs.dd|D}t||}|d||d|S)aUser-friendly way to initialize the MRKL chain. This is intended to be an easy way to get up and running with the MRKL chain. Args: llm: The LLM to use as the agent LLM. chains: The chains the MRKL system has access to. **kwargs: parameters to be passed to initialization. Returns: An initialized MRKL chain. cSs g|] }t|j|j|jdqS))rCfuncr^)rrrr )rEcr&r&r'rGsz)MRKLChain.from_chains..)agentr:Nr&)r+rZ)r0rQrfr.r:rjr&r&r' from_chainss  zMRKLChain.from_chainsN)rQr rfrgr.rr/r)r!r"r#r$rbrkr&r&r&r'resreN),r$ __future__rtypingrrrrrrlangchain_core._apir langchain_core.callbacksr langchain_core.language_modelsr langchain_core.promptsr langchain_core.toolsr rlangchain_core.tools.renderrpydanticrlangchain.agents.agentrrrlangchain.agents.agent_typesr#langchain.agents.mrkl.output_parserrlangchain.agents.mrkl.promptrrrlangchain.agents.utilsrlangchain.chainsrrr+rer&r&r&r's,