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