Заставить агента LangGraph изменить свое внутреннее состояние ⇐ Python

[Anonymous]

Я создаю ИИ-агент, используя LangGraph. Он использует собственный класс состояния, который содержит историю сообщений и строковое значение для имени пользователя.
Я хочу, чтобы агент мог обновлять свое состояние самостоятельно, и, судя по тому, что я видел, вы должны включить это с помощью инструмента, который возвращает команду:

Код: Выделить всё

from langchain.agents import create_agent
from langchain.messages import ToolMessage
from langgraph.types import Command
from langgraph.graph import MessagesState
from langchain.tools import tool

CONFIG = ...

class MyState(MessagesState):
user_name: str

class MyAgent:

def __init__(self):

@tool
def update_name(new_name:str) -> Command:
"""Call this function when you need to update the internal state with a new user name"""
return Command(
update={
"user_name":new_name,
"messages":[ToolMessage("Successfully set new name")]
}
)

self.agent = create_agent(
model=...,
system_prompt=...,
state_schema=MyState,
tools=[update_name,...]
checkpointer=...
)

def run(messages:dict) -> list:
return self.agent.invoke(messages,CONFIG)

Однако, когда агенту дано указание обновить внутреннее имя и он запускает предоставленный инструмент, я получаю сообщение об ошибке:

Код: Выделить всё

ValueError: Expected to have a matching ToolMessage in Command.update for tool 'update_name', got: [ToolMessage(content='Successfully set new name')]. Every tool call (LLM requesting to call a tool) in the message history MUST have a corresponding ToolMessage. You can fix it by modifying the tool to return `Command(update=[ToolMessage("Success", tool_call_id=tool_call_id), ...], ...)`.

Однако это означает, что мне не разрешено обновлять что-либо, кроме списка сообщений, изнутри инструмента. Как предполагается разрешить агенту обновлять свое состояние?


Подробнее здесь: https://stackoverflow.com/questions/798 ... rnal-state