25 lines
1.2 KiB
Python

from __future__ import annotations
from sqlalchemy import Integer, Text
from sqlalchemy.orm import Mapped, mapped_column
from app.db.base import Base
class ToolCall(Base):
__tablename__ = "tool_call"
tool_call_id: Mapped[str] = mapped_column(Text, primary_key=True)
task_id: Mapped[str] = mapped_column(Text, nullable=False, index=True)
request_id: Mapped[str | None] = mapped_column(Text, nullable=True, index=True)
operator_user_id: Mapped[str | None] = mapped_column(Text, nullable=True)
operator_user_name: Mapped[str | None] = mapped_column(Text, nullable=True)
step_id: Mapped[str | None] = mapped_column(Text, nullable=True)
tool_name: Mapped[str] = mapped_column(Text, nullable=False, index=True)
request_payload_json: Mapped[str] = mapped_column(Text, nullable=False)
response_payload_json: Mapped[str] = mapped_column(Text, nullable=False)
success: Mapped[int] = mapped_column(Integer, nullable=False)
duration_ms: Mapped[int | None] = mapped_column(Integer, nullable=True)
started_at: Mapped[str | None] = mapped_column(Text, nullable=True)
finished_at: Mapped[str | None] = mapped_column(Text, nullable=True)