Function queryMultipleVectorsUsingSimilarTo

query/vector/vector_test.go:194–227

(t *testing.T, vector []float32, pred string, topK int)

}

func queryMultipleVectorsUsingSimilarTo(t *testing.T, vector []float32, pred string, topK int) ([][]float32, error) {
	vectorQuery := fmt.Sprintf(`
	 {
		 vector(func: similar_to(%v, %v, "%v")) {
				uid
				%v
		  }
	 }`, pred, topK, vector, pred)

	resp, err := client.Query(vectorQuery)
	require.NoError(t, err)

	type VectorData struct {
		UID   string    `json:"uid"`
		VTest []float32 `json:"vtest"`
	}

	type Data struct {
		Vector []VectorData `json:"vector"`
	}

	var data Data

	err = json.Unmarshal([]byte(resp.Json), &data)
	if err != nil {
		return [][]float32{}, err
	}

	var vectors [][]float32
	for _, vector := range data.Vector {
		vectors = append(vectors, vector.VTest)
	}
	return vectors, nil
}

func querySingleVectorError(t *testing.T, vector, pred string, validateError bool) ([]float32, error) {