compress wip

database/psql/rp.go

@@ -34,13 +34,17 @@ type RPSetting struct {
 	Prompt      string
 }
 type RPUser struct {
-	UserID         int64  `db:"user_id"`
-	UserPrompt     string `db:"user_prompt"`
+	UserID     int64  `db:"user_id"`
+	UserPrompt string `db:"user_prompt"`
+	UsedTokens int64  `db:"used_tokens"`
+
 	SelectedPreset string `db:"selected_preset"`
 	Preset         *RPPreset
 	SelectedModel  string `db:"selected_model"`
 	Model          *RPModel
-	UsedTokens     int64 `db:"used_tokens"`
+
+	CompressMethod string `db:"compress_method"`
+	CompressLimit  int    `db:"compress_limit"`
 }
 
 type RPRepository struct {
@@ -150,3 +154,16 @@ func (rep *RPRepository) GetSetting(id int) (*RPSetting, error) {
 	err := rep.db.Get(setting, "SELECT * FROM rp_settings WHERE id=$1;", id)
 	return setting, err
 }
+
+func (rep *RPRepository) UpdateUserCompressSettings(user *RPUser) (*RPUser, error) {
+	query, args, err := sqlx.In(
+		"UPDATE rp_users SET compress_method=?, compress_limit=? WHERE user_id=?;",
+		user.CompressMethod, user.CompressLimit, user.UserID,
+	)
+	if err != nil {
+		return user, err
+	}
+	query = rep.db.Rebind(query)
+	_, err = rep.db.Exec(query, args...)
+	return user, err
+}