agent

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

type AgentConfig struct {
	// AuthMode is one of "logged-in", "token", or "byok".
	AuthMode string

	// GitHubTokenFile is the path to a file containing a GitHub token (token mode).
	GitHubTokenFile string

	// ModelEndpoint is the Azure AI Foundry endpoint URL (byok mode).
	ModelEndpoint string

	// ModelDeployment is the model deployment name (byok mode).
	ModelDeployment string

	// AzureCredential is used to acquire Entra tokens for BYOK sessions.
	AzureCredential azcore.TokenCredential

	// Model overrides the default model for Copilot sessions.
	Model string

	// MaxRounds is the maximum number of tool-call rounds per session.
	MaxRounds int

	// Verbosity is the log verbosity level from the CLI. When >= 5,
	// the Copilot CLI subprocess is started with --log-level=debug and
	// all session events are traced.
	Verbosity int
}

type AnalyzeOptions struct {
	// Manifest is the raw manifest.json content.
	Manifest []byte

	// TestName is the name of the failed test (used for rendering).
	TestName string

	// TestError is the content of test_logs/error.log, or empty.
	TestError string

	// TestOutput is the content of test_logs/output.log, or empty.
	TestOutput string

	// SiblingTests is the content of sibling_tests.json, or empty.
	SiblingTests string

	// DataDir is the root of the structured data directory.
	DataDir string

	// WorktreePaths maps repository names to local filesystem paths.
	WorktreePaths map[string]string

	// KustoCluster is the Kusto cluster URI for hydration share links.
	KustoCluster string

	// KustoDatabase is the Kusto database name.
	KustoDatabase string

	// MaxValidationRounds is the maximum number of parse/validate correction
	// rounds per validate-draft cycle. Zero defaults to 10.
	MaxValidationRounds int

	// ReviewRounds is the number of review passes. Zero defaults to 3.
	ReviewRounds int

	// NodeConsoleLogs maps console log filenames to their contents.
	// Used for validating and hydrating node_console_log proof items.
	NodeConsoleLogs map[string]string

	// NodeConsoleLogURLs maps console log filenames to artifact download URLs.
	// Used for populating ArtifactURL on hydrated node_console_log proof items.
	NodeConsoleLogURLs map[string]string
}

type AnalyzeResult struct {
	// HydratedChain is the fully validated and hydrated causal chain.
	HydratedChain *HydratedChain

	// DraftChain is the last validated draft before the final hydration.
	DraftChain *DraftChain
}

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

type ChainLink struct {
	Question string      `json:"question"`
	Answer   string      `json:"answer"`
	Notes    string      `json:"notes,omitempty"`
	Proof    []ProofItem `json:"proof"`
}

type ClaudeConfig struct {
	// APIKeyFile is the path to a file containing the Anthropic API key.
	// If empty, the ANTHROPIC_API_KEY environment variable is used.
	// Only used when Backend is "api" (the default).
	APIKeyFile string

	// Model is the Anthropic model to use (e.g. "claude-sonnet-4-20250514").
	// Defaults to DefaultClaudeModel.
	Model string

	// Backend selects the API backend: "api" for the direct Anthropic
	// API (default) or "vertex" for Google Vertex AI.
	Backend string

	// VertexProject is the GCP project ID. Required when Backend is "vertex".
	VertexProject string

	// VertexRegion is the GCP region (e.g. "us-east5"). Required when Backend is "vertex".
	VertexRegion string
}

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

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

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

type DiscoveryItem struct {
	// Label is a human-readable description of what this discovery item establishes.
	Label string `json:"label"`

	// KQL is an agent-authored Kusto query whose results establish provenance
	// for constants used in proof queries.
	KQL string `json:"kql"`
}

type DraftChain struct {
	RootCause   string          `json:"root_cause"`
	Summary     string          `json:"summary"`
	Notes       string          `json:"notes,omitempty"`
	Discovery   []DiscoveryItem `json:"discovery,omitempty"`
	Chain       []ChainLink     `json:"chain"`
	Suggestions []string        `json:"suggestions,omitempty"`
}

type HydratedChain struct {
	RootCause   string              `json:"root_cause"`
	Summary     string              `json:"summary"`
	Notes       string              `json:"notes,omitempty"`
	Discovery   []HydratedDiscovery `json:"discovery,omitempty"`
	Chain       []HydratedLink      `json:"chain"`
	Suggestions []string            `json:"suggestions,omitempty"`
}

type HydratedDiscovery struct {
	Directory string         `json:"directory,omitempty"`
	Label     string         `json:"label"`
	KQL       string         `json:"kql"`
	ShareURI  string         `json:"share_uri"`
	Table     *tabular.Table `json:"table,omitempty"`
}

type HydratedLink struct {
	Question string              `json:"question"`
	Answer   string              `json:"answer"`
	Notes    string              `json:"notes,omitempty"`
	Proof    []HydratedProofItem `json:"proof"`
}

type HydratedProofItem struct {
	ProofItem

	// Kusto proof fields populated during hydration
	ShareURI string         `json:"share_uri,omitempty"`
	Table    *tabular.Table `json:"table,omitempty"`

	// Code proof field populated during hydration by reading the worktree
	CodeExcerpt string `json:"code_excerpt,omitempty"`

	// Log proof field populated during hydration by extracting lines from test logs
	// or node console logs.
	LogExcerpt string `json:"log_excerpt,omitempty"`

	// ArtifactURL is the download link for node console log proofs, populated
	// during hydration from the manifest's node_console_logs entries.
	ArtifactURL string `json:"artifact_url,omitempty"`
}

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

