diff --git a/litellm/tests/test_split_test.py b/litellm/tests/test_split_test.py index 4c195d939..c995a0d88 100644 --- a/litellm/tests/test_split_test.py +++ b/litellm/tests/test_split_test.py @@ -19,6 +19,8 @@ messages = [{ "content": "Hello, how are you?","role": "user"}] # print(completion_with_split_tests(models=split_per_model, messages=messages)) -# test with client +# test +print(completion_with_split_tests(models=split_per_model, messages=messages)) -print(completion_with_split_tests(models=split_per_model, messages=messages)) \ No newline at end of file +# test with client, without id +print(completion_with_split_tests(models=split_per_model, messages=messages, use_client=True)) diff --git a/litellm/utils.py b/litellm/utils.py index f8bd4bb56..68076023c 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1990,7 +1990,8 @@ def completion_with_split_tests(models={}, messages=[], use_client=False, overri model_configs = {} if use_client and not override_client: if "id" not in kwargs or kwargs["id"] is None: - raise ValueError("Please tag this completion call, if you'd like to update it's split test values through the UI. - eg. `completion_with_split_tests(.., id=1234)`.") + kwargs["id"] = str(uuid.uuid4()) + #raise ValueError("Please tag this completion call, if you'd like to update it's split test values through the UI. - eg. `completion_with_split_tests(.., id=1234)`.") # get the most recent model split list from server models, model_configs = get_model_split_test(models=models, completion_call_id=kwargs["id"])