add dockerfile for cuda envirement. Refine table search strategy, (#123)

2024-03-14 19:45:29 +08:00
parent 937048e5fb
commit 675a9f8d9a
18 changed files with 259 additions and 84 deletions
--- a/rag/utils/es_conn.py
+++ b/rag/utils/es_conn.py
@@ -249,6 +249,8 @@ class HuEs:
            except ConnectionTimeout as e:
                es_logger.error("Timeout【Q】：" + sql)
                continue
+            except Exception as e:
+                raise e
        es_logger.error("ES search timeout for 3 times!")
        raise ConnectionTimeout()