22 lines
427 B
Go
22 lines
427 B
Go
package rag
|
|
|
|
import (
|
|
"easy_rag/internal/database"
|
|
"easy_rag/internal/embeddings"
|
|
"easy_rag/internal/llm"
|
|
)
|
|
|
|
type Rag struct {
|
|
LLM llm.LLMService
|
|
Embeddings embeddings.EmbeddingsService
|
|
Database database.Database
|
|
}
|
|
|
|
func NewRag(llm llm.LLMService, embeddings embeddings.EmbeddingsService, database database.Database) *Rag {
|
|
return &Rag{
|
|
LLM: llm,
|
|
Embeddings: embeddings,
|
|
Database: database,
|
|
}
|
|
}
|