OpenAI API Hello World

import os
from dotenv import load_dotenv
from openai import OpenAI

# Load environment variables from .env file
load_dotenv()

# Get OpenAI API key from environment variables
openai_api_key = os.getenv('OPENAI_API_KEY')

# Initialize OpenAI client with API key
client = OpenAI(api_key=openai_api_key)

# Define the company name to get information for
company_name = "Twilio"

# Request completion from the OpenAI API
completion = client.chat.completions.create(
    model="gpt-4-turbo",
    messages=[
        {"role": "system", "content": "Only return the information asked for. No preamble, and no conclusion. Only return CSV format."},
        {"role": "user", "content": (
            f"Return the details for the company in JSON format including founding date, status "
            f"(series a, b funding etc., IPO etc.), a brief description of what {company_name} does, "
            f"and its category in JSON format. Use these keys - company, year, description, stage, category"
        )}
    ]
)

# Print the raw completion response
print("Raw completion response:", completion)

# Print the content of the first choice's message
print("Content of the first choice's message:", completion.choices[0].message.content)