| @@ -173,6 +173,13 @@ class OpenAIEmbedding(BaseEmbedding): | |||
| 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: | |||
| engine = self.deployment_name | |||
| else: | |||
| @@ -187,6 +194,13 @@ class OpenAIEmbedding(BaseEmbedding): | |||
| async def _aget_text_embeddings(self, texts: List[str]) -> List[List[float]]: | |||
| """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: | |||
| engine = self.deployment_name | |||
| else: | |||