diff --git a/litellm/litellm_config.yaml b/litellm/litellm_config.yaml new file mode 100644 index 0000000..b02ad3c --- /dev/null +++ b/litellm/litellm_config.yaml @@ -0,0 +1,18 @@ +model_list: + - model_name: gemini-pro + litellm_params: + model: gemini/gemini-1.5-pro-latest + api_key: os.environ/GEMINI_API_KEY + + - model_name: gemini-flash + litellm_params: + model: gemini/gemini-1.5-flash-latest + api_key: os.environ/GEMINI_API_KEY + + - model_name: gemini-flash-8b + litellm_params: + model: gemini/gemini-1.5-flash-8b-latest + api_key: os.environ/GEMINI_API_KEY + +general_settings: + master_key: os.environ/LITELLM_MASTER_KEY