feat:更新智能体工具

config/domain.py

@@ -0,0 +1,72 @@
+from dataclasses import dataclass
+from typing import Callable, Dict, List, Optional
+
+
+@dataclass
+class MetricConfig:
+    """
+    指标字段配置（完全去业务语义）
+    """
+    field: str  # 原始字段名，如 amount
+    name: str  # 展示名称，如 实收金额 / 时长 / 评分
+    agg: str = "sum"  # 聚合方式：sum / mean / max / min / count
+    format: str = "auto"  # 展示格式：auto / money / number:2 / int / duration
+
+
+class DomainConfig:
+    def __init__(
+            self,
+            name: str,
+            api_url: str,
+            entity_name: str,
+            field_name_map: Dict[str, str],
+            detail_fields: List[str] | None = None,
+            search_fields: List[str] | None = None,
+            metrics: Optional[List[MetricConfig]] = None,
+            token_getter: Callable[[], str] | None = None,
+            auth_header: str = "Authorization",
+            date_field: str = "date",
+            primary_key: str = "id",
+    ):
+        self.name = name
+        self.api_url = api_url
+        self.entity_name = entity_name
+        self.field_name_map = field_name_map
+        self.detail_fields = detail_fields or []
+        self.search_fields = search_fields or []
+        self.metrics: List[MetricConfig] = metrics or []
+
+        self.token_getter = token_getter
+        self.auth_header = auth_header
+        self.date_field = date_field
+        self.primary_key = primary_key
+
+        self._metric_map: Dict[str, MetricConfig] = {
+            m.field: m for m in self.metrics
+        }
+
+    def get_metric(self, field: str) -> Optional[MetricConfig]:
+        return self._metric_map.get(field)
+
+    def metric_fields(self) -> List[str]:
+        return [m.field for m in self.metrics]
+
+    def has_metric(self, field: str) -> bool:
+        return field in self._metric_map
+
+    def get_chinese_name(self, field: str) -> str:
+        return self.field_name_map.get(field, field)
+
+
+_DOMAINS: Dict[str, DomainConfig] = {}
+
+
+def register_domain(config: DomainConfig):
+    print("注册业务域：", config.name)
+    _DOMAINS[config.name] = config
+
+
+def get_domain(name: str) -> DomainConfig:
+    if name not in _DOMAINS:
+        raise ValueError(f"未注册的业务域：{name}")
+    return _DOMAINS[name]