diff --git a/src/crewai/__init__.py b/src/crewai/__init__.py index 67d63b82c..565f89065 100644 --- a/src/crewai/__init__.py +++ b/src/crewai/__init__.py @@ -2,12 +2,14 @@ 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.llms.base_llm import BaseLLM from crewai.process import Process from crewai.task import Task +from crewai.tasks.task_output import TaskOutput warnings.filterwarnings( "ignore", @@ -19,10 +21,12 @@ __version__ = "0.108.0" __all__ = [ "Agent", "Crew", + "CrewOutput", "Process", "Task", "LLM", "BaseLLM", "Flow", "Knowledge", + "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