llmx

type Agent struct {
	Name        string
	Description string
	// contains filtered or unexported fields
}

type AgentOption func(*AgentOptions)

type AgentOptions struct {
	Model                string
	Instruction          string
	Loop                 Loop
	ReasonLevel          int
	Tools                []*Tool
	MaxTurn              int
	Stream               bool
	Compress             *CompressOptions
	EventSink            EventSink
	LLMInterceptors      []LLMInterceptor
	ToolCallInterceptors []ToolCallInterceptor
}

type CompressInput struct {
	Model    string
	Messages []*types.Message
	Options  *CompressOptions

	// Tokens is the context usage the model last reported. Compression is
	// decided exclusively from this value; zero means there is no reported
	// context size to compress yet.
	Tokens int64
}

type CompressOptions struct {
	// KeepLastTurn is the number of recent user turns left unchanged.
	KeepLastTurn int

	ToolResultLevel      float64
	ToolResultReplace    string
	ToolResultReplaceTag string

	SummaryLevel       float64
	SummaryModel       string
	SummaryInstruction string
	SummaryBridge      string
	SummaryBridgeTag   string
	SummaryTimeout     time.Duration
}

type CompressResultSummary struct {
	Message *types.Message        `json:"message,omitempty" yaml:"message,omitempty"`
	Usage   types.CompletionUsage `json:"usage,omitempty" yaml:"usage,omitempty"`
}

type CompressResultToolResult struct {
	Result       string            `json:"result,omitempty" yaml:"result,omitempty"`
	Instructions map[string]string `json:"instructions,omitempty" yaml:"instructions,omitempty"`
}

type Event struct {
	ID string `json:"id,omitempty" yaml:"id,omitempty"`

	Message       *types.Message      `json:"message,omitempty" yaml:"message,omitempty"`
	HumanRequest  *EventHumanRequest  `json:"humanRequest,omitempty" yaml:"humanRequest,omitempty"`
	HumanResponse *EventHumanResponse `json:"humanResponse,omitempty" yaml:"humanResponse,omitempty"`
	Compression   *EventCompression   `json:"compression,omitempty" yaml:"compression,omitempty"`
	Usage         *EventUsage         `json:"usage,omitempty" yaml:"usage,omitempty"`
	State         *EventState         `json:"state,omitempty" yaml:"state,omitempty"`
}

type EventCompression struct {
	//one of
	ToolResult *CompressResultToolResult `json:"toolResult,omitempty" yaml:"toolResult,omitempty"`
	Summary    *CompressResultSummary    `json:"summary,omitempty" yaml:"summary,omitempty"`

	// UntilMessageID is the last history message this compression covers,
	// inclusive. Messages after it are left unchanged.
	UntilMessageID string `json:"untilMessageID,omitempty" yaml:"untilMessageID,omitempty"`
}

type EventHumanRequest struct {
	ID        string                 `json:"id,omitempty" yaml:"id,omitempty"`
	Arguments map[string]any         `json:"arguments,omitempty" yaml:"arguments,omitempty"`
	ToolCall  *types.MessageToolCall `json:"toolCall,omitempty" yaml:"toolCall,omitempty"`
}

type EventHumanResponse struct {
	ID     string         `json:"id,omitempty" yaml:"id,omitempty"`
	Result map[string]any `json:"result,omitempty" yaml:"result,omitempty"`
}

type EventSink func(context.Context, *Event) error

type EventState struct {
	Patch bool              `json:"patch,omitempty" yaml:"patch,omitempty"`
	State map[string]string `json:"state,omitempty" yaml:"state,omitempty"`
}

type EventUsage struct {
	Usage types.CompletionUsage `json:"usage,omitempty" yaml:"usage,omitempty"`
}

type LLMHandler func(context.Context, *SessionTurn) (*types.Completion, error)

type LLMInterceptor func(LLMHandler) LLMHandler

type Loop func(ctx context.Context, turn *SessionTurnResult) (LoopAction, error)

type LoopAction uint8

type Model struct {
	Name     string
	Model    string
	MaxToken int64
	Provider types.Provider
	Fallback *ModelFallback
}

