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,5 +1,6 @@
from __future__ import annotations
import logging
from typing import Any
from pydantic_ai import RunContext
@@ -8,6 +9,8 @@ from db.customer_risk_db import risk_db
from services.service_tuhui_upload import upload_to_tuhui
from core.order_helpers import parse_order_info
logger = logging.getLogger("cs_agent")
def register_tools(agent) -> None:
"""注册所有 Tool让 Agent 可以主动调用。"""
@@ -65,9 +68,15 @@ def register_tools(agent) -> None:
subject=result.get("subject", ""),
quality=result.get("quality", ""),
)
print(f"[Agent] Workflow 任务已创建 | 客户: {ctx.deps.from_id} | 比例: {result.get('aspect_ratio')} | 透视: {result.get('perspective')} | 图片: {image_url[:60]}...")
logger.info(
"[Agent] Workflow 任务已创建 | 客户: %s | 比例: %s | 透视: %s | 图片: %s...",
ctx.deps.from_id,
result.get("aspect_ratio"),
result.get("perspective"),
image_url[:60],
)
except Exception as e:
print(f"[Agent] Workflow 任务创建失败: {e}")
logger.exception("[Agent] Workflow 任务创建失败: %s", e)
# 组装给 AI 的分析报告
risk = result.get("risk", "none")