PYTHON - OPENAI - API
Simon-Pierre Boucher
2024-09-14
In [1]:
import os
import requests
import json
def call_openai_api(
model="gpt-4",
messages=None,
temperature=1,
max_tokens=2048,
top_p=1,
frequency_penalty=0,
presence_penalty=0
):
# Get the API key from environment variables
api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
raise ValueError("The API key was not found in the environment variables. Please set OPENAI_API_KEY.")
url = "https://api.openai.com/v1/chat/completions"
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {api_key}"
}
body = {
"model": model,
"messages": messages,
"temperature": temperature,
"max_tokens": max_tokens,
"top_p": top_p,
"frequency_penalty": frequency_penalty,
"presence_penalty": presence_penalty
}
response = requests.post(url, headers=headers, data=json.dumps(body))
if response.status_code != 200:
raise ValueError(f"The API request failed: {response.text}")
result = response.json()
return result
In [2]:
# Define the messages
messages = [
{"role": "system", "content": "You are an assistant"},
{"role": "user", "content": "give me a list of gift idea for my girlfriend"}
]
In [3]:
# Call the function
response = call_openai_api(
model="gpt-4",
messages=messages,
temperature=1,
max_tokens=2048,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
# Display the assistant's response
print(response['choices'][0]['message']['content'])
In [ ]: