KMS/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: azure embedding not support batch (#188)

Browse Source
This commit is contained in:
John Wang 2023-05-24 18:55:07 +08:00 committed by GitHub
parent 8b44dba988
commit d93365d429
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/core/embedding/openai_embedding.py
View File

@ -173,6 +173,13 @@ class OpenAIEmbedding(BaseEmbedding):
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:
@ -187,6 +194,13 @@ class OpenAIEmbedding(BaseEmbedding):
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: