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

test: removing bedrock claude-v1 testing - bedrock removed this

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-28 11:08:05 -08:00
parent 9124ef60a4
commit a042092faa
3 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_custom_callback_input.py
View file

@ -483,12 +483,12 @@ def test_chat_bedrock_stream():
customHandler = CompletionCustomHandler()
litellm.callbacks = [customHandler]
response = litellm.completion(
model="bedrock/anthropic.claude-v1",
model="bedrock/anthropic.claude-v2",
messages=[{"role": "user", "content": "Hi 👋 - i'm sync bedrock"}],
)
# test streaming
response = litellm.completion(
model="bedrock/anthropic.claude-v1",
model="bedrock/anthropic.claude-v2",
messages=[{"role": "user", "content": "Hi 👋 - i'm sync bedrock"}],
stream=True,
)
@ -497,7 +497,7 @@ def test_chat_bedrock_stream():
# test failure callback
try:
response = litellm.completion(
model="bedrock/anthropic.claude-v1",
model="bedrock/anthropic.claude-v2",
messages=[{"role": "user", "content": "Hi 👋 - i'm sync bedrock"}],
aws_region_name="my-bad-region",
stream=True,
@ -524,12 +524,12 @@ async def test_async_chat_bedrock_stream():
customHandler = CompletionCustomHandler()
litellm.callbacks = [customHandler]
response = await litellm.acompletion(
model="bedrock/anthropic.claude-v1",
model="bedrock/anthropic.claude-v2",
messages=[{"role": "user", "content": "Hi 👋 - i'm async bedrock"}],
)
# test streaming
response = await litellm.acompletion(
model="bedrock/anthropic.claude-v1",
model="bedrock/anthropic.claude-v2",
messages=[{"role": "user", "content": "Hi 👋 - i'm async bedrock"}],
stream=True,
)
@ -540,7 +540,7 @@ async def test_async_chat_bedrock_stream():
## test failure callback
try:
response = await litellm.acompletion(
model="bedrock/anthropic.claude-v1",
model="bedrock/anthropic.claude-v2",
messages=[{"role": "user", "content": "Hi 👋 - i'm async bedrock"}],
aws_region_name="my-bad-key",
stream=True,