Fix Recruitment Workflow

TransformerOptimus · Oct 3, 2023 · d57f107 · d57f107
1 parent 0a24481
commit d57f107
Showing 1 changed file with 12 additions and 13 deletions.
diff --git a/superagi/agent/workflow_seed.py b/superagi/agent/workflow_seed.py
@@ -112,7 +112,7 @@ def build_recruitment_workflow(cls, session):
         step1 = AgentWorkflowStep.find_or_create_tool_workflow_step(session, agent_workflow.id,
                                                                     str(agent_workflow.id) + "_step1",
                                                                     ListFileTool().name,
-                                                                    "Read files from the resource manager",
+                                                                    "List the files from the resource manager",
                                                                     step_type="TRIGGER")
 
         # task queue ends when the elements gets over
@@ -125,29 +125,28 @@ def build_recruitment_workflow(cls, session):
         step3 = AgentWorkflowStep.find_or_create_tool_workflow_step(session, agent_workflow.id,
                                                                     str(agent_workflow.id) + "_step3",
                                                                     ReadFileTool().name,
-                                                                    "Read the resume from above input")
-
-        step4 = AgentWorkflowStep.find_or_create_tool_workflow_step(session, agent_workflow.id,
-                                                                    str(agent_workflow.id) + "_step4",
-                                                                    ReadFileTool().name,
-                                                                    "Read the job description from file mentioned in High-Level GOAL",
-                                                                    "Check if the resume matches the job description in goal")
+                                                                    "Read the resume from above input",
+                                                                    "Check if the resume matches High-Level GOAL")
 
         step5 = AgentWorkflowStep.find_or_create_tool_workflow_step(session, agent_workflow.id,
                                                                     str(agent_workflow.id) + "_step5",
                                                                     SendEmailTool().name,
-                                                                    "Write a custom Email the candidates for job profile based on their experience")
+                                                                    "Write a custom acceptance Email to the candidates")
+
+        step6 = AgentWorkflowStep.find_or_create_tool_workflow_step(session, agent_workflow.id,
+                                                                    str(agent_workflow.id) + "_step6",
+                                                                    SendEmailTool().name,
+                                                                    "Write a custom Reject Email to the candidates")
 
         AgentWorkflowStep.add_next_workflow_step(session, step1.id, step2.id)
         AgentWorkflowStep.add_next_workflow_step(session, step2.id, step3.id)
         AgentWorkflowStep.add_next_workflow_step(session, step2.id, -1, "COMPLETE")
-        AgentWorkflowStep.add_next_workflow_step(session, step3.id, step4.id)
-        AgentWorkflowStep.add_next_workflow_step(session, step4.id, step5.id, "YES")
-        AgentWorkflowStep.add_next_workflow_step(session, step4.id, step2.id, "NO")
+        AgentWorkflowStep.add_next_workflow_step(session, step3.id, step5.id, "YES")
+        AgentWorkflowStep.add_next_workflow_step(session, step3.id, step6.id, "NO")
         AgentWorkflowStep.add_next_workflow_step(session, step5.id, step2.id)
+        AgentWorkflowStep.add_next_workflow_step(session, step6.id, step2.id)
         session.commit()
 
-
     @classmethod
     def build_coding_workflow(cls, session):
         agent_workflow = AgentWorkflow.find_or_create_by_name(session, "SuperCoder", "SuperCoder")