testai.py
#Note: The openai-python library support for Azure OpenAI is in preview.
import os
import openai
openai.api_type = "azure"
openai.api_base = os.getenv("AZURE_OPENAI_ENDPOINT")
openai.api_version = "2023-05-15"
openai.api_key = os.getenv("AZURE_OPENAI_KEY")
response = openai.ChatCompletion.create(
engine="aiplus", # engine = "deployment_name".
messages=[
{
"role": "system", "content": "You are a helpful assistant."},
{
"role": "user", "content": "Does Azure OpenAI support customer managed keys?"},
{
"role": "assistant", "content": "Yes, customer managed keys are supported by Azure OpenAI."},
{
"role": "user", "content": "Do other Azure Cognitive Services support this too?"}
]
)
print(response)