Merge remote-tracking branch 'upstream/master' into load-balance-search-commands-to-shards

Author: ofekshenawa

search_commands.go

@@ -511,6 +511,14 @@ type FTAttribute struct {
 	PhoneticMatcher string
 	CaseSensitive   bool
 	WithSuffixtrie  bool
+
+	// Vector specific attributes
+	Algorithm      string
+	DataType       string
+	Dim            int
+	DistanceMetric string
+	M              int
+	EFConstruction int
 }
 
 type CursorStats struct {
@@ -1410,21 +1418,26 @@ func parseFTInfo(data map[string]interface{}) (FTInfoResult, error) {
 		for _, attr := range attributes {
 			if attrMap, ok := attr.([]interface{}); ok {
 				att := FTAttribute{}
-				for i := 0; i < len(attrMap); i++ {
-					if internal.ToLower(internal.ToString(attrMap[i])) == "attribute" {
+				attrLen := len(attrMap)
+				for i := 0; i < attrLen; i++ {
+					if internal.ToLower(internal.ToString(attrMap[i])) == "attribute" && i+1 < attrLen {
 						att.Attribute = internal.ToString(attrMap[i+1])
+						i++
 						continue
 					}
-					if internal.ToLower(internal.ToString(attrMap[i])) == "identifier" {
+					if internal.ToLower(internal.ToString(attrMap[i])) == "identifier" && i+1 < attrLen {
 						att.Identifier = internal.ToString(attrMap[i+1])
+						i++
 						continue
 					}
-					if internal.ToLower(internal.ToString(attrMap[i])) == "type" {
+					if internal.ToLower(internal.ToString(attrMap[i])) == "type" && i+1 < attrLen {
 						att.Type = internal.ToString(attrMap[i+1])
+						i++
 						continue
 					}
-					if internal.ToLower(internal.ToString(attrMap[i])) == "weight" {
+					if internal.ToLower(internal.ToString(attrMap[i])) == "weight" && i+1 < attrLen {
 						att.Weight = internal.ToFloat(attrMap[i+1])
+						i++
 						continue
 					}
 					if internal.ToLower(internal.ToString(attrMap[i])) == "nostem" {
@@ -1443,7 +1456,7 @@ func parseFTInfo(data map[string]interface{}) (FTInfoResult, error) {
 						att.UNF = true
 						continue
 					}
-					if internal.ToLower(internal.ToString(attrMap[i])) == "phonetic" {
+					if internal.ToLower(internal.ToString(attrMap[i])) == "phonetic" && i+1 < attrLen {
 						att.PhoneticMatcher = internal.ToString(attrMap[i+1])
 						continue
 					}
@@ -1456,6 +1469,38 @@ func parseFTInfo(data map[string]interface{}) (FTInfoResult, error) {
 						continue
 					}
 
+					// vector specific attributes
+					if internal.ToLower(internal.ToString(attrMap[i])) == "algorithm" && i+1 < attrLen {
+						att.Algorithm = internal.ToString(attrMap[i+1])
+						i++
+						continue
+					}
+					if internal.ToLower(internal.ToString(attrMap[i])) == "data_type" && i+1 < attrLen {
+						att.DataType = internal.ToString(attrMap[i+1])
+						i++
+						continue
+					}
+					if internal.ToLower(internal.ToString(attrMap[i])) == "dim" && i+1 < attrLen {
+						att.Dim = internal.ToInteger(attrMap[i+1])
+						i++
+						continue
+					}
+					if internal.ToLower(internal.ToString(attrMap[i])) == "distance_metric" && i+1 < attrLen {
+						att.DistanceMetric = internal.ToString(attrMap[i+1])
+						i++
+						continue
+					}
+					if internal.ToLower(internal.ToString(attrMap[i])) == "m" && i+1 < attrLen {
+						att.M = internal.ToInteger(attrMap[i+1])
+						i++
+						continue
+					}
+					if internal.ToLower(internal.ToString(attrMap[i])) == "ef_construction" && i+1 < attrLen {
+						att.EFConstruction = internal.ToInteger(attrMap[i+1])
+						i++
+						continue
+					}
+
 				}
 				ftInfo.Attributes = append(ftInfo.Attributes, att)
 			}