internal/service/embedding/embedding_service.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package embedding

import (
	"context"
	"fmt"

	"github.com/apache/answer/plugin"
)

// EmbeddingService is a thin facade that delegates semantic search to a VectorSearch plugin.
// If no plugin is enabled, semantic search is unavailable.
type EmbeddingService struct{}

// NewEmbeddingService creates a new EmbeddingService.
func NewEmbeddingService() *EmbeddingService {
	return &EmbeddingService{}
}

// SearchSimilar delegates to the VectorSearch plugin.
// Returns an error if no plugin is enabled.
func (s *EmbeddingService) SearchSimilar(ctx context.Context, query string, topK int) ([]plugin.VectorSearchResult, error) {
	var results []plugin.VectorSearchResult
	var searchErr error
	found := false

	err := plugin.CallVectorSearch(func(vs plugin.VectorSearch) error {
		found = true
		results, searchErr = vs.SearchSimilar(ctx, query, topK)
		return nil
	})
	if err != nil {
		return nil, fmt.Errorf("call vector search plugin failed: %w", err)
	}
	if !found {
		return nil, fmt.Errorf("semantic search is not available: no vector search plugin is enabled")
	}
	if searchErr != nil {
		return nil, searchErr
	}
	return results, nil
}