andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-04 00:32:36 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Move kickoff callbacks to crew's domain

Browse Source
This commit is contained in:
Gui Vieira
2024-11-20 10:06:49 -03:00
parent 0b9092702b
commit 3e003f5e32
6 changed files with 82 additions and 565 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/crewai/project/crew_base.py
View File

@@ -43,8 +43,8 @@ def CrewBase(cls: T) -> T:
for attr in [
"is_task",
"is_agent",
"is_before_crew",
"is_after_crew",
"is_before_kickoff",
"is_after_kickoff",
"is_kickoff",
]
)
@@ -57,11 +57,11 @@ def CrewBase(cls: T) -> T:
self._original_agents = self._filter_functions(
self._original_functions, "is_agent"
)
self._before_crew = self._filter_functions(
self._original_functions, "is_before_crew"
self._before_kickoff = self._filter_functions(
self._original_functions, "is_before_kickoff"
)
self._after_crew = self._filter_functions(
self._original_functions, "is_after_crew"
self._after_kickoff = self._filter_functions(
self._original_functions, "is_after_kickoff"
)
self._kickoff = self._filter_functions(
self._original_functions, "is_kickoff"
@@ -213,25 +213,4 @@ def CrewBase(cls: T) -> T:
callback_functions[callback]() for callback in callbacks
]
def kickoff(self, inputs=None):
# Execute before_crew functions and allow them to modify inputs
for _, func in self._before_crew.items():
modified_inputs = func(self, inputs)
if modified_inputs is not None:
inputs = modified_inputs
# Get the crew instance
crew_instance = self.crew()
# Execute the crew's tasks
result = crew_instance.kickoff(inputs=inputs)
# Execute after_crew functions and allow them to modify the output
for _, func in self._after_crew.items():
modified_result = func(self, result)
if modified_result is not None:
result = modified_result
return result
return cast(T, WrappedClass)