fix(__init__.py): fix models_by_provider to include cohere_chat models

Fixes https://github.com/BerriAI/litellm/issues/5201
This commit is contained in:
Krrish Dholakia
2024-08-16 11:33:23 -07:00
parent 3d7e3d65d0
commit cbdaecb5a8
2 changed files with 9 additions and 14 deletions
+1 -1
View File
@@ -702,7 +702,7 @@ provider_list: List = [
models_by_provider: dict = {
"openai": open_ai_chat_completion_models + open_ai_text_completion_models,
"cohere": cohere_models,
"cohere": cohere_models + cohere_chat_models,
"cohere_chat": cohere_chat_models,
"anthropic": anthropic_models,
"replicate": replicate_models,
+8 -13
View File
@@ -1,14 +1,9 @@
model_list:
- model_name: "gpt-4"
litellm_params:
model: "gpt-4"
model_info:
my_custom_key: "my_custom_value"
# model_list:
# - model_name: "gpt-4"
# litellm_params:
# model: "gpt-4"
# model_info:
# my_custom_key: "my_custom_value"
litellm_settings:
success_callback: ["s3"]
s3_callback_params:
s3_bucket_name: mytestbucketlitellm # AWS Bucket Name for S3
s3_region_name: us-west-2 # AWS Region Name for S3
s3_aws_access_key_id: os.environ/AWS_ACCESS_KEY_ID # us os.environ/<variable name> to pass environment variables. This is AWS Access Key ID for S3
s3_aws_secret_access_key: os.environ/AWS_SECRET_ACCESS_KEY # AWS Secret Access Key for S3
general_settings:
infer_model_from_keys: true