1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
"""
FILE: completions_aoai_quickstart.py
DESCRIPTION:
This sample demonstrates how to get started with Azure OpenAI Completions using the official OpenAI SDK for Python.
USAGE:
python completions_aoai_quickstart.py
Before running the sample:
pip install openai
Set the environment variables with your own values:
1) AZURE_OPENAI_KEY - your Azure OpenAI API key.
2) AZURE_OPENAI_ENDPOINT - the endpoint to your Azure OpenAI resource.
3) AZURE_OPENAI_COMPLETIONS_DEPLOYMENT - the deployment name you chose when deploying your model.
"""
# These lines are intentionally excluded from the sample code, we use them to configure any vars
# or to tweak usage in ways that keep samples looking consistent when rendered in docs and tools
import os
os.environ["AZURE_OPENAI_ENDPOINT"] = os.environ["AZ_OPENAI_ENDPOINT"]
os.environ["AZURE_OPENAI_COMPLETIONS_DEPLOYMENT"] = "gpt-35-turbo-instruct"
def completions_aoai_quickstart() -> None:
import os
from openai import AzureOpenAI
from azure.identity import DefaultAzureCredential, get_bearer_token_provider
token_provider = get_bearer_token_provider(
DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default"
)
client = AzureOpenAI(
azure_ad_token_provider=token_provider,
azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
api_version=os.environ["API_VERSION_GA"]
)
#This will correspond to the custom name you chose for your deployment when you deployed a model.
# Use a gpt-35-turbo-instruct deployment.
deployment_name=os.environ["AZURE_OPENAI_COMPLETIONS_DEPLOYMENT"]
# Send a completion call to generate an answer
prompt = 'Write a tagline for an ice cream shop.'
response = client.completions.create(model=deployment_name, prompt=prompt, max_tokens=10)
print(prompt + response.choices[0].text)
if __name__ == '__main__':
completions_aoai_quickstart()
|
