Przeglądaj źródła

Code format. (#4611)

### What problem does this PR solve?


### Type of change

- [x] Refactoring
tags/v0.16.0
Kevin Hu 9 miesięcy temu
rodzic
commit
55f2b7c4d5
No account linked to committer's email address
1 zmienionych plików z 2 dodań i 3 usunięć
  1. 2
    3
      rag/utils/es_conn.py

+ 2
- 3
rag/utils/es_conn.py Wyświetl plik

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
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
""" """
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(),

Ładowanie…
Anuluj
Zapisz