o Zh @sdZddlmZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZddlmZer8ddlmZd eed d d d fd"d d!Zd S)#z Json agent.) annotations) TYPE_CHECKINGAnyDictListOptional)BaseCallbackManager)BaseLanguageModel) JSON_PREFIX JSON_SUFFIX) JsonToolkit AgentExecutorNFllmr toolkitr callback_managerOptional[BaseCallbackManager]prefixstrsuffixformat_instructions Optional[str]input_variablesOptional[List[str]]verboseboolagent_executor_kwargsOptional[Dict[str, Any]]kwargsrreturnrc Ksddlm} ddlm} ddlm} |} |durd|ini}| j| f|||d|}| |||d}d d | D}| d ||d | }| jd || ||d |pQiS)aConstruct a json agent from an LLM and tools. Args: llm: The language model to use. toolkit: The toolkit to use. callback_manager: The callback manager to use. Default is None. prefix: The prefix to use. Default is JSON_PREFIX. suffix: The suffix to use. Default is JSON_SUFFIX. format_instructions: The format instructions to use. Default is None. input_variables: The input variables to use. Default is None. verbose: Whether to print verbose output. Default is False. agent_executor_kwargs: Optional additional arguments for the agent executor. kwargs: Additional arguments for the agent. Returns: The agent executor. rr ) ZeroShotAgent)LLMChainNr)rrr)rpromptrcSsg|]}|jqS)name).0Ztoolr#r#c/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/agent_toolkits/json/base.py Dsz%create_json_agent..) llm_chainZ allowed_tools)agenttoolsrrr#) langchain.agents.agentrZlangchain.agents.mrkl.baser Zlangchain.chains.llmr!Z get_toolsZ create_promptZfrom_agent_and_tools)rrrrrrrrrrrr r!r*Z prompt_paramsr"r(Z tool_namesr)r#r#r&create_json_agents@   r,)rr rr rrrrrrrrrrrrrrrrrr)__doc__ __future__rtypingrrrrrZlangchain_core.callbacksrZlangchain_core.language_modelsr Z.langchain_community.agent_toolkits.json.promptr r Z/langchain_community.agent_toolkits.json.toolkitr r+rr,r#r#r#r&s"