From 061b725e00192b17d057a1c3ecad3e9a98a502d6 Mon Sep 17 00:00:00 2001 From: Ivan Ogasawara Date: Thu, 21 Nov 2024 14:03:44 -0400 Subject: [PATCH] refactor: Move db package to augmented package --- src/rago/augmented/base.py | 2 +- src/rago/{ => augmented}/db/__init__.py | 4 ++-- src/rago/{ => augmented}/db/base.py | 0 src/rago/{ => augmented}/db/faiss.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) rename src/rago/{ => augmented}/db/__init__.py (54%) rename src/rago/{ => augmented}/db/base.py (100%) rename src/rago/{ => augmented}/db/faiss.py (94%) diff --git a/src/rago/augmented/base.py b/src/rago/augmented/base.py index a2434ee..7787119 100644 --- 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 b/src/rago/augmented/db/__init__.py similarity index 54% rename from src/rago/db/__init__.py rename to src/rago/augmented/db/__init__.py index 365c6c9..7bee677 100644 --- a/src/rago/db/__init__.py +++ b/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 b/src/rago/augmented/db/base.py similarity index 100% rename from src/rago/db/base.py rename to src/rago/augmented/db/base.py diff --git a/src/rago/db/faiss.py b/src/rago/augmented/db/faiss.py similarity index 94% rename from src/rago/db/faiss.py rename to src/rago/augmented/db/faiss.py index cc4eadf..81d287c 100644 --- a/src/rago/db/faiss.py +++ b/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