retrieval

package
v0.1.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 14, 2026 License: Apache-2.0 Imports: 7 Imported by: 0

Documentation

Overview

Package retrieval provides error definitions for retrieval operations.

Package retrieval provides high-level APIs for knowledge retrieval operations.

Index

Constants

This section is empty.

Variables

View Source
var (
	// ErrInvalidTenantID is returned when tenant ID is empty.
	ErrInvalidTenantID = errors.New("invalid tenant ID")

	// ErrInvalidQuery is returned when query is empty.
	ErrInvalidQuery = errors.New("invalid query")

	// ErrNoRetrievalService is returned when no retrieval service is configured.
	ErrNoRetrievalService = errors.New("no retrieval service configured")

	// ErrSearchFailed is returned when search operation fails.
	ErrSearchFailed = errors.New("search failed")
)

Functions

This section is empty.

Types

type Config

type Config struct {
	UseSimpleRetrieval bool
	TopK               int
	MinScore           float64
}

Config configuration for retrieval service.

type Result

type Result struct {
	Content   string  `json:"content"`
	Source    string  `json:"source"`
	Score     float64 `json:"score"`
	SubSource string  `json:"sub_source"`
}

Result represents a single retrieval result.

type Service

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

Service provides retrieval operations for knowledge base.

func NewService

func NewService(
	pool *postgres.Pool,
	embeddingClient *embedding.EmbeddingClient,
	kbRepo *repositories.KnowledgeRepository,
	config *Config,
) (*Service, error)

NewService creates a new retrieval service instance. Args: pool - database connection pool. embeddingClient - embedding service client. kbRepo - knowledge repository for data access. config - retrieval configuration. Returns new retrieval service instance or error.

func (*Service) Search

func (s *Service) Search(ctx context.Context, tenantID, query string) ([]*Result, error)

Search performs knowledge base search. Args: ctx - operation context. tenantID - tenant identifier for isolation. query - search query text. Returns search results or error if search fails.

func (*Service) SearchWithConfig

func (s *Service) SearchWithConfig(ctx context.Context, tenantID, query string, config *Config) ([]*Result, error)

SearchWithConfig performs search with custom configuration. Args: ctx - operation context. tenantID - tenant identifier for isolation. query - search query text. config - custom search configuration. Returns search results or error if search fails.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL