phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test without id

Browse source
This commit is contained in:
ishaan-jaff 2023-09-02 16:54:04 -07:00
parent e128b272fa
commit e6836985c8
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_split_test.py
View file

@ -19,6 +19,8 @@ messages = [{ "content": "Hello, how are you?","role": "user"}]
# print(completion_with_split_tests(models=split_per_model, messages=messages)) # 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))
# test with client, without id
print(completion_with_split_tests(models=split_per_model, messages=messages, use_client=True))

3
litellm/utils.py
View file

@ -1990,7 +1990,8 @@ def completion_with_split_tests(models={}, messages=[], use_client=False, overri
model_configs = {} model_configs = {}
if use_client and not override_client: if use_client and not override_client:
if "id" not in kwargs or kwargs["id"] is None: 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 # get the most recent model split list from server
models, model_configs = get_model_split_test(models=models, completion_call_id=kwargs["id"]) models, model_configs = get_model_split_test(models=models, completion_call_id=kwargs["id"])