don't crash on over-constraints

microsoft · Feb 22, 2024 · 9722795 · 9722795
1 parent 111a0e6
commit 9722795
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/py/pyaici/server.py b/py/pyaici/server.py
@@ -223,8 +223,9 @@ def mid_process(self) -> MidProcessResult:
         if log_level >= 2:
             print("ALLOW:", bias)
         if bias.num_set() == 0:
-            print("Constraint doesn't allow any tokens")
-            raise StopIteration
+            if log_level >= 1:
+                print("Constraint doesn't allow any tokens; adding EOS")
+            bias[eos_token()] = True
         return MidProcessResult.bias(bias)
 
     def post_process(self, tokens: List[Token]):