Ng 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*GddeZ+e dedGddeZ,e dedGddeZ-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)LLMChainc2eZdZUdZded<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__V/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/agents/mrkl/base.pyrrsBr*rz0.1.0z1.0)messageremovalceZdZUdZeeZded<ed dZ e d!d Z e d!d Z e d!d Z eeeed fd"dZed d eeed fd#dZed$fd ZxZS)% ZeroShotAgentzcAgent for the MRKL chain. Parameters: output_parser: Output parser for the agent. )default_factoryr output_parserkwargsrreturnc tS)Nr)clsr2s r+_get_default_output_parserz(ZeroShotAgent._get_default_output_parser4s!!!r*r ctjS)z Return Identifier of agent type.)rZERO_SHOT_REACT_DESCRIPTIONselfs r+ _agent_typezZeroShotAgent._agent_type8s 44r*cdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r)r9s r+observation_prefixz ZeroShotAgent.observation_prefix=s r*cdS)zVPrefix to append the llm call with. Returns: "Thought: " zThought:r)r9s r+ llm_prefixzZeroShotAgent.llm_prefixFs zr*NtoolsSequence[BaseTool]prefixsuffixformat_instructionsinput_variablesOptional[List[str]]r c$tt|}dd|D}||}d||||g}|rt ||St j|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, cg|] }|j Sr)name.0tools r+ z/ZeroShotAgent.create_prompt..hs<<.s222Ddi222r*) llm_chain allowed_toolsr1r))_validate_toolsrVrr6)r5rWr@rXr1rBrCrDrEr2r\r^rO_output_parsers r+from_llm_and_toolsz ZeroShotAgent.from_llm_and_toolsos6 E"""""  3+ #  -   32E222 &J#*H*H*J*Js $(      r*Nonect|j|t|dkrtd|jd|D]!}|jtd|jd"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 descriptionrJsuperr`)r5r@rM __class__s r+r`zZeroShotAgent._validate_toolss#CL%888 u::??WCLWWW   D' >$)>>>( &&&&&r*)r2rr3r)r3r ) r@rArBr rCr rDr rErFr3r )rWr r@rArXrYr1rZrBr rCr rDr rErFr2rr3r)r@rAr3rc)r$r%r&r'rrr1r( classmethodr6propertyr;r=r?rrrrVrbr` __classcell__)ris@r+r/r/&sa  (-u=M'N'N'NMNNNN"""["555X5XX#6/3 6666[6> ;?59#6/3. . . . [. ` ' ' ' ' '[ ' ' ' ' 'r*r/c*eZdZdZed d Zd S) MRKLChainz&Chain that implements the MRKL system.rWr chainsList[ChainConfig]r2rr3rc fd|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. cPg|]#}t|j|j|j$S))rJfuncrg)rr!r"r#)rLcs r+rNz)MRKLChain.from_chains..sI    ]X0      r*)agentr@r))r/rb)r5rWror2r@rus r+ from_chainszMRKLChain.from_chainssV"      00e<<s6e66v666r*N)rWr rorpr2rr3r)r$r%r&r'rjrvr)r*r+rnrns: 10777[777r*rnN).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/rnr)r*r+rsJUU""""""FFFFFFFFFFFFFFFF******888888<<<<<<111111////////??????@@@@@@JJJJJJJJJJ222222@@@@@@LLLLLLLLLL>>>>>>%%%%%%     *     %  B'B'B'B'B'EB'B'  B'J  %  77777 77  777r*