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

fix: 修复无引用分段指定回答设置问题 #317 (#319)

Browse Source
This commit is contained in:
shaohuzhang1 2024-04-29 17:11:19 +08:00 committed by GitHub
parent e5e2ece1bd
commit dedfd31e5c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py
View File

@ -142,7 +142,7 @@ class BaseChatStep(IChatStep):
paragraph.hit_handling_method == 'directly_return'] paragraph.hit_handling_method == 'directly_return']
if directly_return_chunk_list is not None and len(directly_return_chunk_list) > 0: if directly_return_chunk_list is not None and len(directly_return_chunk_list) > 0:
return iter(directly_return_chunk_list), False return iter(directly_return_chunk_list), False
elif no_references_setting.get( elif len(paragraph_list) == 0 and no_references_setting.get(
'status') == 'designated_answer': 'status') == 'designated_answer':
return iter([AIMessageChunk(content=no_references_setting.get('value'))]), False return iter([AIMessageChunk(content=no_references_setting.get('value'))]), False
if chat_model is None: if chat_model is None:
@ -185,7 +185,7 @@ class BaseChatStep(IChatStep):
paragraph.hit_handling_method == 'directly_return'] paragraph.hit_handling_method == 'directly_return']
if directly_return_chunk_list is not None and len(directly_return_chunk_list) > 0: if directly_return_chunk_list is not None and len(directly_return_chunk_list) > 0:
return directly_return_chunk_list[0], False return directly_return_chunk_list[0], False
elif no_references_setting.get( elif len(paragraph_list) == 0 and no_references_setting.get(
'status') == 'designated_answer': 'status') == 'designated_answer':
return AIMessage(no_references_setting.get('value')), False return AIMessage(no_references_setting.get('value')), False
if chat_model is None: if chat_model is None: