message

Documentation

Index

• Constants
• func Delete(ctx context.Context, id string) error
• func DeleteSessionMessages(ctx context.Context, sessionID string) error
• func InitService(dbConn *sql.DB) error
• func Subscribe(ctx context.Context) <-chan pubsub.Event[Message]
• type Attachment
• type BinaryContent
  • func (bc BinaryContent) String(provider models.ModelProvider) string
• type ContentPart
• type CreateMessageParams
• type DBFinish
• type Finish
• type FinishReason
• type ImageURLContent
  • func (iuc ImageURLContent) String() string
• type Message
  • func Create(ctx context.Context, sessionID string, params CreateMessageParams) (Message, error)
  • func Get(ctx context.Context, id string) (Message, error)
  • func List(ctx context.Context, sessionID string) ([]Message, error)
  • func ListAfter(ctx context.Context, sessionID string, timestamp time.Time) ([]Message, error)
  • func Update(ctx context.Context, message Message) (Message, error)
  • func (m *Message) AddBinary(mimeType string, data []byte)
  • func (m *Message) AddFinish(reason FinishReason)
  • func (m *Message) AddImageURL(url, detail string)
  • func (m *Message) AddToolCall(tc ToolCall)
  • func (m *Message) AddToolResult(tr ToolResult)
  • func (m *Message) AppendContent(delta string)
  • func (m *Message) AppendReasoningContent(delta string)
  • func (m *Message) AppendToolCallInput(toolCallID string, inputDelta string)
  • func (m *Message) BinaryContent() []BinaryContent
  • func (m *Message) Content() *TextContent
  • func (m *Message) FinishPart() *Finish
  • func (m *Message) FinishReason() FinishReason
  • func (m *Message) FinishToolCall(toolCallID string)
  • func (m *Message) ImageURLContent() []ImageURLContent
  • func (m *Message) IsFinished() bool
  • func (m *Message) IsThinking() bool
  • func (m *Message) ReasoningContent() ReasoningContent
  • func (m *Message) SetToolCalls(tc []ToolCall)
  • func (m *Message) SetToolResults(tr []ToolResult)
  • func (m *Message) ToolCalls() []ToolCall
  • func (m *Message) ToolResults() []ToolResult
• type MessageRole
• type ReasoningContent
  • func (tc ReasoningContent) String() string
• type Service
  • func GetService() Service
• type TextContent
  • func (tc *TextContent) String() string
• type ToolCall
• type ToolResult

Constants

const (
	EventMessageCreated pubsub.EventType = "message_created"
	EventMessageUpdated pubsub.EventType = "message_updated"
	EventMessageDeleted pubsub.EventType = "message_deleted"
)

Functions

func Delete

func Delete(ctx context.Context, id string) error

func DeleteSessionMessages

func DeleteSessionMessages(ctx context.Context, sessionID string) error

func InitService

func InitService(dbConn *sql.DB) error

func Subscribe

func Subscribe(ctx context.Context) <-chan pubsub.Event[Message]

Types

type Attachment

type Attachment struct {
	FilePath string
	FileName string
	MimeType string
	Content  []byte
}

type BinaryContent

type BinaryContent struct {
	Path     string
	MIMEType string
	Data     []byte
}

func (BinaryContent) String

func (bc BinaryContent) String(provider models.ModelProvider) string

type ContentPart

type ContentPart interface {
	// contains filtered or unexported methods
}

type CreateMessageParams

type CreateMessageParams struct {
	Role  MessageRole
	Parts []ContentPart
	Model models.ModelID
}

type DBFinish

type DBFinish struct {
	Reason FinishReason `json:"reason"`
	Time   int64        `json:"time"`
}

type Finish

type Finish struct {
	Reason FinishReason `json:"reason"`
	Time   time.Time    `json:"time"`
}

type FinishReason

type FinishReason string

const (
	FinishReasonEndTurn          FinishReason = "end_turn"
	FinishReasonMaxTokens        FinishReason = "max_tokens"
	FinishReasonToolUse          FinishReason = "tool_use"
	FinishReasonCanceled         FinishReason = "canceled"
	FinishReasonError            FinishReason = "error"
	FinishReasonPermissionDenied FinishReason = "permission_denied"

	// Should never happen
	FinishReasonUnknown FinishReason = "unknown"
)

