mirror of
https://github.com/marcogll/talia_bot.git
synced 2026-01-13 21:35:19 +00:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
# app/llm.py
|
|
# Este script se encarga de la comunicación con la inteligencia artificial de OpenAI.
|
|
|
|
import openai
|
|
from config import OPENAI_API_KEY
|
|
|
|
def get_smart_response(prompt):
|
|
"""
|
|
Genera una respuesta inteligente usando la API de OpenAI.
|
|
|
|
Parámetros:
|
|
- prompt: El texto o pregunta que le enviamos a la IA.
|
|
"""
|
|
# Verificamos que tengamos la llave de la API configurada
|
|
if not OPENAI_API_KEY:
|
|
return "Error: La llave de la API de OpenAI no está configurada."
|
|
|
|
try:
|
|
# Creamos el cliente de OpenAI
|
|
client = openai.OpenAI(api_key=OPENAI_API_KEY)
|
|
|
|
# Solicitamos una respuesta al modelo GPT-3.5-turbo
|
|
response = client.chat.completions.create(
|
|
model="gpt-3.5-turbo",
|
|
messages=[
|
|
{"role": "system", "content": "Eres un asistente útil."},
|
|
{"role": "user", "content": prompt},
|
|
],
|
|
)
|
|
# Devolvemos el contenido de la respuesta limpia (sin espacios extras)
|
|
return response.choices[0].message.content.strip()
|
|
except Exception as e:
|
|
# Si algo sale mal, devolvemos el error
|
|
return f"Ocurrió un error al comunicarse con OpenAI: {e}"
|