From 97227956622a82f08e379514278bc3597db16a46 Mon Sep 17 00:00:00 2001
From: Michal Moskal <michal@moskal.me>
Date: Thu, 22 Feb 2024 01:55:36 +0000
Subject: [PATCH] don't crash on over-constraints

---
 py/pyaici/server.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/py/pyaici/server.py b/py/pyaici/server.py
index 16e10917..55d04dc5 100644
--- 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]):