Soulter
47 lines
1.0 KiB
Python
47 lines
1.0 KiB
Python
import abc
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class Result:
|
|
similarity: float
|
|
data: dict
|
|
|
|
|
|
class BaseVecDB:
|
|
async def initialize(self):
|
|
"""
|
|
初始化向量数据库
|
|
"""
|
|
pass
|
|
|
|
@abc.abstractmethod
|
|
async def insert(self, content: str, metadata: dict = None, id: str = None) -> int:
|
|
"""
|
|
插入一条文本和其对应向量，自动生成 ID 并保持一致性。
|
|
"""
|
|
...
|
|
|
|
@abc.abstractmethod
|
|
async def retrieve(self, query: str, top_k: int = 5) -> list[Result]:
|
|
"""
|
|
搜索最相似的文档。
|
|
Args:
|
|
query (str): 查询文本
|
|
top_k (int): 返回的最相似文档的数量
|
|
Returns:
|
|
List[Result]: 查询结果
|
|
"""
|
|
...
|
|
|
|
@abc.abstractmethod
|
|
async def delete(self, doc_id: str) -> bool:
|
|
"""
|
|
删除指定文档。
|
|
Args:
|
|
doc_id (str): 要删除的文档 ID
|
|
Returns:
|
|
bool: 删除是否成功
|
|
"""
|
|
...
|