feat: knowledge base search miss config

zhayujie · Sep 30, 2023 · cfd2f27 · cfd2f27
1 parent a2160d1
commit cfd2f27
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/bot/linkai/link_ai_bot.py b/bot/linkai/link_ai_bot.py
@@ -195,7 +195,7 @@ def _fecth_knowledge_search_suffix(self, response) -> str:
                 first_similarity = response.get("knowledge_base").get("first_similarity")
                 logger.info(f"[LINKAI] knowledge base, search_hit={search_hit}, first_similarity={first_similarity}")
                 plugin_config = pconf("linkai")
-                if plugin_config.get("knowledge_base"):
+                if plugin_config.get("knowledge_base") and plugin_config.get("knowledge_base").get("search_miss_text_enabled"):
                     search_miss_similarity = plugin_config.get("knowledge_base").get("search_miss_similarity")
                     search_miss_text =  plugin_config.get("knowledge_base").get("search_miss_text")
                     if not search_hit:

diff --git a/plugins/linkai/config.json.template b/plugins/linkai/config.json.template
@@ -15,5 +15,10 @@
         "enabled": true,
         "group_enabled": true,
         "max_file_size": 5000
+    },
+    "knowledge_base": {
+        "search_miss_text_enabled": false,
+        "search_miss_similarity": 0.8,
+        "search_miss_suffix": "\n-------------\n回复仅供参考，请以官方文档为准"
     }
 }