tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-11 17:05:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

update claude testing

Browse Source
This commit is contained in:
ishaan-jaff
2023-09-29 15:37:05 -07:00
parent 6a81f860f9
commit 76956cce52
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_completion.py
+4 -2
View File
@@ -784,13 +784,15 @@ def test_completion_bedrock_claude():
response = completion(
model="bedrock/anthropic.claude-instant-v1",
messages=messages,
max_tokens=10,
temperature=0.1,
logger_fn=logger_fn
)
# Add any assertions here to check the response
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_bedrock_claude()
test_completion_bedrock_claude()
def test_completion_bedrock_claude_stream():
print("calling claude")
@@ -807,7 +809,7 @@ def test_completion_bedrock_claude_stream():
print(chunk)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_bedrock_claude_stream()
# test_completion_bedrock_claude_stream()
def test_completion_bedrock_ai21():
try: