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

	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
	Session             *Session
	Model               llm.Model
	ThinkingLevel       string
	SystemPrompt        string
	BuildSystemPrompt   func(context.Context, SystemPromptContext) (string, error)
	GetAPIKeyAndHeaders func(context.Context, llm.Model) (AgentHarnessAuth, bool, error)
	Resources           AgentHarnessResources
	Tools               []core.AgentTool
	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 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
}

type BranchSummaryResult struct {
	Summary       string
	ReadFiles     []string
	ModifiedFiles []string
}

type CapturedShellResult = harnessenv.CapturedShellResult

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

type CompactOptions struct {
	APIKey             string
	ThinkingLevel      string
	CustomInstructions string
}

type CompactionError struct {
	Code string
	Err  error
}

type CompactionPreparation struct {
	FirstKeptEntryID    string             `json:"firstKeptEntryId"`
	MessagesToSummarize []llm.Message      `json:"messagesToSummarize"`
	TurnPrefixMessages  []llm.Message      `json:"turnPrefixMessages"`
	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"`
	Details          map[string]any `json:"details,omitempty"`
}

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

type ContextTokenEstimate struct {
	Tokens         int  `json:"tokens"`
	UsageTokens    int  `json:"usageTokens"`
	EstimateTokens int  `json:"estimateTokens"`
	LastUsageIndex *int `json:"lastUsageIndex"`
}

type CreateDirOptions = harnessenv.CreateDirOptions

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"`
	Summary          string      `json:"summary,omitempty"`
	FirstKeptEntryID string      `json:"firstKeptEntryId,omitempty"`
	TokensBefore     int         `json:"tokensBefore,omitempty"`
	FromID           string      `json:"fromId,omitempty"`
	CustomType       string      `json:"customType,omitempty"`
	Content          any         `json:"content,omitempty"`
	Display          bool        `json:"display,omitempty"`
	Details          any         `json:"details,omitempty"`
	FromHook         bool        `json:"fromHook,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
	ModelProvider string
	ModelID       string
}

type SessionEntryOptions struct {
	Details  any
	FromHook bool
}

type SessionError struct {
	Code string
	Err  error
}

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"`
}

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)
	PathToRoot(*string) ([]Entry, error)
	Entries() []Entry
}

type Shell = harnessenv.Shell

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 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