mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
Fix #2547: Add TaskOutput and CrewOutput to public exports
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -2,11 +2,13 @@ import warnings
|
|||||||
|
|
||||||
from crewai.agent import Agent
|
from crewai.agent import Agent
|
||||||
from crewai.crew import Crew
|
from crewai.crew import Crew
|
||||||
|
from crewai.crews.crew_output import CrewOutput
|
||||||
from crewai.flow.flow import Flow
|
from crewai.flow.flow import Flow
|
||||||
from crewai.knowledge.knowledge import Knowledge
|
from crewai.knowledge.knowledge import Knowledge
|
||||||
from crewai.llm import LLM
|
from crewai.llm import LLM
|
||||||
from crewai.process import Process
|
from crewai.process import Process
|
||||||
from crewai.task import Task
|
from crewai.task import Task
|
||||||
|
from crewai.tasks.task_output import TaskOutput
|
||||||
|
|
||||||
warnings.filterwarnings(
|
warnings.filterwarnings(
|
||||||
"ignore",
|
"ignore",
|
||||||
@@ -18,9 +20,11 @@ __version__ = "0.86.0"
|
|||||||
__all__ = [
|
__all__ = [
|
||||||
"Agent",
|
"Agent",
|
||||||
"Crew",
|
"Crew",
|
||||||
"Process",
|
"CrewOutput",
|
||||||
"Task",
|
|
||||||
"LLM",
|
|
||||||
"Flow",
|
"Flow",
|
||||||
"Knowledge",
|
"Knowledge",
|
||||||
|
"LLM",
|
||||||
|
"Process",
|
||||||
|
"Task",
|
||||||
|
"TaskOutput",
|
||||||
]
|
]
|
||||||
|
|||||||
15
tests/imports_test.py
Normal file
15
tests/imports_test.py
Normal file
@@ -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
|
||||||
Reference in New Issue
Block a user