diff --git a/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_budget.py b/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_budget.py
index e2ae1d65a..cbd628071 100644
--- a/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_budget.py
+++ b/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_budget.py
@@ -1,7 +1,5 @@
 """Tests for context budget tracking."""
 
-import pytest
-
 from dbgpt.agent.core.context.budget import (
     ContextBudgetConfig,
     ContextBudgetTracker,
diff --git a/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_compact.py b/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_compact.py
index dc7869eff..fdfc0aea6 100644
--- a/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_compact.py
+++ b/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_compact.py
@@ -1,7 +1,5 @@
 """Tests for four-layer compaction strategies."""
 
-import pytest
-
 from dbgpt.agent.core.agent import AgentMessage
 from dbgpt.agent.core.context.budget import ContextBudgetConfig, ContextBudgetTracker
 from dbgpt.agent.core.context.compact import (
@@ -119,9 +117,7 @@ class TestObservationMicroCompact:
 
 class TestSessionMemoryCompact:
     def test_drops_old_rounds(self):
-        tracker = _make_tracker(
-            min_keep_recent_rounds=1, min_keep_tokens=0
-        )
+        tracker = _make_tracker(min_keep_recent_rounds=1, min_keep_tokens=0)
         layer2 = SessionMemoryCompact()
 
         msgs = [
@@ -163,9 +159,7 @@ class TestSessionMemoryCompact:
 
     def test_triplet_integrity(self):
         """Ensure complete triplets are never split."""
-        tracker = _make_tracker(
-            min_keep_recent_rounds=1, min_keep_tokens=0
-        )
+        tracker = _make_tracker(min_keep_recent_rounds=1, min_keep_tokens=0)
         layer2 = SessionMemoryCompact()
 
         msgs = [
diff --git a/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_manager.py b/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_manager.py
index 96f530f4c..4640074bc 100644
--- a/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_manager.py
+++ b/packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_manager.py
@@ -3,7 +3,7 @@
 import pytest
 
 from dbgpt.agent.core.agent import AgentMessage
-from dbgpt.agent.core.context.budget import ContextBudgetConfig, TokenState
+from dbgpt.agent.core.context.budget import ContextBudgetConfig
 from dbgpt.agent.core.context.manager import ContextManager
 from dbgpt.core import ModelMessageRoleType
 
diff --git a/packages/dbgpt-serve/src/dbgpt_serve/agent/agents/controller.py b/packages/dbgpt-serve/src/dbgpt_serve/agent/agents/controller.py
index d4ad01acb..5ca5542e0 100644
--- a/packages/dbgpt-serve/src/dbgpt_serve/agent/agents/controller.py
+++ b/packages/dbgpt-serve/src/dbgpt_serve/agent/agents/controller.py
@@ -344,7 +344,12 @@ class MultiAgents(BaseComponent, ABC):
                                 "__CONTEXT_STATUS__"
                             ):
                                 payload = chunk[len("__CONTEXT_STATUS__") :]
-                                resp = f"data:{json.dumps({'context_status': json.loads(payload)}, ensure_ascii=False)}\n\n"
+                                context_status = {"context_status": json.loads(payload)}
+                                resp = (
+                                    "data:"
+                                    f"{json.dumps(context_status, ensure_ascii=False)}"
+                                    "\n\n"
+                                )
                                 yield task, resp, agent_conv_id
                                 continue