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

(fix) testing text_completion

Browse source
This commit is contained in:
ishaan-jaff 2023-11-03 16:39:06 -07:00
parent 382e31d7b7
commit e29b2e8ce4
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_text_completion.py
View file

@ -24,9 +24,11 @@ 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?"
) )
print(response)
response_str = response["choices"][0]["text"] response_str = response["choices"][0]["text"]
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_completion_openai_prompt()
def test_completion_openai_prompt_array(): def test_completion_openai_prompt_array():
@ -61,10 +63,10 @@ def test_completion_hf_prompt_array():
model="huggingface/mistralai/Mistral-7B-v0.1", model="huggingface/mistralai/Mistral-7B-v0.1",
prompt=token_prompt, # token prompt is a 2d list prompt=token_prompt, # token prompt is a 2d list
) )
print("\n\n response\n\n") print("\n\n response")
print(response) print(response)
# response_str = response["choices"][0]["text"] # response_str = response["choices"][0]["text"]
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_completion_hf_prompt_array() # test_completion_hf_prompt_array()