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

fix text-003 bug

Browse source
This commit is contained in:
ishaan-jaff 2023-08-17 10:48:22 -07:00
parent 5ef5af8ddc
commit 35fd82fd3f
2 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -158,7 +158,7 @@ def completion(
model=model,
prompt = prompt
)
completion_response = response["choices"]["text"]
completion_response = response["choices"][0]["text"]
## LOGGING
logging(model=model, input=prompt, custom_llm_provider=custom_llm_provider, additional_args={"max_tokens": max_tokens, "original_response": completion_response}, logger_fn=logger_fn)
## RESPONSE OBJECT

8
litellm/tests/test_completion.py
View file

@ -90,6 +90,14 @@ def test_completion_openai():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_text_openai():
try:
response = completion(model="text-davinci-003", messages=messages)
# Add any assertions here to check the response
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_openai_with_optional_params():
try:
response = completion(model="gpt-3.5-turbo", messages=messages, temperature=0.5, top_p=0.1, user="ishaan_dev@berri.ai")