embedding

type Config struct {
	Provider    string      `json:"provider"`
	Model       string      `json:"model"`
	Credentials Credentials `json:"credentials,omitempty"`
}

type Creator interface {
	Generate(ctx context.Context, text, model string, cfg *Config) (*Embedding, error)
}

type Credentials struct {
	APIKey      string `json:"api_key,omitempty"` // #nosec G117
	HeaderName  string `json:"header_name,omitempty"`
	HeaderValue string `json:"header_value,omitempty"`
}

type Embedding struct {
	EntityID  string    `json:"entity_id"`
	Value     []float64 `json:"value"`
	CreatedAt time.Time `json:"created_at"`
}

type Repository interface {
	Count(ctx context.Context, index, keyQuery string) (int, error)
	Store(ctx context.Context, targetID string, embeddingData *Embedding, key string) error
	GetByTargetID(ctx context.Context, targetID string) (*Embedding, error)
	StoreWithHMSet(ctx context.Context, index, key, gatewayID string, embeddingData *Embedding, data []byte) error
	Search(ctx context.Context, index, query string, embeddingData *Embedding) ([]SearchResult, error)
}

type SearchResult struct {
	Key   string
	Score float64
	Data  string
}