mgbam commited on
Commit
bb4d538
·
verified ·
1 Parent(s): 08215dd

Create models/chat.py

Browse files
Files changed (1) hide show
  1. models/chat.py +9 -0
models/chat.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from datetime import datetime
2
+ from sqlmodel import SQLModel, Field
3
+
4
+ class ChatMessage(SQLModel, table=True):
5
+ id: int = Field(default=None, primary_key=True)
6
+ user: str
7
+ prompt: str
8
+ response: str
9
+ timestamp: datetime = Field(default_factory=datetime.utcnow)