From bc6828c45fd6bc2320a82fbf8d2db927590447f2 Mon Sep 17 00:00:00 2001
From: KevinHuSh <kevinhu.sh@gmail.com>
Date: Fri, 12 Apr 2024 14:58:03 +0800
Subject: [PATCH] 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)
---
api/apps/chunk_app.py | 2 +-
api/apps/conversation_app.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/api/apps/chunk_app.py b/api/apps/chunk_app.py
index 0458160..b3cf11a 100644
--- a/api/apps/chunk_app.py
+++ b/api/apps/chunk_app.py
@@ -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"],
diff --git a/api/apps/conversation_app.py b/api/apps/conversation_app.py
index 8c42c80..1d98943 100644
--- a/api/apps/conversation_app.py
+++ b/api/apps/conversation_app.py
@@ -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"]],
--
GitLab