Просмотр исходного кода

fix: segment resort in dataset retrieve by index_node_id_to_position (#721)

tags/0.3.13
takatost 2 лет назад
Родитель
Сommit
654985177f
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 5 добавлений и 1 удалений
  1. 5
    1
      api/core/tool/dataset_retriever_tool.py

+ 5
- 1
api/core/tool/dataset_retriever_tool.py Просмотреть файл

).all() ).all()


if segments: if segments:
for segment in segments:
index_node_id_to_position = {id: position for position, id in enumerate(index_node_ids)}
sorted_segments = sorted(segments,
key=lambda segment: index_node_id_to_position.get(segment.index_node_id,
float('inf')))
for segment in sorted_segments:
if segment.answer: if segment.answer:
document_context_list.append(f'question:{segment.content} \nanswer:{segment.answer}') document_context_list.append(f'question:{segment.content} \nanswer:{segment.answer}')
else: else:

Загрузка…
Отмена
Сохранить