For OpenAI SDK 1+, add deployment for embedding (#34078)

YusakuNo1 · web-flow · commit c09c76b2aa62 · 2024-01-31T10:25:11.000-08:00
diff --git a/sdk/ai/azure-ai-generative/azure/ai/generative/index/_embeddings/openai.py b/sdk/ai/azure-ai-generative/azure/ai/generative/index/_embeddings/openai.py
@@ -79,6 +79,7 @@ def __init__(
                     api_key=self.api_key,
                     api_version=self.api_version,
                     azure_endpoint=self.api_base,
+                    azure_deployment=self.deployment,
                     default_headers={USER_AGENT_HEADER_KEY: USER_AGENT},
                 )
             else:
diff --git a/sdk/ai/azure-ai-resources/azure/ai/resources/_index/_embeddings/openai.py b/sdk/ai/azure-ai-resources/azure/ai/resources/_index/_embeddings/openai.py
@@ -78,6 +78,7 @@ def __init__(
                     api_key=self.api_key,
                     api_version=self.api_version,
                     azure_endpoint=self.api_base,
+                    azure_deployment=self.deployment,
                     default_headers={USER_AGENT_HEADER_KEY: USER_AGENT},
                 )
             else:

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,7 @@ def __init__(`
`79`	`79`	`api_key=self.api_key,`
`80`	`80`	`api_version=self.api_version,`
`81`	`81`	`azure_endpoint=self.api_base,`
	`82`	`+ azure_deployment=self.deployment,`
`82`	`83`	`default_headers={USER_AGENT_HEADER_KEY: USER_AGENT},`
`83`	`84`	`)`
`84`	`85`	`else:`
Original file line number	Diff line number	Diff line change
`@@ -78,6 +78,7 @@ def __init__(`
`78`	`78`	`api_key=self.api_key,`
`79`	`79`	`api_version=self.api_version,`
`80`	`80`	`azure_endpoint=self.api_base,`
	`81`	`+ azure_deployment=self.deployment,`
`81`	`82`	`default_headers={USER_AGENT_HEADER_KEY: USER_AGENT},`
`82`	`83`	`)`
`83`	`84`	`else:`