Implemented the next parity slice.

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
2026-04-07 02:51:30 +02:00
parent a54c90b18f
commit a5629295ac
21 changed files with 1886 additions and 24 deletions
@@ -0,0 +1,76 @@
+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)