From 6cb03d7c63944501608ca4ddab02ee10d6803c98 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 9 Sep 2023 13:17:51 -0700 Subject: [PATCH] modification to arg checking --- litellm/utils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 8c93ede2e4c..95553e29945 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -518,7 +518,12 @@ def client(original_function): result = None litellm_call_id = str(uuid.uuid4()) kwargs["litellm_call_id"] = litellm_call_id - check_args(*args, **kwargs) + # check_args(*args, **kwargs) + try: + model = args[0] if len(args) > 0 else kwargs["model"] + except: + raise ValueError("model param not passed in.") + try: logging_obj = function_setup(start_time, *args, **kwargs) kwargs["litellm_logging_obj"] = logging_obj