training

type AlpacaExample struct {
	Instruction string `json:"instruction"`
	Input       string `json:"input"`
	Output      string `json:"output"`
}

type ExportOptions struct {
	// Format is one of "sharegpt", "openai", "alpaca".
	Format string

	// Output is the destination file path.
	Output string

	// All, when true, exports sessions from every directory scope.
	All bool

	// MinTurns is the minimum number of user+assistant exchanges required.
	MinTurns int

	// MinActions is the minimum number of TaskActions required.
	MinActions int

	// NoToolResults, when true, replaces tool-result messages with short
	// placeholders instead of including the raw content.
	NoToolResults bool

	// IncludeSystem includes system-prompt messages in the output.
	IncludeSystem bool

	// Session, when non-empty, exports only the session with this ID.
	Session string

	// StructuredTools, when true, preserves the OpenAI function-calling
	// schema: assistant messages retain their ToolCalls arrays and tool
	// results keep role:"tool" with tool_call_id. When false (default),
	// tool calls are flattened to text and tool results are converted to
	// user-role messages with a "[Tool Result]" prefix.
	StructuredTools bool

	// IncludeSubagents, when true, extracts single-task examples from
	// run_subagent and run_parallel_subagents tool calls within each
	// session and appends them to the output alongside the regular
	// conversation examples. Only meaningful for the "openai" format.
	IncludeSubagents bool

	// ExcludePaths is a list of absolute path prefixes. Sessions whose
	// WorkingDirectory starts with any of these paths are excluded.
	ExcludePaths []string
}

type ExportResult struct {
	SessionsScanned   int    `json:"sessions_scanned"`
	SessionsExported  int    `json:"sessions_exported"`
	ExamplesGenerated int    `json:"examples_generated"`
	SessionsFiltered  int    `json:"sessions_filtered"`
	OutputPath        string `json:"output_path"`
}

type FileChangeExample struct {
	Messages []OpenAIMessage           `json:"messages"`
	Metadata FileChangeExampleMetadata `json:"metadata"`
}

type FileChangeExampleMetadata struct {
	Source      string `json:"source"`
	Description string `json:"description"`
	Model       string `json:"model,omitempty"`
	File        string `json:"file"`
}

type FileChangeExportOptions struct {
	// Output is the destination file path (OpenAI JSONL).
	Output string

	// MaxSize is the maximum number of characters per file content (before
	// or after). Changes where either side exceeds this are skipped.
	// Defaults to 50000 when zero.
	MaxSize int

	// ExcludePaths is a list of absolute path prefixes. Working directories
	// starting with any of these paths are excluded from the export.
	ExcludePaths []string
}

type FileChangeExportResult struct {
	ChangesScanned  int    `json:"changes_scanned"`
	ChangesExported int    `json:"changes_exported"`
	ChangesFiltered int    `json:"changes_filtered"`
	OutputPath      string `json:"output_path"`
}

type OpenAIMessage struct {
	Role       string         `json:"role"`
	Content    string         `json:"content"`
	ToolCalls  []api.ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string         `json:"tool_call_id,omitempty"`
}

type OpenAITrainingExample struct {
	Messages []OpenAIMessage `json:"messages"`
}

type PIIRedactionConfig struct {
	// HomeDir is the user's home directory to redact (e.g. /Users/alan).
	// When set, all occurrences are replaced with the placeholder.
	HomeDir string

	// Username is the OS username to redact.
	Username string

	// Hostname is the machine hostname to redact.
	Hostname string
}

type ShareGPTConversation struct {
	ID       string            `json:"id"`
	Messages []ShareGPTMessage `json:"messages"`
	Metadata ShareGPTMetadata  `json:"metadata"`
}

type ShareGPTMessage struct {
	Role       string         `json:"role"` // "system", "user", "assistant"
	Content    string         `json:"content"`
	ToolCalls  []api.ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string         `json:"tool_call_id,omitempty"`
}

type ShareGPTMetadata struct {
	SessionID   string  `json:"session_id"`
	SessionName string  `json:"session_name"`
	Source      string  `json:"source"`
	Model       string  `json:"model,omitempty"`
	Provider    string  `json:"provider,omitempty"`
	TotalCost   float64 `json:"total_cost"`
	WorkingDir  string  `json:"working_directory"`
}