mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-15 11:08:33 +00:00
Tests working now!
This commit is contained in:
@@ -158,9 +158,7 @@ class Pipeline(BaseModel):
|
||||
+ list(next_pipeline.stages)
|
||||
+ self.stages[stage_index + 1 :]
|
||||
)
|
||||
traces.append(
|
||||
[{"router": stage.__class__.__name__, "route_taken": route_taken}]
|
||||
)
|
||||
traces.append([{"route_taken": route_taken}])
|
||||
stage_index += 1
|
||||
continue
|
||||
|
||||
@@ -197,16 +195,6 @@ class Pipeline(BaseModel):
|
||||
else:
|
||||
raise ValueError(f"Unsupported stage type: {type(stage)}")
|
||||
|
||||
async def _process_pipeline(
|
||||
self, pipeline: "Pipeline", current_input: Dict[str, Any]
|
||||
) -> Tuple[List[CrewOutput], List[Union[str, Dict[str, Any]]]]:
|
||||
results = await pipeline.process_single_kickoff(current_input)
|
||||
outputs = [result.crews_outputs[-1] for result in results]
|
||||
traces: List[Union[str, Dict[str, Any]]] = [
|
||||
f"Nested Pipeline: {pipeline.__class__.__name__}"
|
||||
]
|
||||
return outputs, traces
|
||||
|
||||
async def _process_single_crew(
|
||||
self, crew: Crew, current_input: Dict[str, Any]
|
||||
) -> Tuple[List[CrewOutput], List[Union[str, Dict[str, Any]]]]:
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
import logging
|
||||
from dataclasses import dataclass
|
||||
from typing import Any, Callable, Dict, Generic, Tuple, TypeVar
|
||||
|
||||
|
||||
Reference in New Issue
Block a user