scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

remove extra open router test

Browse source
This commit is contained in:
ishaan-jaff 2023-09-29 15:10:41 -07:00
parent 17fc4a7b37
commit c8d0a907ac
1 changed files with 18 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
litellm/tests/test_completion.py
View file

@ -492,18 +492,6 @@ def test_completion_hf_model_no_provider_2():
# test_completion_hf_model_no_provider_2()
def test_completion_openrouter2():
try:
response = completion(
model="openrouter/openai/gpt-4-32k",
messages=messages,
max_tokens=5,
)
# 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_more_optional_params():
try:
response = completion(
@ -804,6 +792,24 @@ def test_completion_bedrock_claude():
pytest.fail(f"Error occurred: {e}")
# test_completion_bedrock_claude()
def test_completion_bedrock_claude_stream():
print("calling claude")
litellm.set_verbose = False
try:
response = completion(
model="bedrock/anthropic.claude-instant-v1",
messages=messages,
logger_fn=logger_fn,
stream=True
)
# Add any assertions here to check the response
print(response)
for chunk in response:
print(chunk)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_bedrock_claude_stream()
def test_completion_bedrock_ai21():
try:
litellm.set_verbose = False