ih jdZddlmZddlmZmZmZmZmZm 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*GddeZ+e dedGddeZ,e dedGddeZ-y)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)LLMChainc0eZdZUdZded<ded<ded<y) 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__X/var/www/html/dev/engine/venv/lib/python3.12/site-packages/langchain/agents/mrkl/base.pyrrs r*rz0.1.0z1.0)messageremovalc eZdZUdZeeZded<ed dZ e ddZ e ddZ e ddZ eeeed f dd Zed d eeed f dd Zedfd ZxZS) ZeroShotAgentzcAgent for the MRKL chain. Parameters: output_parser: Output parser for the agent. )default_factoryr output_parserc tS)Nr)clskwargss r+_get_default_output_parserz(ZeroShotAgent._get_default_output_parser4s !!r*c"tjS)z Return Identifier of agent type.)rZERO_SHOT_REACT_DESCRIPTIONselfs r+ _agent_typezZeroShotAgent._agent_type8s444r*cy)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r)r8s r+observation_prefixz ZeroShotAgent.observation_prefix=sr*cy)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r)r8s r+ llm_prefixzZeroShotAgent.llm_prefixFsr*Nctt|}dj|Dcgc]}|jc}}|j |}dj||||g} |r t | |St j | Scc}w)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, ) tool_namesz )templateinput_variables)rlistjoinnameformatr from_template) r3toolsprefixsuffixformat_instructionsrB tool_stringstoolr@rAs r+ create_promptzZeroShotAgent.create_promptOs0/tE{; YYe1VW  D' !$))-=>  &r*)r4rreturnr)r^r ) rHSequence[BaseTool]rIr rJr rKr rBOptional[List[str]]r^r )rPr rHr_rRzOptional[BaseCallbackManager]r1zOptional[AgentOutputParser]rIr rJr rKr rBr`r4rr^r)rHr_r^None)r$r%r&r'rrr1r( classmethodr5propertyr:r<r>rrrrNrWrU __classcell__)r]s@r+r/r/&sd  (-=M'NM$N""55#6/3 6!66 6 ! 6 - 6 66> ;?59#6/3. . ". 8 . 3 .  . . !. -. .  . . ` ' 'r*r/c2eZdZdZe ddZy) MRKLChainz&Chain that implements the MRKL system.c |Dcgc].}t|j|j|j0}}tj ||}|d||d|Scc}w)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. )rEfuncr[)agentrHr))rr!r"r#r/rW)r3rPchainsr4crHris r+ from_chainszMRKLChain.from_chainssm.   ]]XX00   00e<6e6v66 s3AN)rPr rjzList[ChainConfig]r4rr^r)r$r%r&r'rbrlr)r*r+rfrfs9 17#7->7JM7 77r*rfN).r' __future__rtypingrrrrrr langchain_core._apir langchain_core.callbacksr langchain_core.language_modelsr langchain_core.promptsr langchain_core.toolsrrlangchain_core.tools.renderrpydanticrlangchain._api.deprecationrlangchain.agents.agentrrrlangchain.agents.agent_typesr#langchain.agents.mrkl.output_parserrlangchain.agents.mrkl.promptrrrlangchain.agents.utilsrlangchain.chainsrrr/rfr)r*r+r}sU"FF*8<1/?@JJ2@LL>% *   %  B'EB'  B'J  %  7 7  7r*