14 lines
379 B
Python
14 lines
379 B
Python
|
# 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
|