From 55f2b7c4d516541becb451d3f8b511503e00164a Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 23 Jan 2025 18:43:32 +0800 Subject: [PATCH] Code format. (#4611) ### What problem does this PR solve? ### Type of change - [x] Refactoring --- rag/utils/es_conn.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/rag/utils/es_conn.py b/rag/utils/es_conn.py index 02e7b7146..ce05d5d6d 100644 --- a/rag/utils/es_conn.py +++ b/rag/utils/es_conn.py @@ -28,7 +28,6 @@ from rag import settings from rag.settings import TAG_FLD, PAGERANK_FLD from rag.utils import singleton from api.utils.file_utils import get_project_base_directory -import polars as pl from rag.utils.doc_store_conn import DocStoreConnection, MatchExpr, OrderByExpr, MatchTextExpr, MatchDenseExpr, \ FusionExpr from rag.nlp import is_english, rag_tokenizer @@ -142,7 +141,7 @@ class ESConnection(DocStoreConnection): knowledgebaseIds: list[str], aggFields: list[str] = [], rank_feature: dict | None = None - ) -> list[dict] | pl.DataFrame: + ): """ Refers to https://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl.html """ @@ -424,7 +423,7 @@ class ESConnection(DocStoreConnection): logger.debug("ESConnection.delete query: " + json.dumps(qry.to_dict())) for _ in range(ATTEMPT_TIME): try: - print(Search().query(qry).to_dict(), flush=True) + #print(Search().query(qry).to_dict(), flush=True) res = self.es.delete_by_query( index=indexName, body=Search().query(qry).to_dict(),