55 lines
3.0 KiB
Python
55 lines
3.0 KiB
Python
"""AI 调用日志"""
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import BigInteger, DECIMAL, ForeignKey, Integer, JSON, Text, TIMESTAMP, VARCHAR, func
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from .base import BaseNoTenant
|
|
|
|
|
|
class BaseLog(BaseNoTenant):
|
|
"""日志表基类"""
|
|
__abstract__ = True
|
|
tenant_id: Mapped[int] = mapped_column(
|
|
BigInteger, nullable=False, default=1, index=True
|
|
)
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
TIMESTAMP, nullable=False, server_default=func.now()
|
|
)
|
|
|
|
|
|
class AICallLog(BaseLog):
|
|
__tablename__ = "ai_call_logs"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
user_id: Mapped[Optional[int]] = mapped_column(
|
|
BigInteger, ForeignKey("users.id"), nullable=True
|
|
)
|
|
function_name: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
|
|
prompt: Mapped[str] = mapped_column(Text, nullable=False)
|
|
response: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
tokens_used: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
|
|
cost: Mapped[Optional[float]] = mapped_column(DECIMAL(10, 6), nullable=True)
|
|
status: Mapped[str] = mapped_column(VARCHAR(50), nullable=False)
|
|
trace_id: Mapped[Optional[str]] = mapped_column(VARCHAR(80), nullable=True, index=True)
|
|
agent_run_id: Mapped[Optional[str]] = mapped_column(VARCHAR(80), nullable=True, index=True)
|
|
agent_step_key: Mapped[Optional[str]] = mapped_column(VARCHAR(120), nullable=True, index=True)
|
|
provider: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True, index=True)
|
|
model: Mapped[Optional[str]] = mapped_column(VARCHAR(120), nullable=True, index=True)
|
|
latency_ms: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
|
|
routing_meta: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True)
|
|
prompt_key: Mapped[Optional[str]] = mapped_column(VARCHAR(128), nullable=True, index=True)
|
|
prompt_release_id: Mapped[Optional[int]] = mapped_column(Integer, nullable=True, index=True)
|
|
prompt_source: Mapped[Optional[str]] = mapped_column(VARCHAR(32), nullable=True, index=True)
|
|
provider_request_payload: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True)
|
|
provider_response_payload: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True)
|
|
provider_attempts: Mapped[Optional[list]] = mapped_column(JSON, nullable=True)
|
|
planned_tier: Mapped[Optional[str]] = mapped_column(VARCHAR(16), nullable=True, index=True)
|
|
executed_tier: Mapped[Optional[str]] = mapped_column(VARCHAR(16), nullable=True, index=True)
|
|
routing_decision: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True, index=True)
|
|
route_source: Mapped[Optional[str]] = mapped_column(VARCHAR(32), nullable=True)
|
|
route_rule_name: Mapped[Optional[str]] = mapped_column(VARCHAR(255), nullable=True)
|
|
routing_config_version: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True)
|
|
routing_config_updated_at: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True)
|