Refactor code to include document formatting in ChatGroq

app.py

@@ -130,7 +130,9 @@ prompt = PromptTemplate.from_template(
 5. 질문에 대한 답변이 문서에 부분적으로만 포함되어 있는 경우, 가능한 한 많은 정보를 종합하여 답변해 주세요. 또한, 추가적인 연구나 참고자료가 필요할 수 있음을 언급해 주세요.
 
 #참고문서:
+\"\"\"
 {context}
+\"\"\"
 
 #질문:
 {question}
@@ -158,8 +160,18 @@ llm = ChatGroq(
     ),
 )
 
+def format_docs(docs):
+    formatted_docs = []
+    for doc in docs:
+        formatted_doc = f"Page Content:\n{doc.page_content}\n"
+        if doc.metadata.get("source"):
+            formatted_doc += f"Source: {doc.metadata['source']}\n"
+        formatted_docs.append(formatted_doc)
+    return "\n---\n".join(formatted_docs)
+
+
 rag_chain = (
-    {"context": compression_retriever, "question": RunnablePassthrough()}
+    {"context": compression_retriever | format_docs, "question": RunnablePassthrough()}
     | prompt
     | llm
     | StrOutputParser()