Skip to content

Commit 7a4acc7

Browse files
refactor: Set default task as retrieval passage
1 parent a7c6582 commit 7a4acc7

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

fastembed/text/multitask_embedding.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,12 @@ class Task(int, Enum):
3939

4040

4141
class JinaEmbeddingV3(PooledNormalizedEmbedding):
42-
DEFAULT_TASK = Task.TEXT_MATCHING
4342
PASSAGE_TASK = Task.RETRIEVAL_PASSAGE
4443
QUERY_TASK = Task.RETRIEVAL_QUERY
4544

4645
def __init__(self, *args, **kwargs):
4746
super().__init__(*args, **kwargs)
48-
self._current_task_id = self.DEFAULT_TASK
47+
self._current_task_id = self.PASSAGE_TASK
4948

5049
@classmethod
5150
def _get_worker_class(cls) -> Type["TextEmbeddingWorker"]:
@@ -66,7 +65,7 @@ def embed(
6665
documents: Union[str, Iterable[str]],
6766
batch_size: int = 256,
6867
parallel: Optional[int] = None,
69-
task_id: int = DEFAULT_TASK,
68+
task_id: int = PASSAGE_TASK,
7069
**kwargs,
7170
) -> Iterable[np.ndarray]:
7271
self._current_task_id = task_id

0 commit comments

Comments
 (0)