types

type EmbeddingClient interface {
	// Embed returns a vector embedding for the given text.
	Embed(ctx context.Context, text string) ([]float32, error)

	// EmbedBatch returns vector embeddings for multiple texts.
	EmbedBatch(ctx context.Context, texts []string) ([][]float32, error)

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

type OptimizerConfig struct {
	// EmbeddingService is the URL of the embedding service for semantic search.
	EmbeddingService string

	// EmbeddingServiceTimeout is the HTTP request timeout for calls to the embedding service.
	// Zero means use the default timeout (30s).
	EmbeddingServiceTimeout time.Duration

	// MaxToolsToReturn limits the number of tools returned by FindTool.
	MaxToolsToReturn *int

	// HybridSemanticRatio controls the balance between semantic and keyword search.
	HybridSemanticRatio *float64

	// SemanticDistanceThreshold sets the maximum distance for semantic search results (0.0 = identical, 2.0 = opposite).
	SemanticDistanceThreshold *float64
}

type ToolMatch struct {
	// Name is the unique identifier of the tool.
	Name string `json:"name"`

	// Description is the human-readable description of the tool.
	Description string `json:"description"`

	// Score is a distance metric indicating how well this tool matches.
	// Lower values indicate better matches (0 = identical, 2 = opposite).
	Score float64 `json:"score"`
}

type ToolStore interface {
	// UpsertTools adds or updates tools in the store.
	// Tools are identified by name; duplicate names are overwritten.
	UpsertTools(ctx context.Context, tools []server.ServerTool) error

	// Search finds tools matching the query string.
	// The allowedTools parameter limits results to only tools with names in the given set.
	// If allowedTools is empty, no results are returned (empty = no access).
	// Returns matches ranked by relevance.
	Search(ctx context.Context, query string, allowedTools []string) ([]ToolMatch, error)

	// Close releases any resources held by the store (e.g., database connections).
	// For in-memory stores this is a no-op.
	// It is safe to call Close multiple times.
	Close() error
}