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

fix(main.py): handle translating text completion openai to chat completion for async requests

Browse source 
also adds testing for this, to prevent future regressions
This commit is contained in:
Krrish Dholakia 2024-04-09 16:47:49 -07:00
parent ba57434374
commit 855e7ed9d2
2 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/tests/test_completion.py
View file

@ -975,6 +975,19 @@ def test_completion_text_openai():
pytest.fail(f"Error occurred: {e}")
@pytest.mark.asyncio
async def test_completion_text_openai_async():
try:
# litellm.set_verbose =True
response = await litellm.acompletion(
model="gpt-3.5-turbo-instruct", messages=messages
)
print(response["choices"][0]["message"]["content"])
except Exception as e:
print(e)
pytest.fail(f"Error occurred: {e}")
def custom_callback(
kwargs, # kwargs to completion
completion_response, # response from completion