Fix bug with absent type of message - get it from node
This commit is contained in:
parent
c1d990fd93
commit
6da60386ab
|
|
@ -221,7 +221,14 @@ class GraphHistoryManager:
|
||||||
for node_id in path_to_target:
|
for node_id in path_to_target:
|
||||||
node = node_map.get(node_id)
|
node = node_map.get(node_id)
|
||||||
if node and 'message' in node.get('data', {}): # Проверяем наличие поля 'message'
|
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
|
return messages_for_path
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user