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",
},
],
"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",
"bedrock",
"azure",
"cerebras",
]
MODELS = {