rag

Documentation

Index

• type Client
  • func NewClient(embeddingModel embedding.Model, store *VectorStore) *Client
  • func (c *Client) BuildKnowledgeBase(ctx context.Context, knowledgeBasePath string, model string, reBuild bool) error
  • func (c *Client) Query(ctx context.Context, model string, content string, limit int) ([]*DocumentPart, error)
• type DocumentPart
• type VectorStore
  • func NewPgVectorStore() (*VectorStore, error)
  • func (s *VectorStore) CheckDocumentExist(ctx context.Context, documentID string) (bool, error)
  • func (s *VectorStore) DeleteDocuments(ctx context.Context, documentID string) error
  • func (s *VectorStore) GetMostRelevantDocuments(ctx context.Context, embedding []float32, threshold float32, limit int) ([]*DocumentPart, error)
  • func (s *VectorStore) Init(ctx context.Context) error
  • func (s *VectorStore) Release()
  • func (s *VectorStore) SaveDocumentEmbedding(ctx context.Context, documentID string, partIndex int, content string, ...) error

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

func NewClient

func NewClient(embeddingModel embedding.Model, store *VectorStore) *Client

func (*Client) BuildKnowledgeBase

func (c *Client) BuildKnowledgeBase(ctx context.Context, knowledgeBasePath string, model string, reBuild bool) error

func (*Client) Query

func (c *Client) Query(ctx context.Context, model string, content string, limit int) ([]*DocumentPart, error)

type DocumentPart

type DocumentPart struct {
	DocumentID string
	PartIndex  int
	Content    string
}

type VectorStore

type VectorStore struct {
	// contains filtered or unexported fields
}

func NewPgVectorStore

func NewPgVectorStore() (*VectorStore, error)

func (*VectorStore) CheckDocumentExist

func (s *VectorStore) CheckDocumentExist(ctx context.Context, documentID string) (bool, error)

func (*VectorStore) DeleteDocuments

func (s *VectorStore) DeleteDocuments(ctx context.Context, documentID string) error

func (*VectorStore) GetMostRelevantDocuments

func (s *VectorStore) GetMostRelevantDocuments(ctx context.Context, embedding []float32, threshold float32, limit int) ([]*DocumentPart, error)

func (*VectorStore) Init

func (s *VectorStore) Init(ctx context.Context) error

func (*VectorStore) Release

func (s *VectorStore) Release()

func (*VectorStore) SaveDocumentEmbedding

func (s *VectorStore) SaveDocumentEmbedding(ctx context.Context, documentID string, partIndex int, content string, embedding []float32) error