From 927f9979006b463f2317260ac874c81c45e61aac Mon Sep 17 00:00:00 2001 From: Brynn Yin Date: Wed, 29 Nov 2023 12:57:13 +0800 Subject: [PATCH] Fix bug Signed-off-by: Brynn Yin --- examples/flows/standard/basic-with-connection/hello.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/examples/flows/standard/basic-with-connection/hello.py b/examples/flows/standard/basic-with-connection/hello.py index 72906a302fd..06971ccef90 100644 --- a/examples/flows/standard/basic-with-connection/hello.py +++ b/examples/flows/standard/basic-with-connection/hello.py @@ -20,14 +20,16 @@ def get_client(connection: Union[CustomConnection, AzureOpenAIConnection]): "Please upgrade your OpenAI package to version >= 1.0.0 or using the command: pip install --upgrade openai." ) # connection can be extract as a dict object contains the configs and secrets - connection_dict = dict(connection) if isinstance(connection, CustomConnection): + connection_dict = dict(connection) from openai import OpenAI as Client elif isinstance(connection, AzureOpenAIConnection): from openai import AzureOpenAI as Client - connection_dict["azure_endpoint"] = connection.api_base - connection_dict.pop("api_base") - connection_dict.pop("api_type") + connection_dict = { + "azure_endpoint": connection.api_base, + "api_key": connection.api_key, + "api_version": connection.api_version, + } else: raise ValueError(f"Unsupported connection type {type(connection)}") return Client(**connection_dict)