semantic

type Candidate struct {
	Source     Source    `json:"source"`
	Similarity float64   `json:"similarity"`
	CreatedAt  time.Time `json:"created_at"`
	ExpiresAt  time.Time `json:"expires_at"`
}

type Config struct {
	Mode             string            `json:"mode"`
	Family           string            `json:"family"`
	Domain           string            `json:"domain"`
	EvidenceRevision string            `json:"evidence_revision"`
	Bindings         map[string]string `json:"bindings,omitempty"`
	EmbeddingModel   string            `json:"embedding_model"`
	MinSimilarity    float64           `json:"min_similarity,omitempty"`
	TopK             int               `json:"top_k,omitempty"`
	MaxScan          int               `json:"max_scan,omitempty"`
	MaxEntries       int               `json:"max_entries,omitempty"`
	MaxInputBytes    int               `json:"max_input_bytes,omitempty"`
	TimeoutMS        int               `json:"timeout_ms,omitempty"`
}

type Embedder interface {
	Embed(context.Context, string) (Embedding, error)
}

type Embedding struct {
	ModelID     string    `json:"model_id"`
	Vector      []float32 `json:"-"`
	InputTokens *int64    `json:"input_tokens"`
	DurationMS  float64   `json:"duration_ms"`
}

type Entry struct {
	ID        string    `json:"id"`
	Partition string    `json:"partition"`
	ModelID   string    `json:"model_id"`
	Vector    []float32 `json:"vector"`
	Source    Source    `json:"source"`
	CreatedAt time.Time `json:"created_at"`
	ExpiresAt time.Time `json:"expires_at"`
}

type Index interface {
	QuerySemantic(context.Context, Query) (SearchResult, error)
	PutSemantic(context.Context, PutRequest) error
}

type Observation struct {
	Mode                 string        `json:"mode"`
	Status               string        `json:"status"`
	Partition            string        `json:"partition,omitempty"`
	EmbeddingModelID     string        `json:"embedding_model_id,omitempty"`
	EmbeddingInputTokens *int64        `json:"embedding_input_tokens"`
	EmbeddingMS          float64       `json:"embedding_ms,omitempty"`
	TotalMS              float64       `json:"total_ms"`
	Search               *SearchResult `json:"search,omitempty"`
	Indexed              bool          `json:"indexed"`
	GenerationAvoided    bool          `json:"generation_avoided"`
}

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

type PutRequest struct {
	Entry      Entry `json:"entry"`
	MaxEntries int   `json:"max_entries"`
}

type Query struct {
	Partition     string    `json:"partition"`
	ModelID       string    `json:"model_id"`
	Vector        []float32 `json:"vector"`
	MinSimilarity float64   `json:"min_similarity"`
	TopK          int       `json:"top_k"`
	MaxScan       int       `json:"max_scan"`
}

type ReuseConfig struct {
	Config
	ReadOnly                bool   `json:"read_only"`
	Intent                  string `json:"intent"`
	VerifierModel           string `json:"verifier_model"`
	VerifierMaxInputTokens  int64  `json:"verifier_max_input_tokens,omitempty"`
	VerifierMaxOutputTokens int64  `json:"verifier_max_output_tokens,omitempty"`
	VerifierTimeoutMS       int    `json:"verifier_timeout_ms,omitempty"`
	WaitForCandidateMS      int    `json:"wait_for_candidate_ms,omitempty"`
}

type SearchResult struct {
	Candidates []Candidate `json:"candidates"`
	Scanned    int         `json:"scanned"`
	Truncated  bool        `json:"truncated"`
}

type Source struct {
	Key          string `json:"key"`
	Version      int64  `json:"version"`
	SHA256       string `json:"sha256"`
	OutputSHA256 string `json:"output_sha256"`
}

type ValidityRequest struct {
	OriginalQuestion string            `json:"original_question"`
	Question         string            `json:"question"`
	Instructions     string            `json:"instructions"`
	Context          string            `json:"context"`
	CachedAnswer     string            `json:"cached_answer"`
	Bindings         map[string]string `json:"bindings"`
	Intent           string            `json:"intent"`
}

type ValidityResult struct {
	GenerationCalled bool   `json:"generation_called"`
	Complete         bool   `json:"complete"`
	Durable          bool   `json:"durable"`
	Verdict          string `json:"verdict"`
	SameIntent       bool   `json:"same_intent"`
	SameFacts        bool   `json:"same_facts"`
	SameConstraints  bool   `json:"same_constraints"`
	AnswerSupported  bool   `json:"answer_supported"`
	InputTokens      *int64 `json:"input_tokens"`
	OutputTokens     *int64 `json:"output_tokens"`
	Reused           bool   `json:"reused"`
	// SharedWork identifies a zero-incremental-usage decision whose model work
	// is accounted once by its batch's separate work record and shared receipt.
	SharedWork   bool    `json:"shared_work,omitempty"`
	DurationMS   float64 `json:"duration_ms"`
	ReceiptKey   string  `json:"receipt_key,omitempty"`
	OutputSHA256 string  `json:"output_sha256,omitempty"`
}

type Verifier interface {
	Verify(context.Context, ValidityRequest) (ValidityResult, error)
	Identity() string
}