class ChatState(db.Model):
__tablename__ = 'chat_state'
id = db.Column(db.Integer, primary_key=True)
client_id = db.Column(db.String)
prompt_key = db.Column(db.String)
created_at = db.Column(db.DateTime, default=datetime.datetime.now)
def __init__(self, client_id, prompt_key):
self.client_id = client_id
self.prompt_key = prompt_key
class TherapyConfig(db.Model):
__tablename__ = 'therapy_configs'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String)
config = db.Column(db.JSON)
updated_at = db.Column(db.DateTime, default=datetime.datetime.now)
created_at = db.Column(db.DateTime, default=datetime.datetime.now)
def __init__(self, name, config):
self.name = name
self.config = config
def serialize(self):
return {
'id': self.id,
'name': self.name,
'config': self.config,
}
class ClientData(db.Model):
__tablename__ = 'client_data'
id = db.Column(db.Integer, primary_key=True)
client_id = db.Column(db.String)
config_id = db.Column(db.String)
session_id = db.Column(db.Integer)
prompt_key = db.Column(db.String)
prompt_text = db.Column(db.String, nullable=True)
response = db.Column(db.String)
created_at = db.Column(db.DateTime, default=datetime.datetime.now)
def __init__(self, client_id, config_id, session_id, prompt_key, response):
self.client_id = client_id
self.config_id = config_id
self.session_id = session_id
self.prompt_key = prompt_key
self.response = response