mirrors/opencode
1
0
Fork 0
mirror of https://github.com/sst/opencode.git synced 2025-08-04 13:30:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

add initial message handling

Browse source
This commit is contained in:
Kujtim Hoxha 2025-04-12 02:01:45 +02:00
parent 08bd75bb6e
commit 8d874b839d
14 changed files with 1132 additions and 466 deletions
Download patch file Download diff file Expand all files Collapse all files

14
internal/message/message.go
View file

@ -2,17 +2,20 @@ package message
import (
"context"
"database/sql"
"encoding/json"
"fmt"
"github.com/google/uuid"
"github.com/kujtimiihoxha/termai/internal/db"
"github.com/kujtimiihoxha/termai/internal/llm/models"
"github.com/kujtimiihoxha/termai/internal/pubsub"
)
type CreateMessageParams struct {
Role MessageRole
Parts []ContentPart
Model models.ModelID
}
type Service interface {
@ -68,6 +71,7 @@ func (s *service) Create(sessionID string, params CreateMessageParams) (Message,
SessionID: sessionID,
Role: string(params.Role),
Parts: string(partsJSON),
Model: sql.NullString{String: string(params.Model), Valid: true},
})
if err != nil {
return Message{}, err
@ -101,9 +105,15 @@ func (s *service) Update(message Message) error {
if err != nil {
return err
}
finishedAt := sql.NullInt64{}
if f := message.FinishPart(); f != nil {
finishedAt.Int64 = f.Time
finishedAt.Valid = true
}
err = s.q.UpdateMessage(s.ctx, db.UpdateMessageParams{
ID: message.ID,
Parts: string(parts),
ID: message.ID,
Parts: string(parts),
FinishedAt: finishedAt,
})
if err != nil {
return err