fix bug, add dependencies: strsimpy, markdownify

requirements.txt

@@ -27,7 +27,11 @@ pathlib
 pytest
 scikit-learn
 numexpr
+strsimpy
+markdownify
+
 vllm>=0.2.0; sys_platform == "linux"
+
 # online api libs
 # zhipuai
 # dashscope>=1.10.0 # qwen

requirements_api.txt

@@ -26,6 +26,8 @@ pathlib
 pytest
 scikit-learn
 numexpr
+strsimpy
+markdownify
 
 vllm>=0.2.0; sys_platform == "linux"
 

tests/api/test_stream_chat_api.py

@@ -93,8 +93,8 @@ def test_knowledge_chat(api="/chat/knowledge_base_chat"):
         data = json.loads(line)
         if "answer" in data:
             print(data["answer"], end="", flush=True)
+    pprint(data)
     assert "docs" in data and len(data["docs"]) > 0
-    pprint(data["docs"])
     assert response.status_code == 200
 
 

webui_pages/knowledge_base/knowledge_base.py

@@ -50,7 +50,7 @@ def file_exists(kb: str, selected_rows: List) -> Tuple[str, str]:
     return "", ""
 
 
-def knowledge_base_page(api: ApiRequest):
+def knowledge_base_page(api: ApiRequest, is_lite: bool = None):
     try:
         kb_list = {x["kb_name"]: x for x in get_kb_details()}
     except Exception as e: