diff --git a/tests/test_import_llm.py b/tests/test_import_llm.py new file mode 100644 index 000000000..8e23bf39e --- /dev/null +++ b/tests/test_import_llm.py @@ -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}")