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)
This commit is contained in:
KevinHuSh
2024-04-12 14:58:03 +08:00
committed by GitHub
parent 73285d9d07
commit bc6828c45f
2 changed files with 2 additions and 2 deletions

View File

@ -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"]],