o `+ iz@sddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZmZdd lmZdd lmZdd lmZdd lmZmZdd lmZddlmZm Z m!Z!m"Z"ddl#m$Z$ddl%m&Z&ededdGdddeZ'dS))Sequence)AnyOptional) deprecated) AgentAction)BaseCallbackManager)BaseLanguageModel)BasePromptTemplate)ChatPromptTemplateHumanMessagePromptTemplateSystemMessagePromptTemplate)BaseTool)Field)AGENT_DEPRECATION_WARNING)AgentAgentOutputParserChatOutputParser)FORMAT_INSTRUCTIONS HUMAN_MESSAGESYSTEM_MESSAGE_PREFIXSYSTEM_MESSAGE_SUFFIX)validate_tools_single_input)LLMChainz0.1.0z1.0)messageZremovalcsleZdZUdZeedZeed< e de fddZ e de fddZ d e eee fde ffd d Zed edefd dZedeeddffdd Ze de e fddZeeeeedfdeede de de de dee e defddZeddeeeedfdedeedee deede de de de dee e d ede!fddZ"e de fd d!Z#Z$S)" ChatAgentz Chat Agent.)default_factory output_parserreturncCdS)z&Prefix to append the observation with.z Observation: selfr r f/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/langchain/agents/chat/base.pyobservation_prefix)zChatAgent.observation_prefixcCr)z#Prefix to append the llm call with.zThought:r r!r r r# llm_prefix.r%zChatAgent.llm_prefixintermediate_stepscs4t|}t|tsd}t||rd|S|S)Nz*agent_scratchpad should be of type string.zhThis was your previous work (but I haven't seen any of it! I only see what you return as final answer): )super_construct_scratchpad isinstancestr ValueError)r"r'agent_scratchpadmsg __class__r r#r)3s  zChatAgent._construct_scratchpadkwargscKstSNr)clsr1r r r#_get_default_output_parserCz$ChatAgent._get_default_output_parsertoolsNcst|t|j|ddS)N) class_namer6)r(_validate_toolsr__name__)r3r6r/r r#r8Gs zChatAgent._validate_toolscCsdgS)Nz Observation:r r!r r r#_stopLr5zChatAgent._stopsystem_message_prefixsystem_message_suffix human_messageformat_instructionsinput_variablesc Cs~ddd|D}ddd|D}|j|d}|d|d|d|} t| t|g} |dur9d d g}t|| d S) agCreate a prompt from a list of tools. Args: tools: A list of tools. system_message_prefix: The system message prefix. Default is SYSTEM_MESSAGE_PREFIX. system_message_suffix: The system message suffix. Default is SYSTEM_MESSAGE_SUFFIX. human_message: The human message. Default is HUMAN_MESSAGE. format_instructions: The format instructions. Default is FORMAT_INSTRUCTIONS. input_variables: The input variables. Default is None. Returns: A prompt template.  cSsg|] }|jd|jqS)z: )name description.0Ztoolr r r# ksz+ChatAgent.create_prompt..z, cSg|]}|jqSr rArCr r r#rEl) tool_namesz Ninputr-)r?messages)joinformatr Z from_templater r ) r3r6r;r<r=r>r?Z tool_stringsrItemplaterKr r r# create_promptPs  zChatAgent.create_promptllmcallback_managerc Ks^|||j|||||| d} t|| |d} dd|D} |p#|}|d| | |d| S)a=Construct an agent from an LLM and tools. Args: llm: The language model. tools: A list of tools. callback_manager: The callback manager. Default is None. output_parser: The output parser. Default is None. system_message_prefix: The system message prefix. Default is SYSTEM_MESSAGE_PREFIX. system_message_suffix: The system message suffix. Default is SYSTEM_MESSAGE_SUFFIX. human_message: The human message. Default is HUMAN_MESSAGE. format_instructions: The format instructions. Default is FORMAT_INSTRUCTIONS. input_variables: The input variables. Default is None. kwargs: Additional keyword arguments. Returns: An agent. )r;r<r=r>r?)rPpromptrQcSrFr rGrCr r r#rErHz0ChatAgent.from_llm_and_tools..) llm_chainZ allowed_toolsrNr )r8rOrr4)r3rPr6rQrr;r<r=r>r?r1rRrSrIZ_output_parserr r r#from_llm_and_toolsws. " zChatAgent.from_llm_and_toolscCstr2)r,r!r r r# _agent_typeszChatAgent._agent_type)%r9 __module__ __qualname____doc__rrrr__annotations__propertyr+r$r&listtuplerr) classmethodrr4rr r8r:rrrrrr rOrrrrTrU __classcell__r r r/r#rs   &    8rN)(collections.abcrtypingrrZlangchain_core._apirZlangchain_core.agentsrZlangchain_core.callbacksrZlangchain_core.language_modelsrZlangchain_core.promptsr Zlangchain_core.prompts.chatr r r Zlangchain_core.toolsr ZpydanticrZlangchain._api.deprecationrZlangchain.agents.agentrrZ#langchain.agents.chat.output_parserrZlangchain.agents.chat.promptrrrrZlangchain.agents.utilsrZlangchain.chains.llmrrr r r r#s,