From 475b704f95130c98b0ab3125892bfc75a102284a Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 9 Apr 2025 09:35:05 +0000 Subject: [PATCH] Fix #2547: Add TaskOutput and CrewOutput to public exports Co-Authored-By: Joe Moura --- src/crewai/__init__.py | 10 +++++++--- tests/imports_test.py | 15 +++++++++++++++ 2 files changed, 22 insertions(+), 3 deletions(-) create mode 100644 tests/imports_test.py 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