Use outlines-core to translate JSON Schemas into regexes

dottxt-ai · Dec 9, 2024 · 9aca8c7 · 9aca8c7
1 parent e4f96fb
commit 9aca8c7
Show file tree

Hide file tree

Showing 7 changed files with 56 additions and 543 deletions.
diff --git a/benchmarks/bench_json_schema.py b/benchmarks/bench_json_schema.py
@@ -1,6 +1,7 @@
+from outlines_core.fsm.json_schema import build_regex_from_schema
+
 from outlines.caching import cache_disabled
 from outlines.fsm.guide import RegexGuide
-from outlines.fsm.json_schema import build_regex_from_schema
 
 from .common import setup_tokenizer  # noqa: E402
 
@@ -70,10 +71,6 @@ def setup(self, schema_name):
         self.tokenizer = setup_tokenizer()
         self.schema = schemas[schema_name]
 
-    @cache_disabled()
-    def time_json_schema_to_regex(self, schema_name):
-        build_regex_from_schema(self.schema)
-
     @cache_disabled()
     def time_json_schema_to_fsm(self, schema_name):
         regex = build_regex_from_schema(self.schema)