github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2026-07-16 17:15:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore:fmt

Browse Source
This commit is contained in:
aries_ckt
2026-05-11 10:43:00 +08:00
parent 42d94d846f
commit 303aa20928
4 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_budget.py
View File

@@ -1,7 +1,5 @@
"""Tests for context budget tracking."""
import pytest
from dbgpt.agent.core.context.budget import (
ContextBudgetConfig,
ContextBudgetTracker,

10
packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_compact.py
View File

@@ -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 = [

2
packages/dbgpt-core/src/dbgpt/agent/core/context/tests/test_manager.py
View File

@@ -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

7
packages/dbgpt-serve/src/dbgpt_serve/agent/agents/controller.py
View File

@@ -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