From ac686e9c6f9c98215be9a09f83d76405e3f58eea Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Tue, 27 Jun 2023 17:46:48 +0800
Subject: [PATCH] fix: knowledge embedding default parameter 1.knowledge
 embedding default parameter

---
 pilot/embedding_engine/knowledge_embedding.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pilot/embedding_engine/knowledge_embedding.py b/pilot/embedding_engine/knowledge_embedding.py
index a1d148130..5a8a2f944 100644
--- a/pilot/embedding_engine/knowledge_embedding.py
+++ b/pilot/embedding_engine/knowledge_embedding.py
@@ -4,7 +4,7 @@ from chromadb.errors import NotEnoughElementsException
 from langchain.embeddings import HuggingFaceEmbeddings
 
 from pilot.configs.config import Config
-from pilot.embedding_engine.knowledge_type import get_knowledge_embedding
+from pilot.embedding_engine.knowledge_type import get_knowledge_embedding, KnowledgeType
 from pilot.vector_store.connector import VectorStoreConnector
 
 CFG = Config()
@@ -15,7 +15,7 @@ class KnowledgeEmbedding:
         self,
         model_name,
         vector_store_config,
-        knowledge_type: Optional[str],
+        knowledge_type: Optional[str] = KnowledgeType.DOCUMENT.value,
         knowledge_source: Optional[str] = None,
     ):
         """Initialize with knowledge embedding client, model_name, vector_store_config, knowledge_type, knowledge_source"""