feat: add thinking display

Cinnamon · Feb 5, 2025 · 9286bbb · 9286bbb
1 parent e90f895
commit 9286bbb
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/libs/kotaemon/kotaemon/indices/qa/utils.py b/libs/kotaemon/kotaemon/indices/qa/utils.py
@@ -80,3 +80,18 @@ def find_start_end_phrase(
         final_match = None
 
     return final_match, matched_length
+
+
+def replace_think_tag_with_details(text):
+    text = text.replace(
+        "<think>",
+        '<details><summary><span style="color:grey">Thought</span></summary><blockquote>',  # noqa
+    )
+    text = text.replace("</think>", "</blockquote></details>")
+    return text
+
+
+def strip_think_tag(text):
+    if "</think>" in text:
+        text = text.split("</think>")[1]
+    return text