Skip to content

Commit

Permalink
fix: add agent transfer
Browse files Browse the repository at this point in the history
  • Loading branch information
yongenaelf committed Nov 6, 2024
1 parent c3357bc commit aac4f05
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions examples/software_team/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,3 +36,20 @@ def get_status(task):
instructions="You are responsible for managing the project and coordinating between the Developer and Tester agents.",
functions=[manage_project, get_status],
)

def transfer_to_developer():
"""Function to transfer the conversation to the Developer agent."""
return developer_agent

def transfer_to_tester():
"""Function to transfer the conversation to the Tester agent."""
return tester_agent

def transfer_to_project_manager():
"""Function to transfer the conversation to the Project Manager agent."""
return project_manager_agent

developer_agent.functions.append(transfer_to_project_manager)
tester_agent.functions.append(transfer_to_project_manager)
project_manager_agent.functions.append(transfer_to_developer)
project_manager_agent.functions.append(transfer_to_tester)

0 comments on commit aac4f05

Please sign in to comment.