type ImageURLContent

type ImageURLContent struct {
	URL    string `json:"url"`
	Detail string `json:"detail,omitempty"`
}

func (ImageURLContent) String

func (iuc ImageURLContent) String() string

type Message

type Message struct {
	ID        string
	Role      MessageRole
	SessionID string
	Parts     []ContentPart
	Model     models.ModelID
	CreatedAt time.Time
	UpdatedAt time.Time
}

func Create

func Create(ctx context.Context, sessionID string, params CreateMessageParams) (Message, error)

func Get

func Get(ctx context.Context, id string) (Message, error)

func List

func List(ctx context.Context, sessionID string) ([]Message, error)

func ListAfter

func ListAfter(ctx context.Context, sessionID string, timestamp time.Time) ([]Message, error)

func Update

func Update(ctx context.Context, message Message) (Message, error)

func (*Message) AddBinary

func (m *Message) AddBinary(mimeType string, data []byte)

func (*Message) AddFinish

func (m *Message) AddFinish(reason FinishReason)

func (*Message) AddImageURL

func (m *Message) AddImageURL(url, detail string)

func (*Message) AddToolCall

func (m *Message) AddToolCall(tc ToolCall)

func (*Message) AddToolResult

func (m *Message) AddToolResult(tr ToolResult)

func (*Message) AppendContent

func (m *Message) AppendContent(delta string)

func (*Message) AppendReasoningContent

func (m *Message) AppendReasoningContent(delta string)

func (*Message) AppendToolCallInput

func (m *Message) AppendToolCallInput(toolCallID string, inputDelta string)

func (*Message) BinaryContent

func (m *Message) BinaryContent() []BinaryContent

func (*Message) Content

func (m *Message) Content() *TextContent

func (*Message) FinishPart

func (m *Message) FinishPart() *Finish

func (*Message) FinishReason

func (m *Message) FinishReason() FinishReason

func (*Message) FinishToolCall

func (m *Message) FinishToolCall(toolCallID string)

func (*Message) ImageURLContent

func (m *Message) ImageURLContent() []ImageURLContent

func (*Message) IsFinished

func (m *Message) IsFinished() bool

func (*Message) IsThinking

func (m *Message) IsThinking() bool

func (*Message) ReasoningContent

func (m *Message) ReasoningContent() ReasoningContent

func (*Message) SetToolCalls

func (m *Message) SetToolCalls(tc []ToolCall)

func (*Message) SetToolResults

func (m *Message) SetToolResults(tr []ToolResult)

func (*Message) ToolCalls

func (m *Message) ToolCalls() []ToolCall

func (*Message) ToolResults

func (m *Message) ToolResults() []ToolResult

type MessageRole

type MessageRole string

const (
	Assistant MessageRole = "assistant"
	User      MessageRole = "user"
	System    MessageRole = "system"
	Tool      MessageRole = "tool"
)

type ReasoningContent

type ReasoningContent struct {
	Thinking string `json:"thinking"`
}

func (ReasoningContent) String

func (tc ReasoningContent) String() string

type Service

type Service interface {
	pubsub.Subscriber[Message]

	Create(ctx context.Context, sessionID string, params CreateMessageParams) (Message, error)
	Update(ctx context.Context, message Message) (Message, error)
	Get(ctx context.Context, id string) (Message, error)
	List(ctx context.Context, sessionID string) ([]Message, error)
	ListAfter(ctx context.Context, sessionID string, timestamp time.Time) ([]Message, error)
	Delete(ctx context.Context, id string) error
	DeleteSessionMessages(ctx context.Context, sessionID string) error
}

func GetService

func GetService() Service

type TextContent

type TextContent struct {
	Text string `json:"text"`
}

func (*TextContent) String

func (tc *TextContent) String() string

type ToolCall

type ToolCall struct {
	ID       string `json:"id"`
	Name     string `json:"name"`
	Input    string `json:"input"`
	Type     string `json:"type"`
	Finished bool   `json:"finished"`
}

type ToolResult

type ToolResult struct {
	ToolCallID string `json:"tool_call_id"`
	Name       string `json:"name"`
	Content    string `json:"content"`
	Metadata   string `json:"metadata"`
	IsError    bool   `json:"is_error"`
}