fix: add default value for optional fields (#11)

dylibso · Nov 11, 2024 · 4828a02 · 4828a02
1 parent 60242c8
commit 4828a02
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/template/plugin/pdk_types.py.ejs b/template/plugin/pdk_types.py.ejs
@@ -21,7 +21,20 @@ class <%- capitalize(schema.name) %>(extism.Json):
 <% if (p.description) { -%>
   # <%- formatCommentBlock(p.description, "# ") %>
 <% } -%>
+<% if (!p.nullable) {%>
   <%- p.name %>: <%- toPythonType(p) %>
+<% } %>
+<% }) %>
+
+<% schema.properties.forEach(p => { -%>
+<% if (p.description) { -%>
+  # <%- formatCommentBlock(p.description, "# ") %>
+<% } -%>
+<% if (p.nullable) {%>
+  <%- p.name %>: <%- toPythonType(p) %> = None
+<% } %>
 <% }) %>
+
 <% } %>
 <% }); %>
+