Update

setup_knowledge_base.py

@@ -190,7 +190,7 @@ def insert_chunks_to_db(db_path, document_id, chunks):
     conn.close()
     print(f"Saved {len(chunks)} chunks into the database for file ID {document_id}.")
 
-def generate_embeddings(texts, model_name="togethercomputer/m2-bert-80M-32k-retrieval"):
+def generate_embeddings(texts, model_name="BAAI/bge-base-en-v1.5"):
     """
     Generates text embeddings using Together AI's embedding model.
     """
@@ -287,7 +287,7 @@ if __name__ == "__main__":
     db_path = os.path.join(db_directory, "processed_documents.db")
     vector_db_dir = "data/vector_db_chroma"
     collection_name = "my_instrument_manual_chunks"
-    embeddings_model_name = "togethercomputer/m2-bert-80M-32k-retrieval" # 確保與 main.py 中使用的一致
+    embeddings_model_name = "BAAI/bge-base-en-v1.5" # 確保與 main.py 中使用的一致
 
     # Create necessary directories
     os.makedirs(db_directory, exist_ok=True)