# import the ObjectID class
from .object import ObjectID

# define a vector base class
class VectorBase:
    def __init__(self, db_url, model_name) -> None:
        self.db_url = db_url
        self.model_name = model_name

    async def insert(self, vector: [float], id: ObjectID):
        pass

    async def query(self, vector: [float], top_k: int) -> [ObjectID]:
        pass