bupa1018 commited on
Commit
259728c
·
1 Parent(s): fed7cd4

Update ragchain.py

Browse files
Files changed (1) hide show
  1. 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