vector

package

v0.0.3 Latest Latest Go to latest Published: Feb 18, 2025 License: MIT Imports: 8 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Config struct {
	Host     string
	Port     string
	Database string
	User     string
	Password string
	SSLMode  string
}

func EnvConfig() *Config

type Service struct {
	DB        *sqlx.DB
	Table     string
	OpenAICli *openai.Client
}

func New(ctx context.Context, table string, cli *openai.Client) (*Service, error)

func (s *Service) Close()

func (s *Service) GenerateEmbeddings(ctx context.Context, text string) ([]float64, error)

func (s *Service) Query(ctx context.Context, input string) ([]string, error)

func (s *Service) StoreEmbedding(ctx context.Context, text string, embedding []float64) error

func (s *Service) Truncate(ctx context.Context) error