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

updates to testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-16 09:33:12 -07:00
parent 77e4acc7fa
commit d6fb6ea897
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_completion.py
View file

@ -39,15 +39,15 @@ def test_completion_claude_stream():
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
def test_completion_hf_api(): # def test_completion_hf_api():
try: # try:
user_message = "write some code to find the sum of two numbers" # user_message = "write some code to find the sum of two numbers"
messages = [{ "content": user_message,"role": "user"}] # messages = [{ "content": user_message,"role": "user"}]
response = completion(model="stabilityai/stablecode-completion-alpha-3b-4k", messages=messages, custom_llm_provider="huggingface") # response = completion(model="stabilityai/stablecode-completion-alpha-3b-4k", messages=messages, custom_llm_provider="huggingface")
# Add any assertions here to check the response # # Add any assertions here to check the response
print(response) # print(response)
except Exception as e: # except Exception as e:
pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# def test_completion_hf_deployed_api(): # def test_completion_hf_deployed_api():
# try: # try: