refactor: Move db package to augmented package (#27)

osl-incubator · Nov 21, 2024 · b7a13b7 · b7a13b7
1 parent 58c830d
commit b7a13b7
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/src/rago/augmented/base.py b/src/rago/augmented/base.py
@@ -7,7 +7,7 @@
 
 from typeguard import typechecked
 
-from rago.db import DBBase, FaissDB
+from rago.augmented.db import DBBase, FaissDB
 
 if TYPE_CHECKING:
     import numpy as np

diff --git a/src/rago/db/__init__.py → src/rago/augmented/db/__init__.py b/src/rago/db/__init__.py → src/rago/augmented/db/__init__.py
@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
-from rago.db.base import DBBase
-from rago.db.faiss import FaissDB
+from rago.augmented.db.base import DBBase
+from rago.augmented.db.faiss import FaissDB
 
 __all__ = [
     'DBBase',

diff --git a/src/rago/db/base.py → src/rago/augmented/db/base.py b/src/rago/db/base.py → src/rago/augmented/db/base.py
diff --git a/src/rago/db/faiss.py → src/rago/augmented/db/faiss.py b/src/rago/db/faiss.py → src/rago/augmented/db/faiss.py
@@ -8,7 +8,7 @@
 
 from typeguard import typechecked
 
-from rago.db.base import DBBase
+from rago.augmented.db.base import DBBase
 
 
 @typechecked