Skip to content

Commit f1bb1b8

Browse files
update base vectorizer class
1 parent eb8258d commit f1bb1b8

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

redisvl/vectorize/base.py

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,27 +21,35 @@ def set_model(self, model: str, dims: Optional[int] = None) -> None:
2121

2222
def embed_many(
2323
self,
24-
inputs: List[str],
24+
texts: List[str],
2525
preprocess: Optional[Callable] = None,
26-
chunk_size: int = 1000,
26+
batch_size: Optional[int] = 1000,
27+
as_buffer: Optional[bool] = False
2728
) -> List[List[float]]:
2829
raise NotImplementedError
2930

3031
def embed(
31-
self, emb_input: str, preprocess: Optional[Callable] = None
32+
self,
33+
text: str,
34+
preprocess: Optional[Callable] = None,
35+
as_buffer: Optional[bool] = False
3236
) -> List[float]:
3337
raise NotImplementedError
3438

3539
async def aembed_many(
3640
self,
37-
inputs: List[str],
41+
texts: List[str],
3842
preprocess: Optional[Callable] = None,
39-
chunk_size: int = 1000,
43+
batch_size: Optional[int] = 1000,
44+
as_buffer: Optional[bool] = False
4045
) -> List[List[float]]:
4146
raise NotImplementedError
4247

4348
async def aembed(
44-
self, emb_input: str, preprocess: Optional[Callable] = None
49+
self,
50+
text: str,
51+
preprocess: Optional[Callable] = None,
52+
as_buffer: Optional[bool] = False
4553
) -> List[float]:
4654
raise NotImplementedError
4755

0 commit comments

Comments
 (0)