[feat] Refactor VectorDB class hierarchy for flexibility

This commit is contained in:
Sayo
2023-06-22 12:15:26 +08:00
parent 973dc5434f
commit 85a6a0c161
3 changed files with 43 additions and 35 deletions

View File

@@ -0,0 +1,10 @@
class BaseVectorDB:
def __init__(self):
self.client = self._get_or_create_db()
self.collection = self._get_or_create_collection()
def _get_or_create_db(self):
raise NotImplementedError
def _get_or_create_collection(self):
raise NotImplementedError