From e5a8a2d0f98a8bc4499b3e808a67ac753f8da190 Mon Sep 17 00:00:00 2001
From: yueliuli <1628111725@qq.com>
Date: Thu, 23 Apr 2026 16:42:07 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E9=9A=90=E6=82=A3=E7=BA=A7?=
 =?UTF-8?q?=E5=88=AB=E5=88=A4=E5=AE=9A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lib/qwen_fun.py | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/lib/qwen_fun.py b/lib/qwen_fun.py
index 8acc5db..a460e3c 100644
--- a/lib/qwen_fun.py
+++ b/lib/qwen_fun.py
@@ -660,14 +660,22 @@ def get_hazard_level_from_rule(rule_definitions: dict, class_name: str, tag_name
     返回:
         str: 隐患等级（如"重大隐患"），未找到时返回"一般隐患"
     """
-    class_rules = rule_definitions.get(class_name, {})
-    if not class_rules:
-        return "一般隐患"
 
-    for level_name, level_content in class_rules.items():
-        if isinstance(level_content, dict) and tag_name in level_content:
-            return level_name
+    print(f"当前参数: class_name={class_name}, tag_name={tag_name}")
 
+    for scene_name, scene_content in rule_definitions.items():
+        if not isinstance(scene_content, dict):
+            continue
+        for category_name, category_content in scene_content.items():
+            if not isinstance(category_content, dict):
+                continue
+            for level_name, level_content in category_content.items():
+                if isinstance(level_content, dict) and tag_name in level_content:
+                    print(f"找到隐患等级: {level_name}, 类别: {category_name}")
+                    return level_name
+
+    print(f"未找到隐患等级: class_name={class_name}, tag_name={tag_name}")
+    print("默认返回一般隐患")
     return "一般隐患"