clean: split distance strategy model

custom_pgvector.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import contextlib
-import enum
 import json
 import logging
 from typing import Any, Dict, Generator, Iterable, List, Optional, Tuple, Type
@@ -17,14 +16,7 @@ from sqlalchemy import delete, text
 from sqlalchemy.orm import Session, declarative_base
 
 from utils import str_to_list
-
-
-class DistanceStrategy(str, enum.Enum):
-    """Enumerator of the Distance strategies."""
-
-    EUCLIDEAN = "l2"
-    COSINE = "cosine"
-    MAX_INNER_PRODUCT = "inner"
+from distance import DistanceStrategy
 
 
 DEFAULT_DISTANCE_STRATEGY = DistanceStrategy.COSINE

models/distance.py

@@ -0,0 +1,9 @@
+import enum
+
+
+class DistanceStrategy(str, enum.Enum):
+    """Enumerator of the Distance strategies."""
+
+    EUCLIDEAN = "l2"
+    COSINE = "cosine"
+    MAX_INNER_PRODUCT = "inner"