15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:12:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: 问题优化对话错误

This commit is contained in:
shaohuzhang1 2024-06-27 09:48:40 +08:00
parent b3abef38a7
commit 1af9ff351b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/application/flow/step_node/question_node/impl/base_question_node.py
View File

@ -7,6 +7,7 @@
@desc:
"""
import json
from functools import reduce
from typing import List, Dict
from django.db.models import QuerySet
@ -148,9 +149,10 @@ class BaseQuestionNode(IQuestionNode):
@staticmethod
def get_history_message(history_chat_record, dialogue_number):
start_index = len(history_chat_record) - dialogue_number
history_message = [[history_chat_record[index].get_human_message(), history_chat_record[index].get_ai_message()]
for index in
range(start_index if start_index > 0 else 0, len(history_chat_record))]
history_message = reduce(lambda x, y: [*x, *y], [
[history_chat_record[index].get_human_message(), history_chat_record[index].get_ai_message()]
for index in
range(start_index if start_index > 0 else 0, len(history_chat_record))], [])
return history_message
def generate_prompt_question(self, prompt):