fix: custom chat exception (eosphoros-ai#257)

fix custom chat exception
raise user can understand error

pilot/scene/chat_knowledge/custom/chat.py

@@ -1,3 +1,5 @@
+from chromadb.errors import NoIndexException
+
 from pilot.scene.base_chat import BaseChat, logger, headers
 from pilot.scene.base import ChatScene
 from pilot.common.sql_database import Database
@@ -50,13 +52,19 @@ def __init__(
  )
 
  def generate_input_values(self):
- docs = self.knowledge_embedding_client.similar_search(
- self.current_user_input, CFG.KNOWLEDGE_SEARCH_TOP_SIZE
- )
- context = [d.page_content for d in docs]
- self.metadata = [d.metadata for d in docs]
- context = context[:2000]
- input_values = {"context": context, "question": self.current_user_input}
+ try:
+ docs = self.knowledge_embedding_client.similar_search(
+ self.current_user_input, CFG.KNOWLEDGE_SEARCH_TOP_SIZE
+ )
+ context = [d.page_content for d in docs]
+ self.metadata = [d.metadata for d in docs]
+ context = context[:2000]
+ input_values = {"context": context, "question": self.current_user_input}
+ except NoIndexException:
+ raise ValueError(
+ f"you have no {self.knowledge_name} knowledge store, please upload your knowledge"
+ )
+
  return input_values
 
  def do_with_prompt_response(self, prompt_response):

pilot/server/webserver.py

@@ -25,7 +25,6 @@
 from pilot.configs.model_config import (
  DATASETS_DIR,
  KNOWLEDGE_UPLOAD_ROOT_PATH,
- LLM_MODEL_CONFIG,
  LOGDIR,
 )
 
@@ -632,7 +631,7 @@ def knowledge_embedding_store(vs_id, files):
  )
  knowledge_embedding_client = KnowledgeEmbedding(
  file_path=os.path.join(KNOWLEDGE_UPLOAD_ROOT_PATH, vs_id, filename),
- model_name=LLM_MODEL_CONFIG["text2vec"],
+ model_name=CFG.EMBEDDING_MODEL,
  vector_store_config={
  "vector_store_name": vector_store_name["vs_name"],
  "vector_store_path": KNOWLEDGE_UPLOAD_ROOT_PATH,