Fix issue #2353: Add tests for importing LLM from crewai

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-13 01:07:01 +00:00
parent 41a670166a
commit 6be47ee64e

23
tests/test_import_llm.py Normal file
View File

@@ -0,0 +1,23 @@
import pytest
def test_import_llm_from_crewai():
"""Test that LLM can be imported directly from crewai."""
try:
from crewai import LLM
assert LLM is not None
except ImportError as e:
pytest.fail(f"Failed to import LLM from crewai: {e}")
def test_bedrock_llm_creation():
"""Test that a Bedrock LLM can be created."""
try:
from crewai import LLM
# Just test the object creation, not the actual API call
bedrock_llm = LLM(
model="bedrock/anthropic.claude-3-sonnet-20240229-v1:0"
)
assert bedrock_llm is not None
assert bedrock_llm.model == "bedrock/anthropic.claude-3-sonnet-20240229-v1:0"
except Exception as e:
pytest.fail(f"Failed to create Bedrock LLM: {e}")