andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Address PR feedback and fix lint issues

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-01 01:33:43 +00:00
parent c9ed55814e
commit 56afe4fb6d
3 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -107,11 +107,13 @@ ENV_VARS = {
{
"prompt": "Enter your LLAMA API key (press Enter to skip)",
"key_name": "LLAMA_API_KEY",
"description": "API key for Meta's Llama API authentication",
},
{
"prompt": "Enter your LLAMA API base URL (press Enter to skip)",
"key_name": "LLAMA_API_BASE",
"default": "https://api.llama.com/compat/v1",
"description": "Base URL for Meta's Llama API with OpenAI compatibility",
},
],
}
@@ -327,8 +329,8 @@ MODELS = {
"sambanova/Meta-Llama-3.2-1B-Instruct",
],
"meta-llama": [
"meta-llama/llama-4-scout-17b-16e-instruct-fp8",
"meta-llama/llama-4-maverick-17b-128e-instruct-fp8",
"meta-llama/llama-4-scout-17b",
"meta-llama/llama-4-maverick-17b",
"meta-llama/llama-3.3-70b-instruct",
"meta-llama/llama-3.3-8b-instruct",
],

6
src/crewai/llm.py
View File

@@ -120,9 +120,9 @@ LLM_CONTEXT_WINDOW_SIZES = {
"mixtral-8x7b-32768": 32768,
"llama-3.3-70b-versatile": 128000,
"llama-3.3-70b-instruct": 128000,
"llama-4-scout-17b-16e-instruct-fp8": 128000,
"llama-4-maverick-17b-128e-instruct-fp8": 128000,
"llama-3.3-8b-instruct": 128000,
"llama-4-scout-17b": 128000, # Based on official Meta documentation
"llama-4-maverick-17b": 128000, # Based on official Meta documentation
"llama-3.3-8b-instruct": 128000, # Based on official Meta documentation
# sambanova
"Meta-Llama-3.3-70B-Instruct": 131072,
"QwQ-32B-Preview": 8192,

7
tests/llm_test.py
View File

@@ -621,8 +621,11 @@ def test_handle_streaming_tool_calls_no_tools(mock_emit):
def test_llama_api_support():
"""Test that Llama API models are correctly configured."""
from crewai.cli.constants import MODELS, PROVIDERS, ENV_VARS, LITELLM_PARAMS
"""
Test Llama API configuration and integration.
- Verifies provider registration, model availability, environment variables, and context window sizes.
"""
from crewai.cli.constants import ENV_VARS, LITELLM_PARAMS, MODELS, PROVIDERS
from crewai.llm import LLM_CONTEXT_WINDOW_SIZES
assert "meta-llama" in PROVIDERS