vector

package

v0.0.8 Latest Latest Go to latest Published: Feb 27, 2025 License: MIT Imports: 11 Imported by: 0

Details

View Source

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

This section is empty.

This section is empty.

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 []float32) error

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

type Memory struct {
	Role    string `db:"role"`
	Content string `db:"content"`
}

type MemoryService struct {
	V         *Service
	SessionID string
}

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

func (s *MemoryService) Query(ctx context.Context, query string) (string, error)

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

func (s *MemoryService) StoreEmbedding(ctx context.Context, role, content string, embedding []float32) error

type Service struct {
	DB         *sqlx.DB
	OpenAICli  *openai.Client
	Model      string
	Dimensions int64
}

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

func (s *Service) Close()

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