type KustoClient interface {
	// Query executes a KQL query and returns the result as a tabular.Table.
	Query(ctx context.Context, kql string) (*tabular.Table, error)
}

type LLMProvider interface {
	// CreateProviderSession creates a new LLM session configured for
	// analysis. The provider translates the provider-neutral
	// ProviderSessionConfig into its native format (e.g. copilot
	// SessionConfig sections, Anthropic MessageNewParams).
	CreateProviderSession(ctx context.Context, logger logr.Logger, cfg ProviderSessionConfig) (LLMSession, error)

	// Stop shuts down the provider and releases all resources.
	Stop() error
}

type LLMSession interface {
	// SendAndWait sends a user prompt and blocks until the model finishes
	// responding, including any tool-use rounds. Returns the final
	// assistant text content.
	SendAndWait(ctx context.Context, prompt string) (string, error)

	// SaveConversation writes the conversation history to a JSON file at
	// the given path. This is best-effort: implementations should log
	// errors rather than return them.
	SaveConversation(path string)

	// SessionID returns a unique identifier for this session.
	SessionID() string

	// Disconnect releases in-memory session resources while preserving
	// on-disk state for potential later resumption.
	Disconnect() error

	// Delete permanently removes all session data.
	Delete(ctx context.Context) error
}

type ProofItem struct {
	Type string `json:"type"` // "kusto", "code", "log"

	// Kusto proof fields
	KQL  string `json:"kql,omitempty"`
	Note string `json:"note,omitempty"`

	// Code proof fields (File is also used by node_console_log log proofs to
	// specify which console log file to reference)
	Repo  string `json:"repo,omitempty"`
	File  string `json:"file,omitempty"`
	Lines [2]int `json:"lines,omitempty"` // 1-indexed, inclusive; used by code and log proofs

	// Log proof fields
	Source string `json:"source,omitempty"` // "error", "output", or "node_console_log"
}

type ProviderSessionConfig struct {
	// IdentityPrompt carries the identity/role instructions that tell
	// the model who it is (e.g. "You are a senior SRE …").
	IdentityPrompt string

	// TonePrompt carries the tone/style instructions that tell the
	// model how to respond (e.g. "Be precise, evidence-driven …").
	TonePrompt string

	// SystemPrompt carries domain-specific content (system.md, references,
	// exemplars) built by BuildDomainPrompt.
	SystemPrompt string

	// Tools are provider-neutral tool definitions. Each provider converts
	// them to its native tool format (e.g. copilot.Tool, Anthropic tool
	// params).
	Tools []ToolDefinition

	// WorkingDirectory is the workspace root. Providers that support
	// file-access tools scope operations to this directory.
	WorkingDirectory string

	// Model overrides the provider's default model for this session.
	// When empty, the provider uses its configured default.
	Model string
}

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

type SessionConfig struct {
	// WorkingDirectory is the workspace root for the Copilot session.
	// Tool operations (read_file, grep, bash, glob) are relative to this directory.
	WorkingDirectory string

	// SystemMessage configures system prompt customization.
	SystemMessage *copilot.SystemMessageConfig

	// Tools are custom tools (e.g. kusto_query) registered on this session.
	Tools []copilot.Tool

	// Model overrides the default model for this session.
	Model string
}

type ToolDefinition struct {
	// Name is the tool's unique identifier (e.g. "kusto_query").
	Name string

	// Description explains what the tool does. This is shown to the model
	// to help it decide when and how to use the tool.
	Description string

	// ParamSchema is the JSON Schema for the tool's input parameters,
	// serialized as a JSON object. Example:
	//
	//   {"type":"object","properties":{"kql":{"type":"string",
	//     "description":"The KQL query to execute."}},"required":["kql"]}
	ParamSchema json.RawMessage

	// Handler executes the tool with the given JSON-encoded parameters
	// and returns the text result. The context carries cancellation and
	// tracing information from the provider's session.
	Handler func(ctx context.Context, params json.RawMessage) (string, error)
}

type ValidationContext struct {
	// ValidRepos is the set of repository names that have source code worktrees available.
	ValidRepos map[string]bool
	// WorktreePaths maps repository names to local filesystem paths for their git worktrees.
	WorktreePaths map[string]string
	// DataDir is the root of the gathered data directory (for discovery path validation).
	DataDir string
	// TestError is the contents of the test error.log file.
	TestError string
	// TestOutput is the contents of the test output.log file.
	TestOutput string
	// NodeConsoleLogs maps console log filenames to their contents.
	// Used for validating node_console_log proof items.
	NodeConsoleLogs map[string]string
}

type ValidationProblem struct {
	// Category is a machine-readable identifier for the type of problem.
	Category string `json:"category"`
	// Chain is the chain link index where the problem was found, or -1 for top-level issues.
	Chain int `json:"chain"`
	// Proof is the proof item index (0-based) where the problem was found, or -1 if N/A.
	Proof int `json:"proof"`
	// Detail is the human-readable description of the problem.
	Detail string `json:"detail"`
}

type ValidationResult struct {
	// Problems is the list of validation issues found.
	Problems []ValidationProblem
	// Feedback is the human-readable text suitable for sending to the agent as
	// a correction prompt. It is empty when there are no problems.
	Feedback string
}