Check the JSON Schema before translating it

environment.yml

@@ -17,6 +17,7 @@ dependencies:
   - pytest
   - pre-commit
   - referencing
+  - jsonschema
   - transformers
   - pip
   - pip:

outlines/text/json_schema.py

@@ -2,6 +2,7 @@
 import json
 import re
 
+from jsonschema.protocols import Validator
 from referencing import Registry, Resource
 from referencing._core import Resolver
 from referencing.jsonschema import DRAFT202012
@@ -37,6 +38,7 @@ def build_regex_from_schema(schema: str):
     follows the schema.
 
     """
+    Validator.check_schema(schema)
     schema = json.loads(schema)
 
     # Build reference resolver

pyproject.toml

@@ -36,6 +36,7 @@ dependencies = [
    "numba",
    "joblib",
    "referencing",
+   "jsonschema",
 ]
 dynamic = ["version"]
 
@@ -86,6 +87,7 @@ exclude=["examples"]
 module = [
     "jinja2",
     "joblib.*",
+    "jsonschema.*",
     "openai",
     "numpy.*",
     "perscache.*",