restore efficientcy

acheong08 · May 27, 2023 · 83ee790 · 83ee790
1 parent b4b54ec
commit 83ee790
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/semantic_search.go b/semantic_search.go
@@ -7,20 +7,13 @@ import (
 )
 
 func SemanticSearch(query []string, corpus []string, results int, sorted bool) ([][]typings.SearchResult, error) {
-	var encodedCorpus [][]float64 = make([][]float64, len(corpus))
-	var err error
-	for i, v := range corpus {
-		encodedCorpus[i], err = vectors.Encode(v)
-		if err != nil {
-			return [][]typings.SearchResult{}, err
-		}
+	encodedCorpus, err := vectors.EncodeMulti(corpus)
+	if err != nil {
+		return [][]typings.SearchResult{}, err
 	}
-	var encodedQuery [][]float64 = make([][]float64, len(query))
-	for i, v := range query {
-		encodedQuery[i], err = vectors.Encode(v)
-		if err != nil {
-			return [][]typings.SearchResult{}, err
-		}
+	encodedQuery, err := vectors.EncodeMulti(query)
+	if err != nil {
+		return [][]typings.SearchResult{}, err
 	}
 
 	// Semantic search