From a9eaf63c342c8f7e9b33f4d8f50e6158aeab0591 Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Thu, 31 Oct 2024 10:20:59 -0400
Subject: [PATCH] Add support for cerebras

---
 src/crewai/cli/constants.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/crewai/cli/constants.py b/src/crewai/cli/constants.py
index 39b024792..a60922eb0 100644
--- a/src/crewai/cli/constants.py
+++ b/src/crewai/cli/constants.py
@@ -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 = {