apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-13 17:22:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add agent update events for code-based transitions

Browse source
This commit is contained in:
akhisud3195 2025-05-07 14:18:50 +05:30
parent e1b1b63980
commit 92fb1819c7
1 changed files with 82 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

82
apps/rowboat_agents/src/graph/core.py
View file

@ -379,6 +379,47 @@ async def run_turn_streamed(
print(f"\nSkipping agent {current_agent.name} because it has already responded")
if parent_stack:
print(f"-- Returning to parent agent {parent_stack[-1].name}")
# Create tool call for control transition
tool_call_id = str(uuid.uuid4())
transition_message = {
'content': None,
'role': 'assistant',
'sender': current_agent.name,
'tool_calls': [{
'function': {
'name': 'transfer_to_agent',
'arguments': json.dumps({
'assistant': parent_stack[-1].name
})
},
'id': tool_call_id,
'type': 'function'
}],
'tool_call_id': None,
'tool_name': None,
'response_type': ResponseType.INTERNAL.value
}
print('-'*100)
print(f"Yielding control transition message: {transition_message}")
print('-'*100)
yield ('message', transition_message)
# Create tool response for control transition
transition_response = {
'content': json.dumps({
'assistant': parent_stack[-1].name
}),
'role': 'tool',
'sender': None,
'tool_calls': None,
'tool_call_id': tool_call_id,
'tool_name': 'transfer_to_agent'
}
print('-'*100)
print(f"Yielding control transition response: {transition_response}")
print('-'*100)
yield ('message', transition_response)
current_agent = parent_stack.pop()
continue
else:
@ -413,6 +454,47 @@ async def run_turn_streamed(
accumulated_messages.append(message)
# Return to parent or end turn
if is_internal and parent_stack:
# Create tool call for control transition
tool_call_id = str(uuid.uuid4())
transition_message = {
'content': None,
'role': 'assistant',
'sender': current_agent.name,
'tool_calls': [{
'function': {
'name': 'transfer_to_agent',
'arguments': json.dumps({
'assistant': parent_stack[-1].name
})
},
'id': tool_call_id,
'type': 'function'
}],
'tool_call_id': None,
'tool_name': None,
'response_type': ResponseType.INTERNAL.value
}
print('-'*100)
print(f"Yielding control transition message: {transition_message}")
print('-'*100)
yield ('message', transition_message)
# Create tool response for control transition
transition_response = {
'content': json.dumps({
'assistant': parent_stack[-1].name
}),
'role': 'tool',
'sender': None,
'tool_calls': None,
'tool_call_id': tool_call_id,
'tool_name': 'transfer_to_agent'
}
print('-'*100)
print(f"Yielding control transition response: {transition_response}")
print('-'*100)
yield ('message', transition_response)
current_agent = parent_stack.pop()
continue
elif not is_internal: