Fix bug with absent type of message - get it from node

This commit is contained in:
dimitrievgs 2025-09-15 00:51:33 +03:00
parent c1d990fd93
commit 6da60386ab

View File

@ -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