type ModelConfig struct {
	Name     string               `yaml:"name"`
	Model    string               `yaml:"model"`
	Provider string               `yaml:"provider"`
	MaxToken int64                `yaml:"maxToken"`
	Fallback *ModelConfigFallback `yaml:"fallback"`
}

type ModelConfigFallback struct {
	MaxSeconds int                           `yaml:"maxSeconds"`
	Providers  []ModelConfigFallbackProvider `yaml:"providers"`
}

type ModelConfigFallbackProvider struct {
	Provider string `yaml:"provider"`
	Model    string `yaml:"model"`
}

type ModelFallback struct {
	MaxSeconds int
	Providers  []*ModelFallbackProvider
	// contains filtered or unexported fields
}

type ModelFallbackProvider struct {
	ProviderName string
	Provider     types.Provider
	Model        string
}

type ModelInfo struct {
	MaxToken       int64 `yaml:"maxToken"`
	MaxOutputToken int64 `yaml:"maxOutputToken"`
}

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

type ModelsConfig struct {
	Providers []ProviderConfig `yaml:"providers"`
	Models    []ModelConfig    `yaml:"models"`
}

type ProviderConfig struct {
	Name          string                       `yaml:"name"`
	Provider      string                       `yaml:"provider"`
	Url           string                       `yaml:"url"`
	Insecure      bool                         `yaml:"insecure"`
	Sk            string                       `yaml:"sk"`
	ThinkArgs     *types.ProviderThinkArgs     `yaml:"thinkArgs"`
	EmbeddingArgs *types.ProviderEmbeddingArgs `yaml:"embeddingArgs"`
}

type ProviderCreator func(opts ...types.ProviderOption) (types.Provider, error)

type ProviderEnv struct {
	Provider     string
	Url          string
	LegacyURL    string
	ApiKey       string
	LegacyAPIKey string
}

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

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

type SessionHumanResponse struct {
	ID     string         `json:"id,omitempty" yaml:"id,omitempty"`
	Result map[string]any `json:"result,omitempty" yaml:"result,omitempty"`
}

type SessionInput struct {
	// Messages are appended to the model history, one message event each.
	Messages []*types.Message `json:"messages,omitempty" yaml:"messages,omitempty"`

	// HumanResponses resolve human requests that are still pending.
	HumanResponses []*SessionHumanResponse `json:"humanResponses,omitempty" yaml:"humanResponses,omitempty"`

	// State replaces the session state before the rest of this input.
	// StateDelta merges into the current session state instead.
	State      map[string]string `json:"state,omitempty" yaml:"state,omitempty"`
	StateDelta map[string]string `json:"stateDelta,omitempty" yaml:"stateDelta,omitempty"`
}

type SessionResult struct {
	Events []*Event
	State  map[string]string

	Turns      int
	StopReason StopReason
	Cause      error
}

type SessionTurn struct {
	Index       int
	Model       string
	Instruction string
	History     []*types.Message
	Tools       []*Tool
	// contains filtered or unexported fields
}

type SessionTurnResult struct {
	Index          int
	Events         []*Event
	TurnEventIndex int
}

type StopReason string

type Tool struct {
	Name        string
	Description string
	Parameters  map[string]any
	Invoker     ToolInvoker
}

type ToolCallHandler func(ToolContext, *types.MessageToolCall) (any, error)

type ToolCallInterceptor func(ToolCallHandler) ToolCallHandler

type ToolContext interface {
	context.Context
	ToolCallID() string
	State() map[string]string
	SetState(patch map[string]string) error
	// HumanRequest and HumanResponse return immutable session values. Call Clone
	// before modifying nested fields.
	HumanRequest() *EventHumanRequest
	HumanResponse() *EventHumanResponse
}

type ToolInvoker func(ToolContext, string) (any, error)

type ToolResultHuman struct {
	Arguments map[string]any
}

type ToolResultNotify struct{}

type ToolResultWithInstruction struct {
	Result      any
	Instruction map[string]string
}

type ToolRun struct {
	Call         *types.MessageToolCall
	Result       any
	Error        error
	Duration     time.Duration
	StatePatches []map[string]string
}