From 17997d26c5c56e1ca7889f79c236f2fee7ca9ad6 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Thu, 3 Apr 2025 14:07:15 +0530 Subject: [PATCH] Make generated code block extractor less strict to improve code tool --- src/khoj/processor/tools/run_code.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/khoj/processor/tools/run_code.py b/src/khoj/processor/tools/run_code.py index 43a47204..1a34c8aa 100644 --- a/src/khoj/processor/tools/run_code.py +++ b/src/khoj/processor/tools/run_code.py @@ -161,7 +161,7 @@ async def generate_python_code( ) # Extract python code wrapped in markdown code blocks from the response - code_blocks = re.findall(r"```(?:python)?\n(.*?)\n```", response, re.DOTALL) + code_blocks = re.findall(r"```(?:python)?\n(.*?)```", response, re.DOTALL) if not code_blocks: raise ValueError("No Python code blocks found in response")