diff --git a/AutomaticInterface/AutomaticInterface/AutomaticInterfaceGenerator.cs b/AutomaticInterface/AutomaticInterface/AutomaticInterfaceGenerator.cs
index aa8a5b6..f147501 100644
--- a/AutomaticInterface/AutomaticInterface/AutomaticInterfaceGenerator.cs
+++ b/AutomaticInterface/AutomaticInterface/AutomaticInterfaceGenerator.cs
@@ -12,6 +12,7 @@ public class AutomaticInterfaceGenerator : IIncrementalGenerator
public const string IgnoreAutomaticInterfaceAttributeName = "IgnoreAutomaticInterface";
public const string NamespaceParameterName = "namespaceName";
public const string InterfaceParameterName = "interfaceName";
+ public const string AsInternalParameterName = "asInternal";
public void Initialize(IncrementalGeneratorInitializationContext context)
{
diff --git a/AutomaticInterface/AutomaticInterface/RegisterAttributesExtensions.cs b/AutomaticInterface/AutomaticInterface/RegisterAttributesExtensions.cs
index c10df77..a3d3785 100644
--- a/AutomaticInterface/AutomaticInterface/RegisterAttributesExtensions.cs
+++ b/AutomaticInterface/AutomaticInterface/RegisterAttributesExtensions.cs
@@ -24,12 +24,13 @@ namespace AutomaticInterface
///
/// Use source generator to automatically create a Interface from this class
///
- /// Namespace name for the generated interface. Defaults to the same namespace as the class.
- /// Interface name for the generated interface. Defaults to an interface version of the class name.
[AttributeUsage(AttributeTargets.Class)]
internal sealed class {{{AutomaticInterfaceGenerator.DefaultAttributeName}}}Attribute : Attribute
{
- internal {{{AutomaticInterfaceGenerator.DefaultAttributeName}}}Attribute(string {{{AutomaticInterfaceGenerator.NamespaceParameterName}}} = "", string {{{AutomaticInterfaceGenerator.InterfaceParameterName}}} = "", bool asInternal = false) { }
+ /// Namespace name for the generated interface. Defaults to the same namespace as the class.
+ /// Interface name for the generated interface. Defaults to an interface version of the class name.
+ /// If true, set the visibility of the interface to internal.
+ internal {{{AutomaticInterfaceGenerator.DefaultAttributeName}}}Attribute(string {{{AutomaticInterfaceGenerator.NamespaceParameterName}}} = "", string {{{AutomaticInterfaceGenerator.InterfaceParameterName}}} = "", bool {{{AutomaticInterfaceGenerator.AsInternalParameterName}}} = false) { }
}
}
""",