o g @sdZddlmZddlZddlmZddlmZmZm Z ddl m Z ddl m Z mZmZmZGdd d e ZGd d d eZGd d d e ZGddde ZdddZdddZdddZdS)a1Schema definitions for representing agent actions, observations, and return values. **ATTENTION** The schema definitions are provided for backwards compatibility. New agents should be built using the langgraph library (https://github.com/langchain-ai/langgraph)), which provides a simpler and more flexible way to define agents. Please see the migration guide for information on how to migrate existing agents to modern langgraph agents: https://python.langchain.com/docs/how_to/migrate_agent/ Agents use language models to choose a sequence of actions to take. A basic agent works in the following manner: 1. Given a prompt an agent uses an LLM to request an action to take (e.g., a tool to run). 2. The agent executes the action (e.g., runs the tool), and receives an observation. 3. The agent returns the observation to the LLM, which can then be used to generate the next action. 4. When the agent reaches a stopping condition, it returns a final return value. The schemas for the agents themselves are defined in langchain.agents.agent. ) annotationsN)Sequence)AnyLiteralUnion) Serializable) AIMessage BaseMessageFunctionMessage HumanMessagecsxeZdZUdZded< ded< ded< dZded<dfd d ZedddZedddZ e dddZ Z S) AgentActionzRepresents a request to execute an action by an agent. The action consists of the name of the tool to execute and the input to pass to the tool. The log is used to pass along extra information about the action. strtoolUnion[str, dict] tool_inputlogzLiteral['AgentAction']typekwargsrc stjd|||d|dS)N)rrrsuper__init__)selfrrrr __class__rQ/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_core/agents.pyr>szAgentAction.__init__returnboolcCdS)zRReturn whether or not the class is serializable. Default is True. Trclsrrris_lc_serializableCszAgentAction.is_lc_serializable list[str]cCgdS)zhGet the namespace of the langchain object. Default is ["langchain", "schema", "agent"].  langchainschemaagentrrrrrget_lc_namespaceJszAgentAction.get_lc_namespaceSequence[BaseMessage]cCst|S)z3Return the messages that correspond to this action.)!_convert_agent_action_to_messagesrrrrmessagesQzAgentAction.messages)rr rrrr rrrrrr"rr) __name__ __module__ __qualname____doc____annotations__rr classmethodr!r(propertyr, __classcell__rrrrr (s    r c@s(eZdZUdZded< dZded<dS)AgentActionMessageLogaRepresentation of an action to be executed by an agent. This is similar to AgentAction, but includes a message log consisting of chat messages. This is useful when working with ChatModels, and is used to reconstruct conversation history from the agent's perspective. r) message_logz Literal['AgentActionMessageLog']rN)r2r3r4r5r6rrrrrr:Ws  r:c@s4eZdZUdZded< ded< ed dd Zd S) AgentStepz!Result of running an AgentAction.r actionr observationrr)cCst|j|jS)-Messages that correspond to this observation.)&_convert_agent_observation_to_messagesr=r>r+rrrr,uzAgentStep.messagesNr0)r2r3r4r5r6r8r,rrrrr<ms r<csneZdZUdZded< ded< dZded<dfd d ZedddZedddZ e dddZ Z S) AgentFinishzyFinal return value of an ActionAgent. Agents return an AgentFinish when they have reached a stopping condition. dict return_valuesr rzLiteral['AgentFinish']rrrc stjd||d|dS)zGOverride init to support instantiation by position for backward compat.)rDrNrr)rrDrrrrrrszAgentFinish.__init__rrcCr)z0Return whether or not the class is serializable.Trrrrrr!szAgentFinish.is_lc_serializabler"cCr#)z*Get the namespace of the langchain object.r$rrrrrr(r-zAgentFinish.get_lc_namespacer)cCst|jdgS)r?content)rrr+rrrr,rAzAgentFinish.messages)rDrCrr rrr.r/r0r1rrrrrB{s    rB agent_actionrr)cCst|tr|jSt|jdgS)a Convert an agent action to a message. This code is used to reconstruct the original AI message from the agent action. Args: agent_action: Agent action to convert. Returns: AIMessage that corresponds to the original tool invocation. rE) isinstancer:r;rr)rGrrrr*s r*r>rcCs^t|tr t||gS|}t|ts)z tj|dd}Wn ty(t|}Ynwt|dgS)aEConvert an agent action to a message. This code is used to reconstruct the original AI message from the agent action. Args: agent_action: Agent action to convert. observation: Observation to convert to a message. Returns: AIMessage that corresponds to the original tool invocation. F ensure_asciirE)rHr:_create_function_messager jsondumps Exceptionr rGr>rFrrrr@s      r@r cCsJt|tsz tj|dd}Wntyt|}Ynw|}t|j|dS)aConvert agent action and observation into a function message. Args: agent_action: the tool invocation request from the agent. observation: the result of the tool invocation. Returns: FunctionMessage that corresponds to the original tool invocation. FrI)namerF)rHr rLrMrNr rrOrrrrKs   rK)rGr rr))rGr r>rrr))rGr r>rrr )r5 __future__rrLcollections.abcrtypingrrr langchain_core.load.serializablerlangchain_core.messagesrr r r r r:r<rBr*r@rKrrrrs   / %