a ag @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'h/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain/agents/mrkl/base.pyrs rz0.1.0z1.0)messageZremovalc 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_parser4sz(ZeroShotAgent._get_default_output_parserr)r.cCstjS)z Return Identifier of agent type.)rZZERO_SHOT_REACT_DESCRIPTIONselfr'r'r( _agent_type8szZeroShotAgent._agent_typecCsdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r'r1r'r'r(observation_prefix=sz ZeroShotAgent.observation_prefixcCsdS)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r'r1r'r'r( llm_prefixFszZeroShotAgent.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( hz/ZeroShotAgent.create_prompt..) tool_namesz )templater:)rlistjoinformatr Z from_template) r/r6r7r8r9r:Z tool_stringsrBrCr'r'r( create_promptOs   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:)rHpromptrIcSsg|] }|jqSr'r;r=r'r'r(r@rAz4ZeroShotAgent.from_llm_and_tools..) llm_chainZ allowed_toolsr,)_validate_toolsrGrr0)r/rHr6rIr,r7r8r9r:r-rJrKrBZ_output_parserr'r'r(from_llm_and_toolsos,  z ZeroShotAgent.from_llm_and_toolsNone)r6r.cs`t|j|t|dkr*td|jd|D] }|jdur.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<superrL)r/r6r? __class__r'r(rLs     zZeroShotAgent._validate_tools)r"r#r$r%rrr,r& classmethodr0propertyr3r4r5rrrrGrMrL __classcell__r'r'rSr(r*&s4 $0r*c@s(eZdZdZedddddddZd S) MRKLChainz&Chain that implements the MRKL system.r zList[ChainConfig]rr)rHchainsr-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<funcrQ)rrr r!)r>cr'r'r(r@s z)MRKLChain.from_chains..)agentr6)r*rM)r/rHrYr-r6r\r'r'r( from_chainss  zMRKLChain.from_chainsN)r"r#r$r%rUr]r'r'r'r(rXsrXN).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*rXr'r'r'r(s<