Customize embedding model

2024-12-17 19:53:30 +01:00
parent 2561f2f175
commit 6bafb48cec
2 changed files with 15 additions and 12 deletions
--- a/example/webui/main.go
+++ b/example/webui/main.go
@@ -23,6 +23,7 @@ var apiKey = os.Getenv("API_KEY")
 var vectorStore = os.Getenv("VECTOR_STORE")
 var kbdisableIndexing = os.Getenv("KBDISABLEINDEX")
 var timeout = os.Getenv("TIMEOUT")
+var embeddingModel = os.Getenv("EMBEDDING_MODEL")

 const defaultChunkSize = 4098

@@ -63,7 +64,7 @@ func main() {
 		dbStore = rag.NewLocalAIRAGDB(laiStore, lai)
 	default:
 		var err error
-		dbStore, err = rag.NewChromemDB("local-agent-framework", stateDir, lai)
+		dbStore, err = rag.NewChromemDB("local-agent-framework", stateDir, lai, embeddingModel)
 		if err != nil {
 			panic(err)
 		}