fixing llama_rag demo

mgonzs13 · Oct 27, 2024 · 4facfa5 · 4facfa5
1 parent fca5864
commit 4facfa5
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/llama_demos/llama_demos/llama_rag_demo_node.py b/llama_demos/llama_demos/llama_rag_demo_node.py
@@ -61,7 +61,7 @@
     [
         SystemMessage("You are an AI assistant that answer questions."),
         HumanMessagePromptTemplate.from_template(
-            "Taking into account the followin context:\n{context}\nAnswer this question: {question}"
+            "Taking into account the followin context:{context}\n\nAnswer this question: {question}"
         ),
     ]
 )
@@ -74,7 +74,12 @@
 
 
 def format_docs(docs):
-    return "\n\n\t- ".join(doc.page_content for doc in docs)
+    formated_docs = ""
+
+    for d in docs:
+        formated_docs += f"\n\n\t- {d.page_content}"
+
+    return formated_docs
 
 
 # create and use the chain