Hemang Thakur
commited on
Commit
·
d1249fd
1
Parent(s):
6e358d5
fixed index out of range error
Browse files
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
|
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)))
|