To handle responses from OpenAI’s API in Python, you first need to interact with the API in a structured manner. You'll typically use the requests library to send a request to the API endpoint. First, ensure you have the requests library installed in your Python environment. If it's not already installed, you can add it using pip:
pip install requests
Once you have that set up, you'll need to make an API call by preparing your request. This involves sending a POST request to the OpenAI API endpoint with your API key included in the headers. You should also include the payload with your query, which consists of parameters like the model you want to use and the prompt you are providing. Here's a simple example:
import requests
url = "https://api.openai.com/v1/engines/davinci-codex/completions"
headers = {
"Authorization": f"Bearer YOUR_API_KEY",
"Content-Type": "application/json"
}
data = {
"prompt": "What is the capital of France?",
"max_tokens": 50
}
response = requests.post(url, headers=headers, json=data)
After making the request, you need to handle the response appropriately. The API returns a JSON object that contains the result. You can check the HTTP response status to ensure that the request was successful. A response code of 200 indicates success, while any other code could signal an error. To extract the relevant data, you can access the JSON content using the response.json() method, which gives you a Python dictionary. Here’s how you can implement this:
if response.status_code == 200:
result = response.json()
answer = result['choices'][0]['text']
print("Response from OpenAI:", answer.strip())
else:
print("Error:", response.status_code, response.text)
In this example, you check if the status code is 200 before processing the JSON response. If successful, you retrieve the generated text from the choices list in the response. Make sure to handle potential errors and exceptions to make your application robust. This will help you effectively manage the responses from OpenAI’s API and integrate it into your projects.