diff --git a/api/model/embedding_search/page_index.py b/api/model/embedding_search/page_index.py index c995dce..0081050 100644 --- a/api/model/embedding_search/page_index.py +++ b/api/model/embedding_search/page_index.py @@ -36,16 +36,16 @@ def create_page_index_model(collection_id: int): if collection_id in page_index_model_list: return page_index_model_list[collection_id] else: - class PageIndexModel(AbstractPageIndexModel): + class _PageIndexModel(AbstractPageIndexModel): __tablename__ = "embedding_search_page_index_%s" % str(collection_id) embedding_index = sqlalchemy.Index(__tablename__ + "_embedding_idx", AbstractPageIndexModel.embedding, postgresql_using='ivfflat', postgresql_ops={'embedding': 'vector_cosine_ops'}) - page_index_model_list[collection_id] = PageIndexModel + page_index_model_list[collection_id] = _PageIndexModel - return PageIndexModel + return _PageIndexModel class PageIndexHelper: columns = [