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

refactor: update the get_crew_instance return type

Browse Source
This commit is contained in:
Lucas Gomide
2025-05-02 11:24:50 -03:00
parent 41952664b7
commit 88e5ddfcdb
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/crewai/cli/utils.py
View File

@@ -309,35 +309,35 @@ def get_crews(crew_path: str = "crew.py", require: bool = False) -> list[Crew]:
return crew_instances
def get_crew_instance(module_attr) -> tuple[bool, Crew | None]:
def get_crew_instance(module_attr) -> Crew | None:
if (
callable(module_attr)
and hasattr(module_attr, "is_crew_class")
and module_attr.is_crew_class
):
return True, module_attr().crew()
return module_attr().crew()
if (ismethod(module_attr) or isfunction(module_attr)) and get_type_hints(
module_attr
).get("return") is Crew:
return True, module_attr()
return module_attr()
elif isinstance(module_attr, Crew):
return True, module_attr
return module_attr
else:
return False, None
return None
def fetch_crews(module_attr) -> list[Crew]:
crew_instances: list[Crew] = []
is_crew, crew_instance = get_crew_instance(module_attr)
crew_instance = get_crew_instance(module_attr)
if is_crew and crew_instance:
if crew_instance:
crew_instances.append(crew_instance)
if isinstance(module_attr, type) and issubclass(module_attr, Flow):
instance = module_attr()
for attr_name in dir(instance):
attr = getattr(instance, attr_name)
is_crew, crew_instance = get_crew_instance(attr)
if is_crew and crew_instance:
crew_instance = get_crew_instance(attr)
if crew_instance:
crew_instances.append(crew_instance)
return crew_instances