Merge pull request #2548 from crewAIInc/devin/1744191265-fix-taskoutput-import

Fix #2547: Add TaskOutput and CrewOutput to public exports

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",
 ]

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