package psql

import (
	"git.nix13.pw/scuroneko/laniakea"
	"github.com/vinovest/sqlx"
)

type AIModel struct {
	ID          string
	Key         string
	Name        string
	ContextSize int `db:"context_size"`
}

type AIRepository struct {
	db *sqlx.DB
}

func newAiRepository(db *sqlx.DB) *AIRepository {
	return &AIRepository{db}
}
func NewAIRepository(db *laniakea.DatabaseContext) *AIRepository {
	return newAiRepository(db.PostgresSQL)
}

func (rep *AIRepository) GetModel(id string) (*AIModel, error) {
	model := new(AIModel)
	err := rep.db.Get(model, "SELECT * FROM ai_models WHERE id=$1;", id)
	return model, err
}
func (rep *AIRepository) GetAllModels() ([]*AIModel, error) {
	models := make([]*AIModel, 0)
	err := rep.db.Select(&models, "SELECT * FROM ai_models ORDER BY id;")
	return models, err
}