Store errors in thread status (postgresml#1263)

open-prophetdb · Jan 2, 2024 · 689ac9e · 689ac9e
1 parent 75d341b
commit 689ac9e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pgml-extension/src/bindings/transformers/transformers.py b/pgml-extension/src/bindings/transformers/transformers.py
@@ -187,10 +187,9 @@ def streaming_worker(worker_threads, model, **kwargs):
         worker_threads.update_thread(thread_id, "Error setting data")
     try:
         model.generate(**kwargs)
-    except BaseException as error:
-        print(f"Error in streaming_worker: {error}", file=sys.stderr)
-    finally:
         worker_threads.delete_thread(thread_id)
+    except BaseException as error:
+        worker_threads.update_thread(thread_id, f"Error in streaming_worker: {error}")
 
 
 class GGMLPipeline(object):