1// Code generated by sqlc. DO NOT EDIT.
2// versions:
3// sqlc v1.27.0
4// source: messages.sql
5
6package db
7
8import (
9 "context"
10 "database/sql"
11)
12
13const createMessage = `-- name: CreateMessage :one
14INSERT INTO messages (
15 id,
16 session_id,
17 role,
18 finished,
19 content,
20 tool_calls,
21 tool_results,
22 created_at,
23 updated_at
24) VALUES (
25 ?, ?, ?, ?, ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now')
26)
27RETURNING id, session_id, role, content, thinking, finished, tool_calls, tool_results, created_at, updated_at
28`
29
30type CreateMessageParams struct {
31 ID string `json:"id"`
32 SessionID string `json:"session_id"`
33 Role string `json:"role"`
34 Finished bool `json:"finished"`
35 Content string `json:"content"`
36 ToolCalls sql.NullString `json:"tool_calls"`
37 ToolResults sql.NullString `json:"tool_results"`
38}
39
40func (q *Queries) CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error) {
41 row := q.queryRow(ctx, q.createMessageStmt, createMessage,
42 arg.ID,
43 arg.SessionID,
44 arg.Role,
45 arg.Finished,
46 arg.Content,
47 arg.ToolCalls,
48 arg.ToolResults,
49 )
50 var i Message
51 err := row.Scan(
52 &i.ID,
53 &i.SessionID,
54 &i.Role,
55 &i.Content,
56 &i.Thinking,
57 &i.Finished,
58 &i.ToolCalls,
59 &i.ToolResults,
60 &i.CreatedAt,
61 &i.UpdatedAt,
62 )
63 return i, err
64}
65
66const deleteMessage = `-- name: DeleteMessage :exec
67DELETE FROM messages
68WHERE id = ?
69`
70
71func (q *Queries) DeleteMessage(ctx context.Context, id string) error {
72 _, err := q.exec(ctx, q.deleteMessageStmt, deleteMessage, id)
73 return err
74}
75
76const deleteSessionMessages = `-- name: DeleteSessionMessages :exec
77DELETE FROM messages
78WHERE session_id = ?
79`
80
81func (q *Queries) DeleteSessionMessages(ctx context.Context, sessionID string) error {
82 _, err := q.exec(ctx, q.deleteSessionMessagesStmt, deleteSessionMessages, sessionID)
83 return err
84}
85
86const getMessage = `-- name: GetMessage :one
87SELECT id, session_id, role, content, thinking, finished, tool_calls, tool_results, created_at, updated_at
88FROM messages
89WHERE id = ? LIMIT 1
90`
91
92func (q *Queries) GetMessage(ctx context.Context, id string) (Message, error) {
93 row := q.queryRow(ctx, q.getMessageStmt, getMessage, id)
94 var i Message
95 err := row.Scan(
96 &i.ID,
97 &i.SessionID,
98 &i.Role,
99 &i.Content,
100 &i.Thinking,
101 &i.Finished,
102 &i.ToolCalls,
103 &i.ToolResults,
104 &i.CreatedAt,
105 &i.UpdatedAt,
106 )
107 return i, err
108}
109
110const listMessagesBySession = `-- name: ListMessagesBySession :many
111SELECT id, session_id, role, content, thinking, finished, tool_calls, tool_results, created_at, updated_at
112FROM messages
113WHERE session_id = ?
114ORDER BY created_at ASC
115`
116
117func (q *Queries) ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error) {
118 rows, err := q.query(ctx, q.listMessagesBySessionStmt, listMessagesBySession, sessionID)
119 if err != nil {
120 return nil, err
121 }
122 defer rows.Close()
123 items := []Message{}
124 for rows.Next() {
125 var i Message
126 if err := rows.Scan(
127 &i.ID,
128 &i.SessionID,
129 &i.Role,
130 &i.Content,
131 &i.Thinking,
132 &i.Finished,
133 &i.ToolCalls,
134 &i.ToolResults,
135 &i.CreatedAt,
136 &i.UpdatedAt,
137 ); err != nil {
138 return nil, err
139 }
140 items = append(items, i)
141 }
142 if err := rows.Close(); err != nil {
143 return nil, err
144 }
145 if err := rows.Err(); err != nil {
146 return nil, err
147 }
148 return items, nil
149}
150
151const updateMessage = `-- name: UpdateMessage :exec
152UPDATE messages
153SET
154 content = ?,
155 thinking = ?,
156 tool_calls = ?,
157 tool_results = ?,
158 finished = ?,
159 updated_at = strftime('%s', 'now')
160WHERE id = ?
161`
162
163type UpdateMessageParams struct {
164 Content string `json:"content"`
165 Thinking string `json:"thinking"`
166 ToolCalls sql.NullString `json:"tool_calls"`
167 ToolResults sql.NullString `json:"tool_results"`
168 Finished bool `json:"finished"`
169 ID string `json:"id"`
170}
171
172func (q *Queries) UpdateMessage(ctx context.Context, arg UpdateMessageParams) error {
173 _, err := q.exec(ctx, q.updateMessageStmt, updateMessage,
174 arg.Content,
175 arg.Thinking,
176 arg.ToolCalls,
177 arg.ToolResults,
178 arg.Finished,
179 arg.ID,
180 )
181 return err
182}