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

fix(proxy_server.py): If master key is set, only master key can be used to generate new keys

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-22 10:18:28 -08:00
parent 10fe16c965
commit 604ad41eac
2 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_completion.py
View file

@ -164,9 +164,7 @@ def test_completion_gpt4_vision():
def test_completion_perplexity_api():
try:
litellm.set_verbose=True
litellm.num_retries = 0
litellm.drop_params = True
# litellm.set_verbose=True
messages=[{
"role": "system",
"content": "You're a good bot"
@ -180,7 +178,7 @@ def test_completion_perplexity_api():
response = completion(
model="mistral-7b-instruct",
messages=messages,
api_base="https://api.perplexity.ai", stop="Hello")
api_base="https://api.perplexity.ai")
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
@ -446,7 +444,7 @@ def test_completion_openai():
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai()
# test_completion_openai()
def test_completion_text_openai():
try:
@ -832,6 +830,7 @@ def test_completion_replicate_llama2_stream():
# assert len(chunk.choices[0].delta["content"]) > 2
# print(chunk)
assert len(complete_response) > 5
print(f"complete_response: {complete_response}")
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_replicate_llama2_stream()