nodes/luandev-ComfyUI-CrewAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor

Browse Source
This commit is contained in:
Luan Citta
2024-06-05 13:59:15 +01:00
parent 883a449cb8
commit 61de88680e
8 changed files with 36 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/agent_node.py
+6 -5
View File
@@ -10,24 +10,25 @@ class AgentNode:
return {
"required": {
"role": ("STRING", {
"forceInput": True, "multiline": True, "default": ""
"multiline": True, "default": ""
}),
"goal": ("STRING", {
"forceInput": True, "multiline": True, "default": ""
"multiline": True, "default": ""
}),
"backstory": ("STRING", {
"forceInput": True, "multiline": True, "default": ""
"multiline": True, "default": ""
}),
},
"optional": {
"LLM": ("CAI_LLM",),
"LLM": ("CREWAI_LLM",),
"verbose": ("BOOL", {"default": False}),
"allow_delegation": ("BOOL", {"default": True}),
},
}
OUTPUT_NODE = True
RETURN_TYPES = ("CAI_AGENT",)
RETURN_TYPES = ("CREWAI_AGENT",)
RETURN_NAMES = ("agent",)
FUNCTION = "create_agent"
CATEGORY = "📎CrewAi"
nodes/crew_node.py
+2 -2
View File
@@ -10,10 +10,10 @@ class CrewNode(BaseNode):
def INPUT_TYPES(cls):
return {
"required": {
"tasks": ("CAI_TASK", {"default": []}),
"tasks": ("CREWAI_TASK", {"default": []}),
},
"optional": {
"agents": ("CAI_AGENT", {"forceInput": False, "default": []}),
"agents": ("CREWAI_AGENT", {"forceInput": False, "default": []}),
"topic": ("STRING", {
"forceInput": False, "multiline": True, "default": ""
}),
nodes/list_agent_node.py
+4 -2
View File
@@ -4,8 +4,10 @@ from .base_node import BaseNode
class AgentList(BaseNode):
@classmethod
def INPUT_TYPES(cls):
return {"required": {"agent-1": ("CAI_AGENT",)}}
return {"required": {"agent-1": ("CREWAI_AGENT",)}}
OUTPUT_IS_LIST = (True,)
RETURN_TYPES = ("CAI_AGENT",)
RETURN_TYPES = ("CREWAI_AGENT",)
RETURN_NAMES = ("agent",)
FUNCTION = "create_list"
nodes/list_task_node.py
+4 -2
View File
@@ -4,8 +4,10 @@ from .base_node import BaseNode
class TaskList(BaseNode):
@classmethod
def INPUT_TYPES(s):
return {"required": {"task-1": ("CAI_TASK",)}}
return {"required": {"task-1": ("CREWAI_TASK",)}}
OUTPUT_IS_LIST = (True,)
RETURN_TYPES = ("CAI_TASK",)
RETURN_TYPES = ("CREWAI_TASK",)
RETURN_NAMES = ("task",)
FUNCTION = "create_list"
nodes/llm_chatGpt_node.py
+5 -6
View File
@@ -12,19 +12,18 @@ class LlmChatGpt(BaseNode):
def INPUT_TYPES(cls):
return {
"required": {
"api_key": ("STRING", {
"forceInput": True,
"default": "http://localhost:11434/v1"
}),
"api_key": ("STRING", {"default": ""}),
},
"optional": {
"model": ("STRING", {
"forceInput": True, "default": "gpt-4o"
"default": "gpt-4o"
}),
}
}
RETURN_TYPES = ("CAI_LLM",)
RETURN_TYPES = ("CREWAI_LLM",)
RETURN_NAMES = ("llm",)
FUNCTION = "create_task"
def create_task(self, model, api_key):
nodes/llm_huggingface_node.py
+5 -5
View File
@@ -13,22 +13,22 @@ class LlmHuggingFace(BaseNode):
return {
"required": {
"model": ("STRING", {
"forceInput": True, "default": "crewai-llama3"
"default": "crewai-llama3"
}),
"base_url": ("STRING", {
"forceInput": True,
"default": "http://localhost:11434/v1"
}),
},
"optional": {
"api_key": ("STRING", {
"forceInput": True,
"default": "http://localhost:11434/v1"
"default": "..."
}),
}
}
RETURN_TYPES = ("CAI_LLM",)
RETURN_TYPES = ("CREWAI_LLM",)
RETURN_NAMES = ("llm",)
FUNCTION = "create_task"
def create_task(self, base_url, api_key, task, max_new_tokens):
nodes/llm_openai_node.py
+5 -5
View File
@@ -13,22 +13,22 @@ class LlmOpenai(BaseNode):
return {
"required": {
"model": ("STRING", {
"forceInput": True, "default": "crewai-llama3"
"default": "crewai-llama3"
}),
"base_url": ("STRING", {
"forceInput": True,
"default": "http://localhost:11434/v1"
}),
},
"optional": {
"api_key": ("STRING", {
"forceInput": True,
"default": "http://localhost:11434/v1"
"default": ""
}),
}
}
RETURN_TYPES = ("CAI_LLM",)
RETURN_TYPES = ("CREWAI_LLM",)
RETURN_NAMES = ("llm",)
FUNCTION = "create_task"
def create_task(self, model, base_url, api_key):
nodes/task_node.py
+5 -4
View File
@@ -11,18 +11,19 @@ class TaskNode(BaseNode):
return {
"required": {
"description": ("STRING", {
"forceInput": True, "multiline": True, "default": ""
"multiline": True, "default": ""
}),
},
"optional": {
"expected_output": ("STRING", {
"forceInput": False, "multiline": True, "default": ""
"multiline": True, "default": ""
}),
"agent": ("CAI_AGENT",),
"agent": ("CREWAI_AGENT",),
}
}
RETURN_TYPES = ("CAI_TASK",)
RETURN_TYPES = ("CREWAI_TASK",)
RETURN_NAMES = ("task",)
FUNCTION = "create_task"
def create_task(self, description, expected_output, agent):