mcp

type ChunkFileMetadata struct {
	Model      string `json:"model"`
	Dimensions int    `json:"dimensions"`
	ChunkType  string `json:"chunk_type"`
	Generated  string `json:"generated"` // time as string
	Version    string `json:"version"`
}

type ChunkFileWrapper struct {
	Metadata ChunkFileMetadata `json:"_metadata"`
	Chunks   []*ContextChunk   `json:"chunks"`
}

type ContextChunk struct {
	ID        string                 `json:"id"`
	Title     string                 `json:"title"`
	Text      string                 `json:"text"`
	ChunkType string                 `json:"chunk_type,omitempty"`
	Embedding []float32              `json:"embedding,omitempty"` // Excluded via MarshalJSON
	Tags      []string               `json:"tags,omitempty"`
	Metadata  map[string]interface{} `json:"metadata,omitempty"`
	CreatedAt time.Time              `json:"created_at"`
	UpdatedAt time.Time              `json:"updated_at"`
}

type ContextSearcher interface {
	// Query executes a semantic search for the given query string.
	// Returns ranked search results based on vector similarity and filters.
	Query(ctx context.Context, query string, options *SearchOptions) ([]*SearchResult, error)

	// Reload reloads the chunk database from the configured directory.
	// Used for hot reload when chunk files are updated.
	Reload(ctx context.Context) error

	// GetMetrics returns current reload operation metrics.
	// Used for monitoring reload health and statistics.
	GetMetrics() MetricsSnapshot

	// Close releases resources held by the searcher.
	Close() error
}

type CortexSearchRequest struct {
	Query        string   `json:"query" jsonschema:"required,description=Natural language search query"`
	Limit        int      `json:"limit,omitempty" jsonschema:"minimum=1,maximum=100,default=15,description=Maximum number of results"`
	Tags         []string `json:"tags,omitempty" jsonschema:"description=Filter by tags (AND logic)"`
	ChunkTypes   []string `json:"chunk_types,omitempty" jsonschema:"description=Filter by chunk type (documentation|symbols|definitions|data)"`
	IncludeStats bool     `json:"include_stats,omitempty" jsonschema:"default=false,description=Include reload metrics in response"`
}

type CortexSearchResponse struct {
	Results []*SearchResult  `json:"results"`
	Total   int              `json:"total"`
	Metrics *MetricsSnapshot `json:"metrics,omitempty"`
}

type EmbeddingProvider interface {
	// Embed converts a slice of text strings into their vector representations.
	// The mode parameter specifies whether embeddings are for queries or passages.
	Embed(ctx context.Context, texts []string, mode string) ([][]float32, error)

	// Dimensions returns the dimensionality of the embedding vectors.
	Dimensions() int

	// Close releases any resources held by the provider.
	Close() error
}

type EmbeddingServiceConfig struct {
	BaseURL string
}

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

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

type MCPServerConfig struct {
	ChunksDir        string
	EmbeddingService *EmbeddingServiceConfig
}

type MetricsSnapshot struct {
	LastReloadTime     time.Time     `json:"last_reload_time"`
	LastReloadDuration time.Duration `json:"last_reload_duration_ms"`
	LastReloadError    string        `json:"last_reload_error,omitempty"`
	TotalReloads       int64         `json:"total_reloads"`
	SuccessfulReloads  int64         `json:"successful_reloads"`
	FailedReloads      int64         `json:"failed_reloads"`
	CurrentChunkCount  int           `json:"current_chunk_count"`
}

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

type SearchOptions struct {
	// Limit specifies the maximum number of results to return (1-100)
	Limit int `json:"limit,omitempty"`

	// MinScore filters results below this similarity threshold (0.0-1.0)
	MinScore float64 `json:"min_score,omitempty"`

	// Tags filters results to only include chunks with ALL specified tags (AND logic)
	Tags []string `json:"tags,omitempty"`

	// ChunkTypes filters results by chunk type (documentation, symbols, definitions, data)
	ChunkTypes []string `json:"chunk_types,omitempty"`
}

type SearchResult struct {
	Chunk         *ContextChunk `json:"chunk"`
	CombinedScore float64       `json:"combined_score"`
}