diff --git a/requirements.txt b/requirements.txt
index 3da9c84..e5d76ed 100644
--- a/requirements.txt
+++ b/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
diff --git a/requirements_api.txt b/requirements_api.txt
index b8a7f6d..4ef692b 100644
--- a/requirements_api.txt
+++ b/requirements_api.txt
@@ -26,6 +26,8 @@ pathlib
 pytest
 scikit-learn
 numexpr
+strsimpy
+markdownify
 
 vllm>=0.2.0; sys_platform == "linux"
 
diff --git a/tests/api/test_stream_chat_api.py b/tests/api/test_stream_chat_api.py
index 8b98c20..7db8276 100644
--- a/tests/api/test_stream_chat_api.py
+++ b/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
 
 
diff --git a/webui_pages/knowledge_base/knowledge_base.py b/webui_pages/knowledge_base/knowledge_base.py
index 95a1fca..4df7c9c 100644
--- a/webui_pages/knowledge_base/knowledge_base.py
+++ b/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: