harness

type AgentHarness struct {
	Env any
	// contains filtered or unexported fields
}

type AgentHarnessAuth struct {
	APIKey  string
	Headers map[string]string
}

type AgentHarnessError struct {
	Code string
	Err  error
}

type AgentHarnessEvent struct {
	Type string

	Message     llm.Message
	Messages    []llm.Message
	ToolResults []llm.Message

	ToolCallID string
	ToolName   string
	Input      map[string]any
	Content    []llm.ContentPart
	Details    any
	IsError    bool

	Steer    []llm.Message
	FollowUp []llm.Message
	NextTurn []llm.Message

	HadPendingMutations bool
	NextTurnCount       int

	Model         llm.Model
	PreviousModel *llm.Model
	Source        string

	Level         string
	PreviousLevel string

	Resources         AgentHarnessResources
	PreviousResources AgentHarnessResources

	Prompt       string
	Images       []llm.ContentPart
	SystemPrompt string

	SessionID     string
	StreamOptions AgentHarnessStreamOptions
	Payload       any

	Status  int
	Headers map[string]string

	ClearedSteer    []llm.Message
	ClearedFollowUp []llm.Message

	Preparation        *CompactionPreparation
	BranchEntries      []Entry
	CustomInstructions string
	Compaction         CompactionResult
	CompactionEntry    Entry
	FromHook           bool
	TargetID           string
	OldLeafID          *string
	NewLeafID          *string
	CommonAncestorID   *string
	BranchSummary      BranchSummaryResult
	SummaryEntry       *Entry
	Operation          string
	Attempt            int
	MaxAttempts        int
	DelayMs            int
	ErrorMessage       string

	AgentEvent core.AgentEvent
}

type AgentHarnessHandler func(context.Context, AgentHarnessEvent) (*AgentHarnessHookResult, error)

type AgentHarnessHookResult struct {
	Messages    []llm.Message
	HasMessages bool

	SystemPrompt    string
	HasSystemPrompt bool

	StreamOptions *AgentHarnessStreamOptionsPatch

	Payload    any
	HasPayload bool

	Block    bool
	HasBlock bool
	Reason   string

	Content      []llm.ContentPart
	HasContent   bool
	Details      any
	HasDetails   bool
	IsError      bool
	HasIsError   bool
	Terminate    bool
	HasTerminate bool

	Cancel                 bool
	Compaction             CompactionResult
	HasCompaction          bool
	BranchSummary          BranchSummaryResult
	HasBranchSummary       bool
	CustomInstructions     string
	HasCustomInstructions  bool
	ReplaceInstructions    bool
	HasReplaceInstructions bool
}

type AgentHarnessOptions struct {
	Env                 any
	ToolContext         AgentHarnessToolContextSource
	Session             *Session
	Models              SimpleCompletionRuntime
	Model               llm.Model
	ThinkingLevel       string
	SystemPrompt        string
	BuildSystemPrompt   func(context.Context, SystemPromptContext) (string, error)
	GetAPIKeyAndHeaders func(context.Context, llm.Model) (AgentHarnessAuth, bool, error)
	Retry               llm.RetryPolicy
	Resources           AgentHarnessResources
	Tools               []core.AgentTool
	HarnessTools        []AgentHarnessTool
	ActiveToolNames     []string
	StreamOptions       AgentHarnessStreamOptions
	SteeringMode        string
	FollowUpMode        string
	ToolExecution       string
}

type AgentHarnessResources struct {
	Skills          []Skill
	PromptTemplates []PromptTemplate
}

type AgentHarnessStreamOptions struct {
	Transport       string
	TimeoutMillis   int
	MaxRetries      int
	MaxRetryDelayMs int
	Headers         map[string]string
	Metadata        map[string]any
	CacheRetention  string
}

type AgentHarnessStreamOptionsPatch struct {
	Transport       Optional[string]
	TimeoutMillis   Optional[int]
	MaxRetries      Optional[int]
	MaxRetryDelayMs Optional[int]
	CacheRetention  Optional[string]
	Headers         map[string]*string
	ClearHeaders    bool
	Metadata        map[string]Optional[any]
	ClearMetadata   bool
}

type AgentHarnessSubscriber func(context.Context, AgentHarnessEvent) error

type AgentHarnessTool struct {
	Name        string
	Label       string
	Description string
	Parameters  llm.Schema

	PrepareArguments func(args any) (map[string]any, error)
	Execute          func(ctx context.Context, toolCallID string, params map[string]any, onUpdate core.AgentToolUpdateCallback, toolContext any) (core.AgentToolResult, error)
	ExecutionMode    string
}

type AgentHarnessToolContextSource func(context.Context) (any, error)

type BashExecutionTextOptions struct {
	Output         string
	ExitCode       *int
	Cancelled      bool
	Truncated      bool
	FullOutputPath string
}

type BranchPreparation struct {
	Messages    []llm.Message
	FileOps     FileOps
	TotalTokens int
}

