refactor: unify core pipeline logging with cs_agent logger

This commit is contained in:
2026-03-01 16:29:52 +08:00
parent a6b7bf1982
commit 8dd5a11b4b
11 changed files with 82 additions and 55 deletions

View File

@@ -1,8 +1,11 @@
from __future__ import annotations
import logging
from typing import TYPE_CHECKING, Any
from core.post_ops import negotiation_strategy_reply
logger = logging.getLogger("cs_agent")
if TYPE_CHECKING:
from core.pydantic_ai_agent import AgentDeps, ConversationState, CustomerMessage, CustomerServiceAgent
@@ -124,13 +127,14 @@ async def execute_ai_turn(
for part in getattr(msg, "parts", []):
part_type = type(part).__name__
if "ToolCall" in part_type:
print(
f"{agent.C_TOOL}[THINK/TOOL_CALL]{agent.C_RESET} "
f"{getattr(part, 'tool_name', '')}({getattr(part, 'args', '')})"
logger.info(
"[THINK/TOOL_CALL] %s(%s)",
getattr(part, "tool_name", ""),
getattr(part, "args", ""),
)
elif "ToolReturn" in part_type:
ret = str(getattr(part, "content", ""))[:120]
print(f"{agent.C_TOOL}[THINK/TOOL_RETURN]{agent.C_RESET} {ret}")
logger.info("[THINK/TOOL_RETURN] %s", ret)
print(f"{agent.C_THINK}[THINK/RAW_OUTPUT]{agent.C_RESET} {repr(reply_text)}")
logger.info("[THINK/RAW_OUTPUT] %r", reply_text)
return reply_text