config

type ConfigFile struct {
	Configs []SimConfig `json:"configs" yaml:"configs"`
}

type Needle struct {
	Text       string `json:"text"`
	Source     string `json:"source,omitempty"`
	Difficulty string `json:"difficulty,omitempty"`
}

type QueriesFile struct {
	Queries []Query `json:"queries" yaml:"queries"`
}

type Query struct {
	ID           string   `json:"id" yaml:"id"`
	Text         string   `json:"text" yaml:"text"`
	RelevantDocs []string `json:"relevant_docs" yaml:"relevant_docs"`
	Notes        string   `json:"notes,omitempty" yaml:"notes,omitempty"`
	Needles      []Needle `json:"needles,omitempty" yaml:"needles,omitempty"`
}

type SimConfig struct {
	Name      string `json:"name" yaml:"name"`
	TopK      int    `json:"top_k" yaml:"top_k"`
	ChunkSize int    `json:"chunk_size,omitempty" yaml:"chunk_size,omitempty"`
	Overlap   int    `json:"overlap,omitempty" yaml:"overlap,omitempty"`
}