From 52495098b2a4fbbe2714b1c266dcf1992004be34 Mon Sep 17 00:00:00 2001 From: Blaister9 Date: Wed, 2 Oct 2024 21:50:16 -0500 Subject: [PATCH] Ajustes en views.py para integrar correctamente FAISS y OpenAI --- backend/chatbot_laura/views.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/backend/chatbot_laura/views.py b/backend/chatbot_laura/views.py index a8a8407..20c68ed 100644 --- a/backend/chatbot_laura/views.py +++ b/backend/chatbot_laura/views.py @@ -11,7 +11,7 @@ import numpy as np import faiss import pickle -import openai # Importa OpenAI aquí +from openai import OpenAI from dotenv import load_dotenv # Configurar logging @@ -19,9 +19,8 @@ logger = logging.getLogger(__name__) load_dotenv() +client = OpenAI(api_key=os.getenv('OPENAI_API_KEY')) # Configura tu clave de API -# Configura la API Key de OpenAI -openai.api_key = os.getenv('OPENAI_API_KEY') class ChatbotLauraView: def __init__(self): @@ -37,7 +36,7 @@ def __init__(self): def get_embedding(self, text, model="text-embedding-ada-002"): try: # Usamos directamente la función de OpenAI para obtener embeddings - response = openai.Embedding.create(input=text, model=model) + response = client.embeddings.create(input=text, model=model) return response['data'][0]['embedding'] except Exception as e: logger.error(f"Error al obtener embedding: {str(e)}")