Remove markdown code block from llm answer

pierrelefevre · Oct 15, 2024 · 16cc347 · 16cc347
1 parent b4b64c5
commit 16cc347
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/aggregator/llm.py b/aggregator/llm.py
@@ -127,6 +127,7 @@ def _group_headlines(headlines):
         ],
     )
     message = completion.choices[0].message.content
+    message = message.replace("```", "").replace("json", "").strip()
 
     groups = json.loads(message)
     return groups