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

(fix) remove print statements

Browse source
This commit is contained in:
ishaan-jaff 2023-11-03 16:45:28 -07:00
parent f3dc06da04
commit 6c4816e214
1 changed files with 0 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -1902,15 +1902,10 @@ def text_completion(*args, **kwargs):
responses = [None for x in prompt] # init responses responses = [None for x in prompt] # init responses
for i, request in enumerate(prompt): for i, request in enumerate(prompt):
decoded_prompt = tokenizer.decode(request) decoded_prompt = tokenizer.decode(request)
# print("\ndecoded\n", decoded_prompt)
# print("type decoded", type(decoded_prompt))
new_kwargs = copy.deepcopy(kwargs) new_kwargs = copy.deepcopy(kwargs)
new_kwargs["prompt"] = decoded_prompt new_kwargs["prompt"] = decoded_prompt
# print("making new individual request", new_kwargs)
response = text_completion(**new_kwargs) response = text_completion(**new_kwargs)
# print("assigning for ", i)
responses[i] = response["choices"][0] responses[i] = response["choices"][0]
print(responses)
formatted_response_obj = { formatted_response_obj = {
"id": response["id"], "id": response["id"],
"object": "text_completion", "object": "text_completion",