contextcompressor

type ChatModelProxy struct {
	GenerateFunc func(ctx context.Context, messages []compactors.Message) (string, error)
}

type CompactMetadata struct {
	CompactType          string `json:"compact_type,omitempty"`
	PreCompactTokenCount int    `json:"pre_compact_token_count,omitempty"`
	LastMessageID        string `json:"last_message_id,omitempty"`
	UserContext          string `json:"user_context,omitempty"`
	MessagesSummarized   int    `json:"messages_summarized,omitempty"`
}

type CompactionResult struct {
	BoundaryMarker    *SystemMessage `json:"boundary_marker,omitempty"`
	SummaryMessages   []Message      `json:"summary_messages,omitempty"`
	MessagesToKeep    []Message      `json:"messages_to_keep,omitempty"`
	PreCompactTokens  int            `json:"pre_compact_tokens,omitempty"`
	PostCompactTokens int            `json:"post_compact_tokens,omitempty"`
}

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

type Compressor interface {
	Compact(ctx context.Context, messages []Message, opts ...Option) (*CompactionResult, error)
}

type Config struct {
	Model               string
	MaxOutputTokens     int
	CompactBufferTokens int
	CustomThreshold     int // 用户自定义压缩阈值，会覆盖自动计算的阈值
	CustomInstructions  string
	SuppressFollowUp    bool
}

type ContentBlock struct {
	Type       string           `json:"type"`
	Text       string           `json:"text,omitempty"`
	Image      *ImageContent    `json:"image,omitempty"`
	ToolUse    *ToolUseBlock    `json:"tool_use,omitempty"`
	ToolResult *ToolResultBlock `json:"tool_result,omitempty"`
	Document   *DocumentContent `json:"document,omitempty"`
}

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

type DocumentContent struct {
	Type     string `json:"type,omitempty"`
	Source   string `json:"source,omitempty"`
	MimeType string `json:"mime_type,omitempty"`
}

type ImageContent struct {
	Type     string `json:"type"`
	Source   string `json:"source,omitempty"`
	MimeType string `json:"mime_type,omitempty"`
}

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

type Message struct {
	ID       string         `json:"id"`
	Type     MessageType    `json:"type"`
	Role     string         `json:"role,omitempty"`
	Content  []ContentBlock `json:"content"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type MessageGroup struct {
	AssistantID string
	Messages    []Message
}

type MessageGrouper interface {
	Group(messages []Message) []MessageGroup
}

type MessageType string

type Option func(*Config)

type PressureLevel int

type SystemMessage struct {
	Message
	CompactMetadata *CompactMetadata `json:"compact_metadata,omitempty"`
}

type TokenWarningState struct {
	PercentLeft                 int  // 剩余百分比
	IsAboveWarningThreshold     bool // 是否超过警告阈值
	IsAboveErrorThreshold       bool // 是否超过错误阈值
	IsAboveAutoCompactThreshold bool // 是否超过自动压缩阈值
	IsAtBlockingLimit           bool // 是否达到阻塞限制
}

type Tokenizer interface {
	Estimate(text string) int
	EstimateMessages(messages []Message) int
}

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

type ToolResultBlock struct {
	ToolUseID string `json:"tool_use_id,omitempty"`
	Content   any    `json:"content,omitempty"`
	IsError   bool   `json:"is_error,omitempty"`
}

type ToolUseBlock struct {
	ID    string         `json:"id,omitempty"`
	Type  string         `json:"type,omitempty"`
	Name  string         `json:"name,omitempty"`
	Input map[string]any `json:"input,omitempty"`
}