cleanup: only set default values for non-required fields

dylibso · Nov 19, 2024 · 59738e2 · 59738e2
1 parent 4828a02
commit 59738e2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/template/plugin/pdk_types.py.ejs b/template/plugin/pdk_types.py.ejs
@@ -21,7 +21,7 @@ class <%- capitalize(schema.name) %>(extism.Json):
 <% if (p.description) { -%>
   # <%- formatCommentBlock(p.description, "# ") %>
 <% } -%>
-<% if (!p.nullable) {%>
+<% if (!p.nullable || p.required) {%>
   <%- p.name %>: <%- toPythonType(p) %>
 <% } %>
 <% }) %>
@@ -30,8 +30,8 @@ class <%- capitalize(schema.name) %>(extism.Json):
 <% if (p.description) { -%>
   # <%- formatCommentBlock(p.description, "# ") %>
 <% } -%>
-<% if (p.nullable) {%>
-  <%- p.name %>: <%- toPythonType(p) %> = None
+<% if (p.nullable && !p.required) {%>
+  <%- p.name %>: <%- toPythonType(p) %>
 <% } %>
 <% }) %>