vector

package

v0.0.5 Latest Latest Go to latest Published: Feb 22, 2025 License: MIT Imports: 9 Imported by: 0

Details

View Source

const (
	RoleSystem    = "system"
	RoleUser      = "user"
	RoleAssistant = "assistant"
	RoleTool      = "tool"
)

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, query string) ([]string, error)

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

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

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

type MemoryService struct {
	V         *Service
	SessionID string
}

func NewMemory(ctx context.Context, v *Service, sid string) (*MemoryService, error)

func (s *MemoryService) Store(ctx context.Context, role, content string) error

func (s *MemoryService) StoreEmbedding(ctx context.Context, role, content string, embedding []float64) 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)