Merge branch 'main' into support_anthropic_function_result

This commit is contained in:
Krish Dholakia 2024-03-16 09:58:08 -07:00 committed by GitHub
commit 0368a335e6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
42 changed files with 815 additions and 216 deletions

View file

@ -358,7 +358,7 @@ def test_completion_mistral_azure():
}
],
)
# Add any assertions here to check the response
# Add any assertions here to check, the response
print(response)
except Exception as e:
@ -575,6 +575,25 @@ def test_completion_azure_gpt4_vision():
# test_completion_azure_gpt4_vision()
def test_completion_fireworks_ai():
try:
litellm.set_verbose = True
messages = [
{"role": "system", "content": "You're a good bot"},
{
"role": "user",
"content": "Hey",
},
]
response = completion(
model="fireworks_ai/accounts/fireworks/models/mixtral-8x7b-instruct",
messages=messages,
)
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="this test is flaky")
def test_completion_perplexity_api():
try: