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

refactor(bedrock.py-+-cohere.py): making bedrock and cohere compatible with openai v1 sdk

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-11 17:33:19 -08:00
parent 39c2597c33
commit 547598a134
7 changed files with 82 additions and 74 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_completion.py
View file

@ -395,7 +395,7 @@ def test_completion_cohere(): # commenting for now as the cohere endpoint is bei
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_cohere() #
# test_completion_cohere()
def test_completion_openai():
@ -634,7 +634,7 @@ def test_completion_azure():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_azure()
# test_completion_azure()
def test_completion_azure2():
# test if we can pass api_base, api_version and api_key in compleition()
try:
@ -941,7 +941,7 @@ def test_completion_bedrock_claude():
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_bedrock_claude()
test_completion_bedrock_claude()
def test_completion_bedrock_cohere():
print("calling bedrock cohere")