Merge pull request #119 from Suraj3620/pydantic-to-schemas

Added helper function to convert pydantic model to schema

ipywidgets_jsonschema/form.py

@@ -13,6 +13,7 @@
 import re
 import traitlets
 import collections.abc
+from pydantic import BaseModel
 
 # We are providing some compatibility for ipywidgets v7 and v8
 IS_VERSION_8 = version.parse(ipywidgets.__version__).major == 8
@@ -41,6 +42,15 @@ def as_tuple(obj):
         return (obj,)
 
 
+def convert_pydantic_to_schema(model) -> dict:
+    """
+    Converts a Pydantic model class to its corresponding JSON schema.
+    """
+    if isinstance(model, type) and issubclass(model, BaseModel):
+        return model.model_json_schema()
+    return model
+
+
 def minmax_schema_rule(widget, schema):
     """This implements the minimum/maximum rules
 
@@ -115,6 +125,7 @@ def __init__(
             A function that is used to sort the keys in a dictionary. Defaults to
             the built-in sorted, but is no-op if sorted raises a TypeError.
         """
+        schema = convert_pydantic_to_schema(schema)
         # Make sure that the given schema is valid
         Draft7Validator.check_schema(schema)
 

pyproject.toml

@@ -26,6 +26,7 @@ dependencies = [
     "jsonschema<5",
     "packaging",
     "traitlets",
+    "pydantic",
 ]
 
 [tool.setuptools]