Improve: Support xgrammar 0.1.14 (#3593)

2025-02-27 16:42:54 +00:00
parent b0df5d240b
commit d281587989
2 changed files with 2 additions and 5 deletions
--- a/python/sglang/srt/constrained/xgrammar_backend.py
+++ b/python/sglang/srt/constrained/xgrammar_backend.py
@@ -19,7 +19,6 @@ from typing import List, Tuple
 import torch
 from xgrammar import (
    CompiledGrammar,
-    Grammar,
    GrammarCompiler,
    GrammarMatcher,
    TokenizerInfo,
@@ -135,9 +134,7 @@ class XGrammarGrammarBackend(BaseGrammarBackend):
                return None
        elif key_type == "regex":
            try:
-                ctx = self.grammar_compiler.compile_grammar(
-                    Grammar.from_regex(key_string)
-                )
+                ctx = self.grammar_compiler.compile_regex(key_string)
            except RuntimeError as e:
                logging.warning(f"Skip invalid regex: regex={key_string}, {e=}")
                return None