Spaces:
Sleeping
Sleeping
File size: 1,613 Bytes
312c075 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
# -*- coding: utf-8 -*-
"""tools.ipynb
Automatically generated by Colaboratory.
Original file is located at
https://colab.research.google.com/drive/1bf1JxvWIpqRY6Q36aazPA2KALPqrvVAC
"""
import os
from dotenv import load_dotenv
import openai
# Cargar variables de entorno
load_dotenv()
OPENAI_KEY = os.getenv('OPENAI_KEY', '')
# Asignar la API Key
openai.api_key = OPENAI_KEY
def mychatbot(messages):
# Chatbot que hace consultas "query" a una base de conocimiento "contract_knowledge"
# Enviar solicitud a la api OpenAI con el modelo "GPT-3.5-turbo"
res = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages = messages
)
# Del diccionario extraer la informacion correspondiente al id "content"
conclusion = res['choices'][0]['message']['content']
return conclusion
def ask_chatbot(question):
prompt = f"""
Dada la siguiente pregunta realizada por el usuario: {question}.
Reponde de manera efusiva y tajante la conversación.
Relaciona todo con la serie animada "Los Simpsons", con una frase, o con un personaje, o con una situacion o con un capitulo. Si puedes agrega un link de youtube donde se pueda ver lo que acabas de mencionar.
"""
# Dialogo con ChatGPT
messages = [
{"role": "system", "content": "Eres un apasionado de 'Los Simpsons'. Has visto todos los capitulos varias veces, te sabes todos los diaglogos de memoria, y cada situacion de la vida la relacionas con 'Los Simpsons'."},
{"role": "user", "content": prompt}
]
# Output
respuesta = mychatbot(messages)
return respuesta |