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

doc:update knowledge api

Browse Source
This commit is contained in:
aries_ckt
2023-07-10 17:00:45 +08:00
parent 56b32ab094
commit 6ff7ef9da4
6 changed files with 149 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit/embedding_engine/test_url_embedding.py Normal file
View File

@@ -0,0 +1,12 @@
from pilot import EmbeddingEngine, KnowledgeType
url = "https://db-gpt.readthedocs.io/en/latest/getting_started/getting_started.html"
embedding_model = "text2vec"
vector_store_config = {
"vector_store_name": url.replace(":", ""),
}
embedding_engine = EmbeddingEngine(knowledge_source=url, knowledge_type=KnowledgeType.URL.value, model_name=embedding_model, vector_store_config=vector_store_config)
# embedding url content to vector store
embedding_engine.knowledge_embedding()