Update workflow and get tests to pass.

Author: ericpts

.github/workflows/main.yaml

@@ -9,7 +9,7 @@ on:
       - main
 
 jobs:
-  test-apiserver:
+  test-pytest:
     runs-on: ubuntu-latest
 
     strategy:
@@ -28,34 +28,9 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -r apiserver/requirements.txt
+        pip install -r requirements.txt
 
-    - name: Run pytest in ./apiserver
-      working-directory: ./apiserver
+    - name: Run pytest
       run: |
         pytest
 
-  test-puppeteer:
-    runs-on: ubuntu-latest
-    needs: [] # Runs in parallel with test-apiserver
-    
-    defaults:
-      run:
-        working-directory: ./computers/puppeteer
-
-    steps:
-    - name: Check out repository
-      uses: actions/checkout@v4
-
-    - name: Set up Node.js
-      uses: actions/setup-node@v4
-      with:
-        node-version: '22.x'
-        cache: 'npm'
-        cache-dependency-path: computers/puppeteer/package-lock.json
-
-    - name: Install npm dependencies for Puppeteer
-      run: npm ci
-
-    - name: Run npm tests for Puppeteer
-      run: npm test

requirements.txt

@@ -4,4 +4,5 @@ pydantic==2.11.4
 playwright==1.52.0
 browserbase==1.3.0
 rich
-hud-python==0.2.10
+hud-python==0.2.10
+pytest

test_agent.py

@@ -31,7 +31,7 @@ def setUp(self):
         self.agent._client = MagicMock()
 
     def test_multiply_numbers(self):
-        self.assertEqual(multiply_numbers(2, 3), {"result": 5})
+        self.assertEqual(multiply_numbers(2, 3), {"result": 6})
 
     def test_handle_action_open_web_browser(self):
         action = types.FunctionCall(name="open_web_browser", args={})
@@ -86,8 +86,8 @@ def test_run_one_iteration_no_function_calls(self, mock_get_model_response):
         self.assertEqual(self.agent._contents[1], mock_candidate.content)
 
     @patch('agent.BrowserAgent.get_model_response')
-    @patch('agent.BrowserAgent._execute_function_call')
-    def test_run_one_iteration_with_function_call(self, mock_execute_function_call, mock_get_model_response):
+    @patch('agent.BrowserAgent.handle_action')
+    def test_run_one_iteration_with_function_call(self, mock_handle_action, mock_get_model_response):
         mock_response = MagicMock()
         mock_candidate = MagicMock()
         function_call = types.FunctionCall(name="navigate", args={"url": "https://example.com"})
@@ -96,12 +96,12 @@ def test_run_one_iteration_with_function_call(self, mock_execute_function_call,
         mock_get_model_response.return_value = mock_response
 
         mock_env_state = EnvState(screenshot=b"screenshot", url="https://example.com")
-        mock_execute_function_call.return_value = mock_env_state
+        mock_handle_action.return_value = mock_env_state
 
         result = self.agent.run_one_iteration()
 
         self.assertEqual(result, "CONTINUE")
-        mock_execute_function_call.assert_called_once_with(function_call)
+        mock_handle_action.assert_called_once_with(function_call)
         self.assertEqual(len(self.agent._contents), 3)