From 6da60386ab1f6665948df4aa9a6fc69694564597 Mon Sep 17 00:00:00 2001 From: dimitrievgs Date: Mon, 15 Sep 2025 00:51:33 +0300 Subject: [PATCH] Fix bug with absent type of message - get it from node --- app/graph_history_manager.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/app/graph_history_manager.py b/app/graph_history_manager.py index 4b2c728..6ec81c0 100644 --- a/app/graph_history_manager.py +++ b/app/graph_history_manager.py @@ -221,7 +221,14 @@ class GraphHistoryManager: for node_id in path_to_target: node = node_map.get(node_id) if node and 'message' in node.get('data', {}): # Проверяем наличие поля 'message' - messages_for_path.append(node['data']['message']) + original_message = node['data']['message'] + # Извлекаем 'role', 'content' и добавляем 'type' из node['type'] + messages_for_path.append({ + "role": original_message.get("role", "unknown"), + "type": node.get("type", "unknown"), # Используем 'type' узла + "content": original_message.get("content", ""), + "node_id": original_message.get("node_id", node_id) + }) return messages_for_path