Update model.py

model.py

@@ -1311,7 +1311,15 @@ async def query_document_info(niche_cases, query_word, alternative_query_word, s
             ]
             print(f"Number of tasks to gather: {len(tasks)}")
             try:
-                results = await asyncio.gather(*tasks)
+                #results = await asyncio.gather(*tasks)
+                results = await asyncio.gather(*tasks, return_exceptions=True)
+                print(f"Results: {results}")
+                for result in results:
+                    if isinstance(result, Exception):
+                        print(f"Error in task: {result}")
+                    else:
+                        print(f"Task completed successfully")
+
                 print("Finished gathering results")
             except Exception as e:
                 print(f"Error in gathering: {e}")