瀏覽代碼

Fix: the output log is incorrect (#8577)

### What problem does this PR solve?

Fix: the output log is incorrect

### Type of change

- [ ] Bug Fix (non-breaking change which fixes an issue)

Co-authored-by: liang <xiaofeng.liang@landstech.com.cn>
tags/v0.20.0
symvation 4 月之前
父節點
當前提交
32f8b3ad77
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 6 行新增6 行删除
  1. 3
    3
      rag/utils/es_conn.py
  2. 3
    3
      rag/utils/opensearch_coon.py

+ 3
- 3
rag/utils/es_conn.py 查看文件

if str(e).find("Timeout") > 0: if str(e).find("Timeout") > 0:
continue continue
raise e raise e
logger.error("ESConnection.search timeout for 3 times!")
logger.error(f"ESConnection.search timeout for {ATTEMPT_TIME} times!")
raise Exception("ESConnection.search timeout.") raise Exception("ESConnection.search timeout.")


def get(self, chunkId: str, indexName: str, knowledgebaseIds: list[str]) -> dict | None: def get(self, chunkId: str, indexName: str, knowledgebaseIds: list[str]) -> dict | None:
if str(e).find("Timeout") > 0: if str(e).find("Timeout") > 0:
continue continue
raise e raise e
logger.error("ESConnection.get timeout for 3 times!")
logger.error(f"ESConnection.get timeout for {ATTEMPT_TIME} times!")
raise Exception("ESConnection.get timeout.") raise Exception("ESConnection.get timeout.")


def insert(self, documents: list[dict], indexName: str, knowledgebaseId: str = None) -> list[str]: def insert(self, documents: list[dict], indexName: str, knowledgebaseId: str = None) -> list[str]:
except Exception: except Exception:
logger.exception("ESConnection.sql got exception") logger.exception("ESConnection.sql got exception")
return None return None
logger.error("ESConnection.sql timeout for 3 times!")
logger.error(f"ESConnection.sql timeout for {ATTEMPT_TIME} times!")
return None return None

+ 3
- 3
rag/utils/opensearch_coon.py 查看文件

if str(e).find("Timeout") > 0: if str(e).find("Timeout") > 0:
continue continue
raise e raise e
logger.error("OSConnection.search timeout for 3 times!")
logger.error(f"OSConnection.search timeout for {ATTEMPT_TIME} times!")
raise Exception("OSConnection.search timeout.") raise Exception("OSConnection.search timeout.")


def get(self, chunkId: str, indexName: str, knowledgebaseIds: list[str]) -> dict | None: def get(self, chunkId: str, indexName: str, knowledgebaseIds: list[str]) -> dict | None:
if str(e).find("Timeout") > 0: if str(e).find("Timeout") > 0:
continue continue
raise e raise e
logger.error("OSConnection.get timeout for 3 times!")
logger.error(f"OSConnection.get timeout for {ATTEMPT_TIME} times!")
raise Exception("OSConnection.get timeout.") raise Exception("OSConnection.get timeout.")


def insert(self, documents: list[dict], indexName: str, knowledgebaseId: str = None) -> list[str]: def insert(self, documents: list[dict], indexName: str, knowledgebaseId: str = None) -> list[str]:
except Exception: except Exception:
logger.exception("OSConnection.sql got exception") logger.exception("OSConnection.sql got exception")
return None return None
logger.error("OSConnection.sql timeout for 3 times!")
logger.error(f"OSConnection.sql timeout for {ATTEMPT_TIME} times!")
return None return None

Loading…
取消
儲存