drjones/draco
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

CoT filter: handle missing opening think-tag (vornith preamble leak)

Browse Source
This commit is contained in:
drjones
2026-10-01 17:29:44 -07:00
parent e5d85a912e
commit fe67d0114a
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
draco_core.py
View File

@@ -179,16 +179,22 @@ class CoTFilter:
self.in_think = False self.in_think = False
self.buf = self.buf[i + len(self.CLOSE):] self.buf = self.buf[i + len(self.CLOSE):]
else: else:
i = self.buf.find(self.OPEN) i_open = self.buf.find(self.OPEN)
if i == -1: i_close = self.buf.find(self.CLOSE)
keep = min(len(self.buf), len(self.OPEN) - 1) # close-tag with no open-tag seen: drop the CoT preamble wholesale
if i_close != -1 and (i_open == -1 or i_close < i_open):
self.buf = self.buf[i_close + len(self.CLOSE):]
self.in_think = False
continue
if i_open == -1:
keep = min(len(self.buf), max(len(self.OPEN), len(self.CLOSE)) - 1)
cut = len(self.buf) - keep cut = len(self.buf) - keep
out += self.buf[:cut] out += self.buf[:cut]
self.buf = self.buf[cut:] self.buf = self.buf[cut:]
return out return out
out += self.buf[:i] out += self.buf[:i_open]
self.in_think = True self.in_think = True
self.buf = self.buf[i + len(self.OPEN):] self.buf = self.buf[i_open + len(self.OPEN):]
def flush(self): def flush(self):
out, self.buf = ("" if self.in_think else self.buf), "" out, self.buf = ("" if self.in_think else self.buf), ""