phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-27 21:31:00 -07:00
parent 94eba84cf0
commit 3658bdb61b
3 changed files with 4 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -1432,7 +1432,7 @@ def text_completion(*args, **kwargs):
kwargs["messages"] = messages
kwargs.pop("prompt")
response = completion(*args, **kwargs) # assume the response is the openai response object
response_2 = {
formatted_response_obj = {
"id": response["id"],
"object": "text_completion",
"created": response["created"],
@ -1447,7 +1447,7 @@ def text_completion(*args, **kwargs):
],
"usage": response["usage"]
}
return response_2
return formatted_response_obj
else:
raise ValueError("please pass prompt into the `text_completion` endpoint - `text_completion(model, prompt='hello world')`")