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

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

7
litellm/tests/test_completion.py
View file

@ -346,12 +346,7 @@ def test_completion_openai_prompt():
response = text_completion( response = text_completion(
model="gpt-3.5-turbo", prompt="What's the weather in SF?" model="gpt-3.5-turbo", prompt="What's the weather in SF?"
) )
response_str = response["choices"][0]["message"]["content"] response_str = response["choices"][0]["text"]
response_str_2 = response.choices[0].message.content
print(response)
assert response_str == response_str_2
assert type(response_str) == str
assert len(response_str) > 1
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.791" version = "0.1.792"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"