Update pattern match

njbbaer · May 4, 2024 · f36ea1c · f36ea1c
1 parent 6892d43
commit f36ea1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/simulacrum.py b/src/simulacrum.py
@@ -65,7 +65,7 @@ def get_conversation_cost(self):
         return self.context.conversation_cost
 
     def _filter_hidden(self, response):
-        response = re.sub(r"<THINK>.*?</>", "", response, flags=re.DOTALL)
+        response = re.sub(r"<think>.*?</think>", "", response, flags=re.DOTALL)
         return response.strip()
 
     def _set_stats(self, completion):