| |
| |
| |
| |
|
|
| package db |
|
|
| import ( |
| "context" |
| "database/sql" |
| ) |
|
|
| const createMessage = `-- name: CreateMessage :one |
| INSERT INTO messages ( |
| id, |
| session_id, |
| role, |
| parts, |
| model, |
| provider, |
| is_summary_message, |
| created_at, |
| updated_at |
| ) VALUES ( |
| ?, ?, ?, ?, ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now') |
| ) |
| RETURNING id, session_id, role, parts, model, created_at, updated_at, finished_at, provider, is_summary_message, prism_model_id, prism_model_name, prism_hypercredit_savings, prism_dollar_savings |
| ` |
|
|
| type CreateMessageParams struct { |
| ID string `json:"id"` |
| SessionID string `json:"session_id"` |
| Role string `json:"role"` |
| Parts string `json:"parts"` |
| Model sql.NullString `json:"model"` |
| Provider sql.NullString `json:"provider"` |
| IsSummaryMessage int64 `json:"is_summary_message"` |
| } |
|
|
| func (q *Queries) CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error) { |
| row := q.queryRow(ctx, q.createMessageStmt, createMessage, |
| arg.ID, |
| arg.SessionID, |
| arg.Role, |
| arg.Parts, |
| arg.Model, |
| arg.Provider, |
| arg.IsSummaryMessage, |
| ) |
| var i Message |
| err := row.Scan( |
| &i.ID, |
| &i.SessionID, |
| &i.Role, |
| &i.Parts, |
| &i.Model, |
| &i.CreatedAt, |
| &i.UpdatedAt, |
| &i.FinishedAt, |
| &i.Provider, |
| &i.IsSummaryMessage, |
| &i.PrismModelID, |
| &i.PrismModelName, |
| &i.PrismHypercreditSavings, |
| &i.PrismDollarSavings, |
| ) |
| return i, err |
| } |
|
|
| const deleteMessage = `-- name: DeleteMessage :exec |
| DELETE FROM messages |
| WHERE id = ? |
| ` |
|
|
| func (q *Queries) DeleteMessage(ctx context.Context, id string) error { |
| _, err := q.exec(ctx, q.deleteMessageStmt, deleteMessage, id) |
| return err |
| } |
|
|
| const deleteSessionMessages = `-- name: DeleteSessionMessages :exec |
| DELETE FROM messages |
| WHERE session_id = ? |
| ` |
|
|
| func (q *Queries) DeleteSessionMessages(ctx context.Context, sessionID string) error { |
| _, err := q.exec(ctx, q.deleteSessionMessagesStmt, deleteSessionMessages, sessionID) |
| return err |
| } |
|
|
| const getLastAssistantMessageBySession = `-- name: GetLastAssistantMessageBySession :one |
| SELECT id, session_id, role, parts, model, created_at, updated_at, finished_at, provider, is_summary_message, prism_model_id, prism_model_name, prism_hypercredit_savings, prism_dollar_savings |
| FROM messages |
| WHERE session_id = ? AND role = 'assistant' AND is_summary_message = 0 |
| ORDER BY created_at DESC |
| LIMIT 1 |
| ` |
|
|
| func (q *Queries) GetLastAssistantMessageBySession(ctx context.Context, sessionID string) (Message, error) { |
| row := q.queryRow(ctx, q.getLastAssistantMessageBySessionStmt, getLastAssistantMessageBySession, sessionID) |
| var i Message |
| err := row.Scan( |
| &i.ID, |
| &i.SessionID, |
| &i.Role, |
| &i.Parts, |
| &i.Model, |
| &i.CreatedAt, |
| &i.UpdatedAt, |
| &i.FinishedAt, |
| &i.Provider, |
| &i.IsSummaryMessage, |
| &i.PrismModelID, |
| &i.PrismModelName, |
| &i.PrismHypercreditSavings, |
| &i.PrismDollarSavings, |
| ) |
| return i, err |
| } |
|
|
| const getMessage = `-- name: GetMessage :one |
| SELECT id, session_id, role, parts, model, created_at, updated_at, finished_at, provider, is_summary_message, prism_model_id, prism_model_name, prism_hypercredit_savings, prism_dollar_savings |
| FROM messages |
| WHERE id = ? LIMIT 1 |
| ` |
|
|
| func (q *Queries) GetMessage(ctx context.Context, id string) (Message, error) { |
| row := q.queryRow(ctx, q.getMessageStmt, getMessage, id) |
| var i Message |
| err := row.Scan( |
| &i.ID, |
| &i.SessionID, |
| &i.Role, |
| &i.Parts, |
| &i.Model, |
| &i.CreatedAt, |
| &i.UpdatedAt, |
| &i.FinishedAt, |
| &i.Provider, |
| &i.IsSummaryMessage, |
| &i.PrismModelID, |
| &i.PrismModelName, |
| &i.PrismHypercreditSavings, |
| &i.PrismDollarSavings, |
| ) |
| return i, err |
| } |
|
|
| const listAllUserMessages = `-- name: ListAllUserMessages :many |
| SELECT id, session_id, role, parts, model, created_at, updated_at, finished_at, provider, is_summary_message, prism_model_id, prism_model_name, prism_hypercredit_savings, prism_dollar_savings |
| FROM messages |
| WHERE role = 'user' |
| ORDER BY created_at DESC |
| ` |
|
|
| func (q *Queries) ListAllUserMessages(ctx context.Context) ([]Message, error) { |
| rows, err := q.query(ctx, q.listAllUserMessagesStmt, listAllUserMessages) |
| if err != nil { |
| return nil, err |
| } |
| defer rows.Close() |
| items := []Message{} |
| for rows.Next() { |
| var i Message |
| if err := rows.Scan( |
| &i.ID, |
| &i.SessionID, |
| &i.Role, |
| &i.Parts, |
| &i.Model, |
| &i.CreatedAt, |
| &i.UpdatedAt, |
| &i.FinishedAt, |
| &i.Provider, |
| &i.IsSummaryMessage, |
| &i.PrismModelID, |
| &i.PrismModelName, |
| &i.PrismHypercreditSavings, |
| &i.PrismDollarSavings, |
| ); err != nil { |
| return nil, err |
| } |
| items = append(items, i) |
| } |
| if err := rows.Close(); err != nil { |
| return nil, err |
| } |
| if err := rows.Err(); err != nil { |
| return nil, err |
| } |
| return items, nil |
| } |
|
|
| const listMessagesBySession = `-- name: ListMessagesBySession :many |
| SELECT id, session_id, role, parts, model, created_at, updated_at, finished_at, provider, is_summary_message, prism_model_id, prism_model_name, prism_hypercredit_savings, prism_dollar_savings |
| FROM messages |
| WHERE session_id = ? |
| ORDER BY created_at ASC |
| ` |
|
|
| func (q *Queries) ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error) { |
| rows, err := q.query(ctx, q.listMessagesBySessionStmt, listMessagesBySession, sessionID) |
| if err != nil { |
| return nil, err |
| } |
| defer rows.Close() |
| items := []Message{} |
| for rows.Next() { |
| var i Message |
| if err := rows.Scan( |
| &i.ID, |
| &i.SessionID, |
| &i.Role, |
| &i.Parts, |
| &i.Model, |
| &i.CreatedAt, |
| &i.UpdatedAt, |
| &i.FinishedAt, |
| &i.Provider, |
| &i.IsSummaryMessage, |
| &i.PrismModelID, |
| &i.PrismModelName, |
| &i.PrismHypercreditSavings, |
| &i.PrismDollarSavings, |
| ); err != nil { |
| return nil, err |
| } |
| items = append(items, i) |
| } |
| if err := rows.Close(); err != nil { |
| return nil, err |
| } |
| if err := rows.Err(); err != nil { |
| return nil, err |
| } |
| return items, nil |
| } |
|
|
| const listUserMessagesBySession = `-- name: ListUserMessagesBySession :many |
| SELECT id, session_id, role, parts, model, created_at, updated_at, finished_at, provider, is_summary_message, prism_model_id, prism_model_name, prism_hypercredit_savings, prism_dollar_savings |
| FROM messages |
| WHERE session_id = ? AND role = 'user' |
| ORDER BY created_at DESC |
| ` |
|
|
| func (q *Queries) ListUserMessagesBySession(ctx context.Context, sessionID string) ([]Message, error) { |
| rows, err := q.query(ctx, q.listUserMessagesBySessionStmt, listUserMessagesBySession, sessionID) |
| if err != nil { |
| return nil, err |
| } |
| defer rows.Close() |
| items := []Message{} |
| for rows.Next() { |
| var i Message |
| if err := rows.Scan( |
| &i.ID, |
| &i.SessionID, |
| &i.Role, |
| &i.Parts, |
| &i.Model, |
| &i.CreatedAt, |
| &i.UpdatedAt, |
| &i.FinishedAt, |
| &i.Provider, |
| &i.IsSummaryMessage, |
| &i.PrismModelID, |
| &i.PrismModelName, |
| &i.PrismHypercreditSavings, |
| &i.PrismDollarSavings, |
| ); err != nil { |
| return nil, err |
| } |
| items = append(items, i) |
| } |
| if err := rows.Close(); err != nil { |
| return nil, err |
| } |
| if err := rows.Err(); err != nil { |
| return nil, err |
| } |
| return items, nil |
| } |
|
|
| const updateMessage = `-- name: UpdateMessage :exec |
| UPDATE messages |
| SET |
| parts = ?, |
| prism_model_id = ?, |
| prism_model_name = ?, |
| prism_hypercredit_savings = ?, |
| prism_dollar_savings = ?, |
| finished_at = ?, |
| updated_at = strftime('%s', 'now') |
| WHERE id = ? |
| ` |
|
|
| type UpdateMessageParams struct { |
| Parts string `json:"parts"` |
| PrismModelID sql.NullString `json:"prism_model_id"` |
| PrismModelName sql.NullString `json:"prism_model_name"` |
| PrismHypercreditSavings sql.NullFloat64 `json:"prism_hypercredit_savings"` |
| PrismDollarSavings sql.NullFloat64 `json:"prism_dollar_savings"` |
| FinishedAt sql.NullInt64 `json:"finished_at"` |
| ID string `json:"id"` |
| } |
|
|
| func (q *Queries) UpdateMessage(ctx context.Context, arg UpdateMessageParams) error { |
| _, err := q.exec(ctx, q.updateMessageStmt, updateMessage, |
| arg.Parts, |
| arg.PrismModelID, |
| arg.PrismModelName, |
| arg.PrismHypercreditSavings, |
| arg.PrismDollarSavings, |
| arg.FinishedAt, |
| arg.ID, |
| ) |
| return err |
| } |
|
|
