andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 00:02:36 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix #2551: Add Huggingface to provider list in CLI

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-09 10:55:31 +00:00
parent 409892d65f
commit ffb8bedf1b
2 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/crewai/cli/constants.py
View File

@@ -85,6 +85,16 @@ ENV_VARS = {
"key_name": "CEREBRAS_API_KEY",
},
],
"huggingface": [
{
"prompt": "Enter your HUGGINGFACE API key (press Enter to skip)",
"key_name": "HUGGINGFACE_API_KEY",
},
{
"prompt": "Enter your Huggingface API base URL (press Enter to skip)",
"key_name": "base_url",
},
],
}
@@ -93,6 +103,7 @@ PROVIDERS = [
"anthropic",
"gemini",
"groq",
"huggingface",
"ollama",
"watson",
"bedrock",
@@ -156,6 +167,12 @@ MODELS = {
"bedrock/mistral.mistral-7b-instruct-v0:2",
"bedrock/mistral.mixtral-8x7b-instruct-v0:1",
],
"huggingface": [
"huggingface/meta-llama/Meta-Llama-3.1-8B-Instruct",
"huggingface/mistralai/Mixtral-8x7B-Instruct-v0.1",
"huggingface/tiiuae/falcon-180B-chat",
"huggingface/google/gemma-7b-it",
],
}
JSON_URL = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json"