|
|
|
@@ -199,9 +199,10 @@ def completion(): |
|
|
|
conv.message[-1] = {"role": "assistant", "content": ans["answer"]}
|
|
|
|
|
|
|
|
def rename_field(ans):
|
|
|
|
for chunk_i in ans['reference'].get('chunks', []):
|
|
|
|
chunk_i['doc_name'] = chunk_i['docnm_kwd']
|
|
|
|
chunk_i.pop('docnm_kwd')
|
|
|
|
for chunk in ans.get('reference', []):
|
|
|
|
for chunk_i in chunk.get('chunks', []):
|
|
|
|
chunk_i['doc_name'] = chunk_i['docnm_kwd']
|
|
|
|
chunk_i.pop('docnm_kwd')
|
|
|
|
|
|
|
|
def stream():
|
|
|
|
nonlocal dia, msg, req, conv
|
|
|
|
@@ -232,10 +233,7 @@ def completion(): |
|
|
|
API4ConversationService.append_message(conv.id, conv.to_dict())
|
|
|
|
break
|
|
|
|
|
|
|
|
for chunk_i in answer['reference'].get('chunks',[]):
|
|
|
|
chunk_i['doc_name'] = chunk_i['docnm_kwd']
|
|
|
|
chunk_i.pop('docnm_kwd')
|
|
|
|
|
|
|
|
rename_field(answer)
|
|
|
|
return get_json_result(data=answer)
|
|
|
|
|
|
|
|
except Exception as e:
|
|
|
|
@@ -252,6 +250,8 @@ def get(conversation_id): |
|
|
|
|
|
|
|
conv = conv.to_dict()
|
|
|
|
for referenct_i in conv['reference']:
|
|
|
|
if referenct_i is None or len(referenct_i) == 0:
|
|
|
|
continue
|
|
|
|
for chunk_i in referenct_i['chunks']:
|
|
|
|
if 'docnm_kwd' in chunk_i.keys():
|
|
|
|
chunk_i['doc_name'] = chunk_i['docnm_kwd']
|