khadijaaao commited on
Commit
f7a8ca8
·
verified ·
1 Parent(s): 4dd18be

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -2
src/streamlit_app.py CHANGED
@@ -119,8 +119,7 @@ if prompt := st.chat_input("Posez votre question ici..."):
119
  with st.spinner("Je réfléchis..."):
120
  try:
121
  docs = retriever.invoke(prompt)
122
- context = "\n\n".join([doc.page_content for doc in docs])
123
-
124
  # Création du prompt pour CTransformers
125
  full_prompt = f"System: Vous êtes un coach expert. Répondez à la question en vous basant UNIQUEMENT sur le contexte fourni. Ne mentionnez pas le contexte dans votre réponse.\n\nContexte:\n{context}\n\nQuestion: {prompt}\n\nRéponse:"
126
 
 
119
  with st.spinner("Je réfléchis..."):
120
  try:
121
  docs = retriever.invoke(prompt)
122
+ context = "\n".join([doc.dict().get("page_content", "") for doc in docs])
 
123
  # Création du prompt pour CTransformers
124
  full_prompt = f"System: Vous êtes un coach expert. Répondez à la question en vous basant UNIQUEMENT sur le contexte fourni. Ne mentionnez pas le contexte dans votre réponse.\n\nContexte:\n{context}\n\nQuestion: {prompt}\n\nRéponse:"
125