json: fix explosive integral grammar rules

ggerganov · Mar 31, 2024 · 8451cdb · 8451cdb
1 parent 575b6dc
commit 8451cdb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/json_schema_to_grammar.py b/examples/json_schema_to_grammar.py
@@ -12,8 +12,9 @@
 
 PRIMITIVE_RULES = {
     'boolean': '("true" | "false") space',
-    'integral-part': '[0-9] | [1-9] [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]?',
-    'decimal-part': '[0-9] [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]? [0-9]?',
+    'decimal-part': '[0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] [0-9]?)?)?)?)?)?)?)?)?)?',
+    'integral-part': '[0-9] | [1-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] ([0-9] [0-9]?)?)?)?)?)?)?)?)?)?',
+
     # 'number': '("-"? ([0-9] | [1-9] [0-9]*)) ("." [0-9]+)? ([eE] [-+]? [0-9]+)? space',
     # 'integer': '("-"? ([0-9] | [1-9] [0-9]*)) space',
     'number': '("-"? integral-part) ("." decimal-part)? ([eE] [-+]? integral-part)? space',