Propage vector properties into in-memory vector indexes.

Author: nicktobey

memory/table.go

@@ -2012,6 +2012,11 @@ func (t *Table) createIndex(data *TableData, name string, columns []sql.IndexCol
 		}
 	}
 
+	var vectorFunction vector.DistanceType
+	if constraint == sql.IndexConstraint_Vector {
+		vectorFunction = vector.DistanceL2Squared{}
+	}
+
 	return &Index{
 		DB:                      t.dbName(),
 		DriverName:              "",
@@ -2022,7 +2027,7 @@ func (t *Table) createIndex(data *TableData, name string, columns []sql.IndexCol
 		Unique:                  constraint == sql.IndexConstraint_Unique,
 		Spatial:                 constraint == sql.IndexConstraint_Spatial,
 		Fulltext:                constraint == sql.IndexConstraint_Fulltext,
-		SupportedVectorFunction: nil,
+		SupportedVectorFunction: vectorFunction,
 		CommentStr:              comment,
 		PrefixLens:              prefixLengths,
 	}, nil

sql/analyzer/replace_order_by_distance.go

@@ -41,6 +41,11 @@ func replaceIdxOrderByDistanceHelper(ctx *sql.Context, scope *plan.Scope, node s
 		if err != nil {
 			return nil, transform.SameTree, err
 		}
+
+		// Column references have not been assigned their final indexes yet, so do that for the ORDER BY expression now.
+		// We can safely do this because an expression that references other tables won't pass `isSortFieldsValidPrefix` below.
+		sortNode = offsetAssignIndexes(sortNode).(plan.Sortable)
+
 		sfExprs := normalizeExpressions(tableAliases, sortNode.GetSortFields().ToExpressions()...)
 		sfAliases := aliasedExpressionsInNode(sortNode)