Function queryProjectsSimilarByEmbedding

query/vector/vector_graphql_test.go:125–153

(t *testing.T, hc *dgraphapi.HTTPClient, vector []float32, topk int)

}

func queryProjectsSimilarByEmbedding(t *testing.T, hc *dgraphapi.HTTPClient, vector []float32, topk int) []ProjectInput {
	// query similar project by embedding
	queryProduct := `query QuerySimilarProjectByEmbedding($by: ProjectEmbedding!, $topK: Int!, $vector: [Float!]!) {
		 querySimilarProjectByEmbedding(by: $by, topK: $topK, vector: $vector) {
		   title
		   title_v
		 }
	   }

	 `

	params := dgraphapi.GraphQLParams{
		Query: queryProduct,
		Variables: map[string]interface{}{
			"by":     "title_v",
			"topK":   topk,
			"vector": vector,
		}}
	response, err := hc.RunGraphqlQuery(params, false)
	require.NoError(t, err)
	type QueryResult struct {
		QueryProject []ProjectInput `json:"querySimilarProjectByEmbedding"`
	}
	var resp QueryResult
	err = json.Unmarshal([]byte(string(response)), &resp)
	require.NoError(t, err)

	return resp.QueryProject
}

func TestVectorGraphQLAddVectorPredicate(t *testing.T) {
	require.NoError(t, client.DropAll())