Changeset 0.20.0 (#85)

server/store/sqlite/channel.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"database/sql"
 	"encoding/json"
+	"strings"
 	"time"
 
 	"git.gobha.me/xcaliber/chat-switchboard/models"
@@ -218,6 +219,48 @@ func (s *ChannelStore) GetModels(ctx context.Context, channelID string) ([]model
 	return result, rows.Err()
 }
 
+func (s *ChannelStore) GetModelByID(ctx context.Context, id string) (*models.ChannelModel, error) {
+	var cm models.ChannelModel
+	err := DB.QueryRowContext(ctx, `
+		SELECT id, channel_id, model_id, COALESCE(provider_config_id, ''),
+			COALESCE(display_name, ''), COALESCE(system_prompt, ''), is_default
+		FROM channel_models WHERE id = ?`, id).Scan(
+		&cm.ID, &cm.ChannelID, &cm.ModelID, &cm.ProviderConfigID,
+		&cm.DisplayName, &cm.SystemPrompt, &cm.IsDefault)
+	if err != nil {
+		return nil, err
+	}
+	return &cm, nil
+}
+
+func (s *ChannelStore) UpdateModel(ctx context.Context, id string, fields map[string]interface{}) error {
+	if len(fields) == 0 {
+		return nil
+	}
+	sets := make([]string, 0, len(fields))
+	args := make([]interface{}, 0, len(fields)+1)
+	for col, val := range fields {
+		sets = append(sets, col+" = ?")
+		args = append(args, val)
+	}
+	args = append(args, id)
+	query := "UPDATE channel_models SET " + strings.Join(sets, ", ") + " WHERE id = ?"
+	_, err := DB.ExecContext(ctx, query, args...)
+	return err
+}
+
+func (s *ChannelStore) DeleteModel(ctx context.Context, id string) error {
+	res, err := DB.ExecContext(ctx, `DELETE FROM channel_models WHERE id = ?`, id)
+	if err != nil {
+		return err
+	}
+	n, _ := res.RowsAffected()
+	if n == 0 {
+		return sql.ErrNoRows
+	}
+	return nil
+}
+
 func (s *ChannelStore) UserOwns(ctx context.Context, channelID, userID string) (bool, error) {
 	var exists bool
 	err := DB.QueryRowContext(ctx,