RAGFlow报错:ESConnection.sql got exception

发布于:2025-05-02 ⋅ 阅读:(92) ⋅ 点赞:(0)

环境:

Ragflowv0.17.2

问题描述:

RAGFlow报错:ESConnection.sql got exception

_ming_cheng_tks, '浙江', 'operator=OR;minimum_should_match=30%') 
2025-04-25 15:55:06,862 INFO     244867 POST http://localhost:1200/_sql?format=json [status:400 duration:0.002s]
2025-04-25 15:55:06,862 ERROR    244867 ESConnection.sql got exception
Traceback (most recent call last):
  File "/home/www/ragflow/ragflow/rag/utils/es_conn.py", line 553, in sql
    res = self.es.sql.query(body={"query": sql, "fetch_size": fetch_size}, format=format,
  File "/home/www/ragflow/ragflow/.venv/lib/python3.10/site-packages/elasticsearch/_sync/client/utils.py", line 446, in wrapped
    return api(*args, **kwargs)
  File "/home/www/ragflow/ragflow/.venv/lib/python3.10/site-packages/elasticsearch/_sync/client/sql.py", line 330, in query
    return self.perform_request(  # type: ignore[return-value]
  File "/home/www/ragflow/ragflow/.venv/lib/python3.10/site-packages/elasticsearch/_sync/client/_base.py", line 389, in perform_request
    return self._client.perform_request(
  File "/home/www/ragflow/ragflow/.venv/lib/python3.10/site-packages/elasticsearch/_sync/client/_base.py", line 320, in perform_request
    raise HTTP_EXCEPTIONS.get(meta.status, ApiError)(

解决方案:

1.主要修改点说明:

  1. 正则表达式增强
pattern = r"([a-z_]+_l?tks)\s+(like|=)\s*'([^']+)'"
  • 支持大小写混合的字段名
  • 允许等号(=)和LIKE操作符
  • 更严格捕获值部分
  1. 分词处理优化
tokenized = rag_tokenizer.fine_grained_tokenize(rag_tokenizer.tokenize(val))
  • 确保中文值被正确分词
  • 处理特殊字符转义
  1. 参数标准化
'operator=OR, minimum_should_match=30%'
  • 使用逗号替代分号作为参数分隔符
  • 符合Elasticsearch SQL参数规范
  1. 保留大小写
re.sub(re.escape(old), new, sql, flags=re.IGNORECASE)
  • 保持原始SQL的大小写格式
  • 避免意外修改其他部分
  1. 超时时间优化
params={"request_timeout": 30}
  • 从2秒延长到30秒
  • 适应复杂查询场景
  1. 错误日志增强
logger.error(f"ES SQL Error: {str(e)} \nQuery: {sql}")
  • 记录完整错误信息
  • 保留问题查询语句

典型转换示例:

原始查询:

SELECT * WHERE _ming_cheng_tks LIKE '浙江'

转换后:

SELECT * WHERE MATCH(_ming_cheng_tks, '浙 江', 'operator=OR, minimum_should_match=30%')

验证方法:

# 测试用例
test_sql = "SELECT _id FROM index WHERE name_tks = '杭州' OR addr_ltks LIKE '西湖区'"
expected = "SELECT _id FROM index WHERE MATCH(name_tks, '杭 州', 'operator=OR, minimum_should_match=30%') OR MATCH(addr_ltks, '西 湖 区', 'operator=OR, minimum_should_match=30%')"

2.修改源码es_conn.py文件路径和详情

大概531行

ragflow-main\rag\utils\es_conn.py

    def sql(self, sql: str, fetch_size: int, format: str):
        logger.debug(f"ESConnection.sql get sql: {sql}")
        sql = re.sub(r"[ `]+", " ", sql)
        sql = sql.replace("%", "")
        replaces = []
        # 修改点1:增强正则匹配模式
        pattern = r"([a-z_]+_l?tks)\s+(like|=)\s*'([^']+)'"
        for r in re.finditer(pattern, sql, re.IGNORECASE):
            fld, op, val = r.group(1), r.group(2), r.group(3)
            # 修改点2:正确处理分词和特殊字符
            tokenized = rag_tokenizer.fine_grained_tokenize(rag_tokenizer.tokenize(val))
            # 修改点3:参数格式标准化
            match_expr = f"MATCH({fld}, '{tokenized}', 'operator=OR, minimum_should_match=30%')" 
            replaces.append((f"{fld} {op} '{val}'", match_expr))
        
        # 修改点4:保留原始大小写格式
        for old, new in replaces:
            sql = re.sub(re.escape(old), new, sql, flags=re.IGNORECASE)
        
        logger.debug(f"ESConnection.sql transformed: {sql}")

        # 修改点5:增加超时时间
        for i in range(ATTEMPT_TIME):
            try:
                res = self.es.sql.query(
                    body={"query": sql, "fetch_size": fetch_size},
                    format=format,
                    params={"request_timeout": 30}  # 从2秒增加到30秒
                )
                return res
            except ConnectionTimeout:
                logger.exception(f"ESConnection.sql timeout on: {sql}")
                continue
            except Exception as e:
                # 修改点6:记录完整错误信息
                logger.error(f"ES SQL Error: {str(e)} \nQuery: {sql}")
                return None

在这里插入图片描述
3.重启服务后验证正常
在这里插入图片描述


网站公告

今日签到

点亮在社区的每一天
去签到