| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- from typing import List
- from fastapi import HTTPException
- from openai import OpenAI
- from config.settings import OPENAI_API_KEY
- from models.chat import Message
- from services.data_service import data_bg_loaded
- from logging import getLogger
- # Initialize OpenAI client
- openai_client = OpenAI(api_key=OPENAI_API_KEY)
- logger = getLogger(__name__)
- async def generate_completion(messages_array: List[Message], session_id: str) -> str:
- """Generate OpenAI chat completion"""
- if not OPENAI_API_KEY:
- logger.error("Error: OpenAI API key is not configured.")
- raise HTTPException(status_code=500, detail="OpenAI API key not configured on server.")
- logger.info(f"[OpenAI Service Python] Session/Token {session_id} sent: {[msg.model_dump() for msg in messages_array]}")
- data_for_prompt = [
- f'{{"pregunta": "{item.get("q", "")}", "respuesta": "{item.get("ans", "")}"}}'
- for item in data_bg_loaded
- ]
- data_string = "\n".join(data_for_prompt)
- preprompt = f"""
- Eres un asistente de el bar klein, tu nombre es camilo klein, usas emojis para responder.
- y ser carismatico con el cliente.
- tus responsabilidades son:
- - Responder preguntas sobre el menu de el bar klein
- - Proporcionar información sobre el menú de el bar klein
- - Proporcionar recomendaciones sobre el menú de el bar klein
- - Proporcionar información sobre la comida de el bar klein
- - No puedes tomar pedidos de clientes, solo informar
- - Debes evadir cualquier pregunta que no sea relacionada con el bar klein
- para esto usaras los siguientes datos:
- {data_string}
- """
- processed_messages: List[dict] = [{"role": "system", "content": preprompt}]
- processed_messages.extend([msg.model_dump() for msg in messages_array])
- try:
- completion = openai_client.chat.completions.create(
- model="gpt-4o-mini",
- messages=processed_messages, # type: ignore (OpenAI lib expects list of specific dicts)
- temperature=0.3,
- )
- response_content = completion.choices[0].message.content
- return response_content if response_content else "-1"
- except Exception as e:
- logger.error(f"Error calling OpenAI: {e}")
- raise HTTPException(status_code=500, detail="Error al procesar tu solicitud con OpenAI.")
|