mirror of
https://github.com/DrizzleTime/Foxel.git
synced 2026-05-11 18:10:10 +08:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from typing import Any, Dict, List
|
|
|
|
|
|
class BaseVectorProvider:
|
|
"""向量数据库提供者基础类，所有实际实现需继承该类"""
|
|
|
|
type: str = ""
|
|
label: str = ""
|
|
description: str | None = None
|
|
enabled: bool = True
|
|
config_schema: List[Dict[str, Any]] = []
|
|
|
|
def __init__(self, config: Dict[str, Any] | None = None):
|
|
self.config = config or {}
|
|
|
|
async def initialize(self) -> None:
|
|
"""执行初始化逻辑，例如建立连接"""
|
|
raise NotImplementedError
|
|
|
|
def ensure_collection(self, collection_name: str, vector: bool, dim: int) -> None:
|
|
raise NotImplementedError
|
|
|
|
def upsert_vector(self, collection_name: str, data: Dict[str, Any]) -> None:
|
|
raise NotImplementedError
|
|
|
|
def delete_vector(self, collection_name: str, path: str) -> None:
|
|
raise NotImplementedError
|
|
|
|
def search_vectors(self, collection_name: str, query_embedding, top_k: int):
|
|
raise NotImplementedError
|
|
|
|
def search_by_path(self, collection_name: str, query_path: str, top_k: int):
|
|
raise NotImplementedError
|
|
|
|
def get_all_stats(self) -> Dict[str, Any]:
|
|
raise NotImplementedError
|
|
|
|
def clear_all_data(self) -> None:
|
|
raise NotImplementedError
|