Added similar_to

dgraph/cmd/zero/oracle.go

@@ -30,6 +30,7 @@ import (
 	"github.com/dgraph-io/badger/v4/y"
 	"github.com/dgraph-io/dgo/v230/protos/api"
 	"github.com/dgraph-io/dgraph/protos/pb"
+	"github.com/dgraph-io/dgraph/tok/hnsw"
 	"github.com/dgraph-io/dgraph/x"
 	"github.com/dgraph-io/ristretto/z"
 )
@@ -376,6 +377,9 @@ func (s *Server) commit(ctx context.Context, src *api.TxnContext) error {
 				return errors.Wrapf(err, "unable to parse group id from %s", pkey)
 			}
 			pred := splits[1]
+			if strings.Contains(pred, hnsw.VecKeyword) {
+				pred = pred[0:strings.Index(pred, hnsw.VecKeyword)]
+			}
 			tablet := s.ServingTablet(pred)
 			if tablet == nil {
 				return errors.Errorf("Tablet for %s is nil", pred)

dgraphtest/paths.go

@@ -51,9 +51,9 @@ func init() {
 	if err != nil {
 		panic(err)
 	}
-	if err := ensureDgraphClone(); err != nil {
-		panic(err)
-	}
+	//if err := ensureDgraphClone(); err != nil {
+	//	panic(err)
+	//}
 
 	log.Printf("[INFO] baseRepoDir: %v", baseRepoDir)
 	log.Printf("[INFO] repoDir: %v", repoDir)

dql/parser.go

@@ -33,12 +33,13 @@ import (
 )
 
 const (
-	uidFunc   = "uid"
-	valueFunc = "val"
-	typFunc   = "type"
-	lenFunc   = "len"
-	countFunc = "count"
-	uidInFunc = "uid_in"
+	uidFunc     = "uid"
+	valueFunc   = "val"
+	typFunc     = "type"
+	lenFunc     = "len"
+	countFunc   = "count"
+	uidInFunc   = "uid_in"
+	similarToFn = "similar_to"
 )
 
 var (
@@ -356,7 +357,7 @@ func parseValue(v varInfo) (types.Val, error) {
 				}, nil
 			}
 		}
-	case "vfloat":
+	case "vector32float":
 		{
 			if i, err := types.ParseVFloat(v.Value); err != nil {
 				return types.Val{}, errors.Wrapf(err, "Expected a vfloat but got %v", v.Value)
@@ -1711,7 +1712,7 @@ func validFuncName(name string) bool {
 
 	switch name {
 	case "regexp", "anyofterms", "allofterms", "alloftext", "anyoftext",
-		"has", "uid", "uid_in", "anyof", "allof", "type", "match":
+		"has", "uid", "uid_in", "anyof", "allof", "type", "match", "similar_to":
 		return true
 	}
 	return false
@@ -1884,7 +1885,7 @@ L:
 				case IsInequalityFn(function.Name):
 					err = parseFuncArgs(it, function)
 
-				case function.Name == "uid_in":
+				case function.Name == "uid_in" || function.Name == "similar_to":
 					err = parseFuncArgs(it, function)
 
 				default:

go.mod

@@ -59,6 +59,7 @@ require (
 	go.opencensus.io v0.24.0
 	go.uber.org/zap v1.16.0
 	golang.org/x/crypto v0.21.0
+	golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8
 	golang.org/x/net v0.22.0
 	golang.org/x/sync v0.6.0
 	golang.org/x/sys v0.18.0
@@ -141,7 +142,6 @@ require (
 	github.com/xdg/stringprep v1.0.3 // indirect
 	go.uber.org/atomic v1.9.0 // indirect
 	go.uber.org/multierr v1.10.0 // indirect
-	golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8 // indirect
 	golang.org/x/mod v0.16.0 // indirect
 	golang.org/x/time v0.3.0 // indirect
 	google.golang.org/api v0.122.0 // indirect