pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

refine citation (#161)

Browse Source
This commit is contained in:
KevinHuSh
2024-03-28 11:45:50 +08:00
committed by GitHub
parent 37cc673098
commit f3477202fe
3 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/apps/conversation_app.py
View File

@ -194,7 +194,8 @@ def chat(dialog, messages, **kwargs):
# try to use sql if field mapping is good to go
if field_map:
chat_logger.info("Use SQL to retrieval:{}".format(questions[-1]))
return use_sql(questions[-1], field_map, dialog.tenant_id, chat_mdl)
ans = use_sql(questions[-1], field_map, dialog.tenant_id, chat_mdl)
if ans: return ans
prompt_config = dialog.prompt_config
for p in prompt_config["parameters"]:
@ -305,7 +306,7 @@ def use_sql(question, field_map, tenant_id, chat_mdl):
tbl, sql = get_table()
if tbl is None:
return None, None
return None
if tbl.get("error") and tried_times <= 2:
user_promt = """
表名：{}；
@ -333,7 +334,7 @@ def use_sql(question, field_map, tenant_id, chat_mdl):
chat_logger.info("GET table: {}".format(tbl))
print(tbl)
if tbl.get("error") or len(tbl["rows"]) == 0:
return None, None
return None
docid_idx = set([ii for ii, c in enumerate(
tbl["columns"]) if c["name"] == "doc_id"])