File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -39,13 +39,12 @@ class Task(int, Enum):
3939
4040
4141class 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
You can’t perform that action at this time.
0 commit comments