65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
from decimal import Decimal
|
||
|
|
import uuid
|
||
|
|
from enum import Enum
|
||
|
|
|
||
|
|
from sqlalchemy import (
|
||
|
|
DateTime,
|
||
|
|
Enum as SqlEnum,
|
||
|
|
ForeignKey,
|
||
|
|
Integer,
|
||
|
|
Numeric,
|
||
|
|
String,
|
||
|
|
func,
|
||
|
|
text,
|
||
|
|
)
|
||
|
|
from sqlalchemy.dialects.postgresql import UUID
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
|
|
||
|
|
from core.db.base import Base, SoftDeleteMixin, TimestampMixin
|
||
|
|
|
||
|
|
|
||
|
|
class AgentChatSessionStatus(str, Enum):
|
||
|
|
PENDING = "pending"
|
||
|
|
RUNNING = "running"
|
||
|
|
COMPLETED = "completed"
|
||
|
|
FAILED = "failed"
|
||
|
|
|
||
|
|
|
||
|
|
class AgentChatSession(TimestampMixin, SoftDeleteMixin, Base):
|
||
|
|
__tablename__: str = "sessions"
|
||
|
|
|
||
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
|
||
|
|
)
|
||
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
UUID(as_uuid=True),
|
||
|
|
ForeignKey("auth.users.id", ondelete="CASCADE"),
|
||
|
|
nullable=False,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
title: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
||
|
|
status: Mapped[AgentChatSessionStatus] = mapped_column(
|
||
|
|
SqlEnum(
|
||
|
|
AgentChatSessionStatus, name="agent_chat_session_status", native_enum=False
|
||
|
|
),
|
||
|
|
nullable=False,
|
||
|
|
default=AgentChatSessionStatus.PENDING,
|
||
|
|
)
|
||
|
|
last_activity_at: Mapped[datetime] = mapped_column(
|
||
|
|
DateTime(timezone=True),
|
||
|
|
server_default=func.now(),
|
||
|
|
nullable=False,
|
||
|
|
)
|
||
|
|
message_count: Mapped[int] = mapped_column(
|
||
|
|
Integer, nullable=False, server_default=text("0")
|
||
|
|
)
|
||
|
|
total_tokens: Mapped[int] = mapped_column(
|
||
|
|
Integer, nullable=False, server_default=text("0")
|
||
|
|
)
|
||
|
|
total_cost: Mapped[Decimal] = mapped_column(
|
||
|
|
Numeric(12, 6), nullable=False, server_default=text("0")
|
||
|
|
)
|