AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 23:56:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Code format. (#4611)

Browse Source 
### What problem does this PR solve?


### Type of change

- [x] Refactoring
This commit is contained in:
Kevin Hu 2025-01-23 18:43:32 +08:00 committed by GitHub
parent 07b3e55903
commit 55f2b7c4d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rag/utils/es_conn.py
View File

@ -28,7 +28,6 @@ from rag import settings
from rag.settings import TAG_FLD, PAGERANK_FLD from rag.settings import TAG_FLD, PAGERANK_FLD
from rag.utils import singleton from rag.utils import singleton
from api.utils.file_utils import get_project_base_directory 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, \ from rag.utils.doc_store_conn import DocStoreConnection, MatchExpr, OrderByExpr, MatchTextExpr, MatchDenseExpr, \
FusionExpr FusionExpr
from rag.nlp import is_english, rag_tokenizer from rag.nlp import is_english, rag_tokenizer
@ -142,7 +141,7 @@ class ESConnection(DocStoreConnection):
knowledgebaseIds: list[str], knowledgebaseIds: list[str],
aggFields: list[str] = [], aggFields: list[str] = [],
rank_feature: dict | None = None rank_feature: dict | None = None
) -> list[dict] | pl.DataFrame: ):
""" """
Refers to https://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl.html 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())) logger.debug("ESConnection.delete query: " + json.dumps(qry.to_dict()))
for _ in range(ATTEMPT_TIME): for _ in range(ATTEMPT_TIME):
try: try:
print(Search().query(qry).to_dict(), flush=True) #print(Search().query(qry).to_dict(), flush=True)
res = self.es.delete_by_query( res = self.es.delete_by_query(
index=indexName, index=indexName,
body=Search().query(qry).to_dict(), body=Search().query(qry).to_dict(),