diff --git a/pilot/utils/llm_connection.py b/pilot/utils/llm_connection.py index 62f40e9..c563cec 100644 --- a/pilot/utils/llm_connection.py +++ b/pilot/utils/llm_connection.py @@ -162,7 +162,7 @@ def retry_on_exception(func): args[0]['function_buffer'] = e.doc continue elif isinstance(e, ValidationError): - logger.warn('Received invalid JSON response from LLM. Asking to retry...') + logger.warning('Received invalid JSON response from LLM. Asking to retry...') logger.info(f' at {e.json_path} {e.message}') # eg: # json_path: '$.type'