Saif Rehman Nasir
commited on
Commit
•
b1fbaf9
1
Parent(s):
8db9985
Increase communities to 3
Browse files
rag.py
CHANGED
@@ -267,7 +267,7 @@ def global_retriever(query: str, level: int, response_type: str):
|
|
267 |
# print(community_data)
|
268 |
intermediate_results = []
|
269 |
i = 0
|
270 |
-
for community in tqdm(community_data[:
|
271 |
intermediate_response = map_chain.invoke(
|
272 |
{"question": query, "context_data": community["output"]}
|
273 |
)
|
|
|
267 |
# print(community_data)
|
268 |
intermediate_results = []
|
269 |
i = 0
|
270 |
+
for community in tqdm(community_data[:3], desc="Processing communities"):
|
271 |
intermediate_response = map_chain.invoke(
|
272 |
{"question": query, "context_data": community["output"]}
|
273 |
)
|