Abdelrahman Abdallah
a5629295ac
New runtime/code: - src/ask_user_runtime.py - src/team_runtime.py New real tools in src/agent_tools.py: - ask_user_question - team_create - team_delete - team_list - team_get - send_message - team_messages - notebook_edit
77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from src.agent_tools import build_tool_context, default_tool_registry, execute_tool
|
|
from src.agent_types import AgentPermissions, AgentRuntimeConfig
|
|
from src.team_runtime import TeamRuntime
|
|
|
|
|
|
class TeamRuntimeTests(unittest.TestCase):
|
|
def test_team_runtime_persists_team_and_messages(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
workspace = Path(tmp_dir)
|
|
runtime = TeamRuntime.from_workspace(workspace)
|
|
runtime.create_team('reviewers', members=['alice', 'bob'])
|
|
runtime.send_message(
|
|
team_name='reviewers',
|
|
text='Please review the patch.',
|
|
sender='agent',
|
|
recipient='alice',
|
|
)
|
|
restored = TeamRuntime.from_workspace(workspace)
|
|
|
|
self.assertEqual(len(restored.teams), 1)
|
|
self.assertEqual(restored.teams[0].name, 'reviewers')
|
|
self.assertEqual(len(restored.messages), 1)
|
|
self.assertIn('Please review the patch.', restored.render_messages())
|
|
|
|
def test_team_tools_execute_against_runtime(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
workspace = Path(tmp_dir)
|
|
runtime = TeamRuntime.from_workspace(workspace)
|
|
context = build_tool_context(
|
|
AgentRuntimeConfig(
|
|
cwd=workspace,
|
|
permissions=AgentPermissions(allow_file_write=True),
|
|
),
|
|
team_runtime=runtime,
|
|
)
|
|
create_result = execute_tool(
|
|
default_tool_registry(),
|
|
'team_create',
|
|
{'team_name': 'reviewers', 'members': ['alice', 'bob']},
|
|
context,
|
|
)
|
|
send_result = execute_tool(
|
|
default_tool_registry(),
|
|
'send_message',
|
|
{
|
|
'team_name': 'reviewers',
|
|
'message': 'Check notebook changes',
|
|
'sender': 'agent',
|
|
},
|
|
context,
|
|
)
|
|
list_result = execute_tool(
|
|
default_tool_registry(),
|
|
'team_list',
|
|
{},
|
|
context,
|
|
)
|
|
message_result = execute_tool(
|
|
default_tool_registry(),
|
|
'team_messages',
|
|
{'team_name': 'reviewers'},
|
|
context,
|
|
)
|
|
|
|
self.assertTrue(create_result.ok)
|
|
self.assertEqual(create_result.metadata.get('action'), 'team_create')
|
|
self.assertTrue(send_result.ok)
|
|
self.assertEqual(send_result.metadata.get('action'), 'send_message')
|
|
self.assertIn('reviewers', list_result.content)
|
|
self.assertIn('Check notebook changes', message_result.content)
|