From ead25a772a798fc8366abbd484645019ab1f59f7 Mon Sep 17 00:00:00 2001 From: SilasMarvin <19626586+SilasMarvin@users.noreply.github.com> Date: Tue, 2 Jan 2024 13:34:22 -0800 Subject: [PATCH] Store errors in thread status --- pgml-extension/src/bindings/transformers/transformers.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pgml-extension/src/bindings/transformers/transformers.py b/pgml-extension/src/bindings/transformers/transformers.py index 83608ed48..83d953fdd 100644 --- 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):