Browse Source

fix: azure embedding not support batch (#188)

tags/0.2.2
John Wang 2 years ago
parent
commit
d93365d429
No account linked to committer's email address
1 changed files with 14 additions and 0 deletions
  1. 14
    0
      api/core/embedding/openai_embedding.py

+ 14
- 0
api/core/embedding/openai_embedding.py View File

Can be overriden for batch queries. Can be overriden for batch queries.


""" """
if self.openai_api_type and self.openai_api_type == 'azure':
embeddings = []
for text in texts:
embeddings.append(self._get_text_embedding(text))

return embeddings

if self.deployment_name is not None: if self.deployment_name is not None:
engine = self.deployment_name engine = self.deployment_name
else: else:


async def _aget_text_embeddings(self, texts: List[str]) -> List[List[float]]: async def _aget_text_embeddings(self, texts: List[str]) -> List[List[float]]:
"""Asynchronously get text embeddings.""" """Asynchronously get text embeddings."""
if self.openai_api_type and self.openai_api_type == 'azure':
embeddings = []
for text in texts:
embeddings.append(await self._aget_text_embedding(text))

return embeddings

if self.deployment_name is not None: if self.deployment_name is not None:
engine = self.deployment_name engine = self.deployment_name
else: else:

Loading…
Cancel
Save