diff --git a/lmformatenforcer/regexparser.py b/lmformatenforcer/regexparser.py index c782531..ed134b3 100644 --- a/lmformatenforcer/regexparser.py +++ b/lmformatenforcer/regexparser.py @@ -51,7 +51,7 @@ def add_character(self, new_character: str) -> 'RegexParser': return RegexParser(self.context, self.config, RegexParser.INVALID_STATE) def can_end(self) -> bool: - return self.current_state in self.context.pattern.finals + return self.current_state in self.context.pattern.finals or self.current_state == RegexParser.INVALID_STATE def get_allowed_characters(self) -> str: if self.current_state not in self.context.pattern.map: