Add support for cerebras

This commit is contained in:
Brandon Hancock
2024-10-31 10:20:59 -04:00
parent b747d66c52
commit a9eaf63c34

View File

@@ -75,6 +75,16 @@ ENV_VARS = {
"key_name": "AZURE_API_VERSION", "key_name": "AZURE_API_VERSION",
}, },
], ],
"cerebras": [
{
"prompt": "Enter your Cerebras model name (must start with 'cerebras/')",
"key_name": "model",
},
{
"prompt": "Enter your Cerebras API version (press Enter to skip)",
"key_name": "CEREBRAS_API_KEY",
},
],
} }
@@ -87,6 +97,7 @@ PROVIDERS = [
"watson", "watson",
"bedrock", "bedrock",
"azure", "azure",
"cerebras",
] ]
MODELS = { MODELS = {