mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 16:48:30 +00:00
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
import importlib
|
|
import sys
|
|
import warnings
|
|
|
|
import pytest
|
|
|
|
|
|
def test_crew_import_with_numpy():
|
|
"""Test that crewai can be imported even with NumPy compatibility issues."""
|
|
try:
|
|
# Force reload to ensure we test our fix
|
|
if "crewai" in sys.modules:
|
|
importlib.reload(sys.modules["crewai"])
|
|
|
|
# This should not raise an exception
|
|
from crewai import Crew
|
|
assert Crew is not None
|
|
except Exception as e:
|
|
pytest.fail(f"Failed to import Crew: {e}")
|
|
|
|
def test_embedding_configurator_with_numpy():
|
|
"""Test that EmbeddingConfigurator can be imported with NumPy."""
|
|
try:
|
|
# Force reload
|
|
if "crewai.utilities.embedding_configurator" in sys.modules:
|
|
importlib.reload(sys.modules["crewai.utilities.embedding_configurator"])
|
|
|
|
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
|
|
configurator = EmbeddingConfigurator()
|
|
# Test that we can create an embedder (might be unavailable but shouldn't crash)
|
|
embedder = configurator.configure_embedder()
|
|
assert embedder is not None
|
|
except Exception as e:
|
|
pytest.fail(f"Failed to use EmbeddingConfigurator: {e}")
|
|
|
|
def test_rag_storage_with_numpy():
|
|
"""Test that RAGStorage can be imported and used with NumPy."""
|
|
try:
|
|
# Force reload
|
|
if "crewai.memory.storage.rag_storage" in sys.modules:
|
|
importlib.reload(sys.modules["crewai.memory.storage.rag_storage"])
|
|
|
|
from crewai.memory.storage.rag_storage import RAGStorage
|
|
# Initialize with minimal config to avoid actual DB operations
|
|
storage = RAGStorage(type="test", crew=None)
|
|
# Just verify we can create the object without errors
|
|
assert storage is not None
|
|
except Exception as e:
|
|
pytest.fail(f"Failed to use RAGStorage: {e}")
|
|
|
|
def test_knowledge_storage_with_numpy():
|
|
"""Test that KnowledgeStorage can be imported and used with NumPy."""
|
|
try:
|
|
# Force reload
|
|
if "crewai.knowledge.storage.knowledge_storage" in sys.modules:
|
|
importlib.reload(sys.modules["crewai.knowledge.storage.knowledge_storage"])
|
|
|
|
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
|
|
# Initialize with minimal config
|
|
storage = KnowledgeStorage()
|
|
# Just verify we can create the object without errors
|
|
assert storage is not None
|
|
except Exception as e:
|
|
pytest.fail(f"Failed to use KnowledgeStorage: {e}")
|