Saif Rehman Nasir commited on
Commit
b1fbaf9
1 Parent(s): 8db9985

Increase communities to 3

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
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[:2], desc="Processing communities"):
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
  )