mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-27 22:57:50 +01:00
24 lines
773 B
Python
24 lines
773 B
Python
from peewee import *
|
|
from database.config import DATABASE_TYPE
|
|
from database.models.components.base_models import BaseModel
|
|
from database.models.app import App
|
|
from database.models.components.sqlite_middlewares import JSONField
|
|
from playhouse.postgres_ext import BinaryJSONField
|
|
|
|
|
|
class ProgressStep(BaseModel):
|
|
app = ForeignKeyField(App, primary_key=True, on_delete='CASCADE')
|
|
step = CharField()
|
|
|
|
if DATABASE_TYPE == 'postgres':
|
|
app_data = BinaryJSONField()
|
|
data = BinaryJSONField(null=True)
|
|
messages = BinaryJSONField(null=True)
|
|
else:
|
|
app_data = JSONField()
|
|
data = JSONField(null=True)
|
|
messages = JSONField(null=True)
|
|
|
|
completed = BooleanField(default=False)
|
|
completed_at = DateTimeField(null=True)
|