rag

package

v1.1.0 Latest Latest Go to latest Published: Mar 6, 2025 License: MIT Imports: 7 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type OpenAIConfig struct {
	EmbedModel string
	Endpoint   string
	LLMModel   string
	Token      string
}

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

func New(filename string, config *OpenAIConfig) (*RAG, error)

func (r *RAG) AddDocument(id string, document string) error

func (r *RAG) Ask(query string) (string, error)

func (r *RAG) Search(query string) ([]chromem.Result, error)