"""
Data Scientist.: Dr.Eddy Giusepe Chirinos Isidro

Objetivo: Aqui vamos aprender a usar a API da OpenAI.
          Nos conectamos com a API e logo escrevemos
          a query dentro da nossa função e printamos 
          para ver a resposta no Terminal 🤗.
"""
import os
import openai


from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv()) # read local .env file
openai.api_key  = os.getenv('OPENAI_API_KEY')

def get_completion(query, model="gpt-3.5-turbo", temperature=0):
    
    messages = [{"role": "user", "content": query}]
    
    response = openai.ChatCompletion.create(model=model,
                                            messages=messages,
                                            temperature=temperature,
                                           )

    return response.choices[0].message["content"]

# Aqui você realiza a sua query, dentro da função 🤗:
response = get_completion("Qual é a capital de Perú?")
print(response)