o Zh @s<dZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZmZmZdd lm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*GdddeZ+e deddGdddeZ,e deddGdddeZ-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_DEPRECATION_WARNING)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'Q/var/www/html/lang_env/lib/python3.10/site-packages/langchain/agents/mrkl/base.pyrs  rz0.1.0z1.0)messageZremovalcseZdZUdZeedZded<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_parser4sz(ZeroShotAgent._get_default_output_parserrcCstjS)z Return Identifier of agent type.)rZZERO_SHOT_REACT_DESCRIPTIONselfr'r'r( _agent_type8szZeroShotAgent._agent_typecCdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r'r1r'r'r(observation_prefix=z ZeroShotAgent.observation_prefixcCr4)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r'r1r'r'r( llm_prefixFr6zZeroShotAgent.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( hz/ZeroShotAgent.create_prompt..) tool_namesz )templater=)rlistjoinformatr Z from_template) r/r8r:r;r<r=Z tool_stringsrGrHr'r'r( create_promptOs    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=)rMpromptrNcSr?r'r@rBr'r'r(rErFz4ZeroShotAgent.from_llm_and_tools..) llm_chainZ allowed_toolsr,Nr')_validate_toolsrLrr0)r/rMr8rNr,r:r;r<r=r-rQrRrGZ_output_parserr'r'r(from_llm_and_toolsos,  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 descriptionrAsuperrS)r/r8rD __class__r'r(rSs     zZeroShotAgent._validate_tools)r-rr.r)r.r) r8r9r:rr;rr<rr=r>r.r )rMr r8r9rNrOr,rPr:rr;rr<rr=r>r-rr.r)r8r9r.rU)r"r#r$r%rrr,r& classmethodr0propertyr3r5r7rrrrLrTrS __classcell__r'r'rZr(r*&s6     0r*c@seZdZdZed d d Zd S) MRKLChainz&Chain that implements the MRKL system.rMr 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))rAfuncrX)rrr r!)rCcr'r'r(rEsz)MRKLChain.from_chains..)agentr8Nr')r*rT)r/rMr`r-r8rdr'r'r( from_chainss  zMRKLChain.from_chainsN)rMr r`rar-rr.r)r"r#r$r%r\rer'r'r'r(r_sr_N).r% __future__rtypingrrrrrrZlangchain_core._apir Zlangchain_core.callbacksr Zlangchain_core.language_modelsr Zlangchain_core.promptsr Zlangchain_core.toolsr rZlangchain_core.tools.renderrZpydanticrZlangchain._api.deprecationrZlangchain.agents.agentrrrZlangchain.agents.agent_typesrZ#langchain.agents.mrkl.output_parserrZlangchain.agents.mrkl.promptrrrZlangchain.agents.utilsrZlangchain.chainsrrr*r_r'r'r'r(s>