diff --git a/langchain/embeddings/openai.py b/langchain/embeddings/openai.py index 8e028e52bdb..c7ebf277e87 100644 --- a/langchain/embeddings/openai.py +++ b/langchain/embeddings/openai.py @@ -253,7 +253,7 @@ class OpenAIEmbeddings(BaseModel, Embeddings): @property def _invocation_params(self) -> Dict: openai_args = { - "engine": self.deployment, + "model": self.model, "request_timeout": self.request_timeout, "headers": self.headers, "api_key": self.openai_api_key, @@ -262,6 +262,8 @@ class OpenAIEmbeddings(BaseModel, Embeddings): "api_type": self.openai_api_type, "api_version": self.openai_api_version, } + if self.openai_api_type in ("azure", "azure_ad", "azuread"): + openai_args["engine"] = self.deployment if self.openai_proxy: import openai