Fix fake ai client pre-programmed errors

initialcapacity · Jun 5, 2024 · 9167ac4 · 9167ac4
1 parent 50064d0
commit 9167ac4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/internal/app/query_service_test.go b/internal/app/query_service_test.go
@@ -72,16 +72,16 @@ type fakeAi struct {
 }
 
 func (f fakeAi) CreateEmbedding(_ context.Context, _ string) ([]float32, error) {
-	if f.completionError != nil {
-		return nil, f.completionError
+	if f.embeddingError != nil {
+		return nil, f.embeddingError
 	}
 
 	return testsupport.CreateVector(0), nil
 }
 
 func (f fakeAi) GetChatCompletion(_ context.Context, _ []ai.ChatMessage) (chan string, error) {
-	if f.embeddingError != nil {
-		return nil, f.embeddingError
+	if f.completionError != nil {
+		return nil, f.completionError
 	}
 
 	response := make(chan string)