Skip to content

Commit 8b9f9ca

Browse files
committed
Improved code
1 parent 80e7570 commit 8b9f9ca

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

pgvector/django/functions.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,10 @@ class DistanceBase(Func):
88

99
def __init__(self, expression: Any, vector: Any, **extra: Any) -> None:
1010
if not hasattr(vector, 'resolve_expression'):
11-
if isinstance(vector, HalfVector):
12-
vector = Value(HalfVector._to_db(vector))
13-
elif isinstance(vector, SparseVector):
14-
vector = Value(SparseVector._to_db(vector))
15-
else:
16-
vector = Value(Vector._to_db(vector))
11+
if isinstance(vector, (Vector, HalfVector, SparseVector)):
12+
vector = Value(vector.to_text())
13+
elif vector is not None:
14+
vector = Value(Vector(vector).to_text())
1715

1816
# prevent error with unhashable types
1917
self._constructor_args = ((expression, vector), extra)

0 commit comments

Comments
 (0)