Spaces:
Runtime error
Runtime error
Update ragchain.py
Browse files- ragchain.py +1 -31
ragchain.py
CHANGED
|
@@ -125,34 +125,4 @@ class RAGChain:
|
|
| 125 |
Query:
|
| 126 |
{query}
|
| 127 |
"""
|
| 128 |
-
return self.llm.invoke(prompt).content
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
def _format_documents(docs):
|
| 133 |
-
formatted_docs = []
|
| 134 |
-
|
| 135 |
-
for i, doc in enumerate(docs, start=1):
|
| 136 |
-
# Add "Snippet + count of doc" e.g., Snippet 1:
|
| 137 |
-
formatted_docs.append(f"Snippet {i}: \n")
|
| 138 |
-
|
| 139 |
-
# Add one line break
|
| 140 |
-
formatted_docs.append("\n")
|
| 141 |
-
|
| 142 |
-
# Process metadata and add to context
|
| 143 |
-
all_metadata = doc.metadata
|
| 144 |
-
metadata_str = ", ".join(f"{key}: {value}" for key, value in all_metadata.items())
|
| 145 |
-
formatted_docs.append(metadata_str)
|
| 146 |
-
|
| 147 |
-
# Add one line break
|
| 148 |
-
formatted_docs.append("\n")
|
| 149 |
-
|
| 150 |
-
# Append page content
|
| 151 |
-
formatted_docs.append(doc.page_content)
|
| 152 |
-
|
| 153 |
-
# Add two line breaks
|
| 154 |
-
formatted_docs.append("\n\n")
|
| 155 |
-
|
| 156 |
-
return formatted_docs
|
| 157 |
-
|
| 158 |
-
|
|
|
|
| 125 |
Query:
|
| 126 |
{query}
|
| 127 |
"""
|
| 128 |
+
return self.llm.invoke(prompt).content
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|