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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
"""
The `vecs.experimental.adapter.noop` module provides a default no-op (no operation) adapter
that passes the inputs through without any modification. This can be useful when no specific
adapter processing is required.
All public classes, enums, and functions are re-exported by `vecs.adapters` module.
"""
from typing import Any, Dict, Generator, Iterable, Optional, Tuple
from .base import AdapterContext, AdapterStep
class NoOp(AdapterStep):
"""
NoOp is a no-operation AdapterStep. It is a default adapter that passes through
the input records without any modifications.
"""
def __init__(self, dimension: int):
"""
Initializes the NoOp adapter with a dimension.
Args:
dimension (int): The dimension of the input vectors.
"""
self._dimension = dimension
@property
def exported_dimension(self) -> Optional[int]:
"""
Returns the dimension of the adapter.
Returns:
int: The dimension of the input vectors.
"""
return self._dimension
def __call__(
self,
records: Iterable[Tuple[str, Any, Optional[Dict]]],
adapter_context: AdapterContext,
) -> Generator[Tuple[str, Any, Dict], None, None]:
"""
Yields the input records without any modification.
Args:
records: Iterable of tuples each containing an id, a media and an optional dict.
adapter_context: Context of the adapter.
Yields:
Tuple[str, Any, Dict]: The input record.
"""
for id, media, metadata in records:
yield (id, media, metadata or {})
|
