From 2695301c9456b520cd932d8bc7185aac34fe4a6c Mon Sep 17 00:00:00 2001 From: Vincent Cloutier Date: Wed, 17 Apr 2024 13:37:06 -0400 Subject: [PATCH] update azure openai arguments --- .../partner_providers/openai.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py index 382a480e1..a1347073f 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py @@ -75,23 +75,17 @@ class AzureChatOpenAIProvider(BaseProvider, AzureChatOpenAI): id = "azure-chat-openai" name = "Azure OpenAI" models = ["*"] - model_id_key = "deployment_name" + model_id_key = "azure_deployment" model_id_label = "Deployment name" pypi_package_deps = ["langchain_openai"] + # Confusingly, langchain uses both OPENAI_API_KEY and AZURE_OPENAI_API_KEY for azure + # https://github.com/langchain-ai/langchain/blob/f2579096993ae460516a0aae1d3e09f3eb5c1772/libs/partners/openai/langchain_openai/llms/azure.py#L85 auth_strategy = EnvAuthStrategy(name="AZURE_OPENAI_API_KEY") registry = True fields = [ - TextField( - key="openai_api_base", label="Base API URL (required)", format="text" - ), - TextField( - key="openai_api_version", label="API version (required)", format="text" - ), - TextField( - key="openai_organization", label="Organization (optional)", format="text" - ), - TextField(key="openai_proxy", label="Proxy (optional)", format="text"), + TextField(key="azure_endpoint", label="Base API URL (required)", format="text"), + TextField(key="api_version", label="API version (required)", format="text"), ]