Skip to content
Snippets Groups Projects
Unverified Commit bc6828c4 authored by KevinHuSh's avatar KevinHuSh Committed by GitHub
Browse files

add quota config in dialog conf (#339)

### What problem does this PR solve?

Issue link:#337

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
parent 73285d9d
No related branches found
No related tags found
No related merge requests found
......@@ -60,7 +60,7 @@ def list():
for id in sres.ids:
d = {
"chunk_id": id,
"content_with_weight": rmSpace(sres.highlight[id]) if question else sres.field[id].get(
"content_with_weight": rmSpace(sres.highlight[id]) if question and id in sres.highlight else sres.field[id].get(
"content_with_weight", ""),
"doc_id": sres.field[id]["doc_id"],
"docnm_kwd": sres.field[id]["docnm_kwd"],
......
......@@ -241,7 +241,7 @@ def chat(dialog, messages, **kwargs):
chat_logger.info("User: {}|Assistant: {}".format(
msg[-1]["content"], answer))
if knowledges:
if knowledges and prompt_config.get("quote", True):
answer, idx = retrievaler.insert_citations(answer,
[ck["content_ltks"]
for ck in kbinfos["chunks"]],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment