U h@s.dZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZdd lmZmZdd 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)Field)BaseToolTool)render_text_description)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&>/tmp/pip-unpacked-wheel-bo69hh5q/langchain/agents/mrkl/base.pyrs rz0.1.0Zcreate_react_agentz1.0) alternativeremovalc seZdZUdZeedZded<edddddZ e d d d d Z e d d d dZ e d d ddZ eeeedfdd d d dddddZeddeeedfddddd d d dddd ddZeddd fd!d" ZZS)# ZeroShotAgentzcAgent for the MRKL chain. Parameters: output_parser: Output parser for the agent. )default_factoryr output_parserr)kwargsreturncKstS)Nr)clsr-r&r&r'_get_default_output_parser/sz(ZeroShotAgent._get_default_output_parserr)r.cCstjS)z Return Identifier of agent type.)rZZERO_SHOT_REACT_DESCRIPTIONselfr&r&r' _agent_type3szZeroShotAgent._agent_typecCsdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r&r1r&r&r'observation_prefix8sz ZeroShotAgent.observation_prefixcCsdS)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r&r1r&r&r' llm_prefixAszZeroShotAgent.llm_prefixNzSequence[BaseTool]zOptional[List[str]]r )toolsprefixsuffixformat_instructionsinput_variablesr.c CsXtt|}ddd|D}|j|d}d||||g}|rNt||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, cSsg|] }|jqSr&name.0toolr&r&r' csz/ZeroShotAgent.create_prompt..) tool_namesz )templater:)rlistjoinformatr Z from_template) r/r6r7r8r9r:Z tool_stringsrArBr&r&r' create_promptJs   zZeroShotAgent.create_promptr zOptional[BaseCallbackManager]zOptional[AgentOutputParser]r) llmr6callback_managerr,r7r8r9r:r-r.c Ks\|||j|||||d} t|| |d} dd|D} |pD|} |f| | | 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. )r7r8r9r:)rGpromptrHcSsg|] }|jqSr&r;r=r&r&r'r@sz4ZeroShotAgent.from_llm_and_tools..) llm_chainZ allowed_toolsr,)_validate_toolsrFrr0)r/rGr6rHr,r7r8r9r:r-rIrJrAZ_output_parserr&r&r'from_llm_and_toolsjs,  z ZeroShotAgent.from_llm_and_toolsNone)r6r.cs`t|j|t|dkr*td|jd|D] }|jdkr.td|jdq.t|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 descriptionr<superrK)r/r6r? __class__r&r'rKs     zZeroShotAgent._validate_tools)r!r"r#r$r rr,r% classmethodr0propertyr3r4r5rrrrFrLrK __classcell__r&r&rRr'r*%s4 $0r*)r)c@s(eZdZdZedddddddZd S) MRKLChainz&Chain that implements the MRKL system.r zList[ChainConfig]rr)rGchainsr-r.cKs.dd|D}t||}|f||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))r<funcrP)rrrr )r>cr&r&r'r@s z)MRKLChain.from_chains..)agentr6)r*rL)r/rGrXr-r6r[r&r&r' from_chainss  zMRKLChain.from_chainsN)r!r"r#r$rTr\r&r&r&r'rWsrWN),r$ __future__rtypingrrrrrrZlangchain_core._apir Zlangchain_core.callbacksr Zlangchain_core.language_modelsr Zlangchain_core.promptsr Zlangchain_core.pydantic_v1r Zlangchain_core.toolsrrZlangchain_core.tools.renderrZlangchain.agents.agentrrrZlangchain.agents.agent_typesrZ#langchain.agents.mrkl.output_parserrZlangchain.agents.mrkl.promptrrrZlangchain.agents.utilsrZlangchain.chainsrrr*rWr&r&r&r's*