fix: Strip closing ct:trimmed tag

julep-ai · May 28, 2024 · e2eec7f · e2eec7f
1 parent 7f286dd
commit e2eec7f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agents-api/agents_api/rec_sum/trim.py b/agents-api/agents_api/rec_sum/trim.py
@@ -62,7 +62,9 @@ async def trim_messages(
     )
 
     assert "<ct:trimmed>" in result["content"]
-    trimmed_messages = json.loads(result["content"].split("<ct:trimmed>")[-1].strip())
+    trimmed_messages = json.loads(
+        result["content"].split("<ct:trimmed>")[-1].replace("</ct:trimmed>", "").strip()
+    )
 
     assert all((msg.get("index") is not None for msg in trimmed_messages))