lmcli/pkg/lmcli/model/conversation.go

package model

import (
	"database/sql"
	"time"
)

type MessageRole string

const (
	MessageRoleSystem     MessageRole = "system"
	MessageRoleUser       MessageRole = "user"
	MessageRoleAssistant  MessageRole = "assistant"
	MessageRoleToolCall   MessageRole = "tool_call"
	MessageRoleToolResult MessageRole = "tool_result"
)

type Message struct {
	ID             uint `gorm:"primaryKey"`
	ConversationID uint `gorm:"foreignKey:ConversationID"`
	Content        string
	Role           MessageRole
	CreatedAt      time.Time
	ToolCalls      ToolCalls   // a json array of tool calls (from the modl)
	ToolResults    ToolResults // a json array of tool results
}

type Conversation struct {
	ID        uint `gorm:"primaryKey"`
	ShortName sql.NullString
	Title     string
}

type RequestParameters struct {
	Model string

	MaxTokens   int
	Temperature float32
	TopP        float32

	ToolBag []Tool
}

func (m *MessageRole) IsAssistant() bool {
	switch *m {
	case MessageRoleAssistant, MessageRoleToolCall:
		return true
	}
	return false
}

// FriendlyRole returns a human friendly signifier for the message's role.
func (m MessageRole) FriendlyRole() string {
	switch m {
	case MessageRoleUser:
		return "You"
	case MessageRoleSystem:
		return "System"
	case MessageRoleAssistant:
		return "Assistant"
	case MessageRoleToolCall:
		return "Tool Call"
	case MessageRoleToolResult:
		return "Tool Result"
	default:
		return string(m)
	}
}