diff --git a/backend/open_webui/retrieval/models/colbert.py b/backend/open_webui/retrieval/models/colbert.py index d11bde1b09..94e6f8af5e 100644 --- a/backend/open_webui/retrieval/models/colbert.py +++ b/backend/open_webui/retrieval/models/colbert.py @@ -12,7 +12,7 @@ log = logging.getLogger(__name__) class ColBERT(BaseReranker): def __init__(self, name, **kwargs) -> None: - log.info('ColBERT: Loading model', name) + log.info('ColBERT: Loading model %s', name) self.device = 'cuda' if torch.cuda.is_available() else 'cpu' DOCKER = kwargs.get('env') == 'docker' diff --git a/backend/open_webui/retrieval/vector/dbs/opengauss.py b/backend/open_webui/retrieval/vector/dbs/opengauss.py index 900713a7a1..e11e944f3d 100644 --- a/backend/open_webui/retrieval/vector/dbs/opengauss.py +++ b/backend/open_webui/retrieval/vector/dbs/opengauss.py @@ -62,7 +62,6 @@ from open_webui.config import ( OPENGAUSS_POOL_SIZE, OPENGAUSS_POOL_TIMEOUT, ) -from open_webui.env import SRC_LOG_LEVELS from open_webui.retrieval.vector.main import ( GetResult, SearchResult, @@ -75,7 +74,6 @@ VECTOR_LENGTH = OPENGAUSS_INITIALIZE_MAX_VECTOR_LENGTH Base = declarative_base() log = logging.getLogger(__name__) -log.setLevel(SRC_LOG_LEVELS['RAG']) class DocumentChunk(Base):