diff --git a/src/crewai/__init__.py b/src/crewai/__init__.py index 0833afd58..c3835f82f 100644 --- a/src/crewai/__init__.py +++ b/src/crewai/__init__.py @@ -2,11 +2,13 @@ import warnings from crewai.agent import Agent from crewai.crew import Crew +from crewai.crews.crew_output import CrewOutput from crewai.flow.flow import Flow from crewai.knowledge.knowledge import Knowledge from crewai.llm import LLM from crewai.process import Process from crewai.task import Task +from crewai.tasks.task_output import TaskOutput warnings.filterwarnings( "ignore", @@ -18,9 +20,11 @@ __version__ = "0.86.0" __all__ = [ "Agent", "Crew", - "Process", - "Task", - "LLM", + "CrewOutput", "Flow", "Knowledge", + "LLM", + "Process", + "Task", + "TaskOutput", ] diff --git a/tests/imports_test.py b/tests/imports_test.py new file mode 100644 index 000000000..0715e3c50 --- /dev/null +++ b/tests/imports_test.py @@ -0,0 +1,15 @@ +"""Test that all public API classes are properly importable.""" + + +def test_task_output_import(): + """Test that TaskOutput can be imported from crewai.""" + from crewai import TaskOutput + + assert TaskOutput is not None + + +def test_crew_output_import(): + """Test that CrewOutput can be imported from crewai.""" + from crewai import CrewOutput + + assert CrewOutput is not None