exercise.py

import openai
### CODIO SOLUTION BEGIN
import os
from openai import OpenAI

# Configure the OpenAI client with an API key
client = OpenAI()

def fix_spelling(input_text):
    """Correct spelling mistakes in the input text using the gpt-4o-mini model."""
    completion = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "system", "content": "You are a helpful assistant great at fixing spelling errors."},
            {"role": "user", "content": input_text}
        ],
        top_p=0.1
    )
    corrected_text = completion.choices[0].message.content.strip()
    return corrected_text

def get_emb(input_text):
    """Return the embedding of the input text using the text-embedding-3-small model."""
    emb_response = client.embeddings.create(
        input=input_text,
        model="text-embedding-3-small"
    )
    embedding = emb_response.data[0].embedding
    return embedding
### CODIO SOLUTION END


# FREEZE CODE BEGIN
corrected_text = fix_spelling("Ths sentnce has some speling errors.")
print(corrected_text)

embedding = get_emb(corrected_text)
input_texts = [
    "I hve a bananna for brekfast.",
    "The colur of the sky is rde.",
    "The grass is greneer on the other side."
]

for input_text in input_texts:
    output_text = fix_spelling(input_text)
    print(output_text)

# FREEZE CODE END