hWndZddlmZddlmZddlmZmZmZm 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*GddeZ+e dedGddeZ,e dedGddeZ-y)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)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/eduruby.in/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_parser5s !!r)c"tjS)z Return Identifier of agent type.)rZERO_SHOT_REACT_DESCRIPTIONselfs r* _agent_typezZeroShotAgent._agent_type9s444r)cy)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r(r7s r*observation_prefixz ZeroShotAgent.observation_prefix>sr)cy)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r(r7s r* llm_prefixzZeroShotAgent.llm_prefixGsr)Nc tt|}dj|Dcgc]}|jc}}|j |}|d|d|d|} |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) r2toolsprefixsuffixformat_instructionsrA tool_stringstoolr?r@s r* create_promptzZeroShotAgent.create_promptPs0/tE{; YYe1VW S/ ! &D'!$))-=>!o%  & &r))r3rreturnr)r^r) rGSequence[BaseTool]rHrrIrrJrrAOptional[list[str]]r^r )rOr rGr_rQzOptional[BaseCallbackManager]r0zOptional[AgentOutputParser]rHrrIrrJrrAr`r3rr^r)rGr_r^None)r#r$r%r&rrr0r' classmethodr4propertyr9r;r=rrrrMrVrT __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. )rDfuncrZ)agentrGr()rr r!r"r.rV)r2rOchainsr3crGris r* from_chainszMRKLChain.from_chainssm4   ]]XX00   00e<6e6v66 s3AN)rOr rjzlist[ChainConfig]r3rr^r)r#r$r%r&rbrlr(r)r*rfrfs? 17 7"7 7  77r)rfN).r& __future__rcollections.abcrtypingrrrrlangchain_core._apir langchain_core.callbacksr langchain_core.language_modelsr langchain_core.promptsr langchain_core.toolsr rlangchain_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"$66*8<1/?@JJ2@LL>% *   %  D'ED'  D'N  %  7 7  7r)