Testando API da OpenAI no terminal mesmo

main-1.py

@@ -0,0 +1,30 @@
+"""
+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)