Make `generate.json()ˋ work with json schema strings

outlines-dev · Nov 29, 2023 · 12392f2 · 12392f2
1 parent 77378d6
commit 12392f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/outlines/text/generate/regex.py b/outlines/text/generate/regex.py
@@ -393,7 +393,8 @@ def json(
         # TODO: Convert string fields to their respective types
         format_fn = lambda x: pyjson.loads(x)
     else:
-        format_fn = lambda x: x
+        schema = schema_object
+        format_fn = lambda x: pyjson.loads(x)
 
     regex_str = build_regex_from_object(schema)