diff --git a/litellm/main.py b/litellm/main.py index 63649844a..76742eb06 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -563,6 +563,7 @@ def completion( "caching_groups", "ttl", "cache", + "no-log", ] default_params = openai_params + litellm_params non_default_params = { @@ -2417,6 +2418,7 @@ def embedding( "caching_groups", "ttl", "cache", + "no-log", ] default_params = openai_params + litellm_params non_default_params = { diff --git a/litellm/utils.py b/litellm/utils.py index 38836a4bc..01fab699a 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2985,12 +2985,15 @@ def client(original_function): print_verbose( f"Async Wrapper: Completed Call, calling async_success_handler: {logging_obj.async_success_handler}" ) - asyncio.create_task( - logging_obj.async_success_handler(result, start_time, end_time) - ) - threading.Thread( - target=logging_obj.success_handler, args=(result, start_time, end_time) - ).start() + # check if user does not want this to be logged + if kwargs.get("no-log", False) == False: + asyncio.create_task( + logging_obj.async_success_handler(result, start_time, end_time) + ) + threading.Thread( + target=logging_obj.success_handler, + args=(result, start_time, end_time), + ).start() # RETURN RESULT if hasattr(result, "_hidden_params"):