Skip to content

Commit cb785b8

Browse files
authored
Fix incorrect embeddings.md custom embedding example (#20143)
Update embeddings.md
1 parent 4a4bbae commit cb785b8

File tree

1 file changed

+15
-15
lines changed
  • docs/src/content/docs/framework/module_guides/models

1 file changed

+15
-15
lines changed

docs/src/content/docs/framework/module_guides/models/embeddings.md

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)