properties: bubble up HNSW distance types #30

no manhattan in c

model without hamming etc..

Author: dan-obx

objectbox/c.py

@@ -877,3 +877,6 @@ def c_array_pointer(py_list: Union[List[Any], np.ndarray], c_type):
 
 OBXHnswDistanceType_UNKNOWN = 0
 OBXHnswDistanceType_EUCLIDEAN = 1
+OBXHnswDistanceType_COSINE = 2
+OBXHnswDistanceType_DOT_PRODUCT = 3
+OBXHnswDistanceType_DOT_PRODUCT_NON_NORMALIZED = 10

objectbox/model/properties.py

@@ -95,8 +95,11 @@ class HnswFlags(IntEnum):
 
 
 class HnswDistanceType(IntEnum):
-    UNKNOWN = OBXHnswDistanceType_UNKNOWN,
+    UNKNOWN = OBXHnswDistanceType_UNKNOWN
     EUCLIDEAN = OBXHnswDistanceType_EUCLIDEAN
+    COSINE = OBXHnswDistanceType_COSINE
+    DOT_PRODUCT = OBXHnswDistanceType_DOT_PRODUCT
+    DOT_PRODUCT_NON_NORMALIZED = OBXHnswDistanceType_DOT_PRODUCT_NON_NORMALIZED
 
 
 @dataclass

tests/model.py

@@ -62,3 +62,19 @@ class VectorEntity:
                           id=3, uid=40001,
                           dimensions=2, distance_type=HnswDistanceType.EUCLIDEAN)
                       )
+    vector_cosine = Property(np.ndarray, type=PropertyType.floatVector, id=4, uid=4004,
+                      index=HnswIndex(
+                          id=4, uid=40002,
+                          dimensions=2, distance_type=HnswDistanceType.COSINE)
+                      )
+    vector_dot_product = Property(np.ndarray, type=PropertyType.floatVector, id=5, uid=4005,
+                      index=HnswIndex(
+                          id=5, uid=40003,
+                          dimensions=2, distance_type=HnswDistanceType.DOT_PRODUCT)
+                      )
+    #vector_dot_product_non_normalized = Property(np.ndarray, type=PropertyType.floatVector, id=6, uid=4006,
+    #                  index=HnswIndex(
+    #                      id=6, uid=40004,
+    #                      dimensions=2, distance_type=HnswDistanceType.DOT_PRODUCT_NON_NORMALIZED)
+    #                  )
+