agent

type Bucket struct {
	Hour    time.Time `json:"hour"`
	Success int       `json:"success"`
	Error   int       `json:"error"`
}

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

type Repository interface {
	InsertRun(ctx context.Context, r *Run) error
	GetByAgentRunID(ctx context.Context, agentID, runID string) (*Run, error)
	ListRuns(ctx context.Context, agentID string, since time.Time, limit int) ([]*Run, error)
	BucketRuns(ctx context.Context, agentID string, since time.Time) ([]Bucket, error)
	Stats(ctx context.Context, agentID string, since time.Time) (*Stats, error)
}

type Run struct {
	ID         string     `json:"id"`
	AgentID    string     `json:"agent_id"`
	RunID      string     `json:"run_id"`
	StartedAt  time.Time  `json:"started_at"`
	EndedAt    *time.Time `json:"ended_at,omitempty"`
	DurationMs *int       `json:"duration_ms,omitempty"`
	Status     string     `json:"status"`
	Model      string     `json:"model,omitempty"`
	TokensIn   int        `json:"tokens_in"`
	TokensOut  int        `json:"tokens_out"`
	Error      string     `json:"error,omitempty"`
	CreatedAt  time.Time  `json:"created_at"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
}

type RunInput struct {
	AgentMRN  string          `json:"agent_mrn"`
	RunID     string          `json:"run_id"`
	StartedAt time.Time       `json:"started_at"`
	EndedAt   *time.Time      `json:"ended_at,omitempty"`
	Status    string          `json:"status"`
	Model     string          `json:"model,omitempty"`
	TokensIn  int             `json:"tokens_in"`
	TokensOut int             `json:"tokens_out"`
	Error     string          `json:"error,omitempty"`
	ToolCalls []ToolCallInput `json:"tool_calls,omitempty"`
	// ObservedAssets are MRNs the agent touched at runtime that aren't already
	// surfaced through ToolCalls' target_mrn — typically extracted from tool
	// outputs (e.g. catalog-traversal tools that return MRNs). The server
	// emits an AGENT_LOOKUP observed edge for each.
	ObservedAssets []string `json:"observed_assets,omitempty"`
}

type Service interface {
	RecordRun(ctx context.Context, input RunInput) (*Run, error)
	ListRuns(ctx context.Context, assetID string, period time.Duration, limit int) ([]*Run, error)
	BucketRuns(ctx context.Context, assetID string, period time.Duration) ([]Bucket, error)
	Stats(ctx context.Context, assetID string, period time.Duration) (*Stats, error)
}

type Stats struct {
	RunCount      int     `json:"run_count"`
	SuccessRate   float64 `json:"success_rate"`
	MedianLatency int     `json:"median_latency_ms"`
	P95Latency    int     `json:"p95_latency_ms"`
	TokensIn      int     `json:"tokens_in"`
	TokensOut     int     `json:"tokens_out"`
}

type ToolCall struct {
	Ordinal    int       `json:"ordinal"`
	ToolName   string    `json:"tool_name"`
	TargetMRN  string    `json:"target_mrn,omitempty"`
	StartedAt  time.Time `json:"started_at"`
	DurationMs *int      `json:"duration_ms,omitempty"`
	Status     string    `json:"status"`
}

type ToolCallInput struct {
	ToolName   string    `json:"tool_name"`
	TargetMRN  string    `json:"target_mrn,omitempty"`
	StartedAt  time.Time `json:"started_at"`
	DurationMs *int      `json:"duration_ms,omitempty"`
	Status     string    `json:"status"`
}