diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py
index 068101fcb..51d3958de 100644
--- a/pilot/server/webserver.py
+++ b/pilot/server/webserver.py
@@ -246,12 +246,13 @@ def http_bot(
     logger.info(
         f"User message send!{state.conv_id},{selected},{plugin_selector},{mode},{sql_mode},{db_selector},{url_input}"
     )
-    if chat_mode_title["knowledge_qa"] == selected:
-        scene: ChatScene = get_chat_mode(selected, mode)
+    if chat_mode_title["sql_generate_diagnostics"] == selected:
+        scene: ChatScene = get_chat_mode(selected, sql_mode)
     elif chat_mode_title["chat_use_plugin"] == selected:
         scene: ChatScene = get_chat_mode(selected)
     else:
-        scene: ChatScene = get_chat_mode(selected, sql_mode)
+        scene: ChatScene = get_chat_mode(selected, mode)
+
     print(f"chat scene:{scene.value}")
 
     if ChatScene.ChatWithDbExecute == scene:
diff --git a/pilot/vector_store/chroma_store.py b/pilot/vector_store/chroma_store.py
index 3a9de6874..4949924d4 100644
--- a/pilot/vector_store/chroma_store.py
+++ b/pilot/vector_store/chroma_store.py
@@ -1,7 +1,6 @@
 import os
 
 from langchain.vectorstores import Chroma
-
 from pilot.configs.model_config import KNOWLEDGE_UPLOAD_ROOT_PATH
 from pilot.logs import logger
 from pilot.vector_store.vector_store_base import VectorStoreBase