fix: Strip closing ct:summarized-messages tag

julep-ai · May 28, 2024 · 2dfbe42 · 2dfbe42
1 parent e2eec7f
commit 2dfbe42
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agents-api/agents_api/rec_sum/summarize.py b/agents-api/agents_api/rec_sum/summarize.py
@@ -77,7 +77,10 @@ async def summarize_messages(
 
     assert "<ct:summarized-messages>" in result["content"]
     summarized_messages = json.loads(
-        result["content"].split("<ct:summarized-messages>")[-1].strip()
+        result["content"]
+        .split("<ct:summarized-messages>")[-1]
+        .replace("/<ct:summarized-messages>", "")
+        .strip()
     )
 
     assert all((msg.get("summarizes") is not None for msg in summarized_messages))