File tree Expand file tree Collapse file tree 2 files changed +10
-6
lines changed
step_node/search_dataset_node Expand file tree Collapse file tree 2 files changed +10
-6
lines changed Original file line number Diff line number Diff line change @@ -105,13 +105,14 @@ class FlowParamsSerializer(serializers.Serializer):
105
105
106
106
chat_record_id = serializers .CharField (required = True , error_messages = ErrMessage .char ("对话记录id" ))
107
107
108
- stream = serializers .BooleanField (required = True , error_messages = ErrMessage .base ("流式输出" ))
108
+ stream = serializers .BooleanField (required = True , error_messages = ErrMessage .boolean ("流式输出" ))
109
109
110
110
client_id = serializers .CharField (required = False , error_messages = ErrMessage .char ("客户端id" ))
111
111
112
112
client_type = serializers .CharField (required = False , error_messages = ErrMessage .char ("客户端类型" ))
113
113
114
114
user_id = serializers .UUIDField (required = True , error_messages = ErrMessage .char ("用户id" ))
115
+ re_chat = serializers .BooleanField (required = True , error_messages = ErrMessage .boolean ("换个答案" ))
115
116
116
117
117
118
class INode :
Original file line number Diff line number Diff line change @@ -61,11 +61,14 @@ def _run(self):
61
61
question = self .workflow_manage .get_reference_field (
62
62
self .node_params_serializer .data .get ('question_reference_address' )[0 ],
63
63
self .node_params_serializer .data .get ('question_reference_address' )[1 :])
64
- history_chat_record = self .flow_params_serializer .data .get ('history_chat_record' , [])
65
- paragraph_id_list = [p .get ('id' ) for p in flat_map (
66
- [get_paragraph_list (chat_record , self .node .id ) for chat_record in history_chat_record if
67
- chat_record .problem_text == question ])]
68
- exclude_paragraph_id_list = list (set (paragraph_id_list ))
64
+ exclude_paragraph_id_list = []
65
+ if self .flow_params_serializer .data .get ('re_chat' , False ):
66
+ history_chat_record = self .flow_params_serializer .data .get ('history_chat_record' , [])
67
+ paragraph_id_list = [p .get ('id' ) for p in flat_map (
68
+ [get_paragraph_list (chat_record , self .node .id ) for chat_record in history_chat_record if
69
+ chat_record .problem_text == question ])]
70
+ exclude_paragraph_id_list = list (set (paragraph_id_list ))
71
+
69
72
return self .execute (** self .node_params_serializer .data , question = str (question ),
70
73
exclude_paragraph_id_list = exclude_paragraph_id_list )
71
74
You can’t perform that action at this time.
0 commit comments