diff options
Diffstat (limited to 'internal/message/content.go')
| -rw-r--r-- | internal/message/content.go | 14 |
1 files changed, 2 insertions, 12 deletions
diff --git a/internal/message/content.go b/internal/message/content.go index 0a1ba714b..d7b76f8ce 100644 --- a/internal/message/content.go +++ b/internal/message/content.go @@ -106,21 +106,11 @@ func (ToolResult) isPart() {} type Finish struct { Reason FinishReason `json:"reason"` - Time int64 `json:"time"` + Time time.Time `json:"time"` } func (Finish) isPart() {} -type Message struct { - ID string - Role MessageRole - SessionID string - Parts []ContentPart - Model models.ModelID - CreatedAt int64 - UpdatedAt int64 -} - func (m *Message) Content() *TextContent { for _, part := range m.Parts { if c, ok := part.(TextContent); ok { @@ -318,7 +308,7 @@ func (m *Message) AddFinish(reason FinishReason) { break } } - m.Parts = append(m.Parts, Finish{Reason: reason, Time: time.Now().UnixMilli()}) + m.Parts = append(m.Parts, Finish{Reason: reason, Time: time.Now()}) } func (m *Message) AddImageURL(url, detail string) { |