-
Notifications
You must be signed in to change notification settings - Fork 95
Expand file tree
/
Copy pathvector.py
More file actions
35 lines (24 loc) · 1.15 KB
/
Copy pathvector.py
File metadata and controls
35 lines (24 loc) · 1.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from peewee import Expression, Field
from typing import Any
from .. import Vector
class VectorField(Field):
field_type = 'vector'
def __init__(self, dimensions: int | None = None, *args: Any, **kwargs: Any) -> None:
self.dimensions = dimensions
super(VectorField, self).__init__(*args, **kwargs)
def get_modifiers(self) -> list[int] | None:
return [self.dimensions] if self.dimensions else None
def db_value(self, value: object) -> str | None:
return Vector._to_db(value)
def python_value(self, value: Any) -> Vector | None:
return Vector._from_db(value)
def _distance(self, op: str, vector: object) -> Expression:
return Expression(lhs=self, op=op, rhs=self.to_value(vector))
def l2_distance(self, vector: object) -> Expression:
return self._distance('<->', vector)
def max_inner_product(self, vector: object) -> Expression:
return self._distance('<#>', vector)
def cosine_distance(self, vector: object) -> Expression:
return self._distance('<=>', vector)
def l1_distance(self, vector: object) -> Expression:
return self._distance('<+>', vector)