rag

package

v0.0.0-...-99d71fe Latest Latest Go to latest Published: Dec 8, 2025 License: MIT Imports: 11 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type RagAgent struct {
	Name    string
	ModelID string
	// contains filtered or unexported fields
}

func NewRagAgent(ctx context.Context, ragAgentConfig agents.AgentConfig, storeConfig StoreConfig, opts ...RagAgentOption) (*RagAgent, error)

func (agent *RagAgent) AddTextChunksToStore(chunks []text.TextChunk) (int, error)

func (agent *RagAgent) GetInfo() (agents.RagAgentInfo, error)

func (agent *RagAgent) GetName() string

func (agent *RagAgent) GetNumberOfDocuments() int

getNumberOfDocuments returns the number of documents in the store

func (agent *RagAgent) IsStoreInitialized() bool

isStoreInitialized checks if the document store is initialized

func (agent *RagAgent) Kind() agents.AgentKind

func (agent *RagAgent) SearchSimilarities(query string) ([]string, error)

type RagAgentOption func(*RagAgent)

AgentOption defines a functional option for configuring LocalAIAgent

func WithLogLevel(level logger.LogLevel) RagAgentOption

WithLogLevel sets the log level for the agent

func WithLogger(log logger.Logger) RagAgentOption

WithLogger sets a custom logger for the agent

func WithVerbose(verbose bool) RagAgentOption

WithVerbose enables verbose logging (INFO level) with agent name prefix

type StoreConfig struct {
	StoreName string
	StorePath string
}

type TextChunk struct {
	Content  string
	Metadata map[string]any
}