zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: AI dialogue context removes form data (#2257)

Browse Source
This commit is contained in:
shaohuzhang1 2025-02-12 16:07:44 +08:00 committed by GitHub
parent 6a5ec866c0
commit 4874c0e4b0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -6,6 +6,7 @@
@date：2024/6/4 14:30 @date：2024/6/4 14:30
@desc: @desc:
""" """
import re
import time import time
from functools import reduce from functools import reduce
from typing import List, Dict from typing import List, Dict
@ -14,7 +15,6 @@ from django.db.models import QuerySet
from langchain.schema import HumanMessage, SystemMessage from langchain.schema import HumanMessage, SystemMessage
from langchain_core.messages import BaseMessage, AIMessage from langchain_core.messages import BaseMessage, AIMessage
from application.flow.common import Answer
from application.flow.i_step_node import NodeResult, INode from application.flow.i_step_node import NodeResult, INode
from application.flow.step_node.ai_chat_step_node.i_chat_node import IChatNode from application.flow.step_node.ai_chat_step_node.i_chat_node import IChatNode
from application.flow.tools import Reasoning from application.flow.tools import Reasoning
@ -181,6 +181,9 @@ class BaseChatNode(IChatNode):
get_message(history_chat_record[index], dialogue_type, runtime_node_id) get_message(history_chat_record[index], dialogue_type, runtime_node_id)
for index in for index in
range(start_index if start_index > 0 else 0, len(history_chat_record))], []) range(start_index if start_index > 0 else 0, len(history_chat_record))], [])
for message in history_message:
if isinstance(message.content, str):
message.content = re.sub('<form_rander>[\d\D]*?<\/form_rander>', '', message.content)
return history_message return history_message
def generate_prompt_question(self, prompt): def generate_prompt_question(self, prompt):

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

@ -6,6 +6,7 @@
@date：2024/6/4 14:30 @date：2024/6/4 14:30
@desc: @desc:
""" """
import re
import time import time
from functools import reduce from functools import reduce
from typing import List, Dict from typing import List, Dict
@ -114,6 +115,9 @@ class BaseQuestionNode(IQuestionNode):
[history_chat_record[index].get_human_message(), history_chat_record[index].get_ai_message()] [history_chat_record[index].get_human_message(), history_chat_record[index].get_ai_message()]
for index in for index in
range(start_index if start_index > 0 else 0, len(history_chat_record))], []) range(start_index if start_index > 0 else 0, len(history_chat_record))], [])
for message in history_message:
if isinstance(message.content, str):
message.content = re.sub('<form_rander>[\d\D]*?<\/form_rander>', '', message.content)
return history_message return history_message
def generate_prompt_question(self, prompt): def generate_prompt_question(self, prompt):