diff options
Diffstat (limited to 'internal/db/sessions.sql.go')
| -rw-r--r-- | internal/db/sessions.sql.go | 54 |
1 files changed, 32 insertions, 22 deletions
diff --git a/internal/db/sessions.sql.go b/internal/db/sessions.sql.go index 3e0b6eea2..f3ee4ff42 100644 --- a/internal/db/sessions.sql.go +++ b/internal/db/sessions.sql.go @@ -14,7 +14,8 @@ INSERT INTO sessions ( id, title, message_count, - tokens, + prompt_tokens, + completion_tokens, cost, updated_at, created_at @@ -24,26 +25,28 @@ INSERT INTO sessions ( ?, ?, ?, + ?, strftime('%s', 'now'), strftime('%s', 'now') -) RETURNING id, title, message_count, tokens, cost, updated_at, created_at +) RETURNING id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at ` type CreateSessionParams struct { - ID string `json:"id"` - Title string `json:"title"` - MessageCount int64 `json:"message_count"` - Tokens int64 `json:"tokens"` - Cost float64 `json:"cost"` + ID string `json:"id"` + Title string `json:"title"` + MessageCount int64 `json:"message_count"` + PromptTokens int64 `json:"prompt_tokens"` + CompletionTokens int64 `json:"completion_tokens"` + Cost float64 `json:"cost"` } -// sqlfluff:dialect:sqlite func (q *Queries) CreateSession(ctx context.Context, arg CreateSessionParams) (Session, error) { row := q.queryRow(ctx, q.createSessionStmt, createSession, arg.ID, arg.Title, arg.MessageCount, - arg.Tokens, + arg.PromptTokens, + arg.CompletionTokens, arg.Cost, ) var i Session @@ -51,7 +54,8 @@ func (q *Queries) CreateSession(ctx context.Context, arg CreateSessionParams) (S &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, @@ -70,7 +74,7 @@ func (q *Queries) DeleteSession(ctx context.Context, id string) error { } const getSessionByID = `-- name: GetSessionByID :one -SELECT id, title, message_count, tokens, cost, updated_at, created_at +SELECT id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at FROM sessions WHERE id = ? LIMIT 1 ` @@ -82,7 +86,8 @@ func (q *Queries) GetSessionByID(ctx context.Context, id string) (Session, error &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, @@ -91,7 +96,7 @@ func (q *Queries) GetSessionByID(ctx context.Context, id string) (Session, error } const listSessions = `-- name: ListSessions :many -SELECT id, title, message_count, tokens, cost, updated_at, created_at +SELECT id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at FROM sessions ORDER BY created_at DESC ` @@ -109,7 +114,8 @@ func (q *Queries) ListSessions(ctx context.Context) ([]Session, error) { &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, @@ -131,23 +137,26 @@ const updateSession = `-- name: UpdateSession :one UPDATE sessions SET title = ?, - tokens = ?, + prompt_tokens = ?, + completion_tokens = ?, cost = ? WHERE id = ? -RETURNING id, title, message_count, tokens, cost, updated_at, created_at +RETURNING id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at ` type UpdateSessionParams struct { - Title string `json:"title"` - Tokens int64 `json:"tokens"` - Cost float64 `json:"cost"` - ID string `json:"id"` + Title string `json:"title"` + PromptTokens int64 `json:"prompt_tokens"` + CompletionTokens int64 `json:"completion_tokens"` + Cost float64 `json:"cost"` + ID string `json:"id"` } func (q *Queries) UpdateSession(ctx context.Context, arg UpdateSessionParams) (Session, error) { row := q.queryRow(ctx, q.updateSessionStmt, updateSession, arg.Title, - arg.Tokens, + arg.PromptTokens, + arg.CompletionTokens, arg.Cost, arg.ID, ) @@ -156,7 +165,8 @@ func (q *Queries) UpdateSession(ctx context.Context, arg UpdateSessionParams) (S &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, |