NeoAnd/evo-ai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

✨ feat: Create delay node

Browse Source
This commit is contained in:
Victor Calazans 2025-05-16 22:25:39 -03:00
parent 9f176bf0e0
commit 257d50a584
1 changed files with 70 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
src/services/custom_agents/workflow_agent.py
View File

@ -320,10 +320,9 @@ class WorkflowAgent(BaseAgent):
)
]
),
)
]
) ]
content = content + condition_content
yield {
"content": content,
"status": "condition_evaluated",
@ -332,7 +331,7 @@ class WorkflowAgent(BaseAgent):
"conversation_history": conversation_history,
"session_id": session_id,
}
async def message_node_function(
state: State, node_id: str, node_data: Dict[str, Any]
) -> AsyncGenerator[State, None]:
@ -365,15 +364,81 @@ class WorkflowAgent(BaseAgent):
"status": "message_added",
"node_outputs": node_outputs,
"cycle_count": state.get("cycle_count", 0),
"conversation_history": conversation_history, "session_id": session_id,
}
async def delay_node_function(
state: State, node_id: str, node_data: Dict[str, Any]
) -> AsyncGenerator[State, None]:
delay_data = node_data.get("delay", {})
delay_value = delay_data.get("value", 0)
delay_unit = delay_data.get("unit", "seconds")
delay_description = delay_data.get("description", "")
# Convert to seconds based on unit
delay_seconds = delay_value
if delay_unit == "minutes":
delay_seconds = delay_value * 60
elif delay_unit == "hours":
delay_seconds = delay_value * 3600
label = node_data.get("label", "delay_node")
print(f"\n⏱️ DELAY-NODE: {delay_value} {delay_unit} - {delay_description}")
content = state.get("content", [])
session_id = state.get("session_id", "")
conversation_history = state.get("conversation_history", [])
# Add a message indicating the delay
delay_message = f"Aguardando {delay_value} {delay_unit}..."
if delay_description:
delay_message += f" ({delay_description})"
new_event = Event(
author=label,
content=Content(parts=[Part(text=delay_message)]),
)
content = content + [new_event]
# Store node output information
node_outputs = state.get("node_outputs", {})
node_outputs[node_id] = {
"delay_value": delay_value,
"delay_unit": delay_unit,
"delay_seconds": delay_seconds,
"delay_start_time": datetime.now().isoformat(),
}
# Actually perform the delay
import asyncio
await asyncio.sleep(delay_seconds)
# Add completion message
complete_message = f"Delay de {delay_value} {delay_unit} concluído."
complete_event = Event(
author=label,
content=Content(parts=[Part(text=complete_message)]),
)
content = content + [complete_event]
# Update node outputs with completion information
node_outputs[node_id]["delay_end_time"] = datetime.now().isoformat()
node_outputs[node_id]["delay_completed"] = True
yield {
"content": content,
"status": "delay_completed",
"node_outputs": node_outputs, "cycle_count": state.get("cycle_count", 0),
"conversation_history": conversation_history,
"session_id": session_id,
}
return {
"start-node": start_node_function,
"agent-node": agent_node_function,
"condition-node": condition_node_function,
"message-node": message_node_function,
"delay-node": delay_node_function,
}
def _evaluate_condition(self, condition: Dict[str, Any], state: State) -> bool: