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

allow custom provider via model name

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-19 16:34:52 -07:00
parent 245b00f5f5
commit 6cee3b22ab
11 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_completion.py
View file

@ -25,6 +25,18 @@ def logger_fn(user_model_dict):
print(f"user_model_dict: {user_model_dict}")
def test_completion_custom_provider_model_name():
try:
response = completion(
model="together_ai/togethercomputer/llama-2-70b-chat", messages=messages, 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_custom_provider_model_name()
def test_completion_claude():
try:
response = completion(