o {q iW@s@dZddlmZddlmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZmZmZddlm 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)Sequence)AnyCallable NamedTupleOptional) 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&a/home/app/PaddleOCR-VL/.venv_paddleocr/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_parser5sz(ZeroShotAgent._get_default_output_parserrcCstjS)z Return Identifier of agent type.)rZZERO_SHOT_REACT_DESCRIPTIONselfr&r&r' _agent_type9szZeroShotAgent._agent_typecCdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r&r0r&r&r'observation_prefix>z ZeroShotAgent.observation_prefixcCr3)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r&r0r&r&r' llm_prefixGr5zZeroShotAgent.llm_prefixNtoolsSequence[BaseTool]prefixsuffixformat_instructionsinput_variablesOptional[list[str]]r c Cs`tt|}ddd|D}|j|d}|d|d|d|}|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' iz/ZeroShotAgent.create_prompt..) tool_namesz )templater<)rlistjoinformatr Z from_template) r.r7r9r:r;r<Z tool_stringsrFrGr&r&r' create_promptPs    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. )r9r:r;r<)rLpromptrMcSr>r&r?rAr&r&r'rDrEz4ZeroShotAgent.from_llm_and_tools..) llm_chainZ allowed_toolsr+Nr&)_validate_toolsrKrr/)r.rLr7rMr+r9r:r;r<r,rPrQrFZ_output_parserr&r&r'from_llm_and_toolsps,  z ZeroShotAgent.from_llm_and_toolsNonecsht|j|t|dkrd|jd}t||D]}|jdur+d|jd}t|qt|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@superrR)r.r7msgrC __class__r&r'rRs     zZeroShotAgent._validate_tools)r,rr-r)r-r) r7r8r9rr:rr;rr<r=r-r )rLr r7r8rMrNr+rOr9rr:rr;rr<r=r,rr-r)r7r8r-rT)r!r"r#r$rrr+r% classmethodr/propertyr2r4r6rrrrKrSrR __classcell__r&r&rZr'r)'s6     0r)c@seZdZdZed d d Zd S) MRKLChainz&Chain that implements the MRKL system.rLr 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))r@funcrW)r rrr )rBcr&r&r'rDsz)MRKLChain.from_chains..)agentr7Nr&)r)rS)r.rLr`r,r7rdr&r&r' from_chainss  zMRKLChain.from_chainsN)rLr r`rar,rr-r)r!r"r#r$r\rer&r&r&r'r_sr_N).r$ __future__rcollections.abcrtypingrrrrZlangchain_core._apirZlangchain_core.callbacksr Zlangchain_core.language_modelsr Zlangchain_core.promptsr Zlangchain_core.toolsr r Zlangchain_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@