o {q i@sdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZddlmZddlmZmZm Z ddl!m"Z"ddl#m$Z$e deddGdddeZ%dS)zDAn agent designed to hold a conversation in addition to using tools.) annotations)Sequence)AnyOptional) deprecated)BaseCallbackManager)BaseLanguageModel)PromptTemplate)BaseTool)Field)AGENT_DEPRECATION_WARNING)AgentAgentOutputParser) AgentTypeConvoOutputParser)FORMAT_INSTRUCTIONSPREFIXSUFFIX)validate_tools_single_input)LLMChainz0.1.0z1.0)messageZremovalc seZdZUdZdZded< eedZded< e d+d,d d Z e d-d dZ e d-ddZ e d-ddZe eeedddfd.ddZe d/fd!d" Ze ddeeedddfd0d)d*ZZS)1ConversationalAgentz>An agent that holds a conversation in addition to using tools.AIstr ai_prefix)default_factoryr output_parserkwargsrreturncKs t|dS)Nrr)clsrrr"k/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/langchain/agents/conversational/base.py_get_default_output_parser%s z.ConversationalAgent._get_default_output_parsercCstjS)z Return Identifier of agent type.)rZ CONVERSATIONAL_REACT_DESCRIPTIONselfr"r"r# _agent_type-szConversationalAgent._agent_typecCdS)z]Prefix to append the observation with. Returns: "Observation: " z Observation: r"r%r"r"r#observation_prefix2z&ConversationalAgent.observation_prefixcCr()zVPrefix to append the llm call with. Returns: "Thought: " zThought:r"r%r"r"r# llm_prefix;r*zConversationalAgent.llm_prefixZHumanNtoolsSequence[BaseTool]prefixsuffixformat_instructions human_prefixinput_variablesOptional[list[str]]r c Csnddd|D}ddd|D} |j| ||d}|d|d|d|} |dur1gd }t| |d S) auCreate 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 ai_prefix: String to use before AI output. Defaults to "AI". human_prefix: String to use before human output. Defaults to "Human". input_variables: List of input variables the final prompt will expect. Defaults to ["input", "chat_history", "agent_scratchpad"]. Returns: A PromptTemplate with the template assembled from the pieces here.  cSs g|] }d|jd|jqS)z> z: )name description.0Ztoolr"r"r# bs z5ConversationalAgent.create_prompt..z, cSg|]}|jqSr"r5r7r"r"r#r9d) tool_namesrr1z N)inputZ chat_historyZagent_scratchpad)templater2)joinformatr ) r!r,r.r/r0rr1r2Z tool_stringsr=r?r"r"r# create_promptDs  z!ConversationalAgent.create_promptNonecst|t|j|dS)N)super_validate_toolsr__name__)r!r, __class__r"r#rEos z#ConversationalAgent._validate_toolsllmrcallback_managerOptional[BaseCallbackManager]Optional[AgentOutputParser]r c  Ksf|||j||| |||| d} t|| |d} dd|D}|p&|j|d}|d| |||d| S) aConstruct an agent from an LLM and tools. Args: llm: The language model to use. tools: A list of tools to use. callback_manager: The callback manager to use. Default is None. output_parser: The output parser to use. Default is None. prefix: The prefix to use in the prompt. Default is PREFIX. suffix: The suffix to use in the prompt. Default is SUFFIX. format_instructions: The format instructions to use. Default is FORMAT_INSTRUCTIONS. ai_prefix: The prefix to use before AI output. Default is "AI". human_prefix: The prefix to use before human output. Default is "Human". input_variables: The input variables to use. Default is None. **kwargs: Any additional keyword arguments to pass to the agent. Returns: An agent. )rr1r.r/r0r2)rIpromptrJcSr:r"r;r7r"r"r#r9r<z:ConversationalAgent.from_llm_and_tools..r ) llm_chainZ allowed_toolsrrNr")rErBrr$)r!rIr,rJrr.r/r0rr1r2rrMrNr=Z_output_parserr"r"r#from_llm_and_toolsts6 # z&ConversationalAgent.from_llm_and_tools)r)rrrrrr)rr)r,r-r.rr/rr0rrrr1rr2r3rr )r,r-rrC)rIrr,r-rJrKrrLr.rr/rr0rrrr1rr2r3rrrr )rF __module__ __qualname____doc__r__annotations__r rr classmethodr$propertyr'r)r+rrrrBrErO __classcell__r"r"rGr#rsF     *rN)&rR __future__rcollections.abcrtypingrrZlangchain_core._apirZlangchain_core.callbacksrZlangchain_core.language_modelsrZlangchain_core.promptsr Zlangchain_core.toolsr Zpydanticr Zlangchain._api.deprecationr Zlangchain.agents.agentr rZlangchain.agents.agent_typesrZ-langchain.agents.conversational.output_parserrZ&langchain.agents.conversational.promptrrrZlangchain.agents.utilsrZlangchain.chainsrrr"r"r"r#s.