domain

type ActionChain struct {
	// contains filtered or unexported fields
}

type AddAction interface {
	Name() string
	Handle(*AddContext)
}

type AddContext struct {

	// 输入
	Messages Messages

	// 输出
	Episodes  []Episode
	Entities  []Entity
	Edges     []Edge
	Summaries []Summary

	// 配置
	Language       string  // 语言设置
	TopicThreshold float64 // 主题变化阈值 (默认 0.7)
	// contains filtered or unexported fields
}

type AddRequest struct {
	AgentID   string    `json:"agent_id"`
	UserID    string    `json:"user_id"`
	SessionID string    `json:"session_id"`
	Messages  []Message `json:"messages"`
}

type AddResponse struct {
	Success   bool      `json:"success"`
	Episodes  []Episode `json:"episodes,omitempty"`
	Entities  []Entity  `json:"entities,omitempty"`
	Edges     []Edge    `json:"edges,omitempty"`
	Summaries []Summary `json:"summaries,omitempty"`
}

type Edge struct {
	ID       string `json:"id"`
	SourceID string `json:"source_id"` // 起始实体 ID
	TargetID string `json:"target_id"` // 目标实体 ID
	Relation string `json:"relation"`  // 关系类型 (如: 喜欢, 认识, 住在)
	Fact     string `json:"fact"`      // 事实描述 (如: "用户喜欢喝咖啡")

	// 向量
	Embedding []float32 `json:"embedding,omitempty"`

	// 双时间轴 (Bi-temporal)
	ValidAt   *time.Time `json:"valid_at,omitempty"`   // 事实生效时间
	InvalidAt *time.Time `json:"invalid_at,omitempty"` // 事实失效时间

	// 入库时间
	CreatedAt time.Time  `json:"created_at"`
	ExpiredAt *time.Time `json:"expired_at,omitempty"` // 记录过期时间

	// 溯源
	EpisodeIDs []string `json:"episode_ids,omitempty"` // 来源 Episode

	// 检索分数
	Score float64 `json:"score,omitempty"`
}

type Entity struct {
	ID          string     `json:"id"`
	AgentID     string     `json:"agent_id"`
	UserID      string     `json:"user_id"`
	Name        string     `json:"name"`        // 实体名称
	Type        EntityType `json:"type"`        // 实体类型
	Description string     `json:"description"` // 实体描述

	// 向量
	Embedding []float32 `json:"embedding,omitempty"`

	// 时间
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`

	// 检索分数
	Score float64 `json:"score,omitempty"`
}

type EntityType string

type Episode struct {
	ID        string `json:"id"`
	AgentID   string `json:"agent_id"`
	UserID    string `json:"user_id"`
	SessionID string `json:"session_id"`

	// 对话内容
	Role string `json:"role"` // user / assistant
	Name string `json:"name"` // 发言者名称

	// 主题
	Topic          string    `json:"topic"`
	TopicEmbedding []float32 `json:"topic_embedding"`

	// 内容
	Content   string    `json:"content"`
	Embedding []float32 `json:"content_embedding"`

	// 时间
	Timestamp time.Time `json:"timestamp"`  // 对话发生时间
	CreatedAt time.Time `json:"created_at"` // 入库时间

	// 检索分数 (查询时填充)
	Score float64 `json:"score,omitempty"`
}

type Message struct {
	Role    string `json:"role"`           // user / assistant / system
	Content string `json:"content"`        // 消息内容
	Name    string `json:"name,omitempty"` // 发言者名称
}

type Messages []Message

type RecallAction interface {
	Name() string
	HandleRecall(*RecallContext)
}

type RecallChain struct {
	// contains filtered or unexported fields
}

type RecallContext struct {

	// 查询参数
	Query     string
	Embedding []float32
	Limit     int
	Options   RetrieveOptions

	// 检索结果 - Zep 三层结构
	Episodes  []Episode
	Entities  []Entity
	Edges     []Edge
	Summaries []Summary
	// contains filtered or unexported fields
}

type RetrieveOptions struct {
	MaxHops   int `json:"max_hops,omitempty"`   // 图遍历最大跳数
	MaxTokens int `json:"max_tokens,omitempty"` // 总 token 预算，默认 2000

	// 各类型数量限制（-1 禁用，0 使用默认值，>0 自定义）
	MaxSummaries int `json:"max_summaries,omitempty"` // 默认 3
	MaxEdges     int `json:"max_edges,omitempty"`     // 默认 10
	MaxEntities  int `json:"max_entities,omitempty"`  // 默认 5
	MaxEpisodes  int `json:"max_episodes,omitempty"`  // 默认 5
}

type RetrieveRequest struct {
	AgentID   string `json:"agent_id"`
	UserID    string `json:"user_id"`
	SessionID string `json:"session_id,omitempty"`
	Query     string `json:"query"`
	Limit     int    `json:"limit,omitempty"`

	// 检索选项
	Options RetrieveOptions `json:"options,omitempty"`
}

type RetrieveResponse struct {
	Success   bool      `json:"success"`
	Episodes  []Episode `json:"episodes,omitempty"`
	Entities  []Entity  `json:"entities,omitempty"`
	Edges     []Edge    `json:"edges,omitempty"`
	Summaries []Summary `json:"summaries,omitempty"`
	Total     int       `json:"total"`

	// 格式化后的记忆上下文 (用于 LLM prompt)
	MemoryContext string `json:"memory_context,omitempty"`
}

type Summary struct {
	ID         string   `json:"id"`
	AgentID    string   `json:"agent_id"`
	UserID     string   `json:"user_id"`
	EpisodeIDs []string `json:"episode_ids"` // 关联的 Episode ID
	Content    string   `json:"content"`     // 摘要内容
	Topic      string   `json:"topic"`       // 主题标签

	// 向量
	Embedding []float32 `json:"embedding,omitempty"`

	// 时间
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`

	// 检索分数
	Score float64 `json:"score,omitempty"`
}

type TokenUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
}