Ajustes en views.py para integrar correctamente FAISS y OpenAI

Blaister9 · Oct 3, 2024 · 3f890e0 · 3f890e0
1 parent 5249509
commit 3f890e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/chatbot_laura/views.py b/backend/chatbot_laura/views.py
@@ -37,7 +37,7 @@ def get_embedding(self, text, model="text-embedding-ada-002"):
         try:
             # Usamos directamente la función de OpenAI para obtener embeddings
             response = client.embeddings.create(input=text, model=model)
-            return response['data'][0]['embedding']
+            return response.data[0].embedding
         except Exception as e:
             logger.error(f"Error al obtener embedding: {str(e)}")
             raise