feat: add default_default and default_annotation control in json_sche…

…ma_to_signature
i2mint · Mar 22, 2024 · f4a0457 · f4a0457
1 parent 3a81571
commit f4a0457
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/ju/json_schema.py b/ju/json_schema.py
@@ -211,7 +211,11 @@ def function_to_json_schema(
 
 
 def json_schema_to_signature(
-    json_schema: dict, *, type_mapper: Mapper = DFLT_JSON_PY_TYPE_PAIRS
+    json_schema: dict, 
+    *, 
+    type_mapper: Mapper = DFLT_JSON_PY_TYPE_PAIRS,
+    default_default=Parameter.empty,
+    default_annotation=Parameter.empty,
 ):
     """
     Transforms a JSON schema to a Python function signature.
@@ -238,7 +242,7 @@ def json_schema_to_signature(
 
 
     """
-    type_mapper = ensure_callable_mapper(type_mapper)
+    type_mapper = ensure_callable_mapper(type_mapper, default=default_annotation)
     properties = json_schema['properties']
 
     def _params():
@@ -252,7 +256,7 @@ def _params():
                     py_type = Union[tuple(json_types)]
             else:
                 py_type = Parameter.empty
-            default = field.get('default', Parameter.empty)
+            default = field.get('default', default_default)
             yield Parameter(
                 name=name,
                 annotation=py_type,