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

expand support for cohere models

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-04 15:00:00 -07:00
parent 50c2c48cc9
commit a3d2a2fb9d
2 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
community_resources/max_tokens.json
View file

@ -12,5 +12,5 @@
"text-bison-001": {"max_tokens": 8192, "input_cost_per_token": 0.000004, "output_cost_per_token": 0.000004}, "text-bison-001": {"max_tokens": 8192, "input_cost_per_token": 0.000004, "output_cost_per_token": 0.000004},
"chat-bison-001": {"max_tokens": 4096, "input_cost_per_token": 0.000002, "output_cost_per_token": 0.000002}, "chat-bison-001": {"max_tokens": 4096, "input_cost_per_token": 0.000002, "output_cost_per_token": 0.000002},
"command-nightly": {"max_tokens": 4096, "input_cost_per_token": 0.000015, "output_cost_per_token": 0.000015}, "command-nightly": {"max_tokens": 4096, "input_cost_per_token": 0.000015, "output_cost_per_token": 0.000015},
"replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1": {"max_tokens": 4096, "input_cost_per_token": 0.00000608, "output_cost_per_token": 0.00000608}, "replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1": {"max_tokens": 4096, "input_cost_per_token": 0.00000608, "output_cost_per_token": 0.00000608}
}} }}

4
litellm/__init__.py
View file

@ -47,6 +47,10 @@ open_ai_text_completion_models = [
cohere_models = [ cohere_models = [
'command-nightly', 'command-nightly',
"command",
"command-light",
"command-medium-beta",
"command-xlarge-beta"
] ]
anthropic_models = [ anthropic_models = [