sms_to_mum.py

import openai
import os

from dotenv import load_dotenv, find_dotenv

load_dotenv(find_dotenv())

openai.api_key = os.getenv("OPENAI_API_KEY")


def get_completion(prompt, model="gpt-4"):
    messages = [{"role": "user", "content": prompt}]
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=0.9,
    )
    return response.choices[0].message["content"]


instructions = f"""
Write an SMS to my mum, which lives far away from me.
Tell her I'm fine and ask her how she is doing.
Say something nice to her in a concise way. 
The greeting should be very short like "Hi mum!".
Don't say that you're checking in - get to the point.
The message should be between 80 and 160 characters long.
Sign the message at the end with "Anna".
"""
message = get_completion(instructions)
print(message)