chore: remove useless code (#1106)

Author: iziang

aperag/config.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import json
 import os
 from functools import wraps
 from pathlib import Path
@@ -25,6 +26,8 @@
 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
 from sqlalchemy.orm import Session, sessionmaker
 
+from aperag.vectorstore.connector import VectorStoreConnectorAdaptor
+
 BASE_DIR = Path(__file__).resolve().parent.parent
 
 load_dotenv(os.path.join(BASE_DIR, ".env"))
@@ -290,3 +293,11 @@ async def wrapper(*args, **kwargs):
 
 AsyncSessionDep = Annotated[AsyncSession, Depends(get_async_session)]
 SyncSessionDep = Annotated[Session, Depends(get_sync_session)]
+
+
+def get_vector_db_connector(collection: str) -> VectorStoreConnectorAdaptor:
+    # todo: specify the collection for different user
+    # one person one collection
+    ctx = json.loads(settings.vector_db_context)
+    ctx["collection"] = collection
+    return VectorStoreConnectorAdaptor(settings.vector_db_type, ctx=ctx)

aperag/index/vector_index.py

@@ -24,7 +24,7 @@
 from aperag.llm.embed.embedding_utils import create_embeddings_and_store
 from aperag.utils.tokenizer import get_default_tokenizer
 from aperag.utils.utils import generate_vector_db_collection_name
-from config.vector_db import get_vector_db_connector
+from aperag.config import get_vector_db_connector
 
 logger = logging.getLogger(__name__)
 

aperag/tasks/collection.py

@@ -28,7 +28,7 @@
     generate_fulltext_index_name,
     generate_vector_db_collection_name,
 )
-from config.vector_db import get_vector_db_connector
+from aperag.config import get_vector_db_connector
 
 logger = logging.getLogger(__name__)