feat: add parameter attribute to allow customization of name, description, and requiredness

2024-06-30 12:21:30 -05:00
parent 04343bf732
commit 3006ad9b50
3 changed files with 101 additions and 3 deletions
@@ -36,9 +36,11 @@ static class JsonSchemaGenerator
        continue;
      }

-      var paramName = parameter.Name;
-      var paramDescription = string.Empty;
-      var paramRequired = parameter.HasDefaultValue;
+      var attribute = parameter.GetCustomAttribute<FunctionParameterAttribute>();
+
+      var paramName = attribute?.Name ?? parameter.Name;
+      var paramDescription = attribute?.Description ?? string.Empty;
+      var paramRequired = attribute?.Required ?? !parameter.HasDefaultValue;

      var paramObject = new JsonObject();
      paramObject[DescriptionKey] = paramDescription;