vrr/opencode
2
0
Fork 0
mirror of https://github.com/anomalyco/opencode.git synced 2026-05-01 22:10:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

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/content.go
View file

@ -2,6 +2,7 @@ package message
import (
"encoding/base64"
"time"
)
type MessageRole string
@ -64,6 +65,7 @@ type ToolCall struct {
Name string `json:"name"`
Input string `json:"input"`
Type string `json:"type"`
Metadata any `json:"metadata"`
Finished bool `json:"finished"`
}
@ -80,6 +82,7 @@ func (ToolResult) isPart() {}
type Finish struct {
Reason string `json:"reason"`
Time int64 `json:"time"`
}
func (Finish) isPart() {}
@ -161,6 +164,15 @@ func (m *Message) IsFinished() bool {
return false
}
func (m *Message) FinishPart() *Finish {
for _, part := range m.Parts {
if c, ok := part.(Finish); ok {
return &c
}
}
return nil
}
func (m *Message) FinishReason() string {
for _, part := range m.Parts {
if c, ok := part.(Finish); ok {
@ -232,7 +244,7 @@ func (m *Message) SetToolResults(tr []ToolResult) {
}
func (m *Message) AddFinish(reason string) {
m.Parts = append(m.Parts, Finish{Reason: reason})
m.Parts = append(m.Parts, Finish{Reason: reason, Time: time.Now().Unix()})
}
func (m *Message) AddImageURL(url, detail string) {