fix: catch CancellEderror

Author: ChenZiHong-Gavin

graphgen/models/llm/local/vllm_wrapper.py

@@ -89,8 +89,7 @@ async def generate_answer(
             result_text = final_output.outputs[0].text
             return result_text
 
-        except Exception as e:
-            print(f"Error in generate_answer: {e}")
+        except (Exception, asyncio.CancelledError):
             await self.engine.abort(request_id)
             raise
 
@@ -142,8 +141,7 @@ async def generate_topk_per_token(
                 return [main_token]
             return []
 
-        except Exception as e:
-            print(f"Error in generate_topk_per_token: {e}")
+        except (Exception, asyncio.CancelledError):
             await self.engine.abort(request_id)
             raise