Update rag_api.py

rag_api.py

@@ -7,9 +7,8 @@ from pydantic import BaseModel
 from langchain_huggingface import HuggingFaceEmbeddings
 # from langchain_community.chains import RetrievalQA
 # from langchain.schema.runnable import RunnablePassthrough
-#from langchain.prompts import PromptTemplate
 from langchain_core.runnables import RunnablePassthrough
-from langchain_core.prompts import PromptTemplate
+from langchain.prompts import PromptTemplate
 from langchain_groq import ChatGroq
 
 
@@ -133,7 +132,7 @@ def load_and_configure_rag():
             | RAG_PROMPT
             | llm_groq
         )
-        return qa_chain
+        return qa_chain,retriever
         
     except Exception as e:
         # Imprime un mensaje de error claro en los logs
@@ -153,9 +152,9 @@ app = FastAPI(title="NutriActive RAG API")
 qa_chain = None # Inicializamos a None
 
 try:
-    qa_chain = load_and_configure_rag() 
+    qa_chain,retriever = load_and_configure_rag() 
 except RuntimeError:
-    # Si load_and_configure_rag falla con RuntimeError, qa_chain se queda en None, 
+    # Si load,_and_configure_rag falla con RuntimeError, qa_chain se queda en None, 
     # y el endpoint / lo reportará. No es necesario hacer nada más aquí.
     pass