26 lines
676 B
Python
26 lines
676 B
Python
from crewai import Crew, Process
|
|
|
|
class TaskList:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
# Generate a dictionary for up to n agents dynamically
|
|
inputs = {"required": {"task-1": ("TASK",)}}
|
|
# for i in range(2, 101): # Adjust the range as needed
|
|
# inputs["optional"] = inputs.get("optional", {})
|
|
# inputs["optional"][f"agent{i}"] = ("AGENT",)
|
|
return inputs
|
|
|
|
|
|
OUTPUT_IS_LIST = (True,)
|
|
RETURN_TYPES = ("TASK",)
|
|
FUNCTION = "create_list"
|
|
CATEGORY = "📎CrewAi"
|
|
|
|
def create_list(self, **kwargs):
|
|
task = []
|
|
|
|
for k, v in kwargs.items():
|
|
task.append(v)
|
|
|
|
return (task, )
|