From ecdbcd815ed15f66ad3a4dbf5de7d8e7d0d9fc07 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Sun, 15 Sep 2024 18:25:19 -0700 Subject: [PATCH] Simplify code to remove json codeblock from AI response string --- src/khoj/processor/conversation/utils.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py index dbfbfa2a..b7794361 100644 --- a/src/khoj/processor/conversation/utils.py +++ b/src/khoj/processor/conversation/utils.py @@ -291,8 +291,6 @@ def reciprocal_conversation_to_chatml(message_pair): return [ChatMessage(content=message, role=role) for message, role in zip(message_pair, ["user", "assistant"])] -def remove_json_codeblock(response): +def remove_json_codeblock(response: str): """Remove any markdown json codeblock formatting if present. Useful for non schema enforceable models""" - if response.startswith("```json") and response.endswith("```"): - response = response[7:-3] - return response + return response.removeprefix("```json").removesuffix("```")