135 lines
3.5 KiB
Go
135 lines
3.5 KiB
Go
// 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
|
|
}
|