locomo

type Config struct {
	DataPath     string
	Limit        int // Max questions (0 = all)
	JudgeCfg     *common.JudgeConfig
	GraphBoost   bool
	EmbeddingCfg *embedding.Config // If set, enables vector search
}

type Conversation struct {
	SampleID     string           `json:"sample_id"`
	Conversation ConversationData `json:"conversation"`
	QA           []QA             `json:"qa"`
	Observation  json.RawMessage  `json:"observation"` // Complex nested structure, parsed manually
}

type ConversationData struct {
	SpeakerA string            `json:"speaker_a"`
	SpeakerB string            `json:"speaker_b"`
	Sessions map[string][]Turn // Parsed from session_N keys
	Dates    map[string]string // Parsed from session_N_date_time keys
}

type QA struct {
	Question string          `json:"question"`
	Answer   json.RawMessage `json:"answer"` // Can be string or number
	Evidence []string        `json:"evidence"`
	Category int             `json:"category"`
}

type Turn struct {
	Speaker string `json:"speaker"`
	DiaID   string `json:"dia_id"`
	Text    string `json:"text"`
}