andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Merge branch 'feature/procedure_v2' into brandon/cre-107-pipeline-conditional-routing

Browse Source
This commit is contained in:
Brandon Hancock
2024-07-30 09:55:20 -04:00
parent e4b3b32b50 072044c537
commit 4251494c55
4 changed files with 35 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/pipeline/test_pipeline.py
View File

@@ -6,7 +6,7 @@ from crewai.agent import Agent
from crewai.crew import Crew
from crewai.crews.crew_output import CrewOutput
from crewai.pipeline.pipeline import Pipeline
from crewai.pipeline.pipeline_run_result import PipelineRunResult
from crewai.pipeline.pipeline_kickoff_result import PipelineKickoffResult
from crewai.process import Process
from crewai.task import Task
from crewai.tasks.task_output import TaskOutput
@@ -113,7 +113,7 @@ async def test_pipeline_process_streams_single_input(mock_crew_factory):
mock_crew.kickoff_async.assert_called_once_with(inputs={"key": "value"})
for pipeline_result in pipeline_results:
assert isinstance(pipeline_result, PipelineRunResult)
assert isinstance(pipeline_result, PipelineKickoffResult)
assert pipeline_result.raw == "Test output"
assert len(pipeline_result.crews_outputs) == 1
print("pipeline_result.token_usage", pipeline_result.token_usage)
@@ -199,7 +199,7 @@ async def test_pipeline_process_streams_single_input_pydantic_output(mock_crew_f
print("pipeline_result.trace", pipeline_result.trace)
assert isinstance(pipeline_result, PipelineRunResult)
assert isinstance(pipeline_result, PipelineKickoffResult)
assert pipeline_result.raw == "Test output"
assert len(pipeline_result.crews_outputs) == 1
assert pipeline_result.token_usage == {crew_name: DEFAULT_TOKEN_USAGE}