llm

type API string

type Action string

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

type AnthropicConfig = Config

type Candidate struct {
	ID      string `json:"id"`
	Content string `json:"content"`
}

type ChatResult struct {
	Text  string
	Calls []ToolCall
}

type ChatTurn struct {
	// Role is one of RoleUser, RoleAssistant, RoleTool.
	Role string
	// Text is the user/assistant text, or the tool result content.
	Text string
	// Calls are the tool calls an assistant turn requested.
	Calls []ToolCall
	// CallID names the call a tool turn answers.
	CallID string
	// Name is the tool turn's tool name.
	Name string
}

type Client interface {
	Consolidator
	Completer
	Distiller
	Merger
}

type Completer interface {
	Complete(ctx context.Context, system, user string) (string, error)
}

type Config struct {
	BaseURL string // OpenAI: e.g. https://host/v1 ; Anthropic: e.g. https://api.minimax.io/anthropic
	APIKey  string
	Model   string
	// MaxTokens caps the completion length (defaults to defaultMaxTokens). A
	// budget is required for reasoning models, which otherwise spend the server
	// default on hidden reasoning and return empty content.
	MaxTokens  int
	HTTPClient *http.Client
}

type Consolidator interface {
	Consolidate(ctx context.Context, in Input) (Decision, error)
}

type Decision struct {
	Action Action `json:"action"`
	// Target is the candidate ID affected by update/supersede (empty for new).
	Target string `json:"target"`
	// Content is the merged text to persist for an update (else the new content).
	Content string `json:"content"`
	// Summary is an optional one-line summary of the resulting memory.
	Summary string `json:"summary"`
	// Reason explains the decision (for logs/debugging).
	Reason string `json:"reason"`
	// LinkedIDs are IDs of existing memories related to this one (same
	// entity/topic) but neither duplicate nor contradiction. Empty for "new"
	// with no related candidates or update/supersede actions.
	LinkedIDs []string `json:"linked_ids,omitempty"`
}

type DistillInput struct {
	Episodes []Episode `json:"episodes"`
	Now      string    `json:"now,omitempty"`
}

type Distiller interface {
	Distill(ctx context.Context, in DistillInput) ([]Fact, error)
}

type Episode struct {
	Content string `json:"content"`
	// Date is the YYYY-MM-DD the episode was recorded. Empty when unknown.
	Date string `json:"date,omitempty"`
}

type Fact struct {
	Content string `json:"content"`
	Summary string `json:"summary,omitempty"`
	// Category routes the fact to a tier: "procedure" (incl. error→recovery) →
	// procedural; "preference" and "fact" → semantic. Empty defaults to semantic.
	Category string `json:"category,omitempty"`
	// Confidence is the LLM's self-assessed reliability of this fact, in [0.1, 0.7].
	// nil means unset; the service layer falls back to ConfidenceSeedFresh.
	Confidence *float64 `json:"confidence,omitempty"`
}

type Input struct {
	New        string      `json:"new"`
	Tier       string      `json:"tier"`
	Candidates []Candidate `json:"candidates"`
}

type Merger interface {
	MergeMemories(ctx context.Context, contents []string) (string, error)
}

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

type OpenAIConfig = Config

type Tool struct {
	Name        string
	Description string
	Schema      map[string]any
}

type ToolCall struct {
	ID   string
	Name string
	Args json.RawMessage
}

type ToolChat interface {
	ChatTools(ctx context.Context, system string, turns []ChatTurn, tools []Tool, choice ToolChoice) (ChatResult, error)
}

type ToolChoice string