mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-29 15:45:44 +01:00
16 lines
489 B
Python
16 lines
489 B
Python
from peewee import *
|
|
from database.config import DATABASE_TYPE
|
|
from database.models.components.progress_step import ProgressStep
|
|
from database.models.components.sqlite_middlewares import JSONField
|
|
from playhouse.postgres_ext import BinaryJSONField
|
|
|
|
|
|
class Architecture(ProgressStep):
|
|
if DATABASE_TYPE == 'postgres':
|
|
architecture = BinaryJSONField()
|
|
else:
|
|
architecture = JSONField() # Custom JSON field for SQLite
|
|
|
|
class Meta:
|
|
db_table = 'architecture'
|