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

fix(main.py): fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-11 12:11:50 -07:00
parent 1ba3fcc3fb
commit dd1048cb35
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/main.py
View file

@ -522,7 +522,7 @@ def mock_completion(
) )
return response return response
if n is None: if n is None:
model_response["choices"][0]["message"]["content"] = mock_response model_response.choices[0].message.content = mock_response # type: ignore
else: else:
_all_choices = [] _all_choices = []
for i in range(n): for i in range(n):
@ -533,12 +533,12 @@ def mock_completion(
), ),
) )
_all_choices.append(_choice) _all_choices.append(_choice)
model_response["choices"] = _all_choices model_response.choices = _all_choices # type: ignore
model_response["created"] = int(time.time()) model_response.created = int(time.time())
model_response["model"] = model model_response.model = model
if mock_tool_calls: if mock_tool_calls:
model_response["choices"][0]["message"]["tool_calls"] = [ model_response.choices[0].message.tool_calls = [ # type: ignore
ChatCompletionMessageToolCall(**tool_call) ChatCompletionMessageToolCall(**tool_call)
for tool_call in mock_tool_calls for tool_call in mock_tool_calls
] ]
@ -2672,9 +2672,9 @@ def completion(
""" """
string_response = response_json["data"][0]["output"][0] string_response = response_json["data"][0]["output"][0]
## RESPONSE OBJECT ## RESPONSE OBJECT
model_response["choices"][0]["message"]["content"] = string_response model_response.choices[0].message.content = string_response # type: ignore
model_response["created"] = int(time.time()) model_response.created = int(time.time())
model_response["model"] = model model_response.model = model
response = model_response response = model_response
else: else:
raise ValueError( raise ValueError(