mcp

type Config struct {
	// SpanSearcher and Embedder optionally retain the legacy core search tool
	// while search moves to a cassette. They must be supplied together.
	SpanSearcher SpanSearcher
	Embedder     embeddings.Embedder

	// Cassettes is the live registry whose advertised tools are exposed.
	Cassettes *cassetterunner.Registry

	// Client invokes admitted cassette operations. Nil uses a redirect-refusing client.
	Client *http.Client

	Logger *slog.Logger
}

type SearchInput struct {
	Query string `json:"query" jsonschema:"the search query text to find relevant spans"`
	TopK  int    `json:"top_k,omitempty" jsonschema:"number of results to return (default: 5)"`
}

type SearchOutput struct {
	Query   string         `json:"query"`
	Results []SearchResult `json:"results"`
	Count   int            `json:"count"`
}

type SearchResult struct {
	SessionID string  `json:"session_id,omitempty" jsonschema:"session the span belongs to"`
	TraceID   string  `json:"trace_id" jsonschema:"trace (turn) the span belongs to"`
	SpanID    string  `json:"span_id" jsonschema:"the matched span"`
	Score     float32 `json:"score" jsonschema:"vector similarity score"`
	// UserPrompt is the prompt of the turn (trace) the span belongs to.
	UserPrompt string `json:"user_prompt,omitempty" jsonschema:"prompt of the turn the span belongs to"`
	// Snippet previews the matched span's delta-only text.
	Snippet   string    `json:"snippet,omitempty" jsonschema:"preview of the matched span's text"`
	Model     string    `json:"model,omitempty" jsonschema:"model that produced the span"`
	StartedAt time.Time `json:"started_at" jsonschema:"when the span started"`
}

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

type SpanSearcher interface {
	Search(ctx context.Context, orgID string, embedding []float32, topK int) ([]spanembed.Hit, error)
}