type BranchSummaryError struct {
	Code string
	Err  error
}

type BranchSummaryOptions struct {
	APIKey              string
	CustomInstructions  string
	ReplaceInstructions bool
	ReserveTokens       int
	Runtime             SimpleCompletionRuntime
	RequestOptions      llm.ModelsStreamOptions
	Retry               llm.RetryPolicy
	RetryCallbacks      llm.RetryCallbacks
}

type BranchSummaryResult struct {
	Summary       string
	ReadFiles     []string
	ModifiedFiles []string
	Usage         *llm.Usage
}

type CapturedShellResult = harnessenv.CapturedShellResult

type CollectEntriesResult struct {
	Entries          []Entry
	CommonAncestorID *string
}

type CompactOptions struct {
	APIKey             string
	ThinkingLevel      string
	CustomInstructions string
	Runtime            SimpleCompletionRuntime
	RequestOptions     llm.ModelsStreamOptions
	Retry              llm.RetryPolicy
	RetryCallbacks     llm.RetryCallbacks
}

type CompactionError struct {
	Code string
	Err  error
}

type CompactionPreparation struct {
	FirstKeptEntryID    string             `json:"firstKeptEntryId"`
	MessagesToSummarize []llm.Message      `json:"messagesToSummarize"`
	TurnPrefixMessages  []llm.Message      `json:"turnPrefixMessages"`
	RetainedTail        []llm.Message      `json:"retainedTail"`
	IsSplitTurn         bool               `json:"isSplitTurn"`
	TokensBefore        int                `json:"tokensBefore"`
	PreviousSummary     string             `json:"previousSummary,omitempty"`
	FileOps             FileOps            `json:"fileOps"`
	Settings            CompactionSettings `json:"settings"`
}

type CompactionResult struct {
	Summary              string         `json:"summary"`
	FirstKeptEntryID     string         `json:"firstKeptEntryId"`
	TokensBefore         int            `json:"tokensBefore"`
	RetainedTail         []llm.Message  `json:"retainedTail,omitempty"`
	EstimatedTokensAfter int            `json:"estimatedTokensAfter,omitempty"`
	Usage                *llm.Usage     `json:"usage,omitempty"`
	Details              map[string]any `json:"details,omitempty"`
}

type CompactionSettings struct {
	Enabled          bool `json:"enabled"`
	ReserveTokens    int  `json:"reserveTokens"`
	KeepRecentTokens int  `json:"keepRecentTokens"`
}

type ContextEntryTransform func(entries []Entry) []Entry

type ContextTokenEstimate = llm.ContextUsageEstimate

type CreateDirOptions = harnessenv.CreateDirOptions

type CustomEntryContextMessageProjector func(entry Entry, index int, entries []Entry) []llm.Message

type CutPoint struct {
	FirstKeptEntryIndex int  `json:"firstKeptEntryIndex"`
	TurnStartIndex      int  `json:"turnStartIndex"`
	IsSplitTurn         bool `json:"isSplitTurn"`
}

type Entry struct {
	Type             string        `json:"type"`
	ID               string        `json:"id,omitempty"`
	ParentID         *string       `json:"parentId,omitempty"`
	Timestamp        string        `json:"timestamp,omitempty"`
	Message          llm.Message   `json:"message,omitempty"`
	TargetID         *string       `json:"targetId,omitempty"`
	Label            *string       `json:"label,omitempty"`
	ThinkingLevel    string        `json:"thinkingLevel,omitempty"`
	Provider         string        `json:"provider,omitempty"`
	ModelID          string        `json:"modelId,omitempty"`
	ActiveToolNames  []string      `json:"activeToolNames,omitempty"`
	Summary          string        `json:"summary,omitempty"`
	FirstKeptEntryID string        `json:"firstKeptEntryId,omitempty"`
	TokensBefore     int           `json:"tokensBefore,omitempty"`
	RetainedTail     []llm.Message `json:"retainedTail,omitempty"`
	FromID           string        `json:"fromId,omitempty"`
	CustomType       string        `json:"customType,omitempty"`
	Data             any           `json:"data,omitempty"`
	Content          any           `json:"content,omitempty"`
	Display          bool          `json:"display,omitempty"`
	Details          any           `json:"details,omitempty"`
	FromHook         bool          `json:"fromHook,omitempty"`
	Usage            *llm.Usage    `json:"usage,omitempty"`
	Name             string        `json:"name,omitempty"`
}

type ExecOptions = harnessenv.ExecOptions

type ExecResult = harnessenv.ExecResult

type ExecutionEnv = harnessenv.ExecutionEnv

type ExecutionError = harnessenv.ExecutionError

type FileError = harnessenv.FileError

type FileInfo = harnessenv.FileInfo

type FileOps struct {
	Read    map[string]bool `json:"read"`
	Written map[string]bool `json:"written"`
	Edited  map[string]bool `json:"edited"`
}

