feat(vector): add weaviate support
This commit is contained in:
1 parent
09cdf07aab
commit
bf9ef4365b
31 files changed
+180
-28
No files matched your search
@@ -712,10 +712,11 @@ pub async fn test_connection(state: State<'_, Arc<AppState>>, config: Connection
|
||||
.await
|
||||
.map(|_| "Connection successful".to_string())
|
||||
}
|
||||
DatabaseType::Qdrant | DatabaseType::Milvus => {
|
||||
DatabaseType::Qdrant | DatabaseType::Milvus | DatabaseType::Weaviate => {
|
||||
let kind = match config.db_type {
|
||||
DatabaseType::Qdrant => db::vector_driver::VectorDbKind::Qdrant,
|
||||
DatabaseType::Milvus => db::vector_driver::VectorDbKind::Milvus,
|
||||
DatabaseType::Weaviate => db::vector_driver::VectorDbKind::Weaviate,
|
||||
_ => unreachable!(),
|
||||
};
|
||||
let client = db::vector_driver::VectorClient::new(
|
||||
@@ -997,10 +998,11 @@ pub async fn connect_db(state: State<'_, Arc<AppState>>, config: ConnectionConfi
|
||||
db::elasticsearch_driver::test_connection(&mut client, connect_timeout).await?;
|
||||
PoolKind::Elasticsearch(client)
|
||||
}
|
||||
DatabaseType::Qdrant | DatabaseType::Milvus => {
|
||||
DatabaseType::Qdrant | DatabaseType::Milvus | DatabaseType::Weaviate => {
|
||||
let kind = match db_config.db_type {
|
||||
DatabaseType::Qdrant => db::vector_driver::VectorDbKind::Qdrant,
|
||||
DatabaseType::Milvus => db::vector_driver::VectorDbKind::Milvus,
|
||||
DatabaseType::Weaviate => db::vector_driver::VectorDbKind::Weaviate,
|
||||
_ => unreachable!(),
|
||||
};
|
||||
let client = db::vector_driver::VectorClient::new(
|
||||
|
||||
Reference in new issue
Block a user