OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-06 03:09:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

add sqlite db

Browse Source
This commit is contained in:
LeonOstrez
2023-09-06 09:16:35 +02:00
parent 38d56277ff
commit d343f2edea
14 changed files with 147 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pilot/database/models/development_steps.py
View File

@@ -1,21 +1,26 @@
from peewee import *
from playhouse.postgres_ext import BinaryJSONField
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 DevelopmentSteps(BaseModel):
id = AutoField() # This will serve as the primary key
app = ForeignKeyField(App, on_delete='CASCADE')
hash_id = CharField(null=False)
messages = BinaryJSONField(null=True)
llm_response = BinaryJSONField(null=False)
if DATABASE_TYPE == 'postgres':
messages = BinaryJSONField(null=True)
llm_response = BinaryJSONField(null=False)
else:
messages = JSONField(null=True) # Custom JSON field for SQLite
llm_response = JSONField(null=False) # Custom JSON field for SQLite
previous_step = ForeignKeyField('self', null=True, column_name='previous_step')
class Meta:
db_table = 'development_steps'
indexes = (
(('app', 'hash_id'), True),
)
)