tools.py

# -*- 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