diff --git a/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs b/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs
index 00b61df..db695ca 100644
--- a/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs
+++ b/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs
@@ -161,6 +161,14 @@ public sealed class ContextAwareGenerator : MetadataAttributeClassGeneratorBase
         sb.AppendLine("        _contextProvider = provider;");
         sb.AppendLine("    }");
         sb.AppendLine();
+        sb.AppendLine("    /// <summary>");
+        sb.AppendLine("    /// 重置上下文提供者为默认值（用于测试清理）");
+        sb.AppendLine("    /// </summary>");
+        sb.AppendLine("    public static void ResetContextProvider()");
+        sb.AppendLine("    {");
+        sb.AppendLine("        _contextProvider = null;");
+        sb.AppendLine("    }");
+        sb.AppendLine();
     }