diff --git a/src/dataset/pinecone_handler.py b/src/dataset/pinecone_db_handler.py similarity index 99% rename from src/dataset/pinecone_handler.py rename to src/dataset/pinecone_db_handler.py index bca5273..6f53daa 100644 --- a/src/dataset/pinecone_handler.py +++ b/src/dataset/pinecone_db_handler.py @@ -4,13 +4,13 @@ import json import pinecone import os +from .settings import PINECONE_INDEX_NAME, PINECONE_VALUES_DIMS, PINECONE_METRIC, PINECONE_METADATA_ENTRIES + import logging logger = logging.getLogger(__name__) -from .settings import PINECONE_INDEX_NAME, PINECONE_VALUES_DIMS, PINECONE_METRIC, PINECONE_METADATA_ENTRIES - -class PineconeHandler: +class PineconeDB: def __init__( self, create_index: bool = False,