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

style: fmt

Browse Source
This commit is contained in:
aries_ckt 2023-08-03 15:39:16 +08:00
parent 3c8fa10a9b
commit dfa31a39c6
2 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/unit/embedding_engine/document_test.py
View File

@ -15,6 +15,7 @@ embedding_engine = EmbeddingEngine(
knowledge_source=document_path, knowledge_source=document_path,
knowledge_type=KnowledgeType.DOCUMENT.value, knowledge_type=KnowledgeType.DOCUMENT.value,
model_name=embedding_model, model_name=embedding_model,
vector_store_config=vector_store_config) vector_store_config=vector_store_config,
)
# embedding document content to vector store # embedding document content to vector store
embedding_engine.knowledge_embedding() embedding_engine.knowledge_embedding()