fix embedder test

getzep · Sep 17, 2023 · e7ec94a · e7ec94a
1 parent e1689db
commit e7ec94a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/extractors/embedder_test.go b/pkg/extractors/embedder_test.go
@@ -13,8 +13,9 @@ import (
 )
 
 func TestEmbeddingExtractor_Extract_OpenAI(t *testing.T) {
-	appState.Config.LLM.Service = "openai"
-	appState.Config.LLM.Model = "gpt-3.5-turbo"
+	appState.Config.Extractors.Messages.Embeddings.Service = "openai"
+	appState.Config.Extractors.Messages.Embeddings.Dimensions = 1536
+
 	llmClient, err := llms.NewOpenAILLM(testCtx, appState.Config)
 	assert.NoError(t, err)
 	appState.LLMClient = llmClient