response instead of everything

app.py

@@ -35,14 +35,7 @@ def query_rag(study_name: str, question: str, prompt_type: str) -> str:
     # Use the prepared context in the query
     response = rag.query(question, prompt_template=prompt)
 
-    # Format the response as Markdown
-    formatted_response = f"## Question\n\n{question}\n\n## Answer\n\n{response['answer']}\n\n## Sources\n\n"
-    for source in response["sources"]:
-        formatted_response += (
-            f"- {source['title']} ({source.get('year', 'Year not specified')})\n"
-        )
-
-    return formatted_response
+    return response.response
 
 
 def get_study_info(study_name):

rag/rag_pipeline.py

@@ -27,8 +27,8 @@ class RAGPipeline:
             for index, doc_data in enumerate(self.data):
                 doc_content = (
                     f"Title: {doc_data['title']}\n"
+                    f"Abstract: {doc_data['abstract']}\n"
                     f"Authors: {', '.join(doc_data['authors'])}\n"
-                    f"Full Text: {doc_data['full_text']}"
                 )
 
                 metadata = {
@@ -37,6 +37,7 @@ class RAGPipeline:
                     "authors": doc_data.get("authors", []),
                     "year": doc_data.get("year"),
                     "doi": doc_data.get("doi"),
+                    "full_text": doc_data.get("full_text"),
                 }
 
                 self.documents.append(
@@ -89,7 +90,4 @@ class RAGPipeline:
 
         response = query_engine.query(context)
 
-        return {
-            "answer": response.response,
-            "sources": [node.metadata for node in response.source_nodes],
-        }
+        return response