structs

type Chats struct {
	ID              uint32 `gorm:"primaryKey;autoIncrement"`
	LastModelID     uint32
	NowAgent        string
	Root            string
	TraceID         uint64
	State           state.State
	Title           string // 用户设置的标题（/title 命令写入）
	AITitle         string // AI 生成的标题（自动生成/compress 重生成写入）
	ReasoningEffort string

	Stop                     bool                `gorm:"-" json:"-"`
	DB                       *gorm.DB            `gorm:"-" json:"-"`
	CurrentAgentID           string              `gorm:"-" json:"-"`
	CurrentAgentConfig       structs.AgentConfig `gorm:"-" json:"-"`
	CurrentActivatePath      string              `gorm:"-" json:"-"`
	EnableScopes             map[string]bool     `gorm:"-" json:"-"`
	TemporyDataOfRequest     map[string]any      `gorm:"-" json:"-"`
	TemporyDataOfSession     map[string]any      `gorm:"-" json:"-"`
	InTestFlag               bool                `gorm:"-" json:"-"`
	ReferCount               int32               `gorm:"-" json:"-"`
	ToolCallingContext       map[string]any      `gorm:"-" json:"-"`
	ToolCallingType          map[string]string   `gorm:"-" json:"-"`
	CurrentToolID            string              `gorm:"-" json:"-"`
	CurrentMessageID         uint64              `gorm:"-" json:"-"`
	ToolState                uint64              `gorm:"-" json:"-"`
	LatestToolCallingContext map[string]any      `gorm:"-" json:"-"`
	LatestToolCallingType    map[string]string   `gorm:"-" json:"-"`
	// ToolCallingStreaming 标记每个工具调用 id 是否为流式增量预览（true）还是最终状态（false）。
	// OnHook 写入时按 session.State 判定：StateReciving/StateRequesting（AI 正在生成）→ 增量；
	// StateToolCalling（审批后执行）→ 最终。SetCallback 据此选事件名。
	ToolCallingStreaming map[string]bool `gorm:"-" json:"-"`

	// ToolKillFn 由当前正在执行的工具注册，loop.Stop() 调用它来中断工具
	ToolKillFn func() `gorm:"-" json:"-"`
	// contains filtered or unexported fields
}

type ClassifySegment struct {
	ID        uint64 `gorm:"primaryKey;autoIncrement"`
	ChatID    uint32 `gorm:"index"`
	MessageID uint64 `gorm:"index"`
	Label     string `gorm:"type:text"`
	Text      string `gorm:"type:text"`
	TempPath  string `gorm:"type:text"`
}

type Configs struct {
	LastChatID uint32
}

type Messages struct {
	ID            uint64 `gorm:"primaryKey;autoIncrement"`
	ChatID        uint32
	AgentID       *string
	Delta         string `gorm:"type:text"`
	Summary       string `gorm:"type:text"`
	ThinkingDelta string `gorm:"type:text"`
	Chats         *Chats `gorm:"foreignKey:ChatID;constraints:OnDelete:RESTRICT;OnUpdate:CASCADE"`
	// SubAgents             SubAgents         `gorm:"foreignKey:AgentID;constraints:OnDelete:RESTRICT;OnUpdate:CASCADE"`
	Refers                MessagesReferList `gorm:"type:bytes;serialize:gob"`
	ToolCallingJSONString string            `gorm:"type:text"`
	// ToolFinished          bool              `gorm:"default:false"`
	Time             uint64 `gorm:"autoCreateTime"`
	ModelName        string
	ModelID          uint32
	Type             MessagesRole
	PromptTokens     uint32
	CompletionTokens uint32
	TotalTokens      uint32
	CachedTokens     uint32
}

type MessagesRefer struct {
	FilePath     string
	FileType     MessagesReferType
	FileFromLine int32
	FileFromCol  int32
	FileToLine   int32
	FileToCol    int32
	Origin       []byte
}

type MessagesReferList []MessagesRefer

type MessagesReferType uint8

type MessagesRole uint8

type ReferFiles struct {
	ChatID   uint32 `gorm:"primaryKey"`
	Path     string `gorm:"primaryKey"`
	Content  string `gorm:"type:text"`
	ReadOnly bool
	Chats    Chats `gorm:"foreignKey:ChatID;constraints:OnDelete:RESTRICT;OnUpdate:CASCADE"`
}

type Scopes struct {
	ChatID  uint32 `gorm:"primaryKey;column:chat_id"`
	Name    string `gorm:"primaryKey;column:name"`
	Enabled bool   `gorm:"column:enabled"`
	Chats   *Chats `gorm:"foreignKey:ChatID;references:ID"`
}

type SubAgents struct {
	ID string `gorm:"primaryKey"`
	// ChatID      uint32
	AgentID     string
	BindPath    string
	Deleted     bool
	LastSummary string
}

type Terminals struct {
	ID      uint32 `gorm:"primaryKey"`
	ChatID  uint32
	Chats   Chats  `gorm:"foreignKey:ChatID"`
	History []byte `gorm:"type:blob"`
	Title   string
}

type Traces struct {
	Path    string `gorm:"primaryKey"`
	ChatID  uint32 `gorm:"primaryKey"`
	AgentID string `gorm:"primaryKey"`
	TraceID uint64
	Chats   *Chats `gorm:"foreignKey:ChatID"`
}