Files
DB-GPT/tests/unit/embedding_engine/test_url_embedding.py
2023-07-10 17:00:45 +08:00

13 lines
500 B
Python

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()