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

Add missing doc string

Browse Source
This commit is contained in:
Brandon Hancock
2024-07-29 16:22:52 -04:00
parent 467c67c597
commit d72f5a601e
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/crewai/pipeline/pipeline.py
View File

@@ -391,6 +391,15 @@ class Pipeline(BaseModel):
return [crew_outputs + [output] for output in all_stage_outputs[-1]] return [crew_outputs + [output] for output in all_stage_outputs[-1]]
def __rshift__(self, other: PipelineStage) -> Pipeline: def __rshift__(self, other: PipelineStage) -> Pipeline:
"""
Implements the >> operator to add another Stage (Crew or List[Crew]) to an existing Pipeline.
Args:
other (Any): The stage to add.
Returns:
Pipeline: A new pipeline with the added stage.
"""
if isinstance(other, (Crew, Pipeline, PipelineRouter)): if isinstance(other, (Crew, Pipeline, PipelineRouter)):
return type(self)(stages=self.stages + [other]) return type(self)(stages=self.stages + [other])
elif isinstance(other, list) and all(isinstance(crew, Crew) for crew in other): elif isinstance(other, list) and all(isinstance(crew, Crew) for crew in other):