adapters

type ArenaOutputAdapter struct{}

type MediaSource struct {
	MIMEType string
	Data     string
	URI      string
	Path     string
	Size     int64
	Width    int
	Height   int
	Duration int64 // milliseconds
}

type ProviderSpec struct {
	Type  string `json:"type" yaml:"type"`
	Model string `json:"model" yaml:"model"`
	ID    string `json:"id" yaml:"id"`
}

type RecordedEval struct {
	ID      string                 `json:"id,omitempty" yaml:"id,omitempty"`
	Type    string                 `json:"type" yaml:"type"`
	Kind    string                 `json:"kind,omitempty" yaml:"kind,omitempty"`
	Score   *float64               `json:"score,omitempty" yaml:"score,omitempty"`
	Passed  *bool                  `json:"passed,omitempty" yaml:"passed,omitempty"`
	Message string                 `json:"message,omitempty" yaml:"message,omitempty"`
	Params  map[string]interface{} `json:"params,omitempty" yaml:"params,omitempty"`
	Details map[string]interface{} `json:"details,omitempty" yaml:"details,omitempty"`
}

type RecordedTransition struct {
	From         string `json:"from" yaml:"from"`
	To           string `json:"to" yaml:"to"`
	Event        string `json:"event,omitempty" yaml:"event,omitempty"`
	PromptTask   string `json:"prompt_task,omitempty" yaml:"prompt_task,omitempty"`
	MessageIndex int    `json:"message_index" yaml:"message_index"`
}

type RecordingAdapter interface {
	// CanHandle returns true if this adapter supports the given source/type hint.
	// The source could be a file path, glob pattern, database query, etc.
	// typeHint is an optional explicit format indicator from the eval config.
	CanHandle(source, typeHint string) bool

	// Enumerate expands a source into individual recording references.
	// For file-based adapters, this expands glob patterns to matching files.
	// For database adapters, this could execute a query and return record IDs.
	// Returns a single-element slice for non-expandable sources.
	Enumerate(source string) ([]RecordingReference, error)

	// Load converts a recording to Arena message format.
	// The reference should have been obtained from Enumerate.
	// Returns the messages, metadata, and any error encountered.
	Load(ref RecordingReference) ([]types.Message, *RecordingMetadata, error)
}

type RecordingMetadata struct {
	// JudgeTargets maps judge names to provider specifications.
	// Used by LLM judge assertions to determine which provider to use.
	JudgeTargets map[string]ProviderSpec `json:"judge_targets,omitempty" yaml:"judge_targets,omitempty"`

	// ProviderInfo contains information about the original provider(s)
	// that generated the recorded conversation.
	ProviderInfo map[string]interface{} `json:"provider_info,omitempty" yaml:"provider_info,omitempty"`

	// Tags are optional labels for categorizing/filtering recordings.
	Tags []string `json:"tags,omitempty" yaml:"tags,omitempty"`

	// Timestamps contains the timestamp for each turn in the conversation.
	// The length should match the number of messages.
	Timestamps []time.Time `json:"timestamps,omitempty" yaml:"timestamps,omitempty"`

	// SessionID is the unique identifier for the recorded session.
	SessionID string `json:"session_id,omitempty" yaml:"session_id,omitempty"`

	// Duration is the total duration of the conversation.
	Duration time.Duration `json:"duration,omitempty" yaml:"duration,omitempty"`

	// Extras holds any additional metadata from the recording.
	Extras map[string]interface{} `json:"extras,omitempty" yaml:"extras,omitempty"`

	// ConversationAssertions are the conversation-level judged results the
	// recording carries (kind "assertion" or "guardrail"). Nil when the source
	// ran none.
	ConversationAssertions []RecordedEval `json:"conversation_assertions,omitempty" yaml:"conversation_assertions,omitempty"` //nolint:lll

	// TurnAssertions are per-turn judged results keyed by the index of the
	// message they were evaluated against (an assistant message, for arena
	// run output). Nil when the source carries none.
	TurnAssertions map[int][]RecordedEval `json:"turn_assertions,omitempty" yaml:"turn_assertions,omitempty"`

	// EvalResults are the pack-level eval observations the session produced:
	// measurements (kind "eval") with a score and no verdict, and any
	// guardrail outcomes recorded at session level. Conversation-level. Nil
	// when the source carries none.
	EvalResults []RecordedEval `json:"eval_results,omitempty" yaml:"eval_results,omitempty"`

	// WorkflowTransitions are the recorded workflow state changes, in order.
	// Nil when the session ran no workflow or the source does not record them.
	WorkflowTransitions []RecordedTransition `json:"workflow_transitions,omitempty" yaml:"workflow_transitions,omitempty"`
}

