From 44cdea03813ee9ecc342d701ab5c27feac4f2d66 Mon Sep 17 00:00:00 2001 From: Nicholas Albion Date: Fri, 6 Oct 2023 14:21:56 +1100 Subject: [PATCH] renamed `wrapper()` `retry_wrapper()` so that it's more obvious in logs --- pilot/utils/llm_connection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pilot/utils/llm_connection.py b/pilot/utils/llm_connection.py index 32cc01a..5b294cf 100644 --- a/pilot/utils/llm_connection.py +++ b/pilot/utils/llm_connection.py @@ -156,7 +156,7 @@ def retry_on_exception(func): if 'function_buffer' in args[0]: del args[0]['function_buffer'] - def wrapper(*args, **kwargs): + def retry_wrapper(*args, **kwargs): while True: try: # spinner_stop(spinner) @@ -237,7 +237,7 @@ def retry_on_exception(func): if user_message != '': return {} - return wrapper + return retry_wrapper @retry_on_exception