type FileSystem = harnessenv.FileSystem

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

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

type JsonlSessionRepo struct {
	SessionsRoot string
}

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

type LoadSkillsOptions struct {
	IncludeRootMarkdownFiles bool
	RespectGitignore         bool
}

type LocalExecutionEnv = harnessenv.LocalExecutionEnv

type LocalExecutionOption = harnessenv.LocalExecutionOption

type NavigateTreeOptions struct {
	Summarize           bool
	CustomInstructions  string
	ReplaceInstructions bool
	Label               string
}

type NavigateTreeResult struct {
	Cancelled    bool
	EditorText   string
	SummaryEntry *Entry
}

type NowMillisFunc = sessionid.NowMillisFunc

type Optional[T any] struct {
	Set   bool
	Value T
}

type PromptTemplate struct {
	Name        string
	Description string
	Content     string
}

type PromptTemplateDiagnostic struct {
	Type    string
	Code    string
	Message string
	Path    string
	Source  any
}

type PromptTemplateResult struct {
	PromptTemplates []PromptTemplate
	Diagnostics     []PromptTemplateDiagnostic
}

type RandomBytesFunc = sessionid.RandomBytesFunc

type RemoveOptions = harnessenv.RemoveOptions

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

type SessionContext struct {
	Messages        []llm.Message
	ThinkingLevel   string
	Model           *SessionModel
	ActiveToolNames []string

	// ModelProvider and ModelID are kept as compatibility mirrors. Model is the
	// canonical representation for new callers.
	ModelProvider string
	ModelID       string
}

type SessionContextBuildOptions struct {
	EntryTransforms []ContextEntryTransform
	EntryProjectors map[string]CustomEntryContextMessageProjector
}

type SessionCreateOptions struct {
	ParentSessionPath string
	Metadata          map[string]any
}

type SessionEntryCursorOptions struct {
	AfterEntrySeq int
	Limit         int
}

type SessionEntryOptions struct {
	Details      any
	FromHook     bool
	Usage        *llm.Usage
	RetainedTail []llm.Message
}

type SessionError struct {
	Code string
	Err  error
}

type SessionForkOptions struct {
	ParentSessionPath string
	Metadata          map[string]any
}

type SessionMetadata struct {
	ID                string         `json:"id"`
	CreatedAt         string         `json:"createdAt"`
	CWD               string         `json:"cwd,omitempty"`
	Path              string         `json:"path,omitempty"`
	ParentSessionPath string         `json:"parentSessionPath,omitempty"`
	Metadata          map[string]any `json:"metadata,omitempty"`
}

type SessionModel struct {
	Provider string
	ModelID  string
}

type SessionStats struct {
	MessageCount   int     `json:"messageCount"`
	CachedTokens   int     `json:"cachedTokens"`
	UncachedTokens int     `json:"uncachedTokens"`
	TotalTokens    int     `json:"totalTokens"`
	CostTotal      float64 `json:"costTotal"`
}

type SessionStorage interface {
	Metadata() SessionMetadata
	LeafID() (string, bool, error)
	SetLeafID(*string) error
	CreateEntryID() string
	AppendEntry(Entry) error
	Entry(string) (Entry, bool)
	FindEntries(string) []Entry
	Label(string) (string, bool)
	SessionName() (string, bool)
	SessionStats() SessionStats
	PathToRoot(*string) ([]Entry, error)
	PathToRootOrCompaction(*string) ([]Entry, error)
	Entries(...SessionEntryCursorOptions) []Entry
}

type Shell = harnessenv.Shell

type SimpleCompletionRuntime interface {
	CompleteSimple(context.Context, llm.Model, llm.Context, llm.ModelsStreamOptions) (llm.Message, error)
}

type Skill struct {
	Name                   string
	Description            string
	Content                string
	FilePath               string
	SourceInfo             any
	DisableModelInvocation bool
}

type SkillDiagnostic struct {
	Type    string
	Code    string
	Message string
	Path    string
	Source  any
}

type SkillResult struct {
	Skills      []Skill
	Diagnostics []SkillDiagnostic
}

type SourcedPromptTemplate struct {
	PromptTemplate PromptTemplate
	Source         any
}

type SourcedPromptTemplateResult struct {
	PromptTemplates []SourcedPromptTemplate
	Diagnostics     []PromptTemplateDiagnostic
}

type SourcedSkill struct {
	Skill  Skill
	Source any
}

type SourcedSkillResult struct {
	Skills      []SourcedSkill
	Diagnostics []SkillDiagnostic
}

type SummaryWithUsage struct {
	Text  string
	Usage llm.Usage
}

type SystemPromptContext struct {
	Env           any
	Session       *Session
	Model         llm.Model
	ThinkingLevel string
	ActiveTools   []core.AgentTool
	Resources     AgentHarnessResources
}

type TempFileOptions = harnessenv.TempFileOptions

type TruncationOptions = utils.TruncationOptions

type TruncationResult = utils.TruncationResult