type RecordingReference struct {
	// ID is a unique identifier for this recording reference.
	// For file-based adapters, this is typically the file path.
	// For database adapters, this could be a record ID.
	ID string `json:"id" yaml:"id"`

	// Source is the original source pattern/query that produced this reference.
	// For file-based adapters, this is the original glob pattern or path.
	Source string `json:"source" yaml:"source"`

	// TypeHint is an optional format indicator (e.g., "session", "arena_output", "transcript").
	TypeHint string `json:"type_hint,omitempty" yaml:"type_hint,omitempty"`

	// Metadata contains adapter-specific metadata about this reference.
	// This can include file size, modification time, database metadata, etc.
	Metadata map[string]interface{} `json:"metadata,omitempty" yaml:"metadata,omitempty"`
}

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

type SessionRecordingAdapter struct{}

type TranscriptAdapter struct{}

type TranscriptContentPart struct {
	Type  string               `yaml:"type"`
	Text  *string              `yaml:"text,omitempty"`
	Media *TranscriptMediaPart `yaml:"media,omitempty"`
}

type TranscriptFile struct {
	Metadata TranscriptMetadata  `yaml:"metadata"`
	Messages []TranscriptMessage `yaml:"messages"`
}

type TranscriptMediaPart struct {
	MIMEType string `yaml:"mime_type,omitempty"`
	Data     string `yaml:"data,omitempty"` // Base64 encoded
	URI      string `yaml:"uri,omitempty"`  // URL
	Path     string `yaml:"path,omitempty"` // File path
	Size     int64  `yaml:"size,omitempty"`
	Width    int    `yaml:"width,omitempty"`
	Height   int    `yaml:"height,omitempty"`
	Duration int64  `yaml:"duration,omitempty"` // milliseconds
}

type TranscriptMessage struct {
	Role       string                  `yaml:"role"`
	Content    string                  `yaml:"content"`
	Name       string                  `yaml:"name,omitempty"`
	Timestamp  string                  `yaml:"timestamp,omitempty"`
	Parts      []TranscriptContentPart `yaml:"parts,omitempty"`
	ToolCalls  []TranscriptToolCall    `yaml:"tool_calls,omitempty"`
	ToolCallID string                  `yaml:"tool_call_id,omitempty"`
}

type TranscriptMetadata struct {
	SessionID    string                            `yaml:"session_id,omitempty"`
	Provider     string                            `yaml:"provider,omitempty"`
	Model        string                            `yaml:"model,omitempty"`
	Tags         []string                          `yaml:"tags,omitempty"`
	JudgeTargets map[string]TranscriptProviderSpec `yaml:"judge_targets,omitempty"`
}

type TranscriptProviderSpec struct {
	Type  string `yaml:"type"`
	Model string `yaml:"model"`
	ID    string `yaml:"id"`
}

type TranscriptToolCall struct {
	ID       string                     `yaml:"id"`
	Type     string                     `yaml:"type"`
	Function TranscriptToolCallFunction `yaml:"function"`
}

type TranscriptToolCallFunction struct {
	Name      string `yaml:"name"`
	Arguments string `yaml:"arguments"`
}