vector

package

v0.0.4 Latest Latest Go to latest Published: Feb 20, 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 KnowledgeService struct {
	V *Service
}

func NewKnowledge(ctx context.Context, v *Service) (*KnowledgeService, error)

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

func (s *KnowledgeService) Store(ctx context.Context, text string) error

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

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

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

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

func (s *Service) Close()

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