From cecf13eb1b9958ae1fa1b71ec6635b6fec9bceaa Mon Sep 17 00:00:00 2001
From: Sachin Bhat <sachubhat17@gmail.com>
Date: Tue, 16 Jul 2024 08:55:06 +0530
Subject: [PATCH] Add error handling

---
 kor/extraction/api.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/kor/extraction/api.py b/kor/extraction/api.py
index 00abf15..c5083d3 100644
--- a/kor/extraction/api.py
+++ b/kor/extraction/api.py
@@ -24,8 +24,10 @@
 def set_verbose_context(verbose: bool):
     old_verbose = get_verbose()
     set_verbose(verbose)
-    yield
-    set_verbose(old_verbose)
+    try:
+        yield
+    finally:
+        set_verbose(old_verbose)
 
 
 async def _extract_from_document_with_semaphore(