use autocut and certainty in recommending

Southclaws · Jul 3, 2024 · 8779e90 · 8779e90
1 parent 8529f57
commit 8779e90
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app/services/semdex/weaviate/recommend.go b/app/services/semdex/weaviate/recommend.go
@@ -32,7 +32,7 @@ func (w *weaviateSemdexer) Recommend(ctx context.Context, object datagraph.Index
 
 	withNearVector := w.wc.GraphQL().NearVectorArgBuilder().
 		WithVector(wobj.Vector).
-		WithDistance(0.5)
+		WithCertainty(0.5)
 
 	fields := []graphql.Field{
 		{Name: "datagraph_id"},
@@ -43,6 +43,7 @@ func (w *weaviateSemdexer) Recommend(ctx context.Context, object datagraph.Index
 		WithClassName(w.cn.String()).
 		WithFields(fields...).
 		WithNearVector(withNearVector).
+		WithAutocut(2).
 		WithLimit(10).
 		Do(ctx)
 	if err != nil {