Fixes

Signed-off-by: Louis Mandel <[email protected]>

Author: mandel

src/pdl/pdl_llms.py

@@ -183,7 +183,7 @@ def generate_text_stream(
 
 
 def set_structured_decoding_parameters(
-    spec: PdlTypeType,
+    spec: Optional[PdlTypeType],
     parameters: Optional[dict[str, Any]],
 ) -> dict[str, Any]:
     if parameters is None:

src/pdl/pdl_schema_utils.py

@@ -73,15 +73,13 @@ def pdltype_to_jsonschema(
             else:
                 details = constraints.model_dump(exclude_defaults=True)
             schema = {"type": "integer", **details}
-        case ListPdlType(list=ListPdlTypeConstraints()):
-            items_type = PdlTypeParser.model_validate(
-                pdl_type.list.__pydantic_extra__
-            ).root
+        case ListPdlType(list=ListPdlTypeConstraints() as cstr):
+            items_type = PdlTypeParser.model_validate(cstr.__pydantic_extra__).root
             details = {}
-            if pdl_type.list.minItems is not None:
-                details["minItems"] = pdl_type.list.minItems
-            if pdl_type.list.maxItems is not None:
-                details["maxItems"] = pdl_type.list.maxItems
+            if cstr.minItems is not None:
+                details["minItems"] = cstr.minItems
+            if cstr.maxItems is not None:
+                details["maxItems"] = cstr.maxItems
             schema = {
                 "type": "array",
                 "items": pdltype_to_jsonschema(items_type, additional_properties),

tests/test_type_checking.py

@@ -3,6 +3,7 @@
 
 from pdl.pdl import exec_dict
 from pdl.pdl_interpreter import PDLRuntimeError
+from pdl.pdl_parser import PDLParseError
 from pdl.pdl_schema_utils import pdltype_to_jsonschema
 
 _PDLTYPE_TO_JSONSCHEMA_TESTS = [
@@ -314,7 +315,7 @@ def test_function_call7():
 
 
 def test_function_call8():
-    with pytest.raises(PDLRuntimeError):
+    with pytest.raises(PDLParseError):
         exec_dict(function_call8)