Spaces:
Sleeping
Sleeping
File size: 4,510 Bytes
ed5cf78 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 | import uuid
from datetime import datetime
from typing import Optional, List
from enum import Enum
from sqlmodel import Field, SQLModel, Relationship
class TaskStatus(str, Enum):
TODO = "TODO"
IN_PROGRESS = "IN_PROGRESS"
DONE = "DONE"
class TaskPriority(str, Enum):
LOW = "LOW"
MEDIUM = "MEDIUM"
HIGH = "HIGH"
# Better Auth Compatible Models
class UserBase(SQLModel):
email: str = Field(unique=True, index=True)
name: Optional[str] = None
image: Optional[str] = None
class User(UserBase, table=True):
id: str = Field(primary_key=True)
emailVerified: bool = Field(default=False)
createdAt: datetime = Field(default_factory=datetime.utcnow)
updatedAt: datetime = Field(default_factory=datetime.utcnow)
tasks: List["Task"] = Relationship(back_populates="owner")
sessions: List["Session"] = Relationship(back_populates="user")
conversations: List["Conversation"] = Relationship(back_populates="user")
class UserCreate(UserBase):
id: Optional[str] = Field(default_factory=lambda: str(uuid.uuid4()))
password: str
class UserRead(UserBase):
id: str
createdAt: datetime
updatedAt: datetime
class Session(SQLModel, table=True):
id: str = Field(primary_key=True)
expiresAt: datetime
token: str
createdAt: datetime = Field(default_factory=datetime.utcnow)
updatedAt: datetime = Field(default_factory=datetime.utcnow)
ipAddress: Optional[str] = None
userAgent: Optional[str] = None
userId: str = Field(foreign_key="user.id")
user: User = Relationship(back_populates="sessions")
class Account(SQLModel, table=True):
id: str = Field(primary_key=True)
accountId: str
providerId: str
userId: str = Field(foreign_key="user.id")
accessToken: Optional[str] = None
refreshToken: Optional[str] = None
idToken: Optional[str] = None
accessTokenExpiresAt: Optional[datetime] = None
refreshTokenExpiresAt: Optional[datetime] = None
scope: Optional[str] = None
password: Optional[str] = None
createdAt: datetime = Field(default_factory=datetime.utcnow)
updatedAt: datetime = Field(default_factory=datetime.utcnow)
class Verification(SQLModel, table=True):
id: str = Field(primary_key=True)
identifier: str
value: str
expiresAt: datetime
createdAt: datetime = Field(default_factory=datetime.utcnow)
updatedAt: datetime = Field(default_factory=datetime.utcnow)
class TaskBase(SQLModel):
title: str = Field(index=True)
description: Optional[str] = None
status: TaskStatus = Field(default=TaskStatus.TODO)
priority: Optional[TaskPriority] = Field(default=TaskPriority.MEDIUM)
class Task(TaskBase, table=True):
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
created_at: datetime = Field(default_factory=datetime.utcnow)
updated_at: datetime = Field(default_factory=datetime.utcnow)
owner_id: str = Field(foreign_key="user.id")
owner: User = Relationship(back_populates="tasks")
class TaskCreate(TaskBase):
pass
class TaskUpdate(SQLModel):
title: Optional[str] = None
description: Optional[str] = None
status: Optional[TaskStatus] = None
priority: Optional[TaskPriority] = None
class TaskRead(TaskBase):
id: uuid.UUID
created_at: datetime
updated_at: datetime
owner_id: str
class MessageRole(str, Enum):
USER = "user"
ASSISTANT = "assistant"
SYSTEM = "system"
TOOL = "tool"
class Conversation(SQLModel, table=True):
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
user_id: str = Field(foreign_key="user.id")
title: Optional[str] = None
created_at: datetime = Field(default_factory=datetime.utcnow)
updated_at: datetime = Field(default_factory=datetime.utcnow)
user: User = Relationship(back_populates="conversations")
messages: List["Message"] = Relationship(back_populates="conversation")
class Message(SQLModel, table=True):
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
conversation_id: uuid.UUID = Field(foreign_key="conversation.id")
role: MessageRole
content: Optional[str] = None
tool_calls: Optional[str] = None # Store as JSON string
tool_call_id: Optional[str] = None # For 'tool' role messages
name: Optional[str] = None # Name of the tool called
created_at: datetime = Field(default_factory=datetime.utcnow)
conversation: Conversation = Relationship(back_populates="messages")
|