github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-09 04:08:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: knowledge embedding default parameter

Browse Source 
1.knowledge embedding default parameter
This commit is contained in:
aries_ckt 2023-06-27 17:46:48 +08:00
parent 5d69b31e8c
commit ac686e9c6f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/embedding_engine/knowledge_embedding.py
View File

@ -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"""