Hemang Thakur commited on
Commit
d1249fd
·
1 Parent(s): 6e358d5

fixed index out of range error

Browse files
Files changed (1) hide show
  1. main.py +4 -5
main.py CHANGED
@@ -376,14 +376,13 @@ async def process_query(user_query: str, sse_queue: asyncio.Queue):
376
  chunk_counter += 1
377
 
378
  sources_for_answer = []
379
- for idx, result in enumerate(search_results, 1):
380
- if search_contents[idx-1]: # Only include if content was successfully fetched
381
  sources_for_answer.append({
382
  "id": idx,
383
  "title": result.get('title', 'No Title'),
384
- "link": result.get('link', 'No URL')
385
- }
386
- )
387
 
388
  await sse_queue.put(("final_message", response))
389
  await sse_queue.put(("final_sources", json.dumps(sources_for_answer)))
 
376
  chunk_counter += 1
377
 
378
  sources_for_answer = []
379
+ for idx, (result, content) in enumerate(zip(search_results, search_contents), 1):
380
+ if content: # Only include if content was successfully fetched
381
  sources_for_answer.append({
382
  "id": idx,
383
  "title": result.get('title', 'No Title'),
384
+ "link": result.get('link', 'No URL')
385
+ })
 
386
 
387
  await sse_queue.put(("final_message", response))
388
  await sse_queue.put(("final_sources", json.dumps(sources_for_answer)))