Fix variable scope issue in get_authenticated_git_url method

- Move provider and repo_name variable initialization outside try block
- Initialize with default values before attempting repository verification
- Ensures variables are always available regardless of exception path
- Add comprehensive unit tests to verify the fix works correctly

Co-authored-by: openhands <openhands@all-hands.dev>

.github/workflows/enterprise-preview.yml

@@ -1,29 +0,0 @@
-# Feature branch preview for enterprise code
-name: Enterprise Preview
-
-# Run on PRs labeled
-on:
-  pull_request:
-    types: [labeled]
-
-# Match ghcr-build.yml, but don't interrupt it.
-concurrency:
-  group: ${{ github.workflow }}-${{ (github.head_ref && github.ref) || github.run_id }}
-  cancel-in-progress: false
-
-jobs:
-  # This must happen for the PR Docker workflow when the label is present,
-  # and also if it's added after the fact. Thus, it exists in both places.
-  enterprise-preview:
-    name: Enterprise preview
-    if: github.event.label.name == 'deploy'
-    runs-on: blacksmith-4vcpu-ubuntu-2204
-    steps:
-      # This should match the version in ghcr-build.yml
-      - name: Trigger remote job
-        run: |
-          curl --fail-with-body -sS -X POST \
-            -H "Authorization: Bearer ${{ secrets.PAT_TOKEN }}" \
-            -H "Accept: application/vnd.github+json" \
-            -d "{\"ref\": \"main\", \"inputs\": {\"openhandsPrNumber\": \"${{ github.event.pull_request.number }}\", \"deployEnvironment\": \"feature\", \"enterpriseImageTag\": \"pr-${{ github.event.pull_request.number }}\" }}" \
-            https://api.github.com/repos/All-Hands-AI/deploy/actions/workflows/deploy.yaml/dispatches

.github/workflows/ghcr-build.yml

@@ -59,7 +59,6 @@ jobs:
     permissions:
       contents: read
       packages: write
-      security-events: write
     steps:
       - name: Checkout
         uses: actions/checkout@v4
@@ -86,35 +85,14 @@ jobs:
         run: |
           ./containers/build.sh -i openhands -o ${{ env.REPO_OWNER }} --push
 
-      - name: Run Trivy vulnerability scanner
-        if: github.event_name != 'pull_request'
-        uses: aquasecurity/trivy-action@master
-        # This only reports, does not fail the build on CVE.
-        with:
-          image-ref: ghcr.io/${{ env.REPO_OWNER }}/openhands:${{ env.RELEVANT_SHA }}
-          format: 'sarif'
-          output: 'trivy-results.sarif'
-          timeout: '10m'
-          scanners: 'vuln'  # Only scan vulnerabilities, not secrets/config
-          severity: 'CRITICAL,HIGH,MEDIUM'  # Skip LOW severity
-
-      - name: Upload Trivy results to GitHub Security tab
-        if: github.event_name != 'pull_request'
-        uses: github/codeql-action/upload-sarif@v3
-        with:
-          sarif_file: 'trivy-results.sarif'
-          category: 'security/container-openhands-${{ matrix.base_image.tag }}'
-          sha: "${{ env.RELEVANT_SHA }}"
-          ref: refs/heads/${{ github.event.pull_request.head.ref || github.ref_name }}
   # Builds the runtime Docker images
   ghcr_build_runtime:
-    name: Build Runtime Image
+    name: Build Image
     runs-on: blacksmith-8vcpu-ubuntu-2204
     if: "!(github.event_name == 'push' && startsWith(github.ref, 'refs/tags/ext-v'))"
     permissions:
       contents: read
       packages: write
-      security-events: write
     needs: define-matrix
     strategy:
       matrix:
@@ -188,49 +166,18 @@ jobs:
           name: runtime-src-${{ matrix.base_image.tag }}
           path: containers/runtime
 
-      - name: Run Trivy vulnerability scanner
-        if: github.event_name != 'pull_request'
-        uses: aquasecurity/trivy-action@master
-        # This only reports, does not fail the build on CVE.
-        with:
-          image-ref: ghcr.io/${{ env.REPO_OWNER }}/runtime:${{ env.RELEVANT_SHA }}-${{ matrix.base_image.tag }}
-          format: 'sarif'
-          output: 'trivy-results-raw.sarif'
-          timeout: '10m'
-          scanners: 'vuln'  # Only scan vulnerabilities, not secrets/config
-          severity: 'CRITICAL,HIGH,MEDIUM'  # Skip LOW severity
-      - name: Customize SARIF with image flavor
-        if: github.event_name != 'pull_request'
-        run: |
-          # Modify the tool name to include the image flavor
-          jq --arg flavor "${{ matrix.base_image.tag }}" \
-            '.runs[0].tool.driver.name = "Trivy (" + $flavor + ")"' \
-            trivy-results-raw.sarif > trivy-results.sarif
-
-          echo "Modified tool name to: $(jq -r '.runs[0].tool.driver.name' trivy-results.sarif)"
-
-      - name: Upload Trivy results to GitHub Security tab
-        if: github.event_name != 'pull_request'
-        uses: github/codeql-action/upload-sarif@v3
-        with:
-          sarif_file: 'trivy-results.sarif'
-          category: 'security/container-runtime-${{ matrix.base_image.tag }}'
-
   ghcr_build_enterprise:
-    name: Build Enterprise Image
+    name: Push Enterprise Image
     runs-on: blacksmith-8vcpu-ubuntu-2204
     permissions:
       contents: read
       packages: write
-      security-events: write
     needs: [define-matrix, ghcr_build_app]
     # Do not build enterprise in forks
     if: github.event.pull_request.head.repo.fork != true
     steps:
-      - name: Checkout
+      - name: Checkout repository
         uses: actions/checkout@v4
-        with:
-          ref: ${{ github.event.pull_request.head.sha }}
 
       # Set up Docker Buildx for better performance
       - name: Set up Docker Buildx
@@ -262,16 +209,14 @@ jobs:
             latest=auto
             prefix=
             suffix=
-        env:
-          DOCKER_METADATA_PR_HEAD_SHA: true
       - name: Determine app image tag
         shell: bash
         run: |
           # Duplicated with build.sh
-          sanitized_ref_name=$(echo "${{github.event.pull_request.head.sha}}" | sed 's/[^a-zA-Z0-9.-]\+/-/g')
+          sanitized_ref_name=$(echo "$GITHUB_REF_NAME" | sed 's/[^a-zA-Z0-9.-]\+/-/g')
+          OPENHANDS_BUILD_VERSION=$sanitized_ref_name
           sanitized_ref_name=$(echo "$sanitized_ref_name" | tr '[:upper:]' '[:lower:]') # lower case is required in tagging
-          echo "OPENHANDS_BUILD_VERSION=${sanitized_ref_name}" >> $GITHUB_ENV
-          echo REPO_OWNER=$(echo ${{ github.repository_owner }} | tr '[:upper:]' '[:lower:]') >> $GITHUB_ENV
+          echo "OPENHANDS_DOCKER_TAG=${sanitized_ref_name}" >> $GITHUB_ENV
       - name: Build and push Docker image
         uses: useblacksmith/build-push-action@v1
         with:
@@ -281,38 +226,21 @@ jobs:
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           build-args: |
-            OPENHANDS_VERSION=${{ env.OPENHANDS_BUILD_VERSION }}
+            OPENHANDS_VERSION=${{ env.OPENHANDS_DOCKER_TAG }}
           platforms: linux/amd64
           # Add build provenance
           provenance: true
           # Add build attestations for better security
           sbom: true
-      - name: Run Trivy vulnerability scanner
-        uses: aquasecurity/trivy-action@master
-        if: github.event_name != 'pull_request'
-        # This only reports, does not fail the build on CVE.
-        with:
-          image-ref: "ghcr.io/${{ env.REPO_OWNER }}/enterprise-server:sha-${{ env.RELEVANT_SHA }}"
-          format: 'sarif'
-          output: 'trivy-results.sarif'
-          timeout: '10m'
-          scanners: 'vuln'  # Only scan vulnerabilities, not secrets/config
-          severity: 'CRITICAL,HIGH,MEDIUM'  # Skip LOW severity
-
-      - name: Upload Trivy results to GitHub Security tab
-        if: github.event_name != 'pull_request'
-        uses: github/codeql-action/upload-sarif@v3
-        with:
-          sarif_file: 'trivy-results.sarif'
-          category: 'security/container-enterprise-server'
 
   enterprise-preview:
     name: Enterprise preview
-    if: github.event_name == 'pull_request' && contains(github.event.pull_request.labels.*.name, 'deploy')
+    if: |
+      (github.event_name == 'pull_request' && github.event.action == 'labeled' && github.event.label.name == 'deploy') ||
+      (github.event_name == 'pull_request' && github.event.action != 'labeled' && contains(github.event.pull_request.labels.*.name, 'deploy'))
     runs-on: blacksmith-4vcpu-ubuntu-2204
     needs: [ghcr_build_enterprise]
     steps:
-      # This should match the version in enterprise-preview.yml
       - name: Trigger remote job
         run: |
           curl --fail-with-body -sS -X POST \

.github/workflows/mdx-lint.yml

@@ -1,70 +0,0 @@
-# Workflow that checks MDX format in docs/ folder
-name: MDX Lint
-
-# Run on pushes to main and on pull requests that modify docs/ files
-on:
-  push:
-    branches:
-      - main
-    paths:
-      - 'docs/**/*.mdx'
-  pull_request:
-    paths:
-      - 'docs/**/*.mdx'
-
-# If triggered by a PR, it will be in the same group. However, each commit on main will be in its own unique group
-concurrency:
-  group: ${{ github.workflow }}-${{ (github.head_ref && github.ref) || github.run_id }}
-  cancel-in-progress: true
-
-jobs:
-  mdx-lint:
-    name: Lint MDX files
-    runs-on: blacksmith-4vcpu-ubuntu-2204
-    steps:
-      - uses: actions/checkout@v4
-
-      - name: Install Node.js 22
-        uses: useblacksmith/setup-node@v5
-        with:
-          node-version: 22
-
-      - name: Install MDX dependencies
-        run: |
-          npm install @mdx-js/mdx@3 glob@10
-
-      - name: Validate MDX files
-        run: |
-          node -e "
-          const {compile} = require('@mdx-js/mdx');
-          const fs = require('fs');
-          const path = require('path');
-          const glob = require('glob');
-
-          async function validateMDXFiles() {
-            const files = glob.sync('docs/**/*.mdx');
-            console.log('Found', files.length, 'MDX files to validate');
-
-            let hasErrors = false;
-
-            for (const file of files) {
-              try {
-                const content = fs.readFileSync(file, 'utf8');
-                await compile(content);
-                console.log('✅ MDX parsing successful for', file);
-              } catch (err) {
-                console.error('❌ MDX parsing failed for', file, ':', err.message);
-                hasErrors = true;
-              }
-            }
-
-            if (hasErrors) {
-              console.error('\\n❌ Some MDX files have parsing errors. Please fix them before merging.');
-              process.exit(1);
-            } else {
-              console.log('\\n✅ All MDX files are valid!');
-            }
-          }
-
-          validateMDXFiles();
-          "

.github/workflows/py-tests.yml

@@ -19,16 +19,12 @@ jobs:
   # Run python tests on Linux
   test-on-linux:
     name: Python Tests on Linux
-    runs-on: blacksmith-4vcpu-ubuntu-2404
+    runs-on: blacksmith-4vcpu-ubuntu-2204
     env:
       INSTALL_DOCKER: "0" # Set to '0' to skip Docker installation
     strategy:
       matrix:
         python-version: ["3.12"]
-    permissions:
-      # For coverage comment and python-coverage-comment-action branch
-      pull-requests: write
-      contents: write
     steps:
       - uses: actions/checkout@v4
       - name: Set up Docker Buildx
@@ -52,21 +48,10 @@ jobs:
       - name: Build Environment
         run: make build
       - name: Run Unit Tests
-        run: PYTHONPATH=".:$PYTHONPATH" poetry run pytest --forked -n auto -s ./tests/unit --cov=openhands --cov-branch
-        env:
-          COVERAGE_FILE: ".coverage.${{ matrix.python_version }}"
+        run: PYTHONPATH=".:$PYTHONPATH" poetry run pytest --forked -n auto -svv ./tests/unit
       - name: Run Runtime Tests with CLIRuntime
-        run: PYTHONPATH=".:$PYTHONPATH" TEST_RUNTIME=cli poetry run pytest -s tests/runtime/test_bash.py --cov=openhands --cov-branch
-        env:
-          COVERAGE_FILE: ".coverage.runtime.${{ matrix.python_version }}"
-      - name: Store coverage file
-        uses: actions/upload-artifact@v4
-        with:
-          name: coverage-openhands
-          path: |
-            .coverage.${{ matrix.python_version }}
-            .coverage.runtime.${{ matrix.python_version }}
-          include-hidden-files: true
+        run: PYTHONPATH=".:$PYTHONPATH" TEST_RUNTIME=cli poetry run pytest -svv tests/runtime/test_bash.py
+
   # Run specific Windows python tests
   test-on-windows:
     name: Python Tests on Windows
@@ -100,7 +85,7 @@ jobs:
           DEBUG: "1"
   test-enterprise:
     name: Enterprise Python Unit Tests
-    runs-on: blacksmith-4vcpu-ubuntu-2404
+    runs-on: blacksmith-4vcpu-ubuntu-2204
     strategy:
       matrix:
         python-version: ["3.12"]
@@ -117,37 +102,5 @@ jobs:
         working-directory: ./enterprise
         run: poetry install --with dev,test
       - name: Run Unit Tests
-        # Use base working directory for coverage paths to line up.
-        run: PYTHONPATH=".:$PYTHONPATH" poetry run --project=enterprise pytest --forked -n auto -s -p no:ddtrace -p no:ddtrace.pytest_bdd -p no:ddtrace.pytest_benchmark ./enterprise/tests/unit --cov=enterprise --cov-branch
-        env:
-          COVERAGE_FILE: ".coverage.enterprise.${{ matrix.python_version }}"
-      - name: Store coverage file
-        uses: actions/upload-artifact@v4
-        with:
-          name: coverage-enterprise
-          path: ".coverage.enterprise.${{ matrix.python_version }}"
-          include-hidden-files: true
-  coverage-comment:
-    name: Coverage Comment
-    if: github.event_name == 'pull_request'
-    runs-on: ubuntu-latest
-    needs: [test-on-linux, test-enterprise]
-
-    permissions:
-      pull-requests: write
-      contents: write
-    steps:
-      - uses: actions/checkout@v4
-
-      - uses: actions/download-artifact@v5
-        id: download
-        with:
-          pattern: coverage-*
-          merge-multiple: true
-
-      - name: Coverage comment
-        id: coverage_comment
-        uses: py-cov-action/python-coverage-comment-action@v3
-        with:
-          GITHUB_TOKEN: ${{ github.token }}
-          MERGE_COVERAGE_FILES: true
+        working-directory: ./enterprise
+        run: PYTHONPATH=".:$PYTHONPATH" poetry run pytest --forked -n auto -svv -p no:ddtrace -p no:ddtrace.pytest_bdd -p no:ddtrace.pytest_benchmark ./tests/unit

.github/workflows/stale.yml

@@ -15,7 +15,7 @@ jobs:
           stale-issue-message: 'This issue is stale because it has been open for 40 days with no activity. Remove the stale label or leave a comment, otherwise it will be closed in 10 days.'
           stale-pr-message: 'This PR is stale because it has been open for 40 days with no activity. Remove the stale label or leave a comment, otherwise it will be closed in 10 days.'
           days-before-stale: 40
-          exempt-issue-labels: roadmap,backlog,app-team
+          exempt-issue-labels: roadmap,backlog
           close-issue-message: 'This issue was automatically closed due to 50 days of inactivity. We do this to help keep the issues somewhat manageable and focus on active issues.'
           close-pr-message: 'This PR was closed because it had no activity for 50 days. If you feel this was closed in error, and you would like to continue the PR, please resubmit or let us know.'
           days-before-close: 10

Development.md

@@ -159,7 +159,7 @@ poetry run pytest ./tests/unit/test_*.py
 To reduce build time (e.g., if no changes were made to the client-runtime component), you can use an existing Docker
 container image by setting the SANDBOX_RUNTIME_CONTAINER_IMAGE environment variable to the desired Docker image.
 
-Example: `export SANDBOX_RUNTIME_CONTAINER_IMAGE=ghcr.io/all-hands-ai/runtime:0.57-nikolaik`
+Example: `export SANDBOX_RUNTIME_CONTAINER_IMAGE=ghcr.io/all-hands-ai/runtime:0.55-nikolaik`
 
 ## Develop inside Docker container
 

README.md

@@ -79,17 +79,17 @@ You'll find OpenHands running at [http://localhost:3000](http://localhost:3000)
 You can also run OpenHands directly with Docker:
 
 ```bash
-docker pull docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik
+docker pull docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik
 
 docker run -it --rm --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e LOG_ALL_EVENTS=true \
     -v /var/run/docker.sock:/var/run/docker.sock \
     -v ~/.openhands:/.openhands \
     -p 3000:3000 \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57
+    docker.all-hands.dev/all-hands-ai/openhands:0.55
 ```
 
 </details>

README_CN.md

@@ -51,17 +51,17 @@ OpenHands也可以使用Docker在本地系统上运行。
 
 
 ```bash
-docker pull docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik
+docker pull docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik
 
 docker run -it --rm --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e LOG_ALL_EVENTS=true \
     -v /var/run/docker.sock:/var/run/docker.sock \
     -v ~/.openhands:/.openhands \
     -p 3000:3000 \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57
+    docker.all-hands.dev/all-hands-ai/openhands:0.55
 ```
 
 > **注意**: 如果您在0.44版本之前使用过OpenHands，您可能需要运行 `mv ~/.openhands-state ~/.openhands` 来将对话历史迁移到新位置。

README_JA.md

@@ -42,17 +42,17 @@ OpenHandsはDockerを利用してローカル環境でも実行できます。
 > 公共ネットワークで実行していますか？[Hardened Docker Installation Guide](https://docs.all-hands.dev/usage/runtimes/docker#hardened-docker-installation)を参照して、ネットワークバインディングの制限や追加のセキュリティ対策を実施してください。
 
 ```bash
-docker pull docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik
+docker pull docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik
 
 docker run -it --rm --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e LOG_ALL_EVENTS=true \
     -v /var/run/docker.sock:/var/run/docker.sock \
     -v ~/.openhands:/.openhands \
     -p 3000:3000 \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57
+    docker.all-hands.dev/all-hands-ai/openhands:0.55
 ```
 
 **注**: バージョン0.44以前のOpenHandsを使用していた場合は、会話履歴を移行するために `mv ~/.openhands-state ~/.openhands` を実行してください。

config.template.toml

@@ -219,14 +219,6 @@ correct_num = 5
 api_key = ""
 model = "gpt-4o"
 
-# Example routing LLM configuration for multimodal model routing
-# Uncomment and configure to enable model routing with a secondary model
-#[llm.secondary_model]
-#model = "kimi-k2"
-#api_key = ""
-#for_routing = true
-#max_input_tokens = 128000
-
 
 #################################### Agent ###################################
 # Configuration for agents (group name starts with 'agent')
@@ -488,55 +480,3 @@ type = "noop"
 
 # Run the runtime sandbox container in privileged mode for use with docker-in-docker
 #privileged = false
-
-#################################### MCP #####################################
-# Configuration for Model Context Protocol (MCP) servers
-# MCP allows OpenHands to communicate with external tool servers
-##############################################################################
-[mcp]
-# SSE servers - Server-Sent Events transport (legacy)
-#sse_servers = [
-#    # Basic SSE server with just a URL
-#    "http://localhost:8080/mcp/sse",
-#
-#    # SSE server with authentication
-#    {url = "https://api.example.com/mcp/sse", api_key = "your-api-key"}
-#]
-
-# SHTTP servers - Streamable HTTP transport (recommended)
-#shttp_servers = [
-#    # Basic SHTTP server with default 60s timeout
-#    "https://api.example.com/mcp/shttp",
-#
-#    # SHTTP server with custom timeout for long-running tools
-#    {
-#        url = "https://api.example.com/mcp/shttp",
-#        api_key = "your-api-key",
-#        timeout = 180  # 3 minutes for processing-heavy tools (1-3600 seconds)
-#    }
-#]
-
-# Stdio servers - Direct process communication (development only)
-#stdio_servers = [
-#    # Basic stdio server
-#    {name = "filesystem", command = "npx", args = ["@modelcontextprotocol/server-filesystem", "/"]},
-#
-#    # Stdio server with environment variables
-#    {
-#        name = "fetch",
-#        command = "uvx",
-#        args = ["mcp-server-fetch"],
-#        env = {DEBUG = "true"}
-#    }
-#]
-
-#################################### Model Routing ############################
-# Configuration for experimental model routing feature
-# Enables intelligent switching between different LLM models for specific purposes
-##############################################################################
-[model_routing]
-# Router to use for model selection
-# Available options:
-# - "noop_router" (default): No routing, always uses primary LLM
-# - "multimodal_router": A router that switches between primary and secondary models, depending on whether the input is multimodal or not
-#router_name = "noop_router"

containers/dev/compose.yml

@@ -12,7 +12,7 @@ services:
       - SANDBOX_API_HOSTNAME=host.docker.internal
       - DOCKER_HOST_ADDR=host.docker.internal
       #
-      - SANDBOX_RUNTIME_CONTAINER_IMAGE=${SANDBOX_RUNTIME_CONTAINER_IMAGE:-ghcr.io/all-hands-ai/runtime:0.57-nikolaik}
+      - SANDBOX_RUNTIME_CONTAINER_IMAGE=${SANDBOX_RUNTIME_CONTAINER_IMAGE:-ghcr.io/all-hands-ai/runtime:0.55-nikolaik}
       - SANDBOX_USER_ID=${SANDBOX_USER_ID:-1234}
       - WORKSPACE_MOUNT_PATH=${WORKSPACE_BASE:-$PWD/workspace}
     ports:

docker-compose.yml

@@ -7,7 +7,7 @@ services:
     image: openhands:latest
     container_name: openhands-app-${DATE:-}
     environment:
-      - SANDBOX_RUNTIME_CONTAINER_IMAGE=${SANDBOX_RUNTIME_CONTAINER_IMAGE:-docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik}
+      - SANDBOX_RUNTIME_CONTAINER_IMAGE=${SANDBOX_RUNTIME_CONTAINER_IMAGE:-docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik}
       #- SANDBOX_USER_ID=${SANDBOX_USER_ID:-1234} # enable this only if you want a specific non-root sandbox user but you will have to manually adjust permissions of ~/.openhands for this user
       - WORKSPACE_MOUNT_PATH=${WORKSPACE_BASE:-$PWD/workspace}
     ports:

docs/usage/architecture/runtime.mdx

@@ -124,7 +124,7 @@ This tagging approach allows OpenHands to efficiently manage both development an
 OpenHands supports both bind mounts and Docker named volumes in SandboxConfig.volumes:
 
 - Bind mount: "/abs/host/path:/container/path[:mode]"
-- Named volume: "volume:`<name>`:/container/path[:mode]" or any non-absolute host spec treated as a named volume
+- Named volume: "volume:<name>:/container/path[:mode]" or any non-absolute host spec treated as a named volume
 
 Overlay mode (copy-on-write layer) is supported for bind mounts by appending ":overlay" to the mode (e.g., ":ro,overlay").
 To enable overlay COW, set SANDBOX_VOLUME_OVERLAYS to a writable host directory; per-container upper/work dirs are created under it. If SANDBOX_VOLUME_OVERLAYS is unset, overlay mounts are skipped.

docs/usage/cloud/bitbucket-installation.mdx

@@ -8,21 +8,9 @@ description: This guide walks you through the process of installing OpenHands Cl
 
 - Signed in to [OpenHands Cloud](https://app.all-hands.dev) with [a Bitbucket account](/usage/cloud/openhands-cloud).
 
-## Adding Bitbucket Repository Access
-
-Upon signing into OpenHands Cloud with a Bitbucket account, OpenHands will have access to your repositories.
-
-## Working With Bitbucket Repos in Openhands Cloud
-
-After signing in with a Bitbucket account, use the `Open Repository` section to select the appropriate repository and
-branch you'd like OpenHands to work on. Then click on `Launch` to start the conversation!
-
-![Connect Repo](/static/img/connect-repo.png)
-
 ## IP Whitelisting
 
-If your Bitbucket Cloud instance has IP restrictions, you'll need to whitelist the following IP addresses to allow
-OpenHands to access your repositories:
+If your Bitbucket Cloud instance has IP restrictions, you'll need to whitelist the following IP addresses to allow OpenHands to access your repositories:
 
 ### Core App IP
 ```
@@ -43,6 +31,17 @@ OpenHands to access your repositories:
 34.60.55.59
 ```
 
+## Adding Bitbucket Repository Access
+
+Upon signing into OpenHands Cloud with a Bitbucket account, OpenHands will have access to your repositories.
+
+## Working With Bitbucket Repos in Openhands Cloud
+
+After signing in with a Bitbucket account, use the `select a repo` and `select a branch` dropdowns to select the
+appropriate repository and branch you'd like OpenHands to work on. Then click on `Launch` to start the conversation!
+
+![Connect Repo](/static/img/connect-repo-no-github.png)
+
 ## Next Steps
 
 - [Learn about the Cloud UI](/usage/cloud/cloud-ui).

docs/usage/cloud/cloud-api.mdx

@@ -12,10 +12,13 @@ For the available API endpoints, refer to the
 To use the OpenHands Cloud API, you'll need to generate an API key:
 
 1. Log in to your [OpenHands Cloud](https://app.all-hands.dev) account.
-2. Navigate to the [Settings > API Keys](https://app.all-hands.dev/settings/api-keys) page.
-3. Click `Create API Key`.
-4. Give your key a descriptive name (Example: "Development" or "Production") and select `Create`.
-5. Copy the generated API key and store it securely. It will only be shown once.
+2. Navigate to the [Settings page](https://app.all-hands.dev/settings).
+3. Select the `API Keys` tab.
+4. Click `Create API Key`.
+5. Give your key a descriptive name (Example: "Development" or "Production") and select `Create`.
+6. Copy the generated API key and store it securely. It will only be shown once.
+
+![API Key Generation](/static/img/api-key-generation.png)
 
 ## API Usage
 

docs/usage/cloud/cloud-ui.mdx

@@ -8,39 +8,24 @@ description: The Cloud UI provides a web interface for interacting with OpenHand
 
 The landing page is where you can:
 
+- [Add GitHub repository access](/usage/cloud/github-installation#adding-github-repository-access) to OpenHands.
 - [Select a GitHub repo](/usage/cloud/github-installation#working-with-github-repos-in-openhands-cloud),
   [a GitLab repo](/usage/cloud/gitlab-installation#working-with-gitlab-repos-in-openhands-cloud) or
   [a Bitbucket repo](/usage/cloud/bitbucket-installation#working-with-bitbucket-repos-in-openhands-cloud) to start working on.
-- Launch an empty conversation using `New Conversation`.
 - See `Suggested Tasks` for repositories that OpenHands has access to.
-- See your `Recent Conversations`.
+- Launch an empty conversation using `Launch from Scratch`.
 
 ## Settings
 
-Settings are divided across tabs, with each tab focusing on a specific area of configuration.
+The Settings page allows you to:
 
-- `User`
-  - Change your email address.
-- `Integrations`
-  - [Configure GitHub repository access](/usage/cloud/github-installation#modifying-repository-access) for OpenHands.
-  - [Install the OpenHands Slack app](/usage/cloud/slack-installation).
-- `Application`
-  - Set your preferred language, notifications and other preferences.
-  - Toggle task suggestions on GitHub.
-  - Toggle Solvability Analysis.
-  - Set a maximum budget per conversation.
-  - Configure the username and email that OpenHands uses for commits.
-- `LLM` (Available for `Pro Users`)
-  - Choose to use another LLM or use different models from the OpenHands provider.
-- `Billing`
-  - Add credits for using the OpenHands provider.
-  - Cancel your `Pro` subscription.
-- `Secrets`
-  - [Generate custom secrets](/usage/common-settings#secrets-management).
-- `API Keys`
-  - [Create API keys to work with OpenHands programmatically](/usage/cloud/cloud-api).
-- `MCP`
-  - [Setup an MCP server](/usage/mcp)
+- [Configure GitHub repository access](/usage/cloud/github-installation#modifying-repository-access) for OpenHands.
+- [Install the OpenHands Slack app](/usage/cloud/slack-installation).
+- Set application settings like your preferred language, notifications and other preferences.
+- Add credits to your account.
+- [Generate custom secrets](/usage/common-settings#secrets-management).
+- [Create API keys to work with OpenHands programmatically](/usage/cloud/cloud-api).
+- Change your email address.
 
 ## Key Features
 

docs/usage/cloud/github-installation.mdx

@@ -12,7 +12,7 @@ description: This guide walks you through the process of installing OpenHands Cl
 
 You can grant OpenHands access to specific GitHub repositories:
 
-1. Click on `+ Add GitHub Repos` in the repository selection dropdown.
+1. Click on `Add GitHub repos` on the landing page.
 2. Select your organization and choose the specific repositories to grant OpenHands access to.
 <Accordion title="OpenHands permissions">
   - OpenHands requests short-lived tokens (8-hour expiration) with these permissions:
@@ -34,22 +34,20 @@ You can grant OpenHands access to specific GitHub repositories:
 ## Modifying Repository Access
 
 You can modify GitHub repository access at any time by:
-- Selecting `+ Add GitHub Repos` in the repository selection dropdown or
-- Visiting the `Settings > Integrations` page and selecting `Configure GitHub Repositories`
+- Selecting `Add GitHub repos` on the landing page or
+- Visiting the Settings page and selecting `Configure GitHub Repositories` under the `Integrations` tab
 
 ## Working With GitHub Repos in Openhands Cloud
 
-Once you've granted GitHub repository access, you can start working with your GitHub repository. Use the
-`Open Repository` section to select the appropriate repository and branch you'd like OpenHands to work on. Then click
-on `Launch` to start the conversation!
+Once you've granted GitHub repository access, you can start working with your GitHub repository. Use the `select a repo`
+and `select a branch` dropdowns to select the appropriate repository and branch you'd like OpenHands to work on. Then
+click on `Launch` to start the conversation!
 
 ![Connect Repo](/static/img/connect-repo.png)
 
-## Working on GitHub Issues and Pull Requests Using Openhands
+## Working on Github Issues and Pull Requests Using Openhands
 
-To allow OpenHands to work directly from GitHub directly, you must
-[give OpenHands access to your repository](/usage/cloud/github-installation#modifying-repository-access). Once access is
-given, you can use OpenHands by labeling the issue or by tagging `@openhands`.
+Giving GitHub repository access to OpenHands also allows you to work on GitHub issues and pull requests directly.
 
 ### Working with Issues
 
@@ -66,12 +64,7 @@ To get OpenHands to work on pull requests, mention `@openhands` in the comments
 - Request updates
 - Get code explanations
 
-<Note>
-The `@openhands` mention functionality in pull requests only works if the pull request is both
-*to* and *from* a repository that you have added through the interface. This is because OpenHands needs appropriate
-permissions to access both repositories.
-</Note>
-
+**Important Note**: The `@openhands` mention functionality in pull requests only works if the pull request is both *to* and *from* a repository that you have added through the interface. This is because OpenHands needs appropriate permissions to access both repositories.
 
 ## Next Steps
 

docs/usage/cloud/gitlab-installation.mdx

@@ -14,17 +14,16 @@ Upon signing into OpenHands Cloud with a GitLab account, OpenHands will have acc
 
 ## Working With GitLab Repos in Openhands Cloud
 
-After signing in with a Gitlab account, use the `Open Repository` section to select the appropriate repository and
-branch you'd like OpenHands to work on. Then click on `Launch` to start the conversation!
+After signing in with a Gitlab account, use the `select a repo` and `select a branch` dropdowns to select the
+appropriate repository and branch you'd like OpenHands to work on. Then click on `Launch` to start the conversation!
 
-![Connect Repo](/static/img/connect-repo.png)
+![Connect Repo](/static/img/connect-repo-no-github.png)
 
 ## Using Tokens with Reduced Scopes
 
 OpenHands requests an API-scoped token during OAuth authentication. By default, this token is provided to the agent.
-To restrict the agent's permissions, you can define a custom secret `GITLAB_TOKEN`, which will override the default
-token assigned to the agent. While the high-permission API token is still requested and used for other components of
-the application (e.g. opening merge requests), the agent will not have access to it.
+To restrict the agent's permissions, you can define a custom secret `GITLAB_TOKEN`, which will override the default token assigned to the agent.
+While the high-permission API token is still requested and used for other components of the application (e.g. opening merge requests), the agent will not have access to it.
 
 ## Working on GitLab Issues and Merge Requests Using Openhands
 
@@ -33,8 +32,7 @@ This feature works for personal projects and is available for group projects wit
 [Premium or Ultimate tier subscription](https://docs.gitlab.com/user/project/integrations/webhooks/#group-webhooks).
 
 A webhook is automatically installed within a few minutes after the owner/maintainer of the project or group logs into
-OpenHands Cloud. If you decide to delete the webhook, then re-installing will require the support of All Hands AI but
-we are planning to improve this in a future release.
+OpenHands Cloud. If you decide to delete the webhook, then re-installing will require the support of All Hands AI but we are planning to improve this in a future release.
 </Note>
 
 Giving GitLab repository access to OpenHands also allows you to work on GitLab issues and merge requests directly.

docs/usage/cloud/slack-installation.mdx

@@ -13,9 +13,7 @@ description: This guide walks you through installing the OpenHands Slack app.
 </iframe>
 
 <Info>
-OpenHands utilizes a large language model (LLM), which may generate responses that are inaccurate or incomplete.
-While we strive for accuracy, OpenHands' outputs are not guaranteed to be correct, and we encourage users to
-validate critical information independently.
+OpenHands utilizes a large language model (LLM), which may generate responses that are inaccurate or incomplete. While we strive for accuracy, OpenHands' outputs are not guaranteed to be correct, and we encourage users to validate critical information independently.
 </Info>
 
 ## Prerequisites

docs/usage/configuration-options.mdx

@@ -8,11 +8,6 @@ description: This page outlines all available configuration options for OpenHand
    In GUI Mode, any settings applied through the Settings UI will take precedence.
 </Note>
 
-<Note>
-   **Looking for Environment Variables?** All configuration options can also be set using environment variables. 
-   See the [Environment Variables Reference](./environment-variables) for a complete list with examples.
-</Note>
-
 ## Location of the `config.toml` File
 
 When running OpenHands in CLI, headless, or development mode, you can use a project-specific `config.toml` file for configuration, which must be
@@ -23,11 +18,6 @@ specify a different path to the `config.toml` file.
 
 The core configuration options are defined in the `[core]` section of the `config.toml` file.
 
-Core configuration options can be set as environment variables by converting to uppercase. For example:
-- `debug` → `DEBUG`
-- `cache_dir` → `CACHE_DIR`
-- `runtime` → `RUNTIME`
-
 ### Workspace
 - `workspace_base` **(Deprecated)**
   - Type: `str`
@@ -151,11 +141,6 @@ The LLM (Large Language Model) configuration options are defined in the `[llm]`
 
 To use these with the docker command, pass in `-e LLM_<option>`. Example: `-e LLM_NUM_RETRIES`.
 
-All LLM configuration options can be set as environment variables by prefixing with `LLM_` and converting to uppercase. For example:
-- `model` → `LLM_MODEL`
-- `api_key` → `LLM_API_KEY`
-- `base_url` → `LLM_BASE_URL`
-
 <Note>
 For development setups, you can also define custom named LLM configurations. See [Custom LLM Configurations](./llms/custom-llm-configs) for details.
 </Note>
@@ -292,11 +277,6 @@ For development setups, you can also define custom named LLM configurations. See
 
 The agent configuration options are defined in the `[agent]` and `[agent.<agent_name>]` sections of the `config.toml` file.
 
-Agent configuration options can be set as environment variables by prefixing with `AGENT_` and converting to uppercase. For example:
-- `enable_browsing` → `AGENT_ENABLE_BROWSING`
-- `function_calling` → `AGENT_FUNCTION_CALLING`
-- `llm_config` → `AGENT_LLM_CONFIG`
-
 ### LLM Configuration
 - `llm_config`
   - Type: `str`
@@ -348,11 +328,6 @@ The sandbox configuration options are defined in the `[sandbox]` section of the
 
 To use these with the docker command, pass in `-e SANDBOX_<option>`. Example: `-e SANDBOX_TIMEOUT`.
 
-All sandbox configuration options can be set as environment variables by prefixing with `SANDBOX_` and converting to uppercase. For example:
-- `timeout` → `SANDBOX_TIMEOUT`
-- `user_id` → `SANDBOX_USER_ID`
-- `base_container_image` → `SANDBOX_BASE_CONTAINER_IMAGE`
-
 ### Execution
 - `timeout`
   - Type: `int`
@@ -415,10 +390,6 @@ The security configuration options are defined in the `[security]` section of th
 
 To use these with the docker command, pass in `-e SECURITY_<option>`. Example: `-e SECURITY_CONFIRMATION_MODE`.
 
-All security configuration options can be set as environment variables by prefixing with `SECURITY_` and converting to uppercase. For example:
-- `confirmation_mode` → `SECURITY_CONFIRMATION_MODE`
-- `security_analyzer` → `SECURITY_SECURITY_ANALYZER`
-
 ### Confirmation Mode
 - `confirmation_mode`
   - Type: `bool`

docs/usage/environment-variables.mdx

@@ -1,251 +0,0 @@
----
-title: Environment Variables Reference
-description: Complete reference of all environment variables supported by OpenHands
----
-
-This page provides a reference of environment variables that can be used to configure OpenHands. Environment variables provide an alternative to TOML configuration files and are particularly useful for containerized deployments, CI/CD pipelines, and cloud environments.
-
-## Environment Variable Naming Convention
-
-OpenHands follows a consistent naming pattern for environment variables:
-
-- **Core settings**: Direct uppercase mapping (e.g., `debug` → `DEBUG`)
-- **LLM settings**: Prefixed with `LLM_` (e.g., `model` → `LLM_MODEL`)
-- **Agent settings**: Prefixed with `AGENT_` (e.g., `enable_browsing` → `AGENT_ENABLE_BROWSING`)
-- **Sandbox settings**: Prefixed with `SANDBOX_` (e.g., `timeout` → `SANDBOX_TIMEOUT`)
-- **Security settings**: Prefixed with `SECURITY_` (e.g., `confirmation_mode` → `SECURITY_CONFIRMATION_MODE`)
-
-## Core Configuration Variables
-
-These variables correspond to the `[core]` section in `config.toml`:
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `DEBUG` | boolean | `false` | Enable debug logging throughout the application |
-| `DISABLE_COLOR` | boolean | `false` | Disable colored output in terminal |
-| `CACHE_DIR` | string | `"/tmp/cache"` | Directory path for caching |
-| `SAVE_TRAJECTORY_PATH` | string | `"./trajectories"` | Path to store conversation trajectories |
-| `REPLAY_TRAJECTORY_PATH` | string | `""` | Path to load and replay a trajectory file |
-| `FILE_STORE_PATH` | string | `"/tmp/file_store"` | File store directory path |
-| `FILE_STORE` | string | `"memory"` | File store type (`memory`, `local`, etc.) |
-| `FILE_UPLOADS_MAX_FILE_SIZE_MB` | integer | `0` | Maximum file upload size in MB (0 = no limit) |
-| `FILE_UPLOADS_RESTRICT_FILE_TYPES` | boolean | `false` | Whether to restrict file upload types |
-| `FILE_UPLOADS_ALLOWED_EXTENSIONS` | list | `[".*"]` | List of allowed file extensions for uploads |
-| `MAX_BUDGET_PER_TASK` | float | `0.0` | Maximum budget per task (0.0 = no limit) |
-| `MAX_ITERATIONS` | integer | `100` | Maximum number of iterations per task |
-| `RUNTIME` | string | `"docker"` | Runtime environment (`docker`, `local`, `cli`, etc.) |
-| `DEFAULT_AGENT` | string | `"CodeActAgent"` | Default agent class to use |
-| `JWT_SECRET` | string | auto-generated | JWT secret for authentication |
-| `RUN_AS_OPENHANDS` | boolean | `true` | Whether to run as the openhands user |
-| `VOLUMES` | string | `""` | Volume mounts in format `host:container[:mode]` |
-
-## LLM Configuration Variables
-
-These variables correspond to the `[llm]` section in `config.toml`:
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `LLM_MODEL` | string | `"claude-3-5-sonnet-20241022"` | LLM model to use |
-| `LLM_API_KEY` | string | `""` | API key for the LLM provider |
-| `LLM_BASE_URL` | string | `""` | Custom API base URL |
-| `LLM_API_VERSION` | string | `""` | API version to use |
-| `LLM_TEMPERATURE` | float | `0.0` | Sampling temperature |
-| `LLM_TOP_P` | float | `1.0` | Top-p sampling parameter |
-| `LLM_MAX_INPUT_TOKENS` | integer | `0` | Maximum input tokens (0 = no limit) |
-| `LLM_MAX_OUTPUT_TOKENS` | integer | `0` | Maximum output tokens (0 = no limit) |
-| `LLM_MAX_MESSAGE_CHARS` | integer | `30000` | Maximum characters that will be sent to the model in observation content |
-| `LLM_TIMEOUT` | integer | `0` | API timeout in seconds (0 = no timeout) |
-| `LLM_NUM_RETRIES` | integer | `8` | Number of retry attempts |
-| `LLM_RETRY_MIN_WAIT` | integer | `15` | Minimum wait time between retries (seconds) |
-| `LLM_RETRY_MAX_WAIT` | integer | `120` | Maximum wait time between retries (seconds) |
-| `LLM_RETRY_MULTIPLIER` | float | `2.0` | Exponential backoff multiplier |
-| `LLM_DROP_PARAMS` | boolean | `false` | Drop unsupported parameters without error |
-| `LLM_CACHING_PROMPT` | boolean | `true` | Enable prompt caching if supported |
-| `LLM_DISABLE_VISION` | boolean | `false` | Disable vision capabilities for cost reduction |
-| `LLM_CUSTOM_LLM_PROVIDER` | string | `""` | Custom LLM provider name |
-| `LLM_OLLAMA_BASE_URL` | string | `""` | Base URL for Ollama API |
-| `LLM_INPUT_COST_PER_TOKEN` | float | `0.0` | Cost per input token |
-| `LLM_OUTPUT_COST_PER_TOKEN` | float | `0.0` | Cost per output token |
-| `LLM_REASONING_EFFORT` | string | `""` | Reasoning effort for o-series models (`low`, `medium`, `high`) |
-
-### AWS Configuration
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `LLM_AWS_ACCESS_KEY_ID` | string | `""` | AWS access key ID |
-| `LLM_AWS_SECRET_ACCESS_KEY` | string | `""` | AWS secret access key |
-| `LLM_AWS_REGION_NAME` | string | `""` | AWS region name |
-
-## Agent Configuration Variables
-
-These variables correspond to the `[agent]` section in `config.toml`:
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `AGENT_LLM_CONFIG` | string | `""` | Name of LLM config group to use |
-| `AGENT_FUNCTION_CALLING` | boolean | `true` | Enable function calling |
-| `AGENT_ENABLE_BROWSING` | boolean | `false` | Enable browsing delegate |
-| `AGENT_ENABLE_LLM_EDITOR` | boolean | `false` | Enable LLM-based editor |
-| `AGENT_ENABLE_JUPYTER` | boolean | `false` | Enable Jupyter integration |
-| `AGENT_ENABLE_HISTORY_TRUNCATION` | boolean | `true` | Enable history truncation |
-| `AGENT_ENABLE_PROMPT_EXTENSIONS` | boolean | `true` | Enable microagents (prompt extensions) |
-| `AGENT_DISABLED_MICROAGENTS` | list | `[]` | List of microagents to disable |
-
-## Sandbox Configuration Variables
-
-These variables correspond to the `[sandbox]` section in `config.toml`:
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `SANDBOX_TIMEOUT` | integer | `120` | Sandbox timeout in seconds |
-| `SANDBOX_USER_ID` | integer | `1000` | User ID for sandbox processes |
-| `SANDBOX_BASE_CONTAINER_IMAGE` | string | `"nikolaik/python-nodejs:python3.12-nodejs22"` | Base container image |
-| `SANDBOX_USE_HOST_NETWORK` | boolean | `false` | Use host networking |
-| `SANDBOX_RUNTIME_BINDING_ADDRESS` | string | `"0.0.0.0"` | Runtime binding address |
-| `SANDBOX_ENABLE_AUTO_LINT` | boolean | `false` | Enable automatic linting |
-| `SANDBOX_INITIALIZE_PLUGINS` | boolean | `true` | Initialize sandbox plugins |
-| `SANDBOX_RUNTIME_EXTRA_DEPS` | string | `""` | Extra dependencies to install |
-| `SANDBOX_RUNTIME_STARTUP_ENV_VARS` | dict | `{}` | Environment variables for runtime |
-| `SANDBOX_BROWSERGYM_EVAL_ENV` | string | `""` | BrowserGym evaluation environment |
-| `SANDBOX_VOLUMES` | string | `""` | Volume mounts (replaces deprecated workspace settings) |
-| `SANDBOX_RUNTIME_CONTAINER_IMAGE` | string | `""` | Pre-built runtime container image |
-| `SANDBOX_KEEP_RUNTIME_ALIVE` | boolean | `false` | Keep runtime alive after session ends |
-| `SANDBOX_PAUSE_CLOSED_RUNTIMES` | boolean | `false` | Pause instead of stopping closed runtimes |
-| `SANDBOX_CLOSE_DELAY` | integer | `300` | Delay before closing idle runtimes (seconds) |
-| `SANDBOX_RM_ALL_CONTAINERS` | boolean | `false` | Remove all containers when stopping |
-| `SANDBOX_ENABLE_GPU` | boolean | `false` | Enable GPU support |
-| `SANDBOX_CUDA_VISIBLE_DEVICES` | string | `""` | Specify GPU devices by ID |
-| `SANDBOX_VSCODE_PORT` | integer | auto | Specific port for VSCode server |
-
-### Sandbox Environment Variables
-Variables prefixed with `SANDBOX_ENV_` are passed through to the sandbox environment:
-
-| Environment Variable | Description |
-|---------------------|-------------|
-| `SANDBOX_ENV_*` | Any variable with this prefix is passed to the sandbox (e.g., `SANDBOX_ENV_OPENAI_API_KEY`) |
-
-## Security Configuration Variables
-
-These variables correspond to the `[security]` section in `config.toml`:
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `SECURITY_CONFIRMATION_MODE` | boolean | `false` | Enable confirmation mode for actions |
-| `SECURITY_SECURITY_ANALYZER` | string | `"llm"` | Security analyzer to use (`llm`, `invariant`) |
-| `SECURITY_ENABLE_SECURITY_ANALYZER` | boolean | `true` | Enable security analysis |
-
-## Debug and Logging Variables
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `DEBUG` | boolean | `false` | Enable general debug logging |
-| `DEBUG_LLM` | boolean | `false` | Enable LLM-specific debug logging |
-| `DEBUG_RUNTIME` | boolean | `false` | Enable runtime debug logging |
-| `LOG_TO_FILE` | boolean | auto | Log to file (auto-enabled when DEBUG=true) |
-
-## Runtime-Specific Variables
-
-### Docker Runtime
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `SANDBOX_VOLUME_OVERLAYS` | string | `""` | Volume overlay configurations |
-
-### Remote Runtime
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `SANDBOX_API_KEY` | string | `""` | API key for remote runtime |
-| `SANDBOX_REMOTE_RUNTIME_API_URL` | string | `""` | Remote runtime API URL |
-
-### Local Runtime
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `RUNTIME_URL` | string | `""` | Runtime URL for local runtime |
-| `RUNTIME_URL_PATTERN` | string | `""` | Runtime URL pattern |
-| `RUNTIME_ID` | string | `""` | Runtime identifier |
-| `LOCAL_RUNTIME_MODE` | string | `""` | Enable local runtime mode (`1` to enable) |
-
-## Integration Variables
-
-### GitHub Integration
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `GITHUB_TOKEN` | string | `""` | GitHub personal access token |
-
-### Third-Party API Keys
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `OPENAI_API_KEY` | string | `""` | OpenAI API key |
-| `ANTHROPIC_API_KEY` | string | `""` | Anthropic API key |
-| `GOOGLE_API_KEY` | string | `""` | Google API key |
-| `AZURE_API_KEY` | string | `""` | Azure API key |
-| `TAVILY_API_KEY` | string | `""` | Tavily search API key |
-
-## Server Configuration Variables
-
-These are primarily used when running OpenHands as a server:
-
-| Environment Variable | Type | Default | Description |
-|---------------------|------|---------|-------------|
-| `FRONTEND_PORT` | integer | `3000` | Frontend server port |
-| `BACKEND_PORT` | integer | `8000` | Backend server port |
-| `FRONTEND_HOST` | string | `"localhost"` | Frontend host address |
-| `BACKEND_HOST` | string | `"localhost"` | Backend host address |
-| `WEB_HOST` | string | `"localhost"` | Web server host |
-| `SERVE_FRONTEND` | boolean | `true` | Whether to serve frontend |
-
-## Deprecated Variables
-
-These variables are deprecated and should be replaced:
-
-| Environment Variable | Replacement | Description |
-|---------------------|-------------|-------------|
-| `WORKSPACE_BASE` | `SANDBOX_VOLUMES` | Use volume mounting instead |
-| `WORKSPACE_MOUNT_PATH` | `SANDBOX_VOLUMES` | Use volume mounting instead |
-| `WORKSPACE_MOUNT_PATH_IN_SANDBOX` | `SANDBOX_VOLUMES` | Use volume mounting instead |
-| `WORKSPACE_MOUNT_REWRITE` | `SANDBOX_VOLUMES` | Use volume mounting instead |
-
-## Usage Examples
-
-### Basic Setup with OpenAI
-```bash
-export LLM_MODEL="gpt-4o"
-export LLM_API_KEY="your-openai-api-key"
-export DEBUG=true
-```
-
-### Docker Deployment with Custom Volumes
-```bash
-export RUNTIME="docker"
-export SANDBOX_VOLUMES="/host/workspace:/workspace:rw,/host/data:/data:ro"
-export SANDBOX_TIMEOUT=300
-```
-
-### Remote Runtime Configuration
-```bash
-export RUNTIME="remote"
-export SANDBOX_API_KEY="your-remote-api-key"
-export SANDBOX_REMOTE_RUNTIME_API_URL="https://your-runtime-api.com"
-```
-
-### Security-Enhanced Setup
-```bash
-export SECURITY_CONFIRMATION_MODE=true
-export SECURITY_SECURITY_ANALYZER="llm"
-export DEBUG_RUNTIME=true
-```
-
-## Notes
-
-1. **Boolean Values**: Environment variables expecting boolean values accept `true`/`false`, `1`/`0`, or `yes`/`no` (case-insensitive).
-
-2. **List Values**: Lists should be provided as Python literal strings, e.g., `AGENT_DISABLED_MICROAGENTS='["microagent1", "microagent2"]'`.
-
-3. **Dictionary Values**: Dictionaries should be provided as Python literal strings, e.g., `SANDBOX_RUNTIME_STARTUP_ENV_VARS='{"KEY": "value"}'`.
-
-4. **Precedence**: Environment variables take precedence over TOML configuration files.
-
-5. **Docker Usage**: When using Docker, pass environment variables with the `-e` flag:
-   ```bash
-   docker run -e LLM_API_KEY="your-key" -e DEBUG=true openhands/openhands
-   ```
-
-6. **Validation**: Invalid environment variable values will be logged as errors and fall back to defaults.

docs/usage/how-to/cli-mode.mdx

@@ -113,7 +113,7 @@ The conversation history will be saved in `~/.openhands/sessions`.
 ```bash
 docker run -it \
     --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e SANDBOX_USER_ID=$(id -u) \
     -e SANDBOX_VOLUMES=$SANDBOX_VOLUMES \
     -e LLM_API_KEY=$LLM_API_KEY \
@@ -122,7 +122,7 @@ docker run -it \
     -v ~/.openhands:/.openhands \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app-$(date +%Y%m%d%H%M%S) \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57 \
+    docker.all-hands.dev/all-hands-ai/openhands:0.55 \
     python -m openhands.cli.entry --override-cli-mode true
 ```
 

docs/usage/how-to/headless-mode.mdx

@@ -52,7 +52,7 @@ Set environment variables and run the Docker command:
 
 ```bash
 # Set required environment variables
-export SANDBOX_VOLUMES="/path/to/workspace:/workspace:rw"  # Format: host_path:container_path:mode
+export SANDBOX_VOLUMES="/path/to/workspace"  # See SANDBOX_VOLUMES docs for details
 export LLM_MODEL="anthropic/claude-sonnet-4-20250514"
 export LLM_API_KEY="your-api-key"
 export SANDBOX_SELECTED_REPO="owner/repo-name"  # Optional: requires GITHUB_TOKEN
@@ -61,7 +61,7 @@ export GITHUB_TOKEN="your-token"  # Required for repository operations
 # Run OpenHands
 docker run -it \
     --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e SANDBOX_USER_ID=$(id -u) \
     -e SANDBOX_VOLUMES=$SANDBOX_VOLUMES \
     -e LLM_API_KEY=$LLM_API_KEY \
@@ -73,7 +73,7 @@ docker run -it \
     -v ~/.openhands:/.openhands \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app-$(date +%Y%m%d%H%M%S) \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57 \
+    docker.all-hands.dev/all-hands-ai/openhands:0.55 \
     python -m openhands.core.main -t "write a bash script that prints hi"
 ```
 

docs/usage/llms/local-llms.mdx

@@ -68,23 +68,23 @@ Download and install the LM Studio desktop app from [lmstudio.ai](https://lmstud
 1. Check [the installation guide](/usage/local-setup) and ensure all prerequisites are met before running OpenHands, then run:
 
 ```bash
-docker pull docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik
+docker pull docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik
 
 docker run -it --rm --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e LOG_ALL_EVENTS=true \
     -v /var/run/docker.sock:/var/run/docker.sock \
     -v ~/.openhands:/.openhands \
     -p 3000:3000 \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57
+    docker.all-hands.dev/all-hands-ai/openhands:0.55
 ```
 
 2. Wait until the server is running (see log below):
 ```
 Digest: sha256:e72f9baecb458aedb9afc2cd5bc935118d1868719e55d50da73190d3a85c674f
-Status: Image is up to date for docker.all-hands.dev/all-hands-ai/openhands:0.57
+Status: Image is up to date for docker.all-hands.dev/all-hands-ai/openhands:0.55
 Starting OpenHands...
 Running OpenHands as root
 14:22:13 - openhands:INFO: server_config.py:50 - Using config class None

docs/usage/llms/openhands-llms.mdx

@@ -30,20 +30,6 @@ When running OpenHands, you'll need to set the following in the OpenHands UI thr
 
 ## Pricing
 
-Pricing follows official API provider rates. Below are the current pricing details for OpenHands models:
+Pricing follows official API provider rates. [You can view model prices here.](https://github.com/BerriAI/litellm/blob/main/model_prices_and_context_window.json)
 
-| Model | Input Cost (per 1M tokens) | Cached Input Cost (per 1M tokens) | Output Cost (per 1M tokens) | Max Input Tokens | Max Output Tokens |
-|-------|----------------------------|-----------------------------------|------------------------------|------------------|-------------------|
-| claude-opus-4-20250514 | $15.00 | $1.50 | $75.00 | 200,000 | 32,000 |
-| claude-sonnet-4-20250514 | $3.00 | $0.30 | $15.00 | 200,000 | 64,000 |
-| devstral-medium-2507 | $0.40 | N/A | $2.00 | 128,000 | 128,000 |
-| devstral-small-2505 | $0.10 | N/A | $0.30 | 128,000 | 128,000 |
-| devstral-small-2507 | $0.10 | N/A | $0.30 | 128,000 | 128,000 |
-| gemini-2.5-pro | $1.25 | $0.31 | $10.00 | 1,048,576 | 65,535 |
-| gpt-5-2025-08-07 | $1.25 | $0.125 | $10.00 | 400,000 | 128,000 |
-| gpt-5-mini-2025-08-07 | $0.25 | $0.025 | $2.00 | 400,000 | 128,000 |
-| o3 | $2.00 | $0.50 | $8.00 | 200,000 | 100,000 |
-| o4-mini | $1.10 | $0.28 | $4.40 | 200,000 | 100,000 |
-| qwen3-coder-480b | $0.40 | N/A | $1.60 | N/A | N/A |
-
-**Note:** Cached input tokens are charged at a reduced rate when the same content is reused across requests. Models that don't support prompt caching show "N/A" for cached input cost.
+For `qwen3-coder-480b`, we charge the cheapest FP8 rate available on openrouter: \$0.4 per million input tokens and \$1.6 per million output tokens.

docs/usage/local-setup.mdx

@@ -116,17 +116,17 @@ Note that you'll still need `uv` installed for the default MCP servers to work p
 <Accordion title="Docker Command (Click to expand)">
 
 ```bash
-docker pull docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik
+docker pull docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik
 
 docker run -it --rm --pull=always \
-    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.57-nikolaik \
+    -e SANDBOX_RUNTIME_CONTAINER_IMAGE=docker.all-hands.dev/all-hands-ai/runtime:0.55-nikolaik \
     -e LOG_ALL_EVENTS=true \
     -v /var/run/docker.sock:/var/run/docker.sock \
     -v ~/.openhands:/.openhands \
     -p 3000:3000 \
     --add-host host.docker.internal:host-gateway \
     --name openhands-app \
-    docker.all-hands.dev/all-hands-ai/openhands:0.57
+    docker.all-hands.dev/all-hands-ai/openhands:0.55
 ```
 
 </Accordion>

docs/usage/mcp.mdx

@@ -67,19 +67,6 @@ sse_servers = [
     # External MCP service with authentication
     {url="https://api.example.com/mcp/sse", api_key="your-api-key"}
 ]
-
-# SHTTP Servers - Modern streamable HTTP transport (recommended)
-shttp_servers = [
-    # Basic SHTTP server with default 60s timeout
-    "https://api.example.com/mcp/shttp",
-    
-    # Server with custom timeout for heavy operations
-    {
-        url = "https://files.example.com/mcp/shttp",
-        api_key = "your-api-key",
-        timeout = 1800  # 30 minutes for large file processing
-    }
-]
 ```
 
 
@@ -131,17 +118,6 @@ SHTTP (Streamable HTTP) servers are configured using either a string URL or an o
   - Type: `str`
   - Description: API key for authentication
 
-- `timeout` (optional)
-  - Type: `int`
-  - Default: `60`
-  - Range: `1-3600` seconds (1 hour maximum)
-  - Description: Timeout in seconds for tool execution. This prevents tool calls from hanging indefinitely.
-  - **Use Cases:**
-    - **Short timeout (1-30s)**: For lightweight operations like status checks or simple queries
-    - **Medium timeout (30-300s)**: For standard processing tasks like data analysis or API calls  
-    - **Long timeout (300-3600s)**: For heavy operations like file processing, complex calculations, or batch operations
-  - **Note**: This timeout only applies to individual tool calls, not server connection establishment.
-
 ### Stdio Servers
 
 **Note**: While stdio servers are supported, we recommend using MCP proxies (see above) for better reliability and performance.
@@ -216,27 +192,5 @@ SHTTP is the modern HTTP-based transport protocol that provides enhanced feature
 
 SHTTP is the recommended transport for HTTP-based MCP servers as it provides better reliability and features compared to the legacy SSE transport.
 
-#### SHTTP Timeout Best Practices
-
-When configuring SHTTP timeouts, consider these guidelines:
-
-**Timeout Selection:**
-- **Database queries**: 30-60 seconds
-- **File operations**: 60-300 seconds (depending on file size)
-- **Web scraping**: 60-120 seconds
-- **Complex calculations**: 300-1800 seconds
-- **Batch processing**: 1800-3600 seconds (maximum)
-
-**Error Handling:**
-When a tool call exceeds the configured timeout:
-- The operation is cancelled with an `asyncio.TimeoutError`
-- The agent receives a timeout error message
-- The server connection remains active for subsequent requests
-
-**Monitoring:**
-- Set timeouts based on your tool's actual performance characteristics
-- Monitor timeout occurrences to optimize timeout values
-- Consider implementing server-side timeout handling for graceful degradation
-
 ### Standard Input/Output (stdio)
 Stdio transport enables communication through standard input and output streams, making it ideal for local integrations and command-line tools. This transport is used for locally executed MCP servers that run as separate processes.

enterprise/Dockerfile

@@ -7,28 +7,14 @@ LABEL com.datadoghq.tags.service="deploy"
 LABEL com.datadoghq.tags.env="${DD_ENV}"
 
 # Install Node.js v20+ and npm (which includes npx)
-# Apply security updates to fix CVEs
 RUN apt-get update && \
     apt-get install -y curl && \
     curl -fsSL https://deb.nodesource.com/setup_20.x | bash - && \
     apt-get install -y nodejs && \
     apt-get install -y jq gettext && \
-    # Apply security updates for packages with available fixes
-    apt-get upgrade -y \
-        libc-bin \
-        libc6 \
-        libgnutls30 \
-        libsqlite3-0 \
-        perl-base && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
+    apt-get clean
 
-# Install Python packages with security fixes
-RUN pip install alembic psycopg2-binary cloud-sql-python-connector pg8000 gspread stripe python-keycloak asyncpg sqlalchemy[asyncio] resend tenacity slack-sdk ddtrace posthog "limits==5.2.0" coredis prometheus-client shap scikit-learn pandas numpy && \
-    # Update packages with known CVE fixes
-    pip install --upgrade \
-        "mcp>=1.10.0" \
-        "pillow>=11.3.0"
+RUN pip install alembic psycopg2-binary cloud-sql-python-connector pg8000 gspread stripe python-keycloak asyncpg sqlalchemy[asyncio] resend tenacity slack-sdk ddtrace posthog "limits==5.2.0" coredis prometheus-client shap scikit-learn pandas numpy
 
 WORKDIR /app
 COPY enterprise .

enterprise/dev_config/python/.pre-commit-config.yaml

@@ -46,8 +46,7 @@ repos:
           - types-toml
           - types-redis
           - lxml
-          # OpenHands package in repo root
-          - ./
+          # TODO: Add OpenHands in parent
           - stripe==11.5.0
           - pygithub==2.6.1
         # To see gaps add `--html-report mypy-report/`

enterprise/dev_config/python/mypy.ini

@@ -7,11 +7,15 @@ warn_unreachable = True
 warn_redundant_casts = True
 no_implicit_optional = True
 strict_optional = True
-disable_error_code = type-abstract
-exclude = (^enterprise/migrations/.*)
+exclude = (^enterprise/migrations/.*|^openhands/.*)
 
 [mypy-enterprise.tests.unit.test_auth_routes.*]
 disable_error_code = union-attr
 
 [mypy-enterprise.sync.install_gitlab_webhooks.*]
 disable_error_code = redundant-cast
+
+# Let the other config check base openhands packages
+[mypy-openhands.*]
+follow_imports = skip
+ignore_missing_imports = True

enterprise/experiments/experiment_manager.py

@@ -2,6 +2,7 @@ from experiments.constants import (
     ENABLE_EXPERIMENT_MANAGER,
 )
 from experiments.experiment_versions import (
+    handle_claude4_vs_gpt5_experiment,
     handle_condenser_max_step_experiment,
     handle_system_prompt_experiment,
 )
@@ -43,6 +44,9 @@ class SaaSExperimentManager(ExperimentManager):
             return conversation_settings
 
         # Apply conversation-scoped experiments
+        conversation_settings = handle_claude4_vs_gpt5_experiment(
+            user_id, conversation_id, conversation_settings
+        )
         conversation_settings = handle_condenser_max_step_experiment(
             user_id, conversation_id, conversation_settings
         )
@@ -51,7 +55,7 @@ class SaaSExperimentManager(ExperimentManager):
 
     @staticmethod
     def run_config_variant_test(
-        user_id: str | None, conversation_id: str, config: OpenHandsConfig
+        user_id: str, conversation_id: str, config: OpenHandsConfig
     ) -> OpenHandsConfig:
         """
         Run agent config variant test and potentially modify the OpenHands config

enterprise/integrations/github/data_collector.py

@@ -390,24 +390,24 @@ class GitHubDataCollector:
         merged_by = None
         merge_commit_sha = None
         if is_merged:
-            merged_by = (pr_data.get('mergedBy') or {}).get('login')
-            merge_commit_sha = (pr_data.get('mergeCommit') or {}).get('oid')
+            merged_by = pr_data.get('mergedBy', {}).get('login')
+            merge_commit_sha = pr_data.get('mergeCommit', {}).get('oid')
 
         return {
             'repo_metadata': self._extract_repo_metadata(repo_data),
             'pr_metadata': {
-                'username': (pr_data.get('author') or {}).get('login'),
-                'number': pr_data.get('number'),
-                'title': pr_data.get('title'),
-                'body': pr_data.get('body'),
+                'username': pr_data.get('author', {}).get('login'),
+                'number': pr_data['number'],
+                'title': pr_data['title'],
+                'body': pr_data['body'],
                 'comments': pr_comments,
             },
             'commits': commits,
             'review_comments': review_comments,
             'merge_status': {
-                'merged': pr_data.get('merged'),
+                'merged': pr_data['merged'],
                 'merged_by': merged_by,
-                'state': pr_data.get('state'),
+                'state': pr_data['state'],
                 'merge_commit_sha': merge_commit_sha,
             },
             'openhands_stats': {

enterprise/integrations/gitlab/gitlab_manager.py

@@ -62,13 +62,7 @@ class GitlabManager(Manager):
             logger.warning(f'Got invalid keyloak user id for GitLab User {user_id}')
             return False
 
-        # Importing here prevents circular import
-        from integrations.gitlab.gitlab_service import SaaSGitLabService
-
-        gitlab_service: SaaSGitLabService = GitLabServiceImpl(
-            external_auth_id=keycloak_user_id
-        )
-
+        gitlab_service = GitLabServiceImpl(external_auth_id=keycloak_user_id)
         return await gitlab_service.user_has_write_access(project_id)
 
     async def receive_message(self, message: Message):
@@ -125,13 +119,7 @@ class GitlabManager(Manager):
             gitlab_view: The GitLab view object containing issue/PR/comment info
         """
         keycloak_user_id = gitlab_view.user_info.keycloak_user_id
-
-        # Importing here prevents circular import
-        from integrations.gitlab.gitlab_service import SaaSGitLabService
-
-        gitlab_service: SaaSGitLabService = GitLabServiceImpl(
-            external_auth_id=keycloak_user_id
-        )
+        gitlab_service = GitLabServiceImpl(external_auth_id=keycloak_user_id)
 
         outgoing_message = message.message
 

enterprise/integrations/gitlab/gitlab_view.py

@@ -47,14 +47,14 @@ class GitlabIssue(ResolverViewInterface):
         )
 
         self.previous_comments = await gitlab_service.get_issue_or_mr_comments(
-            str(self.project_id), self.issue_number, is_mr=self.is_mr
+            self.project_id, self.issue_number, is_mr=self.is_mr
         )
 
         (
             self.title,
             self.description,
         ) = await gitlab_service.get_issue_or_mr_title_and_body(
-            str(self.project_id), self.issue_number, is_mr=self.is_mr
+            self.project_id, self.issue_number, is_mr=self.is_mr
         )
 
     async def _get_instructions(self, jinja_env: Environment) -> tuple[str, str]:
@@ -199,11 +199,11 @@ class GitlabInlineMRComment(GitlabMRComment):
             self.title,
             self.description,
         ) = await gitlab_service.get_issue_or_mr_title_and_body(
-            str(self.project_id), self.issue_number, is_mr=self.is_mr
+            self.project_id, self.issue_number, is_mr=self.is_mr
         )
 
         self.previous_comments = await gitlab_service.get_review_thread_comments(
-            str(self.project_id), self.issue_number, self.discussion_id
+            self.project_id, self.issue_number, self.discussion_id
         )
 
     async def _get_instructions(self, jinja_env: Environment) -> tuple[str, str]:

enterprise/integrations/utils.py

@@ -172,17 +172,6 @@ def get_summary_for_agent_state(
 
         return f'OpenHands encountered an error: **{reason}**.\n\n[See the conversation]({conversation_link}) for more information.'
 
-    if state == AgentState.AWAITING_USER_INPUT:
-        logger.info(
-            'Agent is awaiting user input',
-            extra={
-                'agent_state': state.value,
-                'conversation_link': conversation_link,
-                'observation_reason': getattr(observation, 'reason', None),
-            },
-        )
-        return f'OpenHands is waiting for your input. [Continue the conversation]({conversation_link}) to provide additional instructions.'
-
     # Log unknown agent state as error
     logger.error(
         'Unknown error: Unhandled agent state',

enterprise/migrations/versions/075_add_cancellation_fields_to_subscription_access.py

@@ -1,50 +0,0 @@
-"""add cancellation fields to subscription_access
-
-Revision ID: 075
-Revises: 074
-Create Date: 2025-01-11
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = '075'
-down_revision: Union[str, None] = '074'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # Add cancelled_at field to track cancellation timestamp
-    op.add_column(
-        'subscription_access',
-        sa.Column('cancelled_at', sa.DateTime(timezone=True), nullable=True),
-    )
-
-    # Add stripe_subscription_id field to enable cancellation via Stripe API
-    op.add_column(
-        'subscription_access',
-        sa.Column('stripe_subscription_id', sa.String(), nullable=True),
-    )
-
-    # Create index on stripe_subscription_id for efficient lookups
-    op.create_index(
-        'ix_subscription_access_stripe_subscription_id',
-        'subscription_access',
-        ['stripe_subscription_id'],
-    )
-
-
-def downgrade() -> None:
-    # Drop index
-    op.drop_index(
-        'ix_subscription_access_stripe_subscription_id', 'subscription_access'
-    )
-
-    # Drop columns
-    op.drop_column('subscription_access', 'stripe_subscription_id')
-    op.drop_column('subscription_access', 'cancelled_at')

enterprise/poetry.lock

@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 2.1.4 and should not be changed by hand.
+# This file is automatically @generated by Poetry 2.1.3 and should not be changed by hand.
 
 [[package]]
 name = "aiofiles"
@@ -766,7 +766,7 @@ version = "1.17.1"
 description = "Foreign Function Interface for Python calling C code."
 optional = false
 python-versions = ">=3.8"
-groups = ["main", "test"]
+groups = ["main"]
 files = [
     {file = "cffi-1.17.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:df8b1c11f177bc2313ec4b2d46baec87a5f3e71fc8b45dab2ee7cae86d9aba14"},
     {file = "cffi-1.17.1-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:8f2cdc858323644ab277e9bb925ad72ae0e67f69e804f4898c070998d50b1a67"},
@@ -836,7 +836,6 @@ files = [
     {file = "cffi-1.17.1-cp39-cp39-win_amd64.whl", hash = "sha256:d016c76bdd850f3c626af19b0542c9677ba156e4ee4fccfdd7848803533ef662"},
     {file = "cffi-1.17.1.tar.gz", hash = "sha256:1c39c6016c32bc48dd54561950ebd6836e1670f2ae46128f67cf49e789c52824"},
 ]
-markers = {test = "platform_python_implementation == \"CPython\" and sys_platform == \"win32\""}
 
 [package.dependencies]
 pycparser = "*"
@@ -1427,73 +1426,73 @@ yaml = ["pyyaml (>=6.0.1)"]
 
 [[package]]
 name = "ddtrace"
-version = "3.13.0"
+version = "3.12.4"
 description = "Datadog APM client library"
 optional = false
 python-versions = ">=3.8"
 groups = ["main"]
 files = [
-    {file = "ddtrace-3.13.0-cp310-cp310-macosx_12_0_arm64.whl", hash = "sha256:12122a8e7089ab40cad2cd6bb51834859aa0a27babf3256a73630e6ee2315455"},
-    {file = "ddtrace-3.13.0-cp310-cp310-macosx_12_0_x86_64.whl", hash = "sha256:02fab2c444b87f290850b3d750e17ccdf49ace3baf8ff3305e8147f6fdf0dc50"},
-    {file = "ddtrace-3.13.0-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:a003ffa4649dab4971d3557ce2d85eb2c5d335ebc7152196cbf780171fd4b5e1"},
-    {file = "ddtrace-3.13.0-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:52b2458b6f0f4725156d46c6cb5410f98568a61cc890bb270515c9caad3a522d"},
-    {file = "ddtrace-3.13.0-cp310-cp310-manylinux_2_28_i686.whl", hash = "sha256:9160222e476e18af95ef687bd548f8e86b3815896bf7cd1d42a9b43005e058e2"},
-    {file = "ddtrace-3.13.0-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:464e245c2114c722ad4240b73b1c598f83cc1c7bdc9001aec3083f914c1cacc0"},
-    {file = "ddtrace-3.13.0-cp310-cp310-musllinux_1_2_i686.whl", hash = "sha256:21901a58e938dbeba0ca6c49b8ba1480d07eea5b057845ae4ff3a706d833137f"},
-    {file = "ddtrace-3.13.0-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:40e00faced483a3eac0b499cf191a38fbf8bb060a3872029ee3299871f87bdd9"},
-    {file = "ddtrace-3.13.0-cp310-cp310-win32.whl", hash = "sha256:d15593cb804d74094df1a71167a70136b7616579259ce2b26279f2762354e709"},
-    {file = "ddtrace-3.13.0-cp310-cp310-win_amd64.whl", hash = "sha256:5de44e7c595d25745665fa1cc44c0f0b4c7ad79be06d0de74f6e0edb2c8ec351"},
-    {file = "ddtrace-3.13.0-cp311-cp311-macosx_12_0_arm64.whl", hash = "sha256:68c38ac75cc3668e9284873f5e84c3e104880d68c3891ed13614e0614c46f5b0"},
-    {file = "ddtrace-3.13.0-cp311-cp311-macosx_12_0_x86_64.whl", hash = "sha256:6d8811c4b7397384aff7e54b7399647f4c1c0e9167792cb45adb2d3553fc20a2"},
-    {file = "ddtrace-3.13.0-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:029b6e6c50984b1976c6b0970e60184919dab9514441d08683a50a5d52a05326"},
-    {file = "ddtrace-3.13.0-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:8de2a060400ee89422ecfd3269dfd2e113f4f9dae00f6fcd3ed9e53e2223a26a"},
-    {file = "ddtrace-3.13.0-cp311-cp311-manylinux_2_28_i686.whl", hash = "sha256:bb0738048ea0e49e6bec9be2bf5c68a24d7ea3b27bf956147378366aacb4ca4b"},
-    {file = "ddtrace-3.13.0-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:04cf4776c52cfb19914bf6e84242d110197d15426c34e45b14fa63d9085767d5"},
-    {file = "ddtrace-3.13.0-cp311-cp311-musllinux_1_2_i686.whl", hash = "sha256:6c32774e90593ebb264d53d6523b71243b9ba794ae5689e38ad522afddd06c0b"},
-    {file = "ddtrace-3.13.0-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:a01f99b0287c2bbd8b305e0cb54b382eaf2a0fe89ba82f2f68fcbdd9fed040cd"},
-    {file = "ddtrace-3.13.0-cp311-cp311-win32.whl", hash = "sha256:b37efa3e7b487bd60e6fb89186d98c1ad1727871074f3519c9ca92feea7e5cd0"},
-    {file = "ddtrace-3.13.0-cp311-cp311-win_amd64.whl", hash = "sha256:112e4d96f02f94247528b65f046c69d360d6eca75b9e7cd2f95fde1c14e2002e"},
-    {file = "ddtrace-3.13.0-cp312-cp312-macosx_12_0_arm64.whl", hash = "sha256:13ac5bc306df5719d00a8b1f6925efbb9dd0ba5e121edcc2acfef24c57b3deb5"},
-    {file = "ddtrace-3.13.0-cp312-cp312-macosx_12_0_x86_64.whl", hash = "sha256:b3bdfc3cabab85f91a4f24264a2d0f6f74984a5b5994c62072c6e3b5e05320f3"},
-    {file = "ddtrace-3.13.0-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:11b10f8dfadb4b1372aee820be6c22071138ede2ddb32f73486255d5879b283f"},
-    {file = "ddtrace-3.13.0-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:a3d68007602797f280c971a286c3f05bdff66c12a68a3e0bd67cb5bbc1c4a67a"},
-    {file = "ddtrace-3.13.0-cp312-cp312-manylinux_2_28_i686.whl", hash = "sha256:abd00a5b83d85a951dd976a59c8673bedacdc1ea9e6acb8e72545f73bddc7879"},
-    {file = "ddtrace-3.13.0-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:5dbe392b2182e6dd617e946cf41da7e3207387b912809ebe8338b794b08750b2"},
-    {file = "ddtrace-3.13.0-cp312-cp312-musllinux_1_2_i686.whl", hash = "sha256:6b38b4ad9e3f1b3421022587748f6a687ed722eae16033392fc875b5c67d6c5a"},
-    {file = "ddtrace-3.13.0-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:f38a1545495c8db3318621400a3d407db457e3550a397e39cf883f41919e1dc8"},
-    {file = "ddtrace-3.13.0-cp312-cp312-win32.whl", hash = "sha256:e01bb1b305b777001d310911bd73d1fd88c9c212258caaf65f1422a0dbef1a3b"},
-    {file = "ddtrace-3.13.0-cp312-cp312-win_amd64.whl", hash = "sha256:8dbb9aa23a36599754932e79df28eb07fdd3aaca515297bf58dfcdac608273da"},
-    {file = "ddtrace-3.13.0-cp313-cp313-macosx_12_0_arm64.whl", hash = "sha256:397a68e476d8bd9aa14f8c097bc9014510948e76a0110842ab6f5fa1143ad153"},
-    {file = "ddtrace-3.13.0-cp313-cp313-macosx_12_0_x86_64.whl", hash = "sha256:fab1b06476169e2cf6a098130c44eeb3d9d8205b5a91ae8afdb7d2b4d2d0b0be"},
-    {file = "ddtrace-3.13.0-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:653f75c3e838366108464f9555120f61ef0589974f346ed2c2c9cb3001d3fc6a"},
-    {file = "ddtrace-3.13.0-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:80f694c3d3984c9bd3bd7818268be7ece02071c67671c6d8c815e6888ae4e78c"},
-    {file = "ddtrace-3.13.0-cp313-cp313-manylinux_2_28_i686.whl", hash = "sha256:be16f9c0583767db13403e78ac7ac7b4c103e8b7eaac6deef7c897408f24b940"},
-    {file = "ddtrace-3.13.0-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:c5490a715fbb70ee03840c6a3146c76d7bfa27d5b679ce4c1a7b368eff7dee9f"},
-    {file = "ddtrace-3.13.0-cp313-cp313-musllinux_1_2_i686.whl", hash = "sha256:45235a81c828e2d6bdb4ac1bbe55582c190bc27e8820eeae5c0478ea11f1ed81"},
-    {file = "ddtrace-3.13.0-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:7a9374a8cf405169a9eab7791cc94d5dc5753eefe806b5bee9909eef3d5e339d"},
-    {file = "ddtrace-3.13.0-cp313-cp313-win32.whl", hash = "sha256:6bc1648a1c046e6061e29d94d2003c17820cc3a7f1c24322dab654abe9bb30db"},
-    {file = "ddtrace-3.13.0-cp313-cp313-win_amd64.whl", hash = "sha256:8823e95f69dd3fc8a884d092fdc54a3c3078daf0f90e824fceda7e0f26acbc70"},
-    {file = "ddtrace-3.13.0-cp38-cp38-macosx_12_0_arm64.whl", hash = "sha256:338932a8511a815d5198ec09d55f6850fcb9c679a1b50a3a28fdc0ff99bd800a"},
-    {file = "ddtrace-3.13.0-cp38-cp38-macosx_12_0_x86_64.whl", hash = "sha256:c14fe68cfc1c11b9d560a3026e3e5dcdd59b725b6ce79cda66d23a26b37751e6"},
-    {file = "ddtrace-3.13.0-cp38-cp38-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:3fd70631f5c70ccafde14df98a9f807e537222f13d6f03fa08bf1308eaf89301"},
-    {file = "ddtrace-3.13.0-cp38-cp38-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:09c71f464afb05d7f1a2758112f4feaf2bca39daa22a6c3f75999227eb40e2ec"},
-    {file = "ddtrace-3.13.0-cp38-cp38-manylinux_2_28_i686.whl", hash = "sha256:481b13365e3cf100bf35f305bd0680695fa369e67a9ec4e1b41788df62ac1d0b"},
-    {file = "ddtrace-3.13.0-cp38-cp38-musllinux_1_2_aarch64.whl", hash = "sha256:0d99ebbef96f406e0436bd21a92354c3c338fc6a8fe85d0a26fe942bc563b721"},
-    {file = "ddtrace-3.13.0-cp38-cp38-musllinux_1_2_i686.whl", hash = "sha256:28086003f1c5ce3e84239eea9d624afcc386b38f2115c3438ea49beff84ff861"},
-    {file = "ddtrace-3.13.0-cp38-cp38-musllinux_1_2_x86_64.whl", hash = "sha256:f280e80560f5c953bb16b168bed1b6f7d527ef98f81860422500040ee57a7aba"},
-    {file = "ddtrace-3.13.0-cp38-cp38-win32.whl", hash = "sha256:82f0b76c83e368c686594f42809d727143ee89a879d1a76cde9f75d4cea07cb4"},
-    {file = "ddtrace-3.13.0-cp38-cp38-win_amd64.whl", hash = "sha256:dd7b3a9933b11b2fce4dd4cb34ee465bc3c87024444a2e6a5a653f424bae8e37"},
-    {file = "ddtrace-3.13.0-cp39-cp39-macosx_12_0_arm64.whl", hash = "sha256:c1ce2123615e4618050ec7fc96e296283f23c45eddcf3a2fe94386f7513795a4"},
-    {file = "ddtrace-3.13.0-cp39-cp39-macosx_12_0_x86_64.whl", hash = "sha256:9dae3459edd5cc7a1124596b524b743b1d2bddf4155ca9679c599740ad71546d"},
-    {file = "ddtrace-3.13.0-cp39-cp39-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:8d36d0cf84a39b29f88dcb06a20fc3f2c7a9eca8eb1fd5d15bc5a51de095962c"},
-    {file = "ddtrace-3.13.0-cp39-cp39-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:d4a55277a3db32fee06030fd0dbf77c2e867541c3e4b65e68e46b03971401173"},
-    {file = "ddtrace-3.13.0-cp39-cp39-manylinux_2_28_i686.whl", hash = "sha256:cb97593d9739f0be6647e19edc6fc6998dfba3e78fb9d2df5fef9ebfb117aa85"},
-    {file = "ddtrace-3.13.0-cp39-cp39-musllinux_1_2_aarch64.whl", hash = "sha256:f905e5bb2db4c154fca25ded15c3e1d633951db2d6ed2989f630ee3afd589cc0"},
-    {file = "ddtrace-3.13.0-cp39-cp39-musllinux_1_2_i686.whl", hash = "sha256:de3ecc6428330117ef063ef6a90326669a9a4cf3e766674228ec384edca52bb1"},
-    {file = "ddtrace-3.13.0-cp39-cp39-musllinux_1_2_x86_64.whl", hash = "sha256:eec340ef5152e971dc6ab075945dfa7c41285f8441bea0a78f5f4cd1f6b9aab6"},
-    {file = "ddtrace-3.13.0-cp39-cp39-win32.whl", hash = "sha256:8c2831f928393f934bfe9f9b5f0eeb22a0f5c88fbebe32cc5106b24409847d6b"},
-    {file = "ddtrace-3.13.0-cp39-cp39-win_amd64.whl", hash = "sha256:e04f4c41e7216422e9cd101bee70a823f56dddb8333158e1e72b73332e1a311d"},
-    {file = "ddtrace-3.13.0.tar.gz", hash = "sha256:d7d3d82795d29cf2385aa692ee5c65e469ebfa34469941055af66eae2eefa374"},
+    {file = "ddtrace-3.12.4-cp310-cp310-macosx_12_0_arm64.whl", hash = "sha256:222dc483f22a065795f473cad6fc6e798ecf9da9f4fc99ca87f1ba70f34d21b1"},
+    {file = "ddtrace-3.12.4-cp310-cp310-macosx_12_0_x86_64.whl", hash = "sha256:196f114a70b75320876f6861c10435c6d4ea50e0f406328b0862a021c344d002"},
+    {file = "ddtrace-3.12.4-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:4200e8b057b29ce3ba0889a9d423e4d105b0ba35d4bd58ba2670763018909623"},
+    {file = "ddtrace-3.12.4-cp310-cp310-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:7fc1449d511e04e8b2596eee6d1ad2d3420dff23f6dfd8a899c5e3e03dfe8ba5"},
+    {file = "ddtrace-3.12.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:2ebae69206957837341cd94bbe78e5242395f7571455dfe911b56ea2f7404ada"},
+    {file = "ddtrace-3.12.4-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:a08cd25234358a2427494d4059ee12afc83e083bad65f2bd62417fd935caa737"},
+    {file = "ddtrace-3.12.4-cp310-cp310-musllinux_1_2_i686.whl", hash = "sha256:fbe90ff2c914c753116807ddffde9065ecbf9944bdc4932862c3f5835485004d"},
+    {file = "ddtrace-3.12.4-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:1b3be9452bc76f730203b86272f8312c7e195b3125f964900df3f41c39ec0c94"},
+    {file = "ddtrace-3.12.4-cp310-cp310-win32.whl", hash = "sha256:b331bc0c3000cea1fd70febcf004b5a617c63b9050094f08100891a23638986d"},
+    {file = "ddtrace-3.12.4-cp310-cp310-win_amd64.whl", hash = "sha256:018d19e2a1e7585df65d938ae51c385d673e8001b66827a47e499ade3b227ad2"},
+    {file = "ddtrace-3.12.4-cp311-cp311-macosx_12_0_arm64.whl", hash = "sha256:0de9563bad27007fd64059e3b5bb3a791184e39619fdb096044e68a454b4427b"},
+    {file = "ddtrace-3.12.4-cp311-cp311-macosx_12_0_x86_64.whl", hash = "sha256:d0c5b84d066ca3d60da9636df526382416dae4288f66fcdaca7a2e765ca2f0bd"},
+    {file = "ddtrace-3.12.4-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:ff1812b1d7e8344088a978f1d4f621257fe1ad5d8efc07317a3c90c280e5bdc4"},
+    {file = "ddtrace-3.12.4-cp311-cp311-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:dd0ac6ba50d36689bf0eeadc88ce91b60bc863036f3dea90dd5656f39bce3ac4"},
+    {file = "ddtrace-3.12.4-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:b8f99761f946b2b7cc2ea4cba821a7a94d05a9eb8cd8a3feabdb49eeacc18bb9"},
+    {file = "ddtrace-3.12.4-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:c4f66c48eca7d6759766fcaf24ac3a65e712e62ae7b1f521a7da2b8d7f101849"},
+    {file = "ddtrace-3.12.4-cp311-cp311-musllinux_1_2_i686.whl", hash = "sha256:42d46f17baaa5040e4f438544603033af8eeec32067c3712a9e620392d75f484"},
+    {file = "ddtrace-3.12.4-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:aa0606a07e7d05881f2ef1172f4175733ae3006bfc3c7cfd58b82ea3ed75c914"},
+    {file = "ddtrace-3.12.4-cp311-cp311-win32.whl", hash = "sha256:efde4b33502f3897993a564ee56d0ea30a65d658d616d16c5ef23c850d0e3417"},
+    {file = "ddtrace-3.12.4-cp311-cp311-win_amd64.whl", hash = "sha256:7d6117fabcd98d3a696d1f80314c9b9e4325b362b31714551efd729a02152ff1"},
+    {file = "ddtrace-3.12.4-cp312-cp312-macosx_12_0_arm64.whl", hash = "sha256:734d782d9f64de378f632516554b9da0dfbf54cf1bb7be4bb1085165e7c052ad"},
+    {file = "ddtrace-3.12.4-cp312-cp312-macosx_12_0_x86_64.whl", hash = "sha256:fbf2543856b4ed5a1d6ac59c82f8c76cef5f4ef65361d59f60ce01db92a4c8d1"},
+    {file = "ddtrace-3.12.4-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:751ce0410405113286bd558fd402f8a58f5b455cee4deb467ae9ae87e5713547"},
+    {file = "ddtrace-3.12.4-cp312-cp312-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:fd804c06d62926cc18a354987f7d5c1fecd1da30983041d3f98bc402d9d23713"},
+    {file = "ddtrace-3.12.4-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:e55b911d5b9f1bd73731870962809f9089677f4d3736d52587b4ba76eee56962"},
+    {file = "ddtrace-3.12.4-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:1e8cc90fdcd7f021d06383b88c0e40726706c06088dddd528e31cf3c65a9fea9"},
+    {file = "ddtrace-3.12.4-cp312-cp312-musllinux_1_2_i686.whl", hash = "sha256:585b7b26f03c64390c800e180304639b4226c34c533f16bc6cd9c328ee4f727a"},
+    {file = "ddtrace-3.12.4-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:fe967af58f2e0033caa977c512a4bfb7af3c6f5ad57e9bdef9241609a4d8a99b"},
+    {file = "ddtrace-3.12.4-cp312-cp312-win32.whl", hash = "sha256:fe03b8f513513e28c35bc792cd7ef0602b21cbcfe71d17a2dd962aee23e980d9"},
+    {file = "ddtrace-3.12.4-cp312-cp312-win_amd64.whl", hash = "sha256:9fd79c44ecffb36ac5b3168f0f196778ed0dd538beb07961ce10e06b8045af35"},
+    {file = "ddtrace-3.12.4-cp313-cp313-macosx_12_0_arm64.whl", hash = "sha256:2edf755f4bfd823ce8b560c233cb17137ef79d097bc1ade7914f684b39011bcb"},
+    {file = "ddtrace-3.12.4-cp313-cp313-macosx_12_0_x86_64.whl", hash = "sha256:6dad7ca193810beb931e81b7430dd074a53bf8f8bd5bdc19acd198d460b2438a"},
+    {file = "ddtrace-3.12.4-cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:9de7aa6b6ea3d41f8f20c5e00dd85b2f2b3bb1591f3b7deab5d4c527620c3cb3"},
+    {file = "ddtrace-3.12.4-cp313-cp313-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:80e0acbbe85365f113bf6e57f77a82f0e0612a7a4cb57f16e9e184748a2bc478"},
+    {file = "ddtrace-3.12.4-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:46de7dd48256d8e347f2ab436644bd8946d3605caedb150eb46327a9f5b005b6"},
+    {file = "ddtrace-3.12.4-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:4d5c9ddacecb0072292360813b453129998ca293e13c542fa51771c7734ef03a"},
+    {file = "ddtrace-3.12.4-cp313-cp313-musllinux_1_2_i686.whl", hash = "sha256:d0b694838e6c7ea2da6de7ccd7b866ec439c49fa40b68ac46f657163cb571d93"},
+    {file = "ddtrace-3.12.4-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:e89a17cdb4b5442b97a219e8522b9c665cf7a5116f7e97049dd145f837bad5b1"},
+    {file = "ddtrace-3.12.4-cp313-cp313-win32.whl", hash = "sha256:d0b3ec8228950e7ff68c39537630cd12880656d96461ef021d6484b2df8dba84"},
+    {file = "ddtrace-3.12.4-cp313-cp313-win_amd64.whl", hash = "sha256:fad78414731b242e86016a124299f2f41575ccf58444edca777b425dbd9faf0c"},
+    {file = "ddtrace-3.12.4-cp38-cp38-macosx_12_0_arm64.whl", hash = "sha256:9f639f70f1689ec1a1049cd64132491ee09bcfe7609d73f8c220e38261611045"},
+    {file = "ddtrace-3.12.4-cp38-cp38-macosx_12_0_x86_64.whl", hash = "sha256:6b5b150e9d362f7242159dd5a5a7107f1be091282c0ee69301fb7ede60f28d3c"},
+    {file = "ddtrace-3.12.4-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:eda3b6ebd275f7f7272f45f4e8ee0e0720c1e217c80140270f8c5e415e11133e"},
+    {file = "ddtrace-3.12.4-cp38-cp38-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:fe644904b44d39a93eb40fb033aef26a03e4096d135ee844b71ed49d1bd647ad"},
+    {file = "ddtrace-3.12.4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:62a48fc36308919afb1fae22a268a96cff3448f1feb860db97d130498ddfa428"},
+    {file = "ddtrace-3.12.4-cp38-cp38-musllinux_1_2_aarch64.whl", hash = "sha256:77de49365f55033d7e14b544f92d0cae71969b78c4ab8642c3340124e0200739"},
+    {file = "ddtrace-3.12.4-cp38-cp38-musllinux_1_2_i686.whl", hash = "sha256:87fbd5126f8339bcb508a52455f58b0c92870a1c3748849a4d6543198b5f8752"},
+    {file = "ddtrace-3.12.4-cp38-cp38-musllinux_1_2_x86_64.whl", hash = "sha256:5845d7c2ed46b44e02bd5d36ca7f8e80a4e942683473c867393b9fd4553f9d64"},
+    {file = "ddtrace-3.12.4-cp38-cp38-win32.whl", hash = "sha256:ebde5af8c5d98f435d7dec960c97151142a4b302e94c20da79ed58fe8a08052e"},
+    {file = "ddtrace-3.12.4-cp38-cp38-win_amd64.whl", hash = "sha256:18dfe9a1a02bfa4ef4f614122135509f454abeff625039b764bc461462ba0923"},
+    {file = "ddtrace-3.12.4-cp39-cp39-macosx_12_0_arm64.whl", hash = "sha256:e78957120c64bd56ce5592bc10587d7c0d1ca68f21f5b46f6a18dafbc43ad234"},
+    {file = "ddtrace-3.12.4-cp39-cp39-macosx_12_0_x86_64.whl", hash = "sha256:3936243dc989b8e8e3bb004262abe68a1cc3e0b9356671c01233b84d2c837903"},
+    {file = "ddtrace-3.12.4-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:ed76d10787fc288ea94808ce601df243fc3953c7142baefac446015bed799790"},
+    {file = "ddtrace-3.12.4-cp39-cp39-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:0c1d3f7f93146653f8ed06d8cd54030b2c902ceca6de55f6df7f40d23037181e"},
+    {file = "ddtrace-3.12.4-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:0f5ab24c82fc7532386b02530f90fed2964718cea296adf6d35fc31bd30d301d"},
+    {file = "ddtrace-3.12.4-cp39-cp39-musllinux_1_2_aarch64.whl", hash = "sha256:30bd9e57923a99d5b4e6562976e9f7307d685caff1544b3d2f7438e6ef8e87e8"},
+    {file = "ddtrace-3.12.4-cp39-cp39-musllinux_1_2_i686.whl", hash = "sha256:3bf18fd5898940fb7f236b4c9796f0ee517eb755fd0c17965d3a0342f865ee5a"},
+    {file = "ddtrace-3.12.4-cp39-cp39-musllinux_1_2_x86_64.whl", hash = "sha256:8ff1c70da37c05a29f0be091b0fdc6bb1d91d448f56861c51df614649441070c"},
+    {file = "ddtrace-3.12.4-cp39-cp39-win32.whl", hash = "sha256:66c007170698e3d12638d03e80f02e93c3bb3e55e96a7f5517e638056562ec1a"},
+    {file = "ddtrace-3.12.4-cp39-cp39-win_amd64.whl", hash = "sha256:a4f2dabbc95e5c6bf4c43eb141e94021789c81a929588f4000f876f89882c124"},
+    {file = "ddtrace-3.12.4.tar.gz", hash = "sha256:c422977fc4f6e9ba7d4eef9b7e6ce00f8b81c68b034682c6a63eb5c9670e37d8"},
 ]
 
 [package.dependencies]
@@ -1902,25 +1901,25 @@ files = [
 
 [[package]]
 name = "fastapi"
-version = "0.117.1"
+version = "0.116.1"
 description = "FastAPI framework, high performance, easy to learn, fast to code, ready for production"
 optional = false
 python-versions = ">=3.8"
 groups = ["main"]
 files = [
-    {file = "fastapi-0.117.1-py3-none-any.whl", hash = "sha256:33c51a0d21cab2b9722d4e56dbb9316f3687155be6b276191790d8da03507552"},
-    {file = "fastapi-0.117.1.tar.gz", hash = "sha256:fb2d42082d22b185f904ca0ecad2e195b851030bd6c5e4c032d1c981240c631a"},
+    {file = "fastapi-0.116.1-py3-none-any.whl", hash = "sha256:c46ac7c312df840f0c9e220f7964bada936781bc4e2e6eb71f1c4d7553786565"},
+    {file = "fastapi-0.116.1.tar.gz", hash = "sha256:ed52cbf946abfd70c5a0dccb24673f0670deeb517a88b3544d03c2a6bf283143"},
 ]
 
 [package.dependencies]
 pydantic = ">=1.7.4,<1.8 || >1.8,<1.8.1 || >1.8.1,<2.0.0 || >2.0.0,<2.0.1 || >2.0.1,<2.1.0 || >2.1.0,<3.0.0"
-starlette = ">=0.40.0,<0.49.0"
+starlette = ">=0.40.0,<0.48.0"
 typing-extensions = ">=4.8.0"
 
 [package.extras]
-all = ["email-validator (>=2.0.0)", "fastapi-cli[standard] (>=0.0.8)", "httpx (>=0.23.0,<1.0.0)", "itsdangerous (>=1.1.0)", "jinja2 (>=3.1.5)", "orjson (>=3.2.1)", "pydantic-extra-types (>=2.0.0)", "pydantic-settings (>=2.0.0)", "python-multipart (>=0.0.18)", "pyyaml (>=5.3.1)", "ujson (>=4.0.1,!=4.0.2,!=4.1.0,!=4.2.0,!=4.3.0,!=5.0.0,!=5.1.0)", "uvicorn[standard] (>=0.12.0)"]
-standard = ["email-validator (>=2.0.0)", "fastapi-cli[standard] (>=0.0.8)", "httpx (>=0.23.0,<1.0.0)", "jinja2 (>=3.1.5)", "python-multipart (>=0.0.18)", "uvicorn[standard] (>=0.12.0)"]
-standard-no-fastapi-cloud-cli = ["email-validator (>=2.0.0)", "fastapi-cli[standard-no-fastapi-cloud-cli] (>=0.0.8)", "httpx (>=0.23.0,<1.0.0)", "jinja2 (>=3.1.5)", "python-multipart (>=0.0.18)", "uvicorn[standard] (>=0.12.0)"]
+all = ["email-validator (>=2.0.0)", "fastapi-cli[standard] (>=0.0.8)", "httpx (>=0.23.0)", "itsdangerous (>=1.1.0)", "jinja2 (>=3.1.5)", "orjson (>=3.2.1)", "pydantic-extra-types (>=2.0.0)", "pydantic-settings (>=2.0.0)", "python-multipart (>=0.0.18)", "pyyaml (>=5.3.1)", "ujson (>=4.0.1,!=4.0.2,!=4.1.0,!=4.2.0,!=4.3.0,!=5.0.0,!=5.1.0)", "uvicorn[standard] (>=0.12.0)"]
+standard = ["email-validator (>=2.0.0)", "fastapi-cli[standard] (>=0.0.8)", "httpx (>=0.23.0)", "jinja2 (>=3.1.5)", "python-multipart (>=0.0.18)", "uvicorn[standard] (>=0.12.0)"]
+standard-no-fastapi-cloud-cli = ["email-validator (>=2.0.0)", "fastapi-cli[standard-no-fastapi-cloud-cli] (>=0.0.8)", "httpx (>=0.23.0)", "jinja2 (>=3.1.5)", "python-multipart (>=0.0.18)", "uvicorn[standard] (>=0.12.0)"]
 
 [[package]]
 name = "fastjsonschema"
@@ -2292,72 +2291,6 @@ test-downstream = ["aiobotocore (>=2.5.4,<3.0.0)", "dask[dataframe,test]", "moto
 test-full = ["adlfs", "aiohttp (!=4.0.0a0,!=4.0.0a1)", "cloudpickle", "dask", "distributed", "dropbox", "dropboxdrivefs", "fastparquet", "fusepy", "gcsfs", "jinja2", "kerchunk", "libarchive-c", "lz4", "notebook", "numpy", "ocifs", "pandas", "panel", "paramiko", "pyarrow", "pyarrow (>=1)", "pyftpdlib", "pygit2", "pytest", "pytest-asyncio (!=0.22.0)", "pytest-benchmark", "pytest-cov", "pytest-mock", "pytest-recording", "pytest-rerunfailures", "python-snappy", "requests", "smbprotocol", "tqdm", "urllib3", "zarr", "zstandard ; python_version < \"3.14\""]
 tqdm = ["tqdm"]
 
-[[package]]
-name = "gevent"
-version = "25.9.1"
-description = "Coroutine-based network library"
-optional = false
-python-versions = ">=3.9"
-groups = ["test"]
-files = [
-    {file = "gevent-25.9.1-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:856b990be5590e44c3a3dc6c8d48a40eaccbb42e99d2b791d11d1e7711a4297e"},
-    {file = "gevent-25.9.1-cp310-cp310-manylinux2014_ppc64le.manylinux_2_17_ppc64le.whl", hash = "sha256:fe1599d0b30e6093eb3213551751b24feeb43db79f07e89d98dd2f3330c9063e"},
-    {file = "gevent-25.9.1-cp310-cp310-manylinux2014_s390x.manylinux_2_17_s390x.whl", hash = "sha256:f0d8b64057b4bf1529b9ef9bd2259495747fba93d1f836c77bfeaacfec373fd0"},
-    {file = "gevent-25.9.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:b56cbc820e3136ba52cd690bdf77e47a4c239964d5f80dc657c1068e0fe9521c"},
-    {file = "gevent-25.9.1-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:c5fa9ce5122c085983e33e0dc058f81f5264cebe746de5c401654ab96dddfca8"},
-    {file = "gevent-25.9.1-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:03c74fec58eda4b4edc043311fca8ba4f8744ad1632eb0a41d5ec25413581975"},
-    {file = "gevent-25.9.1-cp310-cp310-win_amd64.whl", hash = "sha256:a8ae9f895e8651d10b0a8328a61c9c53da11ea51b666388aa99b0ce90f9fdc27"},
-    {file = "gevent-25.9.1-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:18e5aff9e8342dc954adb9c9c524db56c2f3557999463445ba3d9cbe3dada7b7"},
-    {file = "gevent-25.9.1-cp311-cp311-manylinux2014_ppc64le.manylinux_2_17_ppc64le.whl", hash = "sha256:1cdf6db28f050ee103441caa8b0448ace545364f775059d5e2de089da975c457"},
-    {file = "gevent-25.9.1-cp311-cp311-manylinux2014_s390x.manylinux_2_17_s390x.whl", hash = "sha256:812debe235a8295be3b2a63b136c2474241fa5c58af55e6a0f8cfc29d4936235"},
-    {file = "gevent-25.9.1-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:b28b61ff9216a3d73fe8f35669eefcafa957f143ac534faf77e8a19eb9e6883a"},
-    {file = "gevent-25.9.1-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:5e4b6278b37373306fc6b1e5f0f1cf56339a1377f67c35972775143d8d7776ff"},
-    {file = "gevent-25.9.1-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:d99f0cb2ce43c2e8305bf75bee61a8bde06619d21b9d0316ea190fc7a0620a56"},
-    {file = "gevent-25.9.1-cp311-cp311-win_amd64.whl", hash = "sha256:72152517ecf548e2f838c61b4be76637d99279dbaa7e01b3924df040aa996586"},
-    {file = "gevent-25.9.1-cp312-cp312-macosx_11_0_universal2.whl", hash = "sha256:46b188248c84ffdec18a686fcac5dbb32365d76912e14fda350db5dc0bfd4f86"},
-    {file = "gevent-25.9.1-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:f2b54ea3ca6f0c763281cd3f96010ac7e98c2e267feb1221b5a26e2ca0b9a692"},
-    {file = "gevent-25.9.1-cp312-cp312-manylinux2014_ppc64le.manylinux_2_17_ppc64le.whl", hash = "sha256:7a834804ac00ed8a92a69d3826342c677be651b1c3cd66cc35df8bc711057aa2"},
-    {file = "gevent-25.9.1-cp312-cp312-manylinux2014_s390x.manylinux_2_17_s390x.whl", hash = "sha256:323a27192ec4da6b22a9e51c3d9d896ff20bc53fdc9e45e56eaab76d1c39dd74"},
-    {file = "gevent-25.9.1-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:6ea78b39a2c51d47ff0f130f4c755a9a4bbb2dd9721149420ad4712743911a51"},
-    {file = "gevent-25.9.1-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:dc45cd3e1cc07514a419960af932a62eb8515552ed004e56755e4bf20bad30c5"},
-    {file = "gevent-25.9.1-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:34e01e50c71eaf67e92c186ee0196a039d6e4f4b35670396baed4a2d8f1b347f"},
-    {file = "gevent-25.9.1-cp312-cp312-win_amd64.whl", hash = "sha256:4acd6bcd5feabf22c7c5174bd3b9535ee9f088d2bbce789f740ad8d6554b18f3"},
-    {file = "gevent-25.9.1-cp313-cp313-macosx_11_0_universal2.whl", hash = "sha256:4f84591d13845ee31c13f44bdf6bd6c3dbf385b5af98b2f25ec328213775f2ed"},
-    {file = "gevent-25.9.1-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:9cdbb24c276a2d0110ad5c978e49daf620b153719ac8a548ce1250a7eb1b9245"},
-    {file = "gevent-25.9.1-cp313-cp313-manylinux2014_ppc64le.manylinux_2_17_ppc64le.whl", hash = "sha256:88b6c07169468af631dcf0fdd3658f9246d6822cc51461d43f7c44f28b0abb82"},
-    {file = "gevent-25.9.1-cp313-cp313-manylinux2014_s390x.manylinux_2_17_s390x.whl", hash = "sha256:b7bb0e29a7b3e6ca9bed2394aa820244069982c36dc30b70eb1004dd67851a48"},
-    {file = "gevent-25.9.1-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:2951bb070c0ee37b632ac9134e4fdaad70d2e660c931bb792983a0837fe5b7d7"},
-    {file = "gevent-25.9.1-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:e4e17c2d57e9a42e25f2a73d297b22b60b2470a74be5a515b36c984e1a246d47"},
-    {file = "gevent-25.9.1-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:8d94936f8f8b23d9de2251798fcb603b84f083fdf0d7f427183c1828fb64f117"},
-    {file = "gevent-25.9.1-cp313-cp313-win_amd64.whl", hash = "sha256:eb51c5f9537b07da673258b4832f6635014fee31690c3f0944d34741b69f92fa"},
-    {file = "gevent-25.9.1-cp314-cp314-macosx_11_0_universal2.whl", hash = "sha256:1a3fe4ea1c312dbf6b375b416925036fe79a40054e6bf6248ee46526ea628be1"},
-    {file = "gevent-25.9.1-cp314-cp314-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:0adb937f13e5fb90cca2edf66d8d7e99d62a299687400ce2edee3f3504009356"},
-    {file = "gevent-25.9.1-cp314-cp314-manylinux2014_ppc64le.manylinux_2_17_ppc64le.whl", hash = "sha256:427f869a2050a4202d93cf7fd6ab5cffb06d3e9113c10c967b6e2a0d45237cb8"},
-    {file = "gevent-25.9.1-cp314-cp314-manylinux2014_s390x.manylinux_2_17_s390x.whl", hash = "sha256:c049880175e8c93124188f9d926af0a62826a3b81aa6d3074928345f8238279e"},
-    {file = "gevent-25.9.1-cp314-cp314-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:b5a67a0974ad9f24721034d1e008856111e0535f1541499f72a733a73d658d1c"},
-    {file = "gevent-25.9.1-cp314-cp314-musllinux_1_2_aarch64.whl", hash = "sha256:1d0f5d8d73f97e24ea8d24d8be0f51e0cf7c54b8021c1fddb580bf239474690f"},
-    {file = "gevent-25.9.1-cp314-cp314-musllinux_1_2_x86_64.whl", hash = "sha256:ddd3ff26e5c4240d3fbf5516c2d9d5f2a998ef87cfb73e1429cfaeaaec860fa6"},
-    {file = "gevent-25.9.1-cp314-cp314-win_amd64.whl", hash = "sha256:bb63c0d6cb9950cc94036a4995b9cc4667b8915366613449236970f4394f94d7"},
-    {file = "gevent-25.9.1-cp39-cp39-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:f18f80aef6b1f6907219affe15b36677904f7cfeed1f6a6bc198616e507ae2d7"},
-    {file = "gevent-25.9.1-cp39-cp39-musllinux_1_2_x86_64.whl", hash = "sha256:b274a53e818124a281540ebb4e7a2c524778f745b7a99b01bdecf0ca3ac0ddb0"},
-    {file = "gevent-25.9.1-cp39-cp39-win32.whl", hash = "sha256:c6c91f7e33c7f01237755884316110ee7ea076f5bdb9aa0982b6dc63243c0a38"},
-    {file = "gevent-25.9.1-cp39-cp39-win_amd64.whl", hash = "sha256:012a44b0121f3d7c800740ff80351c897e85e76a7e4764690f35c5ad9ec17de5"},
-    {file = "gevent-25.9.1.tar.gz", hash = "sha256:adf9cd552de44a4e6754c51ff2e78d9193b7fa6eab123db9578a210e657235dd"},
-]
-
-[package.dependencies]
-cffi = {version = ">=1.17.1", markers = "platform_python_implementation == \"CPython\" and sys_platform == \"win32\""}
-greenlet = {version = ">=3.2.2", markers = "platform_python_implementation == \"CPython\""}
-"zope.event" = "*"
-"zope.interface" = "*"
-
-[package.extras]
-dnspython = ["dnspython (>=1.16.0,<2.0) ; python_version < \"3.10\"", "idna ; python_version < \"3.10\""]
-docs = ["furo", "repoze.sphinx.autointerface", "sphinx", "sphinxcontrib-programoutput", "zope.schema"]
-monitor = ["psutil (>=5.7.0) ; sys_platform != \"win32\" or platform_python_implementation == \"CPython\""]
-recommended = ["cffi (>=1.17.1) ; platform_python_implementation == \"CPython\"", "dnspython (>=1.16.0,<2.0) ; python_version < \"3.10\"", "idna ; python_version < \"3.10\"", "psutil (>=5.7.0) ; sys_platform != \"win32\" or platform_python_implementation == \"CPython\""]
-test = ["cffi (>=1.17.1) ; platform_python_implementation == \"CPython\"", "coverage (>=5.0) ; sys_platform != \"win32\"", "dnspython (>=1.16.0,<2.0) ; python_version < \"3.10\"", "idna ; python_version < \"3.10\"", "objgraph", "psutil (>=5.7.0) ; sys_platform != \"win32\" or platform_python_implementation == \"CPython\"", "requests"]
-
 [[package]]
 name = "gitdb"
 version = "4.0.12"
@@ -2392,6 +2325,27 @@ gitdb = ">=4.0.1,<5"
 doc = ["sphinx (>=7.1.2,<7.2)", "sphinx-autodoc-typehints", "sphinx_rtd_theme"]
 test = ["coverage[toml]", "ddt (>=1.1.1,!=1.4.3)", "mock ; python_version < \"3.8\"", "mypy", "pre-commit", "pytest (>=7.3.1)", "pytest-cov", "pytest-instafail", "pytest-mock", "pytest-sugar", "typing-extensions ; python_version < \"3.11\""]
 
+[[package]]
+name = "google-ai-generativelanguage"
+version = "0.6.15"
+description = "Google Ai Generativelanguage API client library"
+optional = false
+python-versions = ">=3.7"
+groups = ["main"]
+files = [
+    {file = "google_ai_generativelanguage-0.6.15-py3-none-any.whl", hash = "sha256:5a03ef86377aa184ffef3662ca28f19eeee158733e45d7947982eb953c6ebb6c"},
+    {file = "google_ai_generativelanguage-0.6.15.tar.gz", hash = "sha256:8f6d9dc4c12b065fe2d0289026171acea5183ebf2d0b11cefe12f3821e159ec3"},
+]
+
+[package.dependencies]
+google-api-core = {version = ">=1.34.1,<2.0.dev0 || >=2.11.dev0,<3.0.0dev", extras = ["grpc"]}
+google-auth = ">=2.14.1,<2.24.0 || >2.24.0,<2.25.0 || >2.25.0,<3.0.0dev"
+proto-plus = [
+    {version = ">=1.25.0,<2.0.0dev", markers = "python_version >= \"3.13\""},
+    {version = ">=1.22.3,<2.0.0dev"},
+]
+protobuf = ">=3.20.2,<4.21.0 || >4.21.0,<4.21.1 || >4.21.1,<4.21.2 || >4.21.2,<4.21.3 || >4.21.3,<4.21.4 || >4.21.4,<4.21.5 || >4.21.5,<6.0.0dev"
+
 [[package]]
 name = "google-api-core"
 version = "2.25.1"
@@ -2730,6 +2684,30 @@ websockets = ">=13.0.0,<15.1.0"
 [package.extras]
 aiohttp = ["aiohttp (<4.0.0)"]
 
+[[package]]
+name = "google-generativeai"
+version = "0.8.5"
+description = "Google Generative AI High level API client library and tools."
+optional = false
+python-versions = ">=3.9"
+groups = ["main"]
+files = [
+    {file = "google_generativeai-0.8.5-py3-none-any.whl", hash = "sha256:22b420817fb263f8ed520b33285f45976d5b21e904da32b80d4fd20c055123a2"},
+]
+
+[package.dependencies]
+google-ai-generativelanguage = "0.6.15"
+google-api-core = "*"
+google-api-python-client = "*"
+google-auth = ">=2.15.0"
+protobuf = "*"
+pydantic = "*"
+tqdm = "*"
+typing-extensions = "*"
+
+[package.extras]
+dev = ["Pillow", "absl-py", "black", "ipython", "nose2", "pandas", "pytype", "pyyaml"]
+
 [[package]]
 name = "google-resumable-media"
 version = "2.7.2"
@@ -2774,7 +2752,7 @@ version = "3.2.4"
 description = "Lightweight in-process concurrent programming"
 optional = false
 python-versions = ">=3.9"
-groups = ["main", "test"]
+groups = ["main"]
 files = [
     {file = "greenlet-3.2.4-cp310-cp310-macosx_11_0_universal2.whl", hash = "sha256:8c68325b0d0acf8d91dde4e6f930967dd52a5302cd4062932a6b2e7c2969f47c"},
     {file = "greenlet-3.2.4-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:94385f101946790ae13da500603491f04a76b6e4c059dab271b3ce2e283b2590"},
@@ -2831,7 +2809,6 @@ files = [
     {file = "greenlet-3.2.4-cp39-cp39-win_amd64.whl", hash = "sha256:d2e685ade4dafd447ede19c31277a224a239a0a1a4eca4e6390efedf20260cfb"},
     {file = "greenlet-3.2.4.tar.gz", hash = "sha256:0dca0d95ff849f9a364385f36ab49f50065d76964944638be9691e1832e9f86d"},
 ]
-markers = {test = "platform_python_implementation == \"CPython\""}
 
 [package.extras]
 docs = ["Sphinx", "furo"]
@@ -5431,7 +5408,7 @@ llama = ["llama-index (>=0.12.29,<0.13.0)", "llama-index-core (>=0.12.29,<0.13.0
 
 [[package]]
 name = "openhands-ai"
-version = "0.57.0"
+version = "0.55.0"
 description = "OpenHands: Code Less, Make More"
 optional = false
 python-versions = "^3.12,<3.14"
@@ -5455,7 +5432,7 @@ google-api-python-client = "^2.164.0"
 google-auth-httplib2 = "*"
 google-auth-oauthlib = "*"
 google-cloud-aiplatform = "*"
-google-genai = "*"
+google-generativeai = "*"
 html2text = "*"
 httpx-aiohttp = "^0.1.8"
 ipywidgets = "^8.1.5"
@@ -5465,7 +5442,7 @@ json-repair = "*"
 jupyter_kernel_gateway = "*"
 kubernetes = "^33.1.0"
 libtmux = ">=0.37,<0.40"
-litellm = ">=1.74.3, <1.77.2, !=1.64.4, !=1.67.*"
+litellm = "^1.74.3, !=1.64.4, !=1.67.*"
 memory-profiler = "^0.61.0"
 numpy = "*"
 openai = "1.99.9"
@@ -5474,7 +5451,6 @@ opentelemetry-api = "^1.33.1"
 opentelemetry-exporter-otlp-proto-grpc = "^1.33.1"
 pathspec = "^0.12.1"
 pexpect = "*"
-pillow = "^11.3.0"
 poetry = "^2.1.2"
 prompt-toolkit = "^3.0.50"
 protobuf = "^5.0.0,<6.0.0"
@@ -5482,7 +5458,6 @@ psutil = "*"
 pygithub = "^2.5.0"
 pyjwt = "^2.9.0"
 pylatexenc = "*"
-pypdf = "^6.0.0"
 PyPDF2 = "*"
 python-docx = "*"
 python-dotenv = "*"
@@ -5496,23 +5471,19 @@ pyyaml = "^6.0.2"
 qtconsole = "^5.6.1"
 rapidfuzz = "^3.9.0"
 redis = ">=5.2,<7.0"
-requests = "^2.32.5"
-setuptools = ">=78.1.1"
 shellingham = "^1.5.4"
-sse-starlette = "^3.0.2"
-starlette = "^0.48.0"
+sse-starlette = "^2.1.3"
 tenacity = ">=8.5,<10.0"
 termcolor = "*"
 toml = "*"
 tornado = "*"
 types-toml = "*"
-urllib3 = "^2.5.0"
 uvicorn = "*"
 whatthepatch = "^1.0.6"
 zope-interface = "7.2"
 
 [package.extras]
-third-party-runtimes = ["daytona (==0.24.2)", "e2b-code-interpreter (>=2.0.0,<3.0.0)", "modal (>=0.66.26,<1.2.0)", "runloop-api-client (==0.50.0)"]
+third-party-runtimes = ["daytona (==0.24.2)", "e2b (>=1.0.5,<1.8.0)", "modal (>=0.66.26,<1.2.0)", "runloop-api-client (==0.50.0)"]
 
 [package.source]
 type = "directory"
@@ -6545,12 +6516,11 @@ version = "2.22"
 description = "C parser in Python"
 optional = false
 python-versions = ">=3.8"
-groups = ["main", "test"]
+groups = ["main"]
 files = [
     {file = "pycparser-2.22-py3-none-any.whl", hash = "sha256:c3702b6d3dd8c7abc1afa565d7e63d53a1d0bd86cdc24edd75470f4de499cfcc"},
     {file = "pycparser-2.22.tar.gz", hash = "sha256:491c8be9c040f5390f5bf44a5b07752bd07f56edf992381b05c701439eec10f6"},
 ]
-markers = {test = "platform_python_implementation == \"CPython\" and sys_platform == \"win32\""}
 
 [[package]]
 name = "pydantic"
@@ -8340,7 +8310,7 @@ version = "80.9.0"
 description = "Easily download, build, install, upgrade, and uninstall Python packages"
 optional = false
 python-versions = ">=3.9"
-groups = ["main", "test"]
+groups = ["main"]
 files = [
     {file = "setuptools-80.9.0-py3-none-any.whl", hash = "sha256:062d34222ad13e0cc312a4c02d73f059e86a4acbfbdea8f8f76b28c99f306922"},
     {file = "setuptools-80.9.0.tar.gz", hash = "sha256:f36b47402ecde768dbfafc46e8e4207b4360c654f1f3bb84475f0a28628fb19c"},
@@ -8706,14 +8676,14 @@ sqlcipher = ["sqlcipher3_binary"]
 
 [[package]]
 name = "sse-starlette"
-version = "3.0.2"
+version = "2.4.1"
 description = "SSE plugin for Starlette"
 optional = false
 python-versions = ">=3.9"
 groups = ["main"]
 files = [
-    {file = "sse_starlette-3.0.2-py3-none-any.whl", hash = "sha256:16b7cbfddbcd4eaca11f7b586f3b8a080f1afe952c15813455b162edea619e5a"},
-    {file = "sse_starlette-3.0.2.tar.gz", hash = "sha256:ccd60b5765ebb3584d0de2d7a6e4f745672581de4f5005ab31c3a25d10b52b3a"},
+    {file = "sse_starlette-2.4.1-py3-none-any.whl", hash = "sha256:08b77ea898ab1a13a428b2b6f73cfe6d0e607a7b4e15b9bb23e4a37b087fd39a"},
+    {file = "sse_starlette-2.4.1.tar.gz", hash = "sha256:7c8a800a1ca343e9165fc06bbda45c78e4c6166320707ae30b416c42da070926"},
 ]
 
 [package.dependencies]
@@ -8721,7 +8691,7 @@ anyio = ">=4.7.0"
 
 [package.extras]
 daphne = ["daphne (>=4.2.0)"]
-examples = ["aiosqlite (>=0.21.0)", "fastapi (>=0.115.12)", "sqlalchemy[asyncio] (>=2.0.41)", "starlette (>=0.41.3)", "uvicorn (>=0.34.0)"]
+examples = ["aiosqlite (>=0.21.0)", "fastapi (>=0.115.12)", "sqlalchemy[asyncio,examples] (>=2.0.41)", "starlette (>=0.41.3)", "uvicorn (>=0.34.0)"]
 granian = ["granian (>=2.3.1)"]
 uvicorn = ["uvicorn (>=0.34.0)"]
 
@@ -8777,14 +8747,14 @@ files = [
 
 [[package]]
 name = "starlette"
-version = "0.48.0"
+version = "0.47.3"
 description = "The little ASGI library that shines."
 optional = false
 python-versions = ">=3.9"
 groups = ["main"]
 files = [
-    {file = "starlette-0.48.0-py3-none-any.whl", hash = "sha256:0764ca97b097582558ecb498132ed0c7d942f233f365b86ba37770e026510659"},
-    {file = "starlette-0.48.0.tar.gz", hash = "sha256:7e8cee469a8ab2352911528110ce9088fdc6a37d9876926e73da7ce4aa4c7a46"},
+    {file = "starlette-0.47.3-py3-none-any.whl", hash = "sha256:89c0778ca62a76b826101e7c709e70680a1699ca7da6b44d38eb0a7e61fe4b51"},
+    {file = "starlette-0.47.3.tar.gz", hash = "sha256:6bc94f839cc176c4858894f1f8908f0ab79dfec1a6b8402f6da9be26ebea52e9"},
 ]
 
 [package.dependencies]
@@ -9913,32 +9883,13 @@ enabler = ["pytest-enabler (>=2.2)"]
 test = ["big-O", "jaraco.functools", "jaraco.itertools", "jaraco.test", "more_itertools", "pytest (>=6,!=8.1.*)", "pytest-ignore-flaky"]
 type = ["pytest-mypy"]
 
-[[package]]
-name = "zope-event"
-version = "6.0"
-description = "Very basic event publishing system"
-optional = false
-python-versions = ">=3.9"
-groups = ["test"]
-files = [
-    {file = "zope_event-6.0-py3-none-any.whl", hash = "sha256:6f0922593407cc673e7d8766b492c519f91bdc99f3080fe43dcec0a800d682a3"},
-    {file = "zope_event-6.0.tar.gz", hash = "sha256:0ebac894fa7c5f8b7a89141c272133d8c1de6ddc75ea4b1f327f00d1f890df92"},
-]
-
-[package.dependencies]
-setuptools = ">=75.8.2"
-
-[package.extras]
-docs = ["Sphinx"]
-test = ["zope.testrunner (>=6.4)"]
-
 [[package]]
 name = "zope-interface"
 version = "7.2"
 description = "Interfaces for Python"
 optional = false
 python-versions = ">=3.8"
-groups = ["main", "test"]
+groups = ["main"]
 files = [
     {file = "zope.interface-7.2-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:ce290e62229964715f1011c3dbeab7a4a1e4971fd6f31324c4519464473ef9f2"},
     {file = "zope.interface-7.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:05b910a5afe03256b58ab2ba6288960a2892dfeef01336dc4be6f1b9ed02ab0a"},
@@ -10102,4 +10053,4 @@ cffi = ["cffi (>=1.17) ; python_version >= \"3.13\" and platform_python_implemen
 [metadata]
 lock-version = "2.1"
 python-versions = "^3.12,<3.14"
-content-hash = "8c460070dce6bdec5ee0ee7bc0c2246fcf2602d1e64a0867b4f5e3a0e334fe93"
+content-hash = "0e611931bd3823ee8b6d832b6ef444868a644e21927a9fb72d4aeaab8170028e"

enterprise/pyproject.toml

@@ -37,7 +37,7 @@ sqlalchemy = { extras = [ "asyncio" ], version = "^2.0.40" }
 resend = "^2.7.0"
 tenacity = "^9.1.2"
 slack-sdk = "^3.35.0"
-ddtrace = "3.13.0"                                           #pin to avoid yanked version 3.12.4
+ddtrace = "^3.5.1"
 posthog = "^4.2.0"
 limits = "^5.2.0"
 coredis = "^4.22.0"
@@ -63,7 +63,6 @@ openai = "*"
 opencv-python = "*"
 pandas = "*"
 reportlab = "*"
-gevent = ">=24.2.1,<26.0.0"
 
 [tool.poetry-dynamic-versioning]
 enable = true
@@ -86,7 +85,3 @@ lint.pydocstyle.convention = "google"
 [tool.pytest.ini_options]
 asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "function"
-
-[tool.coverage.run]
-relative_files = true
-omit = [ "tests/*" ]

enterprise/server/auth/token_manager.py

@@ -275,7 +275,9 @@ class TokenManager:
                 self._check_expiration_and_refresh
             )
             if not token_info:
-                logger.info(f'No tokens for user: {username}, identity provider: {idp}')
+                logger.error(
+                    f'No tokens for user: {username}, identity provider: {idp}'
+                )
                 raise ValueError(
                     f'No tokens for user: {username}, identity provider: {idp}'
                 )

enterprise/server/routes/billing.py

@@ -17,13 +17,11 @@ from server.constants import (
     STRIPE_API_KEY,
     STRIPE_WEBHOOK_SECRET,
     SUBSCRIPTION_PRICE_DATA,
-    get_default_litellm_model,
 )
 from server.logger import logger
 from storage.billing_session import BillingSession
 from storage.database import session_maker
 from storage.subscription_access import SubscriptionAccess
-from storage.user_settings import UserSettings
 
 from openhands.server.user_auth import get_user_id
 
@@ -44,8 +42,6 @@ class SubscriptionAccessResponse(BaseModel):
     start_at: datetime
     end_at: datetime
     created_at: datetime
-    cancelled_at: datetime | None = None
-    stripe_subscription_id: str | None = None
 
 
 class CreateCheckoutSessionRequest(BaseModel):
@@ -89,7 +85,7 @@ async def get_credits(user_id: str = Depends(get_user_id)) -> GetCreditsResponse
 async def get_subscription_access(
     user_id: str = Depends(get_user_id),
 ) -> SubscriptionAccessResponse | None:
-    """Get details of the currently valid subscription for the user."""
+    """Get details of the currently valid subscription for the user"""
     with session_maker() as session:
         now = datetime.now(UTC)
         subscription_access = (
@@ -106,8 +102,6 @@ async def get_subscription_access(
             start_at=subscription_access.start_at,
             end_at=subscription_access.end_at,
             created_at=subscription_access.created_at,
-            cancelled_at=subscription_access.cancelled_at,
-            stripe_subscription_id=subscription_access.stripe_subscription_id,
         )
 
 
@@ -119,78 +113,6 @@ async def has_payment_method(user_id: str = Depends(get_user_id)) -> bool:
     return await stripe_service.has_payment_method(user_id)
 
 
-# Endpoint to cancel user's subscription
-@billing_router.post('/cancel-subscription')
-async def cancel_subscription(user_id: str = Depends(get_user_id)) -> JSONResponse:
-    """Cancel user's active subscription at the end of the current billing period."""
-    if not user_id:
-        raise HTTPException(status.HTTP_401_UNAUTHORIZED)
-
-    with session_maker() as session:
-        # Find the user's active subscription
-        now = datetime.now(UTC)
-        subscription_access = (
-            session.query(SubscriptionAccess)
-            .filter(SubscriptionAccess.status == 'ACTIVE')
-            .filter(SubscriptionAccess.user_id == user_id)
-            .filter(SubscriptionAccess.start_at <= now)
-            .filter(SubscriptionAccess.end_at >= now)
-            .filter(SubscriptionAccess.cancelled_at.is_(None))  # Not already cancelled
-            .first()
-        )
-
-        if not subscription_access:
-            raise HTTPException(
-                status_code=status.HTTP_404_NOT_FOUND,
-                detail='No active subscription found',
-            )
-
-        if not subscription_access.stripe_subscription_id:
-            raise HTTPException(
-                status_code=status.HTTP_400_BAD_REQUEST,
-                detail='Cannot cancel subscription: missing Stripe subscription ID',
-            )
-
-        try:
-            # Cancel the subscription in Stripe at period end
-            await stripe.Subscription.modify_async(
-                subscription_access.stripe_subscription_id, cancel_at_period_end=True
-            )
-
-            # Update local database
-            subscription_access.cancelled_at = datetime.now(UTC)
-            session.merge(subscription_access)
-            session.commit()
-
-            logger.info(
-                'subscription_cancelled',
-                extra={
-                    'user_id': user_id,
-                    'stripe_subscription_id': subscription_access.stripe_subscription_id,
-                    'subscription_access_id': subscription_access.id,
-                    'end_at': subscription_access.end_at,
-                },
-            )
-
-            return JSONResponse(
-                {'status': 'success', 'message': 'Subscription cancelled successfully'}
-            )
-
-        except stripe.StripeError as e:
-            logger.error(
-                'stripe_cancellation_failed',
-                extra={
-                    'user_id': user_id,
-                    'stripe_subscription_id': subscription_access.stripe_subscription_id,
-                    'error': str(e),
-                },
-            )
-            raise HTTPException(
-                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-                detail=f'Failed to cancel subscription: {str(e)}',
-            )
-
-
 # Endpoint to create a new setup intent in stripe
 @billing_router.post('/create-customer-setup-session')
 async def create_customer_setup_session(
@@ -268,27 +190,9 @@ async def create_subscription_checkout_session(
     billing_session_type: BillingSessionType = BillingSessionType.MONTHLY_SUBSCRIPTION,
     user_id: str = Depends(get_user_id),
 ) -> CreateBillingSessionResponse:
-    # Prevent duplicate subscriptions for the same user
-    with session_maker() as session:
-        now = datetime.now(UTC)
-        existing_active_subscription = (
-            session.query(SubscriptionAccess)
-            .filter(SubscriptionAccess.status == 'ACTIVE')
-            .filter(SubscriptionAccess.user_id == user_id)
-            .filter(SubscriptionAccess.start_at <= now)
-            .filter(SubscriptionAccess.end_at >= now)
-            .filter(SubscriptionAccess.cancelled_at.is_(None))  # Not cancelled
-            .first()
-        )
-
-        if existing_active_subscription:
-            raise HTTPException(
-                status_code=status.HTTP_400_BAD_REQUEST,
-                detail='Cannot create subscription: User already has an active subscription that has not been cancelled',
-            )
-
     customer_id = await stripe_service.find_or_create_customer(user_id)
     subscription_price_data = SUBSCRIPTION_PRICE_DATA[billing_session_type.value]
+    # TODO: Prevent duplicate subscriptions for the same user
     checkout_session = await stripe.checkout.Session.create_async(
         customer=customer_id,
         line_items=[
@@ -342,7 +246,7 @@ async def create_subscription_checkout_session_via_get(
     billing_session_type: BillingSessionType = BillingSessionType.MONTHLY_SUBSCRIPTION,
     user_id: str = Depends(get_user_id),
 ) -> RedirectResponse:
-    """Create a subscription checkout session using a GET request (For easier copy / paste to URL bar)."""
+    """Create a subscription checkout session using a GET request (For easier copy / paste to URL bar)"""
     response = await create_subscription_checkout_session(
         request, billing_session_type, user_id
     )
@@ -374,7 +278,7 @@ async def success_callback(session_id: str, request: Request):
             != BillingSessionType.DIRECT_PAYMENT.value
         ):
             return RedirectResponse(
-                f'{request.base_url}settings?checkout=success', status_code=302
+                f'{request.base_url}settings/billing?checkout=success', status_code=302
             )
 
         stripe_session = stripe.checkout.Session.retrieve(session_id)
@@ -444,29 +348,14 @@ async def cancel_callback(session_id: str, request: Request):
             session.merge(billing_session)
             session.commit()
 
-            # Redirect credit purchases to billing screen, subscriptions to LLM settings
-            if (
-                billing_session.billing_session_type
-                == BillingSessionType.DIRECT_PAYMENT.value
-            ):
-                return RedirectResponse(
-                    f'{request.base_url}settings/billing?checkout=cancel',
-                    status_code=302,
-                )
-            else:
-                return RedirectResponse(
-                    f'{request.base_url}settings?checkout=cancel', status_code=302
-                )
-
-    # If no billing session found, default to LLM settings (subscription flow)
     return RedirectResponse(
-        f'{request.base_url}settings?checkout=cancel', status_code=302
+        f'{request.base_url}settings/billing?checkout=cancel', status_code=302
     )
 
 
 @billing_router.post('/stripe-webhook')
 async def stripe_webhook(request: Request) -> JSONResponse:
-    """Endpoint for stripe webhooks."""
+    """Endpoint for stripe webhooks"""
     payload = await request.body()
     sig_header = request.headers.get('stripe-signature')
 
@@ -508,111 +397,15 @@ async def stripe_webhook(request: Request) -> JSONResponse:
                 end_at=end_at,
                 amount_paid=amount_paid,
                 stripe_invoice_payment_id=invoice.payment_intent,
-                stripe_subscription_id=invoice.subscription,  # Store Stripe subscription ID
             )
             session.add(subscription_access)
             session.commit()
-    elif event_type == 'customer.subscription.updated':
-        subscription = event['data']['object']
-        subscription_id = subscription['id']
-
-        # Handle subscription cancellation
-        if subscription.get('cancel_at_period_end') is True:
-            with session_maker() as session:
-                subscription_access = (
-                    session.query(SubscriptionAccess)
-                    .filter(
-                        SubscriptionAccess.stripe_subscription_id == subscription_id
-                    )
-                    .filter(SubscriptionAccess.status == 'ACTIVE')
-                    .first()
-                )
-
-                if subscription_access and not subscription_access.cancelled_at:
-                    subscription_access.cancelled_at = datetime.now(UTC)
-                    session.merge(subscription_access)
-                    session.commit()
-
-                    logger.info(
-                        'subscription_cancelled_via_webhook',
-                        extra={
-                            'stripe_subscription_id': subscription_id,
-                            'user_id': subscription_access.user_id,
-                            'subscription_access_id': subscription_access.id,
-                        },
-                    )
-    elif event_type == 'customer.subscription.deleted':
-        subscription = event['data']['object']
-        subscription_id = subscription['id']
-
-        with session_maker() as session:
-            subscription_access = (
-                session.query(SubscriptionAccess)
-                .filter(SubscriptionAccess.stripe_subscription_id == subscription_id)
-                .filter(SubscriptionAccess.status == 'ACTIVE')
-                .first()
-            )
-
-            if subscription_access:
-                subscription_access.status = 'DISABLED'
-                subscription_access.updated_at = datetime.now(UTC)
-                session.merge(subscription_access)
-                session.commit()
-
-                # Reset user settings to free tier defaults
-                reset_user_to_free_tier_settings(subscription_access.user_id)
-
-                logger.info(
-                    'subscription_expired_reset_to_free_tier',
-                    extra={
-                        'stripe_subscription_id': subscription_id,
-                        'user_id': subscription_access.user_id,
-                        'subscription_access_id': subscription_access.id,
-                    },
-                )
     else:
         logger.info('stripe_webhook_unhandled_event_type', extra={'type': event_type})
 
     return JSONResponse({'status': 'success'})
 
 
-def reset_user_to_free_tier_settings(user_id: str) -> None:
-    """Reset user settings to free tier defaults when subscription ends."""
-    with session_maker() as session:
-        user_settings = (
-            session.query(UserSettings)
-            .filter(UserSettings.keycloak_user_id == user_id)
-            .first()
-        )
-
-        if user_settings:
-            user_settings.llm_model = get_default_litellm_model()
-            user_settings.llm_api_key = None
-            user_settings.llm_api_key_for_byor = None
-            user_settings.llm_base_url = LITE_LLM_API_URL
-            user_settings.max_budget_per_task = None
-            user_settings.confirmation_mode = False
-            user_settings.enable_solvability_analysis = False
-            user_settings.security_analyzer = 'llm'
-            user_settings.agent = 'CodeActAgent'
-            user_settings.language = 'en'
-            user_settings.enable_default_condenser = True
-            user_settings.enable_sound_notifications = False
-            user_settings.enable_proactive_conversation_starters = True
-            user_settings.user_consents_to_analytics = False
-
-            session.merge(user_settings)
-            session.commit()
-
-            logger.info(
-                'user_settings_reset_to_free_tier',
-                extra={
-                    'user_id': user_id,
-                    'reset_timestamp': datetime.now(UTC).isoformat(),
-                },
-            )
-
-
 async def _get_litellm_user(client: httpx.AsyncClient, user_id: str) -> dict:
     """Get a user from litellm with the id matching that given.
 

enterprise/server/routes/event_webhook.py

@@ -234,7 +234,7 @@ def _get_user_id(conversation_id: str) -> str:
         return conversation_metadata.user_id
 
 
-async def _get_session_api_key(user_id: str, conversation_id: str) -> str | None:
+async def _get_session_api_key(user_id: str, conversation_id: str) -> str:
     agent_loop_info = await conversation_manager.get_agent_loop_info(
         user_id, filter_to_sids={conversation_id}
     )

enterprise/storage/subscription_access.py

@@ -7,7 +7,7 @@ from storage.base import Base
 class SubscriptionAccess(Base):  # type: ignore
     """
     Represents a user's subscription access record.
-    Tracks subscription status, duration, payment information, and cancellation status.
+    Tracks subscription status, duration, and payment information.
     """
 
     __tablename__ = 'subscription_access'
@@ -27,8 +27,6 @@ class SubscriptionAccess(Base):  # type: ignore
     end_at = Column(DateTime(timezone=True), nullable=True)
     amount_paid = Column(DECIMAL(19, 4), nullable=True)
     stripe_invoice_payment_id = Column(String, nullable=False)
-    cancelled_at = Column(DateTime(timezone=True), nullable=True)
-    stripe_subscription_id = Column(String, nullable=True, index=True)
     created_at = Column(
         DateTime(timezone=True),
         default=lambda: datetime.now(UTC),  # type: ignore[attr-defined]

enterprise/sync/install_gitlab_webhooks.py

@@ -276,12 +276,12 @@ class VerifyWebhookStatus:
                     webhook
                 )
 
-                gitlab_service_impl = GitLabServiceImpl(external_auth_id=user_id)
+                gitlab_service = GitLabServiceImpl(external_auth_id=user_id)
 
-                if not isinstance(gitlab_service_impl, SaaSGitLabService):
+                if not isinstance(gitlab_service, SaaSGitLabService):
                     raise Exception('Only SaaSGitLabService is supported')
                 # Cast needed when mypy can see OpenHands
-                gitlab_service = cast(type[SaaSGitLabService], gitlab_service_impl)
+                gitlab_service = cast(type[SaaSGitLabService], gitlab_service)
 
                 await self.verify_conditions_are_met(
                     gitlab_service=gitlab_service,

enterprise/tests/unit/integrations/test_utils.py

@@ -1,159 +0,0 @@
-"""Tests for enterprise integrations utils module."""
-
-import pytest
-from integrations.utils import get_summary_for_agent_state
-
-from openhands.core.schema.agent import AgentState
-from openhands.events.observation.agent import AgentStateChangedObservation
-
-
-class TestGetSummaryForAgentState:
-    """Test cases for get_summary_for_agent_state function."""
-
-    def setup_method(self):
-        """Set up test fixtures."""
-        self.conversation_link = 'https://example.com/conversation/123'
-
-    def test_empty_observations_list(self):
-        """Test handling of empty observations list."""
-        result = get_summary_for_agent_state([], self.conversation_link)
-
-        assert 'unknown error' in result.lower()
-        assert self.conversation_link in result
-
-    @pytest.mark.parametrize(
-        'state,expected_text,includes_link',
-        [
-            (AgentState.RATE_LIMITED, 'rate limited', False),
-            (AgentState.AWAITING_USER_INPUT, 'waiting for your input', True),
-        ],
-    )
-    def test_handled_agent_states(self, state, expected_text, includes_link):
-        """Test handling of states with specific behavior."""
-        observation = AgentStateChangedObservation(
-            content=f'Agent state: {state.value}', agent_state=state
-        )
-
-        result = get_summary_for_agent_state([observation], self.conversation_link)
-
-        assert expected_text in result.lower()
-        if includes_link:
-            assert self.conversation_link in result
-        else:
-            assert self.conversation_link not in result
-
-    @pytest.mark.parametrize(
-        'state',
-        [
-            AgentState.FINISHED,
-            AgentState.PAUSED,
-            AgentState.STOPPED,
-            AgentState.AWAITING_USER_CONFIRMATION,
-        ],
-    )
-    def test_unhandled_agent_states(self, state):
-        """Test handling of unhandled states (should all return unknown error)."""
-        observation = AgentStateChangedObservation(
-            content=f'Agent state: {state.value}', agent_state=state
-        )
-
-        result = get_summary_for_agent_state([observation], self.conversation_link)
-
-        assert 'unknown error' in result.lower()
-        assert self.conversation_link in result
-
-    @pytest.mark.parametrize(
-        'error_code,expected_text',
-        [
-            (
-                'STATUS$ERROR_LLM_AUTHENTICATION',
-                'authentication with the llm provider failed',
-            ),
-            (
-                'STATUS$ERROR_LLM_SERVICE_UNAVAILABLE',
-                'llm service is temporarily unavailable',
-            ),
-            (
-                'STATUS$ERROR_LLM_INTERNAL_SERVER_ERROR',
-                'llm provider encountered an internal error',
-            ),
-            ('STATUS$ERROR_LLM_OUT_OF_CREDITS', "you've run out of credits"),
-            ('STATUS$ERROR_LLM_CONTENT_POLICY_VIOLATION', 'content policy violation'),
-        ],
-    )
-    def test_error_state_readable_reasons(self, error_code, expected_text):
-        """Test all readable error reason mappings."""
-        observation = AgentStateChangedObservation(
-            content=f'Agent encountered error: {error_code}',
-            agent_state=AgentState.ERROR,
-            reason=error_code,
-        )
-
-        result = get_summary_for_agent_state([observation], self.conversation_link)
-
-        assert 'encountered an error' in result.lower()
-        assert expected_text in result.lower()
-        assert self.conversation_link in result
-
-    def test_error_state_with_custom_reason(self):
-        """Test handling of ERROR state with a custom reason."""
-        observation = AgentStateChangedObservation(
-            content='Agent encountered an error',
-            agent_state=AgentState.ERROR,
-            reason='Test error message',
-        )
-
-        result = get_summary_for_agent_state([observation], self.conversation_link)
-
-        assert 'encountered an error' in result.lower()
-        assert 'test error message' in result.lower()
-        assert self.conversation_link in result
-
-    def test_multiple_observations_uses_first(self):
-        """Test that when multiple observations are provided, only the first is used."""
-        observation1 = AgentStateChangedObservation(
-            content='Agent is awaiting user input',
-            agent_state=AgentState.AWAITING_USER_INPUT,
-        )
-        observation2 = AgentStateChangedObservation(
-            content='Agent encountered an error',
-            agent_state=AgentState.ERROR,
-            reason='Should not be used',
-        )
-
-        result = get_summary_for_agent_state(
-            [observation1, observation2], self.conversation_link
-        )
-
-        # Should handle the first observation (AWAITING_USER_INPUT), not the second (ERROR)
-        assert 'waiting for your input' in result.lower()
-        assert 'error' not in result.lower()
-
-    def test_awaiting_user_input_specific_message(self):
-        """Test that AWAITING_USER_INPUT returns the specific expected message."""
-        observation = AgentStateChangedObservation(
-            content='Agent is awaiting user input',
-            agent_state=AgentState.AWAITING_USER_INPUT,
-        )
-
-        result = get_summary_for_agent_state([observation], self.conversation_link)
-
-        # Test the exact message format
-        assert 'waiting for your input' in result.lower()
-        assert 'continue the conversation' in result.lower()
-        assert self.conversation_link in result
-        assert 'unknown error' not in result.lower()
-
-    def test_rate_limited_specific_message(self):
-        """Test that RATE_LIMITED returns the specific expected message."""
-        observation = AgentStateChangedObservation(
-            content='Agent was rate limited', agent_state=AgentState.RATE_LIMITED
-        )
-
-        result = get_summary_for_agent_state([observation], self.conversation_link)
-
-        # Test the exact message format
-        assert 'rate limited' in result.lower()
-        assert 'try again later' in result.lower()
-        # RATE_LIMITED doesn't include conversation link in response
-        assert self.conversation_link not in result

enterprise/tests/unit/test_billing.py

@@ -5,16 +5,16 @@ import pytest
 import stripe
 from fastapi import HTTPException, Request, status
 from httpx import HTTPStatusError, Response
-from integrations.stripe_service import has_payment_method
+from server.routes import billing
 from server.routes.billing import (
     CreateBillingSessionResponse,
     CreateCheckoutSessionRequest,
     GetCreditsResponse,
     cancel_callback,
-    cancel_subscription,
     create_checkout_session,
-    create_subscription_checkout_session,
+    create_customer_setup_session,
     get_credits,
+    has_payment_method,
     success_callback,
 )
 from sqlalchemy import create_engine
@@ -362,7 +362,8 @@ async def test_cancel_callback_session_not_found():
         response = await cancel_callback('test_session_id', mock_request)
         assert response.status_code == 302
         assert (
-            response.headers['location'] == 'http://test.com/settings?checkout=cancel'
+            response.headers['location']
+            == 'http://test.com/settings/billing?checkout=cancel'
         )
 
         # Verify no database updates occurred
@@ -388,7 +389,8 @@ async def test_cancel_callback_success():
 
         assert response.status_code == 302
         assert (
-            response.headers['location'] == 'http://test.com/settings?checkout=cancel'
+            response.headers['location']
+            == 'http://test.com/settings/billing?checkout=cancel'
         )
 
         # Verify database updates
@@ -400,312 +402,51 @@ async def test_cancel_callback_success():
 @pytest.mark.asyncio
 async def test_has_payment_method_with_payment_method():
     """Test has_payment_method returns True when user has a payment method."""
-    with (
-        patch('integrations.stripe_service.session_maker') as mock_session_maker,
-        patch(
-            'stripe.Customer.list_payment_methods_async',
-            AsyncMock(return_value=MagicMock(data=[MagicMock()])),
-        ) as mock_list_payment_methods,
-    ):
-        # Setup mock session
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.first.return_value = (
-            MagicMock(stripe_customer_id='cus_test123')
-        )
 
+    mock_has_payment_method = AsyncMock(return_value=True)
+    with patch(
+        'integrations.stripe_service.has_payment_method', mock_has_payment_method
+    ):
         result = await has_payment_method('mock_user')
         assert result is True
-        mock_list_payment_methods.assert_called_once_with('cus_test123')
+    mock_has_payment_method.assert_called_once_with('mock_user')
 
 
 @pytest.mark.asyncio
 async def test_has_payment_method_without_payment_method():
     """Test has_payment_method returns False when user has no payment method."""
-    with (
-        patch('integrations.stripe_service.session_maker') as mock_session_maker,
-        patch(
-            'stripe.Customer.list_payment_methods_async',
-            AsyncMock(return_value=MagicMock(data=[])),
-        ) as mock_list_payment_methods,
+    mock_has_payment_method = AsyncMock(return_value=False)
+    with patch(
+        'integrations.stripe_service.has_payment_method', mock_has_payment_method
     ):
-        # Setup mock session
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.first.return_value = (
-            MagicMock(stripe_customer_id='cus_test123')
-        )
-
+        mock_has_payment_method.return_value = False
         result = await has_payment_method('mock_user')
         assert result is False
-        mock_list_payment_methods.assert_called_once_with('cus_test123')
+    mock_has_payment_method.assert_called_once_with('mock_user')
 
 
 @pytest.mark.asyncio
-async def test_cancel_subscription_success():
-    """Test successful subscription cancellation."""
-    from datetime import UTC, datetime
-
-    from storage.subscription_access import SubscriptionAccess
-
-    # Mock active subscription
-    mock_subscription_access = SubscriptionAccess(
-        id=1,
-        status='ACTIVE',
-        user_id='test_user',
-        start_at=datetime.now(UTC),
-        end_at=datetime.now(UTC),
-        amount_paid=2000,
-        stripe_invoice_payment_id='pi_test',
-        stripe_subscription_id='sub_test123',
-        cancelled_at=None,
+async def test_create_customer_setup_session_success():
+    """Test successful creation of customer setup session."""
+    mock_request = Request(
+        scope={'type': 'http', 'state': {'user_id': 'mock_user'}, 'headers': []}
     )
 
-    # Mock Stripe subscription response
-    mock_stripe_subscription = MagicMock()
-    mock_stripe_subscription.cancel_at_period_end = True
-
-    with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
-        patch(
-            'stripe.Subscription.modify_async',
-            AsyncMock(return_value=mock_stripe_subscription),
-        ) as mock_stripe_modify,
-    ):
-        # Setup mock session
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = mock_subscription_access
-
-        # Call the function
-        result = await cancel_subscription('test_user')
-
-        # Verify Stripe API was called
-        mock_stripe_modify.assert_called_once_with(
-            'sub_test123', cancel_at_period_end=True
-        )
-
-        # Verify database was updated
-        assert mock_subscription_access.cancelled_at is not None
-        mock_session.merge.assert_called_once_with(mock_subscription_access)
-        mock_session.commit.assert_called_once()
-
-        # Verify response
-        assert result.status_code == 200
-
-
-@pytest.mark.asyncio
-async def test_cancel_subscription_no_active_subscription():
-    """Test cancellation when no active subscription exists."""
-    with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
-    ):
-        # Setup mock session with no subscription found
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = None
-
-        # Call the function and expect HTTPException
-        with pytest.raises(HTTPException) as exc_info:
-            await cancel_subscription('test_user')
-
-        assert exc_info.value.status_code == 404
-        assert 'No active subscription found' in str(exc_info.value.detail)
-
-
-@pytest.mark.asyncio
-async def test_cancel_subscription_missing_stripe_id():
-    """Test cancellation when subscription has no Stripe ID."""
-    from datetime import UTC, datetime
-
-    from storage.subscription_access import SubscriptionAccess
-
-    # Mock subscription without Stripe ID
-    mock_subscription_access = SubscriptionAccess(
-        id=1,
-        status='ACTIVE',
-        user_id='test_user',
-        start_at=datetime.now(UTC),
-        end_at=datetime.now(UTC),
-        amount_paid=2000,
-        stripe_invoice_payment_id='pi_test',
-        stripe_subscription_id=None,  # Missing Stripe ID
-        cancelled_at=None,
+    mock_customer = stripe.Customer(
+        id='mock-customer', metadata={'user_id': 'mock-user'}
     )
+    mock_session = MagicMock()
+    mock_session.url = 'https://checkout.stripe.com/test-session'
+    mock_create = AsyncMock(return_value=mock_session)
 
     with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
-    ):
-        # Setup mock session
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = mock_subscription_access
-
-        # Call the function and expect HTTPException
-        with pytest.raises(HTTPException) as exc_info:
-            await cancel_subscription('test_user')
-
-        assert exc_info.value.status_code == 400
-        assert 'missing Stripe subscription ID' in str(exc_info.value.detail)
-
-
-@pytest.mark.asyncio
-async def test_cancel_subscription_stripe_error():
-    """Test cancellation when Stripe API fails."""
-    from datetime import UTC, datetime
-
-    from storage.subscription_access import SubscriptionAccess
-
-    # Mock active subscription
-    mock_subscription_access = SubscriptionAccess(
-        id=1,
-        status='ACTIVE',
-        user_id='test_user',
-        start_at=datetime.now(UTC),
-        end_at=datetime.now(UTC),
-        amount_paid=2000,
-        stripe_invoice_payment_id='pi_test',
-        stripe_subscription_id='sub_test123',
-        cancelled_at=None,
-    )
-
-    with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
-        patch(
-            'stripe.Subscription.modify_async',
-            AsyncMock(side_effect=stripe.StripeError('API Error')),
-        ),
-    ):
-        # Setup mock session
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = mock_subscription_access
-
-        # Call the function and expect HTTPException
-        with pytest.raises(HTTPException) as exc_info:
-            await cancel_subscription('test_user')
-
-        assert exc_info.value.status_code == 500
-        assert 'Failed to cancel subscription' in str(exc_info.value.detail)
-
-
-@pytest.mark.asyncio
-async def test_create_subscription_checkout_session_duplicate_prevention():
-    """Test that creating a subscription when user already has active subscription raises error."""
-    from datetime import UTC, datetime
-
-    from storage.subscription_access import SubscriptionAccess
-
-    # Mock active subscription
-    mock_subscription_access = SubscriptionAccess(
-        id=1,
-        status='ACTIVE',
-        user_id='test_user',
-        start_at=datetime.now(UTC),
-        end_at=datetime.now(UTC),
-        amount_paid=2000,
-        stripe_invoice_payment_id='pi_test',
-        stripe_subscription_id='sub_test123',
-        cancelled_at=None,
-    )
-
-    mock_request = Request(scope={'type': 'http'})
-    mock_request._base_url = URL('http://test.com/')
-
-    with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
-    ):
-        # Setup mock session to return existing active subscription
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = mock_subscription_access
-
-        # Call the function and expect HTTPException
-        with pytest.raises(HTTPException) as exc_info:
-            await create_subscription_checkout_session(
-                mock_request, user_id='test_user'
-            )
-
-        assert exc_info.value.status_code == 400
-        assert (
-            'user already has an active subscription'
-            in str(exc_info.value.detail).lower()
-        )
-
-
-@pytest.mark.asyncio
-async def test_create_subscription_checkout_session_allows_after_cancellation():
-    """Test that creating a subscription is allowed when previous subscription was cancelled."""
-    mock_request = Request(scope={'type': 'http'})
-    mock_request._base_url = URL('http://test.com/')
-
-    mock_session_obj = MagicMock()
-    mock_session_obj.url = 'https://checkout.stripe.com/test-session'
-    mock_session_obj.id = 'test_session_id'
-
-    with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
         patch(
             'integrations.stripe_service.find_or_create_customer',
-            AsyncMock(return_value='cus_test123'),
-        ),
-        patch(
-            'stripe.checkout.Session.create_async',
-            AsyncMock(return_value=mock_session_obj),
-        ),
-        patch(
-            'server.routes.billing.SUBSCRIPTION_PRICE_DATA',
-            {'MONTHLY_SUBSCRIPTION': {'unit_amount': 2000}},
+            AsyncMock(return_value=mock_customer),
         ),
+        patch('stripe.checkout.Session.create_async', mock_create),
     ):
-        # Setup mock session - the query should return None because cancelled subscriptions are filtered out
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = None
+        result = await create_customer_setup_session(mock_request)
 
-        # Should succeed
-        result = await create_subscription_checkout_session(
-            mock_request, user_id='test_user'
-        )
-
-        assert isinstance(result, CreateBillingSessionResponse)
-        assert result.redirect_url == 'https://checkout.stripe.com/test-session'
-
-
-@pytest.mark.asyncio
-async def test_create_subscription_checkout_session_success_no_existing():
-    """Test successful subscription creation when no existing subscription."""
-    mock_request = Request(scope={'type': 'http'})
-    mock_request._base_url = URL('http://test.com/')
-
-    mock_session_obj = MagicMock()
-    mock_session_obj.url = 'https://checkout.stripe.com/test-session'
-    mock_session_obj.id = 'test_session_id'
-
-    with (
-        patch('server.routes.billing.session_maker') as mock_session_maker,
-        patch(
-            'integrations.stripe_service.find_or_create_customer',
-            AsyncMock(return_value='cus_test123'),
-        ),
-        patch(
-            'stripe.checkout.Session.create_async',
-            AsyncMock(return_value=mock_session_obj),
-        ),
-        patch(
-            'server.routes.billing.SUBSCRIPTION_PRICE_DATA',
-            {'MONTHLY_SUBSCRIPTION': {'unit_amount': 2000}},
-        ),
-    ):
-        # Setup mock session to return no existing subscription
-        mock_session = MagicMock()
-        mock_session_maker.return_value.__enter__.return_value = mock_session
-        mock_session.query.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.filter.return_value.first.return_value = None
-
-        # Should succeed
-        result = await create_subscription_checkout_session(
-            mock_request, user_id='test_user'
-        )
-
-        assert isinstance(result, CreateBillingSessionResponse)
+        assert isinstance(result, billing.CreateBillingSessionResponse)
         assert result.redirect_url == 'https://checkout.stripe.com/test-session'

evaluation/benchmarks/gaia/run_infer.py

@@ -28,7 +28,6 @@ from evaluation.utils.shared import (
     prepare_dataset,
     reset_logger_for_multiprocessing,
     run_evaluation,
-    update_llm_config_for_completions_logging,
 )
 from openhands.controller.state.state import State
 from openhands.core.config import (
@@ -37,11 +36,7 @@ from openhands.core.config import (
     get_llm_config_arg,
     load_from_toml,
 )
-from openhands.core.config.utils import (
-    get_agent_config_arg,
-    get_llms_for_routing_config,
-    get_model_routing_config_arg,
-)
+from openhands.core.config.utils import get_agent_config_arg
 from openhands.core.logger import openhands_logger as logger
 from openhands.core.main import create_runtime, run_controller
 from openhands.events.action import AgentFinishAction, CmdRunAction, MessageAction
@@ -62,7 +57,6 @@ AGENT_CLS_TO_INST_SUFFIX = {
 
 
 def get_config(
-    instance: pd.Series,
     metadata: EvalMetadata,
 ) -> OpenHandsConfig:
     sandbox_config = get_default_sandbox_config_for_eval()
@@ -72,24 +66,13 @@ def get_config(
         sandbox_config=sandbox_config,
         runtime='docker',
     )
-    config.set_llm_config(
-        update_llm_config_for_completions_logging(
-            metadata.llm_config, metadata.eval_output_dir, instance['instance_id']
-        )
-    )
-    model_routing_config = get_model_routing_config_arg()
-    model_routing_config.llms_for_routing = (
-        get_llms_for_routing_config()
-    )  # Populate with LLMs for routing from config.toml file
-
+    config.set_llm_config(metadata.llm_config)
     if metadata.agent_config:
-        metadata.agent_config.model_routing = model_routing_config
         config.set_agent_config(metadata.agent_config, metadata.agent_class)
     else:
         logger.info('Agent config not provided, using default settings')
         agent_config = config.get_agent_config(metadata.agent_class)
         agent_config.enable_prompt_extensions = False
-        agent_config.model_routing = model_routing_config
 
     config_copy = copy.deepcopy(config)
     load_from_toml(config_copy)
@@ -162,7 +145,7 @@ def process_instance(
     metadata: EvalMetadata,
     reset_logger: bool = True,
 ) -> EvalOutput:
-    config = get_config(instance, metadata)
+    config = get_config(metadata)
 
     # Setup the logger properly, so you can run multi-processing to parallelize the evaluation
     if reset_logger:

evaluation/benchmarks/multi_swe_bench/SWE-Gym.md

@@ -1,152 +0,0 @@
-<h1 align="center"> Training Software Engineering Agents and Verifiers with SWE-Gym </h1>
-
-A Multi-SWE-bench implementation of SWE-Gym.
-
-<p align="center">
-  <a href="https://www.jiayipan.com/" style="text-decoration: none;">Jiayi Pan<sup>*,1</sup></a>,
-  <a href="https://xwang.dev/" style="text-decoration: none;">Xingyao Wang<sup>*,2</sup></a>,
-  <a href="https://www.phontron.com/" style="text-decoration: none;">Graham Neubig<sup>3</sup></a>,
-  <a href="https://www.cs.toronto.edu/~ndjaitly/" style="text-decoration: none;">Navdeep Jaitly<sup>4</sup></a>,
-  <a href="https://blender.cs.illinois.edu/hengji.html" style="text-decoration: none;">Heng Ji<sup>2</sup></a>,
-  <a href="https://www.alanesuhr.com/" style="text-decoration: none;">Alane Suhr<sup>^,1</sup></a>,
-  <a href="https://dreasysnail.github.io/" style="text-decoration: none;">Yizhe Zhang<sup>^,4</sup></a>
-</p>
-
-<p align="center">
-  <sup>1</sup>UC Berkeley, <sup>2</sup>UIUC, <sup>3</sup>CMU, <sup>4</sup>Apple </br>
-  <sub><sup>*</sup>Equal contribution, <sup>^</sup>Equal supervision</sub>
-</p>
-
-<p align="center">
-<a href="https://arxiv.org/abs/2412.21139">📃 Paper</a>
-•
-<a href="https://huggingface.co/SWE-Gym" >🤗 Data & Models</a>
-</p>
-
-We present **SWE-Gym**, the first environment for training real-world software engineering agents.
-We use it to train strong LM agents that achieve state-of-the-art open results on SWE-Bench, with early, promising scaling characteristics as we increase training and inference-time compute.
-
-<p align="center">
-  <img src="https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/teaser.jpg?raw=true" width="100%" alt="teaser">
-</p>
-
----
-# Run SWE-Gym with OpenHands
-
-The process of running SWE-Gym is very similar to how you'd run SWE-Bench evaluation.
-
-
-1. First, clone OpenHands repo `git clone https://github.com/All-Hands-AI/OpenHands.git`
-2. Then setup the repo following [Development.md](https://github.com/All-Hands-AI/OpenHands/blob/main/Development.md)
-3. Then you can simply serve your own model as an OpenAI compatible endpoint, put those info in config.toml. You can do this by following instruction [here](../../README.md#setup).
-4. And then simply do the following to sample for 16x parallelism:
-
-```bash
-export ALLHANDS_API_KEY=ah-yourkey  # You don't need to set this when running these in local docker container
-./evaluation/benchmarks/multi_swe_bench/scripts/rollout_swegym.sh llm.mymodel-temp05 'train-t05' 16
-```
-
-NOTE: SWE-Gym sampling with parallelism is currently only tested with AllHands RemoteRuntime (limited beta). Fill [this form](https://docs.google.com/forms/d/e/1FAIpQLSckVz_JFwg2_mOxNZjCtr7aoBFI2Mwdan3f75J_TrdMS1JV2g/viewform) to apply for access.
-
-
-5. When `rollout_swegym.sh` finishes, you will get a file called `output.with_completions.jsonl.gz`. Then you can use [`./scripts/swegym/convert_data.ipynb`](./scripts/swegym/convert_data.ipynb) to convert them into SFT data format.
-
-## Running the Jupyter Notebook
-
-To run the data conversion notebook, follow these steps:
-
-1. Navigate to the OpenHands repository root:
-```bash
-cd openhands_repo
-```
-
-2. Set the PYTHONPATH and start Jupyter notebook:
-```bash
-PYTHONPATH=$(pwd) jupyter notebook
-```
-
-3. In the Jupyter interface, navigate to `evaluation/benchmarks/swe_bench/scripts/swegym/convert_data.ipynb`
-
-4. Update the file paths in the notebook:
-   - Set `FILE_PATHS` to point to your `output.with_completions.jsonl.gz` files
-   - Set `YOUR_OUTPUT_FOLDER` to your desired output directory
-
-5. Run the notebook cells sequentially to process your data and generate the SFT training format.
-
----
-# More info about SWE-Gym
-
-Progress in agents for software engineering has been limited by the lack of training environments that both include rigorous verification for reinforcement learning and cover the expansive tasks encountered in real-world repository-level engineering.
-
-We introduce SWE-Gym: An Open Environment for Training Software Engineering Agents & Verifiers.
-Our baselines achieve new open SOTA - 32%/26% on SWE-Bench Verified/Lite, with promising scaling trends.
-
-![SWE-Gym Scaling](https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/scaling.jpg?raw=true)
-*SWE-Gym enables scalable improvements for software engineering agents at both training and inference time. Our current results is primarily bottlenecked by training and inference compute, rather than the size of our environment.*
-
-## SWE-Gym Environment
-
-We create SWE-Gym, the first environment for training SWE agents, with **2.4K real tasks from 11 Python repos** & a Lite split of 234 instances. SWE-Gym combines real-world Python tasks, repository context, executable environments, and test verification to train agents for solving software engineering problems.
-
-![SWE-Gym Repo Distribution](https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/swe-gym.jpg?raw=true)
-
-
-## SWE-Gym trains LMs as agents
-
-When fine-tuned on less than 500 agent-environment interaction trajectories sampled from it from GPT-4o and Claude 3.5 Sonnet, we achieve **+14%** absolute gains on SWE-Bench Verified with an 32B LM-powered OpenHands agent.
-
-![OpenHands Performance diff before and after training](https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/oh-agent.jpg?raw=true)
-
-
-## SWE-Gym enables self-improvement
-
-SWE-Gym is also effective across agent scaffolds. With rejection sampling fine-tuning and MoatlessTools scaffold, our 32B and 7B models achieve 20% and 10% respectively on SWE-Bench Lite through self-improvement.
-
-<p align="center">
-  <img src="https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/ml-agent.jpg?raw=true" width="80%" alt="Moatless self-improvement">
-</p>
-
-
-
-## SWE-Gym enables inference-time scaling
-
-SWE-Gym enables inference-time scaling through verifiers trained on agent trajectories.
-These verifiers identify most promising solutions via best-of-n selection, together with our learned agents, they achieve 32%/26% on SWE-Bench Verified/Lite, a new open SoTA.
-
-
-![Inference Time Scaling for Moatless Agent](https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/inference-ml.jpg?raw=true)
-*Inference Time Scaling for Moatless Agent*
-
-![Inference Time Scaling for OpenHands Agent](https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/inference-oh.jpg?raw=true)
-*Inference Time Scaling for OpenHands Agent*
-
-
-## Our baselines on SWE-Gym shows strong scaling trends
-
-Lastly, our ablations reveal strong scaling trends - performance is now bottlenecked by train and inference compute, rather than the size of our dataset. Pushing and improving these scaling trends further is an exciting direction for future work.
-
-![](https://github.com/SWE-Gym/SWE-Gym/blob/main/assets/images/scaling.jpg?raw=true)
-
-## Reproducing Results
-**The Dataset**
-
-To access SWE-Gym dataset, checkout our huggingface hub page [SWE-Gym](https://huggingface.co/SWE-Gym)
-
-The environment constants are currently saved at [SWE-Bench-Fork](https://github.com/SWE-Gym/SWE-Bench-Fork)
-
-We also have pre-built docker images for each instance under [xingyaoww/sweb.eval.x86_64](https://hub.docker.com/search?q=xingyaoww%2Fsweb.eval.x86_64.) prefix at docker hub.
-
-
-## 📚 Citation
-
-```bibtex
-@misc{pan2024trainingsoftwareengineeringagents,
-      title={Training Software Engineering Agents and Verifiers with SWE-Gym},
-      author={Jiayi Pan and Xingyao Wang and Graham Neubig and Navdeep Jaitly and Heng Ji and Alane Suhr and Yizhe Zhang},
-      year={2024},
-      eprint={2412.21139},
-      archivePrefix={arXiv},
-      primaryClass={cs.SE},
-      url={https://arxiv.org/abs/2412.21139},
-}
-```

evaluation/benchmarks/multi_swe_bench/run_infer.py

@@ -51,8 +51,8 @@ RUN_WITH_BROWSING = os.environ.get('RUN_WITH_BROWSING', 'false').lower() == 'tru
 
 # TODO: migrate all swe-bench docker to ghcr.io/openhands
 # TODO: 适应所有的语言
-DOCKER_IMAGE_PREFIX = os.environ.get('EVAL_DOCKER_IMAGE_PREFIX', 'mswebench')
-LANGUAGE = os.environ.get('LANGUAGE', 'java')
+DOCKER_IMAGE_PREFIX = os.environ.get('EVAL_DOCKER_IMAGE_PREFIX', '')
+LANGUAGE = os.environ.get('LANGUAGE', 'python')
 logger.info(f'Using docker image prefix: {DOCKER_IMAGE_PREFIX}')
 
 
@@ -305,19 +305,31 @@ def get_instance_docker_image(instance: pd.Series):
         instance_id = instance.get('instance_id', '')
         tag_suffix = instance_id.split('-')[-1] if instance_id else ''
         container_tag = f'pr-{tag_suffix}'
-        return f'{DOCKER_IMAGE_PREFIX}/{container_name}:{container_tag}'
+        # pdb.set_trace()
+        return f'mswebench/{container_name}:{container_tag}'
+        # return "kong/insomnia:pr-8284"
+        # return "'sweb.eval.x86_64.local_insomnia"
+        # return "local_insomnia_why"
+        # return "local/kong-insomnia:pr-8117"
 
 
 def get_config(
     instance: pd.Series,
     metadata: EvalMetadata,
 ) -> OpenHandsConfig:
-    base_container_image = get_instance_docker_image(instance)
-    logger.info(
-        f'Using instance container image: {base_container_image}. '
-        f'Please make sure this image exists. '
-        f'Submit an issue on https://github.com/All-Hands-AI/OpenHands if you run into any issues.'
-    )
+    SWE_BENCH_CONTAINER_IMAGE = 'ghcr.io/opendevin/eval-swe-bench:full-v1.2.1'
+    if USE_INSTANCE_IMAGE:
+        # We use a different instance image for the each instance of swe-bench eval
+        # base_container_image = get_instance_docker_image(instance['instance_id'])
+        base_container_image = get_instance_docker_image(instance)
+        logger.info(
+            f'Using instance container image: {base_container_image}. '
+            f'Please make sure this image exists. '
+            f'Submit an issue on https://github.com/All-Hands-AI/OpenHands if you run into any issues.'
+        )
+    else:
+        base_container_image = SWE_BENCH_CONTAINER_IMAGE
+        logger.info(f'Using swe-bench container image: {base_container_image}')
 
     sandbox_config = get_default_sandbox_config_for_eval()
     sandbox_config.base_container_image = base_container_image
@@ -760,6 +772,7 @@ if __name__ == '__main__':
     parser.add_argument(
         '--dataset',
         type=str,
+        default='princeton-nlp/SWE-bench',
         help='data set to evaluate on, either full-test or lite-test',
     )
     parser.add_argument(
@@ -774,7 +787,6 @@ if __name__ == '__main__':
     # so we don't need to manage file uploading to OpenHands's repo
     # dataset = load_dataset(args.dataset, split=args.split)
     # dataset = load_dataset(args.dataset)
-    logger.info(f'Loading dataset {args.dataset} with split {args.split} ')
     dataset = load_dataset('json', data_files=args.dataset)
     dataset = dataset[args.split]
     swe_bench_tests = filter_dataset(dataset.to_pandas(), 'instance_id')
@@ -827,7 +839,7 @@ if __name__ == '__main__':
         args.eval_num_workers,
         process_instance,
         timeout_seconds=120 * 60,  # 2 hour PER instance should be more than enough
-        max_retries=3,
+        max_retries=5,
     )
     # Check if any instances reached maximum retries
     check_maximum_retries_exceeded(metadata.eval_output_dir)

evaluation/benchmarks/multi_swe_bench/scripts/data/data_change.py

@@ -1,54 +1,37 @@
-import argparse
 import json
 
+input_file = 'XXX.jsonl'
+output_file = 'YYY.jsonl'
 
-def main(input_file, output_file):
-    with (
-        open(input_file, 'r', encoding='utf-8') as fin,
-        open(output_file, 'w', encoding='utf-8') as fout,
-    ):
-        for line in fin:
-            line = line.strip()
-            if not line:
-                continue
+with (
+    open(input_file, 'r', encoding='utf-8') as fin,
+    open(output_file, 'w', encoding='utf-8') as fout,
+):
+    for line in fin:
+        line = line.strip()
+        if not line:
+            continue
 
-            data = json.loads(line)
-            item = data
+        data = json.loads(line)
+        item = data
 
-            # Skip instances that don't have resolved_issues or have empty resolved_issues
-            if not item.get('resolved_issues') or len(item['resolved_issues']) == 0:
-                print(
-                    f'Skipping instance {item.get("org", "")}/{item.get("repo", "")}-{item.get("number", "")} - no resolved_issues'
-                )
-                continue
+        # 提取原始数据
+        org = item.get('org', '')
+        repo = item.get('repo', '')
+        number = str(item.get('number', ''))
 
-            # 提取原始数据
-            org = item.get('org', '')
-            repo = item.get('repo', '')
-            number = str(item.get('number', ''))
+        new_item = {}
+        new_item['repo'] = f'{org}/{repo}'
+        new_item['instance_id'] = f'{org}__{repo}-{number}'
+        new_item['problem_statement'] = (
+            item['resolved_issues'][0].get('title', '')
+            + '\n'
+            + item['resolved_issues'][0].get('body', '')
+        )
+        new_item['FAIL_TO_PASS'] = []
+        new_item['PASS_TO_PASS'] = []
+        new_item['base_commit'] = item['base'].get('sha', '')
+        new_item['version'] = '0.1'  # depends
 
-            new_item = {}
-            new_item['repo'] = f'{org}/{repo}'
-            new_item['instance_id'] = f'{org}__{repo}-{number}'
-
-            # Get the first resolved issue
-            resolved_issue = item['resolved_issues'][0]
-            title = resolved_issue.get('title') or ''
-            body = resolved_issue.get('body') or ''
-
-            new_item['problem_statement'] = title + '\n' + body
-            new_item['FAIL_TO_PASS'] = []
-            new_item['PASS_TO_PASS'] = []
-            new_item['base_commit'] = item['base'].get('sha', '')
-            new_item['version'] = '0.1'  # depends
-
-            output_data = new_item
-            fout.write(json.dumps(output_data, ensure_ascii=False) + '\n')
-
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--input', required=True, help='Input .jsonl file path')
-    parser.add_argument('--output', required=True, help='Output .jsonl file path')
-    args = parser.parse_args()
-    main(args.input, args.output)
+        output_data = new_item
+        fout.write(json.dumps(output_data, ensure_ascii=False) + '\n')

evaluation/benchmarks/multi_swe_bench/scripts/eval/combine_final_completions.py

@@ -1,69 +0,0 @@
-import argparse
-import gzip
-import json
-import os
-from glob import glob
-
-from tqdm import tqdm
-
-tqdm.pandas()
-
-
-# Load trajectories for resolved instances
-def load_completions(output_dir: str, instance_id: str):
-    glob_path = os.path.join(output_dir, 'llm_completions', instance_id, '*.json')
-    files = sorted(glob(glob_path))  # this is ascending order
-    # pick the last file (last turn)
-    try:
-        file_path = files[-1]
-    except IndexError:
-        # print(f'No files found for instance {instance_id}: files={files}')
-        return None
-    with open(file_path, 'r') as f:
-        result = json.load(f)
-    # create messages
-    messages = result['messages']
-    messages.append(result['response']['choices'][0]['message'])
-    tools = result['kwargs'].get('tools', [])
-    return {
-        'messages': messages,
-        'tools': tools,
-    }
-
-
-parser = argparse.ArgumentParser()
-parser.add_argument('jsonl_path', type=str)
-args = parser.parse_args()
-
-output_dir = os.path.dirname(args.jsonl_path)
-output_path = os.path.join(output_dir, 'output.with_completions.jsonl.gz')
-
-# Check if output would be different from input
-needs_update = False
-with open(args.jsonl_path, 'r') as f_in:
-    for line in tqdm(f_in, desc='Checking for changes'):
-        data = json.loads(line)
-        new_completions = load_completions(output_dir, data['instance_id'])
-        current_completions = data.get('raw_completions')
-        if current_completions != new_completions:
-            needs_update = True
-            break
-
-if not needs_update:
-    print('No updates required. Skipping file update.')
-    exit(0)
-
-if os.path.exists(output_path):
-    print(f'Output file already exists at {output_path}, overwriting? (y/n)')
-    if input() != 'y':
-        print('Exiting...')
-        exit(0)
-
-# Process line by line
-with open(args.jsonl_path, 'r') as f_in, gzip.open(output_path, 'wt') as f_out:
-    for line in tqdm(f_in):
-        data = json.loads(line)
-        data['raw_completions'] = load_completions(output_dir, data['instance_id'])
-        f_out.write(json.dumps(data) + '\n')
-
-print(f'Saved compressed output to {output_path}')

evaluation/benchmarks/multi_swe_bench/scripts/eval/convert.py

@@ -1,11 +1,13 @@
-import argparse
 import json
 import re
 
+IN_FILE = 'output.jsonl'
+OUT_FILE = 'patch.jsonl'
 
-def main(input_file, output_file):
-    with open(input_file, 'r') as fin:
-        with open(output_file, 'w') as fout:
+
+def main():
+    with open(IN_FILE, 'r') as fin:
+        with open(OUT_FILE, 'w') as fout:
             for line in fin:
                 data = json.loads(line)
                 groups = re.match(r'(.*)__(.*)-(.*)', data['instance_id'])
@@ -13,14 +15,10 @@ def main(input_file, output_file):
                     'org': groups.group(1),
                     'repo': groups.group(2),
                     'number': groups.group(3),
-                    'fix_patch': data.get('test_result', {}).get('git_patch', '') or '',
+                    'fix_patch': data['test_result']['git_patch'],
                 }
                 fout.write(json.dumps(patch) + '\n')
 
 
 if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--input', required=True, help='Input .jsonl file path')
-    parser.add_argument('--output', required=True, help='Output .jsonl file path')
-    args = parser.parse_args()
-    main(args.input, args.output)
+    main()

evaluation/benchmarks/multi_swe_bench/scripts/eval/update_multi_swe_bench_config.py

@@ -1,70 +0,0 @@
-import argparse
-import json
-import os
-import subprocess
-
-
-def update_multi_swe_config(output_jsonl_path, config_path, dataset):
-    path_to_parent = os.path.dirname(os.path.abspath(output_jsonl_path))
-    converted_path = os.path.join(path_to_parent, 'output_converted.jsonl')
-
-    # Run the conversion script
-    subprocess.run(
-        [
-            'python3',
-            './evaluation/benchmarks/multi_swe_bench/scripts/eval/convert.py',
-            '--input',
-            output_jsonl_path,
-            '--output',
-            converted_path,
-        ],
-        check=True,
-    )
-
-    # Create required directories
-    os.makedirs(os.path.join(path_to_parent, 'eval_files', 'dataset'), exist_ok=True)
-    os.makedirs(os.path.join(path_to_parent, 'eval_files', 'workdir'), exist_ok=True)
-    os.makedirs(os.path.join(path_to_parent, 'eval_files', 'repos'), exist_ok=True)
-    os.makedirs(os.path.join(path_to_parent, 'eval_files', 'logs'), exist_ok=True)
-
-    # Prepare config dict
-    config = {
-        'mode': 'evaluation',
-        'workdir': os.path.join(path_to_parent, 'eval_files', 'workdir'),
-        'patch_files': [converted_path],
-        'dataset_files': [dataset],
-        'force_build': True,
-        'output_dir': os.path.join(path_to_parent, 'eval_files', 'dataset'),
-        'specifics': [],
-        'skips': [],
-        'repo_dir': os.path.join(path_to_parent, 'eval_files', 'repos'),
-        'need_clone': True,
-        'global_env': [],
-        'clear_env': True,
-        'stop_on_error': False,
-        'max_workers': 5,
-        'max_workers_build_image': 5,
-        'max_workers_run_instance': 5,
-        'log_dir': os.path.join(path_to_parent, 'eval_files', 'logs'),
-        'log_level': 'DEBUG',
-        'fix_patch_run_cmd': (
-            'bash -c "apt update ; apt install -y patch ; '
-            "sed -i 's@git apply.*@patch --batch --fuzz=5 -p1 -i /home/test.patch;"
-            'patch --batch --fuzz=5 -p1 -i /home/fix.patch@g\' /home/fix-run.sh ; chmod +x /home/*.sh  ; /home/fix-run.sh"'
-        ),
-    }
-
-    # Save to multibench.config
-    os.makedirs(os.path.dirname(config_path), exist_ok=True)
-    with open(config_path, 'w') as f:
-        json.dump(config, f, indent=4)
-
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--input', required=True, help='Path to input file')
-    parser.add_argument('--output', required=True, help='Path to create config')
-    parser.add_argument('--dataset', required=True, help='Path to dataset')
-    args = parser.parse_args()
-
-    update_multi_swe_config(args.input, args.output, args.dataset)

evaluation/benchmarks/multi_swe_bench/scripts/eval/update_output_with_eval.py

@@ -1,176 +0,0 @@
-import argparse
-import json
-import os
-from collections import defaultdict
-
-from tqdm import tqdm
-
-parser = argparse.ArgumentParser()
-parser.add_argument('input_file', type=str)
-parser.add_argument(
-    '--force',
-    action='store_true',
-    help='Force update all reports even if no changes are detected',
-)
-parser.add_argument(
-    '--overwrite-backup',
-    action='store_true',
-    help='Automatically overwrite existing backup files without prompting',
-)
-args = parser.parse_args()
-
-dirname = os.path.dirname(args.input_file)
-
-# Initialize counters and data structures
-instance_id_to_status = defaultdict(
-    lambda: {
-        'empty_generation': False,
-        'resolved': False,
-        'failed_apply_patch': False,
-        'error_eval': False,
-        'test_timeout': False,
-    }
-)
-
-# Process official report if it exists
-swebench_official_report_json = os.path.join(
-    dirname, 'eval_files/dataset/final_report.json'
-)
-openhands_remote_report_jsonl = args.input_file.replace(
-    '.jsonl', '.swebench_eval.jsonl'
-)
-
-if os.path.exists(swebench_official_report_json):
-    output_md_filepath = os.path.join(dirname, 'README.md')
-    with open(swebench_official_report_json, 'r') as f:
-        report = json.load(f)
-
-    # Convert instance IDs from "repo/name:pr-123" format to "repo__name-123" format
-    def convert_instance_id(instance_id):
-        """Convert instance ID from slash/colon-pr format to double underscore/dash format."""
-        if '/' in instance_id and ':pr-' in instance_id:
-            # Split on '/' and ':pr-'
-            parts = instance_id.split('/')
-            if len(parts) == 2:
-                repo_part = parts[0]
-                name_and_pr = parts[1]
-                if ':pr-' in name_and_pr:
-                    name, pr_number = name_and_pr.split(':pr-')
-                    return f'{repo_part}__{name}-{pr_number}'
-        return instance_id
-
-    # Convert all instance ID lists in the report
-    for key in [
-        'resolved_ids',
-        'unresolved_ids',
-        'error_ids',
-        'empty_patch_ids',
-        'incomplete_ids',
-    ]:
-        if key in report:
-            report[key] = [
-                convert_instance_id(instance_id) for instance_id in report[key]
-            ]
-
-    output_md = (
-        '# Multi-SWE-bench Report\n'
-        'This folder contains the evaluation results of the SWE-bench using the [official evaluation docker containerization](https://github.com/princeton-nlp/SWE-bench/blob/main/docs/20240627_docker/README.md#choosing-the-right-cache_level).\n\n'
-        '## Summary\n'
-        f'- total instances: {report["total_instances"]}\n'
-        f'- submitted instances: {report["submitted_instances"]}\n'
-        f'- completed instances: {report["completed_instances"]}\n'
-        f'- empty patch instances: {report["empty_patch_instances"]}\n'
-        f'- resolved instances: {report["resolved_instances"]}\n'
-        f'- unresolved instances: {report["unresolved_instances"]}\n'
-        f'- error instances: {report["error_instances"]}\n'
-    )
-
-    output_md += '\n## Resolved Instances\n'
-    # instance_id to status
-    for instance_id in report['resolved_ids']:
-        instance_id_to_status[instance_id]['resolved'] = True
-        output_md += (
-            f'- [{instance_id}](./eval_outputs/{instance_id}/run_instance.log)\n'
-        )
-
-    output_md += '\n## Unresolved Instances\n'
-    for instance_id in report['unresolved_ids']:
-        output_md += (
-            f'- [{instance_id}](./eval_outputs/{instance_id}/run_instance.log)\n'
-        )
-
-    output_md += '\n## Error Instances\n'
-    for instance_id in report['error_ids']:
-        instance_id_to_status[instance_id]['error_eval'] = True
-        output_md += (
-            f'- [{instance_id}](./eval_outputs/{instance_id}/run_instance.log)\n'
-        )
-
-    output_md += '\n## Empty Patch Instances\n'
-    for instance_id in report['empty_patch_ids']:
-        instance_id_to_status[instance_id]['empty_generation'] = True
-        output_md += (
-            f'- [{instance_id}](./eval_outputs/{instance_id}/run_instance.log)\n'
-        )
-
-    output_md += '\n## Incomplete Instances\n'
-    for instance_id in report['incomplete_ids']:
-        output_md += (
-            f'- [{instance_id}](./eval_outputs/{instance_id}/run_instance.log)\n'
-        )
-
-    with open(output_md_filepath, 'w') as f:
-        f.write(output_md)
-
-else:
-    print(
-        f'No report file found: Both {swebench_official_report_json} and {openhands_remote_report_jsonl} do not exist.'
-    )
-    exit()
-
-# Before backup and update, check if any changes would be made (unless --force is used)
-if not args.force:
-    needs_update = False
-    with open(args.input_file, 'r') as infile:
-        for line in tqdm(infile, desc='Checking for changes'):
-            data = json.loads(line)
-            instance_id = data['instance_id']
-            current_report = data.get('report', {})
-            new_report = instance_id_to_status[
-                instance_id
-            ]  # if no report, it's not resolved
-            if current_report != new_report:
-                needs_update = True
-                break
-
-    if not needs_update:
-        print('No updates detected. Skipping file update.')
-        exit()
-else:
-    print('Force flag enabled. Updating all reports regardless of changes.')
-
-# Backup and update the original file row by row
-if os.path.exists(args.input_file + '.bak'):
-    if args.overwrite_backup:
-        print(
-            'Existing backup file found. Overwriting automatically due to --overwrite-backup flag.'
-        )
-        os.remove(args.input_file + '.bak')
-    else:
-        conf = input('Existing backup file found. Do you want to overwrite it? (y/n)')
-        if conf != 'y':
-            exit()
-        os.remove(args.input_file + '.bak')
-
-os.rename(args.input_file, args.input_file + '.bak')
-
-# Process and write file row by row
-with (
-    open(args.input_file + '.bak', 'r') as infile,
-    open(args.input_file, 'w') as outfile,
-):
-    for line in tqdm(infile, desc='Updating output file'):
-        data = json.loads(line)
-        instance_id = data['instance_id']
-        data['report'] = instance_id_to_status[instance_id]
-        outfile.write(json.dumps(data) + '\n')

evaluation/benchmarks/multi_swe_bench/scripts/rollout_multi_swegym.sh

@@ -1,146 +0,0 @@
-#!/bin/bash
-
-# NOTE: this script is for rolling out the Multi-SWE-Gym dataset for **TRAINING**
-# For more information, please refer to
-# 1. the Github Repo: https://github.com/SWE-Gym/SWE-Gym
-# 2. the paper: https://arxiv.org/abs/2412.21139
-
-MODEL=$1  # eg your llm config name in config.toml (eg: "llm.claude-3-5-sonnet-20241022-t05")
-EXP_NAME=$2 # "train-t05"
-EVAL_DATASET=$3  # path to original dataset (jsonl file)
-N_WORKERS=${4:-64}
-N_RUNS=${5:-1}
-
-export EXP_NAME=$EXP_NAME
-# use 2x resources for rollout since some codebases are pretty resource-intensive
-export DEFAULT_RUNTIME_RESOURCE_FACTOR=2
-echo "MODEL: $MODEL"
-echo "EXP_NAME: $EXP_NAME"
-echo "EVAL_DATASET: $EVAL_DATASET"
-# Generate DATASET path by adding _with_runtime_ before .jsonl extension
-DATASET="${EVAL_DATASET%.jsonl}_with_runtime_.jsonl"  # path to converted dataset
-
-# Create the converted dataset file
-echo "Creating converted dataset at: $DATASET"
-poetry run python ./evaluation/benchmarks/multi_swe_bench/scripts/data/data_change.py --input "$EVAL_DATASET" --output "$DATASET"
-
-SPLIT="train"
-export LANGUAGE=java
-
-if [ -z "$ALLHANDS_API_KEY" ] || [ "$RUNTIME" != "remote" ]; then
-    echo "ALLHANDS_API_KEY is not set or RUNTIME is not set to remote. Will rollout and evaluate locally using Docker. WARNING: A large value of N_WORKERS will result in a large number of Docker containers being spun up and may crash your machine."
-    export RUNTIME=docker
-else
-    echo "ALLHANDS_API_KEY is set and RUNTIME is set to remote. Continuing rollout and evaluation with remote runtime..."
-    export SANDBOX_REMOTE_RUNTIME_API_URL="https://runtime.eval.all-hands.dev"
-fi
-
-#EVAL_LIMIT=3000
-MAX_ITER=100
-
-
-# ===== Run inference =====
-source "evaluation/utils/version_control.sh"
-get_openhands_version
-
-echo "OPENHANDS_VERSION: $OPENHANDS_VERSION"
-echo "MODEL_CONFIG: $MODEL_CONFIG"
-echo "DATASET: $DATASET"
-echo "EVAL_DOCKER_IMAGE_PREFIX: $EVAL_DOCKER_IMAGE_PREFIX"
-
-# Default to NOT use Hint
-export USE_INSTANCE_IMAGE=true
-export USE_HINT_TEXT=false
-export RUN_WITH_BROWSING=false
-echo "USE_HINT_TEXT: $USE_HINT_TEXT"
-EVAL_NOTE="$OPENHANDS_VERSION-no-hint-$EXP_NAME"
-
-function run_eval() {
-  local eval_note=$1
-  export LANGUAGE=java
-  echo "About to run command"
-  COMMAND="EVAL_DOCKER_IMAGE_PREFIX=$EVAL_DOCKER_IMAGE_PREFIX; LANGUAGE=java;
-    poetry run python evaluation/benchmarks/multi_swe_bench/run_infer.py \
-    --agent-cls CodeActAgent \
-    --llm-config $MODEL \
-    --max-iterations $MAX_ITER \
-    --eval-num-workers $N_WORKERS \
-    --eval-note $eval_note \
-    --dataset $DATASET \
-    --split $SPLIT"
-
-  echo "Running command: $COMMAND"
-  if [ -n "$EVAL_LIMIT" ]; then
-    echo "EVAL_LIMIT: $EVAL_LIMIT"
-    COMMAND="$COMMAND --eval-n-limit $EVAL_LIMIT"
-  fi
-
-  # Run the command
-  eval $COMMAND
-}
-
-for run_idx in $(seq 1 $N_RUNS); do
-
-    while true; do
-        echo "### Running inference... ###"
-        unset SANDBOX_ENV_GITHUB_TOKEN # prevent the agent from using the github token to push
-        current_eval_note="$EVAL_NOTE-run_$run_idx"
-        echo "EVAL_NOTE: $current_eval_note"
-        echo "DATASET command: $DATASET"
-        #INFER_OUTPUT=$(run_eval $current_eval_note)
-        INFER_OUTPUT=$(run_eval $current_eval_note | tee /dev/stderr)
-        INFER_STATUS=$?  # Capture the exit status of run_infer.sh
-        echo "INFER_STATUS: $INFER_STATUS"
-
-        echo "### Cleaning up remote runtime... ###"
-        ./evaluation/utils/scripts/cleanup_remote_runtime.sh
-
-        if [ $INFER_STATUS -eq 0 ]; then
-            echo "### Inference completed successfully. ###"
-            break
-        else
-            echo "### Inference failed with exit code $INFER_STATUS. Retrying... ###"
-        fi
-    done
-
-    # Extract the output directory using the special delimiters
-    OUTPUT_FILE=$(echo "$INFER_OUTPUT" | grep -o '### OUTPUT FILE:.* ###' | sed 's/### OUTPUT FILE: \(.*\) ###/\1/')
-    echo "Got OUTPUT_FILE: $OUTPUT_FILE"
-
-    while true; do
-        echo "### Evaluating on $OUTPUT_FILE ... ###"
-        OUTPUT_CONFIG_FILE="${OUTPUT_FILE%.jsonl}_config.json"
-        export EVAL_SKIP_BUILD_ERRORS=true
-        pip install multi-swe-bench --quiet --disable-pip-version-check > /dev/null 2>&1
-        COMMAND="poetry run python ./evaluation/benchmarks/multi_swe_bench/scripts/eval/update_multi_swe_bench_config.py --input $OUTPUT_FILE --output $OUTPUT_CONFIG_FILE --dataset $EVAL_DATASET;
-        python -m multi_swe_bench.harness.run_evaluation --config $OUTPUT_CONFIG_FILE
-        "
-
-        if [ -n "$EVAL_LIMIT" ]; then
-        echo "EVAL_LIMIT: $EVAL_LIMIT"
-        COMMAND="$COMMAND --eval-n-limit $EVAL_LIMIT"
-        fi
-        echo "Running command: $COMMAND"
-        # Run the command
-        eval $COMMAND
-        EVAL_STATUS=$?
-        if [ $EVAL_STATUS -eq 0 ]; then
-            echo "### Evaluation completed successfully. ###"
-            break
-        else
-            echo "### Evaluation failed with exit code $EVAL_STATUS. Retrying... ###"
-        fi
-
-        ./evaluation/utils/scripts/cleanup_remote_runtime.sh
-    done
-
-    # update the output with evaluation results
-    echo "### Updating the output with evaluation results... ###"
-    poetry run python evaluation/benchmarks/multi_swe_bench/scripts/eval/update_output_with_eval.py $OUTPUT_FILE
-
-    echo "### Combining the final completions... ###"
-    poetry run python evaluation/benchmarks/multi_swe_bench/scripts/eval/combine_final_completions.py $OUTPUT_FILE
-
-    echo "### DONE for run $run_idx! ###"
-    echo "You can find the final output at $(dirname $OUTPUT_FILE)/$FINAL_OUTPUT_FILE"
-done

evaluation/benchmarks/multi_swe_bench/scripts/run_infer.sh

@@ -47,8 +47,8 @@ if [ -z "$DATASET" ]; then
 fi
 
 if [ -z "$LANGUAGE" ]; then
-  echo "LANGUAGE not specified, use default python"
-  LANGUAGE="java"
+  echo "LANUGUAGE not specified, use default python"
+  LANGUAGE="python"
 fi
 
 if [ -z "$SPLIT" ]; then
@@ -69,10 +69,10 @@ fi
 
 if [ -z "$EVAL_DOCKER_IMAGE_PREFIX" ]; then
   if [ "$LANGUAGE" = "python" ]; then
-  echo "EVAL_DOCKER_IMAGE_PREFIX is docker.io/xingyaoww/ as default as LANGUAGE is python"
+  echo "EVAL_DOCKER_IMAGE_PREFIX is docker.io/xingyaoww/ as default as LANUGUAGE is python"
     EVAL_DOCKER_IMAGE_PREFIX="docker.io/xingyaoww/"
   elif [ "$LANGUAGE" = "java" ]; then
-  echo "EVAL_DOCKER_IMAGE_PREFIX is empty as LANGUAGE is java"
+  echo "EVAL_DOCKER_IMAGE_PREFIX is java_verified as LANUGUAGE is java"
     EVAL_DOCKER_IMAGE_PREFIX=""
   fi
 fi

evaluation/benchmarks/multi_swe_bench/scripts/swegym/convert_data.ipynb

@@ -1,344 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os\n",
-    "\n",
-    "import pandas as pd\n",
-    "from tqdm import tqdm\n",
-    "\n",
-    "tqdm.pandas()"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "# 1. Load raw data and convert to training data"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import gzip\n",
-    "import json\n",
-    "\n",
-    "from tqdm import tqdm\n",
-    "\n",
-    "FILE_PATHS = [\n",
-    "    'YOURPATH-no-hint-train-t05-run_1/output.with_completions.jsonl.gz',\n",
-    "    'YOURPATH-no-hint-train-t05-run_2/output.with_completions.jsonl.gz',\n",
-    "]\n",
-    "\n",
-    "# More memory efficient for large files\n",
-    "# Initialize lists to store the data\n",
-    "data = []\n",
-    "\n",
-    "\n",
-    "# Read file line by line\n",
-    "for FILE_PATH in FILE_PATHS:\n",
-    "    with gzip.open(FILE_PATH, 'rb') as f:  # Use 'rb' for gzipped files\n",
-    "        for i, line in tqdm(\n",
-    "            enumerate(f), desc=f'Processing {FILE_PATH.split(\"/\")[-1]}'\n",
-    "        ):\n",
-    "            # Parse only the fields we need\n",
-    "            raw_data = json.loads(line)\n",
-    "            data.append(\n",
-    "                {\n",
-    "                    'resolved': raw_data['report']['resolved'],\n",
-    "                    'messages': raw_data['raw_completions']['messages']\n",
-    "                    if raw_data['raw_completions'] is not None\n",
-    "                    else None,\n",
-    "                    'git_patch': raw_data['test_result'].get('git_patch', ''),\n",
-    "                    'tools': raw_data['raw_completions']['tools']\n",
-    "                    if raw_data['raw_completions'] is not None\n",
-    "                    and 'tools' in raw_data['raw_completions']\n",
-    "                    else None,\n",
-    "                }\n",
-    "            )\n",
-    "\n",
-    "# Convert to DataFrame after collecting all data\n",
-    "df = pd.DataFrame(data)\n",
-    "print(f'#total amount of data={len(df)}')\n",
-    "df = df[~df['messages'].isna()]\n",
-    "print(f'#total amount of data after removing nan={len(df)}')"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Filter"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "def _contains_multiple_tool_calls(messages: list[dict]) -> bool:\n",
-    "    return any(\n",
-    "        message.get('tool_calls') and len(message['tool_calls']) > 1\n",
-    "        for message in messages\n",
-    "    )\n",
-    "\n",
-    "\n",
-    "df['contains_multiple_tool_calls'] = df['messages'].apply(_contains_multiple_tool_calls)\n",
-    "display(df.groupby(['contains_multiple_tool_calls'])['resolved'].sum())"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "scrolled": true
-   },
-   "outputs": [],
-   "source": [
-    "import copy\n",
-    "\n",
-    "# Convert function calling messages to non-function calling messages\n",
-    "from openhands.llm.fn_call_converter import (\n",
-    "    FunctionCallConversionError,\n",
-    "    convert_fncall_messages_to_non_fncall_messages,\n",
-    "    convert_from_multiple_tool_calls_to_single_tool_call_messages,\n",
-    ")\n",
-    "\n",
-    "total_failed = 0\n",
-    "\n",
-    "\n",
-    "def _convert_messages(messages: list[dict], tools: list[dict]) -> list[dict]:\n",
-    "    global total_failed\n",
-    "    message_copy = copy.deepcopy(messages)\n",
-    "    for message in message_copy:\n",
-    "        if message['content'] is None:\n",
-    "            message['content'] = ''\n",
-    "    try:\n",
-    "        return convert_fncall_messages_to_non_fncall_messages(\n",
-    "            message_copy, tools, add_in_context_learning_example=False\n",
-    "        )\n",
-    "    except FunctionCallConversionError:\n",
-    "        total_failed += 1\n",
-    "        # print(f'Failed to convert messages: {messages}\\nTools: {tools}')\n",
-    "        # traceback.print_exc()\n",
-    "        return None\n",
-    "\n",
-    "\n",
-    "df['converted_messages'] = df.apply(\n",
-    "    lambda row: convert_from_multiple_tool_calls_to_single_tool_call_messages(\n",
-    "        row['messages'], ignore_final_tool_result=True\n",
-    "    ),\n",
-    "    axis=1,\n",
-    ")\n",
-    "df['nonfncall_messages'] = df.apply(\n",
-    "    lambda row: _convert_messages(row['converted_messages'], row['tools']), axis=1\n",
-    ")\n",
-    "print('total nan', df['nonfncall_messages'].isna().sum())\n",
-    "df = df[~df['nonfncall_messages'].isna()]\n",
-    "print(df['nonfncall_messages'].iloc[0])\n",
-    "\n",
-    "print(f'Total failed: {total_failed}')"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Tokenization"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from pandarallel import pandarallel\n",
-    "from transformers import AutoTokenizer\n",
-    "\n",
-    "os.environ['TOKENIZERS_PARALLELISM'] = 'false'\n",
-    "pandarallel.initialize(progress_bar=True, verbose=1, nb_workers=16)\n",
-    "tokenizer = AutoTokenizer.from_pretrained('Qwen/Qwen2.5-7B-Instruct')\n",
-    "\n",
-    "\n",
-    "def clean_messages(messages):\n",
-    "    clean = []\n",
-    "    for msg in messages:\n",
-    "        if not isinstance(msg, dict):\n",
-    "            continue\n",
-    "        role = msg.get('role')\n",
-    "        content = msg.get('content')\n",
-    "        if isinstance(content, str):\n",
-    "            text = content\n",
-    "        elif isinstance(content, dict):\n",
-    "            text = content.get('text')\n",
-    "        elif (\n",
-    "            isinstance(content, list)\n",
-    "            and len(content) == 1\n",
-    "            and isinstance(content[0], dict)\n",
-    "        ):\n",
-    "            text = content[0].get('text')\n",
-    "        else:\n",
-    "            print(f'Format not accepted {content}')\n",
-    "        clean.append({'role': role, 'content': text})\n",
-    "    return clean\n",
-    "\n",
-    "\n",
-    "# Step 1: Clean the messages\n",
-    "df['nonfncall_messages'] = df['nonfncall_messages'].apply(clean_messages)\n",
-    "\n",
-    "# Step 2: Compute token count\n",
-    "df['n_tokens'] = df['nonfncall_messages'].parallel_apply(\n",
-    "    lambda x: len(tokenizer.apply_chat_template(x))\n",
-    ")\n",
-    "\n",
-    "# print(df['nonfncall_messages'])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "print(f'BEFORE: #total={len(df)}')\n",
-    "df_selected = df[df['n_tokens'] < 131072]\n",
-    "print(f'AFTER(truncated to 128k): #total={len(df_selected)}')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "df_selected['n_tokens'].describe()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# ecdf of n_tokens\n",
-    "import matplotlib.pyplot as plt\n",
-    "import seaborn as sns\n",
-    "\n",
-    "display(df.groupby(['resolved'])['n_tokens'].describe())\n",
-    "sns.ecdfplot(x='n_tokens', data=df, hue='resolved')\n",
-    "plt.show()\n",
-    "\n",
-    "print(f'#total={len(df)}')\n",
-    "df_selected = df[df['n_tokens'] < 131072]\n",
-    "print(f'#selected={len(df_selected)}')\n",
-    "display(df_selected.groupby(['resolved'])['n_tokens'].describe())\n",
-    "sns.ecdfplot(x='n_tokens', data=df_selected, hue='resolved')\n",
-    "plt.show()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "df_selected[~df_selected['resolved']]['n_tokens'].describe()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "df_selected['resolved'].value_counts()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "df_selected.groupby(['resolved'])['n_tokens'].describe()"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "# Save Resolved Messages for SFT"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Flatten messages and change format to {\"content\": \"\", \"role\": \"\"}\n",
-    "df_selected[df_selected['resolved']][['nonfncall_messages']].rename(\n",
-    "    columns={'nonfncall_messages': 'messages'}\n",
-    ").to_json(\n",
-    "    os.path.join(\n",
-    "        'PATH_TO_FILE',\n",
-    "        f'policy_traj_128k_swegym_{df_selected[\"resolved\"].value_counts()[True]}i.jsonl',\n",
-    "    ),\n",
-    "    lines=True,\n",
-    "    orient='records',\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.12.11"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 4
-}

evaluation/benchmarks/swe_bench/run_infer.py

@@ -47,8 +47,6 @@ from openhands.core.config import (
     get_agent_config_arg,
     get_evaluation_parser,
     get_llm_config_arg,
-    get_llms_for_routing_config,
-    get_model_routing_config_arg,
 )
 from openhands.core.config.condenser_config import NoOpCondenserConfig
 from openhands.core.config.utils import get_condenser_config_arg
@@ -246,11 +244,6 @@ def get_config(
     # get 'draft_editor' config if exists
     config.set_llm_config(get_llm_config_arg('draft_editor'), 'draft_editor')
 
-    model_routing_config = get_model_routing_config_arg()
-    model_routing_config.llms_for_routing = (
-        get_llms_for_routing_config()
-    )  # Populate with LLMs for routing from config.toml file
-
     agent_config = AgentConfig(
         enable_jupyter=False,
         enable_browsing=RUN_WITH_BROWSING,
@@ -258,10 +251,8 @@ def get_config(
         enable_mcp=False,
         condenser=metadata.condenser_config,
         enable_prompt_extensions=False,
-        model_routing=model_routing_config,
     )
     config.set_agent_config(agent_config)
-
     return config
 
 

evaluation/benchmarks/swe_perf/README.md

@@ -1,81 +0,0 @@
-# SWE-Perf Evaluation
-
-This folder contains the OpenHands inference generation of the [SWE-Perf benchmark](https://swe-perf.github.io/) ([paper](https://arxiv.org/pdf/2507.12415v1)).
-
-The evaluation consists of three steps:
-
-1. Environment setup: [install python environment](../../README.md#development-environment) and [configure LLM config](../../README.md#configure-openhands-and-your-llm).
-2. [Run inference](#running-inference-locally-with-docker): Generate a edit patch for each Github issue
-3. [Evaluate patches](#evaluate-generated-patches)
-
-## Setup Environment and LLM Configuration
-
-Please follow instruction [here](../../README.md#setup) to setup your local development environment and LLM.
-
-## Running inference Locally with Docker
-
-Make sure your Docker daemon is running, and you have ample disk space (at least 200-500GB, depends on the SWE-PErf set you are running on) for the instance-level docker image.
-
-When the `run_infer.sh` script is started, it will automatically pull the relevant SWE-Perf images.
-For example, for instance ID `scikit-learn_scikit-learn-11674`, it will try to pull our pre-build docker image `betty1202/sweb.eval.x86_64.scikit-learn_s_scikit-learn-11674` from DockerHub.
-This image will be used create an OpenHands runtime image where the agent will operate on.
-
-```bash
-./evaluation/benchmarks/swe_perf/scripts/run_infer.sh [model_config] [git-version] [agent] [eval_limit] [max_iter] [num_workers] [dataset] [dataset_split] [n_runs] [mode]
-
-# Example
-./evaluation/benchmarks/swe_bench/scripts/run_infer.sh llm.eval_gpt4_1106_preview HEAD CodeActAgent 500 100 1 SWE-Perf/SWE-Perf test
-```
-
-where `model_config` is mandatory, and the rest are optional.
-
-- `model_config`, e.g. `eval_gpt4_1106_preview`, is the config group name for your
-LLM settings, as defined in your `config.toml`.
-- `git-version`, e.g. `HEAD`, is the git commit hash of the OpenHands version you would
-like to evaluate. It could also be a release tag like `0.6.2`.
-- `agent`, e.g. `CodeActAgent`, is the name of the agent for benchmarks, defaulting
-to `CodeActAgent`.
-- `eval_limit`, e.g. `10`, limits the evaluation to the first `eval_limit` instances. By
-default, the script evaluates the entire SWE-Perf test set (140 issues). Note:
-in order to use `eval_limit`, you must also set `agent`.
-- `max_iter`, e.g. `20`, is the maximum number of iterations for the agent to run. By
-default, it is set to 100.
-- `num_workers`, e.g. `3`, is the number of parallel workers to run the evaluation. By
-default, it is set to 1.
-- `dataset`, a huggingface dataset name. e.g. `SWE-Perf/SWE-Perf`, specifies which dataset to evaluate on.
-- `dataset_split`, split for the huggingface dataset. e.g., `test`, `dev`. Default to `test`.
-
-- `n_runs`, e.g. `3`, is the number of times to run the evaluation. Default is 1.
-- `mode`, e.g. `swt`, `swt-ci`, or `swe`, specifies the evaluation mode. Default is `swe`.
-
-> [!CAUTION]
-> Setting `num_workers` larger than 1 is not officially tested, YMMV.
-
-
-Let's say you'd like to run 10 instances using `llm.eval_gpt4_1106_preview` and CodeActAgent,
-
-then your command would be:
-
-```bash
-./evaluation/benchmarks/swe_bench/scripts/run_infer.sh llm.eval_gpt4_1106_preview HEAD CodeActAgent 10
-```
-
-## Evaluate Generated Patches
-
-
-To evaluate the generated patch, follow these steps:
-
-### 1. Convert output to the evaluation standard format
-Run the following command:
-```bash
-python -m evaluation.benchmarks.swe_perf.format_conversion \
-    --input_path [input_path] \
-    --output_path [output_path]
-```
-
-* `input_path`: Path to the raw generated patch file.
-* `output_path`: Path where the converted file will be saved.
-
-### 2. Run the SWE-Perf benchmark official evaluation
-
-Once the output is converted, use the [official SWE-Perf benchmark evaluation](https://github.com/SWE-Perf/SWE-Perf/tree/main/evaluation) to evaluate it.

evaluation/benchmarks/swe_perf/binary_patch_utils.py

@@ -1,52 +0,0 @@
-"""
-Utilities for handling binary files and patch generation in SWE-Perf evaluation.
-"""
-
-
-def remove_binary_diffs(patch_text):
-    """
-    Remove binary file diffs from a git patch.
-
-    Args:
-        patch_text (str): The git patch text
-
-    Returns:
-        str: The cleaned patch text with binary diffs removed
-    """
-    lines = patch_text.splitlines()
-    cleaned_lines = []
-    block = []
-    is_binary_block = False
-
-    for line in lines:
-        if line.startswith('diff --git '):
-            if block and not is_binary_block:
-                cleaned_lines.extend(block)
-            block = [line]
-            is_binary_block = False
-        elif 'Binary files' in line:
-            is_binary_block = True
-            block.append(line)
-        else:
-            block.append(line)
-
-    if block and not is_binary_block:
-        cleaned_lines.extend(block)
-    return '\n'.join(cleaned_lines)
-
-
-def remove_binary_files_from_git():
-    """
-    Generate a bash command to remove binary files from git staging.
-
-    Returns:
-        str: A bash command that removes binary files from git staging
-    """
-    return """
-    for file in $(git status --porcelain | grep -E "^(M| M|\\?\\?|A| A)" | cut -c4-); do
-        if [ -f "$file" ] && (file "$file" | grep -q "executable" || git check-attr binary "$file" | grep -q "binary: set"); then
-            git rm -f "$file" 2>/dev/null || rm -f "$file"
-            echo "Removed: $file"
-        fi
-    done
-    """.strip()

evaluation/benchmarks/swe_perf/format_conversion.py

@@ -1,45 +0,0 @@
-import json
-import os
-from argparse import ArgumentParser
-
-parser = ArgumentParser()
-parser.add_argument('--input_path', type=str, help='Name of input path to JSON file.')
-parser.add_argument('--output_path', type=str, help='Name of output path to JSON file.')
-args = parser.parse_args()
-
-input_path = args.input_path
-output_path = args.output_path
-os.makedirs(output_path, exist_ok=True)
-
-
-def load_jsonl(file_path):
-    """Load JSONL file into a list of dictionaries."""
-    data = []
-    with open(file_path, 'r') as f:
-        for line in f:
-            data.append(json.loads(line))
-    return data
-
-
-dataset = load_jsonl(input_path)
-ooutput_dataset = []
-for data in dataset:
-    instance_id = data['instance_id']
-    model_name_or_path = 'openhands'
-    model_patch = (
-        data['test_result']['git_patch']
-        if 'test_result' in data and 'git_patch' in data['test_result']
-        else None
-    )
-    ooutput_dataset.append(
-        {
-            'instance_id': instance_id,
-            'model_name_or_path': model_name_or_path,
-            'model_patch': model_patch,
-        }
-    )
-
-with open(os.path.join(output_path, 'output.jsonl'), 'w') as f:
-    for item in ooutput_dataset:
-        json_line = json.dumps(item, ensure_ascii=False)
-        f.write(json_line + '\n')

evaluation/benchmarks/swe_perf/resource/mapping.py

@@ -1,39 +0,0 @@
-"""Mapping instance_id to resource_factor.
-
-Different instances may have different resource requirements.
-e.g., some instances may require more memory/CPU to run inference.
-This file tracks the resource requirements of different instances.
-"""
-
-import json
-import os
-
-from openhands.core.logger import openhands_logger as logger
-
-CUR_DIR = os.path.dirname(os.path.abspath(__file__))
-DEFAULT_RUNTIME_RESOURCE_FACTOR = int(
-    os.environ.get('DEFAULT_RUNTIME_RESOURCE_FACTOR', 1)
-)
-
-# dataset to resource mapping
-_global_resource_mapping: dict[str, dict[str, float]] = {}
-
-
-def get_resource_mapping(dataset_name: str) -> dict[str, float]:
-    if dataset_name not in _global_resource_mapping:
-        file_path = os.path.join(CUR_DIR, f'{dataset_name}.json')
-        if not os.path.exists(file_path):
-            logger.info(f'Resource mapping for {dataset_name} not found.')
-            return None
-
-        with open(file_path, 'r') as f:
-            _global_resource_mapping[dataset_name] = json.load(f)
-        logger.debug(f'Loaded resource mapping for {dataset_name}')
-    return _global_resource_mapping[dataset_name]
-
-
-def get_instance_resource_factor(dataset_name: str, instance_id: str) -> int:
-    resource_mapping = get_resource_mapping(dataset_name)
-    if resource_mapping is None:
-        return DEFAULT_RUNTIME_RESOURCE_FACTOR
-    return int(resource_mapping.get(instance_id, DEFAULT_RUNTIME_RESOURCE_FACTOR))

evaluation/benchmarks/swe_perf/resource/swt_bench_constants.py

@@ -1,842 +0,0 @@
-# Based on https://github.com/logic-star-ai/swt-bench/blob/master/src/constants.py
-
-# Constants - Installation Specifications
-MAP_VERSION_TO_INSTALL_SKLEARN = {
-    k: {
-        'python': '3.6',
-        'packages': 'numpy scipy cython pytest pandas matplotlib',
-        'install': 'python -m pip install -v --no-use-pep517 --no-build-isolation -e .',
-        'pip_packages': [
-            'cython',
-            'numpy==1.19.2',
-            'setuptools',
-            'scipy==1.5.2',
-        ],
-    }
-    for k in ['0.20', '0.21', '0.22']
-}
-MAP_VERSION_TO_INSTALL_SKLEARN.update(
-    {
-        k: {
-            'python': '3.9',
-            'packages': "'numpy==1.19.2' 'scipy==1.5.2' 'cython==3.0.10' pytest 'pandas<2.0.0' 'matplotlib<3.9.0' setuptools pytest joblib threadpoolctl",
-            'install': 'python -m pip install -v --no-use-pep517 --no-build-isolation -e .',
-            'pip_packages': ['cython', 'setuptools', 'numpy', 'scipy'],
-        }
-        for k in ['1.3', '1.4']
-    }
-)
-MAP_VERSION_TO_INSTALL_FLASK = {
-    '2.0': {
-        'python': '3.9',
-        'packages': 'requirements.txt',
-        'install': 'python -m pip install -e .',
-        'pip_packages': [
-            'setuptools==70.0.0',
-            'Werkzeug==2.3.7',
-            'Jinja2==3.0.1',
-            'itsdangerous==2.1.2',
-            'click==8.0.1',
-            'MarkupSafe==2.1.3',
-        ],
-    },
-    '2.1': {
-        'python': '3.10',
-        'packages': 'requirements.txt',
-        'install': 'python -m pip install -e .',
-        'pip_packages': [
-            'click==8.1.3',
-            'itsdangerous==2.1.2',
-            'Jinja2==3.1.2',
-            'MarkupSafe==2.1.1',
-            'Werkzeug==2.3.7',
-        ],
-    },
-}
-MAP_VERSION_TO_INSTALL_FLASK.update(
-    {
-        k: {
-            'python': '3.11',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-            'pip_packages': [
-                'click==8.1.3',
-                'itsdangerous==2.1.2',
-                'Jinja2==3.1.2',
-                'MarkupSafe==2.1.1',
-                'Werkzeug==2.3.7',
-            ],
-        }
-        for k in ['2.2', '2.3']
-    }
-)
-MAP_VERSION_TO_INSTALL_DJANGO = {
-    k: {
-        'python': '3.5',
-        'packages': 'requirements.txt',
-        'pre_install': [
-            'apt-get update && apt-get install -y locales',
-            "echo 'en_US UTF-8' > /etc/locale.gen",
-            'locale-gen en_US.UTF-8',
-        ],
-        'install': 'python setup.py install',
-        'pip_packages': ['setuptools'],
-        'eval_commands': [
-            'export LANG=en_US.UTF-8',
-            'export LC_ALL=en_US.UTF-8',
-            'export PYTHONIOENCODING=utf8',
-            'export LANGUAGE=en_US:en',
-        ],
-    }
-    for k in ['1.7', '1.8', '1.9', '1.10', '1.11', '2.0', '2.1', '2.2']
-}
-MAP_VERSION_TO_INSTALL_DJANGO.update(
-    {
-        k: {'python': '3.5', 'install': 'python setup.py install'}
-        for k in ['1.4', '1.5', '1.6']
-    }
-)
-MAP_VERSION_TO_INSTALL_DJANGO.update(
-    {
-        k: {
-            'python': '3.6',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-            'eval_commands': [
-                "sed -i '/en_US.UTF-8/s/^# //g' /etc/locale.gen && locale-gen",
-                'export LANG=en_US.UTF-8',
-                'export LANGUAGE=en_US:en',
-                'export LC_ALL=en_US.UTF-8',
-            ],
-        }
-        for k in ['3.0', '3.1', '3.2']
-    }
-)
-MAP_VERSION_TO_INSTALL_DJANGO.update(
-    {
-        k: {
-            'python': '3.8',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-        }
-        for k in ['4.0']
-    }
-)
-MAP_VERSION_TO_INSTALL_DJANGO.update(
-    {
-        k: {
-            'python': '3.9',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-        }
-        for k in ['4.1', '4.2']
-    }
-)
-MAP_VERSION_TO_INSTALL_DJANGO.update(
-    {
-        k: {
-            'python': '3.11',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-        }
-        for k in ['5.0']
-    }
-)
-MAP_VERSION_TO_INSTALL_REQUESTS = {
-    k: {'python': '3.9', 'packages': 'pytest', 'install': 'python -m pip install .'}
-    for k in ['0.7', '0.8', '0.9', '0.11', '0.13', '0.14', '1.1', '1.2', '2.0', '2.2']
-    + ['2.3', '2.4', '2.5', '2.7', '2.8', '2.9', '2.10', '2.11', '2.12', '2.17']
-    + ['2.18', '2.19', '2.22', '2.26', '2.25', '2.27', '3.0']
-}
-MAP_VERSION_TO_INSTALL_SEABORN = {
-    k: {
-        'python': '3.9',
-        'install': 'python -m pip install -e .',
-        'pip_packages': [
-            'contourpy==1.1.0',
-            'cycler==0.11.0',
-            'fonttools==4.42.1',
-            'importlib-resources==6.0.1',
-            'kiwisolver==1.4.5',
-            'matplotlib==3.7.2',
-            'numpy==1.25.2',
-            'packaging==23.1',
-            'pandas==1.3.5',  # 2.0.3
-            'pillow==10.0.0',
-            'pyparsing==3.0.9',
-            'pytest',
-            'python-dateutil==2.8.2',
-            'pytz==2023.3.post1',
-            'scipy==1.11.2',
-            'six==1.16.0',
-            'tzdata==2023.1',
-            'zipp==3.16.2',
-        ],
-    }
-    for k in ['0.11']
-}
-MAP_VERSION_TO_INSTALL_SEABORN.update(
-    {
-        k: {
-            'python': '3.9',
-            'install': 'python -m pip install -e .[dev]',
-            'pip_packages': [
-                'contourpy==1.1.0',
-                'cycler==0.11.0',
-                'fonttools==4.42.1',
-                'importlib-resources==6.0.1',
-                'kiwisolver==1.4.5',
-                'matplotlib==3.7.2',
-                'numpy==1.25.2',
-                'packaging==23.1',
-                'pandas==2.0.0',
-                'pillow==10.0.0',
-                'pyparsing==3.0.9',
-                'pytest',
-                'python-dateutil==2.8.2',
-                'pytz==2023.3.post1',
-                'scipy==1.11.2',
-                'six==1.16.0',
-                'tzdata==2023.1',
-                'zipp==3.16.2',
-            ],
-        }
-        for k in ['0.12', '0.13']
-    }
-)
-MAP_VERSION_TO_INSTALL_PYTEST = {
-    k: {'python': '3.9', 'install': 'python -m pip install -e .'}
-    for k in [
-        '4.4',
-        '4.5',
-        '4.6',
-        '5.0',
-        '5.1',
-        '5.2',
-        '5.3',
-        '5.4',
-        '6.0',
-        '6.2',
-        '6.3',
-        '7.0',
-        '7.1',
-        '7.2',
-        '7.4',
-        '8.0',
-    ]
-}
-MAP_VERSION_TO_INSTALL_PYTEST['4.4']['pip_packages'] = [
-    'atomicwrites==1.4.1',
-    'attrs==23.1.0',
-    'more-itertools==10.1.0',
-    'pluggy==0.13.1',
-    'py==1.11.0',
-    'setuptools==68.0.0',
-    'six==1.16.0',
-]
-MAP_VERSION_TO_INSTALL_PYTEST['4.5']['pip_packages'] = [
-    'atomicwrites==1.4.1',
-    'attrs==23.1.0',
-    'more-itertools==10.1.0',
-    'pluggy==0.11.0',
-    'py==1.11.0',
-    'setuptools==68.0.0',
-    'six==1.16.0',
-    'wcwidth==0.2.6',
-]
-MAP_VERSION_TO_INSTALL_PYTEST['4.6']['pip_packages'] = [
-    'atomicwrites==1.4.1',
-    'attrs==23.1.0',
-    'more-itertools==10.1.0',
-    'packaging==23.1',
-    'pluggy==0.13.1',
-    'py==1.11.0',
-    'six==1.16.0',
-    'wcwidth==0.2.6',
-]
-for k in ['5.0', '5.1', '5.2']:
-    MAP_VERSION_TO_INSTALL_PYTEST[k]['pip_packages'] = [
-        'atomicwrites==1.4.1',
-        'attrs==23.1.0',
-        'more-itertools==10.1.0',
-        'packaging==23.1',
-        'pluggy==0.13.1',
-        'py==1.11.0',
-        'wcwidth==0.2.6',
-    ]
-MAP_VERSION_TO_INSTALL_PYTEST['5.3']['pip_packages'] = [
-    'attrs==23.1.0',
-    'more-itertools==10.1.0',
-    'packaging==23.1',
-    'pluggy==0.13.1',
-    'py==1.11.0',
-    'wcwidth==0.2.6',
-]
-MAP_VERSION_TO_INSTALL_PYTEST['5.4']['pip_packages'] = [
-    'py==1.11.0',
-    'packaging==23.1',
-    'attrs==23.1.0',
-    'more-itertools==10.1.0',
-    'pluggy==0.13.1',
-]
-MAP_VERSION_TO_INSTALL_PYTEST['6.0']['pip_packages'] = [
-    'attrs==23.1.0',
-    'iniconfig==2.0.0',
-    'more-itertools==10.1.0',
-    'packaging==23.1',
-    'pluggy==0.13.1',
-    'py==1.11.0',
-    'toml==0.10.2',
-]
-for k in ['6.2', '6.3']:
-    MAP_VERSION_TO_INSTALL_PYTEST[k]['pip_packages'] = [
-        'attrs==23.1.0',
-        'iniconfig==2.0.0',
-        'packaging==23.1',
-        'pluggy==0.13.1',
-        'py==1.11.0',
-        'toml==0.10.2',
-    ]
-MAP_VERSION_TO_INSTALL_PYTEST['7.0']['pip_packages'] = [
-    'attrs==23.1.0',
-    'iniconfig==2.0.0',
-    'packaging==23.1',
-    'pluggy==0.13.1',
-    'py==1.11.0',
-]
-for k in ['7.1', '7.2']:
-    MAP_VERSION_TO_INSTALL_PYTEST[k]['pip_packages'] = [
-        'attrs==23.1.0',
-        'iniconfig==2.0.0',
-        'packaging==23.1',
-        'pluggy==0.13.1',
-        'py==1.11.0',
-        'tomli==2.0.1',
-    ]
-MAP_VERSION_TO_INSTALL_PYTEST['7.4']['pip_packages'] = [
-    'iniconfig==2.0.0',
-    'packaging==23.1',
-    'pluggy==1.3.0',
-    'exceptiongroup==1.1.3',
-    'tomli==2.0.1',
-]
-MAP_VERSION_TO_INSTALL_PYTEST['8.0']['pip_packages'] = [
-    'iniconfig==2.0.0',
-    'packaging==23.1',
-    'pluggy==1.3.0',
-    'exceptiongroup==1.1.3',
-    'tomli==2.0.1',
-]
-MAP_VERSION_TO_INSTALL_MATPLOTLIB = {
-    k: {
-        'python': '3.11',
-        'packages': 'environment.yml',
-        'install': 'python -m pip install -e .',
-        'pre_install': [
-            'apt-get -y update && apt-get -y upgrade && apt-get install -y imagemagick ffmpeg texlive texlive-latex-extra texlive-fonts-recommended texlive-xetex texlive-luatex cm-super dvipng'
-        ],
-        'pip_packages': [
-            'contourpy==1.1.0',
-            'cycler==0.11.0',
-            'fonttools==4.42.1',
-            'ghostscript',
-            'kiwisolver==1.4.5',
-            'numpy==1.25.2',
-            'packaging==23.1',
-            'pillow==10.0.0',
-            'pikepdf',
-            'pyparsing==3.0.9',
-            'python-dateutil==2.8.2',
-            'six==1.16.0',
-            'setuptools==68.1.2',
-            'setuptools-scm==7.1.0',
-            'typing-extensions==4.7.1',
-        ],
-    }
-    for k in ['3.5', '3.6', '3.7']
-}
-MAP_VERSION_TO_INSTALL_MATPLOTLIB.update(
-    {
-        k: {
-            'python': '3.8',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-            'pre_install': [
-                'apt-get -y update && apt-get -y upgrade && apt-get install -y imagemagick ffmpeg libfreetype6-dev pkg-config texlive texlive-latex-extra texlive-fonts-recommended texlive-xetex texlive-luatex cm-super'
-            ],
-            'pip_packages': ['pytest', 'ipython'],
-        }
-        for k in ['3.1', '3.2', '3.3', '3.4']
-    }
-)
-MAP_VERSION_TO_INSTALL_MATPLOTLIB.update(
-    {
-        k: {
-            'python': '3.7',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-            'pre_install': [
-                'apt-get -y update && apt-get -y upgrade && apt-get install -y imagemagick ffmpeg libfreetype6-dev pkg-config'
-            ],
-            'pip_packages': ['pytest'],
-        }
-        for k in ['3.0']
-    }
-)
-MAP_VERSION_TO_INSTALL_MATPLOTLIB.update(
-    {
-        k: {
-            'python': '3.5',
-            'install': 'python setup.py build; python setup.py install',
-            'pre_install': [
-                'apt-get -y update && apt-get -y upgrade && && apt-get install -y imagemagick ffmpeg'
-            ],
-            'pip_packages': ['pytest'],
-            'execute_test_as_nonroot': True,
-        }
-        for k in ['2.0', '2.1', '2.2', '1.0', '1.1', '1.2', '1.3', '1.4', '1.5']
-    }
-)
-MAP_VERSION_TO_INSTALL_SPHINX = {
-    k: {
-        'python': '3.9',
-        'pip_packages': ['tox==4.16.0', 'tox-current-env==0.0.11'],
-        'install': 'python -m pip install -e .[test]',
-        'pre_install': ["sed -i 's/pytest/pytest -rA/' tox.ini"],
-    }
-    for k in ['1.5', '1.6', '1.7', '1.8', '2.0', '2.1', '2.2', '2.3', '2.4', '3.0']
-    + ['3.1', '3.2', '3.3', '3.4', '3.5', '4.0', '4.1', '4.2', '4.3', '4.4']
-    + ['4.5', '5.0', '5.1', '5.2', '5.3', '6.0', '6.2', '7.0', '7.1', '7.2']
-}
-for k in ['3.0', '3.1', '3.2', '3.3', '3.4', '3.5', '4.0', '4.1', '4.2', '4.3', '4.4']:
-    MAP_VERSION_TO_INSTALL_SPHINX[k]['pre_install'].extend(
-        [
-            "sed -i 's/Jinja2>=2.3/Jinja2<3.0/' setup.py",
-            "sed -i 's/sphinxcontrib-applehelp/sphinxcontrib-applehelp<=1.0.7/' setup.py",
-            "sed -i 's/sphinxcontrib-devhelp/sphinxcontrib-devhelp<=1.0.5/' setup.py",
-            "sed -i 's/sphinxcontrib-qthelp/sphinxcontrib-qthelp<=1.0.6/' setup.py",
-            "sed -i 's/alabaster>=0.7,<0.8/alabaster>=0.7,<0.7.12/' setup.py",
-            "sed -i \"s/'packaging',/'packaging', 'markupsafe<=2.0.1',/\" setup.py",
-        ]
-    )
-    if k in ['4.2', '4.3', '4.4']:
-        MAP_VERSION_TO_INSTALL_SPHINX[k]['pre_install'].extend(
-            [
-                "sed -i 's/sphinxcontrib-htmlhelp>=2.0.0/sphinxcontrib-htmlhelp>=2.0.0,<=2.0.4/' setup.py",
-                "sed -i 's/sphinxcontrib-serializinghtml>=1.1.5/sphinxcontrib-serializinghtml>=1.1.5,<=1.1.9/' setup.py",
-            ]
-        )
-    elif k == '4.1':
-        MAP_VERSION_TO_INSTALL_SPHINX[k]['pre_install'].extend(
-            [
-                (
-                    "grep -q 'sphinxcontrib-htmlhelp>=2.0.0' setup.py && "
-                    "sed -i 's/sphinxcontrib-htmlhelp>=2.0.0/sphinxcontrib-htmlhelp>=2.0.0,<=2.0.4/' setup.py || "
-                    "sed -i 's/sphinxcontrib-htmlhelp/sphinxcontrib-htmlhelp<=2.0.4/' setup.py"
-                ),
-                (
-                    "grep -q 'sphinxcontrib-serializinghtml>=1.1.5' setup.py && "
-                    "sed -i 's/sphinxcontrib-serializinghtml>=1.1.5/sphinxcontrib-serializinghtml>=1.1.5,<=1.1.9/' setup.py || "
-                    "sed -i 's/sphinxcontrib-serializinghtml/sphinxcontrib-serializinghtml<=1.1.9/' setup.py"
-                ),
-            ]
-        )
-    else:
-        MAP_VERSION_TO_INSTALL_SPHINX[k]['pre_install'].extend(
-            [
-                "sed -i 's/sphinxcontrib-htmlhelp/sphinxcontrib-htmlhelp<=2.0.4/' setup.py",
-                "sed -i 's/sphinxcontrib-serializinghtml/sphinxcontrib-serializinghtml<=1.1.9/' setup.py",
-            ]
-        )
-MAP_VERSION_TO_INSTALL_SPHINX['7.2']['pre_install'] += [
-    'apt-get update && apt-get install -y graphviz'
-]
-MAP_VERSION_TO_INSTALL_ASTROPY = {
-    k: {
-        'python': '3.9',
-        'install': 'python -m pip install -e .[test] --verbose',
-        'pip_packages': [
-            'attrs==23.1.0',
-            'exceptiongroup==1.1.3',
-            'execnet==2.0.2',
-            'hypothesis==6.82.6',
-            'iniconfig==2.0.0',
-            'numpy==1.25.2',
-            'packaging==23.1',
-            'pluggy==1.3.0',
-            'psutil==5.9.5',
-            'pyerfa==2.0.0.3',
-            'pytest-arraydiff==0.5.0',
-            'pytest-astropy-header==0.2.2',
-            'pytest-astropy==0.10.0',
-            'pytest-cov==4.1.0',
-            'pytest-doctestplus==1.0.0',
-            'pytest-filter-subpackage==0.1.2',
-            'pytest-mock==3.11.1',
-            'pytest-openfiles==0.5.0',
-            'pytest-remotedata==0.4.0',
-            'pytest-xdist==3.3.1',
-            'pytest==7.4.0',
-            'PyYAML==6.0.1',
-            'setuptools==68.0.0',
-            'sortedcontainers==2.4.0',
-            'tomli==2.0.1',
-        ],
-    }
-    for k in ['0.1', '0.2', '0.3', '0.4', '1.1', '1.2', '1.3', '3.0', '3.1', '3.2']
-    + ['4.1', '4.2', '4.3', '5.0', '5.1', '5.2']
-}
-for k in ['4.1', '4.2', '4.3', '5.0', '5.1', '5.2']:
-    MAP_VERSION_TO_INSTALL_ASTROPY[k]['pre_install'] = [
-        'sed -i \'s/requires = \\["setuptools",/requires = \\["setuptools==68.0.0",/\' pyproject.toml'
-    ]
-MAP_VERSION_TO_INSTALL_SYMPY = {
-    k: {
-        'python': '3.9',
-        'packages': 'mpmath flake8',
-        'pip_packages': ['mpmath==1.3.0', 'flake8-comprehensions'],
-        'install': 'python -m pip install -e .',
-    }
-    for k in ['0.7', '1.0', '1.1', '1.10', '1.11', '1.12', '1.2', '1.4', '1.5', '1.6']
-    + ['1.7', '1.8', '1.9']
-}
-MAP_VERSION_TO_INSTALL_SYMPY.update(
-    {
-        k: {
-            'python': '3.9',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-            'pip_packages': ['mpmath==1.3.0'],
-        }
-        for k in ['1.13']
-    }
-)
-MAP_VERSION_TO_INSTALL_PYLINT = {
-    k: {
-        'python': '3.9',
-        'packages': 'requirements.txt',
-        'install': 'python -m pip install -e .',
-    }
-    for k in [
-        '2.10',
-        '2.11',
-        '2.13',
-        '2.14',
-        '2.15',
-        '2.16',
-        '2.17',
-        '2.8',
-        '2.9',
-        '3.0',
-    ]
-}
-MAP_VERSION_TO_INSTALL_PYLINT['2.8']['pip_packages'] = ['pyenchant==3.2']
-MAP_VERSION_TO_INSTALL_PYLINT['2.8']['pre_install'] = [
-    'apt-get update && apt-get install -y libenchant-2-dev hunspell-en-us'
-]
-MAP_VERSION_TO_INSTALL_PYLINT.update(
-    {
-        k: {
-            **MAP_VERSION_TO_INSTALL_PYLINT[k],
-            'pip_packages': ['astroid==3.0.0a6', 'setuptools'],
-        }
-        for k in ['3.0']
-    }
-)
-
-MAP_VERSION_TO_INSTALL_XARRAY = {
-    k: {
-        'python': '3.10',
-        'packages': 'environment.yml',
-        'install': 'python -m pip install -e .',
-        'pip_packages': [
-            'numpy==1.23.0',
-            'packaging==23.1',
-            'pandas==1.5.3',
-            'pytest==7.4.0',
-            'python-dateutil==2.8.2',
-            'pytz==2023.3',
-            'six==1.16.0',
-            'scipy==1.11.1',
-            'setuptools==68.0.0',
-        ],
-        'no_use_env': True,
-    }
-    for k in ['0.12', '0.18', '0.19', '0.20', '2022.03', '2022.06', '2022.09']
-}
-
-MAP_VERSION_TO_INSTALL_SQLFLUFF = {
-    k: {
-        'python': '3.9',
-        'packages': 'requirements.txt',
-        'install': 'python -m pip install -e .',
-    }
-    for k in [
-        '0.10',
-        '0.11',
-        '0.12',
-        '0.13',
-        '0.4',
-        '0.5',
-        '0.6',
-        '0.8',
-        '0.9',
-        '1.0',
-        '1.1',
-        '1.2',
-        '1.3',
-        '1.4',
-        '2.0',
-        '2.1',
-        '2.2',
-    ]
-}
-MAP_VERSION_TO_INSTALL_DBT_CORE = {
-    k: {
-        'python': '3.9',
-        'packages': 'requirements.txt',
-        'install': 'python -m pip install -e .',
-    }
-    for k in [
-        '0.13',
-        '0.14',
-        '0.15',
-        '0.16',
-        '0.17',
-        '0.18',
-        '0.19',
-        '0.20',
-        '0.21',
-        '1.0',
-        '1.1',
-        '1.2',
-        '1.3',
-        '1.4',
-        '1.5',
-        '1.6',
-        '1.7',
-    ]
-}
-MAP_VERSION_TO_INSTALL_PYVISTA = {
-    k: {
-        'python': '3.9',
-        'install': 'python -m pip install -e .',
-        'pip_packages': ['pytest'],
-    }
-    for k in ['0.20', '0.21', '0.22', '0.23']
-}
-MAP_VERSION_TO_INSTALL_PYVISTA.update(
-    {
-        k: {
-            'python': '3.9',
-            'packages': 'requirements.txt',
-            'install': 'python -m pip install -e .',
-            'pip_packages': ['pytest'],
-        }
-        for k in [
-            '0.24',
-            '0.25',
-            '0.26',
-            '0.27',
-            '0.28',
-            '0.29',
-            '0.30',
-            '0.31',
-            '0.32',
-            '0.33',
-            '0.34',
-            '0.35',
-            '0.36',
-            '0.37',
-            '0.38',
-            '0.39',
-            '0.40',
-            '0.41',
-            '0.42',
-            '0.43',
-        ]
-    }
-)
-MAP_VERSION_TO_INSTALL_ASTROID = {
-    k: {
-        'python': '3.9',
-        'install': 'python -m pip install -e .',
-        'pip_packages': ['pytest'],
-    }
-    for k in [
-        '2.10',
-        '2.12',
-        '2.13',
-        '2.14',
-        '2.15',
-        '2.16',
-        '2.5',
-        '2.6',
-        '2.7',
-        '2.8',
-        '2.9',
-        '3.0',
-    ]
-}
-MAP_VERSION_TO_INSTALL_MARSHMALLOW = {
-    k: {
-        'python': '3.9',
-        'install': "python -m pip install -e '.[dev]'",
-    }
-    for k in [
-        '2.18',
-        '2.19',
-        '2.20',
-        '3.0',
-        '3.1',
-        '3.10',
-        '3.11',
-        '3.12',
-        '3.13',
-        '3.15',
-        '3.16',
-        '3.19',
-        '3.2',
-        '3.4',
-        '3.8',
-        '3.9',
-    ]
-}
-MAP_VERSION_TO_INSTALL_PVLIB = {
-    k: {
-        'python': '3.9',
-        'install': 'python -m pip install -e .[all]',
-        'packages': 'pandas scipy',
-        'pip_packages': ['jupyter', 'ipython', 'matplotlib', 'pytest', 'flake8'],
-    }
-    for k in ['0.1', '0.2', '0.3', '0.4', '0.5', '0.6', '0.7', '0.8', '0.9']
-}
-MAP_VERSION_TO_INSTALL_PYDICOM = {
-    k: {'python': '3.6', 'install': 'python -m pip install -e .', 'packages': 'numpy'}
-    for k in [
-        '1.0',
-        '1.1',
-        '1.2',
-        '1.3',
-        '1.4',
-        '2.0',
-        '2.1',
-        '2.2',
-        '2.3',
-        '2.4',
-        '3.0',
-    ]
-}
-MAP_VERSION_TO_INSTALL_PYDICOM.update(
-    {k: {**MAP_VERSION_TO_INSTALL_PYDICOM[k], 'python': '3.8'} for k in ['1.4', '2.0']}
-)
-MAP_VERSION_TO_INSTALL_PYDICOM.update(
-    {k: {**MAP_VERSION_TO_INSTALL_PYDICOM[k], 'python': '3.9'} for k in ['2.1', '2.2']}
-)
-MAP_VERSION_TO_INSTALL_PYDICOM.update(
-    {k: {**MAP_VERSION_TO_INSTALL_PYDICOM[k], 'python': '3.10'} for k in ['2.3']}
-)
-MAP_VERSION_TO_INSTALL_PYDICOM.update(
-    {k: {**MAP_VERSION_TO_INSTALL_PYDICOM[k], 'python': '3.11'} for k in ['2.4', '3.0']}
-)
-MAP_VERSION_TO_INSTALL_HUMANEVAL = {k: {'python': '3.9'} for k in ['1.0']}
-MAP_VERSION_TO_INSTALL_HUMANEVAL_FIX = {
-    k: {'python': '3.10', 'packages': 'pytest'} for k in ['0.0.1']
-}
-
-# Constants - Task Instance Instllation Environment
-MAP_VERSION_TO_INSTALL = {
-    'astropy/astropy': MAP_VERSION_TO_INSTALL_ASTROPY,
-    'dbt-labs/dbt-core': MAP_VERSION_TO_INSTALL_DBT_CORE,
-    'django/django': MAP_VERSION_TO_INSTALL_DJANGO,
-    'matplotlib/matplotlib': MAP_VERSION_TO_INSTALL_MATPLOTLIB,
-    'marshmallow-code/marshmallow': MAP_VERSION_TO_INSTALL_MARSHMALLOW,
-    'mwaskom/seaborn': MAP_VERSION_TO_INSTALL_SEABORN,
-    'pallets/flask': MAP_VERSION_TO_INSTALL_FLASK,
-    'psf/requests': MAP_VERSION_TO_INSTALL_REQUESTS,
-    'pvlib/pvlib-python': MAP_VERSION_TO_INSTALL_PVLIB,
-    'pydata/xarray': MAP_VERSION_TO_INSTALL_XARRAY,
-    'pydicom/pydicom': MAP_VERSION_TO_INSTALL_PYDICOM,
-    'pylint-dev/astroid': MAP_VERSION_TO_INSTALL_ASTROID,
-    'pylint-dev/pylint': MAP_VERSION_TO_INSTALL_PYLINT,
-    'pytest-dev/pytest': MAP_VERSION_TO_INSTALL_PYTEST,
-    'pyvista/pyvista': MAP_VERSION_TO_INSTALL_PYVISTA,
-    'scikit-learn/scikit-learn': MAP_VERSION_TO_INSTALL_SKLEARN,
-    'sphinx-doc/sphinx': MAP_VERSION_TO_INSTALL_SPHINX,
-    'sqlfluff/sqlfluff': MAP_VERSION_TO_INSTALL_SQLFLUFF,
-    'swe-bench/humaneval': MAP_VERSION_TO_INSTALL_HUMANEVAL,
-    'nielstron/humaneval_fix': MAP_VERSION_TO_INSTALL_HUMANEVAL_FIX,
-    'sympy/sympy': MAP_VERSION_TO_INSTALL_SYMPY,
-}
-
-# Constants - Repository Specific Installation Instructions
-MAP_REPO_TO_INSTALL = {}
-
-# Constants - Task Instance Test Frameworks
-TEST_PYTEST_VERBOSE = 'pytest -rA --tb=long -p no:cacheprovider'
-MAP_REPO_TO_TEST_FRAMEWORK_VERBOSE = {
-    'astropy/astropy': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_ASTROPY.keys()
-    },
-    'django/django': {
-        k: './tests/runtests.py --verbosity 2 --settings=test_sqlite --parallel 1'
-        for k in MAP_VERSION_TO_INSTALL_DJANGO.keys()
-    },
-    'marshmallow-code/marshmallow': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_MARSHMALLOW.keys()
-    },
-    'matplotlib/matplotlib': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_MATPLOTLIB.keys()
-    },
-    'mwaskom/seaborn': {
-        k: 'pytest -rA --tb=long' for k in MAP_VERSION_TO_INSTALL_SEABORN.keys()
-    },
-    'pallets/flask': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_FLASK.keys()
-    },
-    'psf/requests': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_REQUESTS.keys()
-    },
-    'pvlib/pvlib-python': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_PVLIB.keys()
-    },
-    'pydata/xarray': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_XARRAY.keys()
-    },
-    'pydicom/pydicom': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_PYDICOM.keys()
-    },
-    'pylint-dev/astroid': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_ASTROID.keys()
-    },
-    'pylint-dev/pylint': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_PYLINT.keys()
-    },
-    'pytest-dev/pytest': {
-        k: 'pytest -rA --tb=long' for k in MAP_VERSION_TO_INSTALL_PYTEST.keys()
-    },
-    'pyvista/pyvista': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_PYVISTA.keys()
-    },
-    'scikit-learn/scikit-learn': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_SKLEARN.keys()
-    },
-    'sphinx-doc/sphinx': {
-        k: 'tox -epy39 -v --' for k in MAP_VERSION_TO_INSTALL_SPHINX.keys()
-    },
-    'sqlfluff/sqlfluff': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_SQLFLUFF.keys()
-    },
-    'swe-bench/humaneval': {
-        k: 'python' for k in MAP_VERSION_TO_INSTALL_HUMANEVAL.keys()
-    },
-    'nielstron/humaneval_fix': {
-        k: TEST_PYTEST_VERBOSE for k in MAP_VERSION_TO_INSTALL_HUMANEVAL.keys()
-    },
-    'sympy/sympy': {
-        k: 'bin/test -C --verbose' for k in MAP_VERSION_TO_INSTALL_SYMPY.keys()
-    },
-}
-MAP_REPO_TO_TEST_FRAMEWORK_VERBOSE['django/django']['1.9'] = (
-    './tests/runtests.py --verbosity 2'
-)

evaluation/benchmarks/swe_perf/run_infer.py

@@ -1,978 +0,0 @@
-import asyncio
-import copy
-import json
-import os
-import tempfile
-from typing import Any, Literal
-
-import pandas as pd
-import toml
-from datasets import load_dataset
-
-import openhands.agenthub
-from evaluation.benchmarks.swe_perf.binary_patch_utils import (
-    remove_binary_diffs,
-    remove_binary_files_from_git,
-)
-from evaluation.benchmarks.swe_perf.resource.mapping import (
-    get_instance_resource_factor,
-)
-from evaluation.benchmarks.swe_perf.resource.swt_bench_constants import (
-    MAP_REPO_TO_INSTALL,
-    MAP_VERSION_TO_INSTALL,
-)
-from evaluation.utils.shared import (
-    EvalException,
-    EvalMetadata,
-    EvalOutput,
-    assert_and_raise,
-    check_maximum_retries_exceeded,
-    codeact_user_response,
-    get_default_sandbox_config_for_eval,
-    get_metrics,
-    is_fatal_evaluation_error,
-    make_metadata,
-    prepare_dataset,
-    reset_logger_for_multiprocessing,
-    run_evaluation,
-    update_llm_config_for_completions_logging,
-)
-from openhands.controller.state.state import State
-from openhands.core.config import (
-    AgentConfig,
-    OpenHandsConfig,
-    get_evaluation_parser,
-    get_llm_config_arg,
-)
-from openhands.core.config.condenser_config import NoOpCondenserConfig
-from openhands.core.config.utils import get_condenser_config_arg
-from openhands.core.logger import openhands_logger as logger
-from openhands.core.main import create_runtime, run_controller
-from openhands.critic import AgentFinishedCritic
-from openhands.events.action import CmdRunAction, FileReadAction, MessageAction
-from openhands.events.observation import (
-    CmdOutputObservation,
-    ErrorObservation,
-    FileReadObservation,
-)
-from openhands.events.serialization.event import event_from_dict, event_to_dict
-from openhands.runtime.base import Runtime
-from openhands.utils.async_utils import call_async_from_sync
-from openhands.utils.shutdown_listener import sleep_if_should_continue
-
-USE_HINT_TEXT = os.environ.get('USE_HINT_TEXT', 'false').lower() == 'true'
-RUN_WITH_BROWSING = os.environ.get('RUN_WITH_BROWSING', 'false').lower() == 'true'
-ENABLE_LLM_EDITOR = os.environ.get('ENABLE_LLM_EDITOR', 'false').lower() == 'true'
-BenchMode = Literal['swe', 'swt', 'swt-ci']
-
-# Global variable to track dataset type
-DATASET_TYPE = 'SWE-Perf'
-
-
-AGENT_CLS_TO_FAKE_USER_RESPONSE_FN = {
-    'CodeActAgent': codeact_user_response,
-}
-
-
-def _get_sweperf_workspace_dir_name(instance: pd.Series) -> str:
-    return f'{instance.repo}__{instance.version}'.replace('/', '__')
-
-
-def get_instruction(instance: pd.Series, metadata: EvalMetadata) -> MessageAction:
-    workspace_dir_name = _get_sweperf_workspace_dir_name(instance)
-
-    # The instruction
-    instruction = f"""
-<uploaded_files>
-/workspace/{workspace_dir_name}
-</uploaded_files>
-
-I've uploaded a python code repository in the directory {workspace_dir_name}. Consider the following issue description:
-
-
-<issue_description>
-{instance.problem_statement_realistic}
-</issue_description>
-
-Can you help me implement the necessary changes to the repository so that the requirements specified in the <issue_description> are met?
-I've already taken care of all changes to any of the test files described in the <issue_description>. This means you DON'T have to modify the testing logic or any of the tests in any way!
-Also the development Python environment is already set up for you (i.e., all dependencies already installed), so you don't need to install other packages.
-Your task is to make the minimal changes to non-test files in the /workspace/{workspace_dir_name} directory to ensure the <issue_description> is satisfied.
-
-Follow these phases to resolve the issue:
-
-## ⚙️ Phase 1: Understand the Problem & Test Reuse
-
-**1.1. Install the package locally:**
-
-```bash
-python -m pip install pyinstrument
-python -m pip install -e .
-```
-
-> Only proceed to README-based install if the above fails.
-
-**1.2. Identify relevant modules and logic:**
-
-* Use test cases mentioned in `<issue_description>` to locate the functions and files involved.
-* Focus on potential performance bottlenecks: loops, I/O, locks, cache access, data structures, etc.
-
-**1.3. Run initial benchmark:**
-
-```bash
-pytest -rA --durations=0 --disable-warnings -p no:warnings --tb=no <test_case>
-```
-
-## 📊 Phase 2: Localization (Hierarchical Bottleneck Detection)
-
-**2.1. Global profiling using `pyinstrument`:**
-
-```bash
-pyinstrument -m pytest -rA --durations=0 --disable-warnings --tb=no --continue-on-collection-errors -p no:warnings <test_case>
-```
-
-**2.2. Analyze performance stack if necessary:**
-
-* 🔍 **Module level**: Identify hot files and methods.
-* 🔬 **Function level**: Focus on top-consuming classes/functions.
-* 🧬 **Line level**: Add fine-grained sampling/logging if needed.
-
-**2.3. Output a layered summary** showing where time is spent and why.
-
-
-## 🧠 Phase 3: Repair (Design Candidate Fixes)
-
-**3.1. Propose multiple optimization ideas:**
-
-* Algorithm refinement
-* Data structure improvement
-* Parallelism / async
-* Caching / batching
-
-**3.2. For each candidate:**
-
-* Describe the idea using pseudocode or `diff`
-* Evaluate expected gain vs implementation complexity
-
----
-
-## 🔬 Phase 4: Patch Validation (Quantitative Benchmarking)
-
-**4.1. Apply each patch separately**
-
-**4.2. Re-run benchmark using the same test case**
-
-**4.3. Record improvements in pytest `call` phase**
-
-**4.4. Build a comparison table:**
-
-| Patch | Call Time (s) | % Improvement | Notes |
-| ----- | ------------- | ------------- | ----- |
-
-**4.5. Select the best-performing, low-risk patch**
-
----
-
-## 📌 Phase 5: Fix Analysis (Explain the Fix)
-
-**5.1.** Clearly describe the **problem**
-
-**5.2.** Identify the **exact location** in the code
-
-**5.3.** Describe **how the test reveals** the problem
-
-**5.4.** State **best practices** relevant to the fix
-
-**5.5.** Explain **how the fix resolves** the issue
-
----
-
-## 🧩 Phase 6: Fix Implementation
-
-**6.1.** Implement the selected fix in the non-test source files
-**6.2.** Ensure the changes are minimal and performance-focused
-
----
-
-## ✅ Phase 7: Final Verification & Review
-
-**7.1.** Run **full regression tests** to ensure correctness
-
-**7.2.** Re-benchmark performance and compare with base commit `{instance['base_commit']}`
-
-**7.3.** Ensure `call` time is significantly improved in `pytest` output
-
-**7.4.** If gain is limited, consider deeper optimization
-
-**7.5.** Write a concise final report including:
-
-* Bottleneck path
-* Fix alternatives considered
-* Final solution and metrics
-* Code diff summary
-
-## Important ##
-1. Please do not optimize specifically for certain test cases, as this may be considered cheating.
-2. Delete all files unrelated to optimization, including temporary files, auxiliary test code, and so on.
-3. Avoid excessive optimization and unnecessary divergence; if the improvement is not significant, stop promptly to maintain efficiency and focus.
-
-Be thorough in your exploration, testing, and reasoning. It's fine if your thinking process is lengthy - quality and completeness are more important than brevity.
-"""
-
-    if RUN_WITH_BROWSING:
-        instruction += (
-            '<IMPORTANT!>\nYou SHOULD NEVER attempt to browse the web. </IMPORTANT!>\n'
-        )
-
-    if 'image_assets' in instance:
-        assets = json.loads(instance['image_assets'])
-        assert 'problem_statement' in assets, (
-            'problem_statement is required in image_assets'
-        )
-        image_urls = assets['problem_statement']
-        return MessageAction(content=instruction, image_urls=image_urls)
-    return MessageAction(content=instruction)
-
-
-def get_instance_docker_image(
-    instance_id: str,
-) -> str:
-    docker_image_prefix = 'docker.io/betty1202/'
-    image_name = 'sweb.eval.x86_64.' + instance_id
-    image_name = image_name.replace(
-        '__', '_s_'
-    )  # to comply with docker image naming convention
-    return (docker_image_prefix.rstrip('/') + '/' + image_name).lower()
-
-
-def get_config(
-    instance: pd.Series,
-    metadata: EvalMetadata,
-) -> OpenHandsConfig:
-    base_container_image = get_instance_docker_image(
-        instance['instance_id'],
-    )
-    logger.info(
-        f'Using instance container image: {base_container_image}. '
-        f'Please make sure this image exists. '
-        f'Submit an issue on https://github.com/All-Hands-AI/OpenHands if you run into any issues.'
-    )
-
-    sandbox_config = get_default_sandbox_config_for_eval()
-    sandbox_config.base_container_image = base_container_image
-    sandbox_config.enable_auto_lint = True
-    sandbox_config.use_host_network = False
-    # Add platform to the sandbox config to solve issue 4401
-    sandbox_config.platform = 'linux/amd64'
-    sandbox_config.remote_runtime_resource_factor = get_instance_resource_factor(
-        dataset_name=metadata.dataset,
-        instance_id=instance['instance_id'],
-    )
-
-    config = OpenHandsConfig(
-        default_agent=metadata.agent_class,
-        run_as_openhands=False,
-        max_iterations=metadata.max_iterations,
-        enable_browser=RUN_WITH_BROWSING,
-        runtime=os.environ.get('RUNTIME', 'docker'),
-        sandbox=sandbox_config,
-        # do not mount workspace
-        workspace_base=None,
-        workspace_mount_path=None,
-    )
-
-    config.set_llm_config(
-        update_llm_config_for_completions_logging(
-            metadata.llm_config, metadata.eval_output_dir, instance['instance_id']
-        )
-    )
-    # get 'draft_editor' config if exists
-    config.set_llm_config(get_llm_config_arg('draft_editor'), 'draft_editor')
-
-    agent_config = AgentConfig(
-        enable_jupyter=False,
-        enable_browsing=RUN_WITH_BROWSING,
-        enable_llm_editor=ENABLE_LLM_EDITOR,
-        enable_mcp=False,
-        condenser=metadata.condenser_config,
-        enable_prompt_extensions=False,
-    )
-    config.set_agent_config(agent_config)
-    return config
-
-
-def initialize_runtime(
-    runtime: Runtime,
-    instance: pd.Series,  # this argument is not required
-    metadata: EvalMetadata,
-):
-    """Initialize the runtime for the agent.
-
-    This function is called before the runtime is used to run the agent.
-    """
-    logger.info('-' * 30)
-    logger.info('BEGIN Runtime Initialization Fn')
-    logger.info('-' * 30)
-    workspace_dir_name = _get_sweperf_workspace_dir_name(instance)
-    obs: CmdOutputObservation
-
-    # Set instance id and git configuration
-    action = CmdRunAction(
-        command=f"""echo 'export SWE_INSTANCE_ID={instance['instance_id']}' >> ~/.bashrc && echo 'export PIP_CACHE_DIR=~/.cache/pip' >> ~/.bashrc && echo "alias git='git --no-pager'" >> ~/.bashrc && git config --global core.pager "" && git config --global diff.binary false"""
-    )
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        obs.exit_code == 0,
-        f'Failed to export SWE_INSTANCE_ID and configure git: {str(obs)}',
-    )
-
-    action = CmdRunAction(command="""export USER=$(whoami); echo USER=${USER} """)
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(obs.exit_code == 0, f'Failed to export USER: {str(obs)}')
-
-    # inject the init script
-    script_dir = os.path.dirname(__file__)
-
-    # inject the instance info
-    action = CmdRunAction(command='mkdir -p /swe_util/eval_data/instances')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        obs.exit_code == 0,
-        f'Failed to create /swe_util/eval_data/instances: {str(obs)}',
-    )
-
-    swe_instance_json_name = 'swe-perf-instance.json'
-    with tempfile.TemporaryDirectory() as temp_dir:
-        # Construct the full path for the desired file name within the temporary directory
-        temp_file_path = os.path.join(temp_dir, swe_instance_json_name)
-        # Write to the file with the desired name within the temporary directory
-        with open(temp_file_path, 'w') as f:
-            if not isinstance(instance, dict):
-                json.dump([instance.to_dict()], f)
-            else:
-                json.dump([instance], f)
-
-        # Copy the file to the desired location
-        runtime.copy_to(temp_file_path, '/swe_util/eval_data/instances/')
-
-        # inject the instance swe entry
-        entry_script_path = 'instance_swe_entry.sh'
-        runtime.copy_to(
-            str(os.path.join(script_dir, f'scripts/setup/{entry_script_path}')),
-            '/swe_util/',
-        )
-
-    action = CmdRunAction(command='cat ~/.bashrc')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(obs.exit_code == 0, f'Failed to cat ~/.bashrc: {str(obs)}')
-
-    action = CmdRunAction(command='source ~/.bashrc')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    if isinstance(obs, ErrorObservation):
-        logger.error(f'Failed to source ~/.bashrc: {str(obs)}')
-    assert_and_raise(obs.exit_code == 0, f'Failed to source ~/.bashrc: {str(obs)}')
-
-    action = CmdRunAction(command=f'source /swe_util/{entry_script_path}')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        obs.exit_code == 0,
-        f'Failed to source /swe_util/{entry_script_path}: {str(obs)}',
-    )
-
-    action = CmdRunAction(command=f'cd /workspace/{workspace_dir_name}')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        obs.exit_code == 0,
-        f'Failed to cd to /workspace/{workspace_dir_name}: {str(obs)}',
-    )
-
-    action = CmdRunAction(command='git reset --hard')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(obs.exit_code == 0, f'Failed to git reset --hard: {str(obs)}')
-
-    action = CmdRunAction(
-        command='for remote_name in $(git remote); do git remote remove "${remote_name}"; done'
-    )
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(obs.exit_code == 0, f'Failed to remove git remotes: {str(obs)}')
-
-    if metadata.details['mode'] == 'swt-ci':
-        # set up repo
-        setup_commands = []
-        if instance['repo'] in MAP_REPO_TO_INSTALL:
-            setup_commands.append(MAP_REPO_TO_INSTALL[instance['repo']])
-
-        # Run pre-install set up if provided
-        install = MAP_VERSION_TO_INSTALL.get(instance['repo'], {}).get(
-            instance['version'], []
-        )
-        if 'pre_install' in install:
-            for pre_install in install['pre_install']:
-                setup_commands.append(pre_install)
-
-        if 'install' in install:
-            setup_commands.append(install['install'])
-
-        for command in setup_commands:
-            action = CmdRunAction(command=command)
-            action.set_hard_timeout(600)
-            logger.info(action, extra={'msg_type': 'ACTION'})
-            obs = runtime.run_action(action)
-            logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-
-    action = CmdRunAction(command='which python')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        obs.exit_code == 0 and 'testbed' in obs.content,
-        f'Expected to find python interpreter from testbed, but got: {str(obs)}',
-    )
-
-    logger.info('-' * 30)
-    logger.info('END Runtime Initialization Fn')
-    logger.info('-' * 30)
-
-
-def complete_runtime(
-    runtime: Runtime,
-    instance: pd.Series,  # this argument is not required, but it is used to get the workspace_dir_name
-) -> dict[str, Any]:
-    """Complete the runtime for the agent.
-
-    This function is called before the runtime is used to run the agent.
-    If you need to do something in the sandbox to get the correctness metric after
-    the agent has run, modify this function.
-    """
-    logger.info('-' * 30)
-    logger.info('BEGIN Runtime Completion Fn')
-    logger.info('-' * 30)
-    obs: CmdOutputObservation
-    workspace_dir_name = _get_sweperf_workspace_dir_name(instance)
-
-    action = CmdRunAction(command=f'cd /workspace/{workspace_dir_name}')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-
-    if obs.exit_code == -1:
-        # The previous command is still running
-        # We need to kill previous command
-        logger.info('The previous command is still running, trying to kill it...')
-        action = CmdRunAction(command='C-c')
-        obs = runtime.run_action(action)
-        logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-
-        # Then run the command again
-        action = CmdRunAction(command=f'cd /workspace/{workspace_dir_name}')
-        action.set_hard_timeout(600)
-        logger.info(action, extra={'msg_type': 'ACTION'})
-        obs = runtime.run_action(action)
-        logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-
-    if obs.exit_code == -1:
-        # The previous command is still running
-        # We need to kill previous command
-        logger.info('The previous command is still running, trying to ctrl+z it...')
-        action = CmdRunAction(command='C-z')
-        obs = runtime.run_action(action)
-        logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-
-        # Then run the command again
-        action = CmdRunAction(command=f'cd /workspace/{workspace_dir_name}')
-        action.set_hard_timeout(600)
-        logger.info(action, extra={'msg_type': 'ACTION'})
-        obs = runtime.run_action(action)
-        logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-
-    assert_and_raise(
-        isinstance(obs, CmdOutputObservation) and obs.exit_code == 0,
-        f'Failed to cd to /workspace/{workspace_dir_name}: {str(obs)}',
-    )
-
-    action = CmdRunAction(command='git config --global core.pager ""')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        isinstance(obs, CmdOutputObservation) and obs.exit_code == 0,
-        f'Failed to git config --global core.pager "": {str(obs)}',
-    )
-
-    # First check for any git repositories in subdirectories
-    action = CmdRunAction(command='find . -type d -name .git -not -path "./.git"')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        isinstance(obs, CmdOutputObservation) and obs.exit_code == 0,
-        f'Failed to find git repositories: {str(obs)}',
-    )
-
-    git_dirs = [p for p in obs.content.strip().split('\n') if p]
-    if git_dirs:
-        # Remove all .git directories in subdirectories
-        for git_dir in git_dirs:
-            action = CmdRunAction(command=f'rm -rf "{git_dir}"')
-            action.set_hard_timeout(600)
-            logger.info(action, extra={'msg_type': 'ACTION'})
-            obs = runtime.run_action(action)
-            logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-            assert_and_raise(
-                isinstance(obs, CmdOutputObservation) and obs.exit_code == 0,
-                f'Failed to remove git directory {git_dir}: {str(obs)}',
-            )
-
-    # add all files
-    action = CmdRunAction(command='git add -A')
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        isinstance(obs, CmdOutputObservation) and obs.exit_code == 0,
-        f'Failed to git add -A: {str(obs)}',
-    )
-
-    # Remove binary files from git staging
-    action = CmdRunAction(command=remove_binary_files_from_git())
-    action.set_hard_timeout(600)
-    logger.info(action, extra={'msg_type': 'ACTION'})
-    obs = runtime.run_action(action)
-    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-    assert_and_raise(
-        isinstance(obs, CmdOutputObservation) and obs.exit_code == 0,
-        f'Failed to remove binary files: {str(obs)}',
-    )
-
-    n_retries = 0
-    git_patch = None
-    while n_retries < 5:
-        action = CmdRunAction(
-            command=f'git diff --no-color --cached {instance["base_commit"]} > patch.diff'
-        )
-        action.set_hard_timeout(max(300 + 100 * n_retries, 600))
-        logger.info(action, extra={'msg_type': 'ACTION'})
-        obs = runtime.run_action(action)
-        logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-        n_retries += 1
-        if isinstance(obs, CmdOutputObservation):
-            if obs.exit_code == 0:
-                # Read the patch file
-                action = FileReadAction(path='patch.diff')
-                action.set_hard_timeout(max(300 + 100 * n_retries, 600))
-                logger.info(action, extra={'msg_type': 'ACTION'})
-                obs = runtime.run_action(action)
-                logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-                if isinstance(obs, FileReadObservation):
-                    git_patch = obs.content
-                    break
-                elif isinstance(obs, ErrorObservation):
-                    # Fall back to cat "patch.diff" to get the patch
-                    assert 'File could not be decoded as utf-8' in obs.content
-                    action = CmdRunAction(command='cat patch.diff')
-                    action.set_hard_timeout(max(300 + 100 * n_retries, 600))
-                    logger.info(action, extra={'msg_type': 'ACTION'})
-                    obs = runtime.run_action(action)
-                    assert isinstance(obs, CmdOutputObservation) and obs.exit_code == 0
-                    logger.info(obs, extra={'msg_type': 'OBSERVATION'})
-                    git_patch = obs.content
-                    break
-                else:
-                    assert_and_raise(False, f'Unexpected observation type: {str(obs)}')
-            else:
-                logger.info('Failed to get git diff, retrying...')
-                sleep_if_should_continue(10)
-        elif isinstance(obs, ErrorObservation):
-            logger.error(f'Error occurred: {obs.content}. Retrying...')
-            sleep_if_should_continue(10)
-        else:
-            assert_and_raise(False, f'Unexpected observation type: {str(obs)}')
-
-    assert_and_raise(git_patch is not None, 'Failed to get git diff (None)')
-
-    # Remove binary diffs from the patch
-    git_patch = remove_binary_diffs(git_patch)
-
-    logger.info('-' * 30)
-    logger.info('END Runtime Completion Fn')
-    logger.info('-' * 30)
-    return {'git_patch': git_patch}
-
-
-def process_instance(
-    instance: pd.Series,
-    metadata: EvalMetadata,
-    reset_logger: bool = True,
-    runtime_failure_count: int = 0,
-) -> EvalOutput:
-    config = get_config(instance, metadata)
-
-    # Setup the logger properly, so you can run multi-processing to parallelize the evaluation
-    if reset_logger:
-        log_dir = os.path.join(metadata.eval_output_dir, 'infer_logs')
-        reset_logger_for_multiprocessing(logger, instance.instance_id, log_dir)
-    else:
-        logger.info(f'Starting evaluation for instance {instance.instance_id}.')
-
-    # Increase resource_factor with increasing attempt_id
-    if runtime_failure_count > 0:
-        config.sandbox.remote_runtime_resource_factor = min(
-            config.sandbox.remote_runtime_resource_factor * (2**runtime_failure_count),
-            8,
-        )
-        logger.warning(
-            f'This is the {runtime_failure_count + 1}th attempt for instance {instance.instance_id}, setting resource factor to {config.sandbox.remote_runtime_resource_factor}'
-        )
-
-    metadata = copy.deepcopy(metadata)
-    metadata.details['runtime_failure_count'] = runtime_failure_count
-    metadata.details['remote_runtime_resource_factor'] = (
-        config.sandbox.remote_runtime_resource_factor
-    )
-
-    runtime = create_runtime(config)
-    call_async_from_sync(runtime.connect)
-
-    try:
-        initialize_runtime(runtime, instance, metadata)
-
-        message_action = get_instruction(instance, metadata)
-
-        # Here's how you can run the agent (similar to the `main` function) and get the final task state
-        state: State | None = asyncio.run(
-            run_controller(
-                config=config,
-                initial_user_action=message_action,
-                runtime=runtime,
-                fake_user_response_fn=AGENT_CLS_TO_FAKE_USER_RESPONSE_FN[
-                    metadata.agent_class
-                ],
-            )
-        )
-
-        # if fatal error, throw EvalError to trigger re-run
-        if is_fatal_evaluation_error(state.last_error):
-            raise EvalException('Fatal error detected: ' + state.last_error)
-
-        # Get git patch
-        complete_runtime_fn = complete_runtime
-        return_val = complete_runtime_fn(runtime, instance)
-        git_patch = return_val['git_patch']
-        logger.info(
-            f'Got git diff for instance {instance.instance_id}:\n--------\n{git_patch}\n--------'
-        )
-    finally:
-        runtime.close()
-    # ==========================================
-
-    # ======= Attempt to evaluate the agent's edits =======
-    # we use eval_infer.sh to evaluate the agent's edits, not here
-    # because the agent may alter the environment / testcases
-    test_result = {
-        'git_patch': git_patch,
-    }
-
-    # If you are working on some simpler benchmark that only evaluates the final model output (e.g., in a MessageAction)
-    # You can simply get the LAST `MessageAction` from the returned `state.history` and parse it for evaluation.
-    if state is None:
-        raise ValueError('State should not be None.')
-
-    # NOTE: this is NO LONGER the event stream, but an agent history that includes delegate agent's events
-    histories = [event_to_dict(event) for event in state.history]
-    metrics = get_metrics(state)
-
-    # Save the output
-    instruction = message_action.content
-    if message_action.image_urls:
-        instruction += (
-            '\n\n<image_urls>' + '\n'.join(message_action.image_urls) + '</image_urls>'
-        )
-    output = EvalOutput(
-        instance_id=instance.instance_id,
-        instruction=instruction,
-        instance=instance.to_dict(),  # SWE Bench specific
-        test_result=test_result,
-        metadata=metadata,
-        history=histories,
-        metrics=metrics,
-        error=state.last_error if state and state.last_error else None,
-    )
-    return output
-
-
-def filter_dataset(dataset: pd.DataFrame, filter_column: str) -> pd.DataFrame:
-    file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'config.toml')
-    if os.path.exists(file_path):
-        with open(file_path, 'r') as file:
-            data = toml.load(file)
-            if 'selected_ids' in data:
-                selected_ids = data['selected_ids']
-                logger.info(
-                    f'Filtering {len(selected_ids)} tasks from "selected_ids"...'
-                )
-                subset = dataset[dataset[filter_column].isin(selected_ids)]
-                logger.info(f'Retained {subset.shape[0]} tasks after filtering')
-                return subset
-            if 'selected_repos' in data:
-                selected_repos = data['selected_repos']
-                if isinstance(selected_repos, str):
-                    selected_repos = [selected_repos]
-                assert isinstance(selected_repos, list)
-                logger.info(
-                    f'Filtering {selected_repos} tasks from "selected_repos"...'
-                )
-                subset = dataset[dataset['repo'].isin(selected_repos)]
-                logger.info(f'Retained {subset.shape[0]} tasks after filtering')
-                return subset
-
-    skip_ids = os.environ.get('SKIP_IDS', '').split(',')
-    if len(skip_ids) > 0:
-        logger.info(f'Filtering {len(skip_ids)} tasks from "SKIP_IDS"...')
-        return dataset[~dataset[filter_column].isin(skip_ids)]
-    return dataset
-
-
-if __name__ == '__main__':
-    parser = get_evaluation_parser()
-    parser.add_argument(
-        '--dataset',
-        type=str,
-        default='SWE-Perf/SWE-Perf',
-        help='data set to evaluate on, either full-test or lite-test',
-    )
-    parser.add_argument(
-        '--split',
-        type=str,
-        default='test',
-        help='split to evaluate on',
-    )
-    parser.add_argument(
-        '--mode',
-        type=str,
-        default='swe',
-        choices=['swe', 'swt', 'swt-ci'],
-        help="mode to run the evaluation, either 'swe', 'swt', or 'swt-ci'",
-    )
-
-    args, _ = parser.parse_known_args()
-
-    # NOTE: It is preferable to load datasets from huggingface datasets and perform post-processing
-    # so we don't need to manage file uploading to OpenHands's repo
-    dataset = load_dataset(args.dataset, split=args.split)
-
-    swe_perf_tests = filter_dataset(dataset.to_pandas(), 'instance_id')
-    logger.info(
-        f'Loaded dataset {args.dataset} with split {args.split}: {len(swe_perf_tests)} tasks'
-    )
-
-    llm_config = None
-    if args.llm_config:
-        llm_config = get_llm_config_arg(args.llm_config)
-        llm_config.log_completions = True
-        # modify_params must be False for evaluation purpose, for reproducibility and accurancy of results
-        llm_config.modify_params = False
-
-    if llm_config is None:
-        raise ValueError(f'Could not find LLM config: --llm_config {args.llm_config}')
-
-    # Get condenser config from environment variable
-    condenser_name = os.environ.get('EVAL_CONDENSER')
-    if condenser_name:
-        condenser_config = get_condenser_config_arg(condenser_name)
-        if condenser_config is None:
-            raise ValueError(
-                f'Could not find Condenser config: EVAL_CONDENSER={condenser_name}'
-            )
-    else:
-        # If no specific condenser config is provided via env var, default to NoOpCondenser
-        condenser_config = NoOpCondenserConfig()
-        logger.debug(
-            'No Condenser config provided via EVAL_CONDENSER, using NoOpCondenser.'
-        )
-
-    details = {'mode': args.mode}
-    _agent_cls = openhands.agenthub.Agent.get_cls(args.agent_cls)
-
-    dataset_descrption = (
-        args.dataset.replace('/', '__') + '-' + args.split.replace('/', '__')
-    )
-    metadata = make_metadata(
-        llm_config,
-        dataset_descrption,
-        args.agent_cls,
-        args.max_iterations,
-        args.eval_note,
-        args.eval_output_dir,
-        details=details,
-        condenser_config=condenser_config,
-    )
-
-    output_file = os.path.join(metadata.eval_output_dir, 'output.jsonl')
-    print(f'### OUTPUT FILE: {output_file} ###')
-
-    # Run evaluation in iterative mode:
-    # If a rollout fails to output AgentFinishAction, we will try again until it succeeds OR total 3 attempts have been made.
-    ITERATIVE_EVAL_MODE = (
-        os.environ.get('ITERATIVE_EVAL_MODE', 'false').lower() == 'true'
-    )
-    ITERATIVE_EVAL_MODE_MAX_ATTEMPTS = int(
-        os.environ.get('ITERATIVE_EVAL_MODE_MAX_ATTEMPTS', '3')
-    )
-
-    if not ITERATIVE_EVAL_MODE:
-        # load the dataset
-        instances = prepare_dataset(swe_perf_tests, output_file, args.eval_n_limit)
-
-        run_evaluation(
-            instances,
-            metadata,
-            output_file,
-            args.eval_num_workers,
-            process_instance,
-            timeout_seconds=8
-            * 60
-            * 60,  # 8 hour PER instance should be more than enough
-            max_retries=5,
-        )
-    else:
-        critic = AgentFinishedCritic()
-
-        def get_cur_output_file_path(attempt: int) -> str:
-            return (
-                f'{output_file.removesuffix(".jsonl")}.critic_attempt_{attempt}.jsonl'
-            )
-
-        eval_ids = None
-        for attempt in range(1, ITERATIVE_EVAL_MODE_MAX_ATTEMPTS + 1):
-            cur_output_file = get_cur_output_file_path(attempt)
-            logger.info(
-                f'Running evaluation with critic {critic.__class__.__name__} for attempt {attempt} of {ITERATIVE_EVAL_MODE_MAX_ATTEMPTS}.'
-            )
-
-            # For deterministic eval, we set temperature to 0.1 for (>1) attempt
-            # so hopefully we get slightly different results
-            if attempt > 1 and metadata.llm_config.temperature == 0:
-                logger.info(
-                    f'Detected temperature is 0 for (>1) attempt {attempt}. Setting temperature to 0.1...'
-                )
-                metadata.llm_config.temperature = 0.1
-
-            # Load instances - at first attempt, we evaluate all instances
-            # On subsequent attempts, we only evaluate the instances that failed the previous attempt determined by critic
-            instances = prepare_dataset(
-                swe_perf_tests, cur_output_file, args.eval_n_limit, eval_ids=eval_ids
-            )
-
-            # Run evaluation - but save them to cur_output_file
-            logger.info(
-                f'Evaluating {len(instances)} instances for attempt {attempt}...'
-            )
-            run_evaluation(
-                instances,
-                metadata,
-                cur_output_file,
-                args.eval_num_workers,
-                process_instance,
-                timeout_seconds=8
-                * 60
-                * 60,  # 8 hour PER instance should be more than enough
-                max_retries=5,
-            )
-
-            # When eval is done, we update eval_ids to the instances that failed the current attempt
-            instances_failed = []
-            logger.info(
-                f'Use critic {critic.__class__.__name__} to check {len(instances)} instances for attempt {attempt}...'
-            )
-            with open(cur_output_file, 'r') as f:
-                for line in f:
-                    instance = json.loads(line)
-                    try:
-                        history = [
-                            event_from_dict(event) for event in instance['history']
-                        ]
-                        critic_result = critic.evaluate(
-                            history, instance['test_result'].get('git_patch', '')
-                        )
-                        if not critic_result.success:
-                            instances_failed.append(instance['instance_id'])
-                    except Exception as e:
-                        logger.error(
-                            f'Error loading history for instance {instance["instance_id"]}: {e}'
-                        )
-                        instances_failed.append(instance['instance_id'])
-            logger.info(
-                f'{len(instances_failed)} instances failed the current attempt {attempt}: {instances_failed}'
-            )
-            eval_ids = instances_failed
-
-            # If no instances failed, we break
-            if len(instances_failed) == 0:
-                break
-
-        # Then we should aggregate the results from all attempts into the original output file
-        # and remove the intermediate files
-        logger.info(
-            'Aggregating results from all attempts into the original output file...'
-        )
-        fout = open(output_file, 'w')
-        added_instance_ids = set()
-        for attempt in reversed(range(1, ITERATIVE_EVAL_MODE_MAX_ATTEMPTS + 1)):
-            cur_output_file = get_cur_output_file_path(attempt)
-            if not os.path.exists(cur_output_file):
-                logger.warning(
-                    f'Intermediate output file {cur_output_file} does not exist. Skipping...'
-                )
-                continue
-
-            with open(cur_output_file, 'r') as f:
-                for line in f:
-                    instance = json.loads(line)
-                    # Also make sure git_patch is not empty - otherwise we fall back to previous attempt (empty patch is worse than anything else)
-                    if (
-                        instance['instance_id'] not in added_instance_ids
-                        and instance['test_result'].get('git_patch', '').strip()
-                    ):
-                        fout.write(line)
-                        added_instance_ids.add(instance['instance_id'])
-            logger.info(
-                f'Aggregated instances from {cur_output_file}. Total instances added so far: {len(added_instance_ids)}'
-            )
-        fout.close()
-        logger.info(
-            f'Done! Total {len(added_instance_ids)} instances added to {output_file}'
-        )
-        # Check if any instances reached maximum retries
-        check_maximum_retries_exceeded(metadata.eval_output_dir)

evaluation/benchmarks/swe_perf/scripts/run_infer.sh

@@ -1,146 +0,0 @@
-#!/usr/bin/env bash
-set -eo pipefail
-
-source "evaluation/utils/version_control.sh"
-
-MODEL_CONFIG=$1
-COMMIT_HASH=$2
-AGENT=$3
-EVAL_LIMIT=$4
-MAX_ITER=$5
-NUM_WORKERS=$6
-DATASET=$7
-SPLIT=$8
-N_RUNS=$9
-MODE=${10}
-
-
-if [ -z "$NUM_WORKERS" ]; then
-  NUM_WORKERS=1
-  echo "Number of workers not specified, use default $NUM_WORKERS"
-fi
-checkout_eval_branch
-
-if [ -z "$AGENT" ]; then
-  echo "Agent not specified, use default CodeActAgent"
-  AGENT="CodeActAgent"
-fi
-
-if [ -z "$MAX_ITER" ]; then
-  echo "MAX_ITER not specified, use default 100"
-  MAX_ITER=100
-fi
-
-if [ -z "$RUN_WITH_BROWSING" ]; then
-  echo "RUN_WITH_BROWSING not specified, use default false"
-  RUN_WITH_BROWSING=false
-fi
-
-
-if [ -z "$DATASET" ]; then
-  echo "DATASET not specified, use default SWE-Perf/SWE-Perf"
-  DATASET="SWE-Perf/SWE-Perf"
-fi
-
-if [ -z "$SPLIT" ]; then
-  echo "SPLIT not specified, use default test"
-  SPLIT="test"
-fi
-
-if [ -z "$MODE" ]; then
-  MODE="swe"
-  echo "MODE not specified, use default $MODE"
-fi
-
-if [ -n "$EVAL_CONDENSER" ]; then
-  echo "Using Condenser Config: $EVAL_CONDENSER"
-else
-  echo "No Condenser Config provided via EVAL_CONDENSER, use default (NoOpCondenser)."
-fi
-
-export RUN_WITH_BROWSING=$RUN_WITH_BROWSING
-echo "RUN_WITH_BROWSING: $RUN_WITH_BROWSING"
-
-get_openhands_version
-
-echo "AGENT: $AGENT"
-echo "OPENHANDS_VERSION: $OPENHANDS_VERSION"
-echo "MODEL_CONFIG: $MODEL_CONFIG"
-echo "DATASET: $DATASET"
-echo "SPLIT: $SPLIT"
-echo "MAX_ITER: $MAX_ITER"
-echo "NUM_WORKERS: $NUM_WORKERS"
-echo "COMMIT_HASH: $COMMIT_HASH"
-echo "MODE: $MODE"
-echo "EVAL_CONDENSER: $EVAL_CONDENSER"
-
-# Default to NOT use Hint
-if [ -z "$USE_HINT_TEXT" ]; then
-  export USE_HINT_TEXT=false
-fi
-echo "USE_HINT_TEXT: $USE_HINT_TEXT"
-EVAL_NOTE="$OPENHANDS_VERSION"
-# if not using Hint, add -no-hint to the eval note
-if [ "$USE_HINT_TEXT" = false ]; then
-  EVAL_NOTE="$EVAL_NOTE-no-hint"
-fi
-
-if [ "$RUN_WITH_BROWSING" = true ]; then
-  EVAL_NOTE="$EVAL_NOTE-with-browsing"
-fi
-
-if [ -n "$EXP_NAME" ]; then
-  EVAL_NOTE="$EVAL_NOTE-$EXP_NAME"
-fi
-# if mode != swe, add mode to the eval note
-if [ "$MODE" != "swe" ]; then
-  EVAL_NOTE="${EVAL_NOTE}-${MODE}"
-fi
-# Add condenser config to eval note if provided
-if [ -n "$EVAL_CONDENSER" ]; then
-  EVAL_NOTE="${EVAL_NOTE}-${EVAL_CONDENSER}"
-fi
-
-function run_eval() {
-  local eval_note="${1}"
-  COMMAND="poetry run python evaluation/benchmarks/swe_perf/run_infer.py \
-    --agent-cls $AGENT \
-    --llm-config $MODEL_CONFIG \
-    --max-iterations $MAX_ITER \
-    --eval-num-workers $NUM_WORKERS \
-    --eval-note $eval_note \
-    --dataset $DATASET \
-    --split $SPLIT \
-    --mode $MODE"
-
-
-
-  if [ -n "$EVAL_LIMIT" ]; then
-    echo "EVAL_LIMIT: $EVAL_LIMIT"
-    COMMAND="$COMMAND --eval-n-limit $EVAL_LIMIT"
-  fi
-
-  # Run the command
-  eval $COMMAND
-}
-
-unset SANDBOX_ENV_GITHUB_TOKEN # prevent the agent from using the github token to push
-if [ -z "$N_RUNS" ]; then
-  N_RUNS=1
-  echo "N_RUNS not specified, use default $N_RUNS"
-fi
-
-# Skip runs if the run number is in the SKIP_RUNS list
-# read from env variable SKIP_RUNS as a comma separated list of run numbers
-SKIP_RUNS=(${SKIP_RUNS//,/ })
-for i in $(seq 1 $N_RUNS); do
-  if [[ " ${SKIP_RUNS[@]} " =~ " $i " ]]; then
-    echo "Skipping run $i"
-    continue
-  fi
-  current_eval_note="$EVAL_NOTE-run_$i"
-  echo "EVAL_NOTE: $current_eval_note"
-  run_eval $current_eval_note
-done
-
-checkout_original_branch

evaluation/benchmarks/swe_perf/scripts/setup/compare_patch_filename.py

@@ -1,54 +0,0 @@
-"""This script compares gold patches with OpenHands-generated patches and check whether
-OpenHands found the right (set of) files to modify.
-"""
-
-import argparse
-import json
-import re
-
-
-def extract_modified_files(patch):
-    modified_files = set()
-    file_pattern = re.compile(r'^diff --git a/(.*?) b/')
-
-    for line in patch.split('\n'):
-        match = file_pattern.match(line)
-        if match:
-            modified_files.add(match.group(1))
-
-    return modified_files
-
-
-def process_report(oh_output_file):
-    succ = 0
-    fail = 0
-    for line in open(oh_output_file):
-        line = json.loads(line)
-        instance_id = line['instance_id']
-        gold_patch = line['swe_instance']['patch']
-        generated_patch = line['git_patch']
-        gold_modified_files = extract_modified_files(gold_patch)
-        # swe-bench lite only: a gold patch always contains exactly one file
-        assert len(gold_modified_files) == 1
-        generated_modified_files = extract_modified_files(generated_patch)
-
-        # Check if all files in gold_patch are also in generated_patch
-        all_files_in_generated = gold_modified_files.issubset(generated_modified_files)
-        if all_files_in_generated:
-            succ += 1
-        else:
-            fail += 1
-            print(
-                f'{instance_id}: file mismatch, gold = {gold_modified_files}, generated = {generated_modified_files}'
-            )
-    print(
-        f'\nSUMMARY: {succ} out of {succ + fail} instances found correct files to edit, success rate = {succ / float(succ + fail)}'
-    )
-
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--oh_output_file', help='Path to the OH output file')
-    args = parser.parse_args()
-
-    process_report(args.oh_output_file)

evaluation/benchmarks/swe_perf/scripts/setup/instance_swe_entry.sh

@@ -1,43 +0,0 @@
-#!/usr/bin/env bash
-
-source ~/.bashrc
-SWEUTIL_DIR=/swe_util
-
-# FIXME: Cannot read SWE_INSTANCE_ID from the environment variable
-# SWE_INSTANCE_ID=django__django-11099
-if [ -z "$SWE_INSTANCE_ID" ]; then
-    echo "Error: SWE_INSTANCE_ID is not set." >&2
-    exit 1
-fi
-
-# Read the swe-bench-test-lite.json file and extract the required item based on instance_id
-item=$(jq --arg INSTANCE_ID "$SWE_INSTANCE_ID" '.[] | select(.instance_id == $INSTANCE_ID)' $SWEUTIL_DIR/eval_data/instances/swe-bench-instance.json)
-
-if [[ -z "$item" ]]; then
-  echo "No item found for the provided instance ID."
-  exit 1
-fi
-
-
-WORKSPACE_NAME=$(echo "$item" | jq -r '(.repo | tostring) + "__" + (.version | tostring) | gsub("/"; "__")')
-
-echo "WORKSPACE_NAME: $WORKSPACE_NAME"
-
-# Clear the workspace
-if [ -d /workspace ]; then
-    rm -rf /workspace/*
-else
-    mkdir /workspace
-fi
-# Copy repo to workspace
-if [ -d /workspace/$WORKSPACE_NAME ]; then
-    rm -rf /workspace/$WORKSPACE_NAME
-fi
-mkdir -p /workspace
-cp -r /testbed /workspace/$WORKSPACE_NAME
-
-# Activate instance-specific environment
-if [ -d /opt/miniconda3 ]; then
-    . /opt/miniconda3/etc/profile.d/conda.sh
-    conda activate testbed
-fi

frontend/.husky/pre-commit

@@ -1,6 +1,8 @@
 # Run frontend checks
 echo "Running frontend checks..."
 cd frontend
+npm run lint
+npm run check-translation-completeness
 npx lint-staged
 
 # Run backend pre-commit

frontend/__tests__/api/file-service/file-service.api.test.ts

@@ -1,5 +1,5 @@
 import { describe, expect, it } from "vitest";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
+import OpenHands from "#/api/open-hands";
 import {
   FILE_VARIANTS_1,
   FILE_VARIANTS_2,
@@ -10,20 +10,20 @@ import {
  * You can find the mock handlers in `frontend/src/mocks/file-service-handlers.ts`.
  */
 
-describe("ConversationService File API", () => {
+describe("OpenHands File API", () => {
   it("should get a list of files", async () => {
-    await expect(
-      ConversationService.getFiles("test-conversation-id"),
-    ).resolves.toEqual(FILE_VARIANTS_1);
+    await expect(OpenHands.getFiles("test-conversation-id")).resolves.toEqual(
+      FILE_VARIANTS_1,
+    );
 
     await expect(
-      ConversationService.getFiles("test-conversation-id-2"),
+      OpenHands.getFiles("test-conversation-id-2"),
     ).resolves.toEqual(FILE_VARIANTS_2);
   });
 
   it("should get content of a file", async () => {
     await expect(
-      ConversationService.getFile("test-conversation-id", "file1.txt"),
+      OpenHands.getFile("test-conversation-id", "file1.txt"),
     ).resolves.toEqual("Content of file1.txt");
   });
 });

frontend/__tests__/components/browser.test.tsx

@@ -13,8 +13,7 @@ vi.mock("react-router", async () => {
 
 vi.mock("#/context/conversation-context", () => ({
   useConversation: () => ({ conversationId: "test-conversation-id" }),
-  ConversationProvider: ({ children }: { children: React.ReactNode }) =>
-    children,
+  ConversationProvider: ({ children }: { children: React.ReactNode }) => children,
 }));
 
 vi.mock("react-i18next", async () => {
@@ -30,18 +29,21 @@ vi.mock("react-i18next", async () => {
   };
 });
 
-// Mock Zustand browser store
+// Mock redux
+const mockDispatch = vi.fn();
 let mockBrowserState = {
   url: "https://example.com",
   screenshotSrc: "",
-  setUrl: vi.fn(),
-  setScreenshotSrc: vi.fn(),
-  reset: vi.fn(),
 };
 
-vi.mock("#/stores/browser-store", () => ({
-  useBrowserStore: () => mockBrowserState,
-}));
+vi.mock("react-redux", async () => {
+  const actual = await vi.importActual("react-redux");
+  return {
+    ...actual,
+    useDispatch: () => mockDispatch,
+    useSelector: () => mockBrowserState,
+  };
+});
 
 // Import the component after all mocks are set up
 import { BrowserPanel } from "#/components/features/browser/browser";
@@ -53,9 +55,6 @@ describe("Browser", () => {
     mockBrowserState = {
       url: "https://example.com",
       screenshotSrc: "",
-      setUrl: vi.fn(),
-      setScreenshotSrc: vi.fn(),
-      reset: vi.fn(),
     };
   });
 
@@ -64,9 +63,6 @@ describe("Browser", () => {
     mockBrowserState = {
       url: "https://example.com",
       screenshotSrc: "",
-      setUrl: vi.fn(),
-      setScreenshotSrc: vi.fn(),
-      reset: vi.fn(),
     };
 
     render(<BrowserPanel />);
@@ -79,11 +75,7 @@ describe("Browser", () => {
     // Set the mock state for this test
     mockBrowserState = {
       url: "https://example.com",
-      screenshotSrc:
-        "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mN0uGvyHwAFCAJS091fQwAAAABJRU5ErkJggg==",
-      setUrl: vi.fn(),
-      setScreenshotSrc: vi.fn(),
-      reset: vi.fn(),
+      screenshotSrc: "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mN0uGvyHwAFCAJS091fQwAAAABJRU5ErkJggg==",
     };
 
     render(<BrowserPanel />);

frontend/__tests__/components/chat/action-suggestions.test.tsx

@@ -0,0 +1,287 @@
+import { describe, expect, it, vi, beforeEach } from "vitest";
+import { render, screen } from "@testing-library/react";
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
+import { ActionSuggestions } from "#/components/features/chat/action-suggestions";
+import OpenHands from "#/api/open-hands";
+import { MOCK_DEFAULT_USER_SETTINGS } from "#/mocks/handlers";
+
+// Mock dependencies
+vi.mock("posthog-js", () => ({
+  default: {
+    capture: vi.fn(),
+  },
+}));
+
+const { useSelectorMock } = vi.hoisted(() => ({
+  useSelectorMock: vi.fn(),
+}));
+
+vi.mock("react-redux", () => ({
+  useSelector: useSelectorMock,
+}));
+
+vi.mock("#/context/auth-context", () => ({
+  useAuth: vi.fn(),
+}));
+
+// Mock react-i18next
+vi.mock("react-i18next", () => ({
+  useTranslation: () => ({
+    t: (key: string) => {
+      const translations: Record<string, string> = {
+        ACTION$PUSH_TO_BRANCH: "Push to Branch",
+        ACTION$PUSH_CREATE_PR: "Push & Create PR",
+        ACTION$PUSH_CHANGES_TO_PR: "Push Changes to PR",
+      };
+      return translations[key] || key;
+    },
+  }),
+}));
+
+vi.mock("react-router", () => ({
+  useParams: () => ({
+    conversationId: "test-conversation-id",
+  }),
+}));
+
+const renderActionSuggestions = () =>
+  render(<ActionSuggestions onSuggestionsClick={() => {}} />, {
+    wrapper: ({ children }) => (
+      <QueryClientProvider client={new QueryClient()}>
+        {children}
+      </QueryClientProvider>
+    ),
+  });
+
+describe("ActionSuggestions", () => {
+  // Setup mocks for each test
+  beforeEach(() => {
+    vi.clearAllMocks();
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
+    getSettingsSpy.mockResolvedValue({
+      ...MOCK_DEFAULT_USER_SETTINGS,
+      provider_tokens_set: {
+        github: "some-token",
+      },
+    });
+
+    useSelectorMock.mockReturnValue({
+      selectedRepository: "test-repo",
+    });
+  });
+
+  it("should render both GitHub buttons when GitHub token is set and repository is selected", async () => {
+    const getConversationSpy = vi.spyOn(OpenHands, "getConversation");
+    // @ts-expect-error - only required for testing
+    getConversationSpy.mockResolvedValue({
+      selected_repository: "test-repo",
+    });
+    renderActionSuggestions();
+
+    // Find all buttons with data-testid="suggestion"
+    const buttons = await screen.findAllByTestId("suggestion");
+
+    // Check if we have at least 2 buttons
+    expect(buttons.length).toBeGreaterThanOrEqual(2);
+
+    // Check if the buttons contain the expected text
+    const pushButton = buttons.find((button) =>
+      button.textContent?.includes("Push to Branch"),
+    );
+    const prButton = buttons.find((button) =>
+      button.textContent?.includes("Push & Create PR"),
+    );
+
+    expect(pushButton).toBeInTheDocument();
+    expect(prButton).toBeInTheDocument();
+  });
+
+  it("should not render buttons when GitHub token is not set", () => {
+    renderActionSuggestions();
+
+    expect(screen.queryByTestId("suggestion")).not.toBeInTheDocument();
+  });
+
+  it("should not render buttons when no repository is selected", () => {
+    useSelectorMock.mockReturnValue({
+      selectedRepository: null,
+    });
+
+    renderActionSuggestions();
+
+    expect(screen.queryByTestId("suggestion")).not.toBeInTheDocument();
+  });
+
+  it("should have different prompts for 'Push to Branch' and 'Push & Create PR' buttons", () => {
+    // This test verifies that the prompts are different in the component
+    renderActionSuggestions();
+
+    // Get the component instance to access the internal values
+    const pushBranchPrompt =
+      "Please push the changes to a remote branch on GitHub, but do NOT create a pull request. Please use the exact SAME branch name as the one you are currently on.";
+    const createPRPrompt =
+      "Please push the changes to GitHub and open a pull request. Please create a meaningful branch name that describes the changes. If a pull request template exists in the repository, please follow it when creating the PR description.";
+
+    // Verify the prompts are different
+    expect(pushBranchPrompt).not.toEqual(createPRPrompt);
+
+    // Verify the PR prompt mentions creating a meaningful branch name
+    expect(createPRPrompt).toContain("meaningful branch name");
+    expect(createPRPrompt).not.toContain("SAME branch name");
+  });
+
+  it("should use correct provider name based on conversation git_provider, not user authenticated providers", async () => {
+    // Test case for GitHub repository
+    const getConversationSpy = vi.spyOn(OpenHands, "getConversation");
+    getConversationSpy.mockResolvedValue({
+      conversation_id: "test-github",
+      title: "GitHub Test",
+      selected_repository: "test-repo",
+      git_provider: "github",
+      selected_branch: "main",
+      last_updated_at: new Date().toISOString(),
+      created_at: new Date().toISOString(),
+      status: "RUNNING",
+      runtime_status: "STATUS$READY",
+      url: null,
+      session_api_key: null,
+    });
+
+    // Mock user having both GitHub and Bitbucket tokens
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
+    getSettingsSpy.mockResolvedValue({
+      ...MOCK_DEFAULT_USER_SETTINGS,
+      provider_tokens_set: {
+        github: "github-token",
+        bitbucket: "bitbucket-token",
+      },
+    });
+
+    const onSuggestionsClick = vi.fn();
+    render(<ActionSuggestions onSuggestionsClick={onSuggestionsClick} />, {
+      wrapper: ({ children }) => (
+        <QueryClientProvider client={new QueryClient()}>
+          {children}
+        </QueryClientProvider>
+      ),
+    });
+
+    const buttons = await screen.findAllByTestId("suggestion");
+    const prButton = buttons.find((button) =>
+      button.textContent?.includes("Push & Create PR"),
+    );
+
+    expect(prButton).toBeInTheDocument();
+
+    if (prButton) {
+      prButton.click();
+    }
+
+    // The suggestion should mention GitHub, not Bitbucket
+    expect(onSuggestionsClick).toHaveBeenCalledWith(
+      expect.stringContaining("GitHub")
+    );
+    expect(onSuggestionsClick).not.toHaveBeenCalledWith(
+      expect.stringContaining("Bitbucket")
+    );
+  });
+
+  it("should use GitLab terminology when git_provider is gitlab", async () => {
+    const getConversationSpy = vi.spyOn(OpenHands, "getConversation");
+    getConversationSpy.mockResolvedValue({
+      conversation_id: "test-gitlab",
+      title: "GitLab Test",
+      selected_repository: "test-repo",
+      git_provider: "gitlab",
+      selected_branch: "main",
+      last_updated_at: new Date().toISOString(),
+      created_at: new Date().toISOString(),
+      status: "RUNNING",
+      runtime_status: "STATUS$READY",
+      url: null,
+      session_api_key: null,
+    });
+
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
+    getSettingsSpy.mockResolvedValue({
+      ...MOCK_DEFAULT_USER_SETTINGS,
+      provider_tokens_set: {
+        gitlab: "gitlab-token",
+      },
+    });
+
+    const onSuggestionsClick = vi.fn();
+    render(<ActionSuggestions onSuggestionsClick={onSuggestionsClick} />, {
+      wrapper: ({ children }) => (
+        <QueryClientProvider client={new QueryClient()}>
+          {children}
+        </QueryClientProvider>
+      ),
+    });
+
+    const buttons = await screen.findAllByTestId("suggestion");
+    const prButton = buttons.find((button) =>
+      button.textContent?.includes("Push & Create PR"),
+    );
+
+    if (prButton) {
+      prButton.click();
+    }
+
+    // Should mention GitLab and "merge request" instead of "pull request"
+    expect(onSuggestionsClick).toHaveBeenCalledWith(
+      expect.stringContaining("GitLab")
+    );
+    expect(onSuggestionsClick).toHaveBeenCalledWith(
+      expect.stringContaining("merge request")
+    );
+  });
+
+  it("should use Bitbucket terminology when git_provider is bitbucket", async () => {
+    const getConversationSpy = vi.spyOn(OpenHands, "getConversation");
+    getConversationSpy.mockResolvedValue({
+      conversation_id: "test-bitbucket",
+      title: "Bitbucket Test",
+      selected_repository: "test-repo",
+      git_provider: "bitbucket",
+      selected_branch: "main",
+      last_updated_at: new Date().toISOString(),
+      created_at: new Date().toISOString(),
+      status: "RUNNING",
+      runtime_status: "STATUS$READY",
+      url: null,
+      session_api_key: null,
+    });
+
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
+    getSettingsSpy.mockResolvedValue({
+      ...MOCK_DEFAULT_USER_SETTINGS,
+      provider_tokens_set: {
+        bitbucket: "bitbucket-token",
+      },
+    });
+
+    const onSuggestionsClick = vi.fn();
+    render(<ActionSuggestions onSuggestionsClick={onSuggestionsClick} />, {
+      wrapper: ({ children }) => (
+        <QueryClientProvider client={new QueryClient()}>
+          {children}
+        </QueryClientProvider>
+      ),
+    });
+
+    const buttons = await screen.findAllByTestId("suggestion");
+    const prButton = buttons.find((button) =>
+      button.textContent?.includes("Push & Create PR"),
+    );
+
+    if (prButton) {
+      prButton.click();
+    }
+
+    // Should mention Bitbucket
+    expect(onSuggestionsClick).toHaveBeenCalledWith(
+      expect.stringContaining("Bitbucket")
+    );
+  });
+});

frontend/__tests__/components/chat/chat-input.test.tsx

@@ -0,0 +1,256 @@
+import userEvent from "@testing-library/user-event";
+import { fireEvent, render, screen } from "@testing-library/react";
+import { describe, afterEach, vi, it, expect } from "vitest";
+import { ChatInput } from "#/components/features/chat/chat-input";
+
+describe("ChatInput", () => {
+  const onSubmitMock = vi.fn();
+
+  afterEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("should render a textarea", () => {
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    expect(screen.getByTestId("chat-input")).toBeInTheDocument();
+    expect(screen.getByRole("textbox")).toBeInTheDocument();
+  });
+
+  it("should call onSubmit when the user types and presses enter", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+
+    await user.type(textarea, "Hello, world!");
+    await user.keyboard("{Enter}");
+
+    expect(onSubmitMock).toHaveBeenCalledWith("Hello, world!");
+  });
+
+  it("should call onSubmit when pressing the submit button", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+    const button = screen.getByRole("button");
+
+    await user.type(textarea, "Hello, world!");
+    await user.click(button);
+
+    expect(onSubmitMock).toHaveBeenCalledWith("Hello, world!");
+  });
+
+  it("should not call onSubmit when the message is empty", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const button = screen.getByRole("button");
+
+    await user.click(button);
+    expect(onSubmitMock).not.toHaveBeenCalled();
+
+    await user.keyboard("{Enter}");
+    expect(onSubmitMock).not.toHaveBeenCalled();
+  });
+
+  it("should not call onSubmit when the message is only whitespace", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+
+    await user.type(textarea, "   ");
+    await user.keyboard("{Enter}");
+
+    expect(onSubmitMock).not.toHaveBeenCalled();
+
+    await user.type(textarea, " \t\n");
+    await user.keyboard("{Enter}");
+
+    expect(onSubmitMock).not.toHaveBeenCalled();
+  });
+
+  it("should disable submit", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput disabled onSubmit={onSubmitMock} />);
+
+    const button = screen.getByRole("button");
+    const textarea = screen.getByRole("textbox");
+
+    await user.type(textarea, "Hello, world!");
+
+    expect(button).toBeDisabled();
+    await user.click(button);
+    expect(onSubmitMock).not.toHaveBeenCalled();
+
+    await user.keyboard("{Enter}");
+    expect(onSubmitMock).not.toHaveBeenCalled();
+  });
+
+  it("should render a placeholder with translation key", () => {
+    render(<ChatInput onSubmit={onSubmitMock} />);
+
+    const textarea = screen.getByPlaceholderText("SUGGESTIONS$WHAT_TO_BUILD");
+    expect(textarea).toBeInTheDocument();
+  });
+
+  it("should create a newline instead of submitting when shift + enter is pressed", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+
+    await user.type(textarea, "Hello, world!");
+    await user.keyboard("{Shift>} {Enter}"); // Shift + Enter
+
+    expect(onSubmitMock).not.toHaveBeenCalled();
+    // expect(textarea).toHaveValue("Hello, world!\n");
+  });
+
+  it("should clear the input message after sending a message", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+    const button = screen.getByRole("button");
+
+    await user.type(textarea, "Hello, world!");
+    await user.keyboard("{Enter}");
+    expect(textarea).toHaveValue("");
+
+    await user.type(textarea, "Hello, world!");
+    await user.click(button);
+    expect(textarea).toHaveValue("");
+  });
+
+  it("should hide the submit button", () => {
+    render(<ChatInput onSubmit={onSubmitMock} showButton={false} />);
+    expect(screen.queryByRole("button")).not.toBeInTheDocument();
+  });
+
+  it("should call onChange when the user types", async () => {
+    const user = userEvent.setup();
+    const onChangeMock = vi.fn();
+    render(<ChatInput onSubmit={onSubmitMock} onChange={onChangeMock} />);
+    const textarea = screen.getByRole("textbox");
+
+    await user.type(textarea, "Hello, world!");
+
+    expect(onChangeMock).toHaveBeenCalledTimes("Hello, world!".length);
+  });
+
+  it("should have set the passed value", () => {
+    render(<ChatInput value="Hello, world!" onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+
+    expect(textarea).toHaveValue("Hello, world!");
+  });
+
+  it("should display the stop button and trigger the callback", async () => {
+    const user = userEvent.setup();
+    const onStopMock = vi.fn();
+    render(
+      <ChatInput onSubmit={onSubmitMock} button="stop" onStop={onStopMock} />,
+    );
+    const stopButton = screen.getByTestId("stop-button");
+
+    await user.click(stopButton);
+    expect(onStopMock).toHaveBeenCalledOnce();
+  });
+
+  it("should call onFocus and onBlur when the textarea is focused and blurred", async () => {
+    const user = userEvent.setup();
+    const onFocusMock = vi.fn();
+    const onBlurMock = vi.fn();
+    render(
+      <ChatInput
+        onSubmit={onSubmitMock}
+        onFocus={onFocusMock}
+        onBlur={onBlurMock}
+      />,
+    );
+    const textarea = screen.getByRole("textbox");
+
+    await user.click(textarea);
+    expect(onFocusMock).toHaveBeenCalledOnce();
+
+    await user.tab();
+    expect(onBlurMock).toHaveBeenCalledOnce();
+  });
+
+  it("should handle text paste correctly", () => {
+    const onSubmit = vi.fn();
+    const onChange = vi.fn();
+
+    render(<ChatInput onSubmit={onSubmit} onChange={onChange} />);
+
+    const input = screen.getByTestId("chat-input").querySelector("textarea");
+    expect(input).toBeTruthy();
+
+    // Fire paste event with text data
+    fireEvent.paste(input!, {
+      clipboardData: {
+        getData: (type: string) => (type === "text/plain" ? "test paste" : ""),
+        files: [],
+      },
+    });
+  });
+
+  it("should handle image paste correctly", () => {
+    const onSubmit = vi.fn();
+    const onFilesPaste = vi.fn();
+
+    render(<ChatInput onSubmit={onSubmit} onFilesPaste={onFilesPaste} />);
+
+    const input = screen.getByTestId("chat-input").querySelector("textarea");
+    expect(input).toBeTruthy();
+
+    // Create a paste event with an image file
+    const file = new File(["dummy content"], "image.png", {
+      type: "image/png",
+    });
+
+    // Fire paste event with image data
+    fireEvent.paste(input!, {
+      clipboardData: {
+        getData: () => "",
+        files: [file],
+      },
+    });
+
+    // Verify file paste was handled
+    expect(onFilesPaste).toHaveBeenCalledWith([file]);
+  });
+
+  it("should use the default maxRows value", () => {
+    // We can't directly test the maxRows prop as it's not exposed in the DOM
+    // Instead, we'll verify the component renders with the default props
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+    expect(textarea).toBeInTheDocument();
+
+    // The actual verification of maxRows=16 is handled internally by the TextareaAutosize component
+    // and affects how many rows the textarea can expand to
+  });
+
+  it("should not submit when Enter is pressed during IME composition", async () => {
+    const user = userEvent.setup();
+    render(<ChatInput onSubmit={onSubmitMock} />);
+    const textarea = screen.getByRole("textbox");
+
+    await user.type(textarea, "こんにちは");
+
+    // Simulate Enter during IME composition
+    fireEvent.keyDown(textarea, {
+      key: "Enter",
+      isComposing: true,
+      nativeEvent: { isComposing: true },
+    });
+
+    expect(onSubmitMock).not.toHaveBeenCalled();
+
+    // Simulate normal Enter after composition is done
+    fireEvent.keyDown(textarea, {
+      key: "Enter",
+      isComposing: false,
+      nativeEvent: { isComposing: false },
+    });
+
+    expect(onSubmitMock).toHaveBeenCalledWith("こんにちは");
+  });
+});

frontend/__tests__/components/chat/chat-interface.test.tsx

@@ -1,254 +1,16 @@
-import {
-  afterEach,
-  beforeAll,
-  beforeEach,
-  describe,
-  expect,
-  it,
-  test,
-  vi,
-} from "vitest";
-import { render, screen, waitFor, within } from "@testing-library/react";
+import { afterEach, beforeAll, describe, expect, it, vi } from "vitest";
+import { screen, waitFor, within } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
-import { MemoryRouter } from "react-router";
-import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
 import { renderWithProviders } from "test-utils";
 import type { Message } from "#/message";
 import { SUGGESTIONS } from "#/utils/suggestions";
 import { ChatInterface } from "#/components/features/chat/chat-interface";
-import { useWsClient } from "#/context/ws-client-provider";
-import { useOptimisticUserMessage } from "#/hooks/use-optimistic-user-message";
-import { useWSErrorMessage } from "#/hooks/use-ws-error-message";
-import { useConfig } from "#/hooks/query/use-config";
-import { useGetTrajectory } from "#/hooks/mutation/use-get-trajectory";
-import { useUploadFiles } from "#/hooks/mutation/use-upload-files";
-import { OpenHandsAction } from "#/types/core/actions";
-
-// Mock the hooks
-vi.mock("#/context/ws-client-provider");
-vi.mock("#/hooks/use-optimistic-user-message");
-vi.mock("#/hooks/use-ws-error-message");
-vi.mock("#/hooks/query/use-config");
-vi.mock("#/hooks/mutation/use-get-trajectory");
-vi.mock("#/hooks/mutation/use-upload-files");
-
-// Mock React Router hooks at the top level
-vi.mock("react-router", async () => {
-  const actual = await vi.importActual("react-router");
-  return {
-    ...actual,
-    useNavigate: () => vi.fn(),
-    useParams: () => ({ conversationId: "test-conversation-id" }),
-    useRouteLoaderData: vi.fn(() => ({})),
-  };
-});
-
-// Mock other hooks that might be used by the component
-vi.mock("#/hooks/use-user-providers", () => ({
-  useUserProviders: () => ({
-    providers: [],
-  }),
-}));
-
-vi.mock("#/hooks/use-conversation-name-context-menu", () => ({
-  useConversationNameContextMenu: () => ({
-    isOpen: false,
-    contextMenuRef: { current: null },
-    handleContextMenu: vi.fn(),
-    handleClose: vi.fn(),
-    handleRename: vi.fn(),
-    handleDelete: vi.fn(),
-  }),
-}));
-
-vi.mock("react-redux", async () => {
-  const actual = await vi.importActual("react-redux");
-  return {
-    ...actual,
-    useSelector: vi.fn((selector) => {
-      // Create a mock state object
-      const mockState = {
-        agent: {
-          curAgentState: "AWAITING_USER_INPUT",
-        },
-        initialQuery: {
-          selectedRepository: null,
-          replayJson: null,
-        },
-        conversation: {
-          messageToSend: null,
-          files: [],
-          images: [],
-          loadingFiles: [],
-          loadingImages: [],
-        },
-        status: {
-          curStatusMessage: null,
-        },
-      };
-
-      // Execute the selector function with our mock state
-      return selector(mockState);
-    }),
-    useDispatch: vi.fn(() => vi.fn()),
-  };
-});
-
-// Helper function to render with Router context
-const renderChatInterfaceWithRouter = () =>
-  renderWithProviders(
-    <MemoryRouter>
-      <ChatInterface />
-    </MemoryRouter>,
-  );
 
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 const renderChatInterface = (messages: Message[]) =>
-  renderWithProviders(
-    <MemoryRouter>
-      <ChatInterface />
-    </MemoryRouter>,
-  );
+  renderWithProviders(<ChatInterface />);
 
-// Helper function to render with QueryClientProvider and Router (for newer tests)
-const renderWithQueryClient = (
-  ui: React.ReactElement,
-  queryClient: QueryClient,
-) =>
-  render(
-    <QueryClientProvider client={queryClient}>
-      <MemoryRouter>{ui}</MemoryRouter>
-    </QueryClientProvider>,
-  );
-
-describe("ChatInterface - Chat Suggestions", () => {
-  // Create a new QueryClient for each test
-  let queryClient: QueryClient;
-
-  beforeEach(() => {
-    queryClient = new QueryClient({
-      defaultOptions: {
-        queries: {
-          retry: false,
-        },
-      },
-    });
-
-    // Default mock implementations
-    (useWsClient as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      send: vi.fn(),
-      isLoadingMessages: false,
-      parsedEvents: [],
-    });
-    (
-      useOptimisticUserMessage as unknown as ReturnType<typeof vi.fn>
-    ).mockReturnValue({
-      setOptimisticUserMessage: vi.fn(),
-      getOptimisticUserMessage: vi.fn(() => null),
-    });
-    (useWSErrorMessage as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      getErrorMessage: vi.fn(() => null),
-      setErrorMessage: vi.fn(),
-      removeErrorMessage: vi.fn(),
-    });
-    (useConfig as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      data: { APP_MODE: "local" },
-    });
-    (useGetTrajectory as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      mutate: vi.fn(),
-      mutateAsync: vi.fn(),
-      isLoading: false,
-    });
-    (useUploadFiles as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      mutateAsync: vi
-        .fn()
-        .mockResolvedValue({ skipped_files: [], uploaded_files: [] }),
-      isLoading: false,
-    });
-  });
-
-  test("should show chat suggestions when there are no events", () => {
-    (useWsClient as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      send: vi.fn(),
-      isLoadingMessages: false,
-      parsedEvents: [],
-    });
-
-    renderWithQueryClient(<ChatInterface />, queryClient);
-
-    // Check if ChatSuggestions is rendered
-    expect(screen.getByTestId("chat-suggestions")).toBeInTheDocument();
-  });
-
-  test("should show chat suggestions when there are only environment events", () => {
-    const environmentEvent: OpenHandsAction = {
-      id: 1,
-      source: "environment",
-      action: "system",
-      args: {
-        content: "source .openhands/setup.sh",
-        tools: null,
-        openhands_version: null,
-        agent_class: null,
-      },
-      message: "Running setup script",
-      timestamp: "2025-07-01T00:00:00Z",
-    };
-
-    (useWsClient as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      send: vi.fn(),
-      isLoadingMessages: false,
-      parsedEvents: [environmentEvent],
-    });
-
-    renderWithQueryClient(<ChatInterface />, queryClient);
-
-    // Check if ChatSuggestions is still rendered with environment events
-    expect(screen.getByTestId("chat-suggestions")).toBeInTheDocument();
-  });
-
-  test("should hide chat suggestions when there is a user message", () => {
-    const userEvent: OpenHandsAction = {
-      id: 1,
-      source: "user",
-      action: "message",
-      args: {
-        content: "Hello",
-        image_urls: [],
-        file_urls: [],
-      },
-      message: "Hello",
-      timestamp: "2025-07-01T00:00:00Z",
-    };
-
-    (useWsClient as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      send: vi.fn(),
-      isLoadingMessages: false,
-      parsedEvents: [userEvent],
-    });
-
-    renderWithQueryClient(<ChatInterface />, queryClient);
-
-    // Check if ChatSuggestions is not rendered with user events
-    expect(screen.queryByTestId("chat-suggestions")).not.toBeInTheDocument();
-  });
-
-  test("should hide chat suggestions when there is an optimistic user message", () => {
-    (
-      useOptimisticUserMessage as unknown as ReturnType<typeof vi.fn>
-    ).mockReturnValue({
-      setOptimisticUserMessage: vi.fn(),
-      getOptimisticUserMessage: vi.fn(() => "Optimistic message"),
-    });
-
-    renderWithQueryClient(<ChatInterface />, queryClient);
-
-    // Check if ChatSuggestions is not rendered with optimistic user message
-    expect(screen.queryByTestId("chat-suggestions")).not.toBeInTheDocument();
-  });
-});
-
-describe("ChatInterface - Empty state", () => {
+describe("Empty state", () => {
   const { send: sendMock } = vi.hoisted(() => ({
     send: vi.fn(),
   }));
@@ -258,52 +20,21 @@ describe("ChatInterface - Empty state", () => {
       send: sendMock,
       status: "CONNECTED",
       isLoadingMessages: false,
-      parsedEvents: [],
     })),
   }));
 
   beforeAll(() => {
+    vi.mock("react-router", async (importActual) => ({
+      ...(await importActual<typeof import("react-router")>()),
+      useRouteLoaderData: vi.fn(() => ({})),
+    }));
+
     vi.mock("#/context/socket", async (importActual) => ({
       ...(await importActual<typeof import("#/context/ws-client-provider")>()),
       useWsClient: useWsClientMock,
     }));
   });
 
-  beforeEach(() => {
-    // Reset mocks to ensure empty state
-    (useWsClient as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      send: sendMock,
-      status: "CONNECTED",
-      isLoadingMessages: false,
-      parsedEvents: [],
-    });
-    (
-      useOptimisticUserMessage as unknown as ReturnType<typeof vi.fn>
-    ).mockReturnValue({
-      setOptimisticUserMessage: vi.fn(),
-      getOptimisticUserMessage: vi.fn(() => null),
-    });
-    (useWSErrorMessage as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      getErrorMessage: vi.fn(() => null),
-      setErrorMessage: vi.fn(),
-      removeErrorMessage: vi.fn(),
-    });
-    (useConfig as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      data: { APP_MODE: "local" },
-    });
-    (useGetTrajectory as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      mutate: vi.fn(),
-      mutateAsync: vi.fn(),
-      isLoading: false,
-    });
-    (useUploadFiles as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
-      mutateAsync: vi
-        .fn()
-        .mockResolvedValue({ skipped_files: [], uploaded_files: [] }),
-      isLoading: false,
-    });
-  });
-
   afterEach(() => {
     vi.clearAllMocks();
   });
@@ -311,9 +42,9 @@ describe("ChatInterface - Empty state", () => {
   it.todo("should render suggestions if empty");
 
   it("should render the default suggestions", () => {
-    renderChatInterfaceWithRouter();
+    renderWithProviders(<ChatInterface />);
 
-    const suggestions = screen.getByTestId("chat-suggestions");
+    const suggestions = screen.getByTestId("suggestions");
     const repoSuggestions = Object.keys(SUGGESTIONS.repo);
 
     // check that there are at most 4 suggestions displayed
@@ -334,19 +65,18 @@ describe("ChatInterface - Empty state", () => {
         send: sendMock,
         status: "CONNECTED",
         isLoadingMessages: false,
-        parsedEvents: [],
       }));
       const user = userEvent.setup();
-      renderChatInterfaceWithRouter();
+      renderWithProviders(<ChatInterface />);
 
-      const suggestions = screen.getByTestId("chat-suggestions");
+      const suggestions = screen.getByTestId("suggestions");
       const displayedSuggestions = within(suggestions).getAllByRole("button");
       const input = screen.getByTestId("chat-input");
 
       await user.click(displayedSuggestions[0]);
 
       // user message loaded to input
-      expect(screen.queryByTestId("chat-suggestions")).toBeInTheDocument();
+      expect(screen.queryByTestId("suggestions")).toBeInTheDocument();
       expect(input).toHaveValue(displayedSuggestions[0].textContent);
     },
   );
@@ -358,12 +88,11 @@ describe("ChatInterface - Empty state", () => {
         send: sendMock,
         status: "CONNECTED",
         isLoadingMessages: false,
-        parsedEvents: [],
       }));
       const user = userEvent.setup();
-      const { rerender } = renderChatInterfaceWithRouter();
+      const { rerender } = renderWithProviders(<ChatInterface />);
 
-      const suggestions = screen.getByTestId("chat-suggestions");
+      const suggestions = screen.getByTestId("suggestions");
       const displayedSuggestions = within(suggestions).getAllByRole("button");
 
       await user.click(displayedSuggestions[0]);
@@ -373,13 +102,8 @@ describe("ChatInterface - Empty state", () => {
         send: sendMock,
         status: "CONNECTED",
         isLoadingMessages: false,
-        parsedEvents: [],
       }));
-      rerender(
-        <MemoryRouter>
-          <ChatInterface />
-        </MemoryRouter>,
-      );
+      rerender(<ChatInterface />);
 
       await waitFor(() =>
         expect(sendMock).toHaveBeenCalledWith(expect.any(String)),
@@ -388,7 +112,7 @@ describe("ChatInterface - Empty state", () => {
   );
 });
 
-describe.skip("ChatInterface - General functionality", () => {
+describe.skip("ChatInterface", () => {
   beforeAll(() => {
     // mock useScrollToBottom hook
     vi.mock("#/hooks/useScrollToBottom", () => ({
@@ -469,11 +193,7 @@ describe.skip("ChatInterface - General functionality", () => {
       },
     ];
 
-    rerender(
-      <MemoryRouter>
-        <ChatInterface />
-      </MemoryRouter>,
-    );
+    rerender(<ChatInterface />);
 
     const imageCarousel = screen.getByTestId("image-carousel");
     expect(imageCarousel).toBeInTheDocument();
@@ -512,11 +232,7 @@ describe.skip("ChatInterface - General functionality", () => {
       pending: true,
     });
 
-    rerender(
-      <MemoryRouter>
-        <ChatInterface />
-      </MemoryRouter>,
-    );
+    rerender(<ChatInterface />);
 
     expect(screen.getByTestId("continue-action-button")).toBeInTheDocument();
   });
@@ -544,7 +260,10 @@ describe.skip("ChatInterface - General functionality", () => {
   });
 
   it("should render both GitHub buttons initially when ghToken is available", () => {
-    // Note: This test may need adjustment since useRouteLoaderData is now globally mocked
+    vi.mock("react-router", async (importActual) => ({
+      ...(await importActual<typeof import("react-router")>()),
+      useRouteLoaderData: vi.fn(() => ({ ghToken: "test-token" })),
+    }));
 
     const messages: Message[] = [
       {
@@ -567,7 +286,10 @@ describe.skip("ChatInterface - General functionality", () => {
   });
 
   it("should render only 'Push changes to PR' button after PR is created", async () => {
-    // Note: This test may need adjustment since useRouteLoaderData is now globally mocked
+    vi.mock("react-router", async (importActual) => ({
+      ...(await importActual<typeof import("react-router")>()),
+      useRouteLoaderData: vi.fn(() => ({ ghToken: "test-token" })),
+    }));
 
     const messages: Message[] = [
       {
@@ -586,11 +308,7 @@ describe.skip("ChatInterface - General functionality", () => {
     await user.click(prButton);
 
     // Re-render to trigger state update
-    rerender(
-      <MemoryRouter>
-        <ChatInterface />
-      </MemoryRouter>,
-    );
+    rerender(<ChatInterface />);
 
     // Verify only one button is shown
     const pushToPrButton = screen.getByRole("button", {
@@ -640,11 +358,7 @@ describe.skip("ChatInterface - General functionality", () => {
       pending: true,
     });
 
-    rerender(
-      <MemoryRouter>
-        <ChatInterface />
-      </MemoryRouter>,
-    );
+    rerender(<ChatInterface />);
 
     expect(screen.getByTestId("feedback-actions")).toBeInTheDocument();
   });

frontend/__tests__/components/chat/expandable-message.test.tsx

@@ -3,7 +3,7 @@ import { screen } from "@testing-library/react";
 import { renderWithProviders } from "test-utils";
 import { createRoutesStub } from "react-router";
 import { ExpandableMessage } from "#/components/features/chat/expandable-message";
-import OptionService from "#/api/option-service/option-service.api";
+import OpenHands from "#/api/open-hands";
 
 vi.mock("react-i18next", async () => {
   const actual = await vi.importActual("react-i18next");
@@ -113,7 +113,7 @@ describe("ExpandableMessage", () => {
   });
 
   it("should render the out of credits message when the user is out of credits", async () => {
-    const getConfigSpy = vi.spyOn(OptionService, "getConfig");
+    const getConfigSpy = vi.spyOn(OpenHands, "getConfig");
     // @ts-expect-error - We only care about the APP_MODE and FEATURE_FLAGS fields
     getConfigSpy.mockResolvedValue({
       APP_MODE: "saas",

frontend/__tests__/components/context-menu/account-settings-context-menu.test.tsx

@@ -2,8 +2,6 @@ import { render, screen } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
 import { afterEach, describe, expect, it, test, vi } from "vitest";
 import { AccountSettingsContextMenu } from "#/components/features/context-menu/account-settings-context-menu";
-import { MemoryRouter } from "react-router";
-import { renderWithProviders } from "../../../test-utils";
 
 describe("AccountSettingsContextMenu", () => {
   const user = userEvent.setup();
@@ -11,11 +9,6 @@ describe("AccountSettingsContextMenu", () => {
   const onLogoutMock = vi.fn();
   const onCloseMock = vi.fn();
 
-  // Create a wrapper with MemoryRouter and renderWithProviders
-  const renderWithRouter = (ui: React.ReactElement) => {
-    return renderWithProviders(<MemoryRouter>{ui}</MemoryRouter>);
-  };
-
   afterEach(() => {
     onClickAccountSettingsMock.mockClear();
     onLogoutMock.mockClear();
@@ -23,7 +16,7 @@ describe("AccountSettingsContextMenu", () => {
   });
 
   it("should always render the right options", () => {
-    renderWithRouter(
+    render(
       <AccountSettingsContextMenu
         onLogout={onLogoutMock}
         onClose={onCloseMock}
@@ -37,7 +30,7 @@ describe("AccountSettingsContextMenu", () => {
   });
 
   it("should call onLogout when the logout option is clicked", async () => {
-    renderWithRouter(
+    render(
       <AccountSettingsContextMenu
         onLogout={onLogoutMock}
         onClose={onCloseMock}
@@ -51,7 +44,7 @@ describe("AccountSettingsContextMenu", () => {
   });
 
   test("logout button is always enabled", async () => {
-    renderWithRouter(
+    render(
       <AccountSettingsContextMenu
         onLogout={onLogoutMock}
         onClose={onCloseMock}
@@ -65,7 +58,7 @@ describe("AccountSettingsContextMenu", () => {
   });
 
   it("should call onClose when clicking outside of the element", async () => {
-    renderWithRouter(
+    render(
       <AccountSettingsContextMenu
         onLogout={onLogoutMock}
         onClose={onCloseMock}

frontend/__tests__/components/features/analytics/analytics-consent-form-modal.test.tsx

@@ -3,13 +3,13 @@ import { describe, expect, it, vi } from "vitest";
 import { render, screen, waitFor } from "@testing-library/react";
 import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
 import { AnalyticsConsentFormModal } from "#/components/features/analytics/analytics-consent-form-modal";
-import SettingsService from "#/settings-service/settings-service.api";
+import OpenHands from "#/api/open-hands";
 
 describe("AnalyticsConsentFormModal", () => {
   it("should call saveUserSettings with consent", async () => {
     const user = userEvent.setup();
     const onCloseMock = vi.fn();
-    const saveUserSettingsSpy = vi.spyOn(SettingsService, "saveSettings");
+    const saveUserSettingsSpy = vi.spyOn(OpenHands, "saveSettings");
 
     render(<AnalyticsConsentFormModal onClose={onCloseMock} />, {
       wrapper: ({ children }) => (

frontend/__tests__/components/features/chat/messages.test.tsx

@@ -8,7 +8,7 @@ import {
   UserMessageAction,
 } from "#/types/core/actions";
 import { OpenHandsObservation } from "#/types/core/observations";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
+import OpenHands from "#/api/open-hands";
 import { Conversation } from "#/api/open-hands.types";
 
 vi.mock("react-router", () => ({
@@ -80,7 +80,7 @@ describe("Messages", () => {
   });
 
   it("should render a launch to microagent action button on chat messages only if it is a user message", () => {
-    const getConversationSpy = vi.spyOn(ConversationService, "getConversation");
+    const getConversationSpy = vi.spyOn(OpenHands, "getConversation");
     const mockConversation: Conversation = {
       conversation_id: "123",
       title: "Test Conversation",

frontend/__tests__/components/features/conversation-panel/conversation-card.test.tsx

@@ -12,7 +12,7 @@ import {
 import userEvent from "@testing-library/user-event";
 import { renderWithProviders } from "test-utils";
 import { formatTimeDelta } from "#/utils/format-time-delta";
-import { ConversationCard } from "#/components/features/conversation-panel/conversation-card/conversation-card";
+import { ConversationCard } from "#/components/features/conversation-panel/conversation-card";
 import { clickOnEditButton } from "./utils";
 
 // We'll use the actual i18next implementation but override the translation function
@@ -64,6 +64,7 @@ describe("ConversationCard", () => {
       <ConversationCard
         onDelete={onDelete}
         onChangeTitle={onChangeTitle}
+        isActive
         title="Conversation 1"
         selectedRepository={null}
         lastUpdatedAt="2021-10-01T12:00:00Z"
@@ -75,6 +76,7 @@ describe("ConversationCard", () => {
     within(card).getByText("Conversation 1");
 
     // Just check that the card contains the expected text content
+    expect(card).toHaveTextContent("Created");
     expect(card).toHaveTextContent("ago");
 
     // Use a regex to match the time part since it might have whitespace
@@ -89,6 +91,7 @@ describe("ConversationCard", () => {
       <ConversationCard
         onDelete={onDelete}
         onChangeTitle={onChangeTitle}
+        isActive
         title="Conversation 1"
         selectedRepository={null}
         lastUpdatedAt="2021-10-01T12:00:00Z"
@@ -103,6 +106,7 @@ describe("ConversationCard", () => {
       <ConversationCard
         onDelete={onDelete}
         onChangeTitle={onChangeTitle}
+        isActive
         title="Conversation 1"
         selectedRepository={{
           selected_repository: "org/selectedRepository",
@@ -123,6 +127,7 @@ describe("ConversationCard", () => {
       <ConversationCard
         onDelete={onDelete}
         onChangeTitle={onChangeTitle}
+        isActive
         title="Conversation 1"
         selectedRepository={null}
         lastUpdatedAt="2021-10-01T12:00:00Z"
@@ -131,14 +136,7 @@ describe("ConversationCard", () => {
       />,
     );
 
-    // Context menu is always in the DOM but hidden by CSS classes when contextMenuOpen is false
-    const contextMenu = screen.queryByTestId("context-menu");
-    if (contextMenu) {
-      const contextMenuParent = contextMenu.parentElement;
-      if (contextMenuParent) {
-        expect(contextMenuParent).toHaveClass("opacity-0", "invisible");
-      }
-    }
+    expect(screen.queryByTestId("context-menu")).not.toBeInTheDocument();
 
     const ellipsisButton = screen.getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
@@ -150,6 +148,7 @@ describe("ConversationCard", () => {
       <ConversationCard
         onDelete={onDelete}
         onChangeTitle={onChangeTitle}
+        isActive
         title="Conversation 1"
         selectedRepository={null}
         lastUpdatedAt="2021-10-01T12:00:00Z"
@@ -171,6 +170,7 @@ describe("ConversationCard", () => {
     renderWithProviders(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         onChangeTitle={onChangeTitle}
         title="Conversation 1"
         selectedRepository={null}
@@ -194,6 +194,7 @@ describe("ConversationCard", () => {
     renderWithProviders(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         onChangeTitle={onChangeTitle}
         title="Conversation 1"
         selectedRepository={{
@@ -222,6 +223,7 @@ describe("ConversationCard", () => {
     const { rerender } = renderWithProviders(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         title="Conversation 1"
         selectedRepository={null}
         lastUpdatedAt="2021-10-01T12:00:00Z"
@@ -237,6 +239,7 @@ describe("ConversationCard", () => {
     rerender(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         title="Conversation 1"
         selectedRepository={null}
         lastUpdatedAt="2021-10-01T12:00:00Z"
@@ -249,14 +252,7 @@ describe("ConversationCard", () => {
     const title = screen.getByTestId("conversation-card-title");
 
     expect(title).toBeEnabled();
-    // Context menu should be hidden after edit button is clicked (check CSS classes on parent div)
-    const contextMenu = screen.queryByTestId("context-menu");
-    if (contextMenu) {
-      const contextMenuParent = contextMenu.parentElement;
-      if (contextMenuParent) {
-        expect(contextMenuParent).toHaveClass("opacity-0", "invisible");
-      }
-    }
+    expect(screen.queryByTestId("context-menu")).not.toBeInTheDocument();
     // expect to be focused
     expect(document.activeElement).toBe(title);
 
@@ -265,14 +261,16 @@ describe("ConversationCard", () => {
     await user.tab();
 
     expect(onChangeTitle).toHaveBeenCalledWith("New Conversation Name");
+    expect(title).toHaveValue("New Conversation Name");
   });
 
-  it("should not call onChange title", async () => {
+  it("should reset title and not call onChangeTitle when the title is empty", async () => {
     const user = userEvent.setup();
     const onContextMenuToggle = vi.fn();
     renderWithProviders(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         onChangeTitle={onChangeTitle}
         title="Conversation 1"
         selectedRepository={null}
@@ -289,7 +287,8 @@ describe("ConversationCard", () => {
     await user.clear(title);
     await user.tab();
 
-    expect(onChangeTitle).not.toBeCalled();
+    expect(onChangeTitle).not.toHaveBeenCalled();
+    expect(title).toHaveValue("Conversation 1");
   });
 
   test("clicking the title should trigger the onClick handler", async () => {
@@ -298,6 +297,7 @@ describe("ConversationCard", () => {
       <ConversationCard
         onClick={onClick}
         onDelete={onDelete}
+        isActive
         onChangeTitle={onChangeTitle}
         title="Conversation 1"
         selectedRepository={null}
@@ -317,6 +317,7 @@ describe("ConversationCard", () => {
     renderWithProviders(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         onChangeTitle={onChangeTitle}
         title="Conversation 1"
         selectedRepository={null}
@@ -340,6 +341,7 @@ describe("ConversationCard", () => {
     renderWithProviders(
       <ConversationCard
         onDelete={onDelete}
+        isActive
         onChangeTitle={onChangeTitle}
         title="Conversation 1"
         selectedRepository={null}
@@ -357,6 +359,72 @@ describe("ConversationCard", () => {
     expect(onClick).not.toHaveBeenCalled();
   });
 
+  it("should show display cost button only when showOptions is true", async () => {
+    const onContextMenuToggle = vi.fn();
+    const { rerender } = renderWithProviders(
+      <ConversationCard
+        onDelete={onDelete}
+        onChangeTitle={onChangeTitle}
+        isActive
+        title="Conversation 1"
+        selectedRepository={null}
+        lastUpdatedAt="2021-10-01T12:00:00Z"
+        contextMenuOpen
+        onContextMenuToggle={onContextMenuToggle}
+      />,
+    );
+
+    // Wait for context menu to appear
+    const menu = await screen.findByTestId("context-menu");
+    expect(
+      within(menu).queryByTestId("display-cost-button"),
+    ).not.toBeInTheDocument();
+
+    rerender(
+      <ConversationCard
+        onDelete={onDelete}
+        onChangeTitle={onChangeTitle}
+        showOptions
+        isActive
+        title="Conversation 1"
+        selectedRepository={null}
+        lastUpdatedAt="2021-10-01T12:00:00Z"
+        contextMenuOpen
+        onContextMenuToggle={onContextMenuToggle}
+      />,
+    );
+
+    // Wait for context menu to appear and check for display cost button
+    const newMenu = await screen.findByTestId("context-menu");
+    within(newMenu).getByTestId("display-cost-button");
+  });
+
+  it("should show metrics modal when clicking the display cost button", async () => {
+    const user = userEvent.setup();
+    const onContextMenuToggle = vi.fn();
+    renderWithProviders(
+      <ConversationCard
+        onDelete={onDelete}
+        isActive
+        onChangeTitle={onChangeTitle}
+        title="Conversation 1"
+        selectedRepository={null}
+        lastUpdatedAt="2021-10-01T12:00:00Z"
+        showOptions
+        contextMenuOpen
+        onContextMenuToggle={onContextMenuToggle}
+      />,
+    );
+
+    const menu = screen.getByTestId("context-menu");
+    const displayCostButton = within(menu).getByTestId("display-cost-button");
+
+    await user.click(displayCostButton);
+
+    // Verify if metrics modal is displayed by checking for the modal content
+    expect(screen.getByTestId("metrics-modal")).toBeInTheDocument();
+  });
+
   it("should not display the edit or delete options if the handler is not provided", async () => {
     const onContextMenuToggle = vi.fn();
     const { rerender } = renderWithProviders(
@@ -431,4 +499,38 @@ describe("ConversationCard", () => {
 
     expect(screen.queryByTestId("ellipsis-button")).not.toBeInTheDocument();
   });
+
+  describe("state indicator", () => {
+    it("should render the 'STOPPED' indicator by default", () => {
+      renderWithProviders(
+        <ConversationCard
+          onDelete={onDelete}
+          isActive
+          onChangeTitle={onChangeTitle}
+          title="Conversation 1"
+          selectedRepository={null}
+          lastUpdatedAt="2021-10-01T12:00:00Z"
+        />,
+      );
+
+      screen.getByTestId("STOPPED-indicator");
+    });
+
+    it("should render the other indicators when provided", () => {
+      renderWithProviders(
+        <ConversationCard
+          onDelete={onDelete}
+          isActive
+          onChangeTitle={onChangeTitle}
+          title="Conversation 1"
+          selectedRepository={null}
+          lastUpdatedAt="2021-10-01T12:00:00Z"
+          conversationStatus="RUNNING"
+        />,
+      );
+
+      expect(screen.queryByTestId("STOPPED-indicator")).not.toBeInTheDocument();
+      screen.getByTestId("RUNNING-indicator");
+    });
+  });
 });

frontend/__tests__/components/features/conversation-panel/conversation-panel.test.tsx

@@ -1,11 +1,12 @@
 import { screen, waitFor, within } from "@testing-library/react";
 import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
+import { QueryClientConfig } from "@tanstack/react-query";
 import userEvent from "@testing-library/user-event";
 import { createRoutesStub } from "react-router";
 import React from "react";
-import { renderWithQueryAndI18n } from "test-utils";
+import { renderWithProviders } from "test-utils";
 import { ConversationPanel } from "#/components/features/conversation-panel/conversation-panel";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
+import OpenHands from "#/api/open-hands";
 import { Conversation } from "#/api/open-hands.types";
 
 describe("ConversationPanel", () => {
@@ -17,7 +18,16 @@ describe("ConversationPanel", () => {
     },
   ]);
 
-  const renderConversationPanel = () => renderWithQueryAndI18n(<RouterStub />);
+  const renderConversationPanel = (config?: QueryClientConfig) =>
+    renderWithProviders(<RouterStub />, {
+      preloadedState: {
+        metrics: {
+          cost: null,
+          max_budget_per_task: null,
+          usage: null,
+        },
+      },
+    });
 
   beforeAll(() => {
     vi.mock("react-router", async (importOriginal) => ({
@@ -75,7 +85,7 @@ describe("ConversationPanel", () => {
     vi.clearAllMocks();
     vi.restoreAllMocks();
     // Setup default mock for getUserConversations
-    vi.spyOn(ConversationService, "getUserConversations").mockResolvedValue({
+    vi.spyOn(OpenHands, "getUserConversations").mockResolvedValue({
       results: [...mockConversations],
       next_page_id: null,
     });
@@ -91,10 +101,7 @@ describe("ConversationPanel", () => {
   });
 
   it("should display an empty state when there are no conversations", async () => {
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockResolvedValue({
       results: [],
       next_page_id: null,
@@ -107,10 +114,7 @@ describe("ConversationPanel", () => {
   });
 
   it("should handle an error when fetching conversations", async () => {
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockRejectedValue(
       new Error("Failed to fetch conversations"),
     );
@@ -126,18 +130,13 @@ describe("ConversationPanel", () => {
     renderConversationPanel();
 
     let cards = await screen.findAllByTestId("conversation-card");
-    // Delete button should not be visible initially (context menu is closed)
-    // The context menu is always in the DOM but hidden by CSS classes on the parent div
-    const contextMenuParent = within(cards[0]).queryByTestId(
-      "context-menu",
-    )?.parentElement;
-    if (contextMenuParent) {
-      expect(contextMenuParent).toHaveClass("opacity-0", "invisible");
-    }
+    expect(
+      within(cards[0]).queryByTestId("delete-button"),
+    ).not.toBeInTheDocument();
 
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
-    const deleteButton = within(cards[0]).getByTestId("delete-button");
+    const deleteButton = screen.getByTestId("delete-button");
 
     // Click the first delete button
     await user.click(deleteButton);
@@ -199,17 +198,14 @@ describe("ConversationPanel", () => {
       },
     ];
 
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockImplementation(async () => ({
       results: mockData,
       next_page_id: null,
     }));
 
     const deleteUserConversationSpy = vi.spyOn(
-      ConversationService,
+      OpenHands,
       "deleteUserConversation",
     );
     deleteUserConversationSpy.mockImplementation(async (id: string) => {
@@ -226,7 +222,7 @@ describe("ConversationPanel", () => {
 
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
-    const deleteButton = within(cards[0]).getByTestId("delete-button");
+    const deleteButton = screen.getByTestId("delete-button");
 
     // Click the first delete button
     await user.click(deleteButton);
@@ -259,10 +255,7 @@ describe("ConversationPanel", () => {
 
   it("should refetch data on rerenders", async () => {
     const user = userEvent.setup();
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockResolvedValue({
       results: [...mockConversations],
       next_page_id: null,
@@ -287,7 +280,15 @@ describe("ConversationPanel", () => {
       },
     ]);
 
-    renderWithQueryAndI18n(<MyRouterStub />);
+    renderWithProviders(<MyRouterStub />, {
+      preloadedState: {
+        metrics: {
+          cost: null,
+          max_budget_per_task: null,
+          usage: null,
+        },
+      },
+    });
 
     const toggleButton = screen.getByText("Toggle");
 
@@ -351,10 +352,7 @@ describe("ConversationPanel", () => {
       },
     ];
 
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockResolvedValue({
       results: mockRunningConversations,
       next_page_id: null,
@@ -370,7 +368,7 @@ describe("ConversationPanel", () => {
     await user.click(ellipsisButton);
 
     // Stop button should be available for RUNNING conversation
-    const stopButton = within(cards[0]).getByTestId("stop-button");
+    const stopButton = screen.getByTestId("stop-button");
     expect(stopButton).toBeInTheDocument();
 
     // Click the stop button
@@ -421,19 +419,13 @@ describe("ConversationPanel", () => {
       },
     ];
 
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockImplementation(async () => ({
       results: mockData,
       next_page_id: null,
     }));
 
-    const stopConversationSpy = vi.spyOn(
-      ConversationService,
-      "stopConversation",
-    );
+    const stopConversationSpy = vi.spyOn(OpenHands, "stopConversation");
     stopConversationSpy.mockImplementation(async (id: string) => {
       const conversation = mockData.find((conv) => conv.conversation_id === id);
       if (conversation) {
@@ -452,7 +444,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const stopButton = within(cards[0]).getByTestId("stop-button");
+    const stopButton = screen.getByTestId("stop-button");
 
     // Click the stop button
     await user.click(stopButton);
@@ -515,10 +507,7 @@ describe("ConversationPanel", () => {
       },
     ];
 
-    const getUserConversationsSpy = vi.spyOn(
-      ConversationService,
-      "getUserConversations",
-    );
+    const getUserConversationsSpy = vi.spyOn(OpenHands, "getUserConversations");
     getUserConversationsSpy.mockResolvedValue({
       results: mockMixedStatusConversations,
       next_page_id: null,
@@ -535,51 +524,29 @@ describe("ConversationPanel", () => {
     );
     await user.click(runningEllipsisButton);
 
-    expect(within(cards[0]).getByTestId("stop-button")).toBeInTheDocument();
+    expect(screen.getByTestId("stop-button")).toBeInTheDocument();
 
     // Click outside to close the menu
     await user.click(document.body);
 
-    // Wait for context menu to close (check CSS classes on parent div)
-    await waitFor(() => {
-      const contextMenuParent = within(cards[0]).queryByTestId(
-        "context-menu",
-      )?.parentElement;
-      if (contextMenuParent) {
-        expect(contextMenuParent).toHaveClass("opacity-0", "invisible");
-      }
-    });
-
     // Test STARTING conversation - should show stop button
     const startingEllipsisButton = within(cards[1]).getByTestId(
       "ellipsis-button",
     );
     await user.click(startingEllipsisButton);
 
-    expect(within(cards[1]).getByTestId("stop-button")).toBeInTheDocument();
+    expect(screen.getByTestId("stop-button")).toBeInTheDocument();
 
     // Click outside to close the menu
     await user.click(document.body);
 
-    // Wait for context menu to close (check CSS classes on parent div)
-    await waitFor(() => {
-      const contextMenuParent = within(cards[1]).queryByTestId(
-        "context-menu",
-      )?.parentElement;
-      if (contextMenuParent) {
-        expect(contextMenuParent).toHaveClass("opacity-0", "invisible");
-      }
-    });
-
     // Test STOPPED conversation - should NOT show stop button
     const stoppedEllipsisButton = within(cards[2]).getByTestId(
       "ellipsis-button",
     );
     await user.click(stoppedEllipsisButton);
 
-    expect(
-      within(cards[2]).queryByTestId("stop-button"),
-    ).not.toBeInTheDocument();
+    expect(screen.queryByTestId("stop-button")).not.toBeInTheDocument();
   });
 
   it("should show edit button in context menu", async () => {
@@ -593,10 +560,10 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    // Edit button should be visible within the first card's context menu
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    // Edit button should be visible
+    const editButton = screen.getByTestId("edit-button");
     expect(editButton).toBeInTheDocument();
-    expect(editButton).toHaveTextContent("BUTTON$RENAME");
+    expect(editButton).toHaveTextContent("BUTTON$EDIT_TITLE");
   });
 
   it("should enter edit mode when edit button is clicked", async () => {
@@ -609,8 +576,8 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    // Click edit button within the first card's context menu
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    // Click edit button
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Should find input field instead of title text
@@ -625,10 +592,7 @@ describe("ConversationPanel", () => {
     const user = userEvent.setup();
 
     // Mock the updateConversation API call
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockResolvedValue(true);
 
     // Mock the toast function
@@ -645,7 +609,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Edit the title
@@ -665,10 +629,7 @@ describe("ConversationPanel", () => {
   it("should save title when Enter key is pressed", async () => {
     const user = userEvent.setup();
 
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockResolvedValue(true);
 
     renderConversationPanel();
@@ -679,7 +640,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Edit the title and press Enter
@@ -697,10 +658,7 @@ describe("ConversationPanel", () => {
   it("should trim whitespace from title", async () => {
     const user = userEvent.setup();
 
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockResolvedValue(true);
 
     renderConversationPanel();
@@ -711,7 +669,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Edit the title with extra whitespace
@@ -724,15 +682,15 @@ describe("ConversationPanel", () => {
     expect(updateConversationSpy).toHaveBeenCalledWith("1", {
       title: "Trimmed Title",
     });
+
+    // Verify input shows trimmed value
+    expect(titleInput).toHaveValue("Trimmed Title");
   });
 
   it("should revert to original title when empty", async () => {
     const user = userEvent.setup();
 
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockResolvedValue(true);
 
     renderConversationPanel();
@@ -743,7 +701,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Clear the title completely
@@ -753,15 +711,15 @@ describe("ConversationPanel", () => {
 
     // Verify API was not called
     expect(updateConversationSpy).not.toHaveBeenCalled();
+
+    // Verify input reverted to original value
+    expect(titleInput).toHaveValue("Conversation 1");
   });
 
   it("should handle API error when updating title", async () => {
     const user = userEvent.setup();
 
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockRejectedValue(new Error("API Error"));
 
     vi.mock("#/utils/custom-toast-handlers", () => ({
@@ -776,7 +734,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Edit the title
@@ -806,32 +764,22 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    // Verify context menu is open within the first card
-    const contextMenu = within(cards[0]).getByTestId("context-menu");
+    // Verify context menu is open
+    const contextMenu = screen.getByTestId("context-menu");
     expect(contextMenu).toBeInTheDocument();
 
-    // Click edit button within the first card's context menu
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    // Click edit button
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
-    // Wait for context menu to close after edit button click (check CSS classes on parent div)
-    await waitFor(() => {
-      const contextMenuParent = within(cards[0]).queryByTestId(
-        "context-menu",
-      )?.parentElement;
-      if (contextMenuParent) {
-        expect(contextMenuParent).toHaveClass("opacity-0", "invisible");
-      }
-    });
+    // Verify context menu is closed
+    expect(screen.queryByTestId("context-menu")).not.toBeInTheDocument();
   });
 
   it("should not call API when title is unchanged", async () => {
     const user = userEvent.setup();
 
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockResolvedValue(true);
 
     renderConversationPanel();
@@ -842,14 +790,15 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Don't change the title, just blur
+    const titleInput = within(cards[0]).getByTestId("conversation-card-title");
     await user.tab();
 
-    // Verify API was NOT called with the same title (since handleConversationTitleChange will always be called)
-    expect(updateConversationSpy).not.toHaveBeenCalledWith("1", {
+    // Verify API was called with the same title (since handleConversationTitleChange will always be called)
+    expect(updateConversationSpy).toHaveBeenCalledWith("1", {
       title: "Conversation 1",
     });
   });
@@ -857,10 +806,7 @@ describe("ConversationPanel", () => {
   it("should handle special characters in title", async () => {
     const user = userEvent.setup();
 
-    const updateConversationSpy = vi.spyOn(
-      ConversationService,
-      "updateConversation",
-    );
+    const updateConversationSpy = vi.spyOn(OpenHands, "updateConversation");
     updateConversationSpy.mockResolvedValue(true);
 
     renderConversationPanel();
@@ -871,7 +817,7 @@ describe("ConversationPanel", () => {
     const ellipsisButton = within(cards[0]).getByTestId("ellipsis-button");
     await user.click(ellipsisButton);
 
-    const editButton = within(cards[0]).getByTestId("edit-button");
+    const editButton = screen.getByTestId("edit-button");
     await user.click(editButton);
 
     // Edit the title with special characters

frontend/__tests__/components/features/conversation/conversation-name.test.tsx

@@ -1,573 +0,0 @@
-import { screen, within } from "@testing-library/react";
-import userEvent from "@testing-library/user-event";
-import { afterEach, beforeAll, describe, expect, it, vi } from "vitest";
-import { renderWithProviders } from "test-utils";
-import { ConversationName } from "#/components/features/conversation/conversation-name";
-import { ConversationNameContextMenu } from "#/components/features/conversation/conversation-name-context-menu";
-import { BrowserRouter } from "react-router";
-
-// Mock the hooks and utilities
-const mockMutate = vi.fn();
-
-vi.mock("#/hooks/query/use-active-conversation", () => ({
-  useActiveConversation: () => ({
-    data: {
-      conversation_id: "test-conversation-id",
-      title: "Test Conversation",
-      status: "RUNNING",
-    },
-  }),
-}));
-
-vi.mock("#/hooks/mutation/use-update-conversation", () => ({
-  useUpdateConversation: () => ({
-    mutate: mockMutate,
-  }),
-}));
-
-vi.mock("#/utils/custom-toast-handlers", () => ({
-  displaySuccessToast: vi.fn(),
-}));
-
-// Mock react-i18next
-vi.mock("react-i18next", async () => {
-  const actual = await vi.importActual("react-i18next");
-  return {
-    ...actual,
-    useTranslation: () => ({
-      t: (key: string) => {
-        const translations: Record<string, string> = {
-          CONVERSATION$TITLE_UPDATED: "Conversation title updated",
-          BUTTON$RENAME: "Rename",
-          BUTTON$EXPORT_CONVERSATION: "Export Conversation",
-          BUTTON$DOWNLOAD_VIA_VSCODE: "Download via VS Code",
-          BUTTON$SHOW_AGENT_TOOLS_AND_METADATA: "Show Agent Tools",
-          CONVERSATION$SHOW_MICROAGENTS: "Show Microagents",
-          BUTTON$DISPLAY_COST: "Display Cost",
-          COMMON$CLOSE_CONVERSATION_STOP_RUNTIME:
-            "Close Conversation (Stop Runtime)",
-          COMMON$DELETE_CONVERSATION: "Delete Conversation",
-        };
-        return translations[key] || key;
-      },
-      i18n: {
-        changeLanguage: () => new Promise(() => {}),
-      },
-    }),
-  };
-});
-
-// Helper function to render ConversationName with Router context
-const renderConversationNameWithRouter = () => {
-  return renderWithProviders(
-    <BrowserRouter>
-      <ConversationName />
-    </BrowserRouter>,
-  );
-};
-
-describe("ConversationName", () => {
-  beforeAll(() => {
-    vi.stubGlobal("window", {
-      open: vi.fn(),
-      addEventListener: vi.fn(),
-      removeEventListener: vi.fn(),
-    });
-  });
-
-  afterEach(() => {
-    vi.clearAllMocks();
-  });
-
-  it("should render the conversation name in view mode", () => {
-    renderConversationNameWithRouter();
-
-    const container = screen.getByTestId("conversation-name");
-    const titleElement = within(container).getByTestId(
-      "conversation-name-title",
-    );
-
-    expect(container).toBeInTheDocument();
-    expect(titleElement).toBeInTheDocument();
-    expect(titleElement).toHaveTextContent("Test Conversation");
-  });
-
-  it("should switch to edit mode on double click", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-
-    // Initially should be in view mode
-    expect(titleElement).toBeInTheDocument();
-    expect(
-      screen.queryByTestId("conversation-name-input"),
-    ).not.toBeInTheDocument();
-
-    // Double click to enter edit mode
-    await user.dblClick(titleElement);
-
-    // Should now be in edit mode
-    expect(
-      screen.queryByTestId("conversation-name-title"),
-    ).not.toBeInTheDocument();
-    const inputElement = screen.getByTestId("conversation-name-input");
-    expect(inputElement).toBeInTheDocument();
-    expect(inputElement).toHaveValue("Test Conversation");
-  });
-
-  it("should update conversation title when input loses focus with valid value", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    await user.clear(inputElement);
-    await user.type(inputElement, "New Conversation Title");
-    await user.tab(); // Trigger blur event
-
-    // Verify that the update function was called
-    expect(mockMutate).toHaveBeenCalledWith(
-      {
-        conversationId: "test-conversation-id",
-        newTitle: "New Conversation Title",
-      },
-      expect.any(Object),
-    );
-  });
-
-  it("should not update conversation when title is unchanged", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    // Keep the same title
-    await user.tab();
-
-    // Should still have the original title
-    expect(inputElement).toHaveValue("Test Conversation");
-  });
-
-  it("should not call the API if user attempts to save an unchanged title", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-
-    // Verify the input has the original title
-    expect(inputElement).toHaveValue("Test Conversation");
-
-    // Trigger blur without changing the title
-    await user.tab();
-
-    // Verify that the API was NOT called
-    expect(mockMutate).not.toHaveBeenCalled();
-  });
-
-  it("should reset input value when title is empty and blur", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    await user.clear(inputElement);
-    await user.tab();
-
-    // Should reset to original title
-    expect(inputElement).toHaveValue("Test Conversation");
-  });
-
-  it("should trim whitespace from input value", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    await user.clear(inputElement);
-    await user.type(inputElement, "  Trimmed Title  ");
-    await user.tab();
-
-    // Should call mutation with trimmed value
-    expect(mockMutate).toHaveBeenCalledWith(
-      {
-        conversationId: "test-conversation-id",
-        newTitle: "Trimmed Title",
-      },
-      expect.any(Object),
-    );
-  });
-
-  it("should handle Enter key to save changes", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    await user.clear(inputElement);
-    await user.type(inputElement, "New Title");
-    await user.keyboard("{Enter}");
-
-    // Should have the new title
-    expect(inputElement).toHaveValue("New Title");
-  });
-
-  it("should prevent event propagation when clicking input in edit mode", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    const clickEvent = new MouseEvent("click", { bubbles: true });
-    const preventDefaultSpy = vi.spyOn(clickEvent, "preventDefault");
-    const stopPropagationSpy = vi.spyOn(clickEvent, "stopPropagation");
-
-    inputElement.dispatchEvent(clickEvent);
-
-    expect(preventDefaultSpy).toHaveBeenCalled();
-    expect(stopPropagationSpy).toHaveBeenCalled();
-  });
-
-  it("should return to view mode after blur", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    // Should be in edit mode
-    expect(screen.getByTestId("conversation-name-input")).toBeInTheDocument();
-
-    await user.tab();
-
-    // Should be back in view mode
-    expect(screen.getByTestId("conversation-name-title")).toBeInTheDocument();
-    expect(
-      screen.queryByTestId("conversation-name-input"),
-    ).not.toBeInTheDocument();
-  });
-
-  it("should focus input when entering edit mode", async () => {
-    const user = userEvent.setup();
-    renderConversationNameWithRouter();
-
-    const titleElement = screen.getByTestId("conversation-name-title");
-    await user.dblClick(titleElement);
-
-    const inputElement = screen.getByTestId("conversation-name-input");
-    expect(inputElement).toHaveFocus();
-  });
-});
-
-describe("ConversationNameContextMenu", () => {
-  const defaultProps = {
-    onClose: vi.fn(),
-  };
-
-  afterEach(() => {
-    vi.clearAllMocks();
-  });
-
-  it("should render all menu options when all handlers are provided", () => {
-    const handlers = {
-      onRename: vi.fn(),
-      onDelete: vi.fn(),
-      onStop: vi.fn(),
-      onDisplayCost: vi.fn(),
-      onShowAgentTools: vi.fn(),
-      onShowMicroagents: vi.fn(),
-      onExportConversation: vi.fn(),
-      onDownloadViaVSCode: vi.fn(),
-    };
-
-    renderWithProviders(
-      <ConversationNameContextMenu {...defaultProps} {...handlers} />,
-    );
-
-    expect(screen.getByTestId("rename-button")).toBeInTheDocument();
-    expect(screen.getByTestId("delete-button")).toBeInTheDocument();
-    expect(screen.getByTestId("stop-button")).toBeInTheDocument();
-    expect(screen.getByTestId("display-cost-button")).toBeInTheDocument();
-    expect(screen.getByTestId("show-agent-tools-button")).toBeInTheDocument();
-    expect(screen.getByTestId("show-microagents-button")).toBeInTheDocument();
-    expect(
-      screen.getByTestId("export-conversation-button"),
-    ).toBeInTheDocument();
-    expect(screen.getByTestId("download-vscode-button")).toBeInTheDocument();
-  });
-
-  it("should not render menu options when handlers are not provided", () => {
-    renderWithProviders(<ConversationNameContextMenu {...defaultProps} />);
-
-    expect(screen.queryByTestId("rename-button")).not.toBeInTheDocument();
-    expect(screen.queryByTestId("delete-button")).not.toBeInTheDocument();
-    expect(screen.queryByTestId("stop-button")).not.toBeInTheDocument();
-    expect(screen.queryByTestId("display-cost-button")).not.toBeInTheDocument();
-    expect(
-      screen.queryByTestId("show-agent-tools-button"),
-    ).not.toBeInTheDocument();
-    expect(
-      screen.queryByTestId("show-microagents-button"),
-    ).not.toBeInTheDocument();
-    expect(
-      screen.queryByTestId("export-conversation-button"),
-    ).not.toBeInTheDocument();
-    expect(
-      screen.queryByTestId("download-vscode-button"),
-    ).not.toBeInTheDocument();
-  });
-
-  it("should call rename handler when rename button is clicked", async () => {
-    const user = userEvent.setup();
-    const onRename = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu {...defaultProps} onRename={onRename} />,
-    );
-
-    const renameButton = screen.getByTestId("rename-button");
-    await user.click(renameButton);
-
-    expect(onRename).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call delete handler when delete button is clicked", async () => {
-    const user = userEvent.setup();
-    const onDelete = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu {...defaultProps} onDelete={onDelete} />,
-    );
-
-    const deleteButton = screen.getByTestId("delete-button");
-    await user.click(deleteButton);
-
-    expect(onDelete).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call stop handler when stop button is clicked", async () => {
-    const user = userEvent.setup();
-    const onStop = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu {...defaultProps} onStop={onStop} />,
-    );
-
-    const stopButton = screen.getByTestId("stop-button");
-    await user.click(stopButton);
-
-    expect(onStop).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call display cost handler when display cost button is clicked", async () => {
-    const user = userEvent.setup();
-    const onDisplayCost = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        onDisplayCost={onDisplayCost}
-      />,
-    );
-
-    const displayCostButton = screen.getByTestId("display-cost-button");
-    await user.click(displayCostButton);
-
-    expect(onDisplayCost).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call show agent tools handler when show agent tools button is clicked", async () => {
-    const user = userEvent.setup();
-    const onShowAgentTools = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        onShowAgentTools={onShowAgentTools}
-      />,
-    );
-
-    const showAgentToolsButton = screen.getByTestId("show-agent-tools-button");
-    await user.click(showAgentToolsButton);
-
-    expect(onShowAgentTools).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call show microagents handler when show microagents button is clicked", async () => {
-    const user = userEvent.setup();
-    const onShowMicroagents = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        onShowMicroagents={onShowMicroagents}
-      />,
-    );
-
-    const showMicroagentsButton = screen.getByTestId("show-microagents-button");
-    await user.click(showMicroagentsButton);
-
-    expect(onShowMicroagents).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call export conversation handler when export conversation button is clicked", async () => {
-    const user = userEvent.setup();
-    const onExportConversation = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        onExportConversation={onExportConversation}
-      />,
-    );
-
-    const exportButton = screen.getByTestId("export-conversation-button");
-    await user.click(exportButton);
-
-    expect(onExportConversation).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call download via VSCode handler when download via VSCode button is clicked", async () => {
-    const user = userEvent.setup();
-    const onDownloadViaVSCode = vi.fn();
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        onDownloadViaVSCode={onDownloadViaVSCode}
-      />,
-    );
-
-    const downloadButton = screen.getByTestId("download-vscode-button");
-    await user.click(downloadButton);
-
-    expect(onDownloadViaVSCode).toHaveBeenCalledTimes(1);
-  });
-
-  it("should render separators between logical groups", () => {
-    const handlers = {
-      onRename: vi.fn(),
-      onShowAgentTools: vi.fn(),
-      onExportConversation: vi.fn(),
-      onDisplayCost: vi.fn(),
-      onStop: vi.fn(),
-    };
-
-    renderWithProviders(
-      <ConversationNameContextMenu {...defaultProps} {...handlers} />,
-    );
-
-    // Look for separator elements using test IDs
-    expect(screen.getByTestId("separator-tools")).toBeInTheDocument();
-    expect(screen.getByTestId("separator-export")).toBeInTheDocument();
-    expect(screen.getByTestId("separator-info-control")).toBeInTheDocument();
-  });
-
-  it("should apply correct positioning class when position is top", () => {
-    const handlers = {
-      onRename: vi.fn(),
-    };
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        {...handlers}
-        position="top"
-      />,
-    );
-
-    const contextMenu = screen.getByTestId("conversation-name-context-menu");
-    expect(contextMenu).toHaveClass("bottom-full");
-  });
-
-  it("should apply correct positioning class when position is bottom", () => {
-    const handlers = {
-      onRename: vi.fn(),
-    };
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        {...handlers}
-        position="bottom"
-      />,
-    );
-
-    const contextMenu = screen.getByTestId("conversation-name-context-menu");
-    expect(contextMenu).toHaveClass("top-full");
-  });
-
-  it("should render correct text content for each menu option", () => {
-    const handlers = {
-      onRename: vi.fn(),
-      onDelete: vi.fn(),
-      onStop: vi.fn(),
-      onDisplayCost: vi.fn(),
-      onShowAgentTools: vi.fn(),
-      onShowMicroagents: vi.fn(),
-      onExportConversation: vi.fn(),
-      onDownloadViaVSCode: vi.fn(),
-    };
-
-    renderWithProviders(
-      <ConversationNameContextMenu {...defaultProps} {...handlers} />,
-    );
-
-    expect(screen.getByTestId("rename-button")).toHaveTextContent("Rename");
-    expect(screen.getByTestId("delete-button")).toHaveTextContent(
-      "Delete Conversation",
-    );
-    expect(screen.getByTestId("stop-button")).toHaveTextContent(
-      "Close Conversation (Stop Runtime)",
-    );
-    expect(screen.getByTestId("display-cost-button")).toHaveTextContent(
-      "Display Cost",
-    );
-    expect(screen.getByTestId("show-agent-tools-button")).toHaveTextContent(
-      "Show Agent Tools",
-    );
-    expect(screen.getByTestId("show-microagents-button")).toHaveTextContent(
-      "Show Microagents",
-    );
-    expect(screen.getByTestId("export-conversation-button")).toHaveTextContent(
-      "Export Conversation",
-    );
-    expect(screen.getByTestId("download-vscode-button")).toHaveTextContent(
-      "Download via VS Code",
-    );
-  });
-
-  it("should call onClose when context menu is closed", () => {
-    const onClose = vi.fn();
-    const handlers = {
-      onRename: vi.fn(),
-    };
-
-    renderWithProviders(
-      <ConversationNameContextMenu
-        {...defaultProps}
-        onClose={onClose}
-        {...handlers}
-      />,
-    );
-
-    // The onClose is typically called by the parent component when clicking outside
-    // This test verifies the prop is properly passed
-    expect(onClose).toBeDefined();
-  });
-});

frontend/__tests__/components/features/conversation/server-status.test.tsx

@@ -1,389 +0,0 @@
-import { screen } from "@testing-library/react";
-import userEvent from "@testing-library/user-event";
-import { afterEach, describe, expect, it, vi } from "vitest";
-import { renderWithProviders } from "test-utils";
-import { ServerStatus } from "#/components/features/controls/server-status";
-import { ServerStatusContextMenu } from "#/components/features/controls/server-status-context-menu";
-import { ConversationStatus } from "#/types/conversation-status";
-import { AgentState } from "#/types/agent-state";
-
-// Mock the conversation slice actions
-vi.mock("#/state/conversation-slice", () => ({
-  setShouldStopConversation: vi.fn(),
-  setShouldStartConversation: vi.fn(),
-  default: {
-    name: "conversation",
-    initialState: {
-      isRightPanelShown: true,
-      shouldStopConversation: false,
-      shouldStartConversation: false,
-    },
-    reducers: {},
-  },
-}));
-
-// Mock react-redux
-vi.mock("react-redux", () => ({
-  useSelector: vi.fn((selector) => {
-    // Mock the selector to return different agent states based on test needs
-    return {
-      curAgentState: AgentState.RUNNING,
-    };
-  }),
-  Provider: ({ children }: { children: React.ReactNode }) => children,
-}));
-
-// Mock the custom hooks
-const mockStartConversationMutate = vi.fn();
-const mockStopConversationMutate = vi.fn();
-
-vi.mock("#/hooks/mutation/use-start-conversation", () => ({
-  useStartConversation: () => ({
-    mutate: mockStartConversationMutate,
-  }),
-}));
-
-vi.mock("#/hooks/mutation/use-stop-conversation", () => ({
-  useStopConversation: () => ({
-    mutate: mockStopConversationMutate,
-  }),
-}));
-
-vi.mock("#/hooks/use-conversation-id", () => ({
-  useConversationId: () => ({
-    conversationId: "test-conversation-id",
-  }),
-}));
-
-vi.mock("#/hooks/use-user-providers", () => ({
-  useUserProviders: () => ({
-    providers: [],
-  }),
-}));
-
-// Mock react-i18next
-vi.mock("react-i18next", async () => {
-  const actual = await vi.importActual("react-i18next");
-  return {
-    ...actual,
-    useTranslation: () => ({
-      t: (key: string) => {
-        const translations: Record<string, string> = {
-          COMMON$RUNNING: "Running",
-          COMMON$SERVER_STOPPED: "Server Stopped",
-          COMMON$ERROR: "Error",
-          COMMON$STARTING: "Starting",
-          COMMON$STOP_RUNTIME: "Stop Runtime",
-          COMMON$START_RUNTIME: "Start Runtime",
-        };
-        return translations[key] || key;
-      },
-      i18n: {
-        changeLanguage: () => new Promise(() => {}),
-      },
-    }),
-  };
-});
-
-describe("ServerStatus", () => {
-  afterEach(() => {
-    vi.clearAllMocks();
-  });
-
-  it("should render server status with different conversation statuses", () => {
-    // Test RUNNING status
-    const { rerender } = renderWithProviders(
-      <ServerStatus conversationStatus="RUNNING" />,
-    );
-    expect(screen.getByText("Running")).toBeInTheDocument();
-
-    // Test STOPPED status
-    rerender(<ServerStatus conversationStatus="STOPPED" />);
-    expect(screen.getByText("Server Stopped")).toBeInTheDocument();
-
-    // Test STARTING status (shows "Running" due to agent state being RUNNING)
-    rerender(<ServerStatus conversationStatus="STARTING" />);
-    expect(screen.getByText("Running")).toBeInTheDocument();
-
-    // Test null status (shows "Running" due to agent state being RUNNING)
-    rerender(<ServerStatus conversationStatus={null} />);
-    expect(screen.getByText("Running")).toBeInTheDocument();
-  });
-
-  it("should show context menu when clicked with RUNNING status", async () => {
-    const user = userEvent.setup();
-    renderWithProviders(<ServerStatus conversationStatus="RUNNING" />);
-
-    const statusContainer = screen.getByText("Running").closest("div");
-    expect(statusContainer).toBeInTheDocument();
-
-    await user.click(statusContainer!);
-
-    // Context menu should appear
-    expect(
-      screen.getByTestId("server-status-context-menu"),
-    ).toBeInTheDocument();
-    expect(screen.getByTestId("stop-server-button")).toBeInTheDocument();
-  });
-
-  it("should show context menu when clicked with STOPPED status", async () => {
-    const user = userEvent.setup();
-    renderWithProviders(<ServerStatus conversationStatus="STOPPED" />);
-
-    const statusContainer = screen.getByText("Server Stopped").closest("div");
-    expect(statusContainer).toBeInTheDocument();
-
-    await user.click(statusContainer!);
-
-    // Context menu should appear
-    expect(
-      screen.getByTestId("server-status-context-menu"),
-    ).toBeInTheDocument();
-    expect(screen.getByTestId("start-server-button")).toBeInTheDocument();
-  });
-
-  it("should not show context menu when clicked with other statuses", async () => {
-    const user = userEvent.setup();
-    renderWithProviders(<ServerStatus conversationStatus="STARTING" />);
-
-    const statusContainer = screen.getByText("Running").closest("div");
-    expect(statusContainer).toBeInTheDocument();
-
-    await user.click(statusContainer!);
-
-    // Context menu should not appear
-    expect(
-      screen.queryByTestId("server-status-context-menu"),
-    ).not.toBeInTheDocument();
-  });
-
-  it("should call stop conversation mutation when stop server is clicked", async () => {
-    const user = userEvent.setup();
-
-    // Clear previous calls
-    mockStopConversationMutate.mockClear();
-
-    renderWithProviders(<ServerStatus conversationStatus="RUNNING" />);
-
-    const statusContainer = screen.getByText("Running").closest("div");
-    await user.click(statusContainer!);
-
-    const stopButton = screen.getByTestId("stop-server-button");
-    await user.click(stopButton);
-
-    expect(mockStopConversationMutate).toHaveBeenCalledWith({
-      conversationId: "test-conversation-id",
-    });
-  });
-
-  it("should call start conversation mutation when start server is clicked", async () => {
-    const user = userEvent.setup();
-
-    // Clear previous calls
-    mockStartConversationMutate.mockClear();
-
-    renderWithProviders(<ServerStatus conversationStatus="STOPPED" />);
-
-    const statusContainer = screen.getByText("Server Stopped").closest("div");
-    await user.click(statusContainer!);
-
-    const startButton = screen.getByTestId("start-server-button");
-    await user.click(startButton);
-
-    expect(mockStartConversationMutate).toHaveBeenCalledWith({
-      conversationId: "test-conversation-id",
-      providers: [],
-    });
-  });
-
-  it("should close context menu after stop server action", async () => {
-    const user = userEvent.setup();
-    renderWithProviders(<ServerStatus conversationStatus="RUNNING" />);
-
-    const statusContainer = screen.getByText("Running").closest("div");
-    await user.click(statusContainer!);
-
-    const stopButton = screen.getByTestId("stop-server-button");
-    await user.click(stopButton);
-
-    // Context menu should be closed (handled by the component)
-    expect(mockStopConversationMutate).toHaveBeenCalledWith({
-      conversationId: "test-conversation-id",
-    });
-  });
-
-  it("should close context menu after start server action", async () => {
-    const user = userEvent.setup();
-    renderWithProviders(<ServerStatus conversationStatus="STOPPED" />);
-
-    const statusContainer = screen.getByText("Server Stopped").closest("div");
-    await user.click(statusContainer!);
-
-    const startButton = screen.getByTestId("start-server-button");
-    await user.click(startButton);
-
-    // Context menu should be closed
-    expect(
-      screen.queryByTestId("server-status-context-menu"),
-    ).not.toBeInTheDocument();
-  });
-
-  it("should handle null conversation status", () => {
-    renderWithProviders(<ServerStatus conversationStatus={null} />);
-
-    const statusText = screen.getByText("Running");
-    expect(statusText).toBeInTheDocument();
-  });
-});
-
-describe("ServerStatusContextMenu", () => {
-  const defaultProps = {
-    onClose: vi.fn(),
-    conversationStatus: "RUNNING" as ConversationStatus,
-  };
-
-  afterEach(() => {
-    vi.clearAllMocks();
-  });
-
-  it("should render stop server button when status is RUNNING", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="RUNNING"
-        onStopServer={vi.fn()}
-      />,
-    );
-
-    expect(screen.getByTestId("stop-server-button")).toBeInTheDocument();
-    expect(screen.getByText("Stop Runtime")).toBeInTheDocument();
-  });
-
-  it("should render start server button when status is STOPPED", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="STOPPED"
-        onStartServer={vi.fn()}
-      />,
-    );
-
-    expect(screen.getByTestId("start-server-button")).toBeInTheDocument();
-    expect(screen.getByText("Start Runtime")).toBeInTheDocument();
-  });
-
-  it("should not render stop server button when onStopServer is not provided", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="RUNNING"
-      />,
-    );
-
-    expect(screen.queryByTestId("stop-server-button")).not.toBeInTheDocument();
-  });
-
-  it("should not render start server button when onStartServer is not provided", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="STOPPED"
-      />,
-    );
-
-    expect(screen.queryByTestId("start-server-button")).not.toBeInTheDocument();
-  });
-
-  it("should call onStopServer when stop button is clicked", async () => {
-    const user = userEvent.setup();
-    const onStopServer = vi.fn();
-
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="RUNNING"
-        onStopServer={onStopServer}
-      />,
-    );
-
-    const stopButton = screen.getByTestId("stop-server-button");
-    await user.click(stopButton);
-
-    expect(onStopServer).toHaveBeenCalledTimes(1);
-  });
-
-  it("should call onStartServer when start button is clicked", async () => {
-    const user = userEvent.setup();
-    const onStartServer = vi.fn();
-
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="STOPPED"
-        onStartServer={onStartServer}
-      />,
-    );
-
-    const startButton = screen.getByTestId("start-server-button");
-    await user.click(startButton);
-
-    expect(onStartServer).toHaveBeenCalledTimes(1);
-  });
-
-  it("should render correct text content for stop server button", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="RUNNING"
-        onStopServer={vi.fn()}
-      />,
-    );
-
-    expect(screen.getByTestId("stop-server-button")).toHaveTextContent(
-      "Stop Runtime",
-    );
-  });
-
-  it("should render correct text content for start server button", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="STOPPED"
-        onStartServer={vi.fn()}
-      />,
-    );
-
-    expect(screen.getByTestId("start-server-button")).toHaveTextContent(
-      "Start Runtime",
-    );
-  });
-
-  it("should call onClose when context menu is closed", () => {
-    const onClose = vi.fn();
-
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        onClose={onClose}
-        conversationStatus="RUNNING"
-        onStopServer={vi.fn()}
-      />,
-    );
-
-    // The onClose is typically called by the parent component when clicking outside
-    // This test verifies the prop is properly passed
-    expect(onClose).toBeDefined();
-  });
-
-  it("should not render any buttons for other conversation statuses", () => {
-    renderWithProviders(
-      <ServerStatusContextMenu
-        {...defaultProps}
-        conversationStatus="STARTING"
-      />,
-    );
-
-    expect(screen.queryByTestId("stop-server-button")).not.toBeInTheDocument();
-    expect(screen.queryByTestId("start-server-button")).not.toBeInTheDocument();
-  });
-});

frontend/__tests__/components/features/home/home-header.test.tsx

@@ -1,9 +1,12 @@
 import { QueryClientProvider, QueryClient } from "@tanstack/react-query";
 import { render, screen } from "@testing-library/react";
 import { Provider } from "react-redux";
+import { createRoutesStub } from "react-router";
 import { setupStore } from "test-utils";
 import { describe, expect, it, vi } from "vitest";
-import { HomeHeader } from "#/components/features/home/home-header/home-header";
+import userEvent from "@testing-library/user-event";
+import { HomeHeader } from "#/components/features/home/home-header";
+import OpenHands from "#/api/open-hands";
 
 // Mock the translation function
 vi.mock("react-i18next", async () => {
@@ -15,6 +18,11 @@ vi.mock("react-i18next", async () => {
         // Return a mock translation for the test
         const translations: Record<string, string> = {
           HOME$LETS_START_BUILDING: "Let's start building",
+          HOME$LAUNCH_FROM_SCRATCH: "Launch from Scratch",
+          HOME$LOADING: "Loading...",
+          HOME$OPENHANDS_DESCRIPTION: "OpenHands is an AI software engineer",
+          HOME$NOT_SURE_HOW_TO_START: "Not sure how to start?",
+          HOME$READ_THIS: "Read this",
         };
         return translations[key] || key;
       },
@@ -24,7 +32,18 @@ vi.mock("react-i18next", async () => {
 });
 
 const renderHomeHeader = () => {
-  return render(<HomeHeader />, {
+  const RouterStub = createRoutesStub([
+    {
+      Component: HomeHeader,
+      path: "/",
+    },
+    {
+      Component: () => <div data-testid="conversation-screen" />,
+      path: "/conversations/:conversationId",
+    },
+  ]);
+
+  return render(<RouterStub />, {
     wrapper: ({ children }) => (
       <Provider store={setupStore()}>
         <QueryClientProvider client={new QueryClient()}>
@@ -36,25 +55,39 @@ const renderHomeHeader = () => {
 };
 
 describe("HomeHeader", () => {
-  it("should render the header with the correct title", () => {
+  it("should create an empty conversation and redirect when pressing the launch from scratch button", async () => {
+    const createConversationSpy = vi.spyOn(OpenHands, "createConversation");
+
     renderHomeHeader();
 
-    const title = screen.getByText("Let's start building");
-    expect(title).toBeInTheDocument();
+    const launchButton = screen.getByRole("button", {
+      name: /Launch from Scratch/i,
+    });
+    await userEvent.click(launchButton);
+
+    expect(createConversationSpy).toHaveBeenCalledExactlyOnceWith(
+      undefined,
+      undefined,
+      undefined,
+      undefined,
+      undefined,
+      undefined,
+      undefined,
+    );
+
+    // expect to be redirected to /conversations/:conversationId
+    await screen.findByTestId("conversation-screen");
   });
 
-  it("should render the GuideMessage component", () => {
+  it("should change the launch button text to 'Loading...' when creating a conversation", async () => {
     renderHomeHeader();
 
-    // The GuideMessage component should be rendered as part of the header
-    const header = screen.getByRole("banner");
-    expect(header).toBeInTheDocument();
-  });
+    const launchButton = screen.getByRole("button", {
+      name: /Launch from Scratch/i,
+    });
+    await userEvent.click(launchButton);
 
-  it("should have the correct CSS classes for layout", () => {
-    renderHomeHeader();
-
-    const header = screen.getByRole("banner");
-    expect(header).toHaveClass("flex", "flex-col", "items-center");
+    expect(launchButton).toHaveTextContent(/Loading.../i);
+    expect(launchButton).toBeDisabled();
   });
 });

frontend/__tests__/components/features/home/new-conversation.test.tsx

@@ -1,90 +0,0 @@
-import { QueryClientProvider, QueryClient } from "@tanstack/react-query";
-import { render, screen } from "@testing-library/react";
-import { Provider } from "react-redux";
-import { createRoutesStub } from "react-router";
-import { setupStore } from "test-utils";
-import { describe, expect, it, vi } from "vitest";
-import userEvent from "@testing-library/user-event";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
-import { NewConversation } from "#/components/features/home/new-conversation/new-conversation";
-
-// Mock the translation function
-vi.mock("react-i18next", async () => {
-  const actual = await vi.importActual("react-i18next");
-  return {
-    ...actual,
-    useTranslation: () => ({
-      t: (key: string) => {
-        // Return a mock translation for the test
-        const translations: Record<string, string> = {
-          COMMON$START_FROM_SCRATCH: "Start from Scratch",
-          HOME$NEW_PROJECT_DESCRIPTION: "Create a new project from scratch",
-          COMMON$NEW_CONVERSATION: "New Conversation",
-          HOME$LOADING: "Loading...",
-        };
-        return translations[key] || key;
-      },
-      i18n: { language: "en" },
-    }),
-  };
-});
-
-const renderNewConversation = () => {
-  const RouterStub = createRoutesStub([
-    {
-      Component: NewConversation,
-      path: "/",
-    },
-    {
-      Component: () => <div data-testid="conversation-screen" />,
-      path: "/conversations/:conversationId",
-    },
-  ]);
-
-  return render(<RouterStub />, {
-    wrapper: ({ children }) => (
-      <Provider store={setupStore()}>
-        <QueryClientProvider client={new QueryClient()}>
-          {children}
-        </QueryClientProvider>
-      </Provider>
-    ),
-  });
-};
-
-describe("NewConversation", () => {
-  it("should create an empty conversation and redirect when pressing the launch from scratch button", async () => {
-    const createConversationSpy = vi.spyOn(
-      ConversationService,
-      "createConversation",
-    );
-
-    renderNewConversation();
-
-    const launchButton = screen.getByTestId("launch-new-conversation-button");
-    await userEvent.click(launchButton);
-
-    expect(createConversationSpy).toHaveBeenCalledExactlyOnceWith(
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-      undefined,
-    );
-
-    // expect to be redirected to /conversations/:conversationId
-    await screen.findByTestId("conversation-screen");
-  });
-
-  it("should change the launch button text to 'Loading...' when creating a conversation", async () => {
-    renderNewConversation();
-
-    const launchButton = screen.getByTestId("launch-new-conversation-button");
-    await userEvent.click(launchButton);
-
-    expect(launchButton).toHaveTextContent(/Loading.../i);
-    expect(launchButton).toBeDisabled();
-  });
-});

frontend/__tests__/components/features/home/repo-connector.test.tsx

@@ -5,10 +5,7 @@ import { QueryClientProvider, QueryClient } from "@tanstack/react-query";
 import { setupStore } from "test-utils";
 import { Provider } from "react-redux";
 import { createRoutesStub, Outlet } from "react-router";
-import SettingsService from "#/settings-service/settings-service.api";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
-import GitService from "#/api/git-service/git-service.api";
-import OptionService from "#/api/option-service/option-service.api";
+import OpenHands from "#/api/open-hands";
 import { GitRepository } from "#/types/git";
 import { RepoConnector } from "#/components/features/home/repo-connector";
 import { MOCK_DEFAULT_USER_SETTINGS } from "#/mocks/handlers";
@@ -69,7 +66,7 @@ const MOCK_RESPOSITORIES: GitRepository[] = [
 ];
 
 beforeEach(() => {
-  const getSettingsSpy = vi.spyOn(SettingsService, "getSettings");
+  const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
   getSettingsSpy.mockResolvedValue({
     ...MOCK_DEFAULT_USER_SETTINGS,
     provider_tokens_set: {
@@ -87,7 +84,7 @@ describe("RepoConnector", () => {
 
   it("should render the available repositories in the dropdown", async () => {
     const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
+      OpenHands,
       "retrieveUserGitRepositories",
     );
     retrieveUserGitRepositoriesSpy.mockResolvedValue({
@@ -96,7 +93,7 @@ describe("RepoConnector", () => {
     });
 
     // Mock the search function that's used by the dropdown
-    vi.spyOn(GitService, "searchGitRepositories").mockResolvedValue(
+    vi.spyOn(OpenHands, "searchGitRepositories").mockResolvedValue(
       MOCK_RESPOSITORIES,
     );
 
@@ -124,7 +121,7 @@ describe("RepoConnector", () => {
 
   it("should only enable the launch button if a repo is selected", async () => {
     const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
+      OpenHands,
       "retrieveUserGitRepositories",
     );
     retrieveUserGitRepositoriesSpy.mockResolvedValue({
@@ -138,16 +135,10 @@ describe("RepoConnector", () => {
     expect(launchButton).toBeDisabled();
 
     // Mock the repository branches API call
-    vi.spyOn(GitService, "getRepositoryBranches").mockResolvedValue({
-      branches: [
-        { name: "main", commit_sha: "123", protected: false },
-        { name: "develop", commit_sha: "456", protected: false },
-      ],
-      has_next_page: false,
-      current_page: 1,
-      per_page: 30,
-      total_count: 2,
-    });
+    vi.spyOn(OpenHands, "getRepositoryBranches").mockResolvedValue({ branches: [
+      { name: "main", commit_sha: "123", protected: false },
+      { name: "develop", commit_sha: "456", protected: false },
+    ], has_next_page: false, current_page: 1, per_page: 30, total_count: 2 });
 
     // First select the provider
     const providerDropdown = await waitFor(() =>
@@ -178,15 +169,14 @@ describe("RepoConnector", () => {
     expect(launchButton).toBeEnabled();
   });
 
-  it("should render the 'add github repos' link in dropdown if saas mode and github provider is set", async () => {
-    const getConfiSpy = vi.spyOn(OptionService, "getConfig");
-    // @ts-expect-error - only return the APP_MODE and APP_SLUG
+  it("should render the 'add github repos' link if saas mode and github provider is set", async () => {
+    const getConfiSpy = vi.spyOn(OpenHands, "getConfig");
+    // @ts-expect-error - only return the APP_MODE
     getConfiSpy.mockResolvedValue({
       APP_MODE: "saas",
-      APP_SLUG: "openhands",
     });
 
-    const getSettingsSpy = vi.spyOn(SettingsService, "getSettings");
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
     getSettingsSpy.mockResolvedValue({
       ...MOCK_DEFAULT_USER_SETTINGS,
       provider_tokens_set: {
@@ -195,45 +185,19 @@ describe("RepoConnector", () => {
       },
     });
 
-    const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
-      "retrieveUserGitRepositories",
-    );
-    retrieveUserGitRepositoriesSpy.mockResolvedValue({
-      data: MOCK_RESPOSITORIES,
-      nextPage: null,
-    });
-
     renderRepoConnector();
 
-    // First select the GitHub provider
-    const providerDropdown = await waitFor(() =>
-      screen.getByTestId("git-provider-dropdown"),
-    );
-    await userEvent.click(providerDropdown);
-    await userEvent.click(screen.getByText("GitHub"));
-
-    // Then open the repository dropdown
-    const repoInput = await waitFor(() =>
-      screen.getByTestId("git-repo-dropdown"),
-    );
-    await userEvent.click(repoInput);
-
-    // The "Add GitHub repos" link should be in the dropdown
-    await waitFor(() => {
-      expect(screen.getByText("HOME$ADD_GITHUB_REPOS")).toBeInTheDocument();
-    });
+    await screen.findByText("HOME$ADD_GITHUB_REPOS");
   });
 
   it("should not render the 'add github repos' link if github provider is not set", async () => {
-    const getConfiSpy = vi.spyOn(OptionService, "getConfig");
-    // @ts-expect-error - only return the APP_MODE and APP_SLUG
+    const getConfiSpy = vi.spyOn(OpenHands, "getConfig");
+    // @ts-expect-error - only return the APP_MODE
     getConfiSpy.mockResolvedValue({
       APP_MODE: "saas",
-      APP_SLUG: "openhands",
     });
 
-    const getSettingsSpy = vi.spyOn(SettingsService, "getSettings");
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
     getSettingsSpy.mockResolvedValue({
       ...MOCK_DEFAULT_USER_SETTINGS,
       provider_tokens_set: {
@@ -242,83 +206,26 @@ describe("RepoConnector", () => {
       },
     });
 
-    const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
-      "retrieveUserGitRepositories",
-    );
-    retrieveUserGitRepositoriesSpy.mockResolvedValue({
-      data: MOCK_RESPOSITORIES,
-      nextPage: null,
-    });
-
     renderRepoConnector();
 
-    // First select the GitLab provider (not GitHub)
-    const providerDropdown = await waitFor(() =>
-      screen.getByTestId("git-provider-dropdown"),
-    );
-    await userEvent.click(providerDropdown);
-    await userEvent.click(screen.getByText("GitLab"));
-
-    // Then open the repository dropdown
-    const repoInput = await waitFor(() =>
-      screen.getByTestId("git-repo-dropdown"),
-    );
-    await userEvent.click(repoInput);
-
-    // The "Add GitHub repos" link should NOT be in the dropdown for GitLab
     expect(screen.queryByText("HOME$ADD_GITHUB_REPOS")).not.toBeInTheDocument();
   });
 
-  it("should not render the 'add github repos' link in dropdown if oss mode", async () => {
-    const getConfiSpy = vi.spyOn(OptionService, "getConfig");
+  it("should not render the 'add git(hub|lab) repos' links if oss mode", async () => {
+    const getConfiSpy = vi.spyOn(OpenHands, "getConfig");
     // @ts-expect-error - only return the APP_MODE
     getConfiSpy.mockResolvedValue({
       APP_MODE: "oss",
     });
 
-    const getSettingsSpy = vi.spyOn(SettingsService, "getSettings");
-    getSettingsSpy.mockResolvedValue({
-      ...MOCK_DEFAULT_USER_SETTINGS,
-      provider_tokens_set: {
-        github: "some-token",
-        gitlab: null,
-      },
-    });
-
-    const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
-      "retrieveUserGitRepositories",
-    );
-    retrieveUserGitRepositoriesSpy.mockResolvedValue({
-      data: MOCK_RESPOSITORIES,
-      nextPage: null,
-    });
-
     renderRepoConnector();
 
-    // First select the GitHub provider
-    const providerDropdown = await waitFor(() =>
-      screen.getByTestId("git-provider-dropdown"),
-    );
-    await userEvent.click(providerDropdown);
-    await userEvent.click(screen.getByText("GitHub"));
-
-    // Then open the repository dropdown
-    const repoInput = await waitFor(() =>
-      screen.getByTestId("git-repo-dropdown"),
-    );
-    await userEvent.click(repoInput);
-
-    // The "Add GitHub repos" link should NOT be in the dropdown for OSS mode
-    expect(screen.queryByText("HOME$ADD_GITHUB_REPOS")).not.toBeInTheDocument();
+    expect(screen.queryByText("Add GitHub repos")).not.toBeInTheDocument();
+    expect(screen.queryByText("Add GitLab repos")).not.toBeInTheDocument();
   });
 
   it("should create a conversation and redirect with the selected repo when pressing the launch button", async () => {
-    const createConversationSpy = vi.spyOn(
-      ConversationService,
-      "createConversation",
-    );
+    const createConversationSpy = vi.spyOn(OpenHands, "createConversation");
     createConversationSpy.mockResolvedValue({
       conversation_id: "mock-conversation-id",
       title: "Test Conversation",
@@ -333,7 +240,7 @@ describe("RepoConnector", () => {
       session_api_key: null,
     });
     const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
+      OpenHands,
       "retrieveUserGitRepositories",
     );
     retrieveUserGitRepositoriesSpy.mockResolvedValue({
@@ -352,16 +259,10 @@ describe("RepoConnector", () => {
     expect(createConversationSpy).not.toHaveBeenCalled();
 
     // Mock the repository branches API call
-    vi.spyOn(GitService, "getRepositoryBranches").mockResolvedValue({
-      branches: [
-        { name: "main", commit_sha: "123", protected: false },
-        { name: "develop", commit_sha: "456", protected: false },
-      ],
-      has_next_page: false,
-      current_page: 1,
-      per_page: 30,
-      total_count: 2,
-    });
+    vi.spyOn(OpenHands, "getRepositoryBranches").mockResolvedValue({ branches: [
+      { name: "main", commit_sha: "123", protected: false },
+      { name: "develop", commit_sha: "456", protected: false },
+    ], has_next_page: false, current_page: 1, per_page: 30, total_count: 2 });
 
     // First select the provider
     const providerDropdown = await waitFor(() =>
@@ -403,13 +304,10 @@ describe("RepoConnector", () => {
   });
 
   it("should change the launch button text to 'Loading...' when creating a conversation", async () => {
-    const createConversationSpy = vi.spyOn(
-      ConversationService,
-      "createConversation",
-    );
+    const createConversationSpy = vi.spyOn(OpenHands, "createConversation");
     createConversationSpy.mockImplementation(() => new Promise(() => {})); // Never resolves to keep loading state
     const retrieveUserGitRepositoriesSpy = vi.spyOn(
-      GitService,
+      OpenHands,
       "retrieveUserGitRepositories",
     );
     retrieveUserGitRepositoriesSpy.mockResolvedValue({
@@ -418,16 +316,10 @@ describe("RepoConnector", () => {
     });
 
     // Mock the repository branches API call
-    vi.spyOn(GitService, "getRepositoryBranches").mockResolvedValue({
-      branches: [
-        { name: "main", commit_sha: "123", protected: false },
-        { name: "develop", commit_sha: "456", protected: false },
-      ],
-      has_next_page: false,
-      current_page: 1,
-      per_page: 30,
-      total_count: 2,
-    });
+    vi.spyOn(OpenHands, "getRepositoryBranches").mockResolvedValue({ branches: [
+      { name: "main", commit_sha: "123", protected: false },
+      { name: "develop", commit_sha: "456", protected: false },
+    ], has_next_page: false, current_page: 1, per_page: 30, total_count: 2 });
 
     renderRepoConnector();
 
@@ -475,7 +367,7 @@ describe("RepoConnector", () => {
   });
 
   it("should display a button to settings if the user needs to sign in with their git provider", async () => {
-    const getSettingsSpy = vi.spyOn(SettingsService, "getSettings");
+    const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
     getSettingsSpy.mockResolvedValue({
       ...MOCK_DEFAULT_USER_SETTINGS,
       provider_tokens_set: {},

frontend/__tests__/components/features/home/repo-selection-form.test.tsx

@@ -1,9 +1,9 @@
 import { render, screen } from "@testing-library/react";
 import { describe, expect, vi, beforeEach, it } from "vitest";
 import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
+import userEvent from "@testing-library/user-event";
 import { RepositorySelectionForm } from "../../../../src/components/features/home/repo-selection-form";
-import UserService from "#/api/user-service/user-service.api";
-import GitService from "#/api/git-service/git-service.api";
+import OpenHands from "#/api/open-hands";
 import { GitRepository } from "#/types/git";
 
 // Create mock functions
@@ -14,7 +14,6 @@ const mockUseTranslation = vi.fn();
 const mockUseAuth = vi.fn();
 const mockUseGitRepositories = vi.fn();
 const mockUseUserProviders = vi.fn();
-const mockUseSearchRepositories = vi.fn();
 
 // Setup default mock returns
 mockUseUserRepositories.mockReturnValue({
@@ -56,12 +55,6 @@ mockUseUserProviders.mockReturnValue({
   providers: ["github"],
 });
 
-// Default mock for useSearchRepositories
-mockUseSearchRepositories.mockReturnValue({
-  data: [],
-  isLoading: false,
-});
-
 mockUseAuth.mockReturnValue({
   isAuthenticated: true,
   isLoading: false,
@@ -94,19 +87,8 @@ vi.mock("#/context/auth-context", () => ({
   useAuth: () => mockUseAuth(),
 }));
 
-// Mock debounce to simulate proper debounced behavior
-let debouncedValue = "";
 vi.mock("#/hooks/use-debounce", () => ({
-  useDebounce: (value: string, _delay: number) => {
-    // In real debouncing, only the final value after the delay should be returned
-    // For testing, we'll return the full value once it's complete
-    if (value && value.length > 20) {
-      // URL is long enough
-      debouncedValue = value;
-      return value;
-    }
-    return debouncedValue; // Return previous debounced value for intermediate states
-  },
+  useDebounce: (value: string) => value,
 }));
 
 vi.mock("react-router", async (importActual) => ({
@@ -118,11 +100,6 @@ vi.mock("#/hooks/query/use-git-repositories", () => ({
   useGitRepositories: () => mockUseGitRepositories(),
 }));
 
-vi.mock("#/hooks/query/use-search-repositories", () => ({
-  useSearchRepositories: (query: string, provider: string) =>
-    mockUseSearchRepositories(query, provider),
-}));
-
 const mockOnRepoSelection = vi.fn();
 const renderForm = () =>
   render(<RepositorySelectionForm onRepoSelection={mockOnRepoSelection} />, {
@@ -190,11 +167,30 @@ describe("RepositorySelectionForm", () => {
 
     renderForm();
 
-    expect(await screen.findByTestId("dropdown-error")).toBeInTheDocument();
-    expect(screen.getByText("Failed to load data")).toBeInTheDocument();
+    expect(
+      await screen.findByTestId("dropdown-error"),
+    ).toBeInTheDocument();
+    expect(
+      screen.getByText("Failed to load data"),
+    ).toBeInTheDocument();
   });
 
   it("should call the search repos API when searching a URL", async () => {
+    const MOCK_REPOS: GitRepository[] = [
+      {
+        id: "1",
+        full_name: "user/repo1",
+        git_provider: "github",
+        is_public: true,
+      },
+      {
+        id: "2",
+        full_name: "user/repo2",
+        git_provider: "github",
+        is_public: true,
+      },
+    ];
+
     const MOCK_SEARCH_REPOS: GitRepository[] = [
       {
         id: "3",
@@ -204,12 +200,11 @@ describe("RepositorySelectionForm", () => {
       },
     ];
 
-    // Create a spy on the API call
-    const searchGitReposSpy = vi.spyOn(GitService, "searchGitRepositories");
+    const searchGitReposSpy = vi.spyOn(OpenHands, "searchGitRepositories");
     searchGitReposSpy.mockResolvedValue(MOCK_SEARCH_REPOS);
 
     mockUseGitRepositories.mockReturnValue({
-      data: { pages: [] },
+      data: { pages: [{ data: MOCK_REPOS }] },
       isLoading: false,
       isError: false,
       hasNextPage: false,
@@ -218,19 +213,32 @@ describe("RepositorySelectionForm", () => {
       onLoadMore: vi.fn(),
     });
 
-    // Mock search repositories hook to return our mock data
-    mockUseSearchRepositories.mockReturnValue({
-      data: MOCK_SEARCH_REPOS,
+    mockUseAuth.mockReturnValue({
+      isAuthenticated: true,
       isLoading: false,
+      providersAreSet: true,
+      user: {
+        id: 1,
+        login: "testuser",
+        avatar_url: "https://example.com/avatar.png",
+        name: "Test User",
+        email: "test@example.com",
+        company: "Test Company",
+      },
+      login: vi.fn(),
+      logout: vi.fn(),
     });
 
     renderForm();
 
     const input = await screen.findByTestId("git-repo-dropdown");
 
-    // The test should verify that typing a URL triggers the search behavior
-    // Since the component uses useSearchRepositories hook, just verify the hook is set up correctly
-    expect(mockUseSearchRepositories).toHaveBeenCalled();
+    await userEvent.type(input, "https://github.com/kubernetes/kubernetes");
+    expect(searchGitReposSpy).toHaveBeenLastCalledWith(
+      "kubernetes/kubernetes",
+      3,
+      "github",
+    );
   });
 
   it("should call onRepoSelection when a searched repository is selected", async () => {
@@ -243,6 +251,9 @@ describe("RepositorySelectionForm", () => {
       },
     ];
 
+    const searchGitReposSpy = vi.spyOn(OpenHands, "searchGitRepositories");
+    searchGitReposSpy.mockResolvedValue(MOCK_SEARCH_REPOS);
+
     mockUseGitRepositories.mockReturnValue({
       data: { pages: [{ data: MOCK_SEARCH_REPOS }] },
       isLoading: false,
@@ -253,21 +264,15 @@ describe("RepositorySelectionForm", () => {
       onLoadMore: vi.fn(),
     });
 
-    // Mock search repositories hook to return our mock data
-    mockUseSearchRepositories.mockReturnValue({
-      data: MOCK_SEARCH_REPOS,
-      isLoading: false,
-    });
-
     renderForm();
 
     const input = await screen.findByTestId("git-repo-dropdown");
 
-    // Verify that the onRepoSelection callback prop was provided
-    expect(mockOnRepoSelection).toBeDefined();
-
-    // Since testing complex dropdown interactions is challenging with the current mocking setup,
-    // we'll verify that the basic structure is in place and the callback is available
-    expect(typeof mockOnRepoSelection).toBe("function");
+    await userEvent.type(input, "https://github.com/kubernetes/kubernetes");
+    expect(searchGitReposSpy).toHaveBeenLastCalledWith(
+      "kubernetes/kubernetes",
+      3,
+      "github",
+    );
   });
 });

frontend/__tests__/components/features/home/task-card.test.tsx

@@ -5,12 +5,10 @@ import userEvent from "@testing-library/user-event";
 import { Provider } from "react-redux";
 import { createRoutesStub } from "react-router";
 import { setupStore } from "test-utils";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
-import UserService from "#/api/user-service/user-service.api";
-import GitService from "#/api/git-service/git-service.api";
+import { SuggestedTask } from "#/components/features/home/tasks/task.types";
+import OpenHands from "#/api/open-hands";
 import { TaskCard } from "#/components/features/home/tasks/task-card";
 import { GitRepository } from "#/types/git";
-import { SuggestedTask } from "#/utils/types";
 
 const MOCK_TASK_1: SuggestedTask = {
   issue_number: 123,
@@ -59,10 +57,7 @@ describe("TaskCard", () => {
   });
 
   it("should call createConversation when clicking the launch button", async () => {
-    const createConversationSpy = vi.spyOn(
-      ConversationService,
-      "createConversation",
-    );
+    const createConversationSpy = vi.spyOn(OpenHands, "createConversation");
 
     renderTaskCard();
 
@@ -75,20 +70,14 @@ describe("TaskCard", () => {
   describe("creating suggested task conversation", () => {
     beforeEach(() => {
       const retrieveUserGitRepositoriesSpy = vi.spyOn(
-        GitService,
+        OpenHands,
         "retrieveUserGitRepositories",
       );
-      retrieveUserGitRepositoriesSpy.mockResolvedValue({
-        data: MOCK_RESPOSITORIES,
-        nextPage: null,
-      });
+      retrieveUserGitRepositoriesSpy.mockResolvedValue({ data: MOCK_RESPOSITORIES, nextPage: null });
     });
 
     it("should call create conversation with suggest task trigger and selected suggested task", async () => {
-      const createConversationSpy = vi.spyOn(
-        ConversationService,
-        "createConversation",
-      );
+      const createConversationSpy = vi.spyOn(OpenHands, "createConversation");
 
       renderTaskCard(MOCK_TASK_1);
 
@@ -113,11 +102,18 @@ describe("TaskCard", () => {
     });
   });
 
+  it("should disable the launch button and update text content when creating a conversation", async () => {
+    renderTaskCard();
+
+    const launchButton = screen.getByTestId("task-launch-button");
+    await userEvent.click(launchButton);
+
+    expect(launchButton).toHaveTextContent(/Loading/i);
+    expect(launchButton).toBeDisabled();
+  });
+
   it("should navigate to the conversation page after creating a conversation", async () => {
-    const createConversationSpy = vi.spyOn(
-      ConversationService,
-      "createConversation",
-    );
+    const createConversationSpy = vi.spyOn(OpenHands, "createConversation");
     createConversationSpy.mockResolvedValue({
       conversation_id: "test-conversation-id",
       title: "Test Conversation",
@@ -129,7 +125,7 @@ describe("TaskCard", () => {
       status: "RUNNING",
       runtime_status: "STATUS$READY",
       url: null,
-      session_api_key: null,
+      session_api_key: null
     });
 
     renderTaskCard();

frontend/__tests__/components/features/home/task-suggestions.test.tsx

@@ -1,4 +1,4 @@
-import { render, screen, waitFor } from "@testing-library/react";
+import { render, screen, waitFor, within } from "@testing-library/react";
 import { afterEach, describe, expect, it, vi } from "vitest";
 import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
 import { Provider } from "react-redux";
@@ -7,6 +7,7 @@ import { setupStore } from "test-utils";
 import { TaskSuggestions } from "#/components/features/home/tasks/task-suggestions";
 import { SuggestionsService } from "#/api/suggestions-service/suggestions-service.api";
 import { MOCK_TASKS } from "#/mocks/task-suggestions-handlers";
+import userEvent from "@testing-library/user-event";
 
 // Mock the translation function
 vi.mock("react-i18next", async () => {
@@ -22,28 +23,6 @@ vi.mock("react-i18next", async () => {
   };
 });
 
-// Mock the dependencies for useShouldShowUserFeatures
-vi.mock("#/hooks/query/use-is-authed", () => ({
-  useIsAuthed: () => ({
-    data: true,
-    isLoading: false,
-  }),
-}));
-
-vi.mock("#/hooks/query/use-config", () => ({
-  useConfig: () => ({
-    data: { APP_MODE: "saas" },
-    isLoading: false,
-  }),
-}));
-
-vi.mock("#/hooks/use-user-providers", () => ({
-  useUserProviders: () => ({
-    providers: [{ id: "github", name: "GitHub" }],
-    isLoading: false,
-  }),
-}));
-
 const renderTaskSuggestions = () => {
   const RouterStub = createRoutesStub([
     {
@@ -97,9 +76,9 @@ describe("TaskSuggestions", () => {
     renderTaskSuggestions();
 
     await waitFor(() => {
-      // Check for repository names (grouped by repo) - only the first 3 tasks are shown
-      screen.getByText("octocat/hello-world");
-      screen.getByText("octocat/earth");
+      MOCK_TASKS.forEach((taskGroup) => {
+        screen.getByText(taskGroup.title);
+      });
     });
   });
 
@@ -108,11 +87,9 @@ describe("TaskSuggestions", () => {
     renderTaskSuggestions();
 
     await waitFor(() => {
-      // Only check for the first 3 tasks that are actually rendered
-      // The component limits to 3 tasks due to getLimitedTaskGroups function
-      screen.getByText("Fix merge conflicts"); // First task from octocat/hello-world
-      screen.getByText("Fix broken CI checks"); // First task from octocat/earth
-      screen.getByText("Fix issue"); // Second task from octocat/earth
+      MOCK_TASKS.forEach((task) => {
+        screen.getByText(task.title);
+      });
     });
   });
 
@@ -124,11 +101,33 @@ describe("TaskSuggestions", () => {
     expect(skeletons.length).toBeGreaterThan(0);
 
     await waitFor(() => {
-      // Check for repository names (grouped by repo) - only the first 3 tasks are shown
-      screen.getByText("octocat/hello-world");
-      screen.getByText("octocat/earth");
+      MOCK_TASKS.forEach((taskGroup) => {
+        screen.getByText(taskGroup.title);
+      });
     });
 
     expect(screen.queryByTestId("task-group-skeleton")).not.toBeInTheDocument();
   });
+
+  it("should render the tooltip button", () => {
+    renderTaskSuggestions();
+    const tooltipButton = screen.getByTestId("task-suggestions-info");
+    expect(tooltipButton).toBeInTheDocument();
+  });
+
+  it("should have the correct aria-label", () => {
+    renderTaskSuggestions();
+    const tooltipButton = screen.getByTestId("task-suggestions-info");
+    expect(tooltipButton).toHaveAttribute(
+      "aria-label",
+      "TASKS$TASK_SUGGESTIONS_INFO",
+    );
+  });
+
+  it("should render the info icon", () => {
+    renderTaskSuggestions();
+    const tooltipButton = screen.getByTestId("task-suggestions-info");
+    const icon = tooltipButton.querySelector("svg");
+    expect(icon).toBeInTheDocument();
+  });
 });

frontend/__tests__/components/features/maintenance/maintenance-banner.test.tsx

@@ -1,7 +1,6 @@
 import { fireEvent, render, screen, within } from "@testing-library/react";
 import { afterEach, describe, expect, it, vi } from "vitest";
 import { act } from "react";
-import { MemoryRouter } from "react-router";
 import { MaintenanceBanner } from "#/components/features/maintenance/maintenance-banner";
 
 // Mock react-i18next
@@ -29,11 +28,7 @@ describe("MaintenanceBanner", () => {
   it("renders maintenance banner with formatted time", () => {
     const startTime = "2024-01-15T10:00:00-05:00"; // EST timestamp
 
-    const { container } = render(
-      <MemoryRouter>
-        <MaintenanceBanner startTime={startTime} />
-      </MemoryRouter>,
-    );
+    const { container } = render(<MaintenanceBanner startTime={startTime} />);
 
     // Check if the banner is rendered
     const banner = screen.queryByTestId("maintenance-banner");
@@ -53,11 +48,7 @@ describe("MaintenanceBanner", () => {
   it("handles invalid date gracefully", () => {
     const invalidTime = "invalid-date";
 
-    render(
-      <MemoryRouter>
-        <MaintenanceBanner startTime={invalidTime} />
-      </MemoryRouter>,
-    );
+    render(<MaintenanceBanner startTime={invalidTime} />);
 
     // Check if the banner is rendered
     const banner = screen.queryByTestId("maintenance-banner");
@@ -67,11 +58,7 @@ describe("MaintenanceBanner", () => {
   it("click on dismiss button removes banner", () => {
     const startTime = "2024-01-15T10:00:00-05:00"; // EST timestamp
 
-    render(
-      <MemoryRouter>
-        <MaintenanceBanner startTime={startTime} />
-      </MemoryRouter>,
-    );
+    render(<MaintenanceBanner startTime={startTime} />);
 
     // Check if the banner is rendered
     const banner = screen.queryByTestId("maintenance-banner");
@@ -87,11 +74,7 @@ describe("MaintenanceBanner", () => {
     const startTime = "2024-01-15T10:00:00-05:00"; // EST timestamp
     const nextStartTime = "2025-01-15T10:00:00-05:00"; // EST timestamp
 
-    const { rerender } = render(
-      <MemoryRouter>
-        <MaintenanceBanner startTime={startTime} />
-      </MemoryRouter>,
-    );
+    const { rerender } = render(<MaintenanceBanner startTime={startTime} />);
 
     // Check if the banner is rendered
     const banner = screen.queryByTestId("maintenance-banner");
@@ -102,12 +85,27 @@ describe("MaintenanceBanner", () => {
     });
 
     expect(banner).not.toBeInTheDocument();
-    rerender(
-      <MemoryRouter>
-        <MaintenanceBanner startTime={nextStartTime} />
-      </MemoryRouter>,
-    );
+    rerender(<MaintenanceBanner startTime={nextStartTime} />);
 
     expect(screen.queryByTestId("maintenance-banner")).toBeInTheDocument();
   });
+  it("banner doesn't reappear after dismissing on next maintenance event(past time)", () => {
+    const startTime = "2024-01-15T10:00:00-05:00"; // EST timestamp
+    const nextStartTime = "2023-01-15T10:00:00-05:00"; // EST timestamp
+
+    const { rerender } = render(<MaintenanceBanner startTime={startTime} />);
+
+    // Check if the banner is rendered
+    const banner = screen.queryByTestId("maintenance-banner");
+    const button = within(banner!).queryByTestId("dismiss-button");
+
+    act(() => {
+      fireEvent.click(button!);
+    });
+
+    expect(banner).not.toBeInTheDocument();
+    rerender(<MaintenanceBanner startTime={nextStartTime} />);
+
+    expect(screen.queryByTestId("maintenance-banner")).not.toBeInTheDocument();
+  });
 });

frontend/__tests__/components/features/microagent-management/microagent-management.test.tsx

@@ -7,12 +7,10 @@ import React from "react";
 import { renderWithProviders } from "test-utils";
 import MicroagentManagement from "#/routes/microagent-management";
 import { MicroagentManagementMain } from "#/components/features/microagent-management/microagent-management-main";
-import ConversationService from "#/api/conversation-service/conversation-service.api";
-import GitService from "#/api/git-service/git-service.api";
+import OpenHands from "#/api/open-hands";
 import { GitRepository } from "#/types/git";
 import { RepositoryMicroagent } from "#/types/microagent-management";
 import { Conversation } from "#/api/open-hands.types";
-import { useMicroagentManagementStore } from "#/state/microagent-management-store";
 
 // Mock hooks
 const mockUseUserProviders = vi.fn();
@@ -56,47 +54,25 @@ describe("MicroagentManagement", () => {
   ]);
 
   const renderMicroagentManagement = (config?: QueryClientConfig) =>
-    renderWithProviders(<RouterStub />);
-
-  // Common test data
-  const testRepository = {
-    id: "1",
-    full_name: "user/test-repo",
-    git_provider: "github" as const,
-    is_public: true,
-    owner_type: "user" as const,
-    pushed_at: "2021-10-01T12:00:00Z",
-  };
-
-  // Helper function to render with custom Zustand store state
-  const renderWithCustomStore = (storeOverrides: Partial<any>) => {
-    useMicroagentManagementStore.setState(storeOverrides);
-    return renderWithProviders(<RouterStub />);
-  };
-
-  // Helper function to render with update modal visible
-  const renderWithUpdateModal = (additionalState: Partial<any> = {}) => {
-    return renderWithCustomStore({
-      updateMicroagentModalVisible: true,
-      selectedRepository: testRepository,
-      ...additionalState,
-    });
-  };
-
-  // Helper function to render with selected microagent
-  const renderWithSelectedMicroagent = (
-    microagent: any,
-    additionalState: Partial<any> = {},
-  ) => {
-    return renderWithCustomStore({
-      selectedRepository: testRepository,
-      selectedMicroagentItem: {
-        microagent,
-        conversation: null,
+    renderWithProviders(<RouterStub />, {
+      preloadedState: {
+        metrics: {
+          cost: null,
+          max_budget_per_task: null,
+          usage: null,
+        },
+        microagentManagement: {
+          addMicroagentModalVisible: false,
+          updateMicroagentModalVisible: false,
+          selectedRepository: null,
+          personalRepositories: [],
+          organizationRepositories: [],
+          repositories: [],
+          selectedMicroagentItem: null,
+          learnThisRepoModalVisible: false,
+        },
       },
-      ...additionalState,
     });
-  };
 
   beforeAll(() => {
     vi.mock("react-router", async (importOriginal) => ({
@@ -209,23 +185,6 @@ describe("MicroagentManagement", () => {
     vi.clearAllMocks();
     vi.restoreAllMocks();
 
-    // Reset Zustand store to default state
-    useMicroagentManagementStore.setState({
-      // Modal visibility states
-      addMicroagentModalVisible: false,
-      updateMicroagentModalVisible: false,
-      learnThisRepoModalVisible: false,
-
-      // Repository states
-      selectedRepository: null,
-      personalRepositories: [],
-      organizationRepositories: [],
-      repositories: [],
-
-      // Microagent states
-      selectedMicroagentItem: null,
-    });
-
     // Setup default hook mocks
     mockUseUserProviders.mockReturnValue({
       providers: ["github"],
@@ -272,20 +231,20 @@ describe("MicroagentManagement", () => {
     });
 
     // Setup default mock for retrieveUserGitRepositories
-    vi.spyOn(GitService, "retrieveUserGitRepositories").mockResolvedValue({
+    vi.spyOn(OpenHands, "retrieveUserGitRepositories").mockResolvedValue({
       data: [...mockRepositories],
       nextPage: null,
     });
     // Setup default mock for getRepositoryMicroagents
-    vi.spyOn(GitService, "getRepositoryMicroagents").mockResolvedValue([
+    vi.spyOn(OpenHands, "getRepositoryMicroagents").mockResolvedValue([
       ...mockMicroagents,
     ]);
     // Setup default mock for searchConversations
-    vi.spyOn(ConversationService, "searchConversations").mockResolvedValue([
+    vi.spyOn(OpenHands, "searchConversations").mockResolvedValue([
       ...mockConversations,
     ]);
     // Setup default mock for getRepositoryMicroagentContent
-    vi.spyOn(GitService, "getRepositoryMicroagentContent").mockResolvedValue({
+    vi.spyOn(OpenHands, "getRepositoryMicroagentContent").mockResolvedValue({
       content: "Original microagent content for testing updates",
       path: ".openhands/microagents/update-test-microagent",
       git_provider: "github",
@@ -1331,7 +1290,7 @@ describe("MicroagentManagement", () => {
   // Add microagent integration tests
   describe("Add microagent functionality", () => {
     beforeEach(() => {
-      vi.spyOn(GitService, "getRepositoryBranches").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryBranches").mockResolvedValue({
         branches: [{ name: "main", commit_sha: "abc123", protected: false }],
         has_next_page: false,
         current_page: 1,
@@ -1387,10 +1346,33 @@ describe("MicroagentManagement", () => {
       });
     });
 
-    it("should render modal when Zustand state is set to visible", async () => {
-      // Render with modal already visible in Zustand state
-      renderWithCustomStore({
-        addMicroagentModalVisible: true,
+    it("should render modal when Redux state is set to visible", async () => {
+      // Render with modal already visible in Redux state
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: null,
+            addMicroagentModalVisible: true, // Start with modal visible
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            updateMicroagentModalVisible: false,
+            learnThisRepoModalVisible: false,
+          },
+        },
       });
 
       // Check that modal is rendered
@@ -1660,16 +1642,34 @@ describe("MicroagentManagement", () => {
       pr_number: null,
     };
 
-    const renderMicroagentManagementMain = (selectedMicroagentItem: any) => {
-      // Set the store with the selected microagent item and a repository
-      useMicroagentManagementStore.setState({
-        selectedMicroagentItem,
-        selectedRepository: testRepository,
+    const renderMicroagentManagementMain = (selectedMicroagentItem: any) =>
+      renderWithProviders(<MicroagentManagementMain />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            addMicroagentModalVisible: false,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            selectedMicroagentItem,
+            updateMicroagentModalVisible: false,
+            learnThisRepoModalVisible: false,
+          },
+        },
       });
 
-      return renderWithProviders(<MicroagentManagementMain />);
-    };
-
     it("should render MicroagentManagementDefault when no microagent or conversation is selected", async () => {
       renderMicroagentManagementMain(null);
 
@@ -1983,7 +1983,7 @@ describe("MicroagentManagement", () => {
     };
 
     beforeEach(() => {
-      vi.spyOn(GitService, "getRepositoryBranches").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryBranches").mockResolvedValue({
         branches: [{ name: "main", commit_sha: "abc123", protected: false }],
         has_next_page: false,
         current_page: 1,
@@ -1994,8 +1994,36 @@ describe("MicroagentManagement", () => {
     });
 
     it("should render update microagent modal when updateMicroagentModalVisible is true", async () => {
-      // Render with update modal visible in Zustand state
-      renderWithUpdateModal();
+      // Render with update modal visible in Redux state
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true, // Start with update modal visible
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Check that update modal is rendered
       expect(screen.getByTestId("add-microagent-modal")).toBeInTheDocument();
@@ -2006,7 +2034,35 @@ describe("MicroagentManagement", () => {
 
     it("should display update microagent title when isUpdate is true", async () => {
       // Render with update modal visible and selected microagent
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Check that the update title is displayed
       expect(
@@ -2016,10 +2072,33 @@ describe("MicroagentManagement", () => {
 
     it("should populate form fields with existing microagent data when updating", async () => {
       // Render with update modal visible and selected microagent
-      renderWithUpdateModal({
-        selectedMicroagentItem: {
-          microagent: mockMicroagentForUpdate,
-          conversation: null,
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
         },
       });
 
@@ -2036,7 +2115,35 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Render with update modal visible and selected microagent
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Wait for modal to be rendered
       await waitFor(() => {
@@ -2064,7 +2171,35 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Render with update modal visible
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Wait for modal to be rendered
       await waitFor(() => {
@@ -2087,7 +2222,35 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Render with update modal visible
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Wait for modal to be rendered
       await waitFor(() => {
@@ -2113,7 +2276,32 @@ describe("MicroagentManagement", () => {
 
     it("should handle update modal with empty microagent data", async () => {
       // Render with update modal visible but no microagent data
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: null,
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Check that update modal is still rendered
       expect(screen.getByTestId("add-microagent-modal")).toBeInTheDocument();
@@ -2126,7 +2314,7 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Mock the content API to return empty content for this test
-      vi.spyOn(GitService, "getRepositoryMicroagentContent").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryMicroagentContent").mockResolvedValue({
         content: "",
         path: ".openhands/microagents/update-test-microagent",
         git_provider: "github",
@@ -2134,7 +2322,35 @@ describe("MicroagentManagement", () => {
       });
 
       // Render with update modal visible and microagent
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Wait for the content to be loaded and check that the form field is empty
       await waitFor(() => {
@@ -2147,7 +2363,7 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Mock the content API to return content without triggers for this test
-      vi.spyOn(GitService, "getRepositoryMicroagentContent").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryMicroagentContent").mockResolvedValue({
         content: "Original microagent content for testing updates",
         path: ".openhands/microagents/update-test-microagent",
         git_provider: "github",
@@ -2155,7 +2371,35 @@ describe("MicroagentManagement", () => {
       });
 
       // Render with update modal visible and microagent
-      renderWithUpdateModal();
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForUpdate,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: true,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Check that the modal is rendered correctly
       expect(screen.getByTestId("add-microagent-modal")).toBeInTheDocument();
@@ -2314,7 +2558,35 @@ describe("MicroagentManagement", () => {
 
     it("should render learn something new button in microagent view", async () => {
       // Render with selected microagent
-      renderWithSelectedMicroagent(mockMicroagentForLearn);
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForLearn,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: false,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Check that the learn something new button is displayed
       expect(
@@ -2326,7 +2598,35 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Render with selected microagent
-      renderWithSelectedMicroagent(mockMicroagentForLearn);
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForLearn,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: false,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Find and click the learn something new button
       const learnButton = screen.getByText("COMMON$LEARN_SOMETHING_NEW");
@@ -2347,7 +2647,7 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Mock the content API to return the expected content for this test
-      vi.spyOn(GitService, "getRepositoryMicroagentContent").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryMicroagentContent").mockResolvedValue({
         content: "Test microagent content for learn functionality",
         path: ".openhands/microagents/learn-test-microagent",
         git_provider: "github",
@@ -2355,7 +2655,35 @@ describe("MicroagentManagement", () => {
       });
 
       // Render with selected microagent
-      renderWithSelectedMicroagent(mockMicroagentForLearn);
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForLearn,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: false,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Find and click the learn something new button
       const learnButton = screen.getByText("COMMON$LEARN_SOMETHING_NEW");
@@ -2379,7 +2707,7 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Mock the content API to return empty content for this test
-      vi.spyOn(GitService, "getRepositoryMicroagentContent").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryMicroagentContent").mockResolvedValue({
         content: "",
         path: ".openhands/microagents/learn-test-microagent",
         git_provider: "github",
@@ -2387,7 +2715,35 @@ describe("MicroagentManagement", () => {
       });
 
       // Render with selected microagent
-      renderWithSelectedMicroagent(mockMicroagentForLearn);
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForLearn,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: false,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Find and click the learn something new button
       const learnButton = screen.getByText("COMMON$LEARN_SOMETHING_NEW");
@@ -2409,7 +2765,7 @@ describe("MicroagentManagement", () => {
       const user = userEvent.setup();
 
       // Mock the content API to return content without triggers for this test
-      vi.spyOn(GitService, "getRepositoryMicroagentContent").mockResolvedValue({
+      vi.spyOn(OpenHands, "getRepositoryMicroagentContent").mockResolvedValue({
         content: "Test microagent content for learn functionality",
         path: ".openhands/microagents/learn-test-microagent",
         git_provider: "github",
@@ -2417,7 +2773,35 @@ describe("MicroagentManagement", () => {
       });
 
       // Render with selected microagent
-      renderWithSelectedMicroagent(mockMicroagentForLearn);
+      renderWithProviders(<RouterStub />, {
+        preloadedState: {
+          metrics: {
+            cost: null,
+            max_budget_per_task: null,
+            usage: null,
+          },
+          microagentManagement: {
+            selectedMicroagentItem: {
+              microagent: mockMicroagentForLearn,
+              conversation: undefined,
+            },
+            addMicroagentModalVisible: false,
+            updateMicroagentModalVisible: false,
+            selectedRepository: {
+              id: "1",
+              full_name: "user/test-repo",
+              git_provider: "github",
+              is_public: true,
+              owner_type: "user",
+              pushed_at: "2021-10-01T12:00:00Z",
+            },
+            personalRepositories: [],
+            organizationRepositories: [],
+            repositories: [],
+            learnThisRepoModalVisible: false,
+          },
+        },
+      });
 
       // Find and click the learn something new button
       const learnButton = screen.getByText("COMMON$LEARN_SOMETHING_NEW");

frontend/__tests__/components/features/payment/payment-form.test.tsx

@@ -1,30 +1,23 @@
-import { screen, waitFor } from "@testing-library/react";
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
+import { render, screen, waitFor } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
 import { afterEach, beforeEach, describe, expect, it, test, vi } from "vitest";
-import BillingService from "#/api/billing-service/billing-service.api";
-import OptionService from "#/api/option-service/option-service.api";
+import OpenHands from "#/api/open-hands";
 import { PaymentForm } from "#/components/features/payment/payment-form";
-import { renderWithProviders } from "../../../../test-utils";
-
-// Mock the stripe checkout hook to avoid JSDOM navigation issues
-const mockMutate = vi.fn().mockResolvedValue(undefined);
-vi.mock("#/hooks/mutation/stripe/use-create-stripe-checkout-session", () => ({
-  useCreateStripeCheckoutSession: () => ({
-    mutate: mockMutate,
-    mutateAsync: vi.fn().mockResolvedValue(undefined),
-    isPending: false,
-  }),
-}));
 
 describe("PaymentForm", () => {
-  const getBalanceSpy = vi.spyOn(BillingService, "getBalance");
-  const createCheckoutSessionSpy = vi.spyOn(
-    BillingService,
-    "createCheckoutSession",
-  );
-  const getConfigSpy = vi.spyOn(OptionService, "getConfig");
+  const getBalanceSpy = vi.spyOn(OpenHands, "getBalance");
+  const createCheckoutSessionSpy = vi.spyOn(OpenHands, "createCheckoutSession");
+  const getConfigSpy = vi.spyOn(OpenHands, "getConfig");
 
-  const renderPaymentForm = () => renderWithProviders(<PaymentForm />);
+  const renderPaymentForm = () =>
+    render(<PaymentForm />, {
+      wrapper: ({ children }) => (
+        <QueryClientProvider client={new QueryClient()}>
+          {children}
+        </QueryClientProvider>
+      ),
+    });
 
   beforeEach(() => {
     // useBalance hook will return the balance only if the APP_MODE is "saas" and the billing feature is enabled
@@ -44,7 +37,6 @@ describe("PaymentForm", () => {
 
   afterEach(() => {
     vi.clearAllMocks();
-    mockMutate.mockClear();
   });
 
   it("should render the users current balance", async () => {
@@ -77,7 +69,7 @@ describe("PaymentForm", () => {
     const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
     await user.click(topUpButton);
 
-    expect(mockMutate).toHaveBeenCalledWith({ amount: 50 });
+    expect(createCheckoutSessionSpy).toHaveBeenCalledWith(50);
   });
 
   it("should only accept integer values", async () => {
@@ -90,7 +82,7 @@ describe("PaymentForm", () => {
     const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
     await user.click(topUpButton);
 
-    expect(mockMutate).toHaveBeenCalledWith({ amount: 50 });
+    expect(createCheckoutSessionSpy).toHaveBeenCalledWith(50);
   });
 
   it("should disable the top-up button if the user enters an invalid amount", async () => {
@@ -130,7 +122,7 @@ describe("PaymentForm", () => {
       const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
       await user.click(topUpButton);
 
-      expect(mockMutate).not.toHaveBeenCalled();
+      expect(createCheckoutSessionSpy).not.toHaveBeenCalled();
     });
 
     test("user enters an empty string", async () => {
@@ -143,7 +135,7 @@ describe("PaymentForm", () => {
       const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
       await user.click(topUpButton);
 
-      expect(mockMutate).not.toHaveBeenCalled();
+      expect(createCheckoutSessionSpy).not.toHaveBeenCalled();
     });
 
     test("user enters a non-numeric value", async () => {
@@ -158,7 +150,7 @@ describe("PaymentForm", () => {
       const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
       await user.click(topUpButton);
 
-      expect(mockMutate).not.toHaveBeenCalled();
+      expect(createCheckoutSessionSpy).not.toHaveBeenCalled();
     });
 
     test("user enters less than the minimum amount", async () => {
@@ -171,7 +163,7 @@ describe("PaymentForm", () => {
       const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
       await user.click(topUpButton);
 
-      expect(mockMutate).not.toHaveBeenCalled();
+      expect(createCheckoutSessionSpy).not.toHaveBeenCalled();
     });
 
     test("user enters a decimal value", async () => {
@@ -185,175 +177,7 @@ describe("PaymentForm", () => {
       const topUpButton = screen.getByText("PAYMENT$ADD_CREDIT");
       await user.click(topUpButton);
 
-      expect(mockMutate).not.toHaveBeenCalled();
-    });
-  });
-
-  describe("Cancel Subscription", () => {
-    const getSubscriptionAccessSpy = vi.spyOn(
-      BillingService,
-      "getSubscriptionAccess",
-    );
-    const cancelSubscriptionSpy = vi.spyOn(
-      BillingService,
-      "cancelSubscription",
-    );
-
-    beforeEach(() => {
-      // Mock active subscription
-      getSubscriptionAccessSpy.mockResolvedValue({
-        start_at: "2024-01-01T00:00:00Z",
-        end_at: "2024-12-31T23:59:59Z",
-        created_at: "2024-01-01T00:00:00Z",
-      });
-    });
-
-    it("should render cancel subscription button when user has active subscription", async () => {
-      renderPaymentForm();
-
-      await waitFor(() => {
-        const cancelButton = screen.getByTestId("cancel-subscription-button");
-        expect(cancelButton).toBeInTheDocument();
-        expect(cancelButton).toHaveTextContent("PAYMENT$CANCEL_SUBSCRIPTION");
-      });
-    });
-
-    it("should not render cancel subscription button when user has no subscription", async () => {
-      getSubscriptionAccessSpy.mockResolvedValue(null);
-      renderPaymentForm();
-
-      await waitFor(() => {
-        const cancelButton = screen.queryByTestId("cancel-subscription-button");
-        expect(cancelButton).not.toBeInTheDocument();
-      });
-    });
-
-    it("should show confirmation modal when cancel subscription button is clicked", async () => {
-      const user = userEvent.setup();
-      renderPaymentForm();
-
-      const cancelButton = await screen.findByTestId(
-        "cancel-subscription-button",
-      );
-      await user.click(cancelButton);
-
-      // Should show confirmation modal
-      expect(
-        screen.getByTestId("cancel-subscription-modal"),
-      ).toBeInTheDocument();
-      expect(
-        screen.getByText("PAYMENT$CANCEL_SUBSCRIPTION_TITLE"),
-      ).toBeInTheDocument();
-      // The message should be rendered (either with Trans component or regular text)
-      const modalContent = screen.getByTestId("cancel-subscription-modal");
-      expect(modalContent).toBeInTheDocument();
-      expect(screen.getByTestId("confirm-cancel-button")).toBeInTheDocument();
-      expect(screen.getByTestId("modal-cancel-button")).toBeInTheDocument();
-    });
-
-    it("should close modal when cancel button in modal is clicked", async () => {
-      const user = userEvent.setup();
-      renderPaymentForm();
-
-      const cancelButton = await screen.findByTestId(
-        "cancel-subscription-button",
-      );
-      await user.click(cancelButton);
-
-      // Modal should be visible
-      expect(
-        screen.getByTestId("cancel-subscription-modal"),
-      ).toBeInTheDocument();
-
-      // Click cancel in modal
-      const modalCancelButton = screen.getByTestId("modal-cancel-button");
-      await user.click(modalCancelButton);
-
-      // Modal should be closed
-      expect(
-        screen.queryByTestId("cancel-subscription-modal"),
-      ).not.toBeInTheDocument();
-    });
-
-    it("should call cancel subscription API when confirm button is clicked", async () => {
-      const user = userEvent.setup();
-      renderPaymentForm();
-
-      const cancelButton = await screen.findByTestId(
-        "cancel-subscription-button",
-      );
-      await user.click(cancelButton);
-
-      // Click confirm in modal
-      const confirmButton = screen.getByTestId("confirm-cancel-button");
-      await user.click(confirmButton);
-
-      // Should call the cancel subscription API
-      expect(cancelSubscriptionSpy).toHaveBeenCalled();
-    });
-
-    it("should close modal after successful cancellation", async () => {
-      const user = userEvent.setup();
-      cancelSubscriptionSpy.mockResolvedValue({
-        status: "success",
-        message: "Subscription cancelled successfully",
-      });
-      renderPaymentForm();
-
-      const cancelButton = await screen.findByTestId(
-        "cancel-subscription-button",
-      );
-      await user.click(cancelButton);
-
-      const confirmButton = screen.getByTestId("confirm-cancel-button");
-      await user.click(confirmButton);
-
-      // Wait for API call to complete and modal to close
-      await waitFor(() => {
-        expect(
-          screen.queryByTestId("cancel-subscription-modal"),
-        ).not.toBeInTheDocument();
-      });
-    });
-
-    it("should show next billing date for active subscription", async () => {
-      // Mock active subscription with end_at as next billing date
-      getSubscriptionAccessSpy.mockResolvedValue({
-        start_at: "2024-01-01T00:00:00Z",
-        end_at: "2025-01-01T00:00:00Z",
-        created_at: "2024-01-01T00:00:00Z",
-        cancelled_at: null,
-        stripe_subscription_id: "sub_123",
-      });
-
-      renderPaymentForm();
-
-      await waitFor(() => {
-        const nextBillingInfo = screen.getByTestId("next-billing-date");
-        expect(nextBillingInfo).toBeInTheDocument();
-        // Check that it contains some date-related content (translation key or actual date)
-        expect(nextBillingInfo).toHaveTextContent(
-          /2025|PAYMENT.*BILLING.*DATE/,
-        );
-      });
-    });
-
-    it("should not show next billing date when subscription is cancelled", async () => {
-      // Mock cancelled subscription
-      getSubscriptionAccessSpy.mockResolvedValue({
-        start_at: "2024-01-01T00:00:00Z",
-        end_at: "2025-01-01T00:00:00Z",
-        created_at: "2024-01-01T00:00:00Z",
-        cancelled_at: "2024-06-15T10:30:00Z",
-        stripe_subscription_id: "sub_123",
-      });
-
-      renderPaymentForm();
-
-      await waitFor(() => {
-        const nextBillingInfo = screen.queryByTestId("next-billing-date");
-        expect(nextBillingInfo).not.toBeInTheDocument();
-      });
+      expect(createCheckoutSessionSpy).not.toHaveBeenCalled();
     });
   });
 });

frontend/__tests__/components/features/sidebar/sidebar.test.tsx

@@ -3,7 +3,7 @@ import { renderWithProviders } from "test-utils";
 import { createRoutesStub } from "react-router";
 import { waitFor } from "@testing-library/react";
 import { Sidebar } from "#/components/features/sidebar/sidebar";
-import SettingsService from "#/settings-service/settings-service.api";
+import OpenHands from "#/api/open-hands";
 
 // These tests will now fail because the conversation panel is rendered through a portal
 // and technically not a child of the Sidebar component.
@@ -19,7 +19,7 @@ const renderSidebar = () =>
   renderWithProviders(<RouterStub initialEntries={["/conversation/123"]} />);
 
 describe("Sidebar", () => {
-  const getSettingsSpy = vi.spyOn(SettingsService, "getSettings");
+  const getSettingsSpy = vi.spyOn(OpenHands, "getSettings");
 
   afterEach(() => {
     vi.clearAllMocks();

frontend/__tests__/components/feedback-actions.test.tsx

@@ -8,6 +8,7 @@ describe("TrajectoryActions", () => {
   const user = userEvent.setup();
   const onPositiveFeedback = vi.fn();
   const onNegativeFeedback = vi.fn();
+  const onExportTrajectory = vi.fn();
 
   afterEach(() => {
     vi.clearAllMocks();
@@ -18,12 +19,14 @@ describe("TrajectoryActions", () => {
       <TrajectoryActions
         onPositiveFeedback={onPositiveFeedback}
         onNegativeFeedback={onNegativeFeedback}
+        onExportTrajectory={onExportTrajectory}
       />,
     );
 
     const actions = screen.getByTestId("feedback-actions");
     within(actions).getByTestId("positive-feedback");
     within(actions).getByTestId("negative-feedback");
+    within(actions).getByTestId("export-trajectory");
   });
 
   it("should call onPositiveFeedback when positive feedback is clicked", async () => {
@@ -31,6 +34,7 @@ describe("TrajectoryActions", () => {
       <TrajectoryActions
         onPositiveFeedback={onPositiveFeedback}
         onNegativeFeedback={onNegativeFeedback}
+        onExportTrajectory={onExportTrajectory}
       />,
     );
 
@@ -45,6 +49,7 @@ describe("TrajectoryActions", () => {
       <TrajectoryActions
         onPositiveFeedback={onPositiveFeedback}
         onNegativeFeedback={onNegativeFeedback}
+        onExportTrajectory={onExportTrajectory}
       />,
     );
 
@@ -54,12 +59,48 @@ describe("TrajectoryActions", () => {
     expect(onNegativeFeedback).toHaveBeenCalled();
   });
 
+  it("should call onExportTrajectory when export button is clicked", async () => {
+    renderWithProviders(
+      <TrajectoryActions
+        onPositiveFeedback={onPositiveFeedback}
+        onNegativeFeedback={onNegativeFeedback}
+        onExportTrajectory={onExportTrajectory}
+      />,
+    );
+
+    const exportButton = screen.getByTestId("export-trajectory");
+    await user.click(exportButton);
+
+    expect(onExportTrajectory).toHaveBeenCalled();
+  });
+
   describe("SaaS mode", () => {
+    it("should only render export button when isSaasMode is true", () => {
+      renderWithProviders(
+        <TrajectoryActions
+          onPositiveFeedback={onPositiveFeedback}
+          onNegativeFeedback={onNegativeFeedback}
+          onExportTrajectory={onExportTrajectory}
+          isSaasMode={true}
+        />,
+      );
+
+      const actions = screen.getByTestId("feedback-actions");
+
+      // Should not render feedback buttons in SaaS mode
+      expect(within(actions).queryByTestId("positive-feedback")).toBeNull();
+      expect(within(actions).queryByTestId("negative-feedback")).toBeNull();
+
+      // Should still render export button
+      within(actions).getByTestId("export-trajectory");
+    });
+
     it("should render all buttons when isSaasMode is false", () => {
       renderWithProviders(
         <TrajectoryActions
           onPositiveFeedback={onPositiveFeedback}
           onNegativeFeedback={onNegativeFeedback}
+          onExportTrajectory={onExportTrajectory}
           isSaasMode={false}
         />,
       );
@@ -67,6 +108,7 @@ describe("TrajectoryActions", () => {
       const actions = screen.getByTestId("feedback-actions");
       within(actions).getByTestId("positive-feedback");
       within(actions).getByTestId("negative-feedback");
+      within(actions).getByTestId("export-trajectory");
     });
 
     it("should render all buttons when isSaasMode is undefined (default behavior)", () => {
@@ -74,12 +116,30 @@ describe("TrajectoryActions", () => {
         <TrajectoryActions
           onPositiveFeedback={onPositiveFeedback}
           onNegativeFeedback={onNegativeFeedback}
+          onExportTrajectory={onExportTrajectory}
         />,
       );
 
       const actions = screen.getByTestId("feedback-actions");
       within(actions).getByTestId("positive-feedback");
       within(actions).getByTestId("negative-feedback");
+      within(actions).getByTestId("export-trajectory");
+    });
+
+    it("should call onExportTrajectory when export button is clicked in SaaS mode", async () => {
+      renderWithProviders(
+        <TrajectoryActions
+          onPositiveFeedback={onPositiveFeedback}
+          onNegativeFeedback={onNegativeFeedback}
+          onExportTrajectory={onExportTrajectory}
+          isSaasMode={true}
+        />,
+      );
+
+      const exportButton = screen.getByTestId("export-trajectory");
+      await user.click(exportButton);
+
+      expect(onExportTrajectory).toHaveBeenCalled();
     });
   });
 });

frontend/__tests__/components/file-operations.test.tsx

@@ -0,0 +1,11 @@
+import { describe, it } from "vitest";
+
+describe("File Operations Messages", () => {
+  it.todo("should show success indicator for successful file read operation");
+
+  it.todo("should show failure indicator for failed file read operation");
+
+  it.todo("should show success indicator for successful file edit operation");
+
+  it.todo("should show failure indicator for failed file edit operation");
+});

frontend/__tests__/components/interactive-chat-box.test.tsx

@@ -1,62 +1,12 @@
-import { screen } from "@testing-library/react";
+import { render, screen, within } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
 import { afterEach, beforeAll, describe, expect, it, vi } from "vitest";
-import { MemoryRouter } from "react-router";
 import { InteractiveChatBox } from "#/components/features/chat/interactive-chat-box";
-import { renderWithProviders } from "../../test-utils";
-import { AgentState } from "#/types/agent-state";
-
-// Mock React Router hooks
-vi.mock("react-router", async () => {
-  const actual = await vi.importActual("react-router");
-  return {
-    ...actual,
-    useNavigate: () => vi.fn(),
-    useParams: () => ({ conversationId: "test-conversation-id" }),
-  };
-});
-
-// Mock the useActiveConversation hook
-vi.mock("#/hooks/query/use-active-conversation", () => ({
-  useActiveConversation: () => ({
-    data: { status: null },
-    isFetched: true,
-    refetch: vi.fn(),
-  }),
-}));
-
-// Mock other hooks that might be used by the component
-vi.mock("#/hooks/use-user-providers", () => ({
-  useUserProviders: () => ({
-    providers: [],
-  }),
-}));
-
-vi.mock("#/hooks/use-conversation-name-context-menu", () => ({
-  useConversationNameContextMenu: () => ({
-    isOpen: false,
-    contextMenuRef: { current: null },
-    handleContextMenu: vi.fn(),
-    handleClose: vi.fn(),
-    handleRename: vi.fn(),
-    handleDelete: vi.fn(),
-  }),
-}));
 
 describe("InteractiveChatBox", () => {
   const onSubmitMock = vi.fn();
   const onStopMock = vi.fn();
 
-  // Helper function to render with Router context
-  const renderInteractiveChatBox = (props: any, options: any = {}) => {
-    return renderWithProviders(
-      <MemoryRouter>
-        <InteractiveChatBox {...props} />
-      </MemoryRouter>,
-      options,
-    );
-  };
-
   beforeAll(() => {
     global.URL.createObjectURL = vi
       .fn()
@@ -68,221 +18,111 @@ describe("InteractiveChatBox", () => {
   });
 
   it("should render", () => {
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: false,
-        hasSubstantiveAgentActions: false,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.INIT,
-          },
-        },
-      },
+    render(<InteractiveChatBox onSubmit={onSubmitMock} onStop={onStopMock} />);
+
+    const chatBox = screen.getByTestId("interactive-chat-box");
+    within(chatBox).getByTestId("chat-input");
+    within(chatBox).getByTestId("upload-image-input");
+  });
+
+  it.fails("should set custom values", () => {
+    render(
+      <InteractiveChatBox
+        onSubmit={onSubmitMock}
+        onStop={onStopMock}
+        value="Hello, world!"
+      />,
     );
 
     const chatBox = screen.getByTestId("interactive-chat-box");
-    expect(chatBox).toBeInTheDocument();
-  });
+    const chatInput = within(chatBox).getByTestId("chat-input");
 
-  it("should set custom values", async () => {
-    const user = userEvent.setup();
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: true,
-        hasSubstantiveAgentActions: true,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.AWAITING_USER_INPUT,
-          },
-          conversation: {
-            isRightPanelShown: true,
-            shouldStopConversation: false,
-            shouldStartConversation: false,
-            images: [],
-            files: [],
-            loadingFiles: [],
-            loadingImages: [],
-            messageToSend: null,
-            shouldShownAgentLoading: false,
-          },
-        },
-      },
-    );
-
-    const textbox = screen.getByTestId("chat-input");
-
-    // Simulate user typing to populate the input
-    await user.type(textbox, "Hello, world!");
-
-    expect(textbox).toHaveTextContent("Hello, world!");
+    expect(chatInput).toHaveValue("Hello, world!");
   });
 
   it("should display the image previews when images are uploaded", async () => {
     const user = userEvent.setup();
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: false,
-        hasSubstantiveAgentActions: false,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.INIT,
-          },
-        },
-      },
-    );
+    render(<InteractiveChatBox onSubmit={onSubmitMock} onStop={onStopMock} />);
 
-    // Create a larger file to ensure it passes validation
-    const fileContent = new Array(1024).fill("a").join(""); // 1KB file
-    const file = new File([fileContent], "chucknorris.png", {
-      type: "image/png",
-    });
-
-    // Click on the paperclip icon to trigger file selection
-    const paperclipIcon = screen.getByTestId("paperclip-icon");
-    await user.click(paperclipIcon);
-
-    // Now trigger the file input change event directly
+    const file = new File(["(⌐□_□)"], "chucknorris.png", { type: "image/png" });
     const input = screen.getByTestId("upload-image-input");
-    await user.upload(input, file);
 
-    // For now, just verify the file input is accessible
-    expect(input).toBeInTheDocument();
+    expect(screen.queryAllByTestId("image-preview")).toHaveLength(0);
+
+    await user.upload(input, file);
+    expect(screen.queryAllByTestId("image-preview")).toHaveLength(1);
+
+    const files = [
+      new File(["(⌐□_□)"], "chucknorris2.png", { type: "image/png" }),
+      new File(["(⌐□_□)"], "chucknorris3.png", { type: "image/png" }),
+    ];
+
+    await user.upload(input, files);
+    expect(screen.queryAllByTestId("image-preview")).toHaveLength(3);
   });
 
   it("should remove the image preview when the close button is clicked", async () => {
     const user = userEvent.setup();
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: false,
-        hasSubstantiveAgentActions: false,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.INIT,
-          },
-        },
-      },
-    );
-
-    const fileContent = new Array(1024).fill("a").join(""); // 1KB file
-    const file = new File([fileContent], "chucknorris.png", {
-      type: "image/png",
-    });
-
-    // Click on the paperclip icon to trigger file selection
-    const paperclipIcon = screen.getByTestId("paperclip-icon");
-    await user.click(paperclipIcon);
+    render(<InteractiveChatBox onSubmit={onSubmitMock} onStop={onStopMock} />);
 
+    const file = new File(["(⌐□_□)"], "chucknorris.png", { type: "image/png" });
     const input = screen.getByTestId("upload-image-input");
-    await user.upload(input, file);
 
-    // For now, just verify the file input is accessible
-    expect(input).toBeInTheDocument();
+    await user.upload(input, file);
+    expect(screen.queryAllByTestId("image-preview")).toHaveLength(1);
+
+    const imagePreview = screen.getByTestId("image-preview");
+    const closeButton = within(imagePreview).getByRole("button");
+    await user.click(closeButton);
+
+    expect(screen.queryAllByTestId("image-preview")).toHaveLength(0);
   });
 
   it("should call onSubmit with the message and images", async () => {
     const user = userEvent.setup();
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: false,
-        hasSubstantiveAgentActions: false,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.INIT,
-          },
-        },
-      },
+    render(<InteractiveChatBox onSubmit={onSubmitMock} onStop={onStopMock} />);
+
+    const textarea = within(screen.getByTestId("chat-input")).getByRole(
+      "textbox",
     );
+    const input = screen.getByTestId("upload-image-input");
+    const file = new File(["(⌐□_□)"], "chucknorris.png", { type: "image/png" });
 
-    const textarea = screen.getByTestId("chat-input");
-
-    // Type the message and ensure it's properly set
+    await user.upload(input, file);
     await user.type(textarea, "Hello, world!");
+    await user.keyboard("{Enter}");
 
-    // Set innerText directly as the component reads this property
-    textarea.innerText = "Hello, world!";
+    expect(onSubmitMock).toHaveBeenCalledWith("Hello, world!", [file], []);
 
-    // Verify the text is in the input before submitting
-    expect(textarea).toHaveTextContent("Hello, world!");
-
-    // Click the submit button instead of pressing Enter for more reliable testing
-    const submitButton = screen.getByTestId("submit-button");
-
-    // Verify the button is enabled before clicking
-    expect(submitButton).not.toBeDisabled();
-
-    await user.click(submitButton);
-
-    expect(onSubmitMock).toHaveBeenCalledWith("Hello, world!", [], []);
+    // clear images after submission
+    expect(screen.queryAllByTestId("image-preview")).toHaveLength(0);
   });
 
-  it("should disable the submit button when agent is loading", async () => {
+  it("should disable the submit button", async () => {
     const user = userEvent.setup();
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: false,
-        hasSubstantiveAgentActions: false,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.LOADING,
-          },
-        },
-      },
+    render(
+      <InteractiveChatBox
+        isDisabled
+        onSubmit={onSubmitMock}
+        onStop={onStopMock}
+      />,
     );
 
-    const button = screen.getByTestId("submit-button");
+    const button = screen.getByRole("button");
     expect(button).toBeDisabled();
 
     await user.click(button);
     expect(onSubmitMock).not.toHaveBeenCalled();
   });
 
-  it("should display the stop button when agent is running and call onStop when clicked", async () => {
+  it("should display the stop button if set and call onStop when clicked", async () => {
     const user = userEvent.setup();
-    renderInteractiveChatBox(
-      {
-        onSubmit: onSubmitMock,
-        onStop: onStopMock,
-        isWaitingForUserInput: false,
-        hasSubstantiveAgentActions: true,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.RUNNING,
-          },
-        },
-      },
+    render(
+      <InteractiveChatBox
+        mode="stop"
+        onSubmit={onSubmitMock}
+        onStop={onStopMock}
+      />,
     );
 
     const stopButton = screen.getByTestId("stop-button");
@@ -296,57 +136,55 @@ describe("InteractiveChatBox", () => {
     const user = userEvent.setup();
     const onSubmit = vi.fn();
     const onStop = vi.fn();
+    const onChange = vi.fn();
 
-    const { rerender } = renderInteractiveChatBox(
-      {
-        onSubmit: onSubmit,
-        onStop: onStop,
-        isWaitingForUserInput: true,
-        hasSubstantiveAgentActions: true,
-        optimisticUserMessage: false,
-      },
-      {
-        preloadedState: {
-          agent: {
-            curAgentState: AgentState.AWAITING_USER_INPUT,
-          },
-          conversation: {
-            isRightPanelShown: true,
-            shouldStopConversation: false,
-            shouldStartConversation: false,
-            images: [],
-            files: [],
-            loadingFiles: [],
-            loadingImages: [],
-            messageToSend: null,
-            shouldShownAgentLoading: false,
-          },
-        },
-      },
+    const { rerender } = render(
+      <InteractiveChatBox
+        onSubmit={onSubmit}
+        onStop={onStop}
+        onChange={onChange}
+        value="test message"
+      />,
     );
 
-    // Verify text input has the initial value
-    const textarea = screen.getByTestId("chat-input");
-    expect(textarea).toHaveTextContent("");
+    // Upload an image via the upload button - this should NOT clear the text input
+    const file = new File(["dummy content"], "test.png", { type: "image/png" });
+    const input = screen.getByTestId("upload-image-input");
+    await user.upload(input, file);
 
-    // Set innerText directly as the component reads this property
-    textarea.innerText = "test message";
+    // Verify text input was not cleared
+    expect(screen.getByRole("textbox")).toHaveValue("test message");
+    expect(onChange).not.toHaveBeenCalledWith("");
 
-    // Submit the message
-    const submitButton = screen.getByTestId("submit-button");
+    // Submit the message with image
+    const submitButton = screen.getByRole("button", { name: "BUTTON$SEND" });
     await user.click(submitButton);
 
-    // Verify onSubmit was called with the message
-    expect(onSubmit).toHaveBeenCalledWith("test message", [], []);
+    // Verify onSubmit was called with the message and image
+    expect(onSubmit).toHaveBeenCalledWith("test message", [file], []);
+
+    // Verify onChange was called to clear the text input
+    expect(onChange).toHaveBeenCalledWith("");
 
     // Simulate parent component updating the value prop
     rerender(
-      <MemoryRouter>
-        <InteractiveChatBox onSubmit={onSubmit} onStop={onStop} />
-      </MemoryRouter>,
+      <InteractiveChatBox
+        onSubmit={onSubmit}
+        onStop={onStop}
+        onChange={onChange}
+        value=""
+      />,
     );
 
     // Verify the text input was cleared
-    expect(screen.getByTestId("chat-input")).toHaveTextContent("");
+    expect(screen.getByRole("textbox")).toHaveValue("");
+
+    // Upload another image - this should NOT clear the text input
+    onChange.mockClear();
+    await user.upload(input, file);
+
+    // Verify text input is still empty and onChange was not called
+    expect(screen.getByRole("textbox")).toHaveValue("");
+    expect(onChange).not.toHaveBeenCalled();
   });
 });

frontend/__tests__/components/jupyter/jupyter.test.tsx

@@ -2,8 +2,8 @@ import { render, screen } from "@testing-library/react";
 import { Provider } from "react-redux";
 import { configureStore } from "@reduxjs/toolkit";
 import { JupyterEditor } from "#/components/features/jupyter/jupyter";
-import { useJupyterStore } from "#/state/jupyter-store";
-import { vi, describe, it, expect, beforeEach } from "vitest";
+import { jupyterReducer } from "#/state/jupyter-slice";
+import { vi, describe, it, expect } from "vitest";
 
 describe("JupyterEditor", () => {
   const mockStore = configureStore({
@@ -15,20 +15,19 @@ describe("JupyterEditor", () => {
       code: () => ({}),
       cmd: () => ({}),
       agent: () => ({}),
+      jupyter: jupyterReducer,
       securityAnalyzer: () => ({}),
       status: () => ({}),
     },
-  });
-
-  beforeEach(() => {
-    // Reset the Zustand store before each test
-    useJupyterStore.setState({
-      cells: Array(20).fill({
-        content: "Test cell content",
-        type: "input",
-        imageUrls: undefined,
-      }),
-    });
+    preloadedState: {
+      jupyter: {
+        cells: Array(20).fill({
+          content: "Test cell content",
+          type: "input",
+          output: "Test output",
+        }),
+      },
+    },
   });
 
   it("should have a scrollable container", () => {
@@ -37,7 +36,7 @@ describe("JupyterEditor", () => {
         <div style={{ height: "100vh" }}>
           <JupyterEditor maxWidth={800} />
         </div>
-      </Provider>,
+      </Provider>
     );
 
     const container = screen.getByTestId("jupyter-container");