diff --git a/frontend.rb b/frontend.rb index b48f7bafd0cb876db6016386b15adbdbf2f63c50..47149fdcbebb51d9c2a5ca2c8600192d16e5a6ce 100755 --- a/frontend.rb +++ b/frontend.rb @@ -35,7 +35,7 @@ Thread.new do if c == "\n" if cur_line.include?('<think>') thinks = 0 - puts "<think>" + puts elsif cur_line.include?('</think>') thinks = nil puts "\n</think>" # now we have 2 newlines