@@ -349,25 +349,25 @@ class InstructorEmbeddings(BaseEmbedding):
349349 self ._model = INSTRUCTOR(instructor_model_name)
350350 self ._instruction = instruction
351351
352- def _get_query_embedding (self , query : str ) -> List[float ]:
353- embeddings = self ._model.encode([[self ._instruction, query]])
354- return embeddings[0 ]
352+ def _get_query_embedding (self , query : str ) -> List[float ]:
353+ embeddings = self ._model.encode([[self ._instruction, query]])
354+ return embeddings[0 ]
355355
356- def _get_text_embedding (self , text : str ) -> List[float ]:
357- embeddings = self ._model.encode([[self ._instruction, text]])
358- return embeddings[0 ]
356+ def _get_text_embedding (self , text : str ) -> List[float ]:
357+ embeddings = self ._model.encode([[self ._instruction, text]])
358+ return embeddings[0 ]
359359
360- def _get_text_embeddings (self , texts : List[str ]) -> List[List[float ]]:
361- embeddings = self ._model.encode(
362- [[self ._instruction, text] for text in texts]
363- )
364- return embeddings
360+ def _get_text_embeddings (self , texts : List[str ]) -> List[List[float ]]:
361+ embeddings = self ._model.encode(
362+ [[self ._instruction, text] for text in texts]
363+ )
364+ return embeddings
365365
366- async def _get_query_embedding (self , query : str ) -> List[float ]:
367- return self ._get_query_embedding(query)
366+ async def _aget_query_embedding (self , query : str ) -> List[float ]:
367+ return self ._get_query_embedding(query)
368368
369- async def _get_text_embedding (self , text : str ) -> List[float ]:
370- return self ._get_text_embedding(text)
369+ async def _aget_text_embedding (self , text : str ) -> List[float ]:
370+ return self ._get_text_embedding(text)
371371```
372372
373373## Standalone Usage
0 commit comments