Other vector functions

Author: griffio

pgvector-module/src/main/kotlin/griffio/PgVectorModule.kt

@@ -62,7 +62,13 @@ private class PgVectorTypeResolver(private val parentResolver: TypeResolver) : P
         when (functionExpr.functionName.text.lowercase()) {
             "binary_quantize" -> IntermediateType(PostgreSqlType.BIT)
             "cosine_distance" -> IntermediateType(PrimitiveType.REAL)
+            "inner_product" -> IntermediateType(PrimitiveType.REAL)
+            "l1_distance" -> IntermediateType(PrimitiveType.REAL)
+            "l2_distance" -> IntermediateType(PrimitiveType.REAL)
+            "l2_normalize" -> IntermediateType(PgVectorSqlType.VECTOR)
             "subvector" -> IntermediateType(PgVectorSqlType.VECTOR)
+            "vector_dims" -> IntermediateType(PostgreSqlType.INTEGER)
+            "vector_norm" -> IntermediateType(PrimitiveType.REAL)
             else -> parentResolver.functionType(functionExpr)
         }
 }

src/main/kotlin/griffio/Main.kt

@@ -26,4 +26,12 @@ fun main() {
     sample.vectorQueries.selectCosineDistance().executeAsList().forEach(::println)
     println("binaryQuantize")
     sample.vectorQueries.selectBinaryQuantize().executeAsList().forEach(::println)
+    println("l2Distance")
+    sample.vectorQueries.selectL2Distance().executeAsList().forEach(::println)
+    println("innerProduct")
+    sample.vectorQueries.selectInnerProduct().executeAsList().forEach(::println)
+    println("euclideanDistance")
+    sample.vectorQueries.selectL1Distance().executeAsList().forEach(::println)
+    println("selectL2Normalize")
+    sample.vectorQueries.selectL2Normalize().executeAsList().forEach(::println)
 }

src/main/sqldelight/griffio/queries/vector.sq

@@ -19,3 +19,21 @@ SELECT cosine_distance('[1,1]'::VECTOR, '[-1,-1]');
 
 selectBinaryQuantize:
 SELECT binary_quantize('[0,0.1,-0.2,-0.3,0.4,0.5,0.6,-0.7,0.8,-0.9,1]'::VECTOR);
+
+selectVectorNorm:
+SELECT round(vector_norm('[1,1]')::NUMERIC, 5);
+
+selectL2Distance:
+SELECT l2_distance('[0,0]'::VECTOR, '[3,4]');
+
+selectInnerProduct:
+SELECT inner_product('[1,2]'::VECTOR, '[3,4]');
+
+selectL1Distance:
+SELECT l1_distance('[1,2,3,4,5,6,7,8,9]'::VECTOR, '[0,3,2,5,4,7,6,9,8]');
+
+selectL2Normalize:
+SELECT l2_normalize('[3,4]'::VECTOR);
+
+
+