From 1ef08c463678e44872ac6ec091bcd599808d8a4a Mon Sep 17 00:00:00 2001
From: mohammed <mohammed18200118@gmail.com>
Date: Fri, 27 Jun 2025 02:20:51 +0300
Subject: [PATCH] avoid serializing optimized code

---
 codeflash/lsp/beta.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/codeflash/lsp/beta.py b/codeflash/lsp/beta.py
index 1e1397bb8..aae2b7bee 100644
--- a/codeflash/lsp/beta.py
+++ b/codeflash/lsp/beta.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import contextlib
-import json
 import os
 from dataclasses import dataclass
 from pathlib import Path
@@ -228,7 +227,7 @@ def perform_function_optimization(
         "status": "success",
         "message": "Optimization completed successfully",
         "extra": f"Speedup: {original_code_baseline.runtime / best_optimization.runtime:.2f}x faster",
-        "optimization": json.dumps(optimized_source, indent=None),
+        "optimization": optimized_source,
     }