// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.31.1
// source: knowledge.sql

package dbgen

import (
	"context"

	"github.com/jackc/pgx/v5/pgtype"
)

const createKnowledgeBase = `-- name: CreateKnowledgeBase :one
INSERT INTO knowledge_bases (id, project_id, key, name, description)
VALUES ($1, $2, $3, $4, $5)
RETURNING id, project_id, key, name, description, created_at, updated_at
`

type CreateKnowledgeBaseParams struct {
	ID          pgtype.UUID `json:"id"`
	ProjectID   pgtype.UUID `json:"project_id"`
	Key         string      `json:"key"`
	Name        string      `json:"name"`
	Description string      `json:"description"`
}

func (q *Queries) CreateKnowledgeBase(ctx context.Context, arg CreateKnowledgeBaseParams) (KnowledgeBasis, error) {
	row := q.db.QueryRow(ctx, createKnowledgeBase,
		arg.ID,
		arg.ProjectID,
		arg.Key,
		arg.Name,
		arg.Description,
	)
	var i KnowledgeBasis
	err := row.Scan(
		&i.ID,
		&i.ProjectID,
		&i.Key,
		&i.Name,
		&i.Description,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}

const getActiveReleaseByAlias = `-- name: GetActiveReleaseByAlias :one
SELECT r.id, r.project_id, r.knowledge_base_id, r.alias, r.version, r.status, r.dataset_version_id, r.index_artifact_id, r.retrieval_policy, r.generation_policy, r.evaluation_run_id, r.created_by_workflow_run_id, r.promoted_at, r.created_at
FROM releases r
JOIN knowledge_bases kb ON kb.id = r.knowledge_base_id
WHERE kb.project_id = $1
  AND kb.id = $2
  AND r.alias = $3
  AND r.status = 'active'
`

type GetActiveReleaseByAliasParams struct {
	ProjectID       pgtype.UUID `json:"project_id"`
	KnowledgeBaseID pgtype.UUID `json:"knowledge_base_id"`
	Alias           string      `json:"alias"`
}

func (q *Queries) GetActiveReleaseByAlias(ctx context.Context, arg GetActiveReleaseByAliasParams) (Release, error) {
	row := q.db.QueryRow(ctx, getActiveReleaseByAlias, arg.ProjectID, arg.KnowledgeBaseID, arg.Alias)
	var i Release
	err := row.Scan(
		&i.ID,
		&i.ProjectID,
		&i.KnowledgeBaseID,
		&i.Alias,
		&i.Version,
		&i.Status,
		&i.DatasetVersionID,
		&i.IndexArtifactID,
		&i.RetrievalPolicy,
		&i.GenerationPolicy,
		&i.EvaluationRunID,
		&i.CreatedByWorkflowRunID,
		&i.PromotedAt,
		&i.CreatedAt,
	)
	return i, err
}

const getKnowledgeBase = `-- name: GetKnowledgeBase :one
SELECT id, project_id, key, name, description, created_at, updated_at FROM knowledge_bases
WHERE id = $1 AND project_id = $2
`

type GetKnowledgeBaseParams struct {
	ID        pgtype.UUID `json:"id"`
	ProjectID pgtype.UUID `json:"project_id"`
}

func (q *Queries) GetKnowledgeBase(ctx context.Context, arg GetKnowledgeBaseParams) (KnowledgeBasis, error) {
	row := q.db.QueryRow(ctx, getKnowledgeBase, arg.ID, arg.ProjectID)
	var i KnowledgeBasis
	err := row.Scan(
		&i.ID,
		&i.ProjectID,
		&i.Key,
		&i.Name,
		&i.Description,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}

const getKnowledgeBaseByKey = `-- name: GetKnowledgeBaseByKey :one
SELECT id, project_id, key, name, description, created_at, updated_at FROM knowledge_bases
WHERE project_id = $1 AND key = $2
`

type GetKnowledgeBaseByKeyParams struct {
	ProjectID pgtype.UUID `json:"project_id"`
	Key       string      `json:"key"`
}

func (q *Queries) GetKnowledgeBaseByKey(ctx context.Context, arg GetKnowledgeBaseByKeyParams) (KnowledgeBasis, error) {
	row := q.db.QueryRow(ctx, getKnowledgeBaseByKey, arg.ProjectID, arg.Key)
	var i KnowledgeBasis
	err := row.Scan(
		&i.ID,
		&i.ProjectID,
		&i.Key,
		&i.Name,
		&i.Description,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}