Обновить список сообщений о состоянии перед прерыванием ⇐ Python

[Anonymous]

Я использую Langgraph версии 1.2.7 для построения рабочего процесса, состоящего из нескольких узлов. Многие из этих узлов могут прерывать поток, запрашивая у пользователя разъяснения или дополнительную информацию. Пользователь должен иметь возможность ответить на вопрос немедленно или вернуться позже и продолжить взаимодействие с разговором.
При получении прошлого разговора мне нужно загрузить полную историю взаимодействия, чтобы пользователь мог следить за контекстом разговора. Другими словами: сообщения о прерываниях, которые запрашивают разъяснения или дополнительную информацию, также должны быть включены в историю.
Проблема в том, что Langgraph не делает этого автоматически, и обновить историю до запуска этих прерываний было очень сложно. Я прошу помощи или рекомендаций о том, как правильно включать сообщения о прерывании в историю разговора.
Ниже приведен код, который помогает проиллюстрировать мою проблему. Очевидно, что в моем реальном рабочем процессе больше узлов, но для простоты и ясности этого примера должно быть достаточно:

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

import asyncio
from typing import Annotated, TypedDict

from langchain.messages import AnyMessage, HumanMessage
from langchain_core.messages import AIMessage
from langchain_core.runnables import RunnableConfig
from langgraph.checkpoint.memory import MemorySaver
from langgraph.graph import END, START, StateGraph
from langgraph.graph.message import add_messages
from langgraph.types import interrupt

class State(TypedDict):
messages: Annotated[list[AnyMessage], add_messages]

def ask_user(state: State):
"""Update messages list before interrupt."""
message = "Need more info"
state["messages"].append(AIMessage(content=message))
interrupt(message)

workflow = StateGraph(State)
workflow.add_node("ask_user", ask_user)
workflow.add_edge(START, "ask_user")
workflow.add_edge("ask_user", END)
graph = workflow.compile(checkpointer=MemorySaver())

async def main():
config = RunnableConfig(configurable={"thread_id": "1"})
response = graph.astream(input={"messages": [HumanMessage(content="Hello")]}, config=config)

async for event in response:
print(event)

hist = await graph.aget_state(config=config)
print(hist.values)

asyncio.run(main())

При запуске кода переменная hist будет иметь только сообщение «Hello» в списке значений вместо двух сообщений.
Некоторые любопытные моменты заключаются в том, что если я изменю стартовый метод с astream на просто поток и синхронизирую цикл for, список сообщений обновится! Но мне действительно нужен асинхронный график, поэтому это не решение.
Я уже пытался обновить состояние с помощью команды и использовать функцию обновления_статуса графика, но безрезультатно.
Заранее благодарю за любую помощь.

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