feat(agent): add debug output switches

This commit is contained in:
2026-07-17 14:19:23 +08:00
parent a879dc2ff1
commit c7bafda135
5 changed files with 60 additions and 4 deletions

View File

@@ -13,12 +13,13 @@ from app.models import Base
from app.models.admin import Admin
from app.models.ai_config import ModelConfig, SystemConfig
from app.schemas.admin import AgentDebugRequest, AgentRuntimeConfigSaveRequest
from app.services.agent_debug_service import AgentDebugService
from app.services.model_stream_service import (
AsyncStreamingModelResponse,
_openai_stream_payload,
_stream_configured_model_async,
)
from app.services.model_service import ModelClientService, _max_output_tokens
from app.services.model_service import ModelClientService, _copy_model_with_overrides, _max_output_tokens
from app.services.rag_service import RagResult
@@ -146,6 +147,23 @@ def test_disabled_stream_returns_one_complete_chunk():
assert chunks == [answer]
def test_debug_stream_setting_overrides_model_without_changing_it():
model = _model()
debug_model = _copy_model_with_overrides(model, {"stream_enabled": 0})
assert model.stream_enabled == 1
assert debug_model.stream_enabled == 0
payload = AgentDebugRequest(
promptContent="测试",
modelId=1,
question="测试",
streamEnabled=0,
reasoningVisible=1,
)
assert AgentDebugService.overrides(payload)["stream_enabled"] == 0
@pytest.mark.parametrize("reasoning_visible", [True, False])
def test_agent_debug_stream_respects_reasoning_visibility(reasoning_visible):
async def chunks():
@@ -179,7 +197,7 @@ def test_agent_debug_stream_respects_reasoning_visibility(reasoning_visible):
db.add(SystemConfig(
id=1,
config_key="show_model_reasoning",
config_value="true" if reasoning_visible else "false",
config_value="false" if reasoning_visible else "true",
))
db.commit()
payload = AgentDebugRequest(
@@ -188,6 +206,7 @@ def test_agent_debug_stream_respects_reasoning_visibility(reasoning_visible):
knowledgeIds=[],
question="怎么冷静",
maxToken=8192,
reasoningVisible=1 if reasoning_visible else 0,
)
with (
patch(