rp setting; some fixes
This commit is contained in:
@@ -27,6 +27,12 @@ type RPModel struct {
|
||||
Name string
|
||||
ContextSize int `db:"context_size"`
|
||||
}
|
||||
type RPSetting struct {
|
||||
ID int
|
||||
Name string
|
||||
Description string
|
||||
Prompt string
|
||||
}
|
||||
type RPUser struct {
|
||||
UserID int64 `db:"user_id"`
|
||||
UserPrompt string `db:"user_prompt"`
|
||||
@@ -133,3 +139,14 @@ func (rep *RPRepository) GetAllModels() ([]*RPModel, error) {
|
||||
err := rep.db.Select(&models, "SELECT * FROM rp_models ORDER BY id;")
|
||||
return models, err
|
||||
}
|
||||
|
||||
func (rep *RPRepository) GetAllSettings() ([]*RPSetting, error) {
|
||||
settings := make([]*RPSetting, 0)
|
||||
err := rep.db.Select(&settings, "SELECT * FROM rp_settings ORDER BY id;")
|
||||
return settings, err
|
||||
}
|
||||
func (rep *RPRepository) GetSetting(id int) (*RPSetting, error) {
|
||||
setting := new(RPSetting)
|
||||
err := rep.db.Get(setting, "SELECT * FROM rp_settings WHERE id=$1;", id)
|
||||
return setting, err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user