test bug fix

Author: Darshan808

packages/jupyter-ai/jupyter_ai/tests/completions/test_handlers.py

@@ -1,5 +1,4 @@
 import json
-import traceback
 from types import SimpleNamespace
 from typing import Union
 
@@ -27,9 +26,9 @@ def __init__(self, **kwargs):
         if "responses" not in kwargs:
             kwargs["responses"] = ["Test response"]
         super().__init__(**kwargs)
-
-    def _acall(self, *args, **kwargs):
-        if self.raise_exc:
+    
+    async def _acall(self, *args, **kwargs):
+        if hasattr(self, 'raise_exc') and self.raise_exc:
             raise Exception("Test exception")
         else:
             return super()._call(*args, **kwargs)