fix: tolerate missing responses output (#9449)

Signed-off-by: Angie Jones <jones.angie@gmail.com>

.github/ISSUE_TEMPLATE/bug_report.md

@@ -2,7 +2,7 @@
 name: Bug report
 about: Create a report to help us improve
 title: ''
-type: bug
+labels: bug
 assignees: ''
 ---
 

.github/ISSUE_TEMPLATE/config.yml

@@ -1 +1,8 @@
 blank_issues_enabled: false
+contact_links:
+  - name: goose Discord discussion
+    url: https://discord.gg/goose-oss
+    about: Please ask and answer questions here.
+  - name: Report a security vulnerability
+    url: https://github.com/aaif-goose/goose/security/policy
+    about: Please report security vulnerabilities here.

.github/ISSUE_TEMPLATE/feature_request.md

@@ -2,14 +2,11 @@
 name: Feature request
 about: Suggest an idea for this project
 title: ''
-type: 'feature'
+labels: 'enhancement'
 assignees: ''
 
 ---
 
-**Before opening a feature request**
-Please start a discussion in the [goose-eng Discord channel](https://discord.com/channels/1287729918100246654/1514412780504088677) before creating a feature request or beginning implementation. This helps align on direction before anyone spends time building.
-
 **Please explain the motivation behind the feature request.**
 Does this feature solve a particular problem you have been experiencing? What opportunities or use cases would be unlocked with this feature?
 

.github/ISSUE_TEMPLATE/submit-recipe.yml

@@ -0,0 +1,99 @@
+name: 🧑‍🍳 Submit a recipe to the goose cookbook
+description: Share a reusable goose recipe with the community!
+title: "[Recipe] <your recipe title here>"
+labels: ["recipe submission"]
+body:
+  - type: markdown
+    attributes:
+      value: |
+        Thanks for contributing to the goose cookbook! 🍳  
+        Recipes are reusable sessions created in goose desktop or CLI and shared with the community to help others vibe code faster.
+
+        📌 **How to Submit**
+        - Create your recipe using goose ("Make recipe from this session")
+        - Fill out the YAML below using the format provided
+        - Paste it into the field and submit the issue — we'll review and add it to the cookbook!
+
+        🪄 **What Happens After?**
+        - If accepted, we'll publish your recipe to the [goose recipes cookbook](https://goose-docs.ai/recipes)
+        - Your GitHub handle will be displayed and linked on the recipe card
+        - If the YAML has any issues, goose will comment with validation errors so you can fix and resubmit.
+
+        🧪 **Pro Tip:** You can test your recipe locally in your terminal with:  
+        `goose recipe validate your-recipe.yaml`
+
+  - type: textarea
+    id: recipe-yaml
+    attributes:
+      label: Paste Your Full Recipe YAML Below
+      description: Use the structure below and we'll auto-fill your GitHub handle for `author.contact` after submission.
+      placeholder: |
+        version: "1.0.0"
+        id: clean-up-feature-flag
+        title: Clean Up Feature Flag
+        description: Automatically clean up all references of a fully rolled out feature flag from a codebase and make the new behavior the default.
+        instructions: |
+          Your job is to systematically remove a fully rolled out feature flag and ensure the new behavior is now the default. Use code search tools like ripgrep to identify all references to the flag, clean up definition files, usage sites, tests, and configuration files. Then create a commit and push changes with clear commit messages documenting the flag removal.
+        prompt: |
+          Task: Remove a feature flag that has been fully rolled out, where the feature flag's functionality should become the default behavior.
+
+          Context:
+            Feature flag key: {{ feature_flag_key }}
+            Project: {{ repo_dir }}
+
+          Steps to follow:
+            1. Check out a *new* branch from main or master named using the feature flag key.
+            2. Find the feature flag constant/object that wraps the key.
+            3. Search for all references to the constant/object using ripgrep or equivalent tools.
+            4. Remove all conditional logic and make the new behavior default.
+            5. Remove unused imports, mocks, config, and tests.
+            6. Commit your changes and push the branch.
+            7. Open a GitHub PR.
+
+          Use commit messages like:
+            chore(flag-cleanup): remove <feature_flag_key> flag from codebase
+
+        parameters:
+          - key: feature_flag_key
+            input_type: string
+            requirement: required
+            description: Key of the feature flag
+
+          - key: repo_dir
+            input_type: string
+            requirement: optional
+            default: ./
+            description: Directory of the codebase
+
+        extensions:
+          - type: stdio
+            name: developer
+            cmd: uvx
+            args:
+              - developer-mcp@latest
+            timeout: 300
+            bundled: true
+            description: Access developer tools
+        
+        activities:
+          - Remove feature flag definitions
+          - Clean up feature flag usage sites
+          - Update affected tests
+          - Remove flag configurations
+          - Document flag removal
+    validations:
+      required: true
+
+  - type: markdown
+    attributes:
+      value: |
+        🛠 **Recipe Field Tips**
+        - `version` must be "1.0.0" for now
+        - `id` should be lowercase, hyphenated, and unique (e.g. `my-awesome-recipe`)
+        - `title` is the display name of your recipe
+        - `description` should clearly explain what the recipe does
+        - `instructions` are specific steps goose should follow — supports template variables like `{{ variable_name }}`
+        - `prompt` is the first thing goose sees when the recipe is launched
+        - `parameters` should include required or optional inputs — optional ones must have `default`
+        - `extensions` must follow the full format with `type`, `cmd`, `args`, `timeout`, etc.
+        - `activities` describe the main actions the recipe performs

.github/actions/generate-release-pr-body/pr_body_template.txt

@@ -2,7 +2,7 @@
 
 ## Test before Release
 
-1. Approve the workflows for this PR (scroll to bottom) to trigger CI and produce the Desktop bundle for testing.
+1. Close and reopen this PR to trigger CI and produce the Desktop bundle for testing. Reason: workflows don't run on PRs opened by `GITHUB_TOKEN` ([docs](https://docs.github.com/en/actions/using-workflows/triggering-a-workflow#triggering-a-workflow-from-a-workflow)).
 2. Make sure all check workflows pass.
 3. Install the Desktop bundle from the download links (posted as a comment below when it is ready).
 4. Complete the goose Release Manual Testing Checklist (posted as a comment below).

.github/copilot-instructions.md

@@ -40,7 +40,7 @@
 ## Project-Specific Context
 
 - This is a Rust project using cargo workspaces
-- Core crates: `goose` (agent logic and ACP server), `goose-cli` (CLI), `goose-mcp` (MCP servers)
+- Core crates: `goose` (agent logic), `goose-cli` (CLI), `goose-server` (backend), `goose-mcp` (MCP servers)
 - Error handling: Use `anyhow::Result`, not `unwrap()` in production code
 - Async runtime: tokio
 - MCP protocol implementations require extra scrutiny
@@ -56,6 +56,7 @@
 - `cargo fmt --check` - Code formatting (rustfmt)
 - `cargo test --jobs 2` - All tests
 - `cargo clippy --all-targets -- -D warnings` - Linting (clippy)
+- `just check-openapi-schema` - OpenAPI schema validation
 
 **Desktop app checks:**
 - `pnpm install --frozen-lockfile` - Fresh dependency install (in `ui/desktop/`)

.github/workflows/autoclose

@@ -0,0 +1,23 @@
+name: Close inactive issues
+on:
+  schedule:
+    - cron: "30 1 * * *"
+
+jobs:
+  close-issues:
+    runs-on: ubuntu-latest
+    permissions:
+      issues: write
+      pull-requests: write
+    steps:
+      - uses: actions/stale@v9
+        with:
+          days-before-issue-stale: 60
+          days-before-issue-close: 21
+          stale-issue-label: "stale"
+          stale-issue-message: "This issue is stale because it has been open for 30 days with no activity."
+          close-issue-message: "This issue was closed because it has been inactive for 14 days since being marked as stale."
+          days-before-pr-stale: -1
+          days-before-pr-close: -1
+          any-of-labels: "bug,Bug"
+          repo-token: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/build-cli.yml

@@ -4,7 +4,7 @@
 #  - canary.yml
 #
 # Platform Build Strategy:
-#  - Linux standard (x86_64 + aarch64): Builds inside manylinux_2_28 container for glibc 2.28+ compat
+#  - Linux standard: Uses native Ubuntu 22.04 runners to keep glibc compatibility with Ubuntu 22.04 LTS
 #  - Linux Vulkan: Uses native Ubuntu 24.04 runners for newer Vulkan headers/tooling
 #  - Linux musl: Uses native Ubuntu 22.04 runners with reduced features for musl compatibility
 #  - macOS: Uses native macOS runners for each architecture
@@ -27,7 +27,6 @@ jobs:
   build-cli:
     name: Build CLI
     runs-on: ${{ matrix.build-on }}
-    container: ${{ matrix.container }}
     env:
       MACOSX_DEPLOYMENT_TARGET: "12.0"
     strategy:
@@ -38,15 +37,11 @@ jobs:
             architecture: x86_64
             target-suffix: unknown-linux-gnu
             build-on: ubuntu-22.04
-            # Pinned by digest for reproducible builds; bump explicitly when newer manylinux_2_28 images ship.
-            container: quay.io/pypa/manylinux_2_28_x86_64@sha256:441c35fdc6ee809ff9260894f8468ab4fea8c15dc880f8700a3f81b7922c1cda
             variant: standard
           - platform: linux
             architecture: aarch64
             target-suffix: unknown-linux-gnu
             build-on: ubuntu-22.04-arm
-            # Pinned by digest for reproducible builds; bump explicitly when newer manylinux_2_28 images ship.
-            container: quay.io/pypa/manylinux_2_28_aarch64@sha256:8b5f2b4e8c072ae5aefeb659f22c03e1ff46e6a82f154b6c904b106c87e65ff7
             variant: standard
           - platform: linux
             architecture: x86_64
@@ -86,12 +81,12 @@ jobs:
           - platform: windows
             architecture: x86_64
             target-suffix: pc-windows-msvc
-            build-on: windows-2022
+            build-on: windows-latest
             variant: cuda
 
     steps:
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ inputs.ref }}
 
@@ -102,8 +97,8 @@ jobs:
           sed -i.bak 's/^version = ".*"/version = "'${{ inputs.version }}'"/' Cargo.toml
           rm -f Cargo.toml.bak
 
-      - name: Install Linux build dependencies (host runner)
-        if: matrix.platform == 'linux' && matrix.container == ''
+      - name: Install Linux build dependencies
+        if: matrix.platform == 'linux'
         run: |
           sudo apt-get update
           sudo apt-get install -y \
@@ -124,28 +119,11 @@ jobs:
             sudo apt-get install -y musl-tools
           fi
 
-      - name: Install Linux build dependencies (manylinux container)
-        if: matrix.platform == 'linux' && matrix.container != ''
-        run: |
-          # perl-core provides FindBin, File::Compare, etc. that openssl-sys's
-          # vendored openssl build needs; in AlmaLinux 8 these aren't standalone packages.
-          # clang provides libclang.so for bindgen (used by llama-cpp-sys-2).
-          # Defensive: avoid actions/checkout falling back to a tarball download if base image changes.
-          dnf install -y --setopt=install_weak_deps=False \
-            openssl-devel \
-            dbus-devel \
-            libxcb-devel \
-            cmake \
-            perl-core \
-            clang \
-            git \
-            tar
-
       - name: Cache Cargo artifacts (Linux/macOS)
         if: matrix.platform != 'windows'
         uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4  # v2.9.1
         with:
-          key: ${{ matrix.architecture }}-${{ matrix.target-suffix }}-${{ matrix.build-on }}-${{ matrix.container || 'native' }}-macos-deployment-target-12
+          key: ${{ matrix.architecture }}-${{ matrix.target-suffix }}-${{ matrix.build-on }}-native-macos-deployment-target-12
 
       - name: Cache Cargo artifacts (Windows)
         if: matrix.platform == 'windows'
@@ -153,8 +131,8 @@ jobs:
         with:
           key: windows-msvc-cli-${{ matrix.variant }}
 
-      - name: Build CLI (Linux/macOS host runner)
-        if: matrix.platform != 'windows' && matrix.container == ''
+      - name: Build CLI (Linux/macOS)
+        if: matrix.platform != 'windows'
         env:
           RUST_LOG: debug
           RUST_BACKTRACE: 1
@@ -179,27 +157,6 @@ jobs:
             cargo build --release --target ${TARGET} -p goose-cli "${FEATURE_ARGS[@]}"
           fi
 
-      - name: Build CLI (manylinux container)
-        if: matrix.platform == 'linux' && matrix.container != ''
-        env:
-          RUST_BACKTRACE: 1
-        run: |
-          # Hermit's tool cache is host-runner-scoped; inside the container we
-          # bootstrap rustup directly and let rust-toolchain.toml pin the channel.
-          curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs \
-            | sh -s -- -y --default-toolchain none --profile minimal --no-modify-path
-          export PATH="$HOME/.cargo/bin:$PATH"
-          TARGET="${{ matrix.architecture }}-${{ matrix.target-suffix }}"
-          RUST_CHANNEL=$(grep '^channel' rust-toolchain.toml | cut -d'"' -f2)
-          if [ -z "$RUST_CHANNEL" ]; then
-            echo "Could not parse channel from rust-toolchain.toml" >&2
-            exit 1
-          fi
-          rustup toolchain install "$RUST_CHANNEL" --profile minimal \
-            --component rustc,cargo --target "$TARGET"
-          rustup show
-          cargo build --release --target "$TARGET" -p goose-cli
-
       - name: Setup Rust (Windows)
         if: matrix.platform == 'windows'
         shell: bash
@@ -258,10 +215,7 @@ jobs:
       - name: Package CLI (Linux/macOS)
         if: matrix.platform != 'windows'
         run: |
-          # Hermit isn't installed in the manylinux container; tar is all this step needs.
-          if [ "${{ matrix.container }}" = '' ]; then
-            source ./bin/activate-hermit
-          fi
+          source ./bin/activate-hermit
           export TARGET="${{ matrix.architecture }}-${{ matrix.target-suffix }}"
           export VARIANT_SUFFIX=""
           if [ "${{ matrix.variant }}" = "vulkan" ]; then
@@ -299,7 +253,7 @@ jobs:
           echo "ARTIFACT_ZIP=target/${TARGET}/release/goose-${TARGET}${VARIANT_SUFFIX}.zip" >> $GITHUB_ENV
 
       - name: Upload CLI artifact
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: goose-${{ matrix.architecture }}-${{ matrix.target-suffix }}${{ matrix.variant != 'standard' && matrix.variant != 'musl' && format('-{0}', matrix.variant) || '' }}
           path: |

.github/workflows/build-notify.yml

@@ -7,8 +7,6 @@ on:
       - Release
       - Canary
     types: [completed]
-    branches-ignore:
-      - "release/**"
 
 
 jobs:
@@ -36,4 +34,4 @@ jobs:
                 "color": 15158332
               }]
             }' \
-            "$DISCORD_WEBHOOK_URL"
+            "$DISCORD_WEBHOOK_URL"

.github/workflows/bundle-desktop-intel.yml

@@ -49,7 +49,7 @@ jobs:
         run: df -h
 
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           # Only pass ref if it's explicitly set, otherwise let checkout action use its default behavior
           ref: ${{ inputs.ref != '' && inputs.ref || '' }}
@@ -73,11 +73,11 @@ jobs:
           key: intel-macos-deployment-target-12
 
 
-      - name: Build desktop backend for Intel macOS (x86_64)
+      - name: Build goose-server for Intel macOS (x86_64)
         run: |
           source ./bin/activate-hermit
           rustup target add x86_64-apple-darwin
-          cargo build --release -p goose-cli --bin goose --target x86_64-apple-darwin
+          cargo build --release -p goose-server --target x86_64-apple-darwin
 
 
 
@@ -95,16 +95,12 @@ jobs:
           # Check disk space after cleanup
           df -h
 
-      - name: Copy backend binary into Electron folder
+      - name: Copy binaries into Electron folder
         run: |
-          mkdir -p ui/desktop/src/bin
-          rm -f ui/desktop/src/bin/goose
-          cp target/x86_64-apple-darwin/release/goose ui/desktop/src/bin/goose
-          chmod +x ui/desktop/src/bin/goose
-          ls -la ui/desktop/src/bin/
+          cp target/x86_64-apple-darwin/release/goosed ui/desktop/src/bin/goosed
 
       - name: Cache pnpm dependencies
-        uses: actions/cache@2c8a9bd7457de244a408f35966fab2fb45fda9c8  # v6.0.0
+        uses: actions/cache@8b402f58fbc84540c8b491a91e594a4576fec3d7  # v5.0.2
         with:
           path: |
             ui/desktop/node_modules
@@ -156,10 +152,6 @@ jobs:
           fi
         working-directory: ui/desktop
 
-      - name: Verify macOS updater resources
-        run: node scripts/verify-mac-update-resources.js "out/Goose-darwin-x64/Goose.app"
-        working-directory: ui/desktop
-
       - name: Clean up signing keychain
         if: always()
         run: |
@@ -176,7 +168,7 @@ jobs:
           df -h
 
       - name: Upload Desktop artifact
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-darwin-x64
           path: ui/desktop/out/Goose-darwin-x64/Goose_intel_mac.zip

.github/workflows/bundle-desktop-linux.yml

@@ -39,7 +39,7 @@ jobs:
 
     steps:
       - name: Checkout repository
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ inputs.ref || inputs.branch }}
 
@@ -104,11 +104,6 @@ jobs:
             sudo apt-get install -y libasound2
           fi
 
-      - name: Configure RPM packaging
-        run: |
-          echo '%_build_id_links none' > ~/.rpmmacros
-          test "$(rpm --eval '%{_build_id_links}')" = "none"
-
       - name: Setup Flatpak Runtimes
         run: |
           flatpak remote-add --if-not-exists --user flathub https://dl.flathub.org/repo/flathub.flatpakrepo
@@ -124,7 +119,7 @@ jobs:
         with:
           key: linux-${{ matrix.build-on }}-${{ matrix.variant }}
 
-      - name: Build desktop backend binary
+      - name: Build goosed binary
         env:
           RUST_LOG: debug
           RUST_BACKTRACE: 1
@@ -137,25 +132,19 @@ jobs:
             FEATURE_ARGS=(--features vulkan)
           fi
 
-          cargo build --release --target ${TARGET} -p goose-cli --bin goose "${FEATURE_ARGS[@]}"
+          cargo build --release --target ${TARGET} -p goose-server "${FEATURE_ARGS[@]}"
 
-      - name: Copy backend binary into Electron folder
+      - name: Copy binaries into Electron folder
         run: |
-          echo "Copying backend binary to ui/desktop/src/bin/"
+          echo "Copying binaries to ui/desktop/src/bin/"
           export TARGET="x86_64-unknown-linux-gnu"
           mkdir -p ui/desktop/src/bin
-          rm -f ui/desktop/src/bin/goose
-          cp target/$TARGET/release/goose ui/desktop/src/bin/
-          chmod +x ui/desktop/src/bin/goose
+          cp target/$TARGET/release/goosed ui/desktop/src/bin/
+          chmod +x ui/desktop/src/bin/goosed
           ls -la ui/desktop/src/bin/
 
-      - name: Free Rust build artifacts before packaging
-        run: |
-          source ./bin/activate-hermit
-          cargo clean
-
       - name: Cache pnpm dependencies
-        uses: actions/cache@2c8a9bd7457de244a408f35966fab2fb45fda9c8  # v6.0.0
+        uses: actions/cache@8b402f58fbc84540c8b491a91e594a4576fec3d7  # v5.0.2
         with:
           path: |
             ui/desktop/node_modules
@@ -192,14 +181,6 @@ jobs:
             done
           fi
 
-          for package in out/make/rpm/x64/*.rpm; do
-            package_contents="$(rpm -qlp "$package")"
-            if grep -qE '^/usr/lib/\.build-id(/|$)' <<< "$package_contents"; then
-              echo "::error file=$package::RPM contains build-id links"
-              exit 1
-            fi
-          done
-
           echo "Build completed. Checking output..."
           ls -la out/
           find out/ -name "*.deb" -o -name "*.rpm" -o -name "*.flatpak" | head -10
@@ -217,7 +198,7 @@ jobs:
 
       - name: Upload .deb package
         if: matrix.variant == 'standard'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-linux-x64-deb
           path: ui/desktop/out/make/deb/x64/*.deb
@@ -225,7 +206,7 @@ jobs:
 
       - name: Upload .deb package (Vulkan)
         if: matrix.variant == 'vulkan'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-linux-x64-vulkan-deb
           path: ui/desktop/out/make/deb/x64/*-vulkan.deb
@@ -233,7 +214,7 @@ jobs:
 
       - name: Upload .rpm package
         if: matrix.variant == 'standard'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-linux-x64-rpm
           path: ui/desktop/out/make/rpm/x64/*.rpm
@@ -241,7 +222,7 @@ jobs:
 
       - name: Upload .rpm package (Vulkan)
         if: matrix.variant == 'vulkan'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-linux-x64-vulkan-rpm
           path: ui/desktop/out/make/rpm/x64/*-vulkan.rpm
@@ -249,7 +230,7 @@ jobs:
 
       - name: Upload .flatpak package
         if: matrix.variant == 'standard'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-linux-x64-flatpak
           path: ui/desktop/out/make/flatpak/x86_64/*.flatpak
@@ -257,7 +238,7 @@ jobs:
 
       - name: Upload .flatpak package (Vulkan)
         if: matrix.variant == 'vulkan'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-linux-x64-vulkan-flatpak
           path: ui/desktop/out/make/flatpak/x86_64/*-vulkan.flatpak

.github/workflows/bundle-desktop-windows.yml

@@ -44,16 +44,16 @@ permissions:
 jobs:
   build-desktop-windows:
     name: Build Desktop (Windows)
-    runs-on: ${{ inputs.windows_variant == 'cuda' && 'windows-2022' || 'windows-latest' }}
+    runs-on: windows-latest
 
     steps:
       - name: Checkout repository
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ inputs.ref != '' && inputs.ref || '' }}
 
       - name: Set up Node.js
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e  # v6.4.0
+        uses: actions/setup-node@6044e13b5dc448c55e2357c09f80417699197238  # v6.2.0
         with:
           node-version: 24.10.0
 
@@ -61,7 +61,7 @@ jobs:
         run: npm install -g pnpm@10.30.3
 
       - name: Cache node_modules
-        uses: actions/cache@2c8a9bd7457de244a408f35966fab2fb45fda9c8  # v6.0.0
+        uses: actions/cache@8b402f58fbc84540c8b491a91e594a4576fec3d7  # v5.0.2
         with:
           path: |
             node_modules
@@ -113,33 +113,28 @@ jobs:
         env:
           CUDA_COMPUTE_CAP: ${{ inputs.windows_variant == 'cuda' && '80' || '' }}
         run: |
-          $isCuda = "${{ inputs.windows_variant }}" -eq "cuda"
-
-          Write-Output "Building Windows ACP backend"
-          if ($isCuda) {
-            cargo build --release --target x86_64-pc-windows-msvc -p goose-cli --bin goose --features cuda
+          Write-Output "Building Windows executable..."
+          if ("${{ inputs.windows_variant }}" -eq "cuda") {
+            cargo build --release --target x86_64-pc-windows-msvc -p goose-server --features cuda
           } else {
-            cargo build --release --target x86_64-pc-windows-msvc -p goose-cli --bin goose
+            cargo build --release --target x86_64-pc-windows-msvc -p goose-server
           }
-          $binaryPath = "./target/x86_64-pc-windows-msvc/release/goose.exe"
 
           # Verify build succeeded
-          if (-not (Test-Path $binaryPath)) {
-            Write-Error "Windows backend binary not found: $binaryPath"
+          if (-not (Test-Path "./target/x86_64-pc-windows-msvc/release/goosed.exe")) {
+            Write-Error "Windows binary not found."
             Get-ChildItem ./target/x86_64-pc-windows-msvc/release/ -ErrorAction SilentlyContinue
             exit 1
           }
 
-          Write-Output "Windows backend binary found."
-          Get-Item $binaryPath
+          Write-Output "Windows binary found."
+          Get-Item ./target/x86_64-pc-windows-msvc/release/goosed.exe
 
       - name: Prepare Windows binary
         shell: bash
         run: |
-          BACKEND_BINARY="./target/x86_64-pc-windows-msvc/release/goose.exe"
-
-          if [ ! -f "$BACKEND_BINARY" ]; then
-            echo "Windows backend binary not found: $BACKEND_BINARY"
+          if [ ! -f "./target/x86_64-pc-windows-msvc/release/goosed.exe" ]; then
+            echo "Windows binary not found."
             exit 1
           fi
 
@@ -147,14 +142,13 @@ jobs:
           rm -rf ./ui/desktop/src/bin
           mkdir -p ./ui/desktop/src/bin
 
-          echo "Copying Windows backend binary..."
-          cp -f "$BACKEND_BINARY" ./ui/desktop/src/bin/
+          echo "Copying Windows binary..."
+          cp -f ./target/x86_64-pc-windows-msvc/release/goosed.exe ./ui/desktop/src/bin/
 
           if [ -d "./ui/desktop/src/platform/windows/bin" ]; then
             echo "Copying Windows platform files..."
             for file in ./ui/desktop/src/platform/windows/bin/*.{exe,dll,cmd}; do
-              filename="$(basename "$file")"
-              if [ -f "$file" ] && [ "$filename" != "goose.exe" ]; then
+              if [ -f "$file" ] && [ "$(basename "$file")" != "goosed.exe" ]; then
                 cp -f "$file" ./ui/desktop/src/bin/
               fi
             done
@@ -201,7 +195,7 @@ jobs:
           ls -la ./dist-windows/resources/bin/
 
       - name: Upload unsigned distribution
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: windows-unsigned${{ inputs.windows_variant == 'cuda' && '-cuda' || '' }}
           path: ui/desktop/dist-windows/
@@ -235,14 +229,14 @@ jobs:
           certificate-profile-name: ${{ secrets.AZURE_CERTIFICATE_PROFILE_NAME }}
           files: |
             ${{ github.workspace }}/dist-windows/Goose.exe
-            ${{ github.workspace }}/dist-windows/resources/bin/goose.exe
+            ${{ github.workspace }}/dist-windows/resources/bin/goosed.exe
 
       - name: Verify signed executables
         shell: pwsh
         run: |
           $files = @(
             "dist-windows/Goose.exe",
-            "dist-windows/resources/bin/goose.exe"
+            "dist-windows/resources/bin/goosed.exe"
           )
           foreach ($file in $files) {
             Write-Output "Verifying signature: $file"
@@ -261,7 +255,7 @@ jobs:
           7z a -tzip "${ZIP_NAME}.zip" dist-windows/
 
       - name: Upload signed Windows build
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-win32-x64${{ inputs.windows_variant == 'cuda' && '-cuda' || '' }}
           path: Goose-win32-x64${{ inputs.windows_variant == 'cuda' && '-cuda' || '' }}.zip
@@ -290,7 +284,7 @@ jobs:
           7z a -tzip "${ZIP_NAME}.zip" dist-windows/
 
       - name: Upload Windows build
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-win32-x64${{ inputs.windows_variant == 'cuda' && '-cuda' || '' }}
           path: Goose-win32-x64${{ inputs.windows_variant == 'cuda' && '-cuda' || '' }}.zip

.github/workflows/bundle-desktop.yml

@@ -77,7 +77,7 @@ jobs:
         run: df -h
 
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           # Only pass ref if it's explicitly set, otherwise let checkout action use its default behavior
           ref: ${{ inputs.ref != '' && inputs.ref || '' }}
@@ -118,10 +118,8 @@ jobs:
           key: macos-deployment-target-12
 
       # Build the project
-      - name: Build desktop backend
-        run: |
-          source ./bin/activate-hermit
-          cargo build --release -p goose-cli --bin goose
+      - name: Build goosed
+        run: source ./bin/activate-hermit && cargo build --release -p goose-server
 
       # Post-build cleanup to free space
       - name: Post-build cleanup
@@ -136,16 +134,12 @@ jobs:
           # Check disk space after cleanup
           df -h
 
-      - name: Copy backend binary into Electron folder
+      - name: Copy binaries into Electron folder
         run: |
-          mkdir -p ui/desktop/src/bin
-          rm -f ui/desktop/src/bin/goose
-          cp target/release/goose ui/desktop/src/bin/goose
-          chmod +x ui/desktop/src/bin/goose
-          ls -la ui/desktop/src/bin/
+          cp target/release/goosed ui/desktop/src/bin/goosed
 
       - name: Cache pnpm dependencies
-        uses: actions/cache@2c8a9bd7457de244a408f35966fab2fb45fda9c8  # v6.0.0
+        uses: actions/cache@8b402f58fbc84540c8b491a91e594a4576fec3d7  # v5.0.2
         with:
           path: |
             ui/desktop/node_modules
@@ -190,10 +184,6 @@ jobs:
           fi
         working-directory: ui/desktop
 
-      - name: Verify macOS updater resources
-        run: node scripts/verify-mac-update-resources.js "out/Goose-darwin-arm64/Goose.app"
-        working-directory: ui/desktop
-
       - name: Clean up signing keychain
         if: always()
         run: |
@@ -211,7 +201,7 @@ jobs:
 
       - name: Upload Desktop artifact
         id: upload-app-bundle
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: Goose-darwin-arm64
           path: ui/desktop/out/Goose-darwin-arm64/Goose.zip

.github/workflows/canary.yml

@@ -33,7 +33,7 @@ jobs:
       version: ${{ steps.set-version.outputs.version }}
     steps:
       # checkout code so we can read the Cargo.toml
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       - name: Generate a canary version
         id: set-version
         run: |
@@ -60,14 +60,14 @@ jobs:
     runs-on: ubuntu-latest
     needs: [build-cli]
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
-      - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+      - uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: download_cli.sh
           path: download_cli.sh
 
   # ------------------------------------------------------------
-  # 4) Bundle Desktop App (macOS only)
+  # 4) Bundle Desktop App (macOS only) - builds goosed and Electron app
   # ------------------------------------------------------------
   bundle-desktop:
     needs: [prepare-version]
@@ -80,7 +80,7 @@ jobs:
       signing: false
 
   # ------------------------------------------------------------
-  # 5) Bundle Desktop App (macOS Intel)
+  # 5) Bundle Desktop App (macOS Intel) - builds goosed and Electron app
   # ------------------------------------------------------------
   bundle-desktop-intel:
     needs: [prepare-version]
@@ -93,7 +93,7 @@ jobs:
       signing: false
 
   # ------------------------------------------------------------
-  # 6) Bundle Desktop App (Linux)
+  # 6) Bundle Desktop App (Linux) - builds goosed and Electron app
   # ------------------------------------------------------------
   bundle-desktop-linux:
     needs: [prepare-version]
@@ -102,7 +102,7 @@ jobs:
       version: ${{ needs.prepare-version.outputs.version }}
 
   # ------------------------------------------------------------
-  # 6) Bundle Desktop App (Windows)
+  # 6) Bundle Desktop App (Windows) - builds goosed and Electron app
   # ------------------------------------------------------------
   bundle-desktop-windows:
     needs: [prepare-version]
@@ -138,7 +138,7 @@ jobs:
           merge-multiple: true
 
       - name: Attest build provenance
-        uses: actions/attest-build-provenance@0f67c3f4856b2e3261c31976d6725780e5e4c373  # v4.1.1
+        uses: actions/attest-build-provenance@a2bbfa25375fe432b6a289bc6b6cd05ecd0c4c32  # v4.1.0
         with:
           subject-path: |
             goose-*.tar.bz2

.github/workflows/cargo-deny.yml

@@ -22,8 +22,8 @@ jobs:
     permissions:
       contents: read
     steps:
-      - uses: actions/checkout@v7.0.0
+      - uses: actions/checkout@v6
       # https://github.com/EmbarkStudios/cargo-deny-action v2.0.15
-      - uses: EmbarkStudios/cargo-deny-action@bb137d7af7e4fb67e5f82a49c4fce4fad40782fe
+      - uses: EmbarkStudios/cargo-deny-action@a531616d8ce3b9177443e48a1159bc945a099823
         with:
           command: check advisories

.github/workflows/cargo-machete.yml

@@ -20,5 +20,5 @@ jobs:
     permissions:
       contents: read
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v6
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6
       - uses: bnjbvr/cargo-machete@ac30a525c0a8d163a92d727b3ff079ee3f6ecb08 # v0.9.2

.github/workflows/check-release-pr.yaml

@@ -13,7 +13,7 @@ jobs:
     runs-on: ubuntu-latest
     if: startsWith(github.head_ref, 'release/')
     steps:
-    - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       with:
         ref: ${{ github.head_ref }}
         fetch-depth: 0

.github/workflows/ci.yml

@@ -20,7 +20,7 @@ jobs:
       code: ${{ steps.filter.outputs.code }}
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Check for file changes
         uses: dorny/paths-filter@fbd0ab8f3e69293af611ebaee6363fc25e6d187d # pin@v3
@@ -39,9 +39,9 @@ jobs:
     if: needs.changes.outputs.code == 'true' || github.event_name != 'pull_request'
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
-      - uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
+      - uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6 # v1.16.1
 
       - name: Run cargo fmt
         run: cargo fmt --check
@@ -53,9 +53,9 @@ jobs:
     if: needs.changes.outputs.code == 'true' || github.event_name != 'pull_request'
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
-      - uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
+      - uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6 # v1.16.1
 
       - name: Install Dependencies
         run: |
@@ -82,7 +82,7 @@ jobs:
     if: github.event_name == 'push' && github.ref == 'refs/heads/main'
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Cache Cargo artifacts
         uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4  # v2.9.1
@@ -106,7 +106,7 @@ jobs:
     needs: changes
     if: needs.changes.outputs.code == 'true' || github.event_name != 'pull_request'
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Read MSRV from Cargo.toml
         id: msrv
@@ -119,7 +119,7 @@ jobs:
           echo "msrv=$msrv" >> "$GITHUB_OUTPUT"
           echo "MSRV: $msrv"
 
-      - uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659  # v1
+      - uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6  # v1
         with:
           toolchain: ${{ steps.msrv.outputs.msrv }}
 
@@ -143,9 +143,9 @@ jobs:
     needs: changes
     if: needs.changes.outputs.code == 'true' || github.event_name != 'pull_request'
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
-      - uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
+      - uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6 # v1.16.1
 
       - uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4  # v2.9.1
 
@@ -165,9 +165,9 @@ jobs:
     if: needs.changes.outputs.code == 'true' || github.event_name != 'pull_request'
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
-      - uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
+      - uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6 # v1.16.1
 
       - name: Install Dependencies
         run: |
@@ -183,18 +183,17 @@ jobs:
           cd ui/desktop && pnpm install --frozen-lockfile
           cd ../sdk && pnpm install --frozen-lockfile
 
+      - name: Check OpenAPI Schema is Up-to-Date
+        run: |
+          source ./bin/activate-hermit
+          hermit uninstall rustup
+          just check-openapi-schema
+
       - name: Check ACP Schema is Up-to-Date
         run: |
           source ./bin/activate-hermit
           just check-acp-schema
 
-      - name: Test ACP Client SDK
-        run: |
-          source ./bin/activate-hermit
-          cd ui/sdk
-          pnpm test
-          pnpm run typecheck:test
-
   desktop-lint:
     name: Test and Lint Electron Desktop App
     runs-on: macos-latest
@@ -202,7 +201,7 @@ jobs:
     if: needs.changes.outputs.code == 'true' || github.event_name != 'pull_request'
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       # Temporarily disabled due to GitHub Actions bug on macOS runners
       # https://github.com/actions/runner-images/issues/13341

.github/workflows/close-release-pr-on-tag.yaml

@@ -15,7 +15,7 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
     - name: Extract version from tag
       id: version

.github/workflows/code-review.yml

@@ -64,7 +64,7 @@ jobs:
           OIDC_TOKEN: ${{ steps.oidc.outputs.token }}
 
       - name: Upload review inputs
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f # v6
         with:
           name: code-review-inputs
           path: /tmp/code-review/
@@ -84,7 +84,7 @@ jobs:
 
     steps:
       - name: Checkout base branch
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Install tools
         run: |
@@ -119,7 +119,7 @@ jobs:
 
       - name: Upload review output
         if: always()
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f # v6
         with:
           name: code-review-output
           path: |

.github/workflows/create-release-branch.yaml

@@ -18,13 +18,13 @@ jobs:
       github.event.pull_request.merged == true &&
       startsWith(github.event.pull_request.head.ref, 'version-bump/')
     steps:
-    - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       with:
         ref: ${{ github.event.pull_request.merge_commit_sha }}
         fetch-depth: 0
 
-    - uses: cashapp/activate-hermit@cea9af7913204a965fd488637a8d1811bba2e616 # v1
-    - uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39  # v8.2.0
+    - uses: cashapp/activate-hermit@e49f5cb4dd64ff0b0b659d1d8df499595451155a # v1
+    - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b  # v7.2.0
 
     - name: Extract version and base branch
       env:

.github/workflows/create-version-bump-pr.yaml

@@ -25,77 +25,64 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
-        with:
-          ref: main
-          fetch-depth: 0
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+      with:
+        ref: main
+        fetch-depth: 0
 
-      - uses: cashapp/activate-hermit@cea9af7913204a965fd488637a8d1811bba2e616 # v1
-      - uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39 # v8.2.0
+    - uses: cashapp/activate-hermit@e49f5cb4dd64ff0b0b659d1d8df499595451155a # v1
+    - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b  # v7.2.0
 
-      - name: install dependencies
-        run: |
-          sudo apt update -y
-          sudo apt install -y libdbus-1-dev gnome-keyring libxcb1-dev
+    - name: install dependencies
+      run: |
+        sudo apt update -y
+        sudo apt install -y libdbus-1-dev gnome-keyring libxcb1-dev
 
-      - name: Compute version and create bump branch
-        env:
-          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
-          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
-          GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
-          OPENROUTER_API_KEY: ${{ secrets.OPENROUTER_API_KEY }}
-          XAI_API_KEY: ${{ secrets.XAI_API_KEY }}
-          TETRATE_API_KEY: ${{ secrets.TETRATE_API_KEY }}
-        run: |
-          VERSION=$(just get-next-minor-version)
-          echo "version=$VERSION" >> $GITHUB_ENV
-          echo "Version: $VERSION (minor bump on main)"
+    - name: Compute version and create bump branch
+      env:
+        ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
+        OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
+        OPENROUTER_API_KEY: ${{ secrets.OPENROUTER_API_KEY }}
+        XAI_API_KEY: ${{ secrets.XAI_API_KEY }}
+        TETRATE_API_KEY: ${{ secrets.TETRATE_API_KEY }}
+      run: |
+        VERSION=$(just get-next-minor-version)
+        echo "version=$VERSION" >> $GITHUB_ENV
+        echo "Version: $VERSION (minor bump on main)"
 
-          git config --local user.email "41898282+github-actions[bot]@users.noreply.github.com"
-          git config --local user.name "github-actions[bot]"
+        git config --local user.email "41898282+github-actions[bot]@users.noreply.github.com"
+        git config --local user.name "github-actions[bot]"
 
-          BRANCH_NAME="version-bump/$VERSION"
-          git switch -C "$BRANCH_NAME"
+        BRANCH_NAME="version-bump/$VERSION"
+        git switch -c "$BRANCH_NAME"
 
-          just prepare-release $VERSION
+        just prepare-release $VERSION
 
-          echo "branch_name=$BRANCH_NAME" >> $GITHUB_ENV
+        echo "branch_name=$BRANCH_NAME" >> $GITHUB_ENV
 
-      - name: Push bump branch
-        run: |
-          git push --force-with-lease origin "${{ env.branch_name }}"
+    - name: Push bump branch
+      run: |
+        git push origin "${{ env.branch_name }}"
 
-      - name: Create Pull Request if needed
-        run: |
-          EXISTING_PR_URL=$(gh pr list \
-            --base main \
-            --head "${{ env.branch_name }}" \
-            --state open \
-            --json url \
-            --jq '.[0].url // empty')
+    - name: Create Pull Request
+      run: |
+        PR_BODY=$(cat <<'EOF'
+        Bumps version to **${{ env.version }}**.
 
-          if [ -n "$EXISTING_PR_URL" ]; then
-            echo "Existing PR found: $EXISTING_PR_URL"
-            echo "pr_url=$EXISTING_PR_URL" >> $GITHUB_ENV
-            exit 0
-          fi
+        **Please follow these steps:**
 
-          PR_BODY=$(cat <<'EOF'
-          Bumps version to **${{ env.version }}**.
-
-          **Please follow these steps:**
-
-          1. Approve workflows for this PR to trigger CI checks.
-          2. Review and resolve any merge conflicts.
-          3. Approve and merge this PR.
-          4. The `release/${{ env.version }}` PR will be created automatically.
-          EOF
-          )
-          PR_URL=$(gh pr create \
-            -B main \
-            -H "${{ env.branch_name }}" \
-            --title "chore(release): bump version to ${{ env.version }} (minor)" \
-            --body "$PR_BODY")
-          echo "pr_url=$PR_URL" >> $GITHUB_ENV
-        env:
-          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        1. Close and reopen this PR to trigger CI checks. Reason: workflows don't run on PRs opened by `GITHUB_TOKEN` ([docs](https://docs.github.com/en/actions/using-workflows/triggering-a-workflow#triggering-a-workflow-from-a-workflow)).
+        2. Review and resolve any merge conflicts.
+        3. Approve and merge this PR.
+        4. The `release/${{ env.version }}` PR will be created automatically.
+        EOF
+        )
+        PR_URL=$(gh pr create \
+          -B main \
+          -H "${{ env.branch_name }}" \
+          --title "chore(release): bump version to ${{ env.version }} (minor)" \
+          --body "$PR_BODY")
+        echo "pr_url=$PR_URL" >> $GITHUB_ENV
+      env:
+        GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/deploy-docs-and-extensions.yml

@@ -24,15 +24,15 @@ jobs:
 
     steps:
       - name: Checkout the branch
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Setup Node.js
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e  # v6.4.0
+        uses: actions/setup-node@6044e13b5dc448c55e2357c09f80417699197238  # v6.2.0
         with:
           node-version: 20
 
       - name: Cache Node.js modules (documentation)
-        uses: actions/cache@2c8a9bd7457de244a408f35966fab2fb45fda9c8  # v6.0.0
+        uses: actions/cache@8b402f58fbc84540c8b491a91e594a4576fec3d7  # v5.0.2
         with:
           path: ./documentation/node_modules
           key: ${{ runner.os }}-documentation-${{ hashFiles('./documentation/package-lock.json') }}
@@ -58,10 +58,10 @@ jobs:
         uses: actions/configure-pages@45bfe0192ca1faeb007ade9deae92b16b8254a0d  # v6.0.0
 
       - name: Upload GitHub Pages artifact
-        uses: actions/upload-pages-artifact@fc324d3547104276b827a68afc52ff2a11cc49c9  # v5.0.0
+        uses: actions/upload-pages-artifact@7b1f4a764d45c48632c6b24a0339c27f5614fb0b  # v4.0.0
         with:
           path: documentation/build
 
       - name: Deploy to GitHub Pages
         id: deployment
-        uses: actions/deploy-pages@cd2ce8fcbc39b97be8ca5fce6e763baed58fa128  # v5.0.0
+        uses: actions/deploy-pages@d6db90164ac5ed86f2b6aed7e0febac5b3c0c03e  # v4.0.5

.github/workflows/docs-update-cli-ref.yml

@@ -43,7 +43,7 @@ jobs:
     
     steps:
       - name: Checkout repository
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
         with:
           fetch-depth: 0  # Fetch all history for version comparison
           fetch-tags: true  # Fetch all tags so we can checkout version tags
@@ -63,7 +63,7 @@ jobs:
           sudo apt-get install -y jq ripgrep
       
       - name: Set up Rust
-        uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
+        uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6 # v1.16.1
         with:
           toolchain: stable
       
@@ -73,7 +73,7 @@ jobs:
           python-version: '3.11'
       
       - name: Set up Node.js
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e # v6.4.0
+        uses: actions/setup-node@49933ea5288caeca8642d1e84afbd3f7d6820020 # v4
         with:
           node-version: '20'
       
@@ -191,7 +191,7 @@ jobs:
       
       - name: Upload automation outputs
         if: always()
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1
+        uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4
         with:
           name: cli-docs-update-${{ steps.versions.outputs.old_version }}-to-${{ steps.versions.outputs.new_version }}
           path: |

.github/workflows/docs-update-recipe-ref.yml

@@ -0,0 +1,292 @@
+# Automatically updates the Recipe Reference Guide when recipe struct fields,
+# validation rules, or schema constraints change between releases.
+#
+# Triggers: Manual (for testing) or on release (production)
+# Testing: Use dry_run mode to review outputs without creating PRs
+# See: documentation/automation/recipe-schema-tracking/TESTING.md
+
+name: Update Recipe Documentation
+
+on:
+  workflow_dispatch:  # Manual trigger for testing
+    inputs:
+      old_version:
+        description: 'Previous version (e.g., v1.14.0). Leave empty to auto-detect.'
+        required: false
+        type: string
+      new_version:
+        description: 'New version (e.g., v1.15.0). Leave empty to use HEAD.'
+        required: false
+        type: string
+      dry_run:
+        description: 'Dry run mode - generate files but do not create PR'
+        required: false
+        type: boolean
+        default: true
+      
+  release:
+    types: [published]
+
+permissions:
+  contents: write       # Create branches and commit files
+  pull-requests: write  # Create PRs
+
+jobs:
+  update-docs:
+    name: Update Recipe Documentation
+    runs-on: ubuntu-latest
+    
+    env:
+      OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+    
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          fetch-depth: 0  # Fetch all history for version comparison
+          fetch-tags: true  # Fetch all tags so we can checkout version tags
+      
+      - name: Fetch upstream tags (for forks)
+        if: github.repository != 'aaif-goose/goose'
+        run: |
+          # Add upstream remote and fetch tags (only needed when testing in forks)
+          git remote add upstream https://github.com/aaif-goose/goose.git || git remote set-url upstream https://github.com/aaif-goose/goose.git
+          git fetch upstream --tags --force
+          echo "✅ Fetched tags from upstream (fork mode)"
+          echo "Total tags available: $(git tag | wc -l)"
+      
+      - name: Install system dependencies
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y jq ripgrep
+      
+      - name: Set up Node.js
+        uses: actions/setup-node@6044e13b5dc448c55e2357c09f80417699197238  # v6.2.0
+        with:
+          node-version: '20'
+      
+      - name: Install goose CLI
+        run: |
+          mkdir -p /home/runner/.local/bin
+          curl -fsSL https://github.com/aaif-goose/goose/releases/download/stable/download_cli.sh \
+            | CONFIGURE=false GOOSE_BIN_DIR=/home/runner/.local/bin bash
+          echo "/home/runner/.local/bin" >> $GITHUB_PATH
+          goose --version
+      
+      - name: Configure goose for CI
+        env:
+          GOOSE_PROVIDER: ${{ vars.GOOSE_PROVIDER || 'openai' }}
+          GOOSE_MODEL: ${{ vars.GOOSE_MODEL || 'gpt-4o' }}
+        run: |
+          mkdir -p ~/.config/goose
+          cat <<EOF > ~/.config/goose/config.yaml
+          GOOSE_PROVIDER: $GOOSE_PROVIDER
+          GOOSE_MODEL: $GOOSE_MODEL
+          keyring: false
+          EOF
+          echo "✅ Created goose config:"
+          cat ~/.config/goose/config.yaml
+      
+      - name: Determine versions to compare
+        id: versions
+        env:
+          GH_TOKEN: ${{ github.token }}
+          INPUT_OLD_VERSION: ${{ github.event.inputs.old_version }}
+          INPUT_NEW_VERSION: ${{ github.event.inputs.new_version }}
+          EVENT_NAME: ${{ github.event_name }}
+          RELEASE_TAG: ${{ github.event.release.tag_name }}
+        run: |
+          get_previous_release() {
+            gh release list --limit 2 --json tagName --jq '.[].tagName' | sed -n '2p'
+          }
+          
+          if [ -n "$INPUT_OLD_VERSION" ]; then
+            OLD_VERSION="$INPUT_OLD_VERSION"
+          else
+            OLD_VERSION=$(get_previous_release)
+          fi
+          
+          if [ -n "$INPUT_NEW_VERSION" ]; then
+            NEW_VERSION="$INPUT_NEW_VERSION"
+          elif [ "$EVENT_NAME" = "release" ]; then
+            NEW_VERSION="$RELEASE_TAG"
+          else
+            NEW_VERSION="HEAD"  # For testing unreleased changes
+          fi
+          
+          if [ -z "$OLD_VERSION" ] || [ -z "$NEW_VERSION" ]; then
+            echo "Error: Could not determine versions to compare"
+            exit 1
+          fi
+          
+          echo "old_version=$OLD_VERSION" >> $GITHUB_OUTPUT
+          echo "new_version=$NEW_VERSION" >> $GITHUB_OUTPUT
+          echo "OLD_VERSION=$OLD_VERSION" >> $GITHUB_ENV
+          echo "NEW_VERSION=$NEW_VERSION" >> $GITHUB_ENV
+          
+          echo "✅ Comparing $OLD_VERSION → $NEW_VERSION"
+      
+      - name: Extract and compare schemas
+        id: extract
+        timeout-minutes: 15
+        working-directory: documentation/automation/recipe-schema-tracking
+        env:
+          GOOSE_REPO: ${{ github.workspace }}
+        run: |
+          set -o pipefail  # Ensure pipeline failures are caught
+          
+          mkdir -p output
+          ./scripts/run-pipeline.sh "$OLD_VERSION" "$NEW_VERSION" 2>&1 | tee output/pipeline.log
+          
+          HAS_CHANGES=$(jq -r '.has_changes' output/validation-changes.json)
+          echo "has_changes=$HAS_CHANGES" >> $GITHUB_OUTPUT
+          
+          if [ "$HAS_CHANGES" = "false" ]; then
+            echo "✅ No changes detected"
+          else
+            echo "✅ Changes detected"
+          fi
+      
+      - name: Update recipe-reference.md (AI synthesis)
+        if: steps.extract.outputs.has_changes == 'true'
+        timeout-minutes: 10
+        working-directory: documentation/automation/recipe-schema-tracking/output
+        env:
+          RECIPE_REF_PATH: ${{ github.workspace }}/documentation/docs/guides/recipes/recipe-reference.md
+        run: |
+          echo "🔍 Environment diagnostics:"
+          echo "  GOOSE_PROVIDER: $GOOSE_PROVIDER"
+          echo "  GOOSE_MODEL: $GOOSE_MODEL"
+          echo "  OPENAI_API_KEY: ${OPENAI_API_KEY:0:8}..." # Show first 8 chars only
+          echo "  RECIPE_REF_PATH: $RECIPE_REF_PATH"
+          echo "  HOME: $HOME"
+          echo "  PATH: $PATH"
+          echo ""
+          echo "📁 Goose config file:"
+          cat ~/.config/goose/config.yaml || echo "Config file not found!"
+          echo ""
+          echo "📁 Current directory:"
+          pwd
+          ls -la
+          echo ""
+          echo "🤖 Step 1: Running validation changes synthesis..."
+          goose run --recipe ../recipes/synthesize-validation-changes.yaml
+          
+          echo ""
+          echo "🤖 Step 2: Applying changes to recipe-reference.md..."
+          goose run --recipe ../recipes/update-recipe-reference.yaml
+      
+      - name: Upload automation outputs
+        if: always()
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
+        with:
+          name: recipe-docs-update-${{ steps.versions.outputs.old_version }}-to-${{ steps.versions.outputs.new_version }}
+          path: |
+            documentation/automation/recipe-schema-tracking/output/*.json
+            documentation/automation/recipe-schema-tracking/output/*.md
+            documentation/automation/recipe-schema-tracking/output/*.log
+          retention-days: 30
+      
+      - name: Create Pull Request
+        if: |
+          steps.extract.outputs.has_changes == 'true' && 
+          (github.event.inputs.dry_run != 'true' || github.event_name == 'release')
+        uses: peter-evans/create-pull-request@5f6978faf089d4d20b00c7766989d076bb2fc7f1  # v8.1.1
+        with:
+          branch: docs/auto-recipe-reference-${{ steps.versions.outputs.new_version }}
+          delete-branch: true
+          
+          commit-message: |
+            docs: Update recipe reference for ${{ steps.versions.outputs.new_version }}
+            
+            Automated update based on schema changes between ${{ steps.versions.outputs.old_version }} and ${{ steps.versions.outputs.new_version }}.
+          
+          title: "docs: Update Recipe Reference Guide for ${{ steps.versions.outputs.new_version }}"
+          body: |
+            ## Summary
+
+            This PR updates the Recipe Reference Guide based on schema and validation changes detected between **${{ steps.versions.outputs.old_version }}** and **${{ steps.versions.outputs.new_version }}**.
+
+            ### Type of Change
+            - [x] Documentation
+
+            ### AI Assistance
+            - [x] This PR was created or reviewed with AI assistance
+
+            #### 🤖 Automation Details
+            
+            - **Workflow**: `docs-update-recipe-ref.yml`
+            - **Triggered by**: ${{ github.event_name }}
+            - **Previous version**: ${{ steps.versions.outputs.old_version }}
+            - **New version**: ${{ steps.versions.outputs.new_version }}
+            
+            #### 📋 Changes Detected
+            
+            Review the workflow artifacts for detailed change analysis:
+            - `validation-changes.json` - Structured diff of changes
+            - `validation-changes.md` - Human-readable change documentation
+            - `update-summary.md` - Summary of documentation updates applied
+            
+            Download artifacts from the [workflow run](${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }}).
+            
+            ### ✅ Review Checklist
+            
+            - [ ] Verify all schema changes are accurately documented
+            - [ ] Check that examples are updated correctly
+            - [ ] Ensure validation rules are clearly explained
+            - [ ] Confirm no unintended changes were made
+            - [ ] Do changes require additional updates in this or other recipe topics?
+            
+            ### 🔗 Related
+            
+            - Release: ${{ github.event.release.html_url || 'N/A' }}
+            
+            ---
+            
+            *This PR was automatically generated by the Recipe Documentation Automation workflow.*
+          
+          labels: |
+            documentation
+            automated
+            recipe-reference
+      
+      - name: Workflow summary
+        if: always()
+        env:
+          OLD_VERSION: ${{ steps.versions.outputs.old_version }}
+          NEW_VERSION: ${{ steps.versions.outputs.new_version }}
+          HAS_CHANGES: ${{ steps.extract.outputs.has_changes }}
+          DRY_RUN: ${{ github.event.inputs.dry_run || 'false' }}
+        run: |
+          echo "## 📊 Recipe Documentation Update Summary" >> $GITHUB_STEP_SUMMARY
+          echo "" >> $GITHUB_STEP_SUMMARY
+          echo "**Version Comparison**: $OLD_VERSION → $NEW_VERSION" >> $GITHUB_STEP_SUMMARY
+          echo "**Changes Detected**: $HAS_CHANGES" >> $GITHUB_STEP_SUMMARY
+          echo "**Dry Run Mode**: $DRY_RUN" >> $GITHUB_STEP_SUMMARY
+          echo "" >> $GITHUB_STEP_SUMMARY
+          
+          if [ "$HAS_CHANGES" = "true" ]; then
+            echo "### ✅ Documentation Updated" >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+            echo "The Recipe Reference Guide has been updated to reflect changes in $NEW_VERSION." >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+            
+            if [ "$DRY_RUN" = "true" ]; then
+              echo "**Note**: Running in dry-run mode - no PR was created. Review the artifacts to see the generated changes." >> $GITHUB_STEP_SUMMARY
+            else
+              echo "A pull request has been created with the documentation updates." >> $GITHUB_STEP_SUMMARY
+            fi
+          else
+            echo "### ℹ️ No Changes Needed" >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+            echo "No recipe schema or validation changes were detected between $OLD_VERSION and $NEW_VERSION." >> $GITHUB_STEP_SUMMARY
+          fi
+          
+          echo "" >> $GITHUB_STEP_SUMMARY
+          echo "### 📦 Artifacts" >> $GITHUB_STEP_SUMMARY
+          echo "" >> $GITHUB_STEP_SUMMARY
+          echo "Download the workflow artifacts to review:" >> $GITHUB_STEP_SUMMARY
+          echo "- Extracted schemas and validation structures" >> $GITHUB_STEP_SUMMARY
+          echo "- Change detection results" >> $GITHUB_STEP_SUMMARY
+          echo "- Human-readable change documentation" >> $GITHUB_STEP_SUMMARY
+          echo "- Documentation update summary" >> $GITHUB_STEP_SUMMARY

.github/workflows/goose-issue-solver.yml

@@ -30,7 +30,7 @@ env:
       - Deletion is a feature: ask "can I delete code instead of adding?"
       - Complete changes: update ALL usages when changing a type. No exceptions.
       - When changing message types, update ALL provider format functions.
-      - When changing ACP request/response types, run `just generate-acp-types`.
+      - When changing server routes, run `just generate-openapi`.
       - Your context degrades. The TODO is your memory. Update it after each step.
 
       Types:
@@ -150,7 +150,7 @@ jobs:
             -d '{"content":"eyes"}'
 
       - name: Checkout
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: main
           fetch-depth: 1

.github/workflows/goose-pr-reviewer.yml

@@ -241,7 +241,7 @@ jobs:
             -d '{"content":"eyes"}'
 
       - name: Checkout PR
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: refs/pull/${{ github.event.issue.number }}/head
           fetch-depth: 1

.github/workflows/goose-release-notes.yml

@@ -15,8 +15,6 @@ on:
       - Release
     types:
       - completed
-    branches-ignore:
-      - "release/**"
 
   # Allow manual trigger for testing
   workflow_dispatch:
@@ -78,7 +76,7 @@ jobs:
   generate-release-notes:
     name: Generate Release Notes
     runs-on: ubuntu-latest
-    # For workflow_run: release branch preflight builds are ignored by the trigger.
+    # For workflow_run: only run if Release succeeded and tag is not 'stable'
     # For workflow_dispatch: only run if tag is not 'stable'
     if: |
       (github.event_name == 'workflow_dispatch' && inputs.tag != 'stable') ||
@@ -115,7 +113,7 @@ jobs:
           fi
 
       - name: Checkout repository
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           fetch-depth: 0
           fetch-tags: true

.github/workflows/minor-release.yaml

@@ -6,61 +6,12 @@ permissions:
 
 on:
   schedule:
-    - cron: "0 0 * * 2"
+    - cron: '0 0 * * 2'
   workflow_dispatch:
-  push:
-    branches:
-      - main
 
 jobs:
-  check-version-bump-pr:
-    if: github.repository == 'aaif-goose/goose' && github.event_name == 'push'
-    runs-on: ubuntu-latest
-    outputs:
-      should_run: ${{ steps.check.outputs.should_run }}
-    steps:
-      - name: Check for conflicted version bump PR
-        id: check
-        env:
-          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-        run: |
-          for attempt in 1 2 3 4 5 6; do
-            CONFLICTING_PR_URLS=$(gh pr list \
-              --repo "$GITHUB_REPOSITORY" \
-              --base main \
-              --state open \
-              --search 'head:version-bump/' \
-              --json mergeStateStatus,url \
-              --jq '.[] | select(.mergeStateStatus == "DIRTY") | .url')
-
-            UNKNOWN_COUNT=$(gh pr list \
-              --repo "$GITHUB_REPOSITORY" \
-              --base main \
-              --state open \
-              --search 'head:version-bump/' \
-              --json mergeStateStatus \
-              --jq '[.[] | select(.mergeStateStatus == "UNKNOWN")] | length')
-
-            if [ -n "$CONFLICTING_PR_URLS" ] || [ "$UNKNOWN_COUNT" = "0" ]; then
-              break
-            fi
-
-            echo "Waiting for GitHub to compute PR mergeability (attempt $attempt)..."
-            sleep 10
-          done
-
-          if [ -n "$CONFLICTING_PR_URLS" ]; then
-            echo "Found conflicted version bump PR(s):"
-            echo "$CONFLICTING_PR_URLS"
-            echo "should_run=true" >> "$GITHUB_OUTPUT"
-          else
-            echo "No conflicted version bump PR found."
-            echo "should_run=false" >> "$GITHUB_OUTPUT"
-          fi
-
   release:
-    needs: check-version-bump-pr
-    if: always() && github.repository == 'aaif-goose/goose' && (github.event_name != 'push' || needs.check-version-bump-pr.outputs.should_run == 'true')
+    if: github.repository == 'aaif-goose/goose'
     uses: ./.github/workflows/create-version-bump-pr.yaml
     secrets:
       ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}

.github/workflows/patch-release.yaml

@@ -19,13 +19,13 @@ jobs:
       TARGET_BRANCH: ${{ inputs.target_branch }}
 
     steps:
-    - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       with:
         ref: ${{ inputs.target_branch }}
         fetch-depth: 0
 
-    - uses: cashapp/activate-hermit@cea9af7913204a965fd488637a8d1811bba2e616 # v1
-    - uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39  # v8.2.0
+    - uses: cashapp/activate-hermit@e49f5cb4dd64ff0b0b659d1d8df499595451155a # v1
+    - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b  # v7.2.0
 
     - name: install dependencies
       run: |

.github/workflows/pr-comment-build-cli.yml

@@ -92,7 +92,7 @@ jobs:
 
       - name: Checkout code
         if: steps.security_check.outputs.authorized == 'true'
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         
       - name: Get PR head SHA with gh
         id: set_head_sha

.github/workflows/pr-comment-bundle-intel.yml

@@ -45,7 +45,7 @@ jobs:
           allowed_contexts: pull_request
 
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         
       - name: Get PR head SHA with gh
         id: set_head_sha

.github/workflows/pr-comment-bundle-windows.yml

@@ -48,7 +48,7 @@ jobs:
           allowed_contexts: pull_request
 
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         
       - name: Get PR head SHA with gh
         id: set_head_sha

.github/workflows/pr-smoke-test.yml

@@ -30,7 +30,7 @@ jobs:
       code: ${{ steps.filter.outputs.code }}
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.inputs.branch || github.ref }}
           fetch-depth: 0
@@ -51,11 +51,11 @@ jobs:
     if: needs.changes.outputs.code == 'true' || github.event_name == 'workflow_dispatch'
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.inputs.branch || github.ref }}
 
-      - uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
+      - uses: actions-rust-lang/setup-rust-toolchain@46268bd060767258de96ed93c1251119784f2ab6 # v1.16.1
 
       - name: Install Dependencies
         run: |
@@ -67,22 +67,29 @@ jobs:
 
       - name: Build Binary for Smoke Tests
         run: |
-          cargo build --bin goose
+          cargo build --bin goose --bin goosed
 
       - name: Upload goose binary
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: goose-binary
           path: target/debug/goose
           retention-days: 1
 
+      - name: Upload goosed binary
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
+        with:
+          name: goosed-binary
+          path: target/debug/goosed
+          retention-days: 1
+
   smoke-tests:
     name: Smoke Tests
     runs-on: ubuntu-latest
     needs: build-binary
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.inputs.branch || github.ref }}
 
@@ -96,7 +103,7 @@ jobs:
         run: chmod +x target/debug/goose
 
       - name: Set up Node.js
-        uses: actions/setup-node@v6
+        uses: actions/setup-node@v4
         with:
           node-version: '22'
 
@@ -135,7 +142,7 @@ jobs:
           python-version: '3.12'
 
       - name: Install uv
-        uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39  # v8.2.0
+        uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b  # v7.2.0
 
       - name: Run MCP Tests
         env:
@@ -170,7 +177,7 @@ jobs:
     needs: build-binary
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.inputs.branch || github.ref }}
 
@@ -213,7 +220,7 @@ jobs:
     needs: build-binary
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.inputs.branch || github.ref }}
 
@@ -232,7 +239,7 @@ jobs:
           python-version: '3.12'
 
       - name: Install uv
-        uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39  # v8.2.0
+        uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b  # v7.2.0
 
       - name: Run Compaction Tests
         env:
@@ -246,3 +253,39 @@ jobs:
           mkdir -p $HOME/.local/share/goose/sessions
           mkdir -p $HOME/.config/goose
           bash scripts/test_compaction.sh
+
+  goosed-integration-tests:
+    name: goose server HTTP integration tests
+    runs-on: ubuntu-latest
+    needs: build-binary
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          ref: ${{ github.event.inputs.branch || github.ref }}
+
+      - name: Download Binary
+        uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c  # v8.0.1
+        with:
+          name: goosed-binary
+          path: target/debug
+
+      - name: Make Binary Executable
+        run: chmod +x target/debug/goosed
+
+      - name: Install Node.js Dependencies
+        run: source ../../bin/activate-hermit && pnpm install --frozen-lockfile
+        working-directory: ui/desktop
+
+      - name: Run Integration Tests
+        env:
+          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
+          GOOSED_BINARY: ../../target/debug/goosed
+          GOOSE_PROVIDER: anthropic
+          GOOSE_MODEL: claude-sonnet-4-5-20250929
+          SHELL: /bin/bash
+          SKIP_BUILD: 1
+        run: |
+            echo 'export PATH=/some/fake/path:$PATH' >> $HOME/.bash_profile
+            source ../../bin/activate-hermit && pnpm run test:integration:goosed
+        working-directory: ui/desktop

.github/workflows/pr-website-preview.yml

@@ -20,16 +20,12 @@ jobs:
     permissions:
       contents: read
       pull-requests: write
-    env:
-      CLOUDFLARE_ACCOUNT_ID: ${{ secrets.PAGES_PR_PREVIEW_CF_ACCOUNT_ID }}
-      CLOUDFLARE_API_TOKEN: ${{ secrets.PAGES_PR_PREVIEW_CF_API_TOKEN }}
-      CLOUDFLARE_PAGES_PROJECT_NAME: ${{ secrets.PAGES_PR_PREVIEW_CF_PAGES_PROJECT_NAME }}
     steps:
       - name: Checkout the branch
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Setup Node.js for docs build
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e  # v6.4.0
+        uses: actions/setup-node@6044e13b5dc448c55e2357c09f80417699197238  # v6.2.0
         with:
           node-version: 20
           cache: npm
@@ -51,16 +47,17 @@ jobs:
         run: ./scripts/verify-build.sh
 
       - name: Setup Node.js for Wrangler
-        if: env.CLOUDFLARE_API_TOKEN != ''
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e  # v6.4.0
+        uses: actions/setup-node@6044e13b5dc448c55e2357c09f80417699197238  # v6.2.0
         with:
           node-version: 22
 
       - name: Deploy preview to Cloudflare Pages
-        if: env.CLOUDFLARE_API_TOKEN != ''
         id: cloudflare-pages
         working-directory: ./documentation
         env:
+          CLOUDFLARE_ACCOUNT_ID: ${{ secrets.PAGES_PR_PREVIEW_CF_ACCOUNT_ID }}
+          CLOUDFLARE_API_TOKEN: ${{ secrets.PAGES_PR_PREVIEW_CF_API_TOKEN }}
+          CLOUDFLARE_PAGES_PROJECT_NAME: ${{ secrets.PAGES_PR_PREVIEW_CF_PAGES_PROJECT_NAME }}
           PR_NUMBER: ${{ github.event.number }}
         run: |
           set -euo pipefail
@@ -79,7 +76,6 @@ jobs:
           echo "preview-url=$preview_url" >> "$GITHUB_OUTPUT"
 
       - name: Comment preview URL
-        if: env.CLOUDFLARE_API_TOKEN != ''
         uses: actions/github-script@3a2844b7e9c422d3c10d287c895573f7108da1b3  # v9.0.0
         env:
           PREVIEW_URL: ${{ steps.cloudflare-pages.outputs.preview-url }}

.github/workflows/publish-ask-ai-bot.yml

@@ -19,13 +19,13 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
 
       - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@bb05f3f5519dd87d3ba754cc423b652a5edd6d2c # v4.2.0
+        uses: docker/setup-buildx-action@8d2750c68a42422c14e847fe6c8ac0403b4cbd6f # v3.12.0
 
       - name: Log in to GitHub Container Registry
-        uses: docker/login-action@650006c6eb7dba73a995cc03b0b2d7f5ca915bee # v4.2.0
+        uses: docker/login-action@4907a6ddec9925e35a0a9e82d7399ccc52663121 # v4.1.0
         with:
           registry: ghcr.io
           username: ${{ github.actor }}
@@ -33,7 +33,7 @@ jobs:
 
       - name: Extract metadata
         id: meta
-        uses: docker/metadata-action@dc802804100637a589fabce1cb79ff13a1411302 # v6.2.0
+        uses: docker/metadata-action@030e881283bb7a6894de51c315a6bfe6a94e05cf # v6.0.0
         with:
           images: ghcr.io/${{ github.repository_owner }}/ask-ai-bot
           tags: |
@@ -42,7 +42,7 @@ jobs:
             type=raw,value=latest,enable={{is_default_branch}}
 
       - name: Build and push Docker image
-        uses: docker/build-push-action@53b7df96c91f9c12dcc8a07bcb9ccacbed38856a # pin@v7.3.0
+        uses: docker/build-push-action@f9f3042f7e2789586610d6e8b85c8f03e5195baf # pin@v7.2.0
         with:
           context: .
           file: services/ask-ai-bot/Dockerfile

.github/workflows/publish-docker.yml

@@ -23,13 +23,13 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       
       - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@bb05f3f5519dd87d3ba754cc423b652a5edd6d2c  # v4.2.0
+        uses: docker/setup-buildx-action@8d2750c68a42422c14e847fe6c8ac0403b4cbd6f  # v3.12.0
       
       - name: Log in to GitHub Container Registry
-        uses: docker/login-action@650006c6eb7dba73a995cc03b0b2d7f5ca915bee  # v4.2.0
+        uses: docker/login-action@4907a6ddec9925e35a0a9e82d7399ccc52663121  # v4.1.0
         with:
           registry: ghcr.io
           username: ${{ github.actor }}
@@ -37,7 +37,7 @@ jobs:
       
       - name: Extract metadata
         id: meta
-        uses: docker/metadata-action@dc802804100637a589fabce1cb79ff13a1411302  # v6.2.0
+        uses: docker/metadata-action@030e881283bb7a6894de51c315a6bfe6a94e05cf  # v6.0.0
         with:
           images: ghcr.io/${{ github.repository_owner }}/goose
           tags: |
@@ -54,7 +54,7 @@ jobs:
       
       - name: Build and push Docker image
         id: docker-push
-        uses: docker/build-push-action@53b7df96c91f9c12dcc8a07bcb9ccacbed38856a # pin@v7.3.0
+        uses: docker/build-push-action@f9f3042f7e2789586610d6e8b85c8f03e5195baf # pin@v7.2.0
         with:
           context: .
           push: true
@@ -65,7 +65,7 @@ jobs:
           platforms: linux/amd64,linux/arm64
 
       - name: Attest Docker image
-        uses: actions/attest-build-provenance@0f67c3f4856b2e3261c31976d6725780e5e4c373  # v4.1.1
+        uses: actions/attest-build-provenance@a2bbfa25375fe432b6a289bc6b6cd05ecd0c4c32  # v4.1.0
         with:
           subject-name: ghcr.io/${{ github.repository_owner }}/goose
           subject-digest: ${{ steps.docker-push.outputs.digest }}

.github/workflows/publish-npm.yml

@@ -22,17 +22,17 @@ jobs:
     needs: [build-cli]
     steps:
       - name: Checkout
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
 
       - name: Setup Node.js
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e # v6.4.0
+        uses: actions/setup-node@49933ea5288caeca8642d1e84afbd3f7d6820020 # v4
         with:
           node-version: '24.10.0'
           registry-url: 'https://registry.npmjs.org'
           always-auth: true
 
       - name: Setup pnpm
-        uses: pnpm/action-setup@0ebf47130e4866e96fce0953f49152a61190b271 # v6.0.9
+        uses: pnpm/action-setup@8912a9102ac27614460f54aedde9e1e7f9aec20d # v6.0.5
         with:
           version: 10.30.3
 
@@ -100,7 +100,10 @@ jobs:
           cd ui/sdk
           pnpm run build:ts
 
-      - name: Compatibility check (SDK client ↔ goose binary)
+          cd ../text
+          pnpm run build
+
+      - name: Compatibility check (TUI client ↔ goose binary)
         env:
           GOOSE_BINARY: ${{ github.workspace }}/ui/goose-binary/goose-binary-linux-x64/bin/goose
         run: |
@@ -108,7 +111,7 @@ jobs:
           pnpm run check:compat
 
       - name: Upload built packages
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1
+        uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4
         with:
           name: npm-packages
           path: ui/
@@ -132,7 +135,7 @@ jobs:
             echo ""
             echo "### npm Packages"
             cd ui
-            for pkg in sdk goose-binary/*/; do
+            for pkg in sdk text goose-binary/*/; do
               if [ -f "$pkg/package.json" ]; then
                 name=$(jq -r '.name' "$pkg/package.json")
                 version=$(jq -r '.version' "$pkg/package.json")
@@ -157,14 +160,14 @@ jobs:
         run: find ui/goose-binary -name 'goose' -type f -exec chmod +x {} +
 
       - name: Setup Node.js
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e # v6.4.0
+        uses: actions/setup-node@49933ea5288caeca8642d1e84afbd3f7d6820020 # v4
         with:
           node-version: '24.10.0'
           registry-url: 'https://registry.npmjs.org'
           always-auth: true
 
       - name: Setup pnpm
-        uses: pnpm/action-setup@0ebf47130e4866e96fce0953f49152a61190b271 # v6.0.9
+        uses: pnpm/action-setup@8912a9102ac27614460f54aedde9e1e7f9aec20d # v6.0.5
         with:
           version: 10.30.3
 
@@ -174,7 +177,7 @@ jobs:
           # Publish each package individually so one failure doesn't abort the rest.
           # Skips packages whose version is already published (409/403).
           failed=0
-          for pkg in sdk goose-binary/*/; do
+          for pkg in sdk goose-binary/*/ text; do
             if [ -f "$pkg/package.json" ]; then
               name=$(jq -r '.name' "$pkg/package.json")
               version=$(jq -r '.version' "$pkg/package.json")
@@ -204,7 +207,7 @@ jobs:
           {
             echo "## 🚀 Published Packages"
             echo ""
-            for pkg in sdk goose-binary/*/; do
+            for pkg in sdk text goose-binary/*/; do
               if [ -f "$pkg/package.json" ]; then
                 name=$(jq -r '.name' "$pkg/package.json")
                 version=$(jq -r '.version' "$pkg/package.json")

.github/workflows/python-sdk-wheels.yml

@@ -1,118 +0,0 @@
-name: Python SDK Wheels
-
-on:
-  workflow_dispatch:
-    inputs:
-      publish:
-        description: "Publish wheels to PyPI after building"
-        required: true
-        default: false
-        type: boolean
-
-permissions:
-  contents: read
-  id-token: write
-
-jobs:
-  build-wheels:
-    name: Build wheel (${{ matrix.name }})
-    runs-on: ${{ matrix.os }}
-    container: ${{ matrix.container || null }}
-    strategy:
-      fail-fast: false
-      matrix:
-        include:
-          - name: macos-arm64
-            os: macos-14
-          - name: macos-x86_64
-            os: macos-13
-          - name: linux-x86_64
-            os: ubuntu-latest
-            container: quay.io/pypa/manylinux_2_28_x86_64@sha256:441c35fdc6ee809ff9260894f8468ab4fea8c15dc880f8700a3f81b7922c1cda
-          - name: windows-x86_64
-            os: windows-latest
-
-    steps:
-      - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
-
-      - name: Set up Rust
-        uses: actions-rust-lang/setup-rust-toolchain@166cdcfd11aee3cb47222f9ddb555ce30ddb9659 # v1.17.0
-
-      - name: Set up Python
-        uses: actions/setup-python@ece7cb06caefa5fff74198d8649806c4678c61a1 # v6
-        with:
-          python-version: "3.12"
-
-      - name: Install uv
-        uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39 # v8.2.0
-        with:
-          enable-cache: true
-
-      - name: Install Linux tools
-        if: runner.os == 'Linux'
-        shell: bash
-        run: |
-          python3 -m pip install --upgrade pip
-          python3 -m pip install uv
-          echo "$HOME/.local/bin" >> "$GITHUB_PATH"
-          curl --proto '=https' --tlsv1.2 -sSf https://just.systems/install.sh | bash -s -- --to /usr/local/bin
-
-      - name: Install just
-        if: runner.os != 'Linux'
-        uses: extractions/setup-just@53165ef7e734c5c07cb06b3c8e7b647c5aa16db3 # v4
-
-      - name: Cache Cargo artifacts
-        uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4 # v2.9.1
-
-      - name: Build wheel
-        shell: bash
-        run: just --justfile crates/goose-sdk/justfile python-wheel
-
-      - name: Repair Linux wheel
-        if: runner.os == 'Linux'
-        shell: bash
-        run: |
-          python3 -m pip install auditwheel
-          mkdir -p crates/goose-sdk/python/wheelhouse
-          auditwheel repair --plat manylinux_2_28_x86_64 -w crates/goose-sdk/python/wheelhouse crates/goose-sdk/python/dist/*.whl
-          rm crates/goose-sdk/python/dist/*.whl
-          mv crates/goose-sdk/python/wheelhouse/*.whl crates/goose-sdk/python/dist/
-
-      - name: Smoke test wheel
-        shell: bash
-        run: |
-          python -m venv .venv-wheel-test
-          if [ "${{ runner.os }}" = "Windows" ]; then
-            python_bin=".venv-wheel-test/Scripts/python.exe"
-          else
-            python_bin=".venv-wheel-test/bin/python"
-          fi
-          UV_NO_CONFIG=1 uv pip install --default-index https://pypi.org/simple --python "$python_bin" crates/goose-sdk/python/dist/*.whl
-          "$python_bin" -c 'import goose; print(goose.__name__)'
-
-      - name: Upload wheel artifact
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7
-        with:
-          name: goose-sdk-wheel-${{ matrix.name }}
-          path: crates/goose-sdk/python/dist/*.whl
-          if-no-files-found: error
-
-  publish:
-    name: Publish wheels to PyPI
-    needs: build-wheels
-    runs-on: ubuntu-latest
-    if: github.event_name == 'workflow_dispatch' && inputs.publish
-    environment: pypi
-    steps:
-      - name: Download wheel artifacts
-        uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c # v8
-        with:
-          pattern: goose-sdk-wheel-*
-          path: dist
-          merge-multiple: true
-
-      - name: Publish to PyPI
-        uses: pypa/gh-action-pypi-publish@cef221092ed1bacb1cc03d23a2d87d1d172e277b # v1.14.0
-        with:
-          packages-dir: dist

.github/workflows/quarantine.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout Code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       - name: Check PR Author
         run: |
           # Get PR author

.github/workflows/rebuild-skills-marketplace.yml

@@ -45,15 +45,15 @@ jobs:
           fi
 
       - name: Checkout the branch
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Setup Node.js
-        uses: actions/setup-node@48b55a011bda9f5d6aeb4c2d9c7362e8dae4041e  # v6.4.0
+        uses: actions/setup-node@6044e13b5dc448c55e2357c09f80417699197238  # v6.2.0
         with:
           node-version: 20
 
       - name: Cache Node.js modules (documentation)
-        uses: actions/cache@2c8a9bd7457de244a408f35966fab2fb45fda9c8  # v6.0.0
+        uses: actions/cache@8b402f58fbc84540c8b491a91e594a4576fec3d7  # v5.0.2
         with:
           path: ./documentation/node_modules
           key: ${{ runner.os }}-documentation-${{ hashFiles('./documentation/package-lock.json') }}
@@ -78,10 +78,10 @@ jobs:
         uses: actions/configure-pages@45bfe0192ca1faeb007ade9deae92b16b8254a0d  # v6.0.0
 
       - name: Upload GitHub Pages artifact
-        uses: actions/upload-pages-artifact@fc324d3547104276b827a68afc52ff2a11cc49c9  # v5.0.0
+        uses: actions/upload-pages-artifact@7b1f4a764d45c48632c6b24a0339c27f5614fb0b  # v4.0.0
         with:
           path: documentation/build
 
       - name: Deploy to GitHub Pages
         id: deployment
-        uses: actions/deploy-pages@cd2ce8fcbc39b97be8ca5fce6e763baed58fa128  # v5.0.0
+        uses: actions/deploy-pages@d6db90164ac5ed86f2b6aed7e0febac5b3c0c03e  # v4.0.5

.github/workflows/recipe-security-scanner.yml

@@ -30,14 +30,14 @@ jobs:
       # We checkout the base branch (trusted) for building the scanner image,
       # and only fetch recipe files from the PR for scanning.
       - name: Checkout base branch (trusted code)
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.pull_request.base.sha }}
           fetch-depth: 0
           path: trusted
 
       - name: Fetch PR recipe files only
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           ref: ${{ github.event.pull_request.head.sha }}
           fetch-depth: 0
@@ -119,7 +119,7 @@ jobs:
 
       - name: Set up Docker Buildx
         if: steps.find_recipes.outputs.has_recipes == 'true' && steps.recipe_changes.outputs.recipe_files_changed == 'true'
-        uses: docker/setup-buildx-action@bb05f3f5519dd87d3ba754cc423b652a5edd6d2c  # v4.2.0
+        uses: docker/setup-buildx-action@8d2750c68a42422c14e847fe6c8ac0403b4cbd6f  # v3.12.0
 
       - name: Prune Docker caches
         if: steps.find_recipes.outputs.has_recipes == 'true' && steps.recipe_changes.outputs.recipe_files_changed == 'true'
@@ -243,7 +243,7 @@ jobs:
 
       - name: Upload scan artifacts
         if: always() && steps.find_recipes.outputs.has_recipes == 'true' && steps.recipe_changes.outputs.recipe_files_changed == 'true'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: security-scan
           path: ${{ runner.temp }}/security-scan/**

.github/workflows/release.yml

@@ -1,10 +1,8 @@
-# This workflow builds release artifacts for release branches and publishes tagged releases.
+# This workflow is main release, needs to be manually tagged & pushed.
 on:
   push:
     paths-ignore:
       - "documentation/**"
-    branches:
-      - "release/*"
     tags:
       - "v1.*"
 
@@ -17,11 +15,6 @@ permissions:
   pull-requests: write  # Required for npm publish workflow
   attestations: write  # Required for SLSA build provenance attestations
 
-env:
-  # Set this repository Actions variable to "true" in GitHub Settings > Secrets and variables
-  # > Actions > Variables after a release containing desktop app-update.yml has shipped.
-  ENABLE_MAC_NATIVE_AUTO_UPDATE: ${{ vars.ENABLE_MAC_NATIVE_AUTO_UPDATE || 'false' }}
-
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true
@@ -41,8 +34,8 @@ jobs:
     runs-on: ubuntu-latest
     needs: [build-cli]
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
-      - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+      - uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: download_cli.sh
           path: download_cli.sh
@@ -56,8 +49,8 @@ jobs:
       id-token: write
       contents: read
     with:
-      signing: ${{ startsWith(github.ref, 'refs/tags/') }}
-      environment: ${{ startsWith(github.ref, 'refs/tags/') && 'signing' || '' }}
+      signing: true
+      environment: signing
     secrets: inherit
 
   # ------------------------------------------------------------
@@ -69,8 +62,8 @@ jobs:
       id-token: write
       contents: read
     with:
-      signing: ${{ startsWith(github.ref, 'refs/tags/') }}
-      environment: ${{ startsWith(github.ref, 'refs/tags/') && 'signing' || '' }}
+      signing: true
+      environment: signing
     secrets: inherit
 
   # ------------------------------------------------------------
@@ -89,7 +82,7 @@ jobs:
       contents: read
       actions: read
     with:
-      signing: ${{ startsWith(github.ref, 'refs/tags/') }}
+      signing: true
     secrets: inherit
 
   bundle-desktop-windows-cuda:
@@ -99,7 +92,7 @@ jobs:
       contents: read
       actions: read
     with:
-      signing: ${{ startsWith(github.ref, 'refs/tags/') }}
+      signing: true
       windows_variant: cuda
     secrets: inherit
 
@@ -108,7 +101,6 @@ jobs:
   # ------------------------------------
   release:
     name: Release
-    if: startsWith(github.ref, 'refs/tags/')
     runs-on: ubuntu-latest
     needs: [build-cli, install-script, bundle-desktop, bundle-desktop-intel, bundle-desktop-linux, bundle-desktop-windows, bundle-desktop-windows-cuda]
     permissions:
@@ -116,25 +108,13 @@ jobs:
       id-token: write      # Required for Sigstore OIDC signing
       attestations: write  # Required for SLSA build provenance attestations
     steps:
-      - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
-
       - name: Download all artifacts
         uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c  # v8.0.1
         with:
           merge-multiple: true
 
-      - name: Generate macOS update manifest
-        if: ${{ env.ENABLE_MAC_NATIVE_AUTO_UPDATE == 'true' }}
-        run: node ui/desktop/scripts/generate-mac-update-manifest.js --version "${GITHUB_REF_NAME}" --directory .
-
-      - name: Attest macOS update manifest
-        if: ${{ env.ENABLE_MAC_NATIVE_AUTO_UPDATE == 'true' }}
-        uses: actions/attest-build-provenance@0f67c3f4856b2e3261c31976d6725780e5e4c373  # v4.1.1
-        with:
-          subject-path: latest-mac.yml
-
       - name: Attest build provenance
-        uses: actions/attest-build-provenance@0f67c3f4856b2e3261c31976d6725780e5e4c373  # v4.1.1
+        uses: actions/attest-build-provenance@a2bbfa25375fe432b6a289bc6b6cd05ecd0c4c32  # v4.1.0
         with:
           subject-path: |
             goose-*.tar.bz2
@@ -183,11 +163,3 @@ jobs:
           allowUpdates: true
           omitBody: true
           omitPrereleaseDuringUpdate: true
-
-      - name: Upload macOS update manifest
-        if: ${{ env.ENABLE_MAC_NATIVE_AUTO_UPDATE == 'true' }}
-        env:
-          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-        run: |
-          gh release upload "${GITHUB_REF_NAME}" latest-mac.yml --clobber
-          gh release upload stable latest-mac.yml --clobber

.github/workflows/scorecard.yml

@@ -34,12 +34,12 @@ jobs:
 
     steps:
       - name: "Checkout code"
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           persist-credentials: false
 
       - name: "Run analysis"
-        uses: ossf/scorecard-action@4eaacf0543bb3f2c246792bd56e8cdeffafb205a # v2.4.3
+        uses: ossf/scorecard-action@f49aabe0b5af0936a0987cfb85d86b75731b0186 # v2.4.1
         with:
           results_file: results.sarif
           results_format: sarif
@@ -64,7 +64,7 @@ jobs:
       # Upload the results as artifacts (optional). Commenting out will disable uploads of run results in SARIF
       # format to the repository Actions tab.
       - name: "Upload artifact"
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: SARIF file
           path: results.sarif

.github/workflows/stale.yml

@@ -29,7 +29,7 @@ jobs:
     steps:
       # Use the official stale action from GitHub
       - name: 'Close Stale PRs'
-        uses: actions/stale@eb5cf3af3ac0a1aa4c9c45633dd1ae542a27a899  # v10.3.0
+        uses: actions/stale@997185467fa4f803885201cee163a9f38240193d  # v10.1.1
         with:
           # Authentication token with required permissions
           repo-token: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/test-finder.yml

@@ -30,7 +30,7 @@ jobs:
 
     steps:
       - name: Checkout code
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
         with:
           fetch-depth: 0
 

.github/workflows/update-hacktoberfest-leaderboard.yml

@@ -16,7 +16,7 @@ jobs:
       issues: write
     steps:
       - name: Checkout repository
-        uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
 
       - name: Update Leaderboard
         uses: actions/github-script@3a2844b7e9c422d3c10d287c895573f7108da1b3  # v9.0.0

.github/workflows/update-health-dashboard.yml

@@ -223,7 +223,7 @@ jobs:
             console.log(`Successfully updated discussion #${discussionNumber}`);
       
       - name: 'Upload metrics artifact'
-        uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
+        uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f  # v6.0.0
         with:
           name: health-metrics
           path: health-metrics.json

.github/workflows/update-release-pr.yaml

@@ -17,12 +17,12 @@ jobs:
     runs-on: ubuntu-latest
     if: startsWith(github.event.pull_request.head.ref, 'release/') && github.event.pull_request.user.login == 'github-actions[bot]'
     steps:
-    - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0  # v7.0.0
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       with:
         fetch-depth: 0
 
-    - uses: cashapp/activate-hermit@cea9af7913204a965fd488637a8d1811bba2e616 # v1
-    - uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39  # v8.2.0
+    - uses: cashapp/activate-hermit@e49f5cb4dd64ff0b0b659d1d8df499595451155a # v1
+    - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b  # v7.2.0
 
     - name: Extract version from branch name
       env:

.gitignore

@@ -6,7 +6,6 @@ tokenizer_files/
 .DS_Store
 .idea
 .vscode
-.zed/
 *.log
 tmp/
 

AGENTS.md

@@ -14,7 +14,7 @@ cargo build
 ```bash
 cargo build                   # debug
 cargo build --release         # release  
-just release-binary           # release binary
+just release-binary           # release + openapi
 ```
 
 ### Test
@@ -33,21 +33,28 @@ cargo clippy --all-targets -- -D warnings
 
 ### UI
 ```bash
+just generate-openapi        # after server changes
 just run-ui                  # start desktop
-cd ui/desktop && pnpm run typecheck
 cd ui/desktop && pnpm test   # test UI
 ```
 
+### Git
+```bash
+git commit -s                # required for DCO sign-off
+```
+
 ## Structure
 ```
 crates/
 ├── goose              # core logic
 ├── goose-acp-macros   # ACP proc macros
 ├── goose-cli          # CLI entry
+├── goose-server       # backend (binary: goosed)
 ├── goose-mcp          # MCP extensions
 ├── goose-test         # test utilities
 └── goose-test-support # test helpers
 
+evals/open-model-gym/  # benchmarking / evals
 ui/desktop/            # Electron app
 ```
 
@@ -63,6 +70,7 @@ ui/desktop/            # Electron app
 # 1. cargo build
 # 2. cargo test -p <crate>
 # 3. cargo clippy --all-targets -- -D warnings
+# 4. [if server] just generate-openapi
 ```
 
 ## Rules
@@ -72,7 +80,7 @@ ui/desktop/            # Electron app
 - Error: Use anyhow::Result
 - Provider: Implement Provider trait see providers/base.rs
 - MCP: Extensions in crates/goose-mcp/
-- UI Desktop: Use ACP SDK types or local `src/types/*` types. Do not import generated OpenAPI types/client code from `ui/desktop/src/api`
+- Server: Changes need just generate-openapi
 
 ## Code Quality
 
@@ -104,7 +112,7 @@ remaining space for dynamic text.
 
 ## Never
 
-- Never: Recreate `ui/desktop/src/api` or add `@hey-api/openapi-ts` to `ui/desktop`
+- Never: Edit ui/desktop/openapi.json manually
 - Cargo.toml: For human-authored dependency changes, use `cargo add` instead of manually editing dependency entries unless there is a specific reason not to.
 - Cargo.toml: Automated dependency bump PRs are exempt; when manual edits are necessary, keep `Cargo.lock` consistent.
 - Never: Skip cargo fmt
@@ -113,5 +121,6 @@ remaining space for dynamic text.
 
 ## Entry Points
 - CLI: crates/goose-cli/src/main.rs
+- Server: crates/goose-server/src/main.rs
 - UI: ui/desktop/src/main.ts
 - Agent: crates/goose/src/agents/agent.rs

BUILDING_LINUX.md

@@ -59,7 +59,13 @@ cd goose
 Build Goose CLI:
 
 ```bash
-cargo build --release -p goose-cli --bin goose
+cargo build --release -p goose-cli
+```
+
+Build Goose Server:
+
+```bash
+cargo build --release -p goose-server
 ```
 
 This command should give you a list of possible packages in the
@@ -74,9 +80,9 @@ cargo test -p
 cd ui/desktop
 pnpm install
 
-# Copy the goose binary to the expected location
+# Copy the server binary to the expected location
 mkdir -p src/bin
-cp ../../target/release/goose src/bin/
+cp ../../target/release/goosed src/bin/
 ```
 
 ### 4. Build the Application
@@ -137,10 +143,10 @@ cd /path/to/goose/ui/desktop/out/goose-linux-x64
 ./goose 2>&1 | grep -v "GLib-GObject" | grep -v "browser_main_loop"
 ```
 
-#### Goose Binary Not Found
-If you see "Goose binary not found", ensure you've:
-1. Built the Rust binary: `cargo build --release -p goose-cli --bin goose`
-2. Copied it to the right location: `cp ../../target/release/goose src/bin/`
+#### Server Binary Not Found
+If you see "Could not find goosed binary", ensure you've:
+1. Built the Rust backend: `cargo build --release -p goose-server`
+2. Copied it to the right location: `cp ../../target/release/goosed src/bin/`
 3. Rebuilt the application: `pnpm run make`
 
 ### Distribution-Specific Notes
@@ -171,7 +177,7 @@ Building as Snap packages is not currently supported but may be added in the fut
 
 For active development:
 
-1. **Backend changes**: Rebuild with `cargo build --release -p goose-cli --bin goose` and copy the binary
+1. **Backend changes**: Rebuild with `cargo build --release -p goose-server` and copy the binary
 2. **Frontend changes**: Use `pnpm run start` for hot reload during development
 3. **Full rebuild**: Run the complete build process above
 

CONTRIBUTING.md

@@ -29,17 +29,16 @@ It just means the change was too large for a first contribution. Start with some
 
 ### Issues
 
-If you spot a bug, please open an issue. This shows the community and the maintainers the direction of your
-thinking.
+If you spot a bug or have a concrete proposal for a feature, please open an issue. This shows the community and
+the maintainers the direction of your thinking.
 
 For bugs, describe how to reproduce the problem as clearly as possible. If the issue involves an interaction
 with an LLM, include a diagnostics report if possible.
 
 ### Discussions
 
-Before opening a feature request or beginning implementation, please start with a discussion in the
-[goose-eng Discord channel](https://discord.com/channels/1287729918100246654/1514412780504088677). Discussions
-are a good place to explore design questions, alternatives, and whether something fits the goals of the project.
+If you have an idea but are not yet sure how it should work, open a discussion instead. Discussions are a good
+place to explore design questions, alternatives, and whether something fits the goals of the project.
 
 If a change is large or touches multiple parts of the codebase, please start with a discussion before opening a PR.
 This helps us align on direction before you spend time implementing something.
@@ -174,38 +173,41 @@ The app opens a window and displays first-time setup. After completing setup, go
 
 Make GUI changes in `ui/desktop`.
 
-#### Troubleshooting: blank screen on `just run-ui`
+### Regenerating the OpenAPI schema
 
-If the app opens to a blank window (logs show `Cannot read properties of null (reading 'useRef')`), your `node_modules` is out of date and is loading two copies of React. Delete it and reinstall:
+The file `ui/desktop/openapi.json` is automatically generated during the build.
+It is written by the `generate_schema` binary in `crates/goose-server`.
+To update the spec without starting the UI, run:
 
 ```
-rm -rf ui/desktop/node_modules
-cd ui && pnpm install
+just generate-openapi
 ```
 
-See #8757.
+This command regenerates `ui/desktop/openapi.json` and then runs the UI's
+`generate-api` script to rebuild the TypeScript client from that spec.
+
+API changes should be made in the Rust source under `crates/goose-server/src/`.
 
 ### Debugging
 
-To debug the external ACP backend, run it from an IDE. The configuration will depend on the IDE. The command to run is:
+To debug the Goose server, run it from an IDE. The configuration will depend on the IDE. The command to run is:
 
 ```
 export GOOSE_SERVER__SECRET_KEY=test
-cargo run --package goose-cli --bin goose -- serve --platform desktop --host 127.0.0.1 --port 3000
+cargo run --package goose-server --bin goosed -- agent   # or: `just run-server`
 ```
 
-The `debug-ui` recipe connects to `http://127.0.0.1:3000` by default. If the
-backend uses another port, set `GOOSE_PORT` when starting the UI, or set
-`GOOSE_EXTERNAL_BACKEND_URL` to the backend's HTTP base URL.
+The server listens on port `3000` by default; this can be changed by setting the
+`GOOSE_PORT` environment variable.
 
-Once the backend is running, start a UI and connect it to the backend by running:
+Once the server is running, start a UI and connect it to the server by running:
 
 ```
 just debug-ui
 ```
 
-The UI connects to the backend started in the IDE, allowing breakpoints
-and stepping through the backend code while interacting with the UI.
+The UI connects to the server started in the IDE, allowing breakpoints
+and stepping through the server code while interacting with the UI.
 
 ## Creating a fork
 
@@ -348,6 +350,16 @@ This project follows the [Conventional Commits](https://www.conventionalcommits.
 [hermit]: https://cashapp.github.io/hermit/
 [just]: https://github.com/casey/just?tab=readme-ov-file#installation
 
+## Developer Certificate of Origin
+
+This project requires a [Developer Certificate of Origin](https://en.wikipedia.org/wiki/Developer_Certificate_of_Origin) sign-offs on all commits. This is a statement indicating that you are allowed to make the contribution and that the project has the right to distribute it under its license. When you are ready to commit, use the `--signoff` or `-s` flag to attach the sign-off to your commit.
+
+```
+git commit --signoff ...
+# OR
+git commit -s ...
+```
+
 ## Other Ways to Contribute
 
 There are numerous ways to be an open source contributor and contribute to goose. We're here to help you on your way! Here are some suggestions to get started. If you have any questions or need help, feel free to reach out to us on [Discord](https://discord.gg/goose-oss).

CUSTOM_DISTROS.md

@@ -26,8 +26,8 @@ goose's architecture is designed for extensibility. Organizations can create "re
           │                │                      │
           ▼                ▼                      ▼
 ┌─────────────────────────────────────────────────────────────────┐
-│                    goose serve (ACP)                            │
-│         ACP HTTP/WebSocket server for custom clients            │
+│                    goose-server (goosed)                        │
+│         REST API for all goose functionality                    │
 └─────────────────────────────────────────────────────────────────┘
           │
           ▼
@@ -49,7 +49,7 @@ goose's architecture is designed for extensibility. Organizations can create "re
 | Bundle custom MCP extensions | `config.yaml` extensions section, `ui/desktop/src/built-in-extensions.json`, `ui/desktop/src/components/settings/extensions/bundled-extensions.json` | Medium |
 | Modify system prompts | `crates/goose/src/prompts/` | Low |
 | Customize desktop branding | `ui/desktop/` (icons, names, colors) | Medium |
-| Build a new UI (web, mobile) | Integrate with `goose serve` over ACP | High |
+| Build a new UI (web, mobile) | Integrate with `goose-server` REST API | High |
 | Create guided workflows | Recipes (YAML-based task definitions) | Low |
 | Build complex multi-step workflows | Recipes with sub-recipes and subagents | Medium |
 
@@ -304,32 +304,40 @@ export GOOSE_BUNDLE_NAME="InsightStream-goose"
 
 **Goal**: Create an entirely new frontend while leveraging goose's backend.
 
-goose provides two ACP transport options for building custom clients:
+goose provides two integration options for building custom UIs:
 
-### Option 1: ACP HTTP/WebSocket (`goose serve`)
+### Option 1: REST API (goose-server)
 
-Use `goose serve` for process-separated integrations such as web apps, desktop shells, and other clients:
+Use goose-server for HTTP-based integrations (web apps, simple clients):
 
 ```bash
 # Start the server
-GOOSE_SERVER__SECRET_KEY='a-long-random-secret' goose serve
+./target/release/goosed
 
-# ACP endpoint available at http://localhost:3284/acp
+# API available at http://localhost:3000
 ```
 
-HTTP clients authenticate with the `X-Secret-Key` header. Browser WebSocket clients use the same secret as a `?token=` query parameter because the browser WebSocket API cannot set custom headers:
+**Reference the OpenAPI spec** at `ui/desktop/openapi.json` for available endpoints:
+- Session management
+- Message streaming  
+- Extension control
+- Configuration
 
-```text
-ws://localhost:3284/acp?token=a-long-random-secret
+**Key endpoints** for a minimal integration:
+
+```
+POST /sessions              # Create a new session
+POST /sessions/{id}/messages # Send a message (streaming response)
+GET  /sessions/{id}         # Get session state
+GET  /extensions            # List available extensions
+POST /extensions/{name}/enable  # Enable an extension
 ```
 
-For browser clients served from a non-loopback origin, pass the exact UI origin with `--allowed-origin`. When you pass `--allowed-origin`, it replaces the default loopback origin allowlist, so include every origin the client needs.
+**Handle streaming responses** - goose uses Server-Sent Events (SSE) for real-time responses.
 
-For the ACP protocol and client flow, see [Agent Client Protocol clients](documentation/docs/guides/acp-clients.md).
+### Option 2: Agent Client Protocol (ACP)
 
-### Option 2: Agent Client Protocol (ACP) over stdio
-
-For richer local integrations (IDEs and embedded agents), run goose as an ACP agent over stdio.
+For richer integrations (IDEs, desktop apps, embedded agents), use the **Agent Client Protocol (ACP)**—a standardized JSON-RPC protocol for AI agent communication over stdio or other transports.
 
 ACP provides:
 - **Bidirectional communication**: Agents can request permissions, stream updates, and receive cancellations
@@ -402,6 +410,11 @@ For the full ACP specification, see the [Agent Client Protocol documentation](ht
 
 ### Technical Details
 
+**REST API (goose-server)**:
+- Server implementation: `crates/goose-server/src/routes/`
+- OpenAPI generation: `just generate-openapi`
+- API client example: `ui/desktop/src/api/` (generated TypeScript client)
+
 **ACP**:
 - ACP server implementation: `crates/goose/src/acp/server.rs`
 - CLI integration: `crates/goose-cli/src/cli.rs` (Command::Acp)
@@ -642,10 +655,10 @@ prompt: |
   - find_patterns: Find similar features to model after
 ```
 
-Recipes that define `sub_recipes` get the Summon extension automatically. The AI can invoke sub-recipes through Summon's `delegate` tool:
+The AI can then invoke these sub-recipes using the `subagent` tool:
 
 ```
-delegate(source: "find_files", parameters: {"search_term": "authentication"})
+subagent(subrecipe: "find_files", parameters: {"search_term": "authentication"})
 ```
 
 ### Subagents: Dynamic Task Delegation
@@ -665,27 +678,28 @@ prompt: |
   To complete this task:
   
   1. Spawn a subagent to analyze the frontend code:
-     delegate(instructions: "Analyze all React components in src/components/ and list their props and state management patterns")
+     subagent(instructions: "Analyze all React components in src/components/ 
+              and list their props and state management patterns")
   
   2. Spawn another subagent for the backend:
-     delegate(instructions: "Document all API endpoints in src/api/ including their request/response schemas")
+     subagent(instructions: "Document all API endpoints in src/api/ 
+              including their request/response schemas")
   
   3. Synthesize findings from both subagents into a unified report.
 ```
 
 #### Parallel Subagent Execution
 
-Use `async: true` to run delegates in parallel, then collect each result with `load(source: "<task_id>")`:
+Multiple subagent calls in the same message execute in parallel:
 
 ```yaml
 prompt: |
-  Run these analyses in parallel:
+  Run these analyses in parallel by making all subagent calls at once:
   
-  delegate(instructions: "Count lines of code by language", async: true)
-  delegate(instructions: "Find all TODO comments", async: true)
-  delegate(instructions: "List external dependencies", async: true)
+  subagent(instructions: "Count lines of code by language")
+  subagent(instructions: "Find all TODO comments") 
+  subagent(instructions: "List external dependencies")
   
-  Use load(source: "<task_id>") for each returned task id.
   Then combine the results into a codebase health report.
 ```
 
@@ -697,18 +711,22 @@ Customize model, provider, or behavior per subagent:
 prompt: |
   Use a faster model for simple tasks:
   
-  delegate(
+  subagent(
     instructions: "List all files modified in the last week",
-    model: "gpt-4o-mini",
-    max_turns: 3
+    settings: {
+      model: "gpt-4o-mini",
+      max_turns: 3
+    }
   )
   
   Use the full model for complex analysis:
   
-  delegate(
+  subagent(
     instructions: "Review this code for security vulnerabilities",
-    model: "claude-sonnet-4-20250514",
-    temperature: 0.1
+    settings: {
+      model: "claude-sonnet-4-20250514",
+      temperature: 0.1
+    }
   )
 ```
 
@@ -720,7 +738,7 @@ Limit which extensions a subagent can access:
 prompt: |
   Create a sandboxed subagent with only file reading capabilities:
   
-  delegate(
+  subagent(
     instructions: "Analyze the README files in this project",
     extensions: ["developer"]  # Only developer extension, no network access
   )
@@ -801,7 +819,7 @@ prompt: |
 2. **Parallelize independent tasks** - Multiple subagent calls in one message run concurrently
 3. **Use `sequential_when_repeated: true`** - For tasks that shouldn't run in parallel (e.g., database migrations)
 4. **Scope extensions appropriately** - Give subagents only the tools they need
-5. **Use background delegation for independent work** - Pass `async: true` to `delegate`, then collect results with `load(source: "<task_id>")`
+5. **Use summary mode (default)** - Subagents return concise summaries; use `summary: false` only when you need full conversation history
 6. **Handle failures gracefully** - Design workflows to continue even if one subagent fails
 
 ### Technical Details

Cargo.toml

@@ -8,8 +8,8 @@ resolver = "2"
 
 [workspace.package]
 edition = "2021"
-version = "1.42.0"
-rust-version = "1.94.1"
+version = "1.36.0"
+rust-version = "1.91.1"
 authors = ["AAIF <ai-oss-tools@block.xyz>"]
 license = "Apache-2.0"
 repository = "https://github.com/aaif-goose/goose"
@@ -21,9 +21,8 @@ string_slice = "warn"
 
 [workspace.dependencies]
 rmcp = { version = "1.4", default-features = false, features = ["schemars", "auth"] }
-agent-client-protocol-schema = { version = "1.1", default-features = false, features = ["unstable"] }
-agent-client-protocol = { version = "1.0", default-features = false }
-agent-client-protocol-http = { version = "1.0", default-features = false, features = ["server", "unstable_cancel_request"] }
+agent-client-protocol-schema = { version = "0.12", default-features = false, features = ["unstable"] }
+agent-client-protocol = { version = "0.11", default-features = false }
 arboard = { version = "3", default-features = false }
 anyhow = { version = "1.0.102", default-features = false, features = ["std"] }
 async-stream = { version = "0.3.6", default-features = false }
@@ -31,11 +30,11 @@ async-trait = { version = "0.1.89", default-features = false }
 axum = { version = "0.8", default-features = false, features = ["http1", "http2", "json", "tokio", "query"] }
 base64 = { version = "0.22.1", default-features = false, features = ["std"] }
 bytes = { version = "1.10.1", default-features = false, features = ["std"] }
-candle-core = { version = "0.11", default-features = false }
-candle-nn = { version = "0.11", default-features = false }
+candle-core = { version = "0.10", default-features = false }
+candle-nn = { version = "0.10", default-features = false }
 chrono = { version = "0.4.44", default-features = false, features = ["serde", "std"] }
 clap = { version = "4.1.14", default-features = false, features = ["derive", "std", "help", "suggestions", "usage", "color", "error-context"] }
-dirs = { version = "6", default-features = false }
+dirs = { version = "5", default-features = false }
 dotenvy = { version = "0.15.7", default-features = false }
 env-lock = { version = "1", default-features = false }
 etcetera = { version = "0.11", default-features = false }
@@ -46,9 +45,9 @@ ignore = { version = "0.4.12", default-features = false }
 include_dir = { version = "0.7", default-features = false }
 indoc = { version = "2", default-features = false }
 keyring = { version = "3.6.3", default-features = false, features = ["vendored"] }
-lru = { version = "0.18", default-features = false }
+lru = { version = "0.16", default-features = false }
 once_cell = { version = "1.21.3", default-features = false, features = ["std"] }
-rand = "0.10.1"
+rand = { version = "0.8.5", default-features = false, features = ["std"] }
 regex = { version = "1.12.3", default-features = false, features = ["std"] }
 reqwest = { version = "0.13.2", default-features = false, features = ["multipart", "form"] }
 rustls = { version = "0.23.31", default-features = false, features = ["aws_lc_rs", "std"] }
@@ -57,27 +56,27 @@ serde = { version = "1.0.228", default-features = false, features = ["derive", "
 serde_json = { version = "1.0.145", default-features = false, features = ["std"] }
 serde_yaml = { version = "0.9.32", default-features = false }
 shellexpand = { version = "3", default-features = false, features = ["base-0", "tilde"] }
-strum = { version = "0.28.0", default-features = false, features = ["derive", "std"] }
+strum = { version = "0.27.1", default-features = false, features = ["derive", "std"] }
 tempfile = { version = "3.10.1", default-features = false }
-thiserror = { version = "2.0.18", default-features = false }
+thiserror = { version = "1.0.49", default-features = false }
 tokio = { version = "1.48", default-features = false }
 tokio-stream = { version = "0.1.16", default-features = false }
 tokio-util = { version = "0.7.12", default-features = false }
-tower-http = { version = "0.7.0", default-features = false }
+tower-http = { version = "0.6.8", default-features = false }
 tracing = { version = "0.1.43", default-features = false, features = ["std"] }
 tracing-appender = { version = "0.2.1", default-features = false }
 tracing-futures = { version = "0.2.4", default-features = false, features = ["futures-03", "std", "std-future"] }
 tracing-subscriber = { version = "0.3.22", default-features = false, features = ["std"] }
 urlencoding = { version = "2.1", default-features = false }
 utoipa = { version = "4.2", default-features = false }
-uuid = { version = "1.23", default-features = false, features = ["v4", "std"] }
+uuid = { version = "1.18", default-features = false, features = ["v4", "std"] }
 webbrowser = { version = "1", default-features = false }
 which = { version = "8", default-features = false, features = ["real-sys"] }
 winapi = { version = "0.3.9", default-features = false, features = ["wincred", "std"] }
 wiremock = { version = "0.6", default-features = false }
 zip = { version = "8", default-features = false, features = ["deflate"] }
 serial_test = { version = "3", default-features = false }
-sha2 = { version = "0.11", default-features = false }
+sha2 = { version = "0.10.8", default-features = false, features = ["std"] }
 shell-words = { version = "1", default-features = false, features = ["std"] }
 test-case = { version = "3", default-features = false }
 url = { version = "2.5.4", default-features = false, features = ["std"] }
@@ -114,12 +113,3 @@ icu_locale = { version = "=2.1.1", default-features = false }
 [patch.crates-io]
 v8 = { path = "vendor/v8" }
 cudaforge = { git = "https://github.com/jbg/cudaforge", rev = "e7c1967340e40673db98dc9e17da0f04834a456f" }
-
-# Dependencies don't need debug info. You rarely step-debug into third-party
-# crates, but their DWARF dominates `target/debug` — roughly 50-70% of each
-# dependency rlib in this workspace is debug info. Dropping it for dependencies
-# only shrinks the debug build substantially and speeds linking. The "*" spec
-# matches all dependencies but NOT workspace members, so first-party crates keep
-# full debug info and stay fully debuggable; release builds are unaffected.
-[profile.dev.package."*"]
-debug = false

I18N.md

@@ -100,15 +100,11 @@ Compiled files go to `src/i18n/compiled/` (gitignored).
 
 The locale is resolved at startup in the following order:
 
-1. **Desktop language setting** — selected in Settings, unless set to System Default
-2. **`GOOSE_LOCALE`** — explicit override from env/app config when the desktop setting is
-   System Default
-3. **`navigator.language`** — the browser/OS locale
-4. **`"en"`** — fallback default
+1. **`GOOSE_LOCALE`** — explicit override (set on the `window` object or via env)
+2. **`navigator.language`** — the browser/OS locale
+3. **`"en"`** — fallback default
 
-The resolved locale is used for both text translations and all Intl formatting (dates, numbers,
-relative times). Changing the desktop language setting reloads the renderer so startup-only locale
-helpers pick up the new value.
+The resolved locale is used for both text translations and all Intl formatting (dates, numbers, relative times).
 
 ## Date and number formatting
 
@@ -138,8 +134,7 @@ This ensures date/number formatting uses the same locale as the rest of the UI.
 1. Copy `src/i18n/messages/en.json` to a new file, e.g., `src/i18n/messages/ja.json`.
 2. Translate the `defaultMessage` values. Keep ICU syntax intact (e.g., `{count, plural, ...}`).
 3. Add the locale code to `SUPPORTED_LOCALES` in `src/i18n/index.ts`.
-4. Add the locale to the language selector in `src/components/settings/app/AppSettingsSection.tsx`.
-5. Optionally run `pnpm i18n:compile` to pre-compile.
+4. Optionally run `pnpm i18n:compile` to pre-compile.
 
 No other code changes are needed — `loadMessages()` dynamically imports the correct catalog at runtime.
 

Justfile

@@ -13,14 +13,18 @@ check-everything:
     cargo clippy --all-targets -- -D warnings
     @echo "  → Checking UI code formatting..."
     cd ui/desktop && pnpm run lint:check
+    @echo "  → Validating OpenAPI schema..."
+    ./scripts/check-openapi-schema.sh
     @echo ""
     @echo "✅ All style checks passed!"
 
 # Default release command
 release-binary:
     @echo "Building release version..."
-    cargo build --release -p goose-cli --bin goose
+    cargo build --release
     @just copy-binary
+    @echo "Generating OpenAPI schema..."
+    cargo run -p goose-server --bin generate_schema
 
 # Build Windows executable on a Windows host
 [unix]
@@ -30,7 +34,7 @@ release-windows:
 
 [windows]
 release-windows:
-    @powershell.exe -NoProfile -ExecutionPolicy Bypass -Command 'rustup target add x86_64-pc-windows-msvc; if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }; cargo build --release --target x86_64-pc-windows-msvc -p goose-cli --bin goose; if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }; Write-Host "Windows executable created at ./target/x86_64-pc-windows-msvc/release/goose.exe"'
+    @powershell.exe -NoProfile -ExecutionPolicy Bypass -Command 'rustup target add x86_64-pc-windows-msvc; if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }; cargo build --release --target x86_64-pc-windows-msvc -p goose-server; if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }; Write-Host "Windows executable created at ./target/x86_64-pc-windows-msvc/release/goosed.exe"'
 
 # Build for Intel Mac
 release-intel:
@@ -39,10 +43,15 @@ release-intel:
     @just copy-binary-intel
 
 copy-binary BUILD_MODE="release":
-    @rm -f ./ui/desktop/src/bin/goosed
+    @if [ -f ./target/{{BUILD_MODE}}/goosed ]; then \
+        echo "Copying goosed binary from target/{{BUILD_MODE}}..."; \
+        cp -p ./target/{{BUILD_MODE}}/goosed ./ui/desktop/src/bin/; \
+    else \
+        echo "Binary not found in target/{{BUILD_MODE}}"; \
+        exit 1; \
+    fi
     @if [ -f ./target/{{BUILD_MODE}}/goose ]; then \
         echo "Copying goose CLI binary from target/{{BUILD_MODE}}..."; \
-        rm -f ./ui/desktop/src/bin/goose; \
         cp -p ./target/{{BUILD_MODE}}/goose ./ui/desktop/src/bin/; \
     else \
         echo "goose CLI binary not found in target/{{BUILD_MODE}}"; \
@@ -51,10 +60,15 @@ copy-binary BUILD_MODE="release":
 
 # Copy binary command for Intel build
 copy-binary-intel:
-    @rm -f ./ui/desktop/src/bin/goosed
+    @if [ -f ./target/x86_64-apple-darwin/release/goosed ]; then \
+        echo "Copying Intel goosed binary to ui/desktop/src/bin with permissions preserved..."; \
+        cp -p ./target/x86_64-apple-darwin/release/goosed ./ui/desktop/src/bin/; \
+    else \
+        echo "Intel release binary not found."; \
+        exit 1; \
+    fi
     @if [ -f ./target/x86_64-apple-darwin/release/goose ]; then \
         echo "Copying Intel goose CLI binary to ui/desktop/src/bin..."; \
-        rm -f ./ui/desktop/src/bin/goose; \
         cp -p ./target/x86_64-apple-darwin/release/goose ./ui/desktop/src/bin/; \
     else \
         echo "Intel goose CLI binary not found."; \
@@ -69,11 +83,10 @@ copy-binary-windows:
 
 [windows]
 copy-binary-windows:
-    @powershell.exe -NoProfile -ExecutionPolicy Bypass -Command 'if (Test-Path ./target/x86_64-pc-windows-msvc/release/goose.exe) { \
+    @powershell.exe -NoProfile -ExecutionPolicy Bypass -Command 'if (Test-Path ./target/x86_64-pc-windows-msvc/release/goosed.exe) { \
         Write-Host "Copying Windows binary to ui/desktop/src/bin..."; \
         New-Item -ItemType Directory -Force "./ui/desktop/src/bin" | Out-Null; \
-        Remove-Item -Path "./ui/desktop/src/bin/goosed.exe" -Force -ErrorAction SilentlyContinue; \
-        Copy-Item -Path "./target/x86_64-pc-windows-msvc/release/goose.exe" -Destination "./ui/desktop/src/bin/" -Force; \
+        Copy-Item -Path "./target/x86_64-pc-windows-msvc/release/goosed.exe" -Destination "./ui/desktop/src/bin/" -Force; \
     } else { \
         Write-Host "Windows binary not found." -ForegroundColor Red; \
         exit 1; \
@@ -99,7 +112,7 @@ run-ui-only:
     cd ui/desktop && pnpm install && pnpm run start-gui
 
 debug-ui:
-    @echo "🚀 Starting goose frontend in external ACP backend mode"
+    @echo "🚀 Starting goose frontend in external backend mode"
     cd ui/desktop && \
     export GOOSE_EXTERNAL_BACKEND=true && \
     export GOOSE_SERVER__SECRET_KEY="${GOOSE_SERVER__SECRET_KEY:-test}" && \
@@ -144,8 +157,19 @@ run-docs:
 
 # Run server
 run-server:
-    @echo "Running external ACP backend..."
-    GOOSE_SERVER__SECRET_KEY="${GOOSE_SERVER__SECRET_KEY:-test}" cargo run -p goose-cli --bin goose -- serve --platform desktop --host 127.0.0.1 --port 3000
+    @echo "Running server..."
+    cargo run -p goose-server --bin goosed agent
+
+# Check if OpenAPI schema is up-to-date
+check-openapi-schema: generate-openapi
+    ./scripts/check-openapi-schema.sh
+
+# Generate OpenAPI specification without starting the UI
+generate-openapi:
+    @echo "Generating OpenAPI schema..."
+    cargo run -p goose-server --bin generate_schema
+    @echo "Generating frontend API..."
+    cd ui/desktop && npx @hey-api/openapi-ts
 
 # Check if generated ACP schema and TypeScript types are up-to-date
 check-acp-schema: generate-acp-types
@@ -290,6 +314,7 @@ bump-version version:
     @cd ui/desktop && npm pkg set "version={{ version }}"
     # update Cargo.lock after bumping versions in Cargo.toml
     @cargo update --workspace
+    @just set-openapi-version {{ version }}
 
 # rebuild canonical model registry and mapping report from models.dev
 build-canonical-models:
@@ -304,10 +329,14 @@ prepare-release version:
         Cargo.lock \
         ui/desktop/package.json \
         ui/pnpm-lock.yaml \
-        crates/goose-provider-types/src/canonical/data/canonical_models.json \
-        crates/goose-provider-types/src/canonical/data/provider_metadata.json
+        ui/desktop/openapi.json \
+        crates/goose/src/providers/canonical/data/canonical_models.json \
+        crates/goose/src/providers/canonical/data/provider_metadata.json
     @git commit --message "chore(release): release version {{ version }}"
 
+set-openapi-version version:
+    @jq '.info.version |= "{{ version }}"' ui/desktop/openapi.json > ui/desktop/openapi.json.tmp && mv ui/desktop/openapi.json.tmp ui/desktop/openapi.json
+
 # extract version from Cargo.toml
 get-tag-version:
     @uvx --from=toml-cli toml get --toml-path=Cargo.toml "workspace.package.version"
@@ -342,6 +371,7 @@ set windows-shell := ["powershell.exe", "-NoLogo", "-Command"]
 ### profile = --release or "" for debug
 ### allparam = OR/AND/ANY/NONE --workspace --all-features --all-targets
 win-bld profile allparam:
+  cargo run {{profile}} -p goose-server --bin  generate_schema
   cargo build {{profile}} {{allparam}}
 
 ### Build just debug
@@ -370,7 +400,6 @@ win-app-deps:
 win-copy-win profile:
   copy target{{s}}{{profile}}{{s}}*.exe ui{{s}}desktop{{s}}src{{s}}bin
   copy target{{s}}{{profile}}{{s}}*.dll ui{{s}}desktop{{s}}src{{s}}bin
-  if exist ui{{s}}desktop{{s}}src{{s}}bin{{s}}goosed.exe del /f /q ui{{s}}desktop{{s}}src{{s}}bin{{s}}goosed.exe
 
 ### "Other" copy {release|debug} files to ui/desktop/src/bin
 ### s = os dependent file separator

README.md

@@ -1,3 +1,5 @@
+> **🦆 goose has moved!** This project has moved from `block/goose` to the [Agentic AI Foundation (AAIF)](https://aaif.io/) at the Linux Foundation. Some links and references are still being updated — please bear with us during the transition.
+
 <div align="center">
 
 # goose
@@ -14,12 +16,8 @@ _your native open source AI agent — desktop app, CLI, and API — for code, wo
   <a href="https://insights.linuxfoundation.org/project/goose"><img src="https://insights.linuxfoundation.org/api/badge/health-score?project=goose"></a>
   <a href="https://repology.org/project/goose-cli/versions"><img src="https://repology.org/badge/tiny-repos/goose-cli.svg" alt="Packaging status"></a>
 </p>
-
-<a href="https://trendshift.io/repositories/25298?utm_source=repository-badge&amp;utm_medium=badge&amp;utm_campaign=badge-repository-25298" target="_blank" rel="noopener noreferrer"><img src="https://trendshift.io/api/badge/repositories/25298" alt="aaif-goose%2Fgoose | Trendshift" width="250" height="55"/></a>
-
 </div>
 
-
 goose is a general-purpose AI agent that runs on your machine. Not just for code — use it for research, writing, automation, data analysis, or anything you need to get done.
 
 A native desktop app for macOS, Linux, and Windows. A full CLI for terminal workflows. An API to embed it anywhere. Built in Rust for performance and portability.

RELEASE_CHECKLIST.md

@@ -1,23 +1,134 @@
 # goose Release Manual Testing Checklist
 
-Download the release builds from this PR. Once a build is ready, the actions bot will post a comment on this PR
-with instructions on how to download and sign.
+## Version: {{VERSION}}
+
+### Identify the high risk changes in this Release
 
-## Use the following script to create a risk assessment and testing plan:
 ```
 ./workflow_recipes/release_risk_check/run.sh {{VERSION}}
 ```
 
-It will generate an analysis report in `/tmp/release_report_final.md` and perform testing is necessary for high risk pr changes.
+It will generate an analysis report in `/tmp/release_report_final.md` and perform testing is necessary for high risk pr changes. 
 
-## Run the goose self-test recipe
 
-goose run --recipe goose-self-test.yaml
+## Regression Testing
 
-## Have goose produce a test plan
+Make a copy of this document for each version and check off as steps are verified.
 
-Open the release candidate desktop app and have goose produce a test plan by pointing it at this PR. Use a prompt like
+### Provider Testing
 
-> Look at the notes in PR <release PR> and the report at `/tmp/release_report_final.md` and investigate potential risks in this release. After familiarizing yourself with the scope of each change, produce a suggested test plan that I should follow before publishing the release.
+- [ ] Run `cd ui/desktop && pnpm run test:integration:providers` locally from the release branch and verify all providers/models work
+- [ ] Launch goose, click reset providers, choose databricks and a model
 
-goose will produce a plan. Follow this plan to finish testing.
+### Starting Conversations
+
+Test various ways to start a conversation:
+
+- [ ] Open home and start a new conversation with "Hello"
+  - [ ] Agent responds
+  - [ ] Token count is updated after agent finishes
+  - [ ] Go to history and see there is a new entry
+- [ ] Go back to the main screen, start a new conversation from the hub and see that it opens a new conversation
+- [ ] Open history and click the Hello conversation - verify it loads
+- [ ] Add a new message to this conversation and see that it is added
+- [ ] Change the working directory of an existing conversation
+  - [ ] Ask "what is your working directory?"
+  - [ ] Response should match the new directory
+- [ ] Open a new window, click chat in left side for new chat
+- [ ] Type "create a tamagotchi game" in the chat input to test developer extension
+
+### Recipes
+
+#### Create Recipe from Session
+
+- [ ] Start a simple chat conversation like "hi"
+- [ ] Click "create a recipe from this session" in the bottom chat bar
+  - [ ] Recipe title, description and instructions should be filled in with details from the chat
+  - [ ] Add a few activities and params (params unused indicator should update if added to instructions/prompts or activities)
+  - [ ] Can launch create and run recipe - launches in a new window showing as a recipe agent chat with parameters filled in and interact with it
+  - [ ] Recipe should be saved in recipe library
+
+#### Use Existing Recipe
+
+- [ ] Pick trip planner from recipe hub (go/gooserecipes)
+  - [ ] See the warning whether to trust this recipe (only on fresh install)
+  - [ ] See the form pop up
+  - [ ] Fill in the form with "Africa" and "14 days"
+  - [ ] Check results are reasonable
+  - [ ] Ask how many days the trip is for - should say 14
+
+#### Recipe Management
+
+- [ ] Go to recipe manager and enter a new recipe to generate a joke
+  - [ ] See that it works if you run it
+  - [ ] Edit the recipe by bottom bar and click "View/Edit Recipe"
+  - [ ] Make it generate a limerick instead
+  - [ ] Check that the updated recipe works
+  - [ ] Delete the recipe from the recipe manager
+  - [ ] Verify recipe is actually deleted
+
+#### Recipe from File
+
+- [ ] Create a file `~/.config/goose/recipes/test-recipe.yaml` with the following content:
+
+```yaml
+recipe:
+  title: test recipe again
+  description: testing recipe again
+  instructions: The value of test_param is {{test_param}}
+  prompt: What is the value of test_param?
+  parameters:
+    - key: test_param
+      input_type: string
+      requirement: required
+      description: Enter value for test_param
+```
+
+- [ ] See that it shows up in the list of installed recipes
+- [ ] Launch the recipe, see that it asks for test_param
+- [ ] Enter a number, see that it pre-fills the prompt and tells you the value after you hit submit
+- [ ] Go to hub and enter "what is the value of test_param"
+- [ ] See a new chat that says it has no idea (recipe is no longer active)
+
+### Extensions
+
+#### Manual Extension Addition
+
+- [ ] Can manually add an extension using random quotes from project
+  - [ ] Add new custom stdio extension with the following command and save:
+    - [ ] `node /ABSOLUTE/PATH/TO/goose/ui/desktop/tests/e2e/basic-mcp.ts` (use your actual project path)
+    - [ ] Should add and can chat to ask for a random quote
+
+#### Playwright Extension
+
+- [ ] Install the playwright extension from the extensions hub
+  - [ ] Tell it to open a browser and search on Google for cats
+  - [ ] Verify that the browser opens and navigates
+
+#### Extension with Environment Variables
+
+- [ ] Install an extension from deeplink that needs env variables:
+  - [ ] Use: `goose://extension?cmd=npx&arg=-y&arg=%40upstash%2Fcontext7-mcp&id=context7&name=Context7&description=Use%20up-to-date%20code%20and%20docs&env=TEST_ACCESS_TOKEN`
+  - [ ] Extension page should load with env variables modal showing
+  - [ ] Allow form input and saving extension
+
+### Speech-to-Text (Local Model)
+
+- [ ] Go to Settings > Chat > Voice dictation provider and select the small model
+- [ ] Run a quick test that speech-to-text is working (click the mic button, speak, verify transcription)
+- [ ] Also try OpenAI using your OpenAI key
+
+### Settings
+
+- [ ] Settings page loads and all tabs load
+- [ ] Can change dark mode setting
+
+### Follow-up Issues
+
+Link any GitHub issues filed during testing:
+
+---
+
+**Tested by:** _____
+**Date:** _____
+**Notes:** _____

crates/goose-cli/Cargo.toml

@@ -20,10 +20,9 @@ name = "generate_manpages"
 path = "src/bin/generate_manpages.rs"
 
 [dependencies]
-clap_mangen = { version = "0.3", default-features = false }
+clap_mangen = { version = "0.2", default-features = false }
 goose = { path = "../goose", default-features = false }
 goose-mcp = { path = "../goose-mcp", default-features = false }
-goose-providers = { path = "../goose-providers", default-features = false }
 rmcp = { workspace = true }
 clap = { workspace = true }
 cliclack = { version = "0.5", default-features = false }
@@ -44,7 +43,7 @@ rustyline = { version = "18", default-features = false, features = ["custom-bind
 tracing = { workspace = true }
 chrono = { workspace = true }
 tracing-subscriber = { workspace = true, features = ["env-filter", "fmt", "json", "time"] }
-shlex = { version = "2.0", default-features = false, features = ["std"] }
+shlex = { version = "1.3", default-features = false, features = ["std"] }
 async-trait = { workspace = true }
 base64 = { workspace = true }
 regex = { workspace = true }
@@ -62,9 +61,8 @@ urlencoding = { workspace = true }
 clap_complete = { version = "4", default-features = false }
 comfy-table = { version = "7", default-features = false }
 sha2 = { workspace = true }
-sigstore-verify = { version = "0.10", default-features = false, optional = true }
+sigstore-verify = { version = "0.8", default-features = false, optional = true }
 axum.workspace = true
-axum-server = { version = "0.8", default-features = false, optional = true }
 clap_complete_nushell = { version = "4", default-features = false }
 
 [target.'cfg(target_os = "windows")'.dependencies]
@@ -75,7 +73,6 @@ winapi = { workspace = true }
 default = [
     "code-mode",
     "local-inference",
-    "tui",
     "aws-providers",
     "telemetry",
     "nostr",
@@ -89,37 +86,28 @@ local-inference = ["goose/local-inference"]
 aws-providers = ["goose/aws-providers"]
 cuda = ["goose/cuda", "local-inference"]
 vulkan = ["goose/vulkan", "local-inference"]
-mlx = ["goose/mlx", "local-inference"]
 telemetry = ["goose/telemetry"]
 nostr = ["goose/nostr"]
 otel = ["goose/otel"]
 system-keyring = ["goose/system-keyring"]
-tui = []
 update = ["dep:sigstore-verify"]
-portable-default = ["rustls-tls", "aws-providers", "telemetry", "otel", "tui"]
+portable-default = ["rustls-tls", "aws-providers", "telemetry", "otel"]
 # disables the update command
 disable-update = []
 rustls-tls = [
-    "dep:axum-server",
     "reqwest/rustls",
-    "axum-server/tls-rustls",
     "sigstore-verify?/rustls",
     "goose/rustls-tls",
     "goose-mcp/rustls-tls",
-    "goose-providers/rustls-tls",
 ]
 native-tls = [
-    "dep:axum-server",
     "reqwest/native-tls",
-    "axum-server/tls-openssl",
     "sigstore-verify?/native-tls",
     "goose/native-tls",
     "goose-mcp/native-tls",
-    "goose-providers/native-tls",
 ]
 
 [dev-dependencies]
-env-lock.workspace = true
 tempfile = { workspace = true }
 test-case = { workspace = true }
 tokio = { workspace = true }

crates/goose-cli/src/cli.rs

@@ -29,7 +29,6 @@ use crate::commands::schedule::{
     handle_schedule_sessions,
 };
 use crate::commands::session::{handle_session_list, handle_session_remove};
-use crate::commands::skills::handle_skills_list;
 use crate::recipes::extract_from_cli::extract_recipe_info_from_cli;
 use crate::recipes::recipe::{explain_recipe, render_recipe_as_yaml};
 use crate::session::{build_session, SessionBuilderConfig};
@@ -43,30 +42,14 @@ use tracing::warn;
 const GOOSE_SERVER_SECRET_KEY_ENV: &str = "GOOSE_SERVER__SECRET_KEY";
 
 fn generate_serve_secret_key() -> String {
-    use rand::distr::{Alphanumeric, SampleString};
+    use rand::distributions::{Alphanumeric, DistString};
 
     format!(
         "goose-acp-{}",
-        Alphanumeric.sample_string(&mut rand::rng(), 32)
+        Alphanumeric.sample_string(&mut rand::thread_rng(), 32)
     )
 }
 
-#[derive(clap::ValueEnum, Clone, Copy, Debug, Default, PartialEq, Eq)]
-enum ServePlatform {
-    #[default]
-    Cli,
-    Desktop,
-}
-
-impl From<ServePlatform> for GoosePlatform {
-    fn from(platform: ServePlatform) -> Self {
-        match platform {
-            ServePlatform::Cli => GoosePlatform::GooseCli,
-            ServePlatform::Desktop => GoosePlatform::GooseDesktop,
-        }
-    }
-}
-
 #[derive(Parser)]
 #[command(name = "goose", author, version, display_name = "", about, long_about = None)]
 pub struct Cli {
@@ -91,7 +74,7 @@ pub struct Identifier {
         alias = "id",
         value_name = "SESSION_ID",
         help = "Session ID (e.g., '20250921_143022')",
-        long_help = "Specify a session ID to resume. Requires --resume."
+        long_help = "Specify a session ID directly. When used with --resume, will resume this specific session if it exists."
     )]
     pub session_id: Option<String>,
 
@@ -376,13 +359,6 @@ pub struct RunBehavior {
     )]
     pub resume: bool,
 
-    /// Print generation statistics after completion
-    #[arg(
-        long = "stats",
-        help = "Print generation statistics after the run completes"
-    )]
-    pub stats: bool,
-
     /// Scheduled job ID (used internally for scheduled executions)
     #[arg(
         long = "scheduled-job-id",
@@ -408,9 +384,7 @@ async fn get_or_create_session_id(
 
     let resolved_id = if resume {
         let Some(id) = identifier else {
-            let sessions = session_manager
-                .list_sessions_by_types(&[SessionType::User])
-                .await?;
+            let sessions = session_manager.list_sessions().await?;
             let session_id = sessions
                 .first()
                 .map(|s| s.id.clone())
@@ -584,13 +558,9 @@ enum SessionCommand {
         )]
         relays: Vec<String>,
     },
-    #[command(
-        about = "Import a session from JSON, a Claude Code / Codex / Pi .jsonl, or an encrypted Nostr share link"
-    )]
+    #[command(about = "Import a session from JSON or an encrypted Nostr share link")]
     Import {
-        #[arg(
-            help = "Path to a goose session export, a Claude Code, Codex, or Pi .jsonl transcript, or a goose://sessions/nostr share link"
-        )]
+        #[arg(help = "Path to a JSON session export, or a goose://sessions/nostr share link")]
         input: String,
 
         #[arg(long = "nostr", help = "Treat input as an encrypted Nostr share link")]
@@ -598,9 +568,11 @@ enum SessionCommand {
     },
     #[command(name = "diagnostics")]
     Diagnostics {
+        /// Session identifier for generating diagnostics
         #[command(flatten)]
         identifier: Option<Identifier>,
 
+        /// Output path for the diagnostics zip file (optional, defaults to current directory)
         #[arg(short = 'o', long)]
         output: Option<PathBuf>,
     },
@@ -727,13 +699,6 @@ enum PluginCommand {
     },
 }
 
-#[derive(Subcommand)]
-enum SkillsCommand {
-    /// List all skills available to the goose agent
-    #[command(about = "List all skills available to the goose agent")]
-    List,
-}
-
 #[derive(Subcommand)]
 enum RecipeCommand {
     /// Validate a recipe file
@@ -849,18 +814,6 @@ enum Command {
         #[arg(long, default_value = "3284")]
         port: u16,
 
-        #[arg(long, help = "Serve ACP over TLS")]
-        tls: bool,
-
-        #[arg(long = "tls-cert-path", value_name = "PATH")]
-        tls_cert_path: Option<String>,
-
-        #[arg(long = "tls-key-path", value_name = "PATH")]
-        tls_key_path: Option<String>,
-
-        #[arg(long, value_enum, default_value_t = ServePlatform::Cli)]
-        platform: ServePlatform,
-
         #[arg(
             long = "with-builtin",
             value_name = "NAME",
@@ -870,20 +823,6 @@ enum Command {
             action = clap::ArgAction::Append
         )]
         builtins: Vec<String>,
-
-        #[arg(
-            long = "dangerously-unauthenticated",
-            help = "Start the ACP endpoint without requiring GOOSE_SERVER__SECRET_KEY"
-        )]
-        dangerously_unauthenticated: bool,
-
-        #[arg(
-            long = "allowed-origin",
-            value_name = "ORIGIN",
-            action = clap::ArgAction::Append,
-            help = "Allow an exact Origin value for ACP CORS; may be specified multiple times and replaces the default loopback origins"
-        )]
-        allowed_origins: Vec<String>,
     },
 
     /// Start or resume interactive chat sessions
@@ -916,15 +855,6 @@ enum Command {
         )]
         fork: bool,
 
-        /// Open the session's conversation in $EDITOR before starting
-        #[arg(
-            long,
-            requires = "resume",
-            help = "Edit the session conversation in $EDITOR before starting",
-            long_help = "Open the session's conversation in your editor ($VISUAL / $EDITOR / vi) for modification before resuming. When combined with --fork, creates a new session from the edited result."
-        )]
-        edit: bool,
-
         /// Show message history when resuming
         #[arg(
             long,
@@ -980,13 +910,6 @@ enum Command {
         command: RecipeCommand,
     },
 
-    /// Skill utilities
-    #[command(about = "Skill utilities")]
-    Skills {
-        #[command(subcommand)]
-        command: SkillsCommand,
-    },
-
     /// Manage plugins
     #[command(about = "Manage plugins")]
     Plugin {
@@ -1048,7 +971,6 @@ enum Command {
     },
 
     /// Launch the goose terminal UI (TUI)
-    #[cfg(feature = "tui")]
     #[command(
         about = "Launch the goose terminal UI",
         long_about = "Launch the goose terminal UI (the @aaif/goose npm package).\n\
@@ -1120,9 +1042,8 @@ enum Command {
         #[arg(long = "override-model", value_name = "MODEL")]
         override_model: Option<String>,
 
-        /// Default `turn-limit` for orchestrated main-pass subprocesses and
-        /// for checks that do not declare their own. Does not cap the legacy
-        /// `--no-orchestrate` in-process main agent.
+        /// Default `turn-limit` applied to checks that do not declare their
+        /// own.
         #[arg(long = "turn-limit", value_name = "N")]
         turn_limit: Option<usize>,
 
@@ -1186,6 +1107,7 @@ enum Command {
         #[arg(long = "severity", value_name = "LEVEL", default_value = "medium")]
         severity: String,
     },
+
     #[command(
         name = "validate-extensions",
         about = "Validate a bundled-extensions.json file",
@@ -1200,8 +1122,8 @@ enum Command {
 #[cfg(feature = "local-inference")]
 #[derive(Subcommand)]
 enum LocalModelsCommand {
-    /// Search HuggingFace for local models
-    #[command(about = "Search HuggingFace for local GGUF and MLX models")]
+    /// Search HuggingFace for GGUF models
+    #[command(about = "Search HuggingFace for GGUF models")]
     Search {
         /// Search query
         query: String,
@@ -1212,9 +1134,9 @@ enum LocalModelsCommand {
     },
 
     /// Download a model from HuggingFace
-    #[command(about = "Download a local model from a search result")]
+    #[command(about = "Download a GGUF model (e.g. bartowski/Llama-3.2-1B-Instruct-GGUF:Q4_K_M)")]
     Download {
-        /// Model spec/download id, e.g. user/repo:Q4_K_M or user/repo
+        /// Model spec in user/repo:quantization format
         spec: String,
     },
 
@@ -1351,10 +1273,8 @@ fn get_command_name(command: &Option<Command>) -> &'static str {
         #[cfg(feature = "update")]
         Some(Command::Update { .. }) => "update",
         Some(Command::Recipe { .. }) => "recipe",
-        Some(Command::Skills { .. }) => "skills",
         Some(Command::Plugin { .. }) => "plugin",
         Some(Command::Term { .. }) => "term",
-        #[cfg(feature = "tui")]
         Some(Command::Tui { .. }) => "tui",
         #[cfg(feature = "local-inference")]
         Some(Command::LocalModels { .. }) => "local-models",
@@ -1377,39 +1297,13 @@ async fn handle_mcp_command(server: McpCommand) -> Result<()> {
     Ok(())
 }
 
-struct ServeCommandArgs {
-    host: String,
-
-    port: u16,
-    tls: bool,
-    tls_cert_path: Option<String>,
-    tls_key_path: Option<String>,
-    platform: ServePlatform,
-    builtins: Vec<String>,
-    dangerously_unauthenticated: bool,
-    allowed_origins: Vec<String>,
-}
-
-async fn handle_serve_command(args: ServeCommandArgs) -> Result<()> {
-    use axum::http::HeaderValue;
+async fn handle_serve_command(host: String, port: u16, builtins: Vec<String>) -> Result<()> {
     use goose::acp::server_factory::{AcpServer, AcpServerFactoryConfig};
     use goose::acp::transport::create_router;
     use goose::config::paths::Paths;
     use std::net::SocketAddr;
     use std::sync::Arc;
-    use tracing::{info, warn};
-
-    let ServeCommandArgs {
-        host,
-        port,
-        tls,
-        tls_cert_path,
-        tls_key_path,
-        platform,
-        builtins,
-        dangerously_unauthenticated,
-        allowed_origins,
-    } = args;
+    use tracing::info;
 
     let builtins = if builtins.is_empty() {
         vec!["developer".to_string()]
@@ -1433,93 +1327,25 @@ async fn handle_serve_command(args: ServeCommandArgs) -> Result<()> {
         builtins,
         data_dir: Paths::data_dir(),
         config_dir: Paths::config_dir(),
-        goose_platform: platform.into(),
+        goose_platform: GoosePlatform::GooseCli,
         additional_source_roots,
     }));
-    let env_secret = std::env::var(GOOSE_SERVER_SECRET_KEY_ENV)
+    let secret_key = std::env::var(GOOSE_SERVER_SECRET_KEY_ENV)
         .ok()
         .map(|secret| secret.trim().to_string())
-        .filter(|secret| !secret.is_empty());
-    let require_token = env_secret.is_some();
-    if !require_token && !dangerously_unauthenticated {
-        anyhow::bail!(
-            "{GOOSE_SERVER_SECRET_KEY_ENV} must be set to start `goose serve`; pass --dangerously-unauthenticated to run without ACP authentication"
-        );
-    }
-    if dangerously_unauthenticated && !require_token {
-        warn!(
-            "{GOOSE_SERVER_SECRET_KEY_ENV} is not set and --dangerously-unauthenticated was passed; the ACP endpoint will accept unauthenticated connections"
-        );
-    }
-    let additional_allowed_origins = allowed_origins
-        .into_iter()
-        .map(|origin| {
-            let origin = origin.trim();
-            if origin.is_empty() || origin == "*" {
-                anyhow::bail!("--allowed-origin must be a non-wildcard Origin value");
-            }
-            HeaderValue::from_str(origin).map_err(|error| {
-                anyhow::anyhow!("invalid --allowed-origin value `{origin}`: {error}")
-            })
-        })
-        .collect::<Result<Vec<_>>>()?;
-    let secret_key = env_secret.unwrap_or_else(generate_serve_secret_key);
-    let router = create_router(
-        server,
-        secret_key,
-        require_token,
-        additional_allowed_origins,
-    );
-
-    let config = Config::global();
-    let tls_cert_path =
-        tls_cert_path.or_else(|| config.get_param::<String>("GOOSE_TLS_CERT_PATH").ok());
-    let tls_key_path =
-        tls_key_path.or_else(|| config.get_param::<String>("GOOSE_TLS_KEY_PATH").ok());
-    let tls = tls
-        || config.get_param::<bool>("GOOSE_TLS").unwrap_or(false)
-        || tls_cert_path.is_some()
-        || tls_key_path.is_some();
+        .filter(|secret| !secret.is_empty())
+        .unwrap_or_else(generate_serve_secret_key);
+    let router = create_router(server, secret_key);
 
     let addr: SocketAddr = format!("{}:{}", host, port).parse()?;
-    if tls {
-        #[cfg(any(feature = "rustls-tls", feature = "native-tls"))]
-        {
-            let tls_setup = goose::acp::transport::tls::setup_tls(
-                tls_cert_path.as_deref(),
-                tls_key_path.as_deref(),
-            )
-            .await?;
-            info!("Starting ACP server on https://{}", addr);
+    info!("Starting ACP server on {}", addr);
 
-            #[cfg(feature = "rustls-tls")]
-            axum_server::bind_rustls(addr, tls_setup.config)
-                .serve(router.into_make_service_with_connect_info::<SocketAddr>())
-                .await?;
-
-            #[cfg(feature = "native-tls")]
-            axum_server::bind_openssl(addr, tls_setup.config)
-                .serve(router.into_make_service_with_connect_info::<SocketAddr>())
-                .await?;
-        }
-
-        #[cfg(not(any(feature = "rustls-tls", feature = "native-tls")))]
-        {
-            let _ = (tls_cert_path, tls_key_path);
-            anyhow::bail!(
-                "TLS was requested but no TLS backend is enabled. \
-                 Enable the `rustls-tls` or `native-tls` feature."
-            );
-        }
-    } else {
-        info!("Starting ACP server on http://{}", addr);
-        let listener = tokio::net::TcpListener::bind(addr).await?;
-        axum::serve(
-            listener,
-            router.into_make_service_with_connect_info::<SocketAddr>(),
-        )
-        .await?;
-    }
+    let listener = tokio::net::TcpListener::bind(addr).await?;
+    axum::serve(
+        listener,
+        router.into_make_service_with_connect_info::<SocketAddr>(),
+    )
+    .await?;
 
     Ok(())
 }
@@ -1604,7 +1430,6 @@ async fn handle_interactive_session(
     identifier: Option<Identifier>,
     resume: bool,
     fork: bool,
-    edit: bool,
     history: bool,
     session_opts: SessionOptions,
     extension_opts: ExtensionOptions,
@@ -1644,31 +1469,12 @@ async fn handle_interactive_session(
     let goose_mode = Config::global().get_goose_mode().unwrap_or_default();
     let mut session_id = get_or_create_session_id(identifier, resume, false, goose_mode).await?;
 
-    if edit || fork {
-        if let Some(ref id) = session_id {
+    if fork {
+        if let Some(id) = session_id {
             let session_manager = SessionManager::instance();
-            let original = session_manager.get_session(id, true).await?;
-
-            let target_id = if fork {
-                let copied = session_manager
-                    .copy_session(id, original.name.clone())
-                    .await?;
-                let copied_id = copied.id.clone();
-                session_id = Some(copied.id);
-                copied_id
-            } else {
-                id.clone()
-            };
-
-            if edit {
-                let conversation = original
-                    .conversation
-                    .ok_or_else(|| anyhow::anyhow!("session has no messages to edit"))?;
-                let edited = crate::session::editor::edit_conversation(&conversation)?;
-                session_manager
-                    .replace_conversation(&target_id, &edited)
-                    .await?;
-            }
+            let original = session_manager.get_session(&id, false).await?;
+            let copied = session_manager.copy_session(&id, original.name).await?;
+            session_id = Some(copied.id);
         }
     }
 
@@ -1693,7 +1499,6 @@ async fn handle_interactive_session(
         quiet: false,
         output_format: "text".to_string(),
         container: session_opts.container.map(Container::new),
-        stats: false,
     })
     .await;
 
@@ -1718,7 +1523,7 @@ async fn log_session_completion(
     let (total_tokens, message_count) = session
         .get_session()
         .await
-        .map(|m| (m.usage.total_tokens.unwrap_or(0), m.message_count))
+        .map(|m| (m.total_tokens.unwrap_or(0), m.message_count))
         .unwrap_or((0, 0));
 
     tracing::info!(
@@ -1906,7 +1711,6 @@ async fn handle_run_command(
         quiet: output_opts.quiet,
         output_format: output_opts.output_format,
         container: session_opts.container.map(Container::new),
-        stats: run_behavior.stats,
     })
     .await;
 
@@ -1995,12 +1799,6 @@ fn handle_recipe_subcommand(command: RecipeCommand) -> Result<()> {
     }
 }
 
-async fn handle_skills_subcommand(command: SkillsCommand) -> Result<()> {
-    match command {
-        SkillsCommand::List => handle_skills_list().await,
-    }
-}
-
 async fn handle_term_subcommand(command: TermCommand) -> Result<()> {
     match command {
         TermCommand::Init {
@@ -2014,40 +1812,20 @@ async fn handle_term_subcommand(command: TermCommand) -> Result<()> {
     }
 }
 
-#[cfg(feature = "local-inference")]
-fn print_download_progress(manager: &goose::download_manager::DownloadManager) {
-    let Some(progress) = manager
-        .list_progress()
-        .into_iter()
-        .find(|progress| progress.status == goose::download_manager::DownloadStatus::Downloading)
-    else {
-        return;
-    };
-
-    print!(
-        "\r  {:.1}% ({:.0}MB / {:.0}MB)",
-        progress.progress_percent,
-        progress.bytes_downloaded as f64 / (1024.0 * 1024.0),
-        progress.total_bytes as f64 / (1024.0 * 1024.0),
-    );
-    use std::io::Write;
-    std::io::stdout().flush().ok();
-}
-
 #[cfg(feature = "local-inference")]
 async fn handle_local_models_command(command: LocalModelsCommand) -> Result<()> {
     use goose::providers::local_inference::hf_models;
-    use goose::providers::local_inference::local_model_registry::get_registry;
-
-    goose::providers::local_inference::configure_huggingface_auth();
+    use goose::providers::local_inference::local_model_registry::{
+        get_registry, mmproj_local_path, model_id_from_repo, LocalModelEntry,
+    };
 
     match command {
         LocalModelsCommand::Search { query, limit } => {
             println!("Searching HuggingFace for '{}'...", query);
-            let results = hf_models::search_local_models(&query, limit).await?;
+            let results = hf_models::search_gguf_models(&query, limit).await?;
 
             if results.is_empty() {
-                println!("No compatible local models found.");
+                println!("No GGUF models found.");
                 return Ok(());
             }
 
@@ -2056,68 +1834,126 @@ async fn handle_local_models_command(command: LocalModelsCommand) -> Result<()>
                     "\n{} (by {}) — {} downloads",
                     model.model_name, model.author, model.downloads
                 );
-                for variant in &model.variants {
-                    let size = if variant.size_bytes > 0 {
+                for file in &model.gguf_files {
+                    let size = if file.size_bytes > 0 {
                         format!(
                             "{:.1}GB",
-                            variant.size_bytes as f64 / (1024.0 * 1024.0 * 1024.0)
+                            file.size_bytes as f64 / (1024.0 * 1024.0 * 1024.0)
                         )
                     } else {
                         "unknown".to_string()
                     };
-                    let support = if variant.supported {
-                        String::new()
-                    } else {
-                        format!(
-                            " ({})",
-                            variant
-                                .unsupported_reason
-                                .as_deref()
-                                .unwrap_or("unsupported on this platform")
-                        )
-                    };
-                    println!(
-                        "  [{}] {} — {} — {}{}",
-                        variant.format, variant.label, size, variant.description, support
-                    );
-                    if variant.supported {
-                        println!(
-                            "    Download: goose local-models download '{}'",
-                            variant.download_id
-                        );
-                    }
+                    println!("  {} — {}", file.quantization, size);
                 }
+                println!(
+                    "  Download: goose local-models download {}:<quantization>",
+                    model.repo_id
+                );
             }
         }
         LocalModelsCommand::Download { spec } => {
             println!("Resolving {}...", spec);
-            let manager = goose::download_manager::get_download_manager();
-            let resolve_task = hf_models::resolve_local_model_spec(&spec);
-            tokio::pin!(resolve_task);
-            let resolved = loop {
-                tokio::select! {
-                    result = &mut resolve_task => break result?,
-                    _ = tokio::time::sleep(std::time::Duration::from_millis(500)) => {
-                        print_download_progress(manager);
-                    }
-                }
-            };
-            let model_id = resolved.model_id();
-            let total_size = resolved.total_size();
+            let (repo_id, resolved) = hf_models::resolve_model_spec_full(&spec).await?;
+            if resolved.files.len() > 1 {
+                anyhow::bail!(
+                    "Model '{}' is sharded ({} files) — download it from the desktop UI",
+                    spec,
+                    resolved.files.len()
+                );
+            }
+            let mmproj = resolved.mmproj;
+            let file = resolved.files.into_iter().next().unwrap();
+            let model_id = model_id_from_repo(&repo_id, &file.quantization);
+            let local_path =
+                goose::config::paths::Paths::in_data_dir("models").join(&file.filename);
+            let mmproj_path = mmproj
+                .as_ref()
+                .map(|mmproj| mmproj_local_path(&repo_id, &mmproj.filename));
+            let mmproj_source_url = mmproj.as_ref().map(|mmproj| mmproj.download_url.clone());
+            let mmproj_size_bytes = mmproj.as_ref().map_or(0, |mmproj| mmproj.size_bytes);
+            let mut download_files = vec![(file.download_url.clone(), local_path.clone())];
+            if let Some(mmproj) = mmproj {
+                download_files.push((mmproj.download_url, mmproj_path.clone().unwrap()));
+            }
 
             println!(
-                "\nDownloaded {} ({}). Registering...",
+                "Downloading {} ({})...",
                 model_id,
-                if total_size > 0 {
-                    format!("{:.1}GB", total_size as f64 / (1024.0 * 1024.0 * 1024.0))
+                if file.size_bytes > 0 {
+                    format!(
+                        "{:.1}GB",
+                        file.size_bytes as f64 / (1024.0 * 1024.0 * 1024.0)
+                    )
                 } else {
                     "unknown size".to_string()
                 }
             );
 
-            let model_id = hf_models::register_resolved_model(resolved, &spec)?;
+            // Register
+            let entry = LocalModelEntry {
+                id: model_id.clone(),
+                repo_id: repo_id.clone(),
+                filename: file.filename.clone(),
+                quantization: file.quantization.clone(),
+                local_path: local_path.clone(),
+                source_url: file.download_url.clone(),
+                settings: Default::default(),
+                size_bytes: file.size_bytes,
+                mmproj_path,
+                mmproj_source_url,
+                mmproj_size_bytes,
+                mmproj_checked: true,
+                shard_files: vec![],
+            };
 
-            println!("Registered: {}", model_id);
+            {
+                let mut registry = get_registry()
+                    .lock()
+                    .map_err(|_| anyhow::anyhow!("Failed to acquire registry lock"))?;
+                registry.add_model(entry)?;
+            }
+
+            // Download
+            let manager = goose::download_manager::get_download_manager();
+            manager
+                .download_model_sharded(
+                    format!("{}-model", model_id),
+                    download_files,
+                    file.size_bytes + mmproj_size_bytes,
+                    None,
+                )
+                .await?;
+
+            // Poll progress
+            loop {
+                if let Some(progress) = manager.get_progress(&format!("{}-model", model_id)) {
+                    match progress.status {
+                        goose::download_manager::DownloadStatus::Downloading => {
+                            print!(
+                                "\r  {:.1}% ({:.0}MB / {:.0}MB)",
+                                progress.progress_percent,
+                                progress.bytes_downloaded as f64 / (1024.0 * 1024.0),
+                                progress.total_bytes as f64 / (1024.0 * 1024.0),
+                            );
+                            use std::io::Write;
+                            std::io::stdout().flush().ok();
+                        }
+                        goose::download_manager::DownloadStatus::Completed => {
+                            println!("\nDownloaded: {}", model_id);
+                            break;
+                        }
+                        goose::download_manager::DownloadStatus::Failed => {
+                            let err = progress.error.unwrap_or_default();
+                            anyhow::bail!("Download failed: {}", err);
+                        }
+                        goose::download_manager::DownloadStatus::Cancelled => {
+                            println!("\nDownload cancelled.");
+                            break;
+                        }
+                    }
+                }
+                tokio::time::sleep(std::time::Duration::from_millis(500)).await;
+            }
         }
         LocalModelsCommand::List => {
             let registry = get_registry()
@@ -2130,16 +1966,12 @@ async fn handle_local_models_command(command: LocalModelsCommand) -> Result<()>
                 return Ok(());
             }
 
-            println!(
-                "{:<50} {:<10} {:<12} Downloaded",
-                "ID", "Backend", "Variant"
-            );
-            println!("{}", "-".repeat(88));
+            println!("{:<50} {:<10} Downloaded", "ID", "Quant");
+            println!("{}", "-".repeat(70));
             for m in models {
                 println!(
-                    "{:<50} {:<10} {:<12} {}",
+                    "{:<50} {:<10} {}",
                     m.id,
-                    m.backend_id.as_deref().unwrap_or("llamacpp"),
                     m.quantization,
                     if m.is_downloaded() { "✓" } else { "✗" }
                 );
@@ -2150,8 +1982,11 @@ async fn handle_local_models_command(command: LocalModelsCommand) -> Result<()>
                 .lock()
                 .map_err(|_| anyhow::anyhow!("Failed to acquire registry lock"))?;
 
-            if registry.get_model(&id).is_some() {
-                registry.delete_model(&id)?;
+            if let Some(entry) = registry.get_model(&id) {
+                if entry.local_path.exists() {
+                    std::fs::remove_file(&entry.local_path)?;
+                }
+                registry.remove_model(&id)?;
                 println!("Deleted model: {}", id);
             } else {
                 println!("Model not found: {}", id);
@@ -2196,7 +2031,6 @@ async fn handle_default_session() -> Result<()> {
         quiet: false,
         output_format: "text".to_string(),
         container: None,
-        stats: false,
     })
     .await;
     session.interactive(None).await
@@ -2232,27 +2066,8 @@ pub async fn cli() -> anyhow::Result<()> {
         Some(Command::Serve {
             host,
             port,
-            tls,
-            tls_cert_path,
-            tls_key_path,
-            platform,
             builtins,
-            dangerously_unauthenticated,
-            allowed_origins,
-        }) => {
-            handle_serve_command(ServeCommandArgs {
-                host,
-                port,
-                tls,
-                tls_cert_path,
-                tls_key_path,
-                platform,
-                builtins,
-                dangerously_unauthenticated,
-                allowed_origins,
-            })
-            .await
-        }
+        }) => handle_serve_command(host, port, builtins).await,
         Some(Command::Session {
             command: Some(cmd), ..
         }) => handle_session_subcommand(cmd).await,
@@ -2261,7 +2076,6 @@ pub async fn cli() -> anyhow::Result<()> {
             identifier,
             resume,
             fork,
-            edit,
             history,
             session_opts,
             extension_opts,
@@ -2270,7 +2084,6 @@ pub async fn cli() -> anyhow::Result<()> {
                 identifier,
                 resume,
                 fork,
-                edit,
                 history,
                 session_opts,
                 extension_opts,
@@ -2316,10 +2129,8 @@ pub async fn cli() -> anyhow::Result<()> {
             Ok(())
         }
         Some(Command::Recipe { command }) => handle_recipe_subcommand(command),
-        Some(Command::Skills { command }) => handle_skills_subcommand(command).await,
         Some(Command::Plugin { command }) => handle_plugin_subcommand(command),
         Some(Command::Term { command }) => handle_term_subcommand(command).await,
-        #[cfg(feature = "tui")]
         Some(Command::Tui { args }) => crate::commands::tui::handle_tui(args),
         #[cfg(feature = "local-inference")]
         Some(Command::LocalModels { command }) => handle_local_models_command(command).await,
@@ -2436,134 +2247,4 @@ mod tests {
         let help = String::from_utf8(buffer).expect("utf8");
         assert!(help.contains("nu"));
     }
-
-    #[test]
-    fn skills_command_accepts_list_subcommand() {
-        let cli = Cli::try_parse_from(["goose", "skills", "list"]).expect("parse failed");
-
-        match cli.command {
-            Some(Command::Skills {
-                command: SkillsCommand::List,
-            }) => {}
-            _ => panic!("expected skills list command"),
-        }
-    }
-
-    #[test]
-    fn serve_command_accepts_dangerously_unauthenticated_flag() {
-        let cli = Cli::try_parse_from([
-            "goose",
-            "serve",
-            "--dangerously-unauthenticated",
-            "--allowed-origin",
-            "app://localhost",
-            "--allowed-origin",
-            "https://app.example",
-        ])
-        .expect("parse failed");
-
-        match cli.command {
-            Some(Command::Serve {
-                dangerously_unauthenticated,
-                allowed_origins,
-                ..
-            }) => {
-                assert!(dangerously_unauthenticated);
-                assert_eq!(
-                    allowed_origins,
-                    vec!["app://localhost", "https://app.example"]
-                );
-            }
-            _ => panic!("expected serve command"),
-        }
-    }
-
-    #[test]
-    fn review_command_accepts_options() {
-        let cli = Cli::try_parse_from([
-            "goose",
-            "review",
-            "origin/main...HEAD",
-            "--prompt",
-            "REVIEW.md",
-            "--model",
-            "test-model",
-            "--provider",
-            "openai",
-            "--override-model",
-            "check-model",
-            "--turn-limit",
-            "4",
-            "--dry-run",
-            "--quiet",
-            "--no-orchestrate",
-            "--instructions",
-            "focus on correctness",
-            "--files",
-            "src/lib.rs",
-            "--check-filter",
-            "security",
-            "--check-scope",
-            ".agents",
-            "--checks-only",
-            "--summary-only",
-            "--severity",
-            "low",
-        ])
-        .expect("parse failed");
-
-        match cli.command {
-            Some(Command::Review {
-                range,
-                prompt,
-                model,
-                provider,
-                override_model,
-                turn_limit,
-                dry_run,
-                quiet,
-                no_orchestrate,
-                instructions,
-                files,
-                check_filter,
-                check_scope,
-                checks_only,
-                summary_only,
-                severity,
-            }) => {
-                assert_eq!(range.as_deref(), Some("origin/main...HEAD"));
-                assert_eq!(prompt.as_deref(), Some(std::path::Path::new("REVIEW.md")));
-                assert_eq!(model.as_deref(), Some("test-model"));
-                assert_eq!(provider.as_deref(), Some("openai"));
-                assert_eq!(override_model.as_deref(), Some("check-model"));
-                assert_eq!(turn_limit, Some(4));
-                assert!(dry_run);
-                assert!(quiet);
-                assert!(no_orchestrate);
-                assert_eq!(instructions.as_deref(), Some("focus on correctness"));
-                assert_eq!(files, vec!["src/lib.rs"]);
-                assert_eq!(check_filter, vec!["security"]);
-                assert_eq!(
-                    check_scope.as_deref(),
-                    Some(std::path::Path::new(".agents"))
-                );
-                assert!(checks_only);
-                assert!(summary_only);
-                assert_eq!(severity, "low");
-            }
-            _ => panic!("expected review command"),
-        }
-    }
-
-    #[cfg(feature = "tui")]
-    #[test]
-    fn tui_command_accepts_trailing_args() {
-        let cli =
-            Cli::try_parse_from(["goose", "tui", "--", "--theme", "dark"]).expect("parse failed");
-
-        match cli.command {
-            Some(Command::Tui { args }) => assert_eq!(args, vec!["--theme", "dark"]),
-            _ => panic!("expected tui command"),
-        }
-    }
 }

crates/goose-cli/src/commands/configure.rs

@@ -19,31 +19,20 @@ use goose::config::{
     configure_tetrate, Config, ConfigError, ExperimentManager, ExtensionEntry, GooseMode,
     PermissionManager,
 };
+use goose::model::ModelConfig;
 #[cfg(feature = "telemetry")]
 use goose::posthog::{get_telemetry_choice, TELEMETRY_ENABLED_KEY};
 use goose::providers::base::ConfigKey;
 use goose::providers::provider_test::test_provider_configuration;
 use goose::providers::{create, providers, retry_operation, RetryConfig};
 use goose::session::SessionType;
-use goose_providers::thinking::ThinkingEffort;
 use serde_json::Value;
 use std::collections::HashMap;
-use std::io::{IsTerminal, Write};
+use std::io::IsTerminal;
 
 // useful for light themes where there is no discernible colour contrast between
 // cursor-selected and cursor-unselected items.
 const MULTISELECT_VISIBILITY_HINT: &str = "<";
-const SHOW_CURSOR: &[u8] = b"\x1b[?25h";
-
-struct CursorRestoreGuard;
-
-impl Drop for CursorRestoreGuard {
-    fn drop(&mut self) {
-        let mut stdout = std::io::stdout().lock();
-        let _ = stdout.write_all(SHOW_CURSOR);
-        let _ = stdout.flush();
-    }
-}
 
 pub async fn handle_configure() -> anyhow::Result<()> {
     if !std::io::stdin().is_terminal() {
@@ -53,7 +42,6 @@ pub async fn handle_configure() -> anyhow::Result<()> {
         );
     }
 
-    let _cursor_restore = CursorRestoreGuard;
     let config = Config::global();
 
     if !config.exists() {
@@ -351,7 +339,8 @@ async fn handle_oauth_configuration(provider_name: &str, key_name: &str) -> anyh
     ));
 
     // Create a temporary provider instance to handle OAuth
-    match create(provider_name, Vec::new()).await {
+    let temp_model = ModelConfig::new("temp")?.with_canonical_limits(provider_name);
+    match create(provider_name, temp_model, Vec::new()).await {
         Ok(provider) => match provider.configure_oauth().await {
             Ok(_) => {
                 let _ = cliclack::log::success("OAuth authentication completed successfully!");
@@ -376,12 +365,7 @@ async fn handle_oauth_configuration(provider_name: &str, key_name: &str) -> anyh
     }
 }
 
-const UNLISTED_MODEL_KEY: &str = "__unlisted__";
-
-fn interactive_model_search(
-    models: &[String],
-    provider_meta: &goose::providers::base::ProviderMetadata,
-) -> anyhow::Result<String> {
+fn interactive_model_search(models: &[String]) -> anyhow::Result<String> {
     const MAX_VISIBLE: usize = 30;
     let mut query = String::new();
 
@@ -411,20 +395,7 @@ fn interactive_model_search(
         };
 
         if filtered.is_empty() {
-            let selection = cliclack::select("No matching models. What would you like to do?")
-                .item(
-                    "__new_search__",
-                    "Start a new search...",
-                    "Enter a different search term",
-                )
-                .item(UNLISTED_MODEL_KEY, "Enter a model not listed...", "")
-                .interact()?;
-
-            if selection == UNLISTED_MODEL_KEY {
-                return prompt_unlisted_model(provider_meta);
-            }
-
-            query.clear();
+            let _ = cliclack::log::warning("No matching models. Try a different search.");
             continue;
         }
 
@@ -458,12 +429,6 @@ fn interactive_model_search(
             );
         }
 
-        items.push((
-            UNLISTED_MODEL_KEY.to_string(),
-            "Enter a model not listed...".to_string(),
-            "",
-        ));
-
         let selection = cliclack::select("Select a model:")
             .items(&items)
             .interact()?;
@@ -473,8 +438,6 @@ fn interactive_model_search(
         } else if selection == "__new_search__" {
             query.clear();
             continue;
-        } else if selection == UNLISTED_MODEL_KEY {
-            return prompt_unlisted_model(provider_meta);
         } else {
             return Ok(selection);
         }
@@ -486,6 +449,7 @@ fn select_model_from_list(
     provider_meta: &goose::providers::base::ProviderMetadata,
 ) -> anyhow::Result<String> {
     const MAX_MODELS: usize = 10;
+    const UNLISTED_MODEL_KEY: &str = "__unlisted__";
 
     // Smart model selection:
     // If we have more than MAX_MODELS models, show the recommended models with additional search option.
@@ -524,14 +488,14 @@ fn select_model_from_list(
                 .interact()?;
 
             if selection == "search_all" {
-                interactive_model_search(models, provider_meta)
+                Ok(interactive_model_search(models)?)
             } else if selection == UNLISTED_MODEL_KEY {
                 prompt_unlisted_model(provider_meta)
             } else {
                 Ok(selection)
             }
         } else {
-            interactive_model_search(models, provider_meta)
+            Ok(interactive_model_search(models)?)
         }
     } else {
         let mut model_items: Vec<(String, String, &str)> =
@@ -772,11 +736,11 @@ pub async fn configure_provider_dialog() -> anyhow::Result<bool> {
 
     let spin = spinner();
     spin.start("Attempting to fetch supported models...");
-    let temp_provider = create(provider_name, Vec::new()).await?;
+    let temp_model_config =
+        ModelConfig::new(&provider_meta.default_model)?.with_canonical_limits(provider_name);
+    let temp_provider = create(provider_name, temp_model_config, Vec::new()).await?;
     let models_res = retry_operation(&RetryConfig::default(), || async {
-        temp_provider
-            .fetch_recommended_models(goose::model_config::global_toolshim())
-            .await
+        temp_provider.fetch_recommended_models().await
     })
     .await;
     spin.stop(style("Model fetch complete").green());
@@ -801,20 +765,20 @@ pub async fn configure_provider_dialog() -> anyhow::Result<bool> {
     {
         let supports_thinking = match temp_provider.fetch_model_info(&model).await {
             Ok(model_info) => model_info.reasoning,
-            Err(_) => goose_providers::model::ModelConfig::new(&model).is_reasoning_model(),
+            Err(_) => goose::model::ModelConfig::new(&model)
+                .map(|c| c.is_reasoning_model())
+                .unwrap_or(false),
         };
 
         if supports_thinking {
-            let effort: ThinkingEffort = cliclack::select("Select thinking effort:")
+            let effort: &str = cliclack::select("Select thinking effort:")
                 .item("off", "Off - No extended thinking", "")
                 .item("low", "Low - Better latency, lighter reasoning", "")
                 .item("medium", "Medium - Moderate thinking", "")
                 .item("high", "High - Deep reasoning", "")
                 .item("max", "Max - No constraints on thinking depth", "")
                 .initial_value("off")
-                .interact()?
-                .parse()
-                .map_err(|_| anyhow::anyhow!("invalid thinking effort"))?;
+                .interact()?;
             config.set_goose_thinking_effort(effort)?;
         }
     }
@@ -1116,7 +1080,6 @@ fn configure_stdio_extension() -> anyhow::Result<()> {
             env_keys,
             description,
             timeout: Some(timeout),
-            cwd: None,
             bundled: None,
             available_tools: Vec::new(),
         },
@@ -1588,7 +1551,7 @@ pub async fn configure_tool_permissions_dialog() -> anyhow::Result<()> {
     let model: String = config
         .get_goose_model()
         .expect("No model configured. Please set model first");
-    let model_config = goose::model_config::model_config_from_user_config(&provider_name, &model)?;
+    let model_config = ModelConfig::new(&model)?.with_canonical_limits(&provider_name);
 
     let agent = Agent::new();
 
@@ -1625,10 +1588,8 @@ pub async fn configure_tool_permissions_dialog() -> anyhow::Result<()> {
     }
 
     let extensions = extension_config.into_iter().collect::<Vec<_>>();
-    let new_provider = create(&provider_name, extensions).await?;
-    agent
-        .update_provider(new_provider, model_config, &session.id)
-        .await?;
+    let new_provider = create(&provider_name, model_config, extensions).await?;
+    agent.update_provider(new_provider, &session.id).await?;
 
     let permission_manager = PermissionManager::instance();
     let selected_tools = agent
@@ -1810,21 +1771,22 @@ pub async fn handle_openrouter_auth() -> anyhow::Result<()> {
     // Test configuration - get the model that was configured
     println!("\nTesting configuration...");
     let configured_model: String = config.get_goose_model()?;
-    let model_config =
-        match goose::model_config::model_config_from_user_config("openrouter", &configured_model) {
-            Ok(config) => config,
-            Err(e) => {
-                eprintln!("⚠️  Invalid model configuration: {}", e);
-                eprintln!("Your settings have been saved. Please check your model configuration.");
-                return Ok(());
-            }
-        };
+    let model_config = match goose::model::ModelConfig::new(&configured_model) {
+        Ok(config) => config.with_canonical_limits("openrouter"),
+        Err(e) => {
+            eprintln!("⚠️  Invalid model configuration: {}", e);
+            eprintln!("Your settings have been saved. Please check your model configuration.");
+            return Ok(());
+        }
+    };
 
-    match create("openrouter", Vec::new()).await {
+    match create("openrouter", model_config, Vec::new()).await {
         Ok(provider) => {
+            let provider_model_config = provider.get_model_config();
             let test_result = provider
                 .complete(
-                    &model_config,
+                    &provider_model_config,
+                    "",
                     "You are goose, an AI assistant.",
                     &[Message::user().with_text("Say 'Configuration test successful!'")],
                     &[],
@@ -1890,14 +1852,16 @@ pub async fn handle_tetrate_auth() -> anyhow::Result<()> {
     // Test configuration
     println!("\nTesting configuration...");
     let configured_model: String = config.get_goose_model()?;
-    if let Err(e) = goose::model_config::model_config_from_user_config("tetrate", &configured_model)
-    {
-        eprintln!("⚠️  Invalid model configuration: {}", e);
-        eprintln!("Your settings have been saved. Please check your model configuration.");
-        return Ok(());
-    }
+    let model_config = match goose::model::ModelConfig::new(&configured_model) {
+        Ok(config) => config.with_canonical_limits("tetrate"),
+        Err(e) => {
+            eprintln!("⚠️  Invalid model configuration: {}", e);
+            eprintln!("Your settings have been saved. Please check your model configuration.");
+            return Ok(());
+        }
+    };
 
-    match create("tetrate", Vec::new()).await {
+    match create("tetrate", model_config, Vec::new()).await {
         Ok(provider) => {
             let test_result = provider.fetch_supported_models().await;
 

crates/goose-cli/src/commands/info.rs

@@ -3,8 +3,8 @@ use console::style;
 use goose::config::paths::Paths;
 use goose::config::Config;
 use goose::conversation::message::Message;
+use goose::providers::errors::ProviderError;
 use goose::session::session_manager::{DB_NAME, SESSIONS_FOLDER};
-use goose_providers::errors::ProviderError;
 use serde_yaml;
 use std::time::Duration;
 
@@ -73,10 +73,11 @@ async fn check_provider(
         }
     };
 
-    let model_config = goose::model_config::model_config_from_user_config(&provider, &model)
-        .map_err(|e| ProviderCheckError::InvalidModel(e.to_string()))?;
+    let model_config = goose::model::ModelConfig::new(&model)
+        .map_err(|e| ProviderCheckError::InvalidModel(e.to_string()))?
+        .with_canonical_limits(&provider);
 
-    let provider_client = goose::providers::create(&provider, Vec::new())
+    let provider_client = goose::providers::create(&provider, model_config, Vec::new())
         .await
         .map_err(|e| {
             let error = e.to_string();
@@ -87,13 +88,12 @@ async fn check_provider(
         })?;
 
     let test_msg = Message::user().with_text("Say 'ok'");
+    let model_config = provider_client.get_model_config();
     let start = std::time::Instant::now();
-    goose::session_context::with_session_id(
-        Some("check".to_string()),
-        provider_client.complete(&model_config, "", &[test_msg], &[]),
-    )
-    .await
-    .map_err(ProviderCheckError::ProviderRequest)?;
+    provider_client
+        .complete(&model_config, "check", "", &[test_msg], &[])
+        .await
+        .map_err(ProviderCheckError::ProviderRequest)?;
 
     Ok(ProviderCheckSuccess {
         provider,

crates/goose-cli/src/commands/mod.rs

@@ -8,9 +8,7 @@ pub mod recipe;
 pub mod review;
 pub mod schedule;
 pub mod session;
-pub mod skills;
 pub mod term;
-#[cfg(feature = "tui")]
 pub mod tui;
 #[cfg(feature = "update")]
 pub mod update;

crates/goose-cli/src/commands/project.rs

@@ -36,7 +36,7 @@ pub fn handle_project_default() -> Result<()> {
     }
 
     // Sort projects by last_accessed (newest first)
-    projects.sort_by_key(|project| std::cmp::Reverse(project.last_accessed));
+    projects.sort_by(|a, b| b.last_accessed.cmp(&a.last_accessed));
 
     // Get the most recent project
     let project = &projects[0];
@@ -178,7 +178,7 @@ pub fn handle_projects_interactive() -> Result<()> {
     }
 
     // Sort projects by last_accessed (newest first)
-    projects.sort_by_key(|project| std::cmp::Reverse(project.last_accessed));
+    projects.sort_by(|a, b| b.last_accessed.cmp(&a.last_accessed));
 
     // Format project paths for display
     let project_choices: Vec<(String, String)> = projects

crates/goose-cli/src/commands/review/handler.rs

@@ -29,9 +29,7 @@ pub struct ReviewOptions {
     /// Force every discovered check to run with this model, regardless of
     /// the check's own `model:` field.
     pub override_model: Option<String>,
-    /// Default `turn-limit` for orchestrated main-pass subprocesses and for
-    /// checks that do not declare their own. Does not cap the legacy
-    /// `--no-orchestrate` in-process main agent.
+    /// Default `turn-limit` applied to checks that do not declare their own.
     pub default_turn_limit: Option<usize>,
     /// Print the assembled prompt and discovered checks instead of dispatching
     /// the review.

crates/goose-cli/src/commands/review/orchestrator.rs

@@ -12,8 +12,7 @@
 //!
 //! - One subprocess per check (`goose run -q -t <prompt>`)
 //! - Concurrency capped at [`MAX_WORKERS`] via a Tokio semaphore
-//! - Per-subprocess turn limit via `--max-turns` (see
-//!   [`resolve_main_turn_limit`] and [`Check::resolved_turn_limit`])
+//! - Per-check timeout of [`CHECK_TIMEOUT_SECS`]
 //! - Each check is given a strict, tool-free prompt and is required to
 //!   return only `{"findings": [...]}` JSON
 //! - Findings are tagged with the originating `check` name in Rust, not
@@ -32,19 +31,27 @@ use anyhow::{Context, Result};
 use serde::{Deserialize, Serialize};
 use std::process::Stdio;
 use std::sync::Arc;
+use std::time::Duration;
 use tokio::io::AsyncWriteExt;
 use tokio::process::Command;
 use tokio::sync::Semaphore;
 use tokio::task::JoinSet;
+use tokio::time::timeout;
 
 use super::handler::ReviewOptions;
-use goose::checks::{Check, DEFAULT_CHECK_TURN_LIMIT};
+use goose::checks::Check;
 
 /// Maximum number of check subprocesses we run concurrently. 4 is
 /// empirically the sweet spot before LLM-side rate limits and local
 /// resource contention start hurting wall-clock.
 pub const MAX_WORKERS: usize = 4;
 
+/// Hard wall-clock cap for a single check subprocess. A check that
+/// takes longer than this is almost always stuck in a tool-call loop
+/// or a retry storm; we'd rather surface the timeout than block the
+/// whole review.
+pub const CHECK_TIMEOUT_SECS: u64 = 5 * 60;
+
 /// One review finding emitted by a check or by the main correctness
 /// pass.
 #[derive(Debug, Clone, Serialize, Deserialize)]
@@ -80,7 +87,7 @@ struct RawFinding {
 /// Run all discovered checks concurrently as `goose run` subprocesses.
 ///
 /// Returns one `Vec<Finding>` per check, in the same order as `checks`.
-/// A failed check (subprocess error, turn-limit exhaustion, malformed JSON) yields an
+/// A failed check (subprocess error, timeout, malformed JSON) yields an
 /// empty findings list and a warning on stderr; a single broken check
 /// must never block the rest of the review.
 pub async fn run_checks_in_parallel(
@@ -179,14 +186,6 @@ fn resolve_check_model(check: &Check, opts: &ReviewOptions) -> Option<String> {
     opts.default_model.clone()
 }
 
-/// Resolve the turn limit for a main-pass subprocess.
-///
-/// Uses `goose review --turn-limit` when set, otherwise
-/// [`DEFAULT_CHECK_TURN_LIMIT`].
-fn resolve_main_turn_limit(default_turn_limit: Option<usize>) -> usize {
-    default_turn_limit.unwrap_or(DEFAULT_CHECK_TURN_LIMIT)
-}
-
 /// Spawn a single `goose run` subprocess for one check and parse its
 /// output into [`Finding`]s.
 async fn run_single_check_subprocess(
@@ -197,8 +196,7 @@ async fn run_single_check_subprocess(
     instructions: Option<&str>,
     max_turns: Option<usize>,
 ) -> Result<Vec<Finding>> {
-    let turns = max_turns.expect("check subprocess always has a resolved turn limit");
-    let prompt = build_check_prompt(check, diff, instructions, turns);
+    let prompt = build_check_prompt(check, diff, instructions);
     let raw = run_subprocess_for_findings(
         &prompt,
         &format!("check '{}'", check.name),
@@ -224,7 +222,8 @@ async fn run_single_check_subprocess(
 /// Generic `goose run` subprocess that hands a prompt to the model
 /// and parses `{"findings": [...]}` JSON out of the response. Shared
 /// by the per-check and per-file main-pass orchestrators so both get
-/// the same robust JSON extraction and error reporting.
+/// the same robust JSON extraction, timeout handling, and error
+/// reporting.
 async fn run_subprocess_for_findings(
     prompt: &str,
     label: &str,
@@ -244,8 +243,11 @@ async fn run_subprocess_for_findings(
         .stdin(Stdio::piped())
         .stdout(Stdio::piped())
         .stderr(Stdio::piped())
-        // If this future is dropped, kill the child so it does not keep
-        // running (and racking up tokens) in the background.
+        // Drop-on-cancel safety: when the outer `timeout` fires it drops
+        // this future, which drops the Child handle. Without
+        // kill_on_drop, the Tokio runtime leaves the subprocess running
+        // (and racking up tokens) in the background — kill_on_drop sends
+        // SIGKILL on Drop instead.
         .kill_on_drop(true);
 
     if let Some(p) = provider {
@@ -271,10 +273,13 @@ async fn run_subprocess_for_findings(
         drop(stdin);
     }
 
-    let output = child
-        .wait_with_output()
-        .await
-        .with_context(|| format!("wait on {label}"))?;
+    let wait = child.wait_with_output();
+    let output = match timeout(Duration::from_secs(CHECK_TIMEOUT_SECS), wait).await {
+        Ok(o) => o.with_context(|| format!("wait on {label}"))?,
+        Err(_) => {
+            anyhow::bail!("{label} timed out after {}s", CHECK_TIMEOUT_SECS);
+        }
+    };
 
     if !output.status.success() {
         let stderr = String::from_utf8_lossy(&output.stderr);
@@ -316,7 +321,6 @@ pub async fn run_main_pass_in_parallel(
 
     let semaphore = Arc::new(Semaphore::new(MAX_WORKERS));
     let mut set: JoinSet<(usize, String, Result<Vec<RawFinding>>, bool)> = JoinSet::new();
-    let max_turns = resolve_main_turn_limit(opts.default_turn_limit);
 
     for (idx, (path, file_diff)) in per_file.iter().enumerate() {
         let sem = semaphore.clone();
@@ -330,20 +334,15 @@ pub async fn run_main_pass_in_parallel(
 
         set.spawn(async move {
             let _permit = sem.acquire().await.expect("semaphore is never closed");
-            let prompt = build_main_pass_prompt(
-                &path,
-                &file_diff,
-                &base_prompt,
-                instructions.as_deref(),
-                max_turns,
-            );
+            let prompt =
+                build_main_pass_prompt(&path, &file_diff, &base_prompt, instructions.as_deref());
             let label = format!("main:{path}");
             let result = run_subprocess_for_findings(
                 &prompt,
                 &label,
                 provider.as_deref(),
                 model.as_deref(),
-                Some(max_turns),
+                None,
             )
             .await;
             (idx, path, result, quiet)
@@ -568,22 +567,6 @@ fn take_quoted(s: &str) -> Option<(String, &str)> {
     None
 }
 
-/// Prompt section telling review subprocesses about the `--max-turns`
-/// cap enforced by goose. Without this, models routinely burn turns on
-/// tool loops and return nothing when the limit stops the session.
-fn build_subprocess_turn_budget_section(max_turns: usize) -> String {
-    format!(
-        "## Turn budget\n\n\
-         You may take at most {max_turns} agent turns (model/tool iterations) in this run. \
-         goose enforces this via `--max-turns`; when you exhaust it, the session stops and \
-         any findings not yet emitted as JSON are lost.\n\n\
-         Plan for the limit:\n\
-         - As turns run low, stop exploring and return JSON with the findings you have verified.\n\
-         - Always emit valid JSON (`{{\"findings\":[...]}}` or `{{\"findings\":[]}}`) before \
-           the turn limit — an empty or missing response counts as failure.\n\n"
-    )
-}
-
 /// Build the strict, JSON-only prompt sent to one main-pass
 /// subprocess. The base prompt (custom or
 /// [`DEFAULT_REVIEW_PROMPT`]) supplies the reviewer voice; we then
@@ -594,7 +577,6 @@ fn build_main_pass_prompt(
     file_diff: &str,
     base_prompt: &str,
     instructions: Option<&str>,
-    max_turns: usize,
 ) -> String {
     let mut s = String::new();
     s.push_str(base_prompt.trim_end());
@@ -607,7 +589,6 @@ fn build_main_pass_prompt(
             s.push_str("\n\n");
         }
     }
-    s.push_str(&build_subprocess_turn_budget_section(max_turns));
     s.push_str("## File under review\n\n");
     s.push_str(&format!("Path: `{path}`\n\n"));
     s.push_str(
@@ -630,12 +611,7 @@ fn build_main_pass_prompt(
 /// Shape matches the prompt format Amp-authored checks already expect,
 /// so a check written for `amp review` runs the same way under
 /// `goose review`.
-fn build_check_prompt(
-    check: &Check,
-    diff: &str,
-    instructions: Option<&str>,
-    max_turns: usize,
-) -> String {
+fn build_check_prompt(check: &Check, diff: &str, instructions: Option<&str>) -> String {
     let mut s = String::new();
     s.push_str("You are running an automated code review check.\n\n");
     s.push_str(&format!("Check name: {}\n", check.name));
@@ -657,9 +633,7 @@ fn build_check_prompt(
             s.push('\n');
         }
     }
-    s.push('\n');
-    s.push_str(&build_subprocess_turn_budget_section(max_turns));
-    s.push_str("Review ONLY the git diff provided below.\n");
+    s.push_str("\nReview ONLY the git diff provided below.\n");
     s.push_str("Do not ask for missing context.\n");
     s.push_str("Use repo-relative file paths.\n");
     s.push_str("Use post-change line numbers from the diff.\n");
@@ -846,7 +820,7 @@ mod tests {
 
     #[test]
     fn check_prompt_is_strict_and_diff_aware() {
-        let p = build_check_prompt(&ck("perf"), "diff content", None, DEFAULT_CHECK_TURN_LIMIT);
+        let p = build_check_prompt(&ck("perf"), "diff content", None);
         assert!(p.contains("automated code review check"));
         assert!(p.contains("Check name: perf"));
         assert!(p.contains("```diff\ndiff content\n```"));
@@ -859,7 +833,7 @@ mod tests {
     fn check_prompt_restricts_findings_to_added_or_modified_lines() {
         // Mirrors Amp's prompt language; without these the model
         // happily flags pre-existing code shown for context.
-        let p = build_check_prompt(&ck("perf"), "diff content", None, DEFAULT_CHECK_TURN_LIMIT);
+        let p = build_check_prompt(&ck("perf"), "diff content", None);
         assert!(p.contains("ONLY in the changed lines"));
         assert!(p.contains("lines beginning with `+`"));
         assert!(p.contains("ONLY for code that was added or modified"));
@@ -872,7 +846,6 @@ mod tests {
             &ck("perf"),
             "diff content",
             Some("This is a refactor; flag any behavior change."),
-            DEFAULT_CHECK_TURN_LIMIT,
         );
         assert!(p.contains("Reviewer instructions:"));
         assert!(p.contains("flag any behavior change"));
@@ -880,29 +853,10 @@ mod tests {
 
     #[test]
     fn check_prompt_skips_blank_reviewer_instructions() {
-        let p = build_check_prompt(
-            &ck("perf"),
-            "diff content",
-            Some("   \n  "),
-            DEFAULT_CHECK_TURN_LIMIT,
-        );
+        let p = build_check_prompt(&ck("perf"), "diff content", Some("   \n  "));
         assert!(!p.contains("Reviewer instructions"));
     }
 
-    #[test]
-    fn check_prompt_includes_turn_budget() {
-        let p = build_check_prompt(&ck("perf"), "diff content", None, 12);
-        assert!(p.contains("## Turn budget"));
-        assert!(p.contains("at most 12 agent turns"));
-        assert!(p.contains("--max-turns"));
-    }
-
-    #[test]
-    fn resolve_main_turn_limit_uses_cli_default_or_fallback() {
-        assert_eq!(resolve_main_turn_limit(Some(40)), 40);
-        assert_eq!(resolve_main_turn_limit(None), DEFAULT_CHECK_TURN_LIMIT);
-    }
-
     #[test]
     fn parse_findings_accepts_bare_json() {
         let raw = r#"{"findings":[{"severity":"high","path":"a.py","line_start":1,"line_end":2,"summary":"bad"}]}"#;
@@ -1136,7 +1090,6 @@ rename to new/name.rs
             "diff --git a/src/foo.rs b/src/foo.rs\n@@ -1 +1 @@\n-old\n+new\n",
             "BASE PROMPT",
             None,
-            DEFAULT_CHECK_TURN_LIMIT,
         );
         assert!(p.starts_with("BASE PROMPT"));
         assert!(p.contains("Path: `src/foo.rs`"));
@@ -1155,7 +1108,6 @@ rename to new/name.rs
             "diff body",
             "BASE",
             Some("PR is a refactor; flag behavior changes."),
-            DEFAULT_CHECK_TURN_LIMIT,
         );
         assert!(p.contains("## Reviewer instructions"));
         assert!(p.contains("flag behavior changes"));
@@ -1163,20 +1115,7 @@ rename to new/name.rs
 
     #[test]
     fn main_pass_prompt_skips_blank_reviewer_instructions() {
-        let p = build_main_pass_prompt(
-            "src/foo.rs",
-            "diff body",
-            "BASE",
-            Some("   \n  \t\n"),
-            DEFAULT_CHECK_TURN_LIMIT,
-        );
+        let p = build_main_pass_prompt("src/foo.rs", "diff body", "BASE", Some("   \n  \t\n"));
         assert!(!p.contains("Reviewer instructions"));
     }
-
-    #[test]
-    fn main_pass_prompt_includes_turn_budget() {
-        let p = build_main_pass_prompt("src/foo.rs", "diff body", "BASE", None, 18);
-        assert!(p.contains("## Turn budget"));
-        assert!(p.contains("at most 18 agent turns"));
-    }
 }

crates/goose-cli/src/commands/schedule.rs

@@ -217,11 +217,11 @@ pub async fn handle_schedule_sessions(schedule_id: String, limit: Option<usize>)
                 println!("No sessions found for schedule ID '{}'.", schedule_id);
             } else {
                 println!("Sessions for schedule ID '{}':", schedule_id);
+                // sessions is now Vec<(String, SessionMetadata)>
                 for (session_name, metadata) in sessions {
                     println!(
-                        "  - Session ID: {}, Messages: {}, Working Dir: {}, Description: \"{}\", Schedule ID: {:?}",
-                        session_name,
-                        metadata.message_count,
+                        "  - Session ID: {}, Working Dir: {}, Description: \"{}\", Schedule ID: {:?}",
+                        session_name, // Display the session_name as Session ID
                         metadata.working_dir.display(),
                         metadata.name,
                         metadata.schedule_id.as_deref().unwrap_or("N/A")

crates/goose-cli/src/commands/session.rs

@@ -7,9 +7,7 @@ use etcetera::home_dir;
 use goose::config::Config;
 #[cfg(feature = "nostr")]
 use goose::session::nostr_share;
-use goose::session::{
-    generate_diagnostics, DiagnosticsLevel, Session, SessionManager, SessionType,
-};
+use goose::session::{generate_diagnostics, Session, SessionManager, SessionType};
 use goose::utils::safe_truncate;
 use regex::Regex;
 use std::fs;
@@ -70,8 +68,7 @@ fn prompt_interactive_session_removal(sessions: &[Session]) -> Result<Vec<Sessio
                 &s.name
             };
             let truncated_desc = safe_truncate(desc, TRUNCATED_DESC_LENGTH);
-            let display_text =
-                format!("{} - {} ({})", session_activity_at(s), truncated_desc, s.id);
+            let display_text = format!("{} - {} ({})", s.updated_at, truncated_desc, s.id);
             (display_text, s.clone())
         })
         .collect();
@@ -151,10 +148,6 @@ fn write_line_or_broken_pipe_ok<W: Write>(out: &mut W, line: &str) -> Result<boo
     }
 }
 
-fn session_activity_at(session: &Session) -> chrono::DateTime<chrono::Utc> {
-    session.last_message_at.unwrap_or(session.updated_at)
-}
-
 pub async fn handle_session_list(
     format: String,
     ascending: bool,
@@ -175,9 +168,9 @@ pub async fn handle_session_list(
     }
 
     if ascending {
-        sessions.sort_by_key(session_activity_at);
+        sessions.sort_by(|a, b| a.updated_at.cmp(&b.updated_at));
     } else {
-        sessions.sort_by_key(|b| std::cmp::Reverse(session_activity_at(b)));
+        sessions.sort_by(|a, b| b.updated_at.cmp(&a.updated_at));
     }
 
     if let Some(n) = limit {
@@ -211,7 +204,7 @@ pub async fn handle_session_list(
                     "{} - {} - {} - {}",
                     session.id,
                     session.name,
-                    session_activity_at(&session),
+                    session.updated_at,
                     display_path_with_tilde(&session.working_dir)
                 );
                 if !write_line_or_broken_pipe_ok(&mut out, &output)? {
@@ -307,15 +300,6 @@ pub async fn handle_session_import(input: String, nostr: bool) -> Result<()> {
             .with_context(|| format!("Failed to read session import file: {input}"))?
     };
 
-    let format = goose::session::import_formats::detect_format(&json);
-    let label = match format {
-        goose::session::import_formats::ImportFormat::Goose => "goose",
-        goose::session::import_formats::ImportFormat::ClaudeCode => "Claude Code",
-        goose::session::import_formats::ImportFormat::Codex => "Codex",
-        goose::session::import_formats::ImportFormat::Pi => "Pi",
-    };
-    println!("Detected format: {}", label);
-
     let session_manager = SessionManager::instance();
     let session = session_manager
         .import_session(&json, Some(SessionType::User))
@@ -329,27 +313,24 @@ pub async fn handle_session_import(input: String, nostr: bool) -> Result<()> {
 
 pub async fn handle_diagnostics(session_id: &str, output_path: Option<PathBuf>) -> Result<()> {
     println!(
-        "Generating diagnostics report for session '{}'...",
+        "Generating diagnostics bundle for session '{}'...",
         session_id
     );
 
     let session_manager = SessionManager::instance();
-    let diagnostics_report =
-        generate_diagnostics(&session_manager, session_id, DiagnosticsLevel::Full)
-            .await
-            .with_context(|| {
-                format!(
-                    "Failed to generate diagnostics report for session '{}'",
-                    session_id
-                )
-            })?;
-    let diagnostics_data = serde_json::to_vec_pretty(&diagnostics_report)
-        .context("Failed to serialize diagnostics report")?;
+    let diagnostics_data = generate_diagnostics(&session_manager, session_id)
+        .await
+        .with_context(|| {
+            format!(
+                "Failed to write to generate diagnostics bundle for session '{}'",
+                session_id
+            )
+        })?;
 
     let output_file = if let Some(path) = output_path {
         path.clone()
     } else {
-        PathBuf::from(format!("diagnostics_{}.json", session_id))
+        PathBuf::from(format!("diagnostics_{}.zip", session_id))
     };
 
     let mut file = fs::File::create(&output_file).context(format!(
@@ -360,7 +341,7 @@ pub async fn handle_diagnostics(session_id: &str, output_path: Option<PathBuf>)
     file.write_all(&diagnostics_data)
         .context("Failed to write diagnostics data")?;
 
-    println!("Diagnostics report saved to: {}", output_file.display());
+    println!("Diagnostics bundle saved to: {}", output_file.display());
 
     Ok(())
 }

crates/goose-cli/src/commands/skills.rs

@@ -1,228 +0,0 @@
-use anyhow::Result;
-use console::{measure_text_width, Term};
-use goose::skills::list_installed_skills;
-use goose::token_counter::create_token_counter;
-
-const DESCRIPTION_PREVIEW_CHARS: usize = 50;
-const SEPARATOR: &str = " | ";
-const MIN_DESCRIPTION_WIDTH: usize = 4;
-const MIN_LOCATION_WIDTH: usize = 4;
-const NAME_HEADER: &str = "Name";
-const DESCRIPTION_HEADER: &str = "Description";
-const DESCRIPTION_TOKENS_HEADER: &str = "Description tokens";
-const CONTENT_TOKENS_HEADER: &str = "Content tokens";
-const LOCATION_HEADER: &str = "Location";
-
-struct SkillRow {
-    name: String,
-    description: String,
-    description_tokens: String,
-    content_tokens: String,
-    location: String,
-}
-
-struct ColumnWidths {
-    name: usize,
-    description: usize,
-    description_tokens: usize,
-    content_tokens: usize,
-    location: usize,
-}
-
-pub async fn handle_skills_list() -> Result<()> {
-    let cwd = std::env::current_dir()?;
-    let terminal_width = terminal_width();
-    let token_counter = create_token_counter().await.map_err(anyhow::Error::msg)?;
-    let mut skills = list_installed_skills(Some(&cwd));
-    skills.sort_by(|a, b| a.name.cmp(&b.name));
-
-    let rows = skills
-        .iter()
-        .map(|skill| SkillRow {
-            name: skill.name.clone(),
-            description: description_preview(&skill.description),
-            description_tokens: token_counter.count_tokens(&skill.description).to_string(),
-            content_tokens: token_counter.count_tokens(&skill.content).to_string(),
-            location: skill.path.clone(),
-        })
-        .collect::<Vec<_>>();
-    let widths = column_widths(&rows, terminal_width);
-
-    println!("{}", header_line(&widths, terminal_width));
-    for row in rows {
-        println!("{}", skill_line(&row, &widths, terminal_width));
-    }
-
-    Ok(())
-}
-
-fn terminal_width() -> Option<usize> {
-    Term::stdout()
-        .size_checked()
-        .map(|(_height, width)| width as usize)
-}
-
-fn column_widths(rows: &[SkillRow], max_display_width: Option<usize>) -> ColumnWidths {
-    let description_tokens = max_width(
-        DESCRIPTION_TOKENS_HEADER,
-        rows.iter().map(|row| row.description_tokens.as_str()),
-    );
-    let content_tokens = max_width(
-        CONTENT_TOKENS_HEADER,
-        rows.iter().map(|row| row.content_tokens.as_str()),
-    );
-    let longest_name = max_width(NAME_HEADER, rows.iter().map(|row| row.name.as_str()));
-    let description = max_width(
-        DESCRIPTION_HEADER,
-        rows.iter().map(|row| row.description.as_str()),
-    );
-    let location = max_width(
-        LOCATION_HEADER,
-        rows.iter().map(|row| row.location.as_str()),
-    );
-
-    let Some(width) = max_display_width else {
-        return ColumnWidths {
-            name: longest_name,
-            description,
-            description_tokens,
-            content_tokens,
-            location,
-        };
-    };
-
-    let separator_width = measure_text_width(SEPARATOR) * 4;
-    let available_width = width.saturating_sub(separator_width);
-    let dynamic_width = available_width.saturating_sub(description_tokens + content_tokens);
-
-    let name =
-        longest_name.min(dynamic_width.saturating_sub(MIN_DESCRIPTION_WIDTH + MIN_LOCATION_WIDTH));
-    let remaining_after_name = dynamic_width.saturating_sub(name);
-    let description = description.min(remaining_after_name.saturating_sub(MIN_LOCATION_WIDTH));
-    let remaining_after_description = remaining_after_name.saturating_sub(description);
-    let location = location.min(remaining_after_description);
-
-    ColumnWidths {
-        name,
-        description,
-        description_tokens,
-        content_tokens,
-        location,
-    }
-}
-
-fn max_width<'a>(header: &str, values: impl Iterator<Item = &'a str>) -> usize {
-    values
-        .map(measure_text_width)
-        .chain(std::iter::once(measure_text_width(header)))
-        .max()
-        .unwrap_or(0)
-}
-
-fn header_line(widths: &ColumnWidths, max_display_width: Option<usize>) -> String {
-    let line = format_line(
-        NAME_HEADER,
-        DESCRIPTION_HEADER,
-        DESCRIPTION_TOKENS_HEADER,
-        CONTENT_TOKENS_HEADER,
-        LOCATION_HEADER,
-        widths,
-    );
-
-    match max_display_width {
-        Some(width) => truncate_to_display_width(&line, width),
-        None => line,
-    }
-}
-
-fn skill_line(row: &SkillRow, widths: &ColumnWidths, max_display_width: Option<usize>) -> String {
-    let line = format_line(
-        &row.name,
-        &row.description,
-        &row.description_tokens,
-        &row.content_tokens,
-        &row.location,
-        widths,
-    );
-
-    match max_display_width {
-        Some(width) => truncate_to_display_width(&line, width),
-        None => line,
-    }
-}
-
-fn format_line(
-    name: &str,
-    description: &str,
-    description_tokens: &str,
-    content_tokens: &str,
-    location: &str,
-    widths: &ColumnWidths,
-) -> String {
-    format!(
-        "{}{}{}{}{}{}{}{}{}",
-        pad_to_display_width(&truncate_to_display_width(name, widths.name), widths.name),
-        SEPARATOR,
-        pad_to_display_width(
-            &truncate_to_display_width(description, widths.description),
-            widths.description
-        ),
-        SEPARATOR,
-        pad_to_display_width(description_tokens, widths.description_tokens),
-        SEPARATOR,
-        pad_to_display_width(content_tokens, widths.content_tokens),
-        SEPARATOR,
-        pad_to_display_width(
-            &truncate_to_display_width(location, widths.location),
-            widths.location
-        ),
-    )
-}
-
-fn description_preview(description: &str) -> String {
-    let normalized = description.split_whitespace().collect::<Vec<_>>().join(" ");
-    truncate_to_chars(&normalized, DESCRIPTION_PREVIEW_CHARS)
-}
-
-fn truncate_to_chars(text: &str, max_chars: usize) -> String {
-    if text.chars().count() <= max_chars {
-        return text.to_string();
-    }
-
-    if max_chars <= 3 {
-        return ".".repeat(max_chars);
-    }
-
-    let mut output = text.chars().take(max_chars - 3).collect::<String>();
-    output.push_str("...");
-    output
-}
-
-fn truncate_to_display_width(text: &str, max_width: usize) -> String {
-    if measure_text_width(text) <= max_width {
-        return text.to_string();
-    }
-
-    if max_width <= 3 {
-        return ".".repeat(max_width);
-    }
-
-    let mut output = String::new();
-    let suffix_width = measure_text_width("...");
-
-    for ch in text.chars() {
-        output.push(ch);
-        if measure_text_width(&output) + suffix_width > max_width {
-            output.pop();
-            break;
-        }
-    }
-
-    output.push_str("...");
-    output
-}
-
-fn pad_to_display_width(text: &str, width: usize) -> String {
-    let padding = width.saturating_sub(measure_text_width(text));
-    format!("{}{}", text, " ".repeat(padding))
-}

crates/goose-cli/src/commands/term.rs

@@ -280,15 +280,9 @@ pub async fn handle_term_run(prompt: Vec<String>) -> Result<()> {
         };
 
     if let Some(oldest_user) = user_messages_after_last_assistant.last() {
-        if let Some(message_id) = oldest_user.id.as_deref() {
-            session_manager
-                .truncate_conversation_from_message(&session_id, message_id)
-                .await?;
-        } else {
-            session_manager
-                .truncate_conversation(&session_id, oldest_user.created)
-                .await?;
-        }
+        session_manager
+            .truncate_conversation(&session_id, oldest_user.created)
+            .await?;
     }
 
     let prompt_with_context = if user_messages_after_last_assistant.is_empty() {
@@ -330,10 +324,7 @@ pub async fn handle_term_info() -> Result<()> {
 
     let session_manager = SessionManager::instance();
     let session = session_manager.get_session(&session_id, false).await.ok();
-    let total_tokens = session
-        .as_ref()
-        .and_then(|s| s.usage.total_tokens)
-        .unwrap_or(0) as usize;
+    let total_tokens = session.as_ref().and_then(|s| s.total_tokens).unwrap_or(0) as usize;
 
     let config = goose::config::Config::global();
     let model_name = config
@@ -354,7 +345,9 @@ pub async fn handle_term_info() -> Result<()> {
         .ok()
         .and_then(|model_name| {
             config.get_goose_provider().ok().and_then(|provider_name| {
-                goose::model_config::model_config_from_user_config(&provider_name, &model_name).ok()
+                goose::model::ModelConfig::new(&model_name)
+                    .ok()
+                    .map(|c| c.with_canonical_limits(&provider_name))
             })
         })
         .map(|mc| mc.context_limit())

crates/goose-cli/src/commands/update.rs

@@ -1,8 +1,4 @@
 use anyhow::{bail, Context, Result};
-use reqwest::{
-    header::{HeaderValue, AUTHORIZATION},
-    StatusCode,
-};
 use sha2::{Digest, Sha256};
 use sigstore_verify::trust_root::{TrustedRoot, SIGSTORE_PRODUCTION_TRUSTED_ROOT};
 use sigstore_verify::types::{Bundle, Sha256Hash};
@@ -68,7 +64,7 @@ fn binary_name() -> &'static str {
 fn sha256_hex(data: &[u8]) -> String {
     let mut hasher = Sha256::new();
     hasher.update(data);
-    goose::utils::bytes_to_hex(hasher.finalize())
+    format!("{:x}", hasher.finalize())
 }
 
 #[derive(serde::Deserialize)]
@@ -83,32 +79,6 @@ struct AttestationEntry {
 
 const GITHUB_ACTIONS_ISSUER: &str = "https://token.actions.githubusercontent.com";
 
-fn sanitized_token(token: Option<&str>) -> Option<&str> {
-    token.map(str::trim).filter(|tok| !tok.is_empty())
-}
-
-fn authorization_header_value(token: &str) -> Option<HeaderValue> {
-    HeaderValue::from_str(&format!("Bearer {token}")).ok()
-}
-
-fn github_token() -> Option<String> {
-    env::var("GITHUB_TOKEN")
-        .ok()
-        .and_then(|tok| sanitized_token(Some(&tok)).map(str::to_owned))
-        .or_else(|| {
-            env::var("GH_TOKEN")
-                .ok()
-                .and_then(|tok| sanitized_token(Some(&tok)).map(str::to_owned))
-        })
-}
-
-fn should_retry_attestations_without_token(status: StatusCode, token: Option<&str>) -> bool {
-    sanitized_token(token)
-        .and_then(authorization_header_value)
-        .is_some()
-        && matches!(status, StatusCode::UNAUTHORIZED | StatusCode::FORBIDDEN)
-}
-
 async fn fetch_attestations(digest: &str, token: Option<&str>) -> Result<Vec<serde_json::Value>> {
     let url = format!(
         "https://api.github.com/repos/aaif-goose/goose/attestations/sha256:{digest}\
@@ -116,14 +86,17 @@ async fn fetch_attestations(digest: &str, token: Option<&str>) -> Result<Vec<ser
     );
 
     let client = reqwest::Client::new();
-    let token = sanitized_token(token);
-    let resp = fetch_attestations_response(&client, &url, token).await?;
+    let mut req = client
+        .get(&url)
+        .header("Accept", "application/vnd.github+json")
+        .header("X-GitHub-Api-Version", "2022-11-28")
+        .header("User-Agent", "goose-cli");
 
-    let resp = if should_retry_attestations_without_token(resp.status(), token) {
-        fetch_attestations_response(&client, &url, None).await?
-    } else {
-        resp
-    };
+    if let Some(tok) = token {
+        req = req.header("Authorization", format!("Bearer {tok}"));
+    }
+
+    let resp = req.send().await.context("Failed to fetch attestations")?;
 
     if !resp.status().is_success() {
         bail!("GitHub attestation API returned HTTP {}", resp.status());
@@ -137,24 +110,6 @@ async fn fetch_attestations(digest: &str, token: Option<&str>) -> Result<Vec<ser
     Ok(body.attestations.into_iter().map(|a| a.bundle).collect())
 }
 
-async fn fetch_attestations_response(
-    client: &reqwest::Client,
-    url: &str,
-    token: Option<&str>,
-) -> Result<reqwest::Response> {
-    let mut req = client
-        .get(url)
-        .header("Accept", "application/vnd.github+json")
-        .header("X-GitHub-Api-Version", "2022-11-28")
-        .header("User-Agent", "goose-cli");
-
-    if let Some(value) = token.and_then(authorization_header_value) {
-        req = req.header(AUTHORIZATION, value);
-    }
-
-    req.send().await.context("Failed to fetch attestations")
-}
-
 // Verify a single attestation bundle against the artifact digest and workflow.
 fn verify_bundle(
     bundle_json: &serde_json::Value,
@@ -187,8 +142,8 @@ fn verify_bundle(
     Ok(())
 }
 
-/// Returns `Ok(())` when the downloaded archive has verified provenance.
-async fn verify_provenance(archive_data: &[u8], tag: &str) -> Result<()> {
+/// Returns `Ok(true)` verified, `Ok(false)` skipped (soft warning), `Err` hard failure.
+async fn verify_provenance(archive_data: &[u8], tag: &str) -> Result<bool> {
     let digest = sha256_hex(archive_data);
     println!("Archive SHA-256: {digest}");
 
@@ -197,19 +152,30 @@ async fn verify_provenance(archive_data: &[u8], tag: &str) -> Result<()> {
         _ => "release.yml",
     };
 
-    let token = github_token();
+    let token = env::var("GITHUB_TOKEN")
+        .ok()
+        .or_else(|| env::var("GH_TOKEN").ok());
 
     println!("Verifying SLSA provenance via Sigstore...");
 
-    let bundles = fetch_attestations(&digest, token.as_deref())
-        .await
-        .context(
-            "Sigstore provenance check could not complete; refusing to install unverifiable update",
-        )?;
-
-    if bundles.is_empty() {
-        bail!("No Sigstore attestation found for downloaded archive; refusing to install unverifiable update");
-    }
+    let bundles = match fetch_attestations(&digest, token.as_deref()).await {
+        Ok(b) if b.is_empty() => {
+            eprintln!(
+                "Warning: No Sigstore attestation found for this build. \
+                 This may be expected for canary or nightly builds."
+            );
+            return Ok(false);
+        }
+        Ok(b) => b,
+        Err(e) => {
+            eprintln!(
+                "Warning: Sigstore provenance check could not complete: {e}\n\
+                 This may be expected for releases published before provenance \
+                 attestations were enabled."
+            );
+            return Ok(false);
+        }
+    };
 
     let trusted_root = TrustedRoot::from_json(SIGSTORE_PRODUCTION_TRUSTED_ROOT)
         .context("Failed to load Sigstore trusted root")?;
@@ -229,7 +195,7 @@ async fn verify_provenance(archive_data: &[u8], tag: &str) -> Result<()> {
         ) {
             Ok(()) => {
                 println!("Sigstore provenance verification passed.");
-                return Ok(());
+                return Ok(true);
             }
             Err(e) => last_err = Some(e),
         }
@@ -281,7 +247,7 @@ pub async fn update(canary: bool, reconfigure: bool) -> Result<()> {
         println!("Downloaded {} bytes.", bytes.len());
 
         // --- Verify SLSA provenance via Sigstore --------------------------------
-        verify_provenance(&bytes, tag).await?;
+        let provenance_verified = verify_provenance(&bytes, tag).await?;
 
         // --- Extract to temp dir (hardened against path traversal) --------------
         let tmp_dir = tempfile::tempdir().context("Failed to create temp directory")?;
@@ -308,7 +274,11 @@ pub async fn update(canary: bool, reconfigure: bool) -> Result<()> {
         #[cfg(target_os = "windows")]
         copy_dlls(&extracted_binary, &current_exe)?;
 
-        println!("goose updated successfully (verified with Sigstore SLSA provenance).");
+        if provenance_verified {
+            println!("goose updated successfully (verified with Sigstore SLSA provenance).");
+        } else {
+            println!("goose updated successfully.");
+        }
 
         // --- Reconfigure if requested -------------------------------------------
         if reconfigure {
@@ -767,48 +737,6 @@ mod tests {
         );
     }
 
-    #[test]
-    fn test_sanitized_token_trims_blank_values() {
-        assert_eq!(sanitized_token(None), None);
-        assert_eq!(sanitized_token(Some("")), None);
-        assert_eq!(sanitized_token(Some("   ")), None);
-        assert_eq!(sanitized_token(Some(" token\n")), Some("token"));
-    }
-
-    #[test]
-    fn test_authorization_header_value_rejects_malformed_tokens() {
-        assert!(authorization_header_value("token").is_some());
-        assert!(authorization_header_value("bad\ntoken").is_none());
-    }
-
-    #[test]
-    fn test_attestation_lookup_retries_auth_failures_without_token() {
-        assert!(should_retry_attestations_without_token(
-            StatusCode::UNAUTHORIZED,
-            Some("token")
-        ));
-        assert!(should_retry_attestations_without_token(
-            StatusCode::FORBIDDEN,
-            Some("token")
-        ));
-        assert!(!should_retry_attestations_without_token(
-            StatusCode::INTERNAL_SERVER_ERROR,
-            Some("token")
-        ));
-        assert!(!should_retry_attestations_without_token(
-            StatusCode::UNAUTHORIZED,
-            Some("")
-        ));
-        assert!(!should_retry_attestations_without_token(
-            StatusCode::UNAUTHORIZED,
-            Some("bad\ntoken")
-        ));
-        assert!(!should_retry_attestations_without_token(
-            StatusCode::UNAUTHORIZED,
-            None
-        ));
-    }
-
     // -----------------------------------------------------------------------
     // Path validation and extraction hardening tests
     // -----------------------------------------------------------------------
@@ -880,11 +808,13 @@ mod tests {
     // -----------------------------------------------------------------------
 
     #[tokio::test]
-    async fn test_verify_provenance_fails_closed_when_unverifiable() {
+    async fn test_verify_provenance_warns_on_missing_attestation() {
         let result = verify_provenance(b"not a real archive", "stable").await;
-        assert!(
-            result.is_err(),
-            "verify_provenance must fail closed when provenance cannot be verified"
+        // Network failures and missing attestations are soft warnings: Ok(false), not hard errors.
+        assert_eq!(
+            result.ok(),
+            Some(false),
+            "verify_provenance should return Ok(false) when attestations cannot be fetched"
         );
     }
 

crates/goose-cli/src/lib.rs

@@ -1,5 +1,3 @@
-#![recursion_limit = "256"]
-
 #[cfg(not(any(feature = "rustls-tls", feature = "native-tls")))]
 compile_error!("At least one of `rustls-tls` or `native-tls` features must be enabled");
 

crates/goose-cli/src/logging.rs

@@ -1,31 +1,53 @@
 use anyhow::Result;
-use goose::providers::utils::init_goose_request_log;
-use std::sync::OnceLock;
+use std::sync::Once;
 
 // Used to ensure we only set up tracing once
-static INIT: OnceLock<Result<()>> = OnceLock::new();
+static INIT: Once = Once::new();
 
 /// Sets up the logging infrastructure for the CLI.
 /// Logs go to a JSON file only (no console output).
-pub fn setup_logging(name: Option<&str>) -> &'static Result<()> {
-    INIT.get_or_init(|| {
-        use tracing_subscriber::util::SubscriberInitExt;
+pub fn setup_logging(name: Option<&str>) -> Result<()> {
+    setup_logging_internal(name, false)
+}
 
-        init_goose_request_log()?;
-        let config = goose::logging::LoggingConfig {
-            component: "cli",
-            name,
-            extra_directives: &["goose_cli=info"],
-            console: false,
-            json: true,
-        };
-        let subscriber = goose::logging::build_logging_subscriber(&config)?;
+fn setup_logging_internal(name: Option<&str>, force: bool) -> Result<()> {
+    let mut result = Ok(());
 
-        subscriber
-            .try_init()
-            .map_err(|e| anyhow::anyhow!("Failed to set global subscriber: {}", e))?;
-        Ok(())
-    })
+    let mut setup = || {
+        result = (|| {
+            use tracing_subscriber::util::SubscriberInitExt;
+
+            let config = goose::logging::LoggingConfig {
+                component: "cli",
+                name,
+                extra_directives: &["goose_cli=info"],
+                console: false,
+                json: true,
+            };
+            let subscriber = goose::logging::build_logging_subscriber(&config)?;
+
+            if force {
+                let _guard = subscriber.set_default();
+                tracing::warn!("Test log entry from setup");
+                tracing::info!("Another test log entry from setup");
+                std::thread::sleep(std::time::Duration::from_millis(100));
+                Ok(())
+            } else {
+                subscriber
+                    .try_init()
+                    .map_err(|e| anyhow::anyhow!("Failed to set global subscriber: {}", e))?;
+                Ok(())
+            }
+        })();
+    };
+
+    if force {
+        setup();
+    } else {
+        INIT.call_once(setup);
+    }
+
+    result
 }
 
 #[cfg(test)]

crates/goose-cli/src/main.rs

@@ -1,5 +1,3 @@
-#![recursion_limit = "256"]
-
 use anyhow::Result;
 use goose_cli::cli::cli;
 

crates/goose-cli/src/recipes/secret_discovery.rs

@@ -178,7 +178,6 @@ mod tests {
                     envs: Envs::new(HashMap::new()),
                     env_keys: vec!["SLACK_TOKEN".to_string()],
                     timeout: None,
-                    cwd: None,
                     description: "slack-mcp".to_string(),
                     bundled: None,
                     available_tools: Vec::new(),
@@ -276,7 +275,6 @@ mod tests {
                     envs: Envs::new(HashMap::new()),
                     env_keys: vec!["API_KEY".to_string()], // Same original key, different extension
                     timeout: None,
-                    cwd: None,
                     description: "service-b".to_string(),
                     bundled: None,
                     available_tools: Vec::new(),

crates/goose-cli/src/scenario_tests/scenario_runner.rs

@@ -9,6 +9,7 @@ use anyhow::Result;
 use goose::agents::{Agent, AgentConfig, GoosePlatform};
 use goose::config::permission::PermissionManager;
 use goose::config::GooseMode;
+use goose::model::ModelConfig;
 use goose::providers::{create, testprovider::TestProvider};
 use goose::session::session_manager::SessionType;
 use goose::session::SessionManager;
@@ -185,7 +186,12 @@ where
 
         let original_env = setup_environment(config)?;
 
-        let inner_provider = create(&factory_name, Vec::new()).await?;
+        let inner_provider = create(
+            &factory_name,
+            ModelConfig::new(config.model_name)?.with_canonical_limits(&factory_name),
+            Vec::new(),
+        )
+        .await?;
 
         let test_provider = Arc::new(TestProvider::new_recording(inner_provider, &file_path));
         (
@@ -240,12 +246,9 @@ where
         )
         .await?;
 
-    let scenario_model_config =
-        goose::model_config::model_config_from_user_config(&factory_name, config.model_name)?;
     agent
         .update_provider(
             provider_arc as Arc<dyn goose::providers::base::Provider>,
-            scenario_model_config,
             &session.id,
         )
         .await?;
@@ -259,7 +262,6 @@ where
         None,
         None,
         "text".to_string(),
-        false,
     )
     .await;
 

crates/goose-cli/src/scenario_tests/scenarios.rs

@@ -80,7 +80,7 @@ mod tests {
     //     run_scenario(
     //         "context_length_exceeded",
     //         Box::new(|provider| {
-    //             let model_config = resolve_global_model_config();
+    //             let model_config = provider.get_model_config();
     //             let context_length = model_config.context_limit.unwrap_or(300_000);
     //             // "hello " is only one token in most models, since the hello and space often
     //             // occur together in the training data.

crates/goose-cli/src/session/builder.rs

@@ -6,7 +6,6 @@ use console::style;
 use goose::agents::{Agent, Container, ExtensionError};
 use goose::config::resolve_extensions_for_new_session;
 use goose::config::{Config, ExtensionConfig, GooseMode};
-use goose::model_config::model_config_from_user_config;
 use goose::providers::create;
 use goose::recipe::Recipe;
 use goose::session::session_manager::SessionType;
@@ -117,8 +116,6 @@ pub struct SessionBuilderConfig {
     pub output_format: String,
     /// Docker container to run stdio extensions inside
     pub container: Option<Container>,
-    /// Print generation statistics after headless runs.
-    pub stats: bool,
 }
 
 /// Manual implementation of Default to ensure proper initialization of output_format
@@ -146,7 +143,6 @@ impl Default for SessionBuilderConfig {
             quiet: false,
             output_format: "text".to_string(),
             container: None,
-            stats: false,
         }
     }
 }
@@ -230,14 +226,14 @@ async fn load_extensions(
 struct ResolvedProviderConfig {
     provider_name: String,
     model_name: String,
-    model_config: goose_providers::model::ModelConfig,
+    model_config: goose::model::ModelConfig,
 }
 
 fn resolve_provider_and_model(
     session_config: &SessionBuilderConfig,
     config: &Config,
     saved_provider: Option<String>,
-    saved_model_config: Option<goose_providers::model::ModelConfig>,
+    saved_model_config: Option<goose::model::ModelConfig>,
 ) -> ResolvedProviderConfig {
     let recipe_settings = session_config
         .recipe
@@ -278,16 +274,14 @@ fn resolve_provider_and_model(
         }
         config
     } else {
-        let mut config =
-            goose::model_config::model_config_from_user_config(&provider_name, &model_name)
-                .unwrap_or_else(|e| {
-                    output::render_error(&format!("Failed to create model configuration: {}", e));
-                    process::exit(1);
-                });
-        if let Some(temp) = recipe_settings.and_then(|s| s.temperature) {
-            config = config.with_temperature(Some(temp));
-        }
-        config
+        let temperature = recipe_settings.and_then(|s| s.temperature);
+        goose::model::ModelConfig::new(&model_name)
+            .unwrap_or_else(|e| {
+                output::render_error(&format!("Failed to create model configuration: {}", e));
+                process::exit(1);
+            })
+            .with_canonical_limits(&provider_name)
+            .with_temperature(temperature)
     };
 
     ResolvedProviderConfig {
@@ -333,10 +327,7 @@ async fn resolve_session_id(
                 }
             }
         } else {
-            match session_manager
-                .list_sessions_by_types(&[SessionType::User])
-                .await
-            {
+            match session_manager.list_sessions().await {
                 Ok(sessions) if !sessions.is_empty() => sessions[0].id.clone(),
                 _ => {
                     output::render_error("Cannot resume - no previous sessions found");
@@ -417,7 +408,6 @@ async fn collect_extension_configs(
     recipe: Option<&Recipe>,
     session_id: &str,
 ) -> Result<Vec<ExtensionConfig>, ExtensionError> {
-    let recipe_extensions = recipe.and_then(|r| r.extensions.as_deref());
     let configured_extensions: Vec<ExtensionConfig> = if session_config.resume {
         EnabledExtensionsState::for_session(
             &agent.config.session_manager,
@@ -428,7 +418,7 @@ async fn collect_extension_configs(
     } else if session_config.no_profile {
         Vec::new()
     } else {
-        resolve_extensions_for_new_session(recipe_extensions, None)
+        resolve_extensions_for_new_session(recipe.and_then(|r| r.extensions.as_deref()), None)
     };
 
     let cli_flag_extensions = parse_cli_flag_extensions(
@@ -438,12 +428,6 @@ async fn collect_extension_configs(
     );
 
     let mut all: Vec<ExtensionConfig> = configured_extensions;
-    if !session_config.no_profile && !session_config.resume && recipe_extensions.is_none() {
-        let project_root = std::env::current_dir().ok();
-        all.extend(goose::plugins::mcp_servers::enabled_plugin_mcp_servers(
-            project_root.as_deref(),
-        ));
-    }
     all.extend(cli_flag_extensions.into_iter().map(|(_, cfg)| cfg));
 
     Ok(all)
@@ -547,79 +531,29 @@ pub async fn build_session(session_config: SessionBuilderConfig) -> CliSession {
             }
         };
 
-    let (new_provider, effective_provider_name, effective_model_name, effective_model_config) =
-        match create(&resolved.provider_name, extensions_for_provider.clone()).await {
-            Ok(provider) => (
-                provider,
-                resolved.provider_name.clone(),
-                resolved.model_name.clone(),
-                resolved.model_config.clone(),
-            ),
-            Err(e)
-                if session_config.resume
-                    && session_config.provider.is_none()
-                    && is_provider_unavailable_error(&e) =>
-            {
-                let fallback_provider = config.get_goose_provider().unwrap_or_else(|_| {
-                    output::render_error("No provider configured. Run 'goose configure' first.");
-                    process::exit(1);
-                });
-                let fallback_model = config.get_goose_model().unwrap_or_else(|_| {
-                    output::render_error("No model configured. Run 'goose configure' first.");
-                    process::exit(1);
-                });
-                eprintln!(
-                    "{}",
-                    style(format!(
-                        "Warning: Could not create the session's original provider '{}' ({}). \
-                    Falling back to the default provider '{}'.",
-                        resolved.provider_name, e, fallback_provider
-                    ))
-                    .yellow()
-                );
-                let fallback_model_config =
-                    model_config_from_user_config(fallback_provider.as_str(), &fallback_model)
-                        .unwrap_or_else(|e| {
-                            output::render_error(&format!(
-                                "Failed to create model configuration: {}",
-                                e
-                            ));
-                            process::exit(1);
-                        });
-                match create(&fallback_provider, extensions_for_provider.clone()).await {
-                    Ok(provider) => (
-                        provider,
-                        fallback_provider,
-                        fallback_model,
-                        fallback_model_config,
-                    ),
-                    Err(e2) => {
-                        output::render_error(&format!(
-                        "Error {}.\n\
-                        Please check your system keychain and run 'goose configure' again.\n\
-                        If your system is unable to use the keyring, please try setting secret key(s) via environment variables.\n\
-                        For more info, see: https://goose-docs.ai/docs/troubleshooting/#keychainkeyring-errors",
-                        e2
-                    ));
-                        process::exit(1);
-                    }
-                }
-            }
-            Err(e) => {
-                output::render_error(&format!(
+    let new_provider = match create(
+        &resolved.provider_name,
+        resolved.model_config,
+        extensions_for_provider.clone(),
+    )
+    .await
+    {
+        Ok(provider) => provider,
+        Err(e) => {
+            output::render_error(&format!(
                 "Error {}.\n\
                 Please check your system keychain and run 'goose configure' again.\n\
                 If your system is unable to use the keyring, please try setting secret key(s) via environment variables.\n\
                 For more info, see: https://goose-docs.ai/docs/troubleshooting/#keychainkeyring-errors",
                 e
             ));
-                process::exit(1);
-            }
-        };
-    tracing::info!("🤖 Using model: {}", effective_model_name);
+            process::exit(1);
+        }
+    };
+    tracing::info!("🤖 Using model: {}", resolved.model_name);
 
     agent
-        .update_provider(new_provider, effective_model_config, &session_id)
+        .update_provider(new_provider, &session_id)
         .await
         .unwrap_or_else(|e| {
             output::render_error(&format!("Failed to initialize agent: {}", e));
@@ -671,7 +605,6 @@ pub async fn build_session(session_config: SessionBuilderConfig) -> CliSession {
         edit_mode,
         recipe.and_then(|r| r.retry.clone()),
         session_config.output_format.clone(),
-        session_config.stats,
     )
     .await;
 
@@ -680,26 +613,17 @@ pub async fn build_session(session_config: SessionBuilderConfig) -> CliSession {
     if !session_config.quiet {
         output::display_session_info(
             session_config.resume,
-            &effective_provider_name,
-            &effective_model_name,
+            &resolved.provider_name,
+            &resolved.model_name,
             &Some(session_id),
         );
     }
     session
 }
 
-fn is_provider_unavailable_error(e: &anyhow::Error) -> bool {
-    let msg = e.to_string();
-    msg.contains("is not set")
-        || msg.contains("not configured")
-        || msg.contains("Configuration value not found")
-}
-
 #[cfg(test)]
 mod tests {
     use super::*;
-    use goose::session::SessionManager;
-    use tempfile::TempDir;
 
     #[test]
     fn test_session_builder_config_creation() {
@@ -727,7 +651,6 @@ mod tests {
             quiet: false,
             output_format: "text".to_string(),
             container: None,
-            stats: false,
         };
 
         assert_eq!(config.extensions.len(), 1);
@@ -763,44 +686,6 @@ mod tests {
         assert!(!config.fork);
     }
 
-    #[tokio::test]
-    async fn test_implicit_resume_ignores_newer_scheduled_sessions() {
-        let temp_dir = TempDir::new().unwrap();
-        let session_manager = SessionManager::new(temp_dir.path().to_path_buf());
-        let goose_mode = GooseMode::default();
-
-        let user_session = session_manager
-            .create_session(
-                temp_dir.path().to_path_buf(),
-                "User session".to_string(),
-                SessionType::User,
-                goose_mode,
-            )
-            .await
-            .unwrap();
-        session_manager
-            .create_session(
-                temp_dir.path().to_path_buf(),
-                "Scheduled job: test".to_string(),
-                SessionType::Scheduled,
-                goose_mode,
-            )
-            .await
-            .unwrap();
-
-        let resolved = resolve_session_id(
-            &SessionBuilderConfig {
-                resume: true,
-                ..SessionBuilderConfig::default()
-            },
-            &session_manager,
-            goose_mode,
-        )
-        .await;
-
-        assert_eq!(resolved, user_session.id);
-    }
-
     #[test]
     fn test_truncate_with_ellipsis() {
         assert_eq!(truncate_with_ellipsis("abc", 5), "abc");

crates/goose-cli/src/session/completion.rs

@@ -1,4 +1,3 @@
-use goose::agents::execute_commands::list_commands;
 use goose::config::GooseMode;
 use rustyline::completion::{Completer, FilenameCompleter, Pair};
 use rustyline::highlight::{CmdKind, Highlighter};
@@ -37,8 +36,8 @@ impl GooseCompleter {
         // Create completion candidates that match the prefix
         let candidates: Vec<Pair> = cache
             .prompts
-            .values()
-            .flatten()
+            .iter()
+            .flat_map(|(_, names)| names)
             .filter(|name| name.starts_with(prefix.trim()))
             .map(|name| Pair {
                 display: name.clone(),
@@ -130,6 +129,7 @@ impl GooseCompleter {
         let skills = list_installed_skills(Some(&cwd));
         let skill_names: Vec<String> = skills.iter().map(|s| s.name.clone()).collect();
 
+        // Complete the last letter being typed (e.g. "/skills coding in<tab>")
         let last = line.rsplit_once(' ').map_or("", |(_, w)| w);
         let pos = line.len() - last.len();
 
@@ -146,32 +146,23 @@ impl GooseCompleter {
         Ok((pos, candidates))
     }
 
-    /// Complete model names for the /model command.
-    fn complete_model_names(&self, line: &str) -> Result<(usize, Vec<Pair>)> {
-        Ok((line.len(), vec![]))
-    }
-
     /// Complete slash commands
     fn complete_slash_commands(&self, line: &str) -> Result<(usize, Vec<Pair>)> {
-        let mut commands = vec![
-            "/exit".to_string(),
-            "/quit".to_string(),
-            "/help".to_string(),
-            "/?".to_string(),
-            "/t".to_string(),
-            "/extension".to_string(),
-            "/builtin".to_string(),
-            "/mode".to_string(),
-            "/model".to_string(),
-            "/recipe".to_string(),
+        // Define available slash commands
+        let commands = [
+            "/exit",
+            "/quit",
+            "/help",
+            "/?",
+            "/t",
+            "/extension",
+            "/builtin",
+            "/prompts",
+            "/prompt",
+            "/mode",
+            "/recipe",
+            "/skills",
         ];
-        commands.extend(
-            list_commands()
-                .iter()
-                .map(|command| format!("/{}", command.name)),
-        );
-        commands.sort();
-        commands.dedup();
 
         // Find commands that match the prefix
         let matching_commands: Vec<Pair> = commands
@@ -405,10 +396,6 @@ impl Completer for GooseCompleter {
                 }
             }
 
-            if line.starts_with("/model") {
-                return self.complete_model_names(line);
-            }
-
             if line.starts_with("/mode") {
                 return self.complete_mode_flags(line);
             }
@@ -581,35 +568,12 @@ mod tests {
         let (pos, candidates) = completer.complete_slash_commands("/").unwrap();
         assert_eq!(pos, 0);
         assert!(candidates.len() > 1);
-        for command in list_commands() {
-            assert!(
-                candidates
-                    .iter()
-                    .any(|candidate| candidate.display == format!("/{}", command.name)),
-                "slash completion should list /{}",
-                command.name
-            );
-        }
 
         // Test no match
         let (_pos, candidates) = completer.complete_slash_commands("/nonexistent").unwrap();
         assert_eq!(candidates.len(), 0);
     }
 
-    #[test]
-    fn test_complete_model_names() {
-        let cache = create_test_cache();
-        let completer = GooseCompleter::new(cache);
-
-        let (pos, candidates) = completer.complete_model_names("/model ").unwrap();
-        assert_eq!(pos, "/model ".len());
-        assert!(candidates.is_empty());
-
-        let (pos, candidates) = completer.complete_model_names("/model gpt").unwrap();
-        assert_eq!(pos, "/model gpt".len());
-        assert!(candidates.is_empty());
-    }
-
     #[test]
     fn test_complete_prompt_names() {
         let cache = create_test_cache();

crates/goose-cli/src/session/editor.rs

@@ -1,10 +1,7 @@
-use anyhow::{Context, Result};
+use anyhow::Result;
 use goose::config::Config;
-use goose::conversation::message::Message;
-use goose::conversation::Conversation;
 use std::fs;
 use std::io::Read;
-use std::io::Write;
 use std::path::PathBuf;
 use std::process::Command;
 use tempfile::Builder;
@@ -24,6 +21,9 @@ pub fn resolve_editor_command() -> Option<String> {
     )
 }
 
+/// Inner resolution logic, separated for testability.
+/// Checks sources in priority order: config, VISUAL, EDITOR.
+/// Skips empty strings at each level.
 fn resolve_editor_from_sources(
     config_editor: Option<&str>,
     visual: Option<&str>,
@@ -37,57 +37,6 @@ fn resolve_editor_from_sources(
     None
 }
 
-/// Resolve the editor command, falling back to vi (or notepad on Windows).
-pub fn resolve_editor_or_default() -> String {
-    let config = Config::global();
-    let config_editor = config.get_goose_prompt_editor().ok().flatten();
-    let visual = std::env::var("VISUAL").ok();
-    let editor_env = std::env::var("EDITOR").ok();
-    resolve_editor_or_default_from_sources(
-        config_editor.as_deref(),
-        visual.as_deref(),
-        editor_env.as_deref(),
-    )
-}
-
-fn resolve_editor_default() -> String {
-    if cfg!(windows) {
-        "notepad".to_string()
-    } else {
-        "vi".to_string()
-    }
-}
-
-fn resolve_editor_or_default_from_sources(
-    config_editor: Option<&str>,
-    visual: Option<&str>,
-    editor_env: Option<&str>,
-) -> String {
-    resolve_editor_from_sources(config_editor, visual, editor_env)
-        .unwrap_or_else(resolve_editor_default)
-}
-
-/// Open a YAML temp file with the user's editor to edit a conversation.
-/// Returns the edited conversation, or an error if the editor failed or YAML was invalid.
-pub fn edit_conversation(conversation: &Conversation) -> Result<Conversation> {
-    let yaml = serde_yaml::to_string(conversation.messages())?;
-
-    let mut tmp = NamedTempFile::with_suffix(".yaml")?;
-    tmp.write_all(yaml.as_bytes())?;
-    tmp.flush()?;
-
-    let editor = resolve_editor_or_default();
-    let path = tmp.path().to_path_buf();
-
-    launch_editor(&editor, &path).with_context(|| format!("failed to launch editor '{editor}'"))?;
-
-    let edited = std::fs::read_to_string(&path)?;
-    let messages: Vec<Message> =
-        serde_yaml::from_str(&edited).context("invalid YAML — session unchanged")?;
-
-    Ok(Conversation::new_unvalidated(messages))
-}
-
 /// Build the markdown template content for the editor prompt.
 fn build_template(messages: &[&str], prefill: Option<&str>) -> String {
     let mut content = String::from("# Goose Prompt Editor\n\n");
@@ -135,36 +84,21 @@ impl SymlinkCleanup {
 
 impl Drop for SymlinkCleanup {
     fn drop(&mut self) {
+        // Always try to clean up the symlink, ignoring any errors
         let _ = std::fs::remove_file(&self.symlink_path);
     }
 }
 
-/// Split an editor command into program and arguments.
-///
-/// Uses shell-word splitting only when the command contains quotes, so values like
-/// `"/Applications/Sublime Text.app/.../subl" -w` work. Unquoted commands are split on
-/// whitespace to avoid shlex stripping backslashes from Windows paths like
-/// `C:\Windows\System32\notepad.exe`.
-fn split_editor_command(editor_cmd: &str) -> Result<Vec<String>> {
-    if editor_cmd.contains(['"', '\'']) {
-        shlex::split(editor_cmd).ok_or_else(|| {
-            anyhow::anyhow!("Invalid editor command: unmatched quotes in '{editor_cmd}'")
-        })
-    } else {
-        Ok(editor_cmd.split_whitespace().map(String::from).collect())
-    }
-}
-
 /// Launch editor and wait for completion
 fn launch_editor(editor_cmd: &str, file_path: &PathBuf) -> Result<()> {
     use std::process::Stdio;
 
-    let parts = split_editor_command(editor_cmd)?;
+    let parts: Vec<&str> = editor_cmd.split_whitespace().collect();
     if parts.is_empty() {
         return Err(anyhow::anyhow!("Empty editor command"));
     }
 
-    let mut cmd = Command::new(&parts[0]);
+    let mut cmd = Command::new(parts[0]);
     if let Ok(cwd) = std::env::current_dir() {
         cmd.current_dir(cwd);
     }
@@ -480,64 +414,54 @@ with multiple lines.
         );
     }
 
+    // --- resolve_editor_from_sources tests ---
+
     #[test]
-    fn test_resolve_editor_resolution_priority() {
-        assert_eq!(
-            resolve_editor_from_sources(Some("config-val"), Some("visual-val"), Some("editor-val")),
-            Some("config-val".to_string())
-        );
-
-        assert_eq!(
-            resolve_editor_from_sources(Some(""), Some("visual-val"), Some("editor-val")),
-            Some("visual-val".to_string())
-        );
-
-        assert_eq!(
-            resolve_editor_from_sources(None, Some(""), Some("editor-val")),
-            Some("editor-val".to_string())
-        );
-
-        assert_eq!(resolve_editor_from_sources(None, None, None), None);
-        assert_eq!(
-            resolve_editor_from_sources(Some(""), Some(""), Some("")),
-            None
-        );
-
-        let default_val = resolve_editor_default();
-        assert_eq!(
-            resolve_editor_or_default_from_sources(None, None, None),
-            default_val
-        );
-        assert_eq!(
-            resolve_editor_or_default_from_sources(Some(""), Some(""), Some("")),
-            default_val
-        );
+    fn test_resolve_editor_returns_config_when_set() {
+        let result = resolve_editor_from_sources(Some("code"), Some("vim"), Some("nano"));
+        assert_eq!(result.as_deref(), Some("code"));
     }
 
     #[test]
-    fn test_split_editor_command() {
-        assert_eq!(
-            split_editor_command("code --wait").unwrap(),
-            vec!["code", "--wait"]
-        );
+    fn test_resolve_editor_falls_back_to_visual() {
+        let result = resolve_editor_from_sources(None, Some("vim"), Some("nano"));
+        assert_eq!(result.as_deref(), Some("vim"));
+    }
 
-        assert_eq!(
-            split_editor_command(
-                r#""/Applications/Sublime Text.app/Contents/SharedSupport/bin/subl" -w"#
-            )
-            .unwrap(),
-            vec![
-                "/Applications/Sublime Text.app/Contents/SharedSupport/bin/subl",
-                "-w"
-            ]
-        );
+    #[test]
+    fn test_resolve_editor_falls_back_to_editor_env() {
+        let result = resolve_editor_from_sources(None, None, Some("nano"));
+        assert_eq!(result.as_deref(), Some("nano"));
+    }
 
-        assert_eq!(
-            split_editor_command(r"C:\Windows\System32\notepad.exe").unwrap(),
-            vec![r"C:\Windows\System32\notepad.exe"]
-        );
+    #[test]
+    fn test_resolve_editor_returns_none_when_nothing_set() {
+        let result = resolve_editor_from_sources(None, None, None);
+        assert_eq!(result, None);
+    }
 
-        assert!(split_editor_command(r#"code --wait "unclosed"#).is_err());
+    #[test]
+    fn test_resolve_editor_skips_empty_config() {
+        let result = resolve_editor_from_sources(Some(""), Some("vim"), None);
+        assert_eq!(result.as_deref(), Some("vim"));
+    }
+
+    #[test]
+    fn test_resolve_editor_skips_empty_visual() {
+        let result = resolve_editor_from_sources(None, Some(""), Some("nano"));
+        assert_eq!(result.as_deref(), Some("nano"));
+    }
+
+    #[test]
+    fn test_resolve_editor_skips_all_empty() {
+        let result = resolve_editor_from_sources(Some(""), Some(""), Some(""));
+        assert_eq!(result, None);
+    }
+
+    #[test]
+    fn test_resolve_editor_skips_empty_config_and_visual() {
+        let result = resolve_editor_from_sources(Some(""), Some(""), Some("emacs"));
+        assert_eq!(result.as_deref(), Some("emacs"));
     }
 
     // --- build_template edge case tests ---
@@ -545,7 +469,9 @@ with multiple lines.
     #[test]
     fn test_build_template_empty_prefill_string() {
         let content = build_template(&["## User: Hello"], Some(""));
+        // Empty prefill should not appear in content
         assert!(content.contains("# Your prompt:\n\n#"));
+        // Should go directly to conversation context
         assert!(content.contains("# Recent conversation for context"));
     }
 
@@ -572,8 +498,11 @@ with multiple lines.
         assert!(prefill_pos < context_pos);
     }
 
+    // --- extract_user_input with prefilled content tests ---
+
     #[test]
     fn test_extract_user_input_with_prefill_kept() {
+        // Simulates a user who opened the editor with prefill and kept it unchanged
         let content = build_template(&["## User: Hello"], Some("fix the login bug"));
         let result = extract_user_input(&content);
         assert_eq!(result, "fix the login bug");
@@ -581,6 +510,7 @@ with multiple lines.
 
     #[test]
     fn test_extract_user_input_with_prefill_edited() {
+        // Simulates a user who edited the prefill text
         let mut content = build_template(&["## User: Hello"], Some("fix the login bug"));
         content = content.replace(
             "fix the login bug",
@@ -592,6 +522,7 @@ with multiple lines.
 
     #[test]
     fn test_extract_user_input_prefill_replaced() {
+        // Simulates a user who deleted the prefill and wrote something new
         let mut content = build_template(&["## User: Hello"], Some("fix the login bug"));
         content = content.replace("fix the login bug\n", "completely different prompt\n");
         let result = extract_user_input(&content);
@@ -600,6 +531,7 @@ with multiple lines.
 
     #[test]
     fn test_extract_user_input_prefill_cleared() {
+        // Simulates a user who deleted the prefill and left nothing
         let mut content = build_template(&["## User: Hello"], Some("fix the login bug"));
         content = content.replace("fix the login bug\n", "");
         let result = extract_user_input(&content);

crates/goose-cli/src/session/elicitation.rs

@@ -1,15 +1,12 @@
 use console::style;
-use rmcp::model::ElicitationAction;
 use serde_json::Value;
 use std::collections::HashMap;
 use std::io::{self, BufRead, IsTerminal, Write};
 
-pub struct ElicitationInput {
-    pub action: ElicitationAction,
-    pub user_data: HashMap<String, Value>,
-}
-
-pub fn collect_elicitation_input(message: &str, schema: &Value) -> io::Result<ElicitationInput> {
+pub fn collect_elicitation_input(
+    message: &str,
+    schema: &Value,
+) -> io::Result<Option<HashMap<String, Value>>> {
     if !message.is_empty() {
         println!("\n{}", style(message).cyan());
     }
@@ -27,18 +24,9 @@ pub fn collect_elicitation_input(message: &str, schema: &Value) -> io::Result<El
                 "Approve?"
             };
             return match cliclack::confirm(prompt).initial_value(true).interact() {
-                Ok(true) => Ok(ElicitationInput {
-                    action: ElicitationAction::Accept,
-                    user_data: HashMap::new(),
-                }),
-                Ok(false) => Ok(ElicitationInput {
-                    action: ElicitationAction::Decline,
-                    user_data: HashMap::new(),
-                }),
-                Err(e) if e.kind() == io::ErrorKind::Interrupted => Ok(ElicitationInput {
-                    action: ElicitationAction::Cancel,
-                    user_data: HashMap::new(),
-                }),
+                Ok(true) => Ok(Some(HashMap::new())),
+                Ok(false) => Ok(None),
+                Err(e) if e.kind() == io::ErrorKind::Interrupted => Ok(None),
                 Err(e) => Err(e),
             };
         }
@@ -77,12 +65,7 @@ pub fn collect_elicitation_input(message: &str, schema: &Value) -> io::Result<El
                 Ok(v) => {
                     data.insert(name.clone(), Value::Bool(v));
                 }
-                Err(e) if e.kind() == io::ErrorKind::Interrupted => {
-                    return Ok(ElicitationInput {
-                        action: ElicitationAction::Cancel,
-                        user_data: HashMap::new(),
-                    });
-                }
+                Err(e) if e.kind() == io::ErrorKind::Interrupted => return Ok(None),
                 Err(e) => return Err(e),
             }
             continue;
@@ -110,10 +93,7 @@ pub fn collect_elicitation_input(message: &str, schema: &Value) -> io::Result<El
 
         // Handle Ctrl+C / EOF for cancellation
         if input.is_none() {
-            return Ok(ElicitationInput {
-                action: ElicitationAction::Cancel,
-                user_data: HashMap::new(),
-            });
+            return Ok(None);
         }
         let input = input.unwrap();
 
@@ -134,18 +114,12 @@ pub fn collect_elicitation_input(message: &str, schema: &Value) -> io::Result<El
                 "{}",
                 style(format!("Required field '{}' is missing", name)).red()
             );
-            return Ok(ElicitationInput {
-                action: ElicitationAction::Decline,
-                user_data: HashMap::new(),
-            });
+            return Ok(None);
         }
     }
 
     println!();
-    Ok(ElicitationInput {
-        action: ElicitationAction::Accept,
-        user_data: data,
-    })
+    Ok(Some(data))
 }
 
 fn read_line() -> io::Result<Option<String>> {

crates/goose-cli/src/session/export.rs

@@ -353,7 +353,7 @@ pub fn message_to_markdown(message: &Message, export_all_content: bool) -> Strin
                         message
                     ));
                 }
-                ActionRequiredData::ElicitationResponse { id, user_data, .. } => {
+                ActionRequiredData::ElicitationResponse { id, user_data } => {
                     md.push_str(&format!(
                         "**Action Required** (elicitation_response): {}\n```json\n{}\n```\n\n",
                         id,

crates/goose-cli/src/session/input.rs

@@ -20,7 +20,6 @@ pub enum InputResult {
     ListPrompts(Option<String>),
     PromptCommand(PromptCommandOptions),
     GooseMode(String),
-    Model(Option<String>),
     Plan(PlanCommandOptions),
     EndPlan,
     Clear,
@@ -197,8 +196,6 @@ fn handle_slash_command(input: &str) -> Option<InputResult> {
     const CMD_EXTENSION: &str = "/extension ";
     const CMD_BUILTIN: &str = "/builtin ";
     const CMD_MODE: &str = "/mode ";
-    const CMD_MODEL: &str = "/model";
-    const CMD_MODEL_WITH_SPACE: &str = "/model ";
     const CMD_PLAN: &str = "/plan";
     const CMD_ENDPLAN: &str = "/endplan";
     const CMD_CLEAR: &str = "/clear";
@@ -264,19 +261,6 @@ fn handle_slash_command(input: &str) -> Option<InputResult> {
         s if s.starts_with(CMD_MODE) => Some(InputResult::GooseMode(
             s.get(CMD_MODE.len()..).unwrap_or("").to_string(),
         )),
-        s if s == CMD_MODEL => Some(InputResult::Model(None)),
-        s if s.starts_with(CMD_MODEL_WITH_SPACE) => {
-            let model = s
-                .get(CMD_MODEL_WITH_SPACE.len()..)
-                .unwrap_or("")
-                .trim()
-                .to_string();
-            if model.is_empty() {
-                Some(InputResult::Model(None))
-            } else {
-                Some(InputResult::Model(Some(model)))
-            }
-        }
         s if s.starts_with(CMD_PLAN) => {
             parse_plan_command(s.get(CMD_PLAN.len()..).unwrap_or("").trim().to_string())
         }
@@ -401,17 +385,10 @@ fn parse_plan_command(input: String) -> Option<InputResult> {
     Some(InputResult::Plan(options))
 }
 
-fn help_text() -> String {
+fn print_help() {
     let newline_key = get_newline_key().to_ascii_uppercase();
     let modes = GooseMode::VARIANTS.join(", ");
-    let additional_builtin_help = additional_builtin_help();
-    let additional_builtin_help = if additional_builtin_help.is_empty() {
-        String::new()
-    } else {
-        format!("{additional_builtin_help}\n")
-    };
-
-    format!(
+    println!(
         "Available commands:
 /exit or /quit - Exit the session
 /t - Toggle Light/Dark/Ansi theme
@@ -422,7 +399,6 @@ fn help_text() -> String {
 /prompts [--extension <name>] - List all available prompts, optionally filtered by extension
 /prompt <n> [--info] [key=value...] - Get prompt info or execute a prompt
 /mode <name> - Set the goose mode to use ({modes})
-/model [name] - Show the current model, or switch models for this session while keeping the same provider
 /plan <message_text> -  Enters 'plan' mode with optional message. Create a plan based on the current messages and asks user if they want to act on it.
                         If user acts on the plan, goose mode is set to 'auto' and returns to 'normal' goose mode.
                         To warm up goose before using '/plan', we recommend setting '/mode approve' & putting appropriate context into goose.
@@ -432,7 +408,6 @@ fn help_text() -> String {
 /recipe [filepath] - Generate a recipe from the current conversation and save it to the specified filepath (must end with .yaml).
                        If no filepath is provided, it will be saved to ./recipe.yaml.
 /compact - Compact the current conversation to reduce context length while preserving key information.
-{additional_builtin_help}/status - Show session status: model, provider, mode, and token usage.
 /edit [text] - Open your prompt editor to compose a message. Optionally pre-fill with text.
                Uses $GOOSE_PROMPT_EDITOR, $VISUAL, or $EDITOR (in that order).
 /skills - List available skills or enable skills by name (usage: /skills [<name>...])
@@ -443,23 +418,7 @@ Navigation:
 Ctrl+C - Clear current line if text is entered, otherwise exit the session
 Ctrl+{newline_key} - Add a newline (configurable via GOOSE_CLI_NEWLINE_KEY)
 Up/Down arrows - Navigate through command history"
-    )
-}
-
-fn additional_builtin_help() -> String {
-    const DOCUMENTED_BUILTINS: &[&str] =
-        &["prompts", "prompt", "compact", "clear", "skills", "status"];
-
-    goose::agents::execute_commands::list_commands()
-        .iter()
-        .filter(|command| !DOCUMENTED_BUILTINS.contains(&command.name))
-        .map(|command| format!("/{} - {}", command.name, command.description))
-        .collect::<Vec<_>>()
-        .join("\n")
-}
-
-fn print_help() {
-    println!("{}", help_text());
+    );
 }
 
 /// Extract recent messages for editor context
@@ -539,38 +498,10 @@ mod tests {
             panic!("Expected AddBuiltin");
         }
 
-        // Test model command
-        assert!(matches!(
-            handle_slash_command("/model"),
-            Some(InputResult::Model(None))
-        ));
-        assert!(matches!(
-            handle_slash_command("/model   "),
-            Some(InputResult::Model(None))
-        ));
-        if let Some(InputResult::Model(Some(model))) = handle_slash_command("/model gpt-4.1") {
-            assert_eq!(model, "gpt-4.1");
-        } else {
-            panic!("Expected Model");
-        }
-
         // Test unknown commands
         assert!(handle_slash_command("/unknown").is_none());
     }
 
-    #[test]
-    fn help_lists_builtin_agent_commands() {
-        let help = help_text();
-
-        for command in goose::agents::execute_commands::list_commands() {
-            assert!(
-                help.contains(&format!("/{}", command.name)),
-                "help output should list /{}",
-                command.name
-            );
-        }
-    }
-
     #[test]
     fn test_prompts_command() {
         // Test basic prompts command

crates/goose-cli/src/session/mod.rs

@@ -1,6 +1,6 @@
 mod builder;
 mod completion;
-pub mod editor;
+mod editor;
 mod elicitation;
 mod export;
 mod input;
@@ -13,7 +13,6 @@ use crate::session::task_execution_display::{
     format_task_execution_notification, TASK_EXECUTION_NOTIFICATION_TYPE,
 };
 use goose::conversation::Conversation;
-use std::env;
 use std::io::Write;
 use std::str::FromStr;
 use tokio::signal::ctrl_c;
@@ -28,7 +27,6 @@ use goose::permission::permission_confirmation::PrincipalType;
 use goose::permission::Permission;
 use goose::permission::PermissionConfirmation;
 use goose::providers::base::Provider;
-use goose::providers::base::ProviderUsage;
 use goose::utils::safe_truncate;
 
 use anyhow::{Context, Result};
@@ -39,8 +37,8 @@ use goose::agents::{Agent, SessionConfig, COMPACT_TRIGGERS};
 use goose::config::extensions::name_to_key;
 use goose::config::{Config, GooseMode};
 use input::InputResult;
+use rmcp::model::PromptMessage;
 use rmcp::model::ServerNotification;
-use rmcp::model::{ElicitationAction, PromptMessage};
 use rmcp::model::{ErrorCode, ErrorData};
 use strum::VariantNames;
 
@@ -53,13 +51,11 @@ use std::collections::{HashMap, HashSet};
 use std::io::IsTerminal;
 use std::path::PathBuf;
 use std::sync::Arc;
-use std::time::{Duration, Instant};
+use std::time::Instant;
 use tokio;
 use tokio_util::sync::CancellationToken;
 use tracing::warn;
 
-const GOOSE_PLANNER_CONTEXT_LIMIT: &str = "GOOSE_PLANNER_CONTEXT_LIMIT";
-
 #[derive(Serialize, Deserialize, Debug)]
 struct JsonOutput {
     messages: Vec<Message>,
@@ -176,7 +172,6 @@ pub struct CliSession {
     edit_mode: Option<EditMode>,
     retry_config: Option<RetryConfig>,
     output_format: String,
-    stats: bool,
 }
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]
@@ -219,23 +214,22 @@ pub async fn classify_planner_response(
     session_id: &str,
     message_text: String,
     provider: Arc<dyn Provider>,
-    model_config: goose_providers::model::ModelConfig,
 ) -> Result<PlannerResponseType> {
     let prompt = format!(
         "The text below is the output from an AI model which can either provide a plan or list of clarifying questions. Based on the text below, decide if the output is a \"plan\" or \"clarifying questions\".\n---\n{message_text}"
     );
 
     let message = Message::user().with_text(&prompt);
-    let (result, _usage) = goose::session_context::with_session_id(
-        Some(session_id.to_string()),
-        provider.complete(
+    let model_config = provider.get_model_config();
+    let (result, _usage) = provider
+        .complete(
             &model_config,
+            session_id,
             "Reply only with the classification label: \"plan\" or \"clarifying questions\"",
             &[message],
             &[],
-        ),
-    )
-    .await?;
+        )
+        .await?;
 
     let predicted = result.as_concat_text();
     if predicted.to_lowercase().contains("plan") {
@@ -256,7 +250,6 @@ impl CliSession {
         edit_mode: Option<EditMode>,
         retry_config: Option<RetryConfig>,
         output_format: String,
-        stats: bool,
     ) -> Self {
         let messages = agent
             .config
@@ -278,7 +271,6 @@ impl CliSession {
             edit_mode,
             retry_config,
             output_format,
-            stats,
         }
     }
 
@@ -320,7 +312,6 @@ impl CliSession {
             env_keys: Vec::new(),
             description: goose::config::DEFAULT_EXTENSION_DESCRIPTION.to_string(),
             timeout: Some(goose::config::DEFAULT_EXTENSION_TIMEOUT),
-            cwd: None,
             bundled: None,
             available_tools: Vec::new(),
         })
@@ -487,6 +478,10 @@ impl CliSession {
 
     /// Start an interactive session, optionally with an initial message
     pub async fn interactive(&mut self, prompt: Option<String>) -> Result<()> {
+        self.agent
+            .emit_hook(goose::hooks::HookEvent::SessionStart, &self.session_id)
+            .await;
+
         let result = self.run_interactive(prompt).await;
 
         self.agent
@@ -613,10 +608,6 @@ impl CliSession {
                 history.save(editor);
                 self.handle_goose_mode(&mode).await?;
             }
-            InputResult::Model(model) => {
-                history.save(editor);
-                self.handle_model(model.as_deref()).await?;
-            }
             InputResult::Plan(options) => {
                 self.handle_plan_mode(options).await?;
             }
@@ -652,8 +643,6 @@ impl CliSession {
                             prefill.as_deref(),
                         ) {
                             Ok((message, true)) => {
-                                editor.add_history_entry(message.as_str())?;
-                                history.save(editor);
                                 self.handle_message_input(&message, history, editor).await?;
                             }
                             Ok((_, false)) => {}
@@ -721,8 +710,8 @@ impl CliSession {
             RunMode::Plan => {
                 let mut plan_messages = self.messages.clone();
                 plan_messages.push(Message::user().with_text(content));
-                let (reasoner, reasoner_model_config) = get_reasoner().await?;
-                self.plan_with_reasoner_model(plan_messages, reasoner, reasoner_model_config)
+                let reasoner = get_reasoner().await?;
+                self.plan_with_reasoner_model(plan_messages, reasoner)
                     .await?;
             }
         }
@@ -806,78 +795,6 @@ impl CliSession {
         Ok(())
     }
 
-    async fn handle_model(&self, model: Option<&str>) -> Result<()> {
-        let provider = self.agent.provider().await?;
-        let current_provider_name = provider.get_name().to_string();
-        let current_model_config = self
-            .agent
-            .model_config_for_session(&self.session_id)
-            .await?;
-        let current_model_name = current_model_config.model_name.clone();
-
-        if model.is_none() {
-            output::goose_mode_message(&format!(
-                "Current session model: '{}' (provider '{}')",
-                current_model_name, current_provider_name
-            ));
-            return Ok(());
-        }
-
-        let model_name = model.unwrap_or_default().trim();
-        if model_name.is_empty() {
-            output::render_error("Model name cannot be empty");
-            return Ok(());
-        }
-
-        if current_provider_name.ends_with("-acp") {
-            output::render_error(
-                "Session model switching is not supported for ACP providers in the CLI.",
-            );
-            return Ok(());
-        }
-
-        if provider.manages_own_context() {
-            output::render_error(&format!(
-                "Session model switching is not supported for provider '{}' because it manages its own conversation context.",
-                current_provider_name
-            ));
-            return Ok(());
-        }
-
-        let new_model_config =
-            build_switched_model_config(&current_provider_name, model_name, &current_model_config)?;
-
-        let configured_effort = Config::global().get_goose_thinking_effort();
-        let new_effort = new_model_config.thinking_effort().or(configured_effort);
-        let current_effort = current_model_config.thinking_effort().or(configured_effort);
-        if new_model_config.model_name == current_model_config.model_name
-            && new_effort == current_effort
-        {
-            output::goose_mode_message(&format!(
-                "Session already using model '{}' for provider '{}'",
-                current_model_name, current_provider_name
-            ));
-            return Ok(());
-        }
-
-        let extensions = self.agent.get_extension_configs().await;
-        let new_provider = goose::providers::create(&current_provider_name, extensions)
-            .await
-            .map_err(|e| anyhow::anyhow!("Failed to create provider: {e}"))?;
-
-        self.agent
-            .update_provider(new_provider, new_model_config, &self.session_id)
-            .await?;
-
-        let mode = self.agent.goose_mode().await;
-        self.agent.update_goose_mode(mode, &self.session_id).await?;
-        output::goose_mode_message(&format!(
-            "Session model switched from '{}' to '{}' for provider '{}'",
-            current_model_name, model_name, current_provider_name
-        ));
-        Ok(())
-    }
-
     async fn handle_plan_mode(&mut self, options: input::PlanCommandOptions) -> Result<()> {
         self.run_mode = RunMode::Plan;
         output::render_enter_plan_mode();
@@ -889,9 +806,8 @@ impl CliSession {
         let mut plan_messages = self.messages.clone();
         plan_messages.push(Message::user().with_text(&options.message_text));
 
-        let (reasoner, reasoner_model_config) = get_reasoner().await?;
-        self.plan_with_reasoner_model(plan_messages, reasoner, reasoner_model_config)
-            .await
+        let reasoner = get_reasoner().await?;
+        self.plan_with_reasoner_model(plan_messages, reasoner).await
     }
 
     async fn handle_clear(&mut self) -> Result<()> {
@@ -911,11 +827,9 @@ impl CliSession {
             .config
             .session_manager
             .update(&self.session_id)
-            .usage(goose_providers::conversation::token_usage::Usage::new(
-                Some(0),
-                Some(0),
-                Some(0),
-            ))
+            .total_tokens(Some(0))
+            .input_tokens(Some(0))
+            .output_tokens(Some(0))
             .apply()
             .await
         {
@@ -1053,24 +967,25 @@ impl CliSession {
         &mut self,
         plan_messages: Conversation,
         reasoner: Arc<dyn Provider>,
-        model_config: goose_providers::model::ModelConfig,
     ) -> Result<(), anyhow::Error> {
         let plan_prompt = self.agent.get_plan_prompt(&self.session_id).await?;
         output::show_thinking();
-        let (plan_response, _usage) = goose::session_context::with_session_id(
-            Some(self.session_id.clone()),
-            reasoner.complete(&model_config, &plan_prompt, plan_messages.messages(), &[]),
-        )
-        .await?;
+        let model_config = reasoner.get_model_config();
+        let (plan_response, _usage) = reasoner
+            .complete(
+                &model_config,
+                &self.session_id,
+                &plan_prompt,
+                plan_messages.messages(),
+                &[],
+            )
+            .await?;
         output::render_message(&plan_response, self.debug);
         output::hide_thinking();
         let planner_response_type = classify_planner_response(
             &self.session_id,
             plan_response.as_concat_text(),
             self.agent.provider().await?,
-            self.agent
-                .model_config_for_session(&self.session_id)
-                .await?,
         )
         .await?;
 
@@ -1135,6 +1050,9 @@ impl CliSession {
 
     /// Process a single message and exit
     pub async fn headless(&mut self, prompt: String) -> Result<()> {
+        self.agent
+            .emit_hook(goose::hooks::HookEvent::SessionStart, &self.session_id)
+            .await;
         let message = Message::user().with_text(&prompt);
         let result = self
             .process_message(message, CancellationToken::default(), false)
@@ -1187,9 +1105,6 @@ impl CliSession {
         let mut markdown_buffer = streaming_buffer::MarkdownBuffer::new();
         let mut prompted_credits_urls: HashSet<String> = HashSet::new();
         let mut thinking_header_shown = false;
-        let run_started = Instant::now();
-        let mut first_token_at: Option<Instant> = None;
-        let mut last_usage: Option<ProviderUsage> = None;
 
         use futures::StreamExt;
         loop {
@@ -1197,9 +1112,6 @@ impl CliSession {
                 result = stream.next() => {
                     match result {
                         Some(Ok(AgentEvent::Message(message))) => {
-                            if first_token_at.is_none() && message_has_text(&message) {
-                                first_token_at = Some(Instant::now());
-                            }
                             if let Some((id, security_prompt)) = find_tool_confirmation(&message) {
                                 let permission = if interactive {
                                     prompt_tool_confirmation(&security_prompt)?
@@ -1265,37 +1177,25 @@ impl CliSession {
                                 let _ = progress_bars.hide();
 
                                 match elicitation::collect_elicitation_input(&elicitation_message, &schema) {
-                                    Ok(input) => {
-                                        match &input.action {
-                                            ElicitationAction::Decline => {
-                                                output::render_text("Information request declined.", Some(Color::Yellow), true);
-                                            }
-                                            ElicitationAction::Cancel => {
-                                                output::render_text("Information request cancelled.", Some(Color::Yellow), true);
-                                            }
-                                            ElicitationAction::Accept => {}
-                                        }
-
-                                        let should_cancel = input.action == ElicitationAction::Cancel;
-                                        let action = input.action;
-                                        let user_data_value = serde_json::to_value(input.user_data)
+                                    Ok(Some(user_data)) => {
+                                        let user_data_value = serde_json::to_value(user_data)
                                             .unwrap_or(serde_json::Value::Object(serde_json::Map::new()));
                                         let response_message = Message::user()
                                             .with_content(MessageContent::action_required_elicitation_response(
                                                 elicitation_id,
                                                 user_data_value,
-                                                action,
                                             ))
                                             .with_visibility(false, true);
                                         self.messages.push(response_message.clone());
                                         // Elicitation responses return an empty stream - the response
                                         // unblocks the waiting tool call via ActionRequiredManager
                                         let _ = self.agent.reply(response_message, session_config.clone(), Some(cancel_token.clone())).await?;
-                                        if should_cancel {
-                                            cancel_token_clone.cancel();
-                                            drop(stream);
-                                            break;
-                                        }
+                                    }
+                                    Ok(None) => {
+                                        output::render_text("Information request cancelled.", Some(Color::Yellow), true);
+                                        cancel_token_clone.cancel();
+                                        drop(stream);
+                                        break;
                                     }
                                     Err(e) => {
                                         output::render_error(&format!("Failed to collect input: {}", e));
@@ -1323,10 +1223,6 @@ impl CliSession {
                                 }
                             }
                         }
-                        Some(Ok(AgentEvent::Usage(usage))) => {
-                            last_usage = Some(usage);
-                        }
-                        Some(Ok(AgentEvent::MessageUsage { .. })) => {}
                         Some(Ok(AgentEvent::McpNotification((extension_id, notification)))) => {
                             handle_mcp_notification(
                                 &extension_id,
@@ -1383,18 +1279,9 @@ impl CliSession {
                 .await
             {
                 Ok(session) => JsonMetadata {
-                    total_tokens: session
-                        .accumulated_usage
-                        .total_tokens
-                        .or(session.usage.total_tokens),
-                    input_tokens: session
-                        .accumulated_usage
-                        .input_tokens
-                        .or(session.usage.input_tokens),
-                    output_tokens: session
-                        .accumulated_usage
-                        .output_tokens
-                        .or(session.usage.output_tokens),
+                    total_tokens: session.accumulated_total_tokens.or(session.total_tokens),
+                    input_tokens: session.accumulated_input_tokens.or(session.input_tokens),
+                    output_tokens: session.accumulated_output_tokens.or(session.output_tokens),
                     status: "completed".to_string(),
                 },
                 Err(_) => JsonMetadata {
@@ -1419,9 +1306,9 @@ impl CliSession {
                 .ok();
             let (total_tokens, input_tokens, output_tokens) = match session {
                 Some(s) => (
-                    s.accumulated_usage.total_tokens.or(s.usage.total_tokens),
-                    s.accumulated_usage.input_tokens.or(s.usage.input_tokens),
-                    s.accumulated_usage.output_tokens.or(s.usage.output_tokens),
+                    s.accumulated_total_tokens.or(s.total_tokens),
+                    s.accumulated_input_tokens.or(s.input_tokens),
+                    s.accumulated_output_tokens.or(s.output_tokens),
                 ),
                 None => (None, None, None),
             };
@@ -1432,9 +1319,6 @@ impl CliSession {
             });
         } else {
             println!();
-            if self.stats {
-                print_run_stats(run_started, first_token_at, last_usage.as_ref());
-            }
         }
 
         Ok(())
@@ -1589,20 +1473,14 @@ impl CliSession {
 
     pub async fn get_total_token_usage(&self) -> Result<Option<i32>> {
         let metadata = self.get_session().await?;
-        Ok(metadata.accumulated_usage.total_tokens)
+        Ok(metadata.accumulated_total_tokens)
     }
 
     /// Display enhanced context usage with session totals
     pub async fn display_context_usage(&self) -> Result<()> {
         let provider = self.agent.provider().await?;
-        let model_config = self
-            .agent
-            .model_config_for_session(&self.session_id)
-            .await?;
-        let context_limit = provider
-            .get_context_limit(&model_config)
-            .await
-            .unwrap_or_else(|_| model_config.context_limit());
+        let model_config = provider.get_model_config();
+        let context_limit = model_config.context_limit();
 
         let config = Config::global();
         let show_cost = config
@@ -1615,15 +1493,18 @@ impl CliSession {
 
         match self.get_session().await {
             Ok(metadata) => {
-                let total_tokens = metadata.usage.total_tokens.unwrap_or(0) as usize;
+                let total_tokens = metadata.total_tokens.unwrap_or(0) as usize;
 
                 output::display_context_usage(total_tokens, context_limit);
 
                 if show_cost {
+                    let input_tokens = metadata.input_tokens.unwrap_or(0) as usize;
+                    let output_tokens = metadata.output_tokens.unwrap_or(0) as usize;
                     output::display_cost_usage(
                         &provider_name,
                         &model_config.model_name,
-                        &metadata.usage,
+                        input_tokens,
+                        output_tokens,
                     );
                 }
             }
@@ -1757,87 +1638,6 @@ impl CliSession {
     }
 }
 
-fn message_has_text(message: &Message) -> bool {
-    message.content.iter().any(
-        |content| matches!(content, MessageContent::Text(text) if !text.text.trim().is_empty()),
-    )
-}
-
-fn print_run_stats(
-    run_started: Instant,
-    first_token_at: Option<Instant>,
-    usage: Option<&ProviderUsage>,
-) {
-    let elapsed = run_started.elapsed();
-    let stats = usage.and_then(|usage| usage.stats.as_ref());
-    let generation_elapsed = stats
-        .and_then(|stats| stats.elapsed_ms)
-        .map(Duration::from_millis);
-    let output_tokens = usage
-        .and_then(|usage| usage.usage.output_tokens)
-        .and_then(|tokens| usize::try_from(tokens).ok())
-        .or_else(|| stats.and_then(|stats| stats.output_tokens));
-    let tokens_per_second = output_tokens.map(|tokens| {
-        let rate_elapsed = generation_elapsed.unwrap_or(elapsed);
-        if rate_elapsed.as_secs_f64() > 0.0 {
-            tokens as f64 / rate_elapsed.as_secs_f64()
-        } else {
-            0.0
-        }
-    });
-    let model_load_ms = stats.and_then(|stats| stats.model_load_ms);
-    let generation_time_to_first_token_ms = stats.and_then(|stats| stats.time_to_first_token_ms);
-
-    eprintln!("\nStats:");
-    if let Some(ms) = model_load_ms {
-        eprintln!("  Model load: {:.2}s", ms as f64 / 1000.0);
-    }
-    if model_load_ms.is_some() {
-        match generation_time_to_first_token_ms {
-            Some(ms) => eprintln!(
-                "  Generation time to first token: {:.2}s",
-                ms as f64 / 1000.0
-            ),
-            None => eprintln!("  Generation time to first token: unavailable"),
-        }
-        match first_token_at {
-            Some(first) => eprintln!(
-                "  End-to-end time to first token: {:.2}s",
-                first.duration_since(run_started).as_secs_f64()
-            ),
-            None => eprintln!("  End-to-end time to first token: unavailable"),
-        }
-    } else if let Some(ms) = generation_time_to_first_token_ms {
-        eprintln!("  Time to first token: {:.2}s", ms as f64 / 1000.0);
-    } else {
-        match first_token_at {
-            Some(first) => eprintln!(
-                "  Time to first token: {:.2}s",
-                first.duration_since(run_started).as_secs_f64()
-            ),
-            None => eprintln!("  Time to first token: unavailable"),
-        }
-    }
-    match tokens_per_second {
-        Some(rate) => eprintln!("  Tokens/sec: {:.2}", rate),
-        None => eprintln!("  Tokens/sec: unavailable"),
-    }
-    if let Some(tokens) = output_tokens {
-        eprintln!("  Output tokens: {tokens}");
-    }
-
-    if let Some(draft) = stats.and_then(|stats| stats.draft.as_ref()) {
-        eprintln!("  Draft accept rate: {:.1}%", draft.accept_rate * 100.0);
-        eprintln!(
-            "  Draft tokens: {} accepted: {} target verified: {} rounds: {}",
-            draft.draft_tokens, draft.accepted_tokens, draft.target_tokens, draft.rounds
-        );
-        if let Some(model) = &draft.model {
-            eprintln!("  Draft model: {model}");
-        }
-    }
-}
-
 fn maybe_open_credits_top_up_url(
     message: &Message,
     interactive: bool,
@@ -2220,11 +2020,11 @@ fn handle_agent_error(e: &anyhow::Error, is_stream_json_mode: bool) {
         });
     }
 
-    if e.downcast_ref::<goose_providers::errors::ProviderError>()
+    if e.downcast_ref::<goose::providers::errors::ProviderError>()
         .map(|provider_error| {
             matches!(
                 provider_error,
-                goose_providers::errors::ProviderError::ContextLengthExceeded(_)
+                goose::providers::errors::ProviderError::ContextLengthExceeded(_)
             )
         })
         .unwrap_or(false)
@@ -2244,8 +2044,8 @@ fn handle_agent_error(e: &anyhow::Error, is_stream_json_mode: bool) {
     }
 }
 
-async fn get_reasoner(
-) -> Result<(Arc<dyn Provider>, goose_providers::model::ModelConfig), anyhow::Error> {
+async fn get_reasoner() -> Result<Arc<dyn Provider>, anyhow::Error> {
+    use goose::model::ModelConfig;
     use goose::providers::create;
 
     let config = Config::global();
@@ -2270,23 +2070,12 @@ async fn get_reasoner(
             .expect("No model configured. Run 'goose configure' first")
     };
 
-    let planner_context_limit = match env::var(GOOSE_PLANNER_CONTEXT_LIMIT)
-        .ok()
-        .map(|v| v.parse::<usize>())
-    {
-        Some(Ok(n)) if n >= 4096 => Some(n),
-        Some(Ok(_)) => anyhow::bail!("{} must be at least 4096", GOOSE_PLANNER_CONTEXT_LIMIT),
-        Some(Err(e)) => anyhow::bail!("{}: {}", GOOSE_PLANNER_CONTEXT_LIMIT, e),
-        None => None,
-    };
-
     let model_config =
-        goose::model_config::model_config_from_user_config(&provider, model.as_str())?
-            .with_context_limit(planner_context_limit);
+        ModelConfig::new_with_context_env(model, &provider, Some("GOOSE_PLANNER_CONTEXT_LIMIT"))?;
     let extensions = goose::config::extensions::get_enabled_extensions_with_config(config);
-    let reasoner = create(&provider, extensions).await?;
+    let reasoner = create(&provider, model_config, extensions).await?;
 
-    Ok((reasoner, model_config))
+    Ok(reasoner)
 }
 
 /// Format elapsed time duration
@@ -2302,27 +2091,11 @@ fn format_elapsed_time(duration: std::time::Duration) -> String {
     }
 }
 
-fn build_switched_model_config(
-    provider_name: &str,
-    model_name: &str,
-    current_model_config: &goose_providers::model::ModelConfig,
-) -> Result<goose_providers::model::ModelConfig> {
-    goose::model_config::model_config_from_user_config(provider_name, model_name)
-        .map(|config| {
-            config
-                .with_temperature(current_model_config.temperature)
-                .with_toolshim(current_model_config.toolshim)
-                .with_toolshim_model(current_model_config.toolshim_model.clone())
-        })
-        .map_err(|e| anyhow::anyhow!("Failed to create model configuration: {e}"))
-}
-
 #[cfg(test)]
 mod tests {
     use super::*;
     use goose::agents::extension::Envs;
     use goose::config::ExtensionConfig;
-    use std::collections::HashMap;
     use std::time::Duration;
     use test_case::test_case;
 
@@ -2402,7 +2175,6 @@ mod tests {
             env_keys: vec![],
             description: goose::config::DEFAULT_EXTENSION_DESCRIPTION.to_string(),
             timeout: Some(goose::config::DEFAULT_EXTENSION_TIMEOUT),
-            cwd: None,
             bundled: None,
             available_tools: vec![],
         }
@@ -2418,7 +2190,6 @@ mod tests {
             env_keys: vec![],
             description: goose::config::DEFAULT_EXTENSION_DESCRIPTION.to_string(),
             timeout: Some(goose::config::DEFAULT_EXTENSION_TIMEOUT),
-            cwd: None,
             bundled: None,
             available_tools: vec![],
         }
@@ -2434,7 +2205,6 @@ mod tests {
             env_keys: vec![],
             description: goose::config::DEFAULT_EXTENSION_DESCRIPTION.to_string(),
             timeout: Some(goose::config::DEFAULT_EXTENSION_TIMEOUT),
-            cwd: None,
             bundled: None,
             available_tools: vec![],
         }
@@ -2449,73 +2219,6 @@ mod tests {
         assert!(CliSession::parse_stdio_extension("").is_err());
     }
 
-    #[test]
-    fn test_build_switched_model_config_rebuilds_target_model_settings() {
-        let _guard = env_lock::lock_env([
-            ("GOOSE_MAX_TOKENS", None::<&str>),
-            ("GOOSE_TEMPERATURE", None::<&str>),
-            ("GOOSE_CONTEXT_LIMIT", None::<&str>),
-            ("GOOSE_TOOLSHIM", None::<&str>),
-            ("GOOSE_TOOLSHIM_OLLAMA_MODEL", None::<&str>),
-        ]);
-
-        let current_model_config = goose_providers::model::ModelConfig {
-            model_name: "gpt-4o".to_string(),
-            context_limit: Some(128_000),
-            temperature: Some(0.25),
-            max_tokens: Some(16_384),
-            toolshim: true,
-            toolshim_model: Some("qwen2.5-coder".to_string()),
-            request_params: Some(HashMap::from([(
-                "anthropic_beta".to_string(),
-                serde_json::json!(["output-128k-2025-02-19"]),
-            )])),
-            reasoning: Some(false),
-        };
-
-        let switched =
-            build_switched_model_config("openai", "gpt-5.4", &current_model_config).unwrap();
-        let expected = goose_providers::model::ModelConfig::new("gpt-5.4")
-            .with_canonical_limits("openai")
-            .with_temperature(Some(0.25))
-            .with_toolshim(true)
-            .with_toolshim_model(Some("qwen2.5-coder".to_string()));
-
-        assert_eq!(switched.model_name, expected.model_name);
-        assert_eq!(switched.context_limit, expected.context_limit);
-        assert_eq!(switched.max_tokens, expected.max_tokens);
-        assert_eq!(switched.request_params, expected.request_params);
-        assert_eq!(switched.reasoning, expected.reasoning);
-        assert_eq!(switched.temperature, Some(0.25));
-        assert!(switched.toolshim);
-        assert_eq!(switched.toolshim_model.as_deref(), Some("qwen2.5-coder"));
-    }
-
-    #[test]
-    fn test_build_switched_model_config_detects_effort_suffix_change() {
-        let _guard = env_lock::lock_env([
-            ("GOOSE_MAX_TOKENS", None::<&str>),
-            ("GOOSE_TEMPERATURE", None::<&str>),
-            ("GOOSE_CONTEXT_LIMIT", None::<&str>),
-            ("GOOSE_TOOLSHIM", None::<&str>),
-            ("GOOSE_TOOLSHIM_OLLAMA_MODEL", None::<&str>),
-            ("GOOSE_THINKING_EFFORT", None::<&str>),
-        ]);
-
-        let current = goose_providers::model::ModelConfig::new("gpt-5.4-high")
-            .with_canonical_limits("openai");
-        assert_eq!(current.model_name, "gpt-5.4");
-        assert_eq!(
-            current.thinking_effort(),
-            Some(goose_providers::thinking::ThinkingEffort::High)
-        );
-
-        let switched = build_switched_model_config("openai", "gpt-5.4", &current).unwrap();
-
-        assert_eq!(switched.model_name, current.model_name);
-        assert_ne!(switched.thinking_effort(), current.thinking_effort());
-    }
-
     #[test]
     fn test_split_command_args_windows_paths() {
         assert_eq!(

crates/goose-cli/src/session/output.rs

@@ -10,7 +10,6 @@ use goose::providers::canonical::maybe_get_canonical_model;
 #[cfg(target_os = "windows")]
 use goose::subprocess::SubprocessExt;
 use goose::utils::safe_truncate;
-use goose_providers::conversation::token_usage::Usage;
 use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
 use rmcp::model::{CallToolRequestParams, JsonObject, PromptArgument};
 use serde_json::Value;
@@ -248,8 +247,7 @@ pub fn render_message(message: &Message, debug: bool) {
             }
             MessageContent::SystemNotification(notification) => {
                 match notification.notification_type {
-                    SystemNotificationType::ThinkingMessage
-                    | SystemNotificationType::ProgressMessage => {
+                    SystemNotificationType::ThinkingMessage => {
                         show_thinking();
                         set_thinking_message(&notification.msg);
                     }
@@ -331,8 +329,7 @@ pub fn render_message_streaming(
             }
             MessageContent::SystemNotification(notification) => {
                 match notification.notification_type {
-                    SystemNotificationType::ThinkingMessage
-                    | SystemNotificationType::ProgressMessage => {
+                    SystemNotificationType::ThinkingMessage => {
                         show_thinking();
                         set_thinking_message(&notification.msg);
                     }
@@ -1417,33 +1414,31 @@ pub fn display_context_usage(total_tokens: usize, context_limit: usize) {
     );
 }
 
-fn estimate_cost_usd(provider: &str, model: &str, usage: &Usage) -> Option<f64> {
+fn estimate_cost_usd(
+    provider: &str,
+    model: &str,
+    input_tokens: usize,
+    output_tokens: usize,
+) -> Option<f64> {
     let canonical_model = maybe_get_canonical_model(provider, model)?;
-    canonical_model.cost.estimate_cost(usage)
+
+    let input_cost_per_token = canonical_model.cost.input? / 1_000_000.0;
+    let output_cost_per_token = canonical_model.cost.output? / 1_000_000.0;
+
+    let input_cost = input_cost_per_token * input_tokens as f64;
+    let output_cost = output_cost_per_token * output_tokens as f64;
+    Some(input_cost + output_cost)
 }
 
 /// Display cost information, if price data is available.
-pub fn display_cost_usage(provider: &str, model: &str, usage: &Usage) {
-    if let Some(cost) = estimate_cost_usd(provider, model, usage) {
+pub fn display_cost_usage(provider: &str, model: &str, input_tokens: usize, output_tokens: usize) {
+    if let Some(cost) = estimate_cost_usd(provider, model, input_tokens, output_tokens) {
         use console::style;
-        let input_tokens = usage.input_tokens.unwrap_or(0);
-        let output_tokens = usage.output_tokens.unwrap_or(0);
-        let cache_read = usage.cache_read_input_tokens.unwrap_or(0);
-        let cache_write = usage.cache_write_input_tokens.unwrap_or(0);
-
-        let cache_breakdown = match (cache_read, cache_write) {
-            (0, 0) => String::new(),
-            (read, 0) => format!(" ({} cache read)", read),
-            (0, write) => format!(" ({} cache write)", write),
-            (read, write) => format!(" ({} cache read, {} cache write)", read, write),
-        };
-
         eprintln!(
-            "Cost: {} USD ({} tokens: in {}{}, out {})",
+            "Cost: {} USD ({} tokens: in {}, out {})",
             style(format!("${:.4}", cost)).cyan(),
             input_tokens + output_tokens,
             input_tokens,
-            cache_breakdown,
             output_tokens
         );
     }

crates/goose-cli/src/session/streaming_buffer.rs

@@ -524,8 +524,10 @@ impl MarkdownBuffer {
                 }
             }
             "]" => {}
-            ")" if state.in_link_url => {
-                state.in_link_url = false;
+            ")" => {
+                if state.in_link_url {
+                    state.in_link_url = false;
+                }
             }
             _ => {}
         }

crates/goose-cli/src/session/thinking.rs

@@ -1,4 +1,4 @@
-use rand::seq::IndexedRandom;
+use rand::seq::SliceRandom;
 
 /// Extended list of playful thinking messages including both goose and general AI actions
 const THINKING_MESSAGES: &[&str] = &[
@@ -215,6 +215,6 @@ const THINKING_MESSAGES: &[&str] = &[
 /// Returns a random thinking message from the extended list
 pub fn get_random_thinking_message() -> &'static str {
     THINKING_MESSAGES
-        .choose(&mut rand::rng())
+        .choose(&mut rand::thread_rng())
         .unwrap_or(&THINKING_MESSAGES[0])
 }

crates/goose-download-manager/Cargo.toml

@@ -1,21 +0,0 @@
-[package]
-name = "goose-download-manager"
-version = "0.1.0-alpha.0"
-edition.workspace = true
-rust-version.workspace = true
-authors.workspace = true
-license.workspace = true
-repository.workspace = true
-description.workspace = true
-
-[lints]
-workspace = true
-
-[dependencies]
-anyhow = { workspace = true }
-once_cell = { workspace = true }
-reqwest = { workspace = true, features = ["stream"] }
-serde = { workspace = true }
-tokio = { workspace = true, features = ["fs", "io-util", "time"] }
-tracing = { workspace = true }
-utoipa = { workspace = true }

crates/goose-download-manager/src/lib.rs

@@ -1,691 +0,0 @@
-use anyhow::Result;
-use serde::{Deserialize, Serialize};
-use std::collections::HashMap;
-use std::path::{Path, PathBuf};
-use std::sync::{Arc, Mutex};
-use tokio::io::AsyncWriteExt;
-use tracing::info;
-use utoipa::ToSchema;
-
-fn partial_path_for(destination: &Path) -> PathBuf {
-    destination.with_extension(
-        destination
-            .extension()
-            .map(|e| format!("{}.part", e.to_string_lossy()))
-            .unwrap_or_else(|| "part".to_string()),
-    )
-}
-
-/// Remove orphaned `.part` files in the given directory (and one level of subdirectories).
-/// Preserves `.part` files whose final destination is in `registered_paths` so that
-/// in-progress shard downloads can resume after a restart.
-pub fn cleanup_partial_downloads(
-    dir: &Path,
-    registered_paths: &std::collections::HashSet<PathBuf>,
-) {
-    let should_keep = |part_path: &Path| -> bool {
-        // Derive the final path by stripping the trailing ".part" extension
-        let final_path = part_path.with_extension("");
-        registered_paths.contains(&final_path)
-    };
-
-    if let Ok(entries) = std::fs::read_dir(dir) {
-        for entry in entries.flatten() {
-            let path = entry.path();
-            if path.extension().is_some_and(|e| e == "part") && !should_keep(&path) {
-                let _ = std::fs::remove_file(&path);
-            }
-            if path.is_dir() {
-                if let Ok(sub_entries) = std::fs::read_dir(&path) {
-                    for sub in sub_entries.flatten() {
-                        let sub_path = sub.path();
-                        if sub_path.extension().is_some_and(|e| e == "part")
-                            && !should_keep(&sub_path)
-                        {
-                            let _ = std::fs::remove_file(&sub_path);
-                        }
-                    }
-                }
-            }
-        }
-    }
-}
-
-#[derive(Debug, Clone, Serialize, Deserialize, ToSchema)]
-pub struct DownloadProgress {
-    /// Model ID being downloaded
-    pub model_id: String,
-    /// Download status
-    pub status: DownloadStatus,
-    /// Bytes downloaded so far
-    pub bytes_downloaded: u64,
-    /// Total bytes to download
-    pub total_bytes: u64,
-    /// Download progress percentage (0-100)
-    pub progress_percent: f32,
-    /// Download speed in bytes per second
-    pub speed_bps: Option<u64>,
-    /// Estimated time remaining in seconds
-    pub eta_seconds: Option<u64>,
-    /// Error message if failed
-    pub error: Option<String>,
-    /// Whether the background download task has exited
-    #[serde(skip)]
-    pub task_exited: bool,
-}
-
-#[derive(Debug, Clone, Serialize, Deserialize, ToSchema, PartialEq)]
-#[serde(rename_all = "lowercase")]
-pub enum DownloadStatus {
-    Downloading,
-    Completed,
-    Failed,
-    Cancelled,
-}
-
-type DownloadMap = Arc<Mutex<HashMap<String, DownloadProgress>>>;
-
-pub struct DownloadManager {
-    downloads: DownloadMap,
-}
-
-impl Default for DownloadManager {
-    fn default() -> Self {
-        Self::new()
-    }
-}
-
-impl DownloadManager {
-    pub fn new() -> Self {
-        Self {
-            downloads: Arc::new(Mutex::new(HashMap::new())),
-        }
-    }
-
-    pub fn get_progress(&self, model_id: &str) -> Option<DownloadProgress> {
-        self.downloads.lock().ok()?.get(model_id).cloned()
-    }
-
-    pub fn is_downloading(&self, model_id: &str) -> bool {
-        self.get_progress(model_id)
-            .is_some_and(|progress| progress.status == DownloadStatus::Downloading)
-    }
-
-    pub fn list_progress(&self) -> Vec<DownloadProgress> {
-        self.downloads
-            .lock()
-            .map(|downloads| downloads.values().cloned().collect())
-            .unwrap_or_default()
-    }
-
-    pub fn set_progress(&self, progress: DownloadProgress) {
-        if let Ok(mut downloads) = self.downloads.lock() {
-            downloads.insert(progress.model_id.clone(), progress);
-        }
-    }
-
-    pub fn reserve_download(&self, progress: DownloadProgress) -> Result<bool> {
-        let mut downloads = self
-            .downloads
-            .lock()
-            .map_err(|_| anyhow::anyhow!("Failed to acquire lock"))?;
-
-        if let Some(existing) = downloads.get(&progress.model_id) {
-            if existing.status == DownloadStatus::Downloading
-                || (existing.status == DownloadStatus::Cancelled && !existing.task_exited)
-            {
-                return Ok(false);
-            }
-        }
-
-        downloads.insert(progress.model_id.clone(), progress);
-        Ok(true)
-    }
-
-    pub fn update_progress(&self, model_id: &str, update: impl FnOnce(&mut DownloadProgress)) {
-        if let Ok(mut downloads) = self.downloads.lock() {
-            if let Some(progress) = downloads.get_mut(model_id) {
-                update(progress);
-            }
-        }
-    }
-
-    pub fn cancel_download(&self, model_id: &str) -> Result<()> {
-        let mut downloads = self
-            .downloads
-            .lock()
-            .map_err(|_| anyhow::anyhow!("Failed to acquire lock"))?;
-
-        if let Some(progress) = downloads.get_mut(model_id) {
-            progress.status = DownloadStatus::Cancelled;
-            Ok(())
-        } else {
-            anyhow::bail!("Download not found")
-        }
-    }
-
-    pub async fn download_model(
-        &self,
-        model_id: String,
-        url: String,
-        destination: PathBuf,
-        on_complete: Option<Box<dyn FnOnce() + Send + 'static>>,
-    ) -> Result<()> {
-        self.download_model_sharded(model_id, vec![(url, destination)], 0, on_complete)
-            .await
-    }
-
-    pub async fn download_model_with_bearer_token(
-        &self,
-        model_id: String,
-        url: String,
-        destination: PathBuf,
-        bearer_token: Option<String>,
-        on_complete: Option<Box<dyn FnOnce() + Send + 'static>>,
-    ) -> Result<()> {
-        self.download_model_sharded_with_bearer_token(
-            model_id,
-            vec![(url, destination)],
-            0,
-            bearer_token,
-            on_complete,
-        )
-        .await
-    }
-
-    pub async fn download_model_sharded(
-        &self,
-        model_id: String,
-        files: Vec<(String, PathBuf)>,
-        total_size_hint: u64,
-        on_complete: Option<Box<dyn FnOnce() + Send + 'static>>,
-    ) -> Result<()> {
-        self.download_model_sharded_with_bearer_token(
-            model_id,
-            files,
-            total_size_hint,
-            None,
-            on_complete,
-        )
-        .await
-    }
-
-    pub async fn download_model_sharded_with_bearer_token(
-        &self,
-        model_id: String,
-        files: Vec<(String, PathBuf)>,
-        total_size_hint: u64,
-        bearer_token: Option<String>,
-        on_complete: Option<Box<dyn FnOnce() + Send + 'static>>,
-    ) -> Result<()> {
-        info!(model_id = %model_id, file_count = files.len(), "Starting model download");
-        {
-            let mut downloads = self
-                .downloads
-                .lock()
-                .map_err(|_| anyhow::anyhow!("Failed to acquire lock"))?;
-
-            if let Some(existing) = downloads.get(&model_id) {
-                if existing.status == DownloadStatus::Downloading {
-                    anyhow::bail!("Download already in progress");
-                }
-                if existing.status == DownloadStatus::Cancelled && !existing.task_exited {
-                    anyhow::bail!(
-                        "Download is being cancelled; wait for it to finish before restarting"
-                    );
-                }
-            }
-
-            downloads.insert(
-                model_id.clone(),
-                DownloadProgress {
-                    model_id: model_id.clone(),
-                    status: DownloadStatus::Downloading,
-                    bytes_downloaded: 0,
-                    total_bytes: total_size_hint,
-                    progress_percent: 0.0,
-                    speed_bps: None,
-                    eta_seconds: None,
-                    error: None,
-                    task_exited: false,
-                },
-            );
-        }
-
-        // Create parent directories for all files
-        for (_, dest) in &files {
-            if let Some(parent) = dest.parent() {
-                tokio::fs::create_dir_all(parent)
-                    .await
-                    .map_err(|e| anyhow::anyhow!("Failed to create directory: {}", e))?;
-            }
-        }
-
-        let downloads = self.downloads.clone();
-        let model_id_clone = model_id.clone();
-        let files_for_cleanup: Vec<PathBuf> = files.iter().map(|(_, d)| d.clone()).collect();
-
-        tokio::spawn(async move {
-            let result = Self::download_files_sequentially(
-                &files,
-                &downloads,
-                &model_id_clone,
-                bearer_token.as_deref(),
-            )
-            .await;
-
-            match result {
-                Ok(_) => {
-                    info!(model_id = %model_id_clone, "Download completed successfully");
-                    if let Ok(mut downloads) = downloads.lock() {
-                        if let Some(progress) = downloads.get_mut(&model_id_clone) {
-                            progress.status = DownloadStatus::Completed;
-                            progress.progress_percent = 100.0;
-                            progress.task_exited = true;
-                        }
-                    }
-
-                    if let Some(callback) = on_complete {
-                        callback();
-                    }
-                }
-                Err(e) => {
-                    for dest in &files_for_cleanup {
-                        let partial = partial_path_for(dest);
-                        let _ = tokio::fs::remove_file(&partial).await;
-                    }
-
-                    if let Ok(mut downloads) = downloads.lock() {
-                        if let Some(progress) = downloads.get_mut(&model_id_clone) {
-                            if progress.status != DownloadStatus::Cancelled {
-                                progress.status = DownloadStatus::Failed;
-                            }
-                            progress.error = Some(e.to_string());
-                            progress.task_exited = true;
-                        }
-                    }
-                }
-            }
-        });
-
-        Ok(())
-    }
-
-    const MAX_RETRIES: u32 = 10;
-    const RETRY_BASE_DELAY: std::time::Duration = std::time::Duration::from_secs(2);
-    const RETRY_MAX_DELAY: std::time::Duration = std::time::Duration::from_secs(60);
-
-    async fn cancellable_sleep(
-        delay: std::time::Duration,
-        downloads: &DownloadMap,
-        model_id: &str,
-    ) -> Result<(), anyhow::Error> {
-        let check_interval = std::time::Duration::from_millis(500);
-        let start = std::time::Instant::now();
-        while start.elapsed() < delay {
-            if Self::is_cancelled(downloads, model_id) {
-                anyhow::bail!("Download cancelled");
-            }
-            let remaining = delay.saturating_sub(start.elapsed());
-            tokio::time::sleep(std::cmp::min(check_interval, remaining)).await;
-        }
-        Ok(())
-    }
-
-    fn is_cancelled(downloads: &DownloadMap, model_id: &str) -> bool {
-        if let Ok(downloads) = downloads.lock() {
-            if let Some(progress) = downloads.get(model_id) {
-                return progress.status == DownloadStatus::Cancelled;
-            }
-        }
-        false
-    }
-
-    #[allow(clippy::too_many_arguments)]
-    /// Download multiple files sequentially, tracking cumulative progress under one model_id.
-    async fn download_files_sequentially(
-        files: &[(String, PathBuf)],
-        downloads: &DownloadMap,
-        model_id: &str,
-        bearer_token: Option<&str>,
-    ) -> Result<(), anyhow::Error> {
-        let client = reqwest::Client::builder()
-            .connect_timeout(std::time::Duration::from_secs(30))
-            .read_timeout(std::time::Duration::from_secs(120))
-            .build()?;
-
-        // HEAD each file to get accurate total size. Only replace the hint if
-        // every file returned a size; partial results would underestimate.
-        let mut total: u64 = 0;
-        let mut all_resolved = true;
-        for (url, _) in files {
-            let size = Self::apply_bearer_token(client.head(url), bearer_token)
-                .send()
-                .await
-                .ok()
-                .and_then(|r| r.content_length())
-                .unwrap_or(0);
-            if size == 0 {
-                all_resolved = false;
-            }
-            total += size;
-        }
-        if all_resolved && total > 0 {
-            if let Ok(mut dl) = downloads.lock() {
-                if let Some(progress) = dl.get_mut(model_id) {
-                    progress.total_bytes = total;
-                }
-            }
-        }
-
-        let start_time = std::time::Instant::now();
-        let mut cumulative_bytes: u64 = 0;
-        // Account for already-downloaded shards
-        for (_, dest) in files {
-            let partial = partial_path_for(dest);
-            if dest.exists() {
-                if let Ok(meta) = tokio::fs::metadata(dest).await {
-                    cumulative_bytes += meta.len();
-                }
-            } else if partial.exists() {
-                if let Ok(meta) = tokio::fs::metadata(&partial).await {
-                    cumulative_bytes += meta.len();
-                }
-            }
-        }
-        let bytes_at_start = cumulative_bytes;
-
-        for (url, destination) in files {
-            if Self::is_cancelled(downloads, model_id) {
-                anyhow::bail!("Download cancelled");
-            }
-
-            // Skip already-completed shards
-            if destination.exists() {
-                continue;
-            }
-
-            Self::download_one_file(
-                &client,
-                url,
-                destination,
-                downloads,
-                model_id,
-                &mut cumulative_bytes,
-                start_time,
-                bytes_at_start,
-                bearer_token,
-            )
-            .await?;
-        }
-
-        Ok(())
-    }
-
-    #[allow(clippy::too_many_arguments)]
-    async fn download_one_file(
-        client: &reqwest::Client,
-        url: &str,
-        destination: &Path,
-        downloads: &DownloadMap,
-        model_id: &str,
-        cumulative_bytes: &mut u64,
-        start_time: std::time::Instant,
-        bytes_at_start: u64,
-        bearer_token: Option<&str>,
-    ) -> Result<(), anyhow::Error> {
-        let partial_path = partial_path_for(destination);
-        let mut retries = 0u32;
-
-        let mut file_bytes: u64 = if partial_path.exists() {
-            tokio::fs::metadata(&partial_path).await?.len()
-        } else {
-            0
-        };
-
-        // Get this file's total size
-        let mut file_total: u64 = Self::apply_bearer_token(client.head(url), bearer_token)
-            .send()
-            .await
-            .ok()
-            .and_then(|r| r.content_length())
-            .unwrap_or(0);
-
-        // If partial matches expected size exactly, promote it
-        if file_total > 0 && file_bytes == file_total {
-            tokio::fs::rename(&partial_path, destination).await?;
-            // cumulative_bytes already accounts for this file from the pre-scan
-            return Ok(());
-        }
-
-        // If partial is oversized or remote changed, discard and re-download
-        if file_total > 0 && file_bytes > file_total {
-            info!(model_id = %model_id, file_bytes, file_total, "Partial file oversized, re-downloading");
-            *cumulative_bytes = cumulative_bytes.saturating_sub(file_bytes);
-            file_bytes = 0;
-            let _ = tokio::fs::remove_file(&partial_path).await;
-        }
-
-        loop {
-            if Self::is_cancelled(downloads, model_id) {
-                let _ = tokio::fs::remove_file(&partial_path).await;
-                anyhow::bail!("Download cancelled");
-            }
-
-            let mut request = Self::apply_bearer_token(client.get(url), bearer_token);
-            if file_bytes > 0 {
-                request = request.header("Range", format!("bytes={}-", file_bytes));
-            }
-
-            let response = match request.send().await {
-                Ok(r) => r,
-                Err(e) => {
-                    if retries >= Self::MAX_RETRIES {
-                        anyhow::bail!("Download failed after {} retries: {}", retries, e);
-                    }
-                    retries += 1;
-                    let delay = std::cmp::min(
-                        Self::RETRY_BASE_DELAY * 2u32.saturating_pow(retries - 1),
-                        Self::RETRY_MAX_DELAY,
-                    );
-                    info!(model_id = %model_id, retry = retries, delay_secs = ?delay.as_secs(), error = %e, "Retrying download after connection error");
-                    Self::cancellable_sleep(delay, downloads, model_id).await?;
-                    continue;
-                }
-            };
-
-            let status = response.status();
-            if status == reqwest::StatusCode::RANGE_NOT_SATISFIABLE {
-                if file_total > 0 && file_bytes == file_total {
-                    break;
-                }
-                *cumulative_bytes = cumulative_bytes.saturating_sub(file_bytes);
-                file_bytes = 0;
-                let _ = tokio::fs::remove_file(&partial_path).await;
-                continue;
-            }
-
-            if !status.is_success() && status != reqwest::StatusCode::PARTIAL_CONTENT {
-                let is_transient = status.is_server_error()
-                    || status == reqwest::StatusCode::REQUEST_TIMEOUT
-                    || status == reqwest::StatusCode::TOO_MANY_REQUESTS;
-
-                if !is_transient || retries >= Self::MAX_RETRIES {
-                    anyhow::bail!("Failed to download: HTTP {}", status);
-                }
-                retries += 1;
-                let delay = std::cmp::min(
-                    Self::RETRY_BASE_DELAY * 2u32.saturating_pow(retries - 1),
-                    Self::RETRY_MAX_DELAY,
-                );
-                info!(model_id = %model_id, retry = retries, http_status = %status, "Retrying download after transient HTTP error");
-                Self::cancellable_sleep(delay, downloads, model_id).await?;
-                continue;
-            }
-
-            if file_bytes > 0 && status == reqwest::StatusCode::OK {
-                info!(model_id = %model_id, "Server ignored Range header, restarting file from scratch");
-                // Subtract already-counted partial bytes from cumulative
-                *cumulative_bytes = cumulative_bytes.saturating_sub(file_bytes);
-                file_bytes = 0;
-                let _ = tokio::fs::remove_file(&partial_path).await;
-            }
-
-            // If HEAD didn't return this file's size, learn it from the GET response.
-            // This block only fires once per file (file_total stays non-zero after),
-            // so retries don't double-count. Since download_files_sequentially's HEAD
-            // pass contributed 0 for this file, we add the discovered size to the
-            // shared total so progress/ETA are accurate.
-            if file_total == 0 {
-                let new_file_total = if file_bytes > 0 {
-                    response
-                        .headers()
-                        .get("content-range")
-                        .and_then(|v| v.to_str().ok())
-                        .and_then(|s| s.rsplit('/').next())
-                        .and_then(|s| s.parse::<u64>().ok())
-                } else {
-                    response.content_length()
-                };
-                if let Some(t) = new_file_total {
-                    file_total = t;
-                    if let Ok(mut dl) = downloads.lock() {
-                        if let Some(progress) = dl.get_mut(model_id) {
-                            progress.total_bytes = progress.total_bytes.saturating_add(t);
-                        }
-                    }
-                }
-            }
-
-            let mut file = tokio::fs::OpenOptions::new()
-                .create(true)
-                .append(true)
-                .open(&partial_path)
-                .await?;
-
-            let file_len = tokio::fs::metadata(&partial_path).await?.len();
-            if file_len != file_bytes {
-                file.set_len(file_bytes).await?;
-            }
-
-            let mut stream_error = false;
-            let mut resp = response;
-
-            loop {
-                let chunk_result = resp.chunk().await;
-                match chunk_result {
-                    Ok(Some(chunk)) => {
-                        if Self::is_cancelled(downloads, model_id) {
-                            let _ = tokio::fs::remove_file(&partial_path).await;
-                            anyhow::bail!("Download cancelled");
-                        }
-
-                        file.write_all(&chunk).await?;
-                        let chunk_len = chunk.len() as u64;
-                        file_bytes += chunk_len;
-                        *cumulative_bytes += chunk_len;
-
-                        let elapsed = start_time.elapsed().as_secs_f64();
-                        let bytes_this_session = cumulative_bytes.saturating_sub(bytes_at_start);
-                        let speed_bps = if elapsed > 0.0 {
-                            Some((bytes_this_session as f64 / elapsed) as u64)
-                        } else {
-                            None
-                        };
-
-                        let current_total = if let Ok(dl) = downloads.lock() {
-                            dl.get(model_id).map(|p| p.total_bytes).unwrap_or(0)
-                        } else {
-                            0
-                        };
-
-                        let eta_seconds = if let Some(speed) = speed_bps {
-                            if speed > 0 && current_total > 0 {
-                                Some(current_total.saturating_sub(*cumulative_bytes) / speed)
-                            } else {
-                                None
-                            }
-                        } else {
-                            None
-                        };
-
-                        if let Ok(mut dl) = downloads.lock() {
-                            if let Some(progress) = dl.get_mut(model_id) {
-                                progress.bytes_downloaded = *cumulative_bytes;
-                                progress.progress_percent = if current_total > 0 {
-                                    (*cumulative_bytes as f64 / current_total as f64 * 100.0) as f32
-                                } else {
-                                    0.0
-                                };
-                                progress.speed_bps = speed_bps;
-                                progress.eta_seconds = eta_seconds;
-                            }
-                        }
-                    }
-                    Ok(None) => break,
-                    Err(e) => {
-                        info!(model_id = %model_id, bytes = *cumulative_bytes, error = %e, "Download stream interrupted, will retry");
-                        stream_error = true;
-                        break;
-                    }
-                }
-            }
-
-            file.flush().await?;
-            drop(file);
-
-            if stream_error {
-                if retries >= Self::MAX_RETRIES {
-                    anyhow::bail!(
-                        "Download failed after {} retries due to stream interruption",
-                        retries
-                    );
-                }
-                retries += 1;
-                let delay = std::cmp::min(
-                    Self::RETRY_BASE_DELAY * 2u32.saturating_pow(retries - 1),
-                    Self::RETRY_MAX_DELAY,
-                );
-                info!(model_id = %model_id, retry = retries, delay_secs = ?delay.as_secs(), "Retrying download with resume");
-                Self::cancellable_sleep(delay, downloads, model_id).await?;
-                continue;
-            }
-
-            break;
-        }
-
-        tokio::fs::rename(&partial_path, destination).await?;
-        Ok(())
-    }
-
-    pub fn clear_completed(&self, model_id: &str) {
-        if let Ok(mut downloads) = self.downloads.lock() {
-            if let Some(progress) = downloads.get(model_id) {
-                let is_terminal = progress.status == DownloadStatus::Completed
-                    || progress.status == DownloadStatus::Failed
-                    || progress.status == DownloadStatus::Cancelled;
-                if is_terminal && progress.task_exited {
-                    downloads.remove(model_id);
-                }
-            }
-        }
-    }
-
-    fn apply_bearer_token(
-        request: reqwest::RequestBuilder,
-        bearer_token: Option<&str>,
-    ) -> reqwest::RequestBuilder {
-        if let Some(token) = bearer_token.filter(|token| !token.is_empty()) {
-            request.header("Authorization", format!("Bearer {}", token))
-        } else {
-            request
-        }
-    }
-}
-
-static DOWNLOAD_MANAGER: once_cell::sync::Lazy<DownloadManager> =
-    once_cell::sync::Lazy::new(DownloadManager::new);
-
-pub fn get_download_manager() -> &'static DownloadManager {
-    &DOWNLOAD_MANAGER
-}

crates/goose-local-inference/Cargo.toml

@@ -1,54 +0,0 @@
-[package]
-name = "goose-local-inference"
-version = "0.1.0-alpha.0"
-edition.workspace = true
-rust-version.workspace = true
-authors.workspace = true
-license.workspace = true
-repository.workspace = true
-description.workspace = true
-
-[lints]
-workspace = true
-
-[features]
-default = []
-cuda = ["llama-cpp-2/cuda"]
-vulkan = ["llama-cpp-2/vulkan"]
-mlx = ["dep:safemlx", "dep:safemlx-lm", "dep:safemlx-lm-utils"]
-
-[dependencies]
-anyhow = { workspace = true }
-async-stream = { workspace = true }
-async-trait = { workspace = true }
-base64 = { workspace = true }
-chrono = { workspace = true }
-etcetera = { workspace = true }
-encoding_rs = { version = "0.8.35", default-features = false }
-fs2 = { workspace = true }
-futures = { workspace = true }
-goose-download-manager = { version = "0.1.0-alpha.0", path = "../goose-download-manager" }
-goose-provider-types = { version = "0.1.0-alpha.0", path = "../goose-provider-types", default-features = false }
-goose-sdk-types = { version = "0.1.0-alpha.0", path = "../goose-sdk-types", default-features = false }
-hf-hub = { version = "1.0.0-rc.1", default-features = false }
-include_dir = { workspace = true }
-llama-cpp-2 = { workspace = true }
-llama-cpp-sys-2 = { workspace = true }
-minijinja = { version = "2.18", default-features = false, features = ["loader", "multi_template", "serde"] }
-reqwest = { workspace = true, features = ["json", "stream"] }
-regex = { workspace = true }
-rmcp = { workspace = true }
-serde = { workspace = true }
-serde_json = { workspace = true }
-tokio = { workspace = true, features = ["fs", "rt-multi-thread", "sync", "time"] }
-tracing = { workspace = true }
-utoipa = { workspace = true, features = ["chrono"] }
-uuid = { workspace = true, features = ["v4", "std"] }
-tempfile = { workspace = true }
-
-safemlx = { default-features = false, features = ["accelerate", "metal", "safetensors"], optional = true, version = "0.1.2" }
-safemlx-lm = { optional = true, version = "0.1.5" }
-safemlx-lm-utils = { optional = true, version = "0.1.2" }
-
-[target.'cfg(target_os = "macos")'.dependencies]
-llama-cpp-2 = { workspace = true, features = ["sampler", "metal", "mtmd"] }

crates/goose-local-inference/src/config_resolver.rs

@@ -1,30 +0,0 @@
-use anyhow::Result;
-use std::sync::OnceLock;
-
-pub type StringParamResolver = fn(&'static str) -> Result<Option<String>>;
-pub type BoolParamResolver = fn(&'static str) -> Result<Option<bool>>;
-
-static STRING_PARAM_RESOLVER: OnceLock<StringParamResolver> = OnceLock::new();
-static BOOL_PARAM_RESOLVER: OnceLock<BoolParamResolver> = OnceLock::new();
-
-pub fn set_string_param_resolver(resolve_param: StringParamResolver) {
-    let _ = STRING_PARAM_RESOLVER.set(resolve_param);
-}
-
-pub fn set_bool_param_resolver(resolve_param: BoolParamResolver) {
-    let _ = BOOL_PARAM_RESOLVER.set(resolve_param);
-}
-
-pub fn string_param(key: &'static str) -> Result<Option<String>> {
-    match STRING_PARAM_RESOLVER.get() {
-        Some(resolve_param) => resolve_param(key),
-        None => Ok(None),
-    }
-}
-
-pub fn bool_param(key: &'static str) -> Result<Option<bool>> {
-    match BOOL_PARAM_RESOLVER.get() {
-        Some(resolve_param) => resolve_param(key),
-        None => Ok(None),
-    }
-}

crates/goose-local-inference/src/huggingface_auth.rs

@@ -1,64 +0,0 @@
-use crate::{config_resolver, paths::Paths};
-use anyhow::Result;
-use chrono::{DateTime, Utc};
-use futures::future::BoxFuture;
-use serde::{Deserialize, Serialize};
-use std::path::{Path, PathBuf};
-use std::sync::OnceLock;
-
-pub const HUGGINGFACE_TOKEN_SECRET_KEY: &str = "HF_TOKEN";
-pub const HUGGINGFACE_OAUTH_CACHE_PATH: &str = "huggingface/oauth/tokens.json";
-
-pub type TokenResolver = fn() -> BoxFuture<'static, Result<Option<String>>>;
-
-static TOKEN_RESOLVER: OnceLock<TokenResolver> = OnceLock::new();
-
-#[derive(Debug, Clone, Serialize, Deserialize)]
-pub struct HuggingFaceTokenData {
-    pub access_token: String,
-    #[serde(default)]
-    pub refresh_token: Option<String>,
-    #[serde(default)]
-    pub expires_at: Option<DateTime<Utc>>,
-}
-
-impl HuggingFaceTokenData {
-    pub fn is_expired(&self) -> bool {
-        self.expires_at
-            .is_some_and(|expires_at| expires_at <= Utc::now())
-    }
-}
-
-pub fn oauth_cache_path() -> PathBuf {
-    Paths::in_config_dir(HUGGINGFACE_OAUTH_CACHE_PATH)
-}
-
-fn load_oauth_token_from_path(path: &Path) -> Option<HuggingFaceTokenData> {
-    let contents = std::fs::read_to_string(path).ok()?;
-    serde_json::from_str(&contents).ok()
-}
-
-pub fn usable_oauth_token() -> Option<String> {
-    let token = load_oauth_token_from_path(&oauth_cache_path())?;
-    (!token.is_expired()).then_some(token.access_token)
-}
-
-pub fn hf_token_secret() -> Result<Option<String>> {
-    Ok(config_resolver::string_param(HUGGINGFACE_TOKEN_SECRET_KEY)?
-        .filter(|token| !token.trim().is_empty()))
-}
-
-pub fn set_token_resolver(resolve_token: TokenResolver) {
-    let _ = TOKEN_RESOLVER.set(resolve_token);
-}
-
-pub async fn resolve_token_async() -> Result<Option<String>> {
-    if let Some(resolve_token) = TOKEN_RESOLVER.get() {
-        return resolve_token().await;
-    }
-
-    if let Some(token) = usable_oauth_token() {
-        return Ok(Some(token));
-    }
-    hf_token_secret()
-}

crates/goose-local-inference/src/lib.rs

@@ -1,950 +0,0 @@
-pub mod config_resolver;
-pub use goose_download_manager as download_manager;
-pub mod huggingface_auth;
-pub mod paths;
-pub mod prompt_template;
-pub mod provider_utils;
-
-mod backend;
-pub mod hf_models;
-mod llamacpp;
-pub mod local_model_registry;
-pub mod management;
-mod mlx;
-pub(crate) mod multimodal;
-#[cfg(feature = "mlx")]
-mod native_tool_parsing;
-pub(crate) mod thinking_output;
-#[cfg(feature = "mlx")]
-mod tool_emulation;
-mod tool_parsing;
-
-use anyhow::Result;
-use async_stream::try_stream;
-use async_trait::async_trait;
-use backend::{BackendLoadedModel, LocalInferenceBackend};
-use goose_provider_types::base::{MessageStream, Provider, ProviderDescriptor, ProviderMetadata};
-use goose_provider_types::conversation::message::{
-    Message, MessageContent, SystemNotificationType,
-};
-use goose_provider_types::conversation::token_usage::{ProviderUsage, Usage};
-use goose_provider_types::errors::ProviderError;
-use goose_provider_types::images::ImageFormat;
-use goose_provider_types::model::ModelConfig;
-use goose_provider_types::request_log::{start_log, LoggerHandleExt, RequestLogHandle};
-use llamacpp::{LlamaCppBackend, LLAMACPP_BACKEND_ID};
-use local_model_registry::ChatTemplate;
-use mlx::{MlxBackend, MLX_BACKEND_ID};
-use rmcp::model::Tool;
-use serde_json::{json, Value};
-use std::collections::{HashMap, HashSet};
-use std::path::PathBuf;
-use std::sync::{Arc, Mutex as StdMutex};
-use tokio::sync::{Mutex, Notify};
-use uuid::Uuid;
-
-type ModelSlotHandle = Arc<ModelSlot>;
-
-struct ModelSlot {
-    state: Mutex<ModelSlotState>,
-    notify: Notify,
-}
-
-enum ModelSlotState {
-    Empty,
-    Loading,
-    Loaded(Box<dyn BackendLoadedModel>),
-}
-
-impl ModelSlot {
-    fn new() -> Self {
-        Self {
-            state: Mutex::new(ModelSlotState::Empty),
-            notify: Notify::new(),
-        }
-    }
-}
-
-#[derive(Clone, Debug, Eq, Hash, PartialEq)]
-struct ModelCacheKey {
-    backend_id: &'static str,
-    model_id: String,
-    chat_template: ChatTemplate,
-}
-
-impl ModelCacheKey {
-    fn new(
-        backend_id: &'static str,
-        model_id: impl Into<String>,
-        chat_template: ChatTemplate,
-    ) -> Self {
-        Self {
-            backend_id,
-            model_id: model_id.into(),
-            chat_template,
-        }
-    }
-}
-
-pub struct InferenceRuntime {
-    models: StdMutex<HashMap<ModelCacheKey, ModelSlotHandle>>,
-    cold_load_lock: Mutex<()>,
-    backends: HashMap<&'static str, Arc<dyn LocalInferenceBackend>>,
-}
-
-pub fn builtin_chat_template_names() -> Vec<String> {
-    llamacpp::builtin_chat_template_names()
-}
-
-static RUNTIME: StdMutex<Option<Arc<InferenceRuntime>>> = StdMutex::new(None);
-
-fn current_runtime() -> Option<Arc<InferenceRuntime>> {
-    RUNTIME.lock().expect("runtime lock poisoned").clone()
-}
-
-impl InferenceRuntime {
-    pub fn get_or_init() -> Result<Arc<Self>> {
-        let mut guard = RUNTIME.lock().expect("runtime lock poisoned");
-        if let Some(runtime) = guard.as_ref() {
-            return Ok(runtime.clone());
-        }
-        let llamacpp_backend: Arc<dyn LocalInferenceBackend> = Arc::new(LlamaCppBackend::new()?);
-        let mlx_backend: Arc<dyn LocalInferenceBackend> = Arc::new(MlxBackend::new());
-        let mut backends = HashMap::new();
-        backends.insert(LLAMACPP_BACKEND_ID, llamacpp_backend);
-        backends.insert(MLX_BACKEND_ID, mlx_backend);
-        let runtime = Arc::new(Self {
-            models: StdMutex::new(HashMap::new()),
-            cold_load_lock: Mutex::new(()),
-            backends,
-        });
-        *guard = Some(runtime.clone());
-        Ok(runtime)
-    }
-
-    fn default_backend(&self) -> &dyn LocalInferenceBackend {
-        self.backends
-            .get(LLAMACPP_BACKEND_ID)
-            .expect("default local inference backend registered")
-            .as_ref()
-    }
-
-    fn backend_for_model(
-        &self,
-        resolved: &ResolvedModelPaths,
-    ) -> Result<Arc<dyn LocalInferenceBackend>, ProviderError> {
-        let backend_id = resolved
-            .backend_id
-            .as_deref()
-            .unwrap_or(LLAMACPP_BACKEND_ID);
-        self.backends.get(backend_id).cloned().ok_or_else(|| {
-            ProviderError::ExecutionError(format!(
-                "Local inference backend '{}' unavailable",
-                backend_id
-            ))
-        })
-    }
-
-    fn get_or_create_model_slot(&self, key: ModelCacheKey) -> ModelSlotHandle {
-        let mut map = self.models.lock().expect("model cache lock poisoned");
-        map.entry(key)
-            .or_insert_with(|| Arc::new(ModelSlot::new()))
-            .clone()
-    }
-
-    fn model_slot(&self, key: &ModelCacheKey) -> Option<ModelSlotHandle> {
-        let map = self.models.lock().expect("model cache lock poisoned");
-        map.get(key).cloned()
-    }
-
-    fn other_model_slots(&self, keep_key: &ModelCacheKey) -> Vec<ModelSlotHandle> {
-        let map = self.models.lock().expect("model cache lock poisoned");
-        map.iter()
-            .filter(|(key, _)| *key != keep_key)
-            .map(|(_, slot)| slot.clone())
-            .collect()
-    }
-}
-
-pub async fn is_model_loaded(model_name: &str) -> Result<bool, ProviderError> {
-    let resolved = match resolve_model_path(model_name) {
-        Some(resolved) => resolved,
-        None => return Ok(false),
-    };
-    let runtime = InferenceRuntime::get_or_init().map_err(|error| {
-        ProviderError::ExecutionError(format!("Failed to initialize local inference: {error}"))
-    })?;
-    let backend = runtime.backend_for_model(&resolved)?;
-    let key = ModelCacheKey::new(
-        backend.id(),
-        model_name.to_string(),
-        resolved.settings.chat_template,
-    );
-    let Some(slot) = runtime.model_slot(&key) else {
-        return Ok(false);
-    };
-
-    let state = slot.state.lock().await;
-    Ok(matches!(*state, ModelSlotState::Loaded(_)))
-}
-
-pub async fn loaded_model_ids() -> Result<HashSet<String>, ProviderError> {
-    let Some(runtime) = current_runtime() else {
-        return Ok(HashSet::new());
-    };
-    let slots = {
-        let map = runtime.models.lock().expect("model cache lock poisoned");
-        map.iter()
-            .map(|(key, slot)| (key.model_id.clone(), slot.clone()))
-            .collect::<Vec<_>>()
-    };
-
-    let mut loaded = HashSet::new();
-    for (model_id, slot) in slots {
-        if let Ok(state) = slot.state.try_lock() {
-            if matches!(*state, ModelSlotState::Loaded(_)) {
-                loaded.insert(model_id);
-            }
-        } else {
-            loaded.insert(model_id);
-        }
-    }
-    Ok(loaded)
-}
-
-pub async fn evict_model(model_name: &str) -> Result<bool, ProviderError> {
-    let Some(runtime) = current_runtime() else {
-        return Ok(false);
-    };
-    let slots = {
-        let map = runtime.models.lock().expect("model cache lock poisoned");
-        map.iter()
-            .filter(|(key, _)| key.model_id == model_name)
-            .map(|(_, slot)| slot.clone())
-            .collect::<Vec<_>>()
-    };
-
-    let mut evicted = false;
-    for slot in slots {
-        let mut state = slot.state.lock().await;
-        if matches!(*state, ModelSlotState::Loaded(_)) {
-            *state = ModelSlotState::Empty;
-            evicted = true;
-            slot.notify.notify_waiters();
-        }
-    }
-    Ok(evicted)
-}
-
-const PROVIDER_NAME: &str = "local";
-const DEFAULT_MODEL: &str = "bartowski/Llama-3.2-1B-Instruct-GGUF:Q4_K_M";
-
-pub const LOCAL_LLM_MODEL_CONFIG_KEY: &str = "LOCAL_LLM_MODEL";
-
-#[derive(Clone)]
-pub(crate) struct ResolvedModelPaths {
-    pub model_path: PathBuf,
-    pub context_limit: usize,
-    pub settings: crate::local_model_registry::ModelSettings,
-    pub mmproj_path: Option<PathBuf>,
-    pub backend_id: Option<String>,
-    pub draft_model_path: Option<PathBuf>,
-}
-
-fn resolve_model_local_path(model_id: &str) -> Option<PathBuf> {
-    use crate::local_model_registry::get_registry;
-
-    get_registry()
-        .lock()
-        .ok()?
-        .get_model(model_id)
-        .map(|entry| entry.local_path.clone())
-}
-
-/// Resolve model path, context limit, settings, and mmproj path for a model ID from the registry.
-fn resolve_model_path(model_id: &str) -> Option<ResolvedModelPaths> {
-    use crate::local_model_registry::{default_settings_for_model, get_registry};
-
-    if let Ok(registry) = get_registry().lock() {
-        if let Some(entry) = registry.get_model(model_id) {
-            let ctx = entry.settings.context_size.unwrap_or(0) as usize;
-            let mut settings = entry.settings.clone();
-            let defaults = default_settings_for_model(model_id);
-            settings.vision_capable = defaults.vision_capable;
-            settings.mmproj_size_bytes = entry.mmproj_size_bytes;
-            let mmproj_path = entry.mmproj_path.as_ref().filter(|p| p.exists()).cloned();
-            let backend_id = entry
-                .backend_id
-                .clone()
-                .or_else(|| settings.backend_id.clone());
-            let draft_model = settings
-                .draft_model
-                .clone()
-                .or_else(|| {
-                    config_resolver::string_param("GOOSE_LOCAL_DRAFT_MODEL")
-                        .ok()
-                        .flatten()
-                })
-                .filter(|draft_model| draft_model != model_id);
-            let draft_model_path = draft_model.as_deref().and_then(resolve_model_local_path);
-            return Some(ResolvedModelPaths {
-                model_path: entry.local_path.clone(),
-                context_limit: ctx,
-                settings,
-                mmproj_path,
-                backend_id,
-                draft_model_path,
-            });
-        }
-    }
-
-    None
-}
-
-pub fn available_inference_memory_bytes(runtime: &InferenceRuntime) -> u64 {
-    runtime.default_backend().available_memory_bytes()
-}
-
-pub fn recommend_local_model(runtime: &InferenceRuntime) -> String {
-    use local_model_registry::{get_registry, is_featured_model, FEATURED_MODELS};
-
-    let available_memory = available_inference_memory_bytes(runtime);
-
-    if let Ok(registry) = get_registry().lock() {
-        let mut models: Vec<_> = registry
-            .list_models()
-            .iter()
-            .filter(|m| is_featured_model(&m.id) && m.size_bytes > 0)
-            .collect();
-        models.sort_by_key(|model| std::cmp::Reverse(model.size_bytes));
-
-        // Return largest that fits in available memory
-        for model in &models {
-            if available_memory >= model.size_bytes {
-                return model.id.clone();
-            }
-        }
-
-        // If nothing fits, return smallest
-        if let Some(smallest) = models.last() {
-            return smallest.id.clone();
-        }
-    }
-
-    // Fallback to first featured model
-    FEATURED_MODELS[0].spec.to_string()
-}
-
-fn build_openai_messages_json(
-    system: &str,
-    messages: &[Message],
-    media_marker: Option<&str>,
-) -> String {
-    use goose_provider_types::formats::openai::format_messages;
-
-    let mut arr: Vec<Value> = vec![json!({"role": "system", "content": system})];
-    arr.extend(format_messages(messages, &ImageFormat::OpenAi));
-    strip_image_parts_from_messages(&mut arr);
-    if let Some(marker) = media_marker {
-        convert_text_media_markers(&mut arr, marker);
-    }
-    serde_json::to_string(&arr).unwrap_or_else(|_| "[]".to_string())
-}
-
-fn build_openai_text_messages_json(
-    system: &str,
-    messages: &[Message],
-    media_marker: Option<&str>,
-) -> String {
-    let mut arr: Vec<Value> = vec![json!({"role": "system", "content": system})];
-    arr.extend(messages.iter().filter_map(|m| {
-        let content = extract_text_content(m);
-        if content.trim().is_empty() {
-            return None;
-        }
-        let role = match m.role {
-            rmcp::model::Role::User => "user",
-            rmcp::model::Role::Assistant => "assistant",
-        };
-        Some(json!({"role": role, "content": content}))
-    }));
-    if let Some(marker) = media_marker {
-        convert_text_media_markers(&mut arr, marker);
-    }
-    serde_json::to_string(&arr).unwrap_or_else(|_| "[]".to_string())
-}
-
-fn convert_text_media_markers(messages: &mut [Value], marker: &str) {
-    if marker.is_empty() {
-        return;
-    }
-
-    for msg in messages {
-        let Some(content) = msg.get_mut("content") else {
-            continue;
-        };
-
-        if let Some(text) = content.as_str() {
-            if let Some(parts) = split_media_marker_text(text, marker) {
-                *content = json!(parts);
-            }
-            continue;
-        }
-
-        let Some(content_parts) = content.as_array_mut() else {
-            continue;
-        };
-        let mut updated = Vec::new();
-        let mut changed = false;
-        for part in content_parts.iter() {
-            if part.get("type").and_then(|v| v.as_str()) == Some("text") {
-                if let Some(text) = part.get("text").and_then(|v| v.as_str()) {
-                    if let Some(parts) = split_media_marker_text(text, marker) {
-                        updated.extend(parts);
-                        changed = true;
-                        continue;
-                    }
-                }
-            }
-            updated.push(part.clone());
-        }
-        if changed {
-            *content_parts = updated;
-        }
-    }
-}
-
-fn split_media_marker_text(text: &str, marker: &str) -> Option<Vec<Value>> {
-    let mut parts = Vec::new();
-    let mut rest = text;
-    let mut found_marker = false;
-    while let Some((before, after)) = rest.split_once(marker) {
-        found_marker = true;
-        let before = before.strip_suffix('\n').unwrap_or(before);
-        if !before.is_empty() {
-            parts.push(json!({"type": "text", "text": before}));
-        }
-        parts.push(json!({"type": "media_marker", "text": marker}));
-        rest = after;
-        rest = rest.strip_prefix('\n').unwrap_or(rest);
-    }
-    if !found_marker {
-        return None;
-    }
-    if !rest.is_empty() {
-        parts.push(json!({"type": "text", "text": rest}));
-    }
-    Some(parts)
-}
-
-/// Remove `image_url` content parts from OpenAI-format messages JSON, replacing
-/// each with a text note. This prevents an FFI crash in llama.cpp which does not
-/// accept `image_url` content-part types.
-fn strip_image_parts_from_messages(messages: &mut [Value]) {
-    let mut stripped = false;
-    for msg in messages.iter_mut() {
-        if let Some(content) = msg.get_mut("content").and_then(|c| c.as_array_mut()) {
-            for part in content.iter_mut() {
-                if part.get("type").and_then(|t| t.as_str()) == Some("image_url") {
-                    *part = json!({
-                        "type": "text",
-                        "text": "[Image attached — image input is not supported with the currently selected model]"
-                    });
-                    stripped = true;
-                }
-            }
-        }
-    }
-    if stripped {
-        tracing::warn!("Stripped image content parts from messages — vision encoder not available for this model");
-    }
-}
-
-/// Convert a message into plain text for the emulator path's chat history.
-///
-/// This is the emulator-path counterpart of [`format_messages`] used by the native
-/// path. It reconstructs the text-based tool syntax that the emulator prompt teaches
-/// the model:
-///
-/// - `ToolRequest` with a `"command"` argument → `$ command`
-/// - `ToolRequest` with a `"code"` argument → `` ```execute_typescript\n…\n``` ``
-/// - `ToolResponse` → `Command output:\n…`
-///
-/// Only `developer__shell` and `code_execution__execute_typescript` style tool calls are
-/// recognized (by argument shape, not tool name). Tool calls from other extensions
-/// (e.g. custom MCP tools made by a native-tool-calling model earlier in the
-/// conversation) are silently dropped, since the emulator path has no syntax to
-/// represent them.
-fn extract_text_content(msg: &Message) -> String {
-    let mut parts = Vec::new();
-
-    for content in &msg.content {
-        match content {
-            MessageContent::Text(text) => {
-                let text = strip_info_messages(&text.text);
-                if !text.trim().is_empty() {
-                    parts.push(text);
-                }
-            }
-            MessageContent::ToolRequest(req) => {
-                if let Ok(call) = &req.tool_call {
-                    if let Some(cmd) = call
-                        .arguments
-                        .as_ref()
-                        .and_then(|a| a.get("command"))
-                        .and_then(|v| v.as_str())
-                    {
-                        parts.push(format!("$ {}", cmd));
-                    } else if let Some(code) = call
-                        .arguments
-                        .as_ref()
-                        .and_then(|a| a.get("code"))
-                        .and_then(|v| v.as_str())
-                    {
-                        parts.push(format!("```execute_typescript\n{}\n```", code));
-                    }
-                }
-            }
-            MessageContent::ToolResponse(response) => match &response.tool_result {
-                Ok(result) => {
-                    let mut output_parts = Vec::new();
-                    for content_item in &result.content {
-                        if let Some(text_content) = content_item.as_text() {
-                            output_parts.push(text_content.text.to_string());
-                        }
-                    }
-                    if !output_parts.is_empty() {
-                        parts.push(format!("Command output:\n{}", output_parts.join("\n")));
-                    }
-                }
-                Err(e) => {
-                    parts.push(format!("Command error: {}", e));
-                }
-            },
-            MessageContent::Image(_) => {
-                parts.push(
-                    "[Image attached — image input is not supported with the currently selected model]"
-                        .to_string(),
-                );
-            }
-            _ => {}
-        }
-    }
-
-    parts.join("\n")
-}
-
-fn strip_info_messages(text: &str) -> String {
-    let mut remaining = text;
-    let mut output = String::new();
-
-    while let Some((before, after_start)) = remaining.split_once("<info-msg>") {
-        output.push_str(before);
-        if let Some((_, after_end)) = after_start.split_once("</info-msg>") {
-            remaining = after_end;
-        } else {
-            remaining = "";
-            break;
-        }
-    }
-
-    output.push_str(remaining);
-    output.trim().to_string()
-}
-
-/// Build a `ProviderUsage` and write the request log entry.
-fn finalize_usage(
-    log: &mut Option<Box<dyn RequestLogHandle>>,
-    model_name: String,
-    path_label: &str,
-    prompt_token_count: usize,
-    output_token_count: i32,
-    extra_log_fields: Option<(&str, &str)>,
-) -> ProviderUsage {
-    let input_tokens = prompt_token_count as i32;
-    let total_tokens = input_tokens + output_token_count;
-    let usage = Usage::new(
-        Some(input_tokens),
-        Some(output_token_count),
-        Some(total_tokens),
-    );
-    let mut log_json = serde_json::json!({
-        "path": path_label,
-        "prompt_tokens": input_tokens,
-        "output_tokens": output_token_count,
-    });
-    if let Some((key, value)) = extra_log_fields {
-        log_json[key] = serde_json::json!(value);
-    }
-    let _ = log.write(&log_json, Some(&usage));
-    ProviderUsage::new(model_name, usage)
-}
-
-type StreamSender =
-    tokio::sync::mpsc::Sender<Result<(Option<Message>, Option<ProviderUsage>), ProviderError>>;
-
-pub struct LocalInferenceProvider {
-    runtime: Arc<InferenceRuntime>,
-    name: String,
-}
-
-impl LocalInferenceProvider {
-    pub async fn from_env() -> Result<Self> {
-        let runtime = InferenceRuntime::get_or_init()?;
-        Ok(Self {
-            runtime,
-            name: PROVIDER_NAME.to_string(),
-        })
-    }
-}
-
-impl ProviderDescriptor for LocalInferenceProvider {
-    fn metadata() -> ProviderMetadata
-    where
-        Self: Sized,
-    {
-        use crate::local_model_registry::{get_registry, FEATURED_MODELS};
-
-        let mut known_models: Vec<&str> = FEATURED_MODELS.iter().map(|m| m.spec).collect();
-
-        // Add any registry models not already in the featured list
-        let mut dynamic_models = Vec::new();
-        if let Ok(registry) = get_registry().lock() {
-            for entry in registry.list_models() {
-                if !known_models.contains(&entry.id.as_str()) {
-                    dynamic_models.push(entry.id.clone());
-                }
-            }
-        }
-        let dynamic_refs: Vec<&str> = dynamic_models.iter().map(|s| s.as_str()).collect();
-        known_models.extend(dynamic_refs);
-
-        ProviderMetadata::new(
-            PROVIDER_NAME,
-            "Local Inference",
-            "Local inference using quantized GGUF models (llama.cpp)",
-            DEFAULT_MODEL,
-            known_models,
-            "https://github.com/utilityai/llama-cpp-rs",
-            vec![],
-        )
-    }
-}
-
-#[async_trait]
-impl Provider for LocalInferenceProvider {
-    fn get_name(&self) -> &str {
-        &self.name
-    }
-
-    async fn fetch_supported_models(&self) -> Result<Vec<String>, ProviderError> {
-        use crate::local_model_registry::get_registry;
-
-        let mut all_models: Vec<String> = Vec::new();
-
-        if let Ok(registry) = get_registry().lock() {
-            for entry in registry.list_models() {
-                all_models.push(entry.id.clone());
-            }
-        }
-
-        Ok(all_models)
-    }
-
-    async fn stream(
-        &self,
-        model_config: &ModelConfig,
-        system: &str,
-        messages: &[Message],
-        tools: &[Tool],
-    ) -> Result<MessageStream, ProviderError> {
-        let resolved = resolve_model_path(&model_config.model_name).ok_or_else(|| {
-            ProviderError::ExecutionError(format!("Model not found: {}", model_config.model_name))
-        })?;
-        let backend = self.runtime.backend_for_model(&resolved)?;
-        let model_context_limit = resolved.context_limit;
-
-        // Allow request_params to override thinking
-        let mut model_settings = resolved.settings.clone();
-        if let Some(false) = model_config
-            .request_param::<bool>("enable_thinking")
-            .or_else(|| {
-                config_resolver::bool_param("GOOSE_LOCAL_ENABLE_THINKING")
-                    .ok()
-                    .flatten()
-            })
-        {
-            model_settings.enable_thinking = false;
-        }
-
-        let cache_key = ModelCacheKey::new(
-            backend.id(),
-            model_config.model_name.clone(),
-            model_settings.chat_template.clone(),
-        );
-        let model_slot = self.runtime.get_or_create_model_slot(cache_key.clone());
-        let runtime = self.runtime.clone();
-
-        let cache_key = cache_key.clone();
-        let model_arc = model_slot.clone();
-        let backend = backend.clone();
-        let model_name = model_config.model_name.clone();
-        let temperature = model_config.temperature;
-        let max_tokens = model_config.max_tokens;
-        let context_limit = model_context_limit;
-        let settings = model_settings;
-        let resolved_model = resolved.clone();
-        let system = system.to_string();
-        let messages = messages.to_vec();
-        let tools = tools.to_vec();
-        let log_payload = serde_json::json!({
-            "system": &system,
-            "messages": messages.iter().map(|m| {
-                serde_json::json!({
-                    "role": match m.role { rmcp::model::Role::User => "user", rmcp::model::Role::Assistant => "assistant" },
-                    "content": extract_text_content(m),
-                })
-            }).collect::<Vec<_>>(),
-            "tools": tools.iter().map(|t| &t.name).collect::<Vec<_>>(),
-            "settings": {
-                "tool_calling": settings.tool_calling,
-                "chat_template": settings.chat_template,
-                "context_size": settings.context_size,
-                "sampling": settings.sampling,
-            },
-        });
-
-        let (tx, mut rx) = tokio::sync::mpsc::channel::<
-            Result<(Option<Message>, Option<ProviderUsage>), ProviderError>,
-        >(32);
-        let mut log = start_log(model_config, &log_payload)?;
-
-        tokio::spawn(async move {
-            let mut model_load_ms = None;
-
-            // Ensure model is loaded — unload any other models first to free memory.
-            loop {
-                let state = model_slot.state.lock().await;
-                match &*state {
-                    ModelSlotState::Loaded(_) => break,
-                    ModelSlotState::Loading => {
-                        let notified = model_slot.notify.notified();
-                        drop(state);
-                        notified.await;
-                    }
-                    ModelSlotState::Empty => {
-                        drop(state);
-
-                        let cold_load_guard = runtime.cold_load_lock.lock().await;
-                        let mut state = model_slot.state.lock().await;
-                        match &*state {
-                            ModelSlotState::Loaded(_) => break,
-                            ModelSlotState::Loading => {
-                                let notified = model_slot.notify.notified();
-                                drop(state);
-                                drop(cold_load_guard);
-                                notified.await;
-                                continue;
-                            }
-                            ModelSlotState::Empty => {}
-                        }
-                        *state = ModelSlotState::Loading;
-                        drop(state);
-
-                        let loading_message = Message::assistant().with_system_notification(
-                            SystemNotificationType::ProgressMessage,
-                            format!("Loading local model {model_name}..."),
-                        );
-                        if tx.send(Ok((Some(loading_message), None))).await.is_err() {
-                            let mut state = model_slot.state.lock().await;
-                            *state = ModelSlotState::Empty;
-                            model_slot.notify.notify_waiters();
-                            return;
-                        }
-
-                        let other_model_slots = runtime.other_model_slots(&cache_key);
-                        for slot in other_model_slots {
-                            let mut other = slot.state.lock().await;
-                            if matches!(*other, ModelSlotState::Loaded(_)) {
-                                tracing::info!("Unloading previous model to free memory");
-                                *other = ModelSlotState::Empty;
-                            }
-                        }
-
-                        let model_id = model_name.clone();
-                        let resolved_for_load = resolved_model.clone();
-                        let settings_for_load = settings.clone();
-                        let backend_for_load = backend.clone();
-                        let load_started = std::time::Instant::now();
-                        let loaded = match tokio::task::spawn_blocking(move || {
-                            backend_for_load.load_model(
-                                &model_id,
-                                &resolved_for_load,
-                                &settings_for_load,
-                            )
-                        })
-                        .await
-                        {
-                            Ok(Ok(loaded)) => loaded,
-                            Ok(Err(err)) => {
-                                let mut state = model_slot.state.lock().await;
-                                *state = ModelSlotState::Empty;
-                                model_slot.notify.notify_waiters();
-                                let _ = log.error(&err);
-                                let _ = tx.send(Err(err)).await;
-                                return;
-                            }
-                            Err(err) => {
-                                let mut state = model_slot.state.lock().await;
-                                *state = ModelSlotState::Empty;
-                                model_slot.notify.notify_waiters();
-                                let err = ProviderError::ExecutionError(err.to_string());
-                                let _ = log.error(&err);
-                                let _ = tx.send(Err(err)).await;
-                                return;
-                            }
-                        };
-                        let elapsed_ms =
-                            u64::try_from(load_started.elapsed().as_millis()).unwrap_or(u64::MAX);
-                        model_load_ms = Some(elapsed_ms);
-                        tracing::info!(
-                            backend = backend.id(),
-                            model = %model_name,
-                            model_load_ms = elapsed_ms,
-                            "Loaded local inference model"
-                        );
-                        let _ = log.write(
-                            &json!({
-                                "path": "model_load",
-                                "backend": backend.id(),
-                                "model": &model_name,
-                                "model_load_ms": elapsed_ms,
-                            }),
-                            None,
-                        );
-
-                        let mut state = model_slot.state.lock().await;
-                        *state = ModelSlotState::Loaded(loaded);
-                        model_slot.notify.notify_waiters();
-                        drop(cold_load_guard);
-                        break;
-                    }
-                }
-            }
-
-            tokio::task::spawn_blocking(move || {
-                // Macro to log errors before sending them through the channel
-                macro_rules! send_err {
-                    ($err:expr) => {{
-                        let err = $err;
-                        let msg = match &err {
-                            ProviderError::ExecutionError(s) => s.as_str(),
-                            ProviderError::ContextLengthExceeded(s) => s.as_str(),
-                            _ => "unknown error",
-                        };
-                        let _ = log.error(msg);
-                        let _ = tx.blocking_send(Err(err));
-                        return;
-                    }};
-                }
-
-                let mut model_guard = model_arc.state.blocking_lock();
-                let loaded = match &mut *model_guard {
-                    ModelSlotState::Loaded(loaded) => loaded.as_mut(),
-                    ModelSlotState::Empty | ModelSlotState::Loading => {
-                        send_err!(ProviderError::ExecutionError(
-                            "Model not loaded".to_string()
-                        ));
-                    }
-                };
-
-                let message_id = Uuid::new_v4().to_string();
-
-                let request = backend::LocalGenerationRequest {
-                    model_name,
-                    system: &system,
-                    messages: &messages,
-                    tools: &tools,
-                    settings: &settings,
-                    temperature,
-                    max_tokens,
-                    context_limit,
-                    model_load_ms,
-                    resolved_model: &resolved_model,
-                    draft_model_path: resolved_model.draft_model_path.clone(),
-                    message_id: &message_id,
-                    tx: &tx,
-                    log: &mut log,
-                };
-
-                let result = backend.generate(loaded, request);
-
-                if let Err(err) = result {
-                    let msg = match &err {
-                        ProviderError::ExecutionError(s) => s.as_str(),
-                        ProviderError::ContextLengthExceeded(s) => s.as_str(),
-                        _ => "unknown error",
-                    };
-                    let _ = log.error(msg);
-                    let _ = tx.blocking_send(Err(err));
-                }
-            });
-        });
-
-        Ok(Box::pin(try_stream! {
-            while let Some(result) = rx.recv().await {
-                let item = result?;
-                yield item;
-            }
-
-        }))
-    }
-}
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn converts_marker_in_string_content_to_media_marker_part() {
-        let mut messages = vec![json!({
-            "role": "user",
-            "content": "look\n<__media__>\nclosely",
-        })];
-
-        convert_text_media_markers(&mut messages, "<__media__>");
-
-        assert_eq!(
-            messages[0]["content"],
-            json!([
-                {"type": "text", "text": "look"},
-                {"type": "media_marker", "text": "<__media__>"},
-                {"type": "text", "text": "closely"},
-            ])
-        );
-    }
-
-    #[test]
-    fn converts_marker_inside_text_content_parts() {
-        let mut messages = vec![json!({
-            "role": "user",
-            "content": [
-                {"type": "text", "text": "<__media__>describe"},
-                {"type": "text", "text": "next"},
-                {"type": "media_marker", "text": "<__media__>"},
-            ],
-        })];
-
-        convert_text_media_markers(&mut messages, "<__media__>");
-
-        assert_eq!(
-            messages[0]["content"],
-            json!([
-                {"type": "media_marker", "text": "<__media__>"},
-                {"type": "text", "text": "describe"},
-                {"type": "text", "text": "next"},
-                {"type": "media_marker", "text": "<__media__>"},
-            ])
-        );
-    }
-}

crates/goose-local-inference/src/management.rs

@@ -1,860 +0,0 @@
-use super::hf_models::{
-    self, register_resolved_model, resolve_local_model_selection, resolve_local_model_spec,
-    resolve_model_spec, HfGgufFile, HfModelInfo, HfModelVariant,
-};
-use super::local_model_registry::{
-    default_settings_for_model, featured_mmproj_spec, get_registry, model_id_from_repo,
-    ChatTemplate, LocalModelEntry, LocalModelStorage, ModelDownloadStatus, ModelSettings,
-    SamplingConfig, ToolCallingMode, FEATURED_MODELS,
-};
-use super::{
-    available_inference_memory_bytes, builtin_chat_template_names, recommend_local_model,
-    InferenceRuntime,
-};
-use crate::download_manager::{get_download_manager, DownloadProgress, DownloadStatus};
-use crate::huggingface_auth;
-use crate::paths::Paths;
-use anyhow::{anyhow, Result};
-use futures::future::join_all;
-use goose_sdk_types::custom_requests::{
-    LocalInferenceBuiltinChatTemplatesListResponse, LocalInferenceChatTemplate,
-    LocalInferenceDownloadProgressDto, LocalInferenceDownloadState, LocalInferenceHfGgufFileDto,
-    LocalInferenceHfModelInfoDto, LocalInferenceHfModelVariantDto,
-    LocalInferenceHuggingFaceRepoVariantsResponse, LocalInferenceHuggingFaceSearchResponse,
-    LocalInferenceModelDownloadRequest, LocalInferenceModelDownloadResponse,
-    LocalInferenceModelDownloadStatusDto, LocalInferenceModelDto, LocalInferenceModelSettingsDto,
-    LocalInferenceModelSettingsReadResponse, LocalInferenceModelSettingsUpdateResponse,
-    LocalInferenceModelsListResponse, LocalInferenceSamplingConfig, LocalInferenceToolCallingMode,
-};
-use std::collections::HashSet;
-use std::path::PathBuf;
-use std::sync::{Arc, OnceLock};
-
-static MANAGEMENT_RUNTIME: OnceLock<Arc<InferenceRuntime>> = OnceLock::new();
-
-#[derive(Clone)]
-struct LocalModelSelection {
-    repo_id: String,
-    backend_id: String,
-    variant_id: Option<String>,
-}
-
-pub async fn list_models() -> Result<LocalInferenceModelsListResponse> {
-    ensure_featured_models_current().await?;
-
-    let runtime = management_runtime()?;
-    let recommended_id = recommend_local_model(&runtime);
-
-    let loaded_model_ids = crate::loaded_model_ids()
-        .await
-        .map_err(|error| anyhow!(error.to_string()))?;
-    let registry = get_registry()
-        .lock()
-        .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-    let mut models: Vec<LocalInferenceModelDto> = registry
-        .list_models()
-        .iter()
-        .map(|entry| local_model_to_dto(entry, &recommended_id, &loaded_model_ids))
-        .collect();
-
-    models.sort_by(|a, b| {
-        let a_downloaded = a.status.state == LocalInferenceDownloadState::Downloaded;
-        let b_downloaded = b.status.state == LocalInferenceDownloadState::Downloaded;
-        match (b_downloaded, a_downloaded) {
-            (true, false) => std::cmp::Ordering::Greater,
-            (false, true) => std::cmp::Ordering::Less,
-            _ => a.id.cmp(&b.id),
-        }
-    });
-
-    Ok(LocalInferenceModelsListResponse { models })
-}
-
-pub async fn search_huggingface_models(
-    query: String,
-    limit: Option<usize>,
-) -> Result<LocalInferenceHuggingFaceSearchResponse> {
-    let limit = limit.unwrap_or(20).min(50);
-    let models = hf_models::search_local_models(&query, limit)
-        .await?
-        .into_iter()
-        .map(hf_model_info_to_dto)
-        .collect();
-    Ok(LocalInferenceHuggingFaceSearchResponse { models })
-}
-
-pub async fn huggingface_repo_variants(
-    repo_id: String,
-) -> Result<LocalInferenceHuggingFaceRepoVariantsResponse> {
-    let variants = hf_models::get_repo_local_variants(&repo_id).await?;
-
-    let runtime = management_runtime()?;
-    let available_memory = available_inference_memory_bytes(&runtime);
-    let gguf_variants: Vec<_> = variants
-        .iter()
-        .filter(|variant| variant.backend_id == "llamacpp")
-        .map(|variant| hf_models::HfQuantVariant {
-            quantization: variant.variant_id.clone(),
-            size_bytes: variant.size_bytes,
-            filename: variant.filename.clone().unwrap_or_default(),
-            download_url: variant.download_url.clone().unwrap_or_default(),
-            description: "",
-            quality_rank: variant.quality_rank,
-            sharded: variant.sharded,
-        })
-        .collect();
-    let recommended_index = hf_models::recommend_variant(&gguf_variants, available_memory);
-
-    let (downloaded_quants, downloaded_variants) = {
-        let registry = get_registry()
-            .lock()
-            .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-        let models: Vec<_> = registry
-            .list_models()
-            .iter()
-            .filter(|m| m.repo_id == repo_id && m.is_downloaded())
-            .collect();
-        (
-            models.iter().map(|m| m.quantization.clone()).collect(),
-            models.iter().map(|m| m.id.clone()).collect(),
-        )
-    };
-
-    Ok(LocalInferenceHuggingFaceRepoVariantsResponse {
-        variants: variants.into_iter().map(hf_model_variant_to_dto).collect(),
-        recommended_index,
-        available_memory_bytes: available_memory,
-        downloaded_quants,
-        downloaded_variants,
-    })
-}
-
-pub async fn download_model(
-    req: LocalInferenceModelDownloadRequest,
-) -> Result<LocalInferenceModelDownloadResponse> {
-    let selection = explicit_model_selection(&req)?;
-    let model_id = local_model_id_from_request(&req, selection.as_ref()).await?;
-    let download_id = format!("{}-model", model_id);
-    let download_reserved = get_download_manager().reserve_download(DownloadProgress {
-        model_id: download_id,
-        status: DownloadStatus::Downloading,
-        bytes_downloaded: 0,
-        total_bytes: 0,
-        progress_percent: 0.0,
-        speed_bps: None,
-        eta_seconds: None,
-        error: None,
-        task_exited: false,
-    })?;
-    if !download_reserved {
-        return Ok(LocalInferenceModelDownloadResponse { model_id });
-    }
-
-    if let Err(error) = register_pending_download_model(&model_id, &req, selection.as_ref()) {
-        mark_download_failed(&model_id, &error);
-        return Err(error.context("Failed to register download"));
-    }
-
-    let spec = req.spec.clone();
-    let selection_for_task = selection.clone();
-    let model_id_for_task = model_id.clone();
-    tokio::spawn(async move {
-        let resolved = if let Some(selection) = selection_for_task {
-            resolve_local_model_selection(
-                &selection.repo_id,
-                &selection.backend_id,
-                selection.variant_id.as_deref(),
-            )
-            .await
-        } else {
-            resolve_local_model_spec(&spec).await
-        };
-        match resolved {
-            Ok(resolved) => {
-                if !model_download_completed(&model_id_for_task) {
-                    return;
-                }
-                if let Err(error) = register_resolved_model(resolved, &spec) {
-                    mark_download_failed(&model_id_for_task, error);
-                }
-            }
-            Err(error) => mark_download_failed(&model_id_for_task, error),
-        }
-    });
-
-    Ok(LocalInferenceModelDownloadResponse { model_id })
-}
-
-pub fn download_progress(model_id: &str) -> Result<Option<LocalInferenceDownloadProgressDto>> {
-    Ok(get_download_manager()
-        .get_progress(&format!("{}-model", model_id))
-        .map(download_progress_to_dto))
-}
-
-pub fn cancel_download(model_id: &str) -> Result<()> {
-    let manager = get_download_manager();
-    manager.cancel_download(&format!("{}-model", model_id))?;
-    let _ = manager.cancel_download(&format!("{}-mmproj", model_id));
-    Ok(())
-}
-
-pub fn delete_model(model_id: &str) -> Result<()> {
-    let mut registry = get_registry()
-        .lock()
-        .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-    if registry.get_model(model_id).is_none() {
-        anyhow::bail!("Model not found");
-    }
-    registry.delete_model(model_id)
-}
-
-pub fn model_exists(model_id: &str) -> Result<bool> {
-    let registry = get_registry()
-        .lock()
-        .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-    Ok(registry.get_model(model_id).is_some())
-}
-
-pub async fn evict_model(model_id: &str) -> Result<()> {
-    crate::evict_model(model_id)
-        .await
-        .map(|_| ())
-        .map_err(|error| anyhow!(error.to_string()))
-}
-
-pub fn get_model_settings(model_id: &str) -> Result<LocalInferenceModelSettingsReadResponse> {
-    let registry = get_registry()
-        .lock()
-        .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-    let settings = registry
-        .get_model_settings(model_id)
-        .ok_or_else(|| anyhow!("Model not found"))?;
-    Ok(LocalInferenceModelSettingsReadResponse {
-        settings: model_settings_to_dto(settings),
-    })
-}
-
-pub fn update_model_settings(
-    model_id: &str,
-    settings: LocalInferenceModelSettingsDto,
-) -> Result<LocalInferenceModelSettingsUpdateResponse> {
-    let settings = model_settings_from_dto(settings);
-    let mut registry = get_registry()
-        .lock()
-        .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-    registry.update_model_settings(model_id, settings.clone())?;
-    Ok(LocalInferenceModelSettingsUpdateResponse {
-        settings: model_settings_to_dto(&settings),
-    })
-}
-
-pub fn list_builtin_chat_templates() -> LocalInferenceBuiltinChatTemplatesListResponse {
-    LocalInferenceBuiltinChatTemplatesListResponse {
-        templates: builtin_chat_template_names(),
-    }
-}
-
-fn management_runtime() -> Result<Arc<InferenceRuntime>> {
-    if let Some(runtime) = MANAGEMENT_RUNTIME.get() {
-        return Ok(runtime.clone());
-    }
-
-    let runtime = InferenceRuntime::get_or_init()?;
-    match MANAGEMENT_RUNTIME.set(runtime.clone()) {
-        Ok(()) => Ok(runtime),
-        Err(_) => Ok(MANAGEMENT_RUNTIME
-            .get()
-            .expect("local inference management runtime initialized by another thread")
-            .clone()),
-    }
-}
-
-pub async fn ensure_featured_models_current() -> Result<()> {
-    let mut mmproj_downloads_needed: Vec<(String, String, PathBuf)> = Vec::new();
-
-    struct PendingResolve {
-        spec: &'static str,
-        repo_id: String,
-        quantization: String,
-        model_id: String,
-    }
-    let mut to_resolve = Vec::new();
-
-    for featured in FEATURED_MODELS {
-        let (repo_id, quantization) = match hf_models::parse_model_spec(featured.spec) {
-            Ok(parts) => parts,
-            Err(_) => continue,
-        };
-
-        let model_id = model_id_from_repo(&repo_id, &quantization);
-
-        {
-            let registry = get_registry()
-                .lock()
-                .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-            if let Some(existing) = registry.get_model(&model_id) {
-                let needs_backfill = existing.mmproj_path.is_none() && featured.mmproj.is_some();
-                let needs_download = existing.is_downloaded()
-                    && featured.mmproj.is_some()
-                    && !existing.mmproj_path.as_ref().is_some_and(|p| p.exists());
-
-                if needs_download {
-                    if let Some(mmproj) = featured.mmproj.as_ref() {
-                        let path = mmproj.local_path();
-                        let url = format!(
-                            "https://huggingface.co/{}/resolve/main/{}",
-                            mmproj.repo, mmproj.filename
-                        );
-                        mmproj_downloads_needed.push((model_id.clone(), url, path));
-                    }
-                }
-
-                if !needs_backfill {
-                    continue;
-                }
-            }
-        }
-
-        to_resolve.push(PendingResolve {
-            spec: featured.spec,
-            repo_id,
-            quantization,
-            model_id,
-        });
-    }
-
-    let resolved: Vec<(PendingResolve, HfGgufFile)> =
-        join_all(to_resolve.into_iter().map(|pending| async move {
-            let hf_file = match resolve_model_spec(pending.spec).await {
-                Ok((_repo, file)) => file,
-                Err(_) => {
-                    let filename = format!(
-                        "{}-{}.gguf",
-                        pending.repo_id.split('/').next_back().unwrap_or("model"),
-                        pending.quantization
-                    );
-                    HfGgufFile {
-                        filename: filename.clone(),
-                        size_bytes: 0,
-                        quantization: pending.quantization.to_string(),
-                        download_url: format!(
-                            "https://huggingface.co/{}/resolve/main/{}",
-                            pending.repo_id, filename
-                        ),
-                    }
-                }
-            };
-            (pending, hf_file)
-        }))
-        .await;
-
-    let entries_to_add: Vec<LocalModelEntry> = resolved
-        .into_iter()
-        .map(|(pending, hf_file)| {
-            let local_path = Paths::in_data_dir("models").join(&hf_file.filename);
-            let settings = default_settings_for_model(&pending.model_id);
-            LocalModelEntry {
-                id: pending.model_id,
-                repo_id: pending.repo_id,
-                filename: hf_file.filename,
-                quantization: pending.quantization,
-                local_path,
-                source_url: hf_file.download_url,
-                backend_id: settings.backend_id.clone(),
-                storage: LocalModelStorage::GooseManaged,
-                settings,
-                size_bytes: hf_file.size_bytes,
-                mmproj_path: None,
-                mmproj_source_url: None,
-                mmproj_size_bytes: 0,
-                mmproj_checked: false,
-                shard_files: vec![],
-            }
-        })
-        .collect();
-
-    {
-        let mut registry = get_registry()
-            .lock()
-            .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-
-        if !entries_to_add.is_empty() {
-            registry.sync_with_featured(entries_to_add);
-        }
-
-        for model in registry.list_models_mut() {
-            model.enrich_with_featured_mmproj();
-            if model.is_downloaded() {
-                if let Some(mmproj) = featured_mmproj_spec(&model.id) {
-                    let path = mmproj.local_path();
-                    if !path.exists() {
-                        let url = format!(
-                            "https://huggingface.co/{}/resolve/main/{}",
-                            mmproj.repo, mmproj.filename
-                        );
-                        mmproj_downloads_needed.push((model.id.clone(), url, path));
-                    }
-                }
-            }
-        }
-        let _ = registry.save();
-    }
-
-    let dm = get_download_manager();
-    let hf_token = huggingface_auth::resolve_token_async().await.ok().flatten();
-    let mut started_paths = std::collections::HashSet::new();
-    for (model_id, url, path) in mmproj_downloads_needed {
-        if !path.exists() && started_paths.insert(path.clone()) {
-            let download_id = format!("{}-mmproj", model_id);
-            let dominated_by_active = dm
-                .get_progress(&download_id)
-                .is_some_and(|p| p.status == DownloadStatus::Downloading);
-            if !dominated_by_active {
-                tracing::info!(model_id = %model_id, "Auto-downloading vision encoder for existing model");
-                if let Err(e) = dm
-                    .download_model_with_bearer_token(
-                        download_id,
-                        url,
-                        path,
-                        hf_token.clone(),
-                        None,
-                    )
-                    .await
-                {
-                    tracing::warn!(model_id = %model_id, error = %e, "Failed to start mmproj download");
-                }
-            }
-        }
-    }
-
-    Ok(())
-}
-
-fn local_model_to_dto(
-    entry: &LocalModelEntry,
-    recommended_id: &str,
-    loaded_model_ids: &HashSet<String>,
-) -> LocalInferenceModelDto {
-    let vision_capable = entry.settings.vision_capable;
-    LocalInferenceModelDto {
-        id: entry.id.clone(),
-        repo_id: entry.repo_id.clone(),
-        filename: entry.filename.clone(),
-        quantization: entry.quantization.clone(),
-        size_bytes: entry.file_size(),
-        status: model_download_status_to_dto(entry.download_status()),
-        recommended: recommended_id == entry.id,
-        is_loaded: loaded_model_ids.contains(&entry.id),
-        settings: model_settings_to_dto(&entry.settings),
-        vision_capable,
-        mmproj_status: vision_capable
-            .then(|| model_download_status_to_dto(entry.mmproj_download_status())),
-    }
-}
-
-fn model_download_status_to_dto(
-    status: ModelDownloadStatus,
-) -> LocalInferenceModelDownloadStatusDto {
-    match status {
-        ModelDownloadStatus::NotDownloaded => LocalInferenceModelDownloadStatusDto {
-            state: LocalInferenceDownloadState::NotDownloaded,
-            ..Default::default()
-        },
-        ModelDownloadStatus::Downloading {
-            progress_percent,
-            bytes_downloaded,
-            total_bytes,
-            speed_bps,
-        } => LocalInferenceModelDownloadStatusDto {
-            state: LocalInferenceDownloadState::Downloading,
-            progress_percent: Some(progress_percent),
-            bytes_downloaded: Some(bytes_downloaded),
-            total_bytes: Some(total_bytes),
-            speed_bps: Some(speed_bps),
-        },
-        ModelDownloadStatus::Downloaded => LocalInferenceModelDownloadStatusDto {
-            state: LocalInferenceDownloadState::Downloaded,
-            ..Default::default()
-        },
-    }
-}
-
-fn download_progress_to_dto(progress: DownloadProgress) -> LocalInferenceDownloadProgressDto {
-    LocalInferenceDownloadProgressDto {
-        model_id: progress.model_id,
-        status: serde_json::to_value(progress.status)
-            .ok()
-            .and_then(|value| value.as_str().map(ToOwned::to_owned))
-            .unwrap_or_else(|| "unknown".to_string()),
-        bytes_downloaded: progress.bytes_downloaded,
-        total_bytes: progress.total_bytes,
-        progress_percent: progress.progress_percent,
-        speed_bps: progress.speed_bps,
-        eta_seconds: progress.eta_seconds,
-        error: progress.error,
-        task_exited: progress.task_exited,
-    }
-}
-
-fn hf_model_info_to_dto(model: HfModelInfo) -> LocalInferenceHfModelInfoDto {
-    LocalInferenceHfModelInfoDto {
-        repo_id: model.repo_id,
-        author: model.author,
-        model_name: model.model_name,
-        downloads: model.downloads,
-        gguf_files: model
-            .gguf_files
-            .into_iter()
-            .map(|file| LocalInferenceHfGgufFileDto {
-                filename: file.filename,
-                size_bytes: file.size_bytes,
-                quantization: file.quantization,
-                download_url: file.download_url,
-            })
-            .collect(),
-        variants: model
-            .variants
-            .into_iter()
-            .map(hf_model_variant_to_dto)
-            .collect(),
-    }
-}
-
-fn hf_model_variant_to_dto(variant: HfModelVariant) -> LocalInferenceHfModelVariantDto {
-    LocalInferenceHfModelVariantDto {
-        variant_id: variant.variant_id,
-        label: variant.label,
-        backend_id: variant.backend_id,
-        format: variant.format,
-        model_id: variant.model_id,
-        download_id: variant.download_id,
-        size_bytes: variant.size_bytes,
-        filename: variant.filename,
-        download_url: variant.download_url,
-        description: variant.description,
-        quality_rank: variant.quality_rank,
-        sharded: variant.sharded,
-        supported: variant.supported,
-        unsupported_reason: variant.unsupported_reason,
-    }
-}
-
-pub fn model_settings_to_dto(settings: &ModelSettings) -> LocalInferenceModelSettingsDto {
-    LocalInferenceModelSettingsDto {
-        backend_id: settings.backend_id.clone(),
-        context_size: settings.context_size,
-        max_output_tokens: settings.max_output_tokens,
-        draft_model: settings.draft_model.clone(),
-        sampling: sampling_to_dto(&settings.sampling),
-        repeat_penalty: settings.repeat_penalty,
-        repeat_last_n: settings.repeat_last_n,
-        frequency_penalty: settings.frequency_penalty,
-        presence_penalty: settings.presence_penalty,
-        n_batch: settings.n_batch,
-        n_gpu_layers: settings.n_gpu_layers,
-        use_mlock: settings.use_mlock,
-        flash_attention: settings.flash_attention,
-        n_threads: settings.n_threads,
-        tool_calling: tool_calling_to_dto(settings.tool_calling),
-        chat_template: chat_template_to_dto(&settings.chat_template),
-        enable_thinking: settings.enable_thinking,
-        vision_capable: settings.vision_capable,
-        image_token_estimate: settings.image_token_estimate,
-        mmproj_size_bytes: settings.mmproj_size_bytes,
-    }
-}
-
-pub fn model_settings_from_dto(settings: LocalInferenceModelSettingsDto) -> ModelSettings {
-    ModelSettings {
-        backend_id: settings.backend_id,
-        context_size: settings.context_size,
-        max_output_tokens: settings.max_output_tokens,
-        draft_model: settings.draft_model,
-        sampling: sampling_from_dto(settings.sampling),
-        repeat_penalty: settings.repeat_penalty,
-        repeat_last_n: settings.repeat_last_n,
-        frequency_penalty: settings.frequency_penalty,
-        presence_penalty: settings.presence_penalty,
-        n_batch: settings.n_batch,
-        n_gpu_layers: settings.n_gpu_layers,
-        use_mlock: settings.use_mlock,
-        flash_attention: settings.flash_attention,
-        n_threads: settings.n_threads,
-        tool_calling: tool_calling_from_dto(settings.tool_calling),
-        chat_template: chat_template_from_dto(settings.chat_template),
-        enable_thinking: settings.enable_thinking,
-        vision_capable: settings.vision_capable,
-        image_token_estimate: settings.image_token_estimate,
-        mmproj_size_bytes: settings.mmproj_size_bytes,
-    }
-}
-
-fn sampling_to_dto(sampling: &SamplingConfig) -> LocalInferenceSamplingConfig {
-    match sampling {
-        SamplingConfig::Greedy => LocalInferenceSamplingConfig::Greedy,
-        SamplingConfig::Temperature {
-            temperature,
-            top_k,
-            top_p,
-            min_p,
-            seed,
-        } => LocalInferenceSamplingConfig::Temperature {
-            temperature: *temperature,
-            top_k: *top_k,
-            top_p: *top_p,
-            min_p: *min_p,
-            seed: *seed,
-        },
-        SamplingConfig::MirostatV2 { tau, eta, seed } => LocalInferenceSamplingConfig::MirostatV2 {
-            tau: *tau,
-            eta: *eta,
-            seed: *seed,
-        },
-    }
-}
-
-fn sampling_from_dto(sampling: LocalInferenceSamplingConfig) -> SamplingConfig {
-    match sampling {
-        LocalInferenceSamplingConfig::Greedy => SamplingConfig::Greedy,
-        LocalInferenceSamplingConfig::Temperature {
-            temperature,
-            top_k,
-            top_p,
-            min_p,
-            seed,
-        } => SamplingConfig::Temperature {
-            temperature,
-            top_k,
-            top_p,
-            min_p,
-            seed,
-        },
-        LocalInferenceSamplingConfig::MirostatV2 { tau, eta, seed } => {
-            SamplingConfig::MirostatV2 { tau, eta, seed }
-        }
-    }
-}
-
-fn tool_calling_to_dto(mode: ToolCallingMode) -> LocalInferenceToolCallingMode {
-    match mode {
-        ToolCallingMode::Auto => LocalInferenceToolCallingMode::Auto,
-        ToolCallingMode::ForceNative => LocalInferenceToolCallingMode::ForceNative,
-        ToolCallingMode::ForceEmulated => LocalInferenceToolCallingMode::ForceEmulated,
-    }
-}
-
-fn tool_calling_from_dto(mode: LocalInferenceToolCallingMode) -> ToolCallingMode {
-    match mode {
-        LocalInferenceToolCallingMode::Auto => ToolCallingMode::Auto,
-        LocalInferenceToolCallingMode::ForceNative => ToolCallingMode::ForceNative,
-        LocalInferenceToolCallingMode::ForceEmulated => ToolCallingMode::ForceEmulated,
-    }
-}
-
-fn chat_template_to_dto(template: &ChatTemplate) -> LocalInferenceChatTemplate {
-    match template {
-        ChatTemplate::Embedded => LocalInferenceChatTemplate::Embedded,
-        ChatTemplate::Builtin { name } => {
-            LocalInferenceChatTemplate::Builtin { name: name.clone() }
-        }
-        ChatTemplate::CustomInline { template } => LocalInferenceChatTemplate::CustomInline {
-            template: template.clone(),
-        },
-    }
-}
-
-fn chat_template_from_dto(template: LocalInferenceChatTemplate) -> ChatTemplate {
-    match template {
-        LocalInferenceChatTemplate::Embedded => ChatTemplate::Embedded,
-        LocalInferenceChatTemplate::Builtin { name } => ChatTemplate::Builtin { name },
-        LocalInferenceChatTemplate::CustomInline { template } => {
-            ChatTemplate::CustomInline { template }
-        }
-    }
-}
-
-fn explicit_model_selection(
-    req: &LocalInferenceModelDownloadRequest,
-) -> Result<Option<LocalModelSelection>> {
-    if let Some(backend_id) = req.backend_id.as_deref() {
-        let (repo_id, parsed_variant_id) = hf_models::parse_model_spec(&req.spec)
-            .map(|(repo_id, quantization)| (repo_id, Some(quantization)))
-            .unwrap_or_else(|_| (req.spec.clone(), None));
-        let variant_id = req.variant_id.clone().or(parsed_variant_id);
-        match backend_id {
-            "mlx" | "llamacpp" => Ok(Some(LocalModelSelection {
-                repo_id,
-                backend_id: backend_id.to_string(),
-                variant_id,
-            })),
-            _ => anyhow::bail!("Unknown local inference backend '{}'", backend_id),
-        }
-    } else {
-        Ok(None)
-    }
-}
-
-async fn local_model_id_from_request(
-    req: &LocalInferenceModelDownloadRequest,
-    selection: Option<&LocalModelSelection>,
-) -> Result<String> {
-    if let Some(selection) = selection {
-        return match selection.backend_id.as_str() {
-            "mlx" => Ok(selection.repo_id.clone()),
-            "llamacpp" => {
-                let quantization = selection.variant_id.as_deref().ok_or_else(|| {
-                    anyhow!(
-                        "llama.cpp model '{}' is missing a quantization",
-                        selection.repo_id
-                    )
-                })?;
-                Ok(model_id_from_repo(&selection.repo_id, quantization))
-            }
-            _ => anyhow::bail!("Unknown local inference backend '{}'", selection.backend_id),
-        };
-    }
-
-    if let Ok((repo_id, quantization)) = hf_models::parse_model_spec(&req.spec) {
-        return Ok(model_id_from_repo(&repo_id, &quantization));
-    }
-
-    let variants = hf_models::get_repo_local_variants(&req.spec).await?;
-    let has_llamacpp = variants
-        .iter()
-        .any(|variant| variant.backend_id == "llamacpp");
-    let mlx_variants: Vec<_> = variants
-        .iter()
-        .filter(|variant| variant.backend_id == "mlx")
-        .collect();
-    if mlx_variants.len() == 1 && !has_llamacpp {
-        Ok(req.spec.clone())
-    } else {
-        anyhow::bail!(
-            "Model spec '{}' is ambiguous; choose one of: {}",
-            req.spec,
-            variants
-                .iter()
-                .map(|variant| variant.download_id.as_str())
-                .collect::<Vec<_>>()
-                .join(", ")
-        )
-    }
-}
-
-fn mark_download_failed(model_id: &str, error: impl std::fmt::Display) {
-    let manager = get_download_manager();
-    let download_id = format!("{}-model", model_id);
-    if manager.get_progress(&download_id).is_none() {
-        manager.set_progress(DownloadProgress {
-            model_id: download_id.clone(),
-            status: DownloadStatus::Failed,
-            bytes_downloaded: 0,
-            total_bytes: 0,
-            progress_percent: 0.0,
-            speed_bps: None,
-            eta_seconds: None,
-            error: Some(error.to_string()),
-            task_exited: true,
-        });
-        return;
-    }
-
-    manager.update_progress(&download_id, |progress| {
-        if progress.status != DownloadStatus::Cancelled {
-            progress.status = DownloadStatus::Failed;
-            progress.error = Some(error.to_string());
-        }
-        progress.task_exited = true;
-    });
-}
-
-fn model_download_completed(model_id: &str) -> bool {
-    get_download_manager()
-        .get_progress(&format!("{}-model", model_id))
-        .is_some_and(|progress| progress.status == DownloadStatus::Completed)
-}
-
-fn register_pending_download_model(
-    model_id: &str,
-    req: &LocalInferenceModelDownloadRequest,
-    selection: Option<&LocalModelSelection>,
-) -> Result<()> {
-    let (repo_id, backend_id, variant_id) = if let Some(selection) = selection {
-        (
-            selection.repo_id.clone(),
-            selection.backend_id.clone(),
-            selection
-                .variant_id
-                .clone()
-                .unwrap_or_else(|| "default".to_string()),
-        )
-    } else if let Ok((repo_id, quantization)) = hf_models::parse_model_spec(&req.spec) {
-        (repo_id, "llamacpp".to_string(), quantization)
-    } else {
-        (req.spec.clone(), "mlx".to_string(), "default".to_string())
-    };
-
-    let mut registry = get_registry()
-        .lock()
-        .map_err(|_| anyhow!("Failed to acquire registry lock"))?;
-    if registry.has_model(model_id) {
-        return Ok(());
-    }
-
-    let mut settings = default_settings_for_model(model_id);
-    if backend_id != "llamacpp" {
-        settings.backend_id = Some(backend_id.clone());
-    }
-
-    let filename = variant_id.clone();
-    registry.add_model(LocalModelEntry {
-        id: model_id.to_string(),
-        repo_id,
-        filename: filename.clone(),
-        quantization: variant_id,
-        local_path: Paths::in_data_dir("models").join(filename),
-        source_url: req.spec.clone(),
-        backend_id: settings.backend_id.clone(),
-        storage: LocalModelStorage::HuggingFaceCache,
-        settings,
-        size_bytes: 0,
-        mmproj_path: None,
-        mmproj_source_url: None,
-        mmproj_size_bytes: 0,
-        mmproj_checked: false,
-        shard_files: vec![],
-    })
-}
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn settings_round_trip_preserves_defaults() {
-        let settings = ModelSettings::default();
-        let dto = model_settings_to_dto(&settings);
-        let round_trip = model_settings_from_dto(dto);
-        assert_eq!(round_trip.repeat_penalty, settings.repeat_penalty);
-        assert_eq!(round_trip.repeat_last_n, settings.repeat_last_n);
-        assert_eq!(round_trip.enable_thinking, settings.enable_thinking);
-        assert_eq!(
-            round_trip.image_token_estimate,
-            settings.image_token_estimate
-        );
-    }
-
-    #[tokio::test]
-    async fn explicit_llamacpp_selection_derives_quantized_model_id() {
-        let req = LocalInferenceModelDownloadRequest {
-            spec: "test/repo".to_string(),
-            backend_id: Some("llamacpp".to_string()),
-            variant_id: Some("Q4_K_M".to_string()),
-        };
-        let selection = explicit_model_selection(&req).unwrap();
-        let model_id = local_model_id_from_request(&req, selection.as_ref())
-            .await
-            .unwrap();
-        assert_eq!(model_id, "test/repo:Q4_K_M");
-    }
-}

crates/goose-local-inference/src/native_tool_parsing.rs

@@ -1,323 +0,0 @@
-use goose_provider_types::conversation::message::{Message, MessageContent};
-use goose_provider_types::errors::ProviderError;
-use goose_provider_types::formats::openai::is_valid_function_name;
-use goose_provider_types::json::safely_parse_json;
-use rmcp::model::{object, CallToolRequestParams, ErrorCode, ErrorData};
-use serde_json::{json, Value};
-use std::borrow::Cow;
-use uuid::Uuid;
-
-pub(crate) fn message_from_native_tool_text(
-    generated_text: &str,
-    message_id: &str,
-) -> Result<Option<Message>, ProviderError> {
-    let mut content = Vec::new();
-
-    if let Some(message) = parse_openai_message_json(generated_text) {
-        append_text(&mut content, message.get("content"));
-        append_tool_calls(&mut content, message.get("tool_calls"));
-    } else if let Some(tool_calls) = parse_tool_calls_json(generated_text) {
-        append_tool_calls(&mut content, Some(&tool_calls));
-    } else if generated_text.contains("<function=") {
-        let (prefix, tool_calls) = parse_xml_tool_calls(generated_text);
-        if let Some(prefix) = prefix {
-            content.push(MessageContent::text(prefix));
-        }
-        content.extend(tool_calls);
-    } else {
-        return Ok(None);
-    }
-
-    if content
-        .iter()
-        .any(|content| matches!(content, MessageContent::ToolRequest(_)))
-    {
-        let mut message = Message::new(
-            rmcp::model::Role::Assistant,
-            chrono::Utc::now().timestamp(),
-            content,
-        );
-        message.id = Some(message_id.to_string());
-        Ok(Some(message))
-    } else {
-        Ok(None)
-    }
-}
-
-fn parse_openai_message_json(generated_text: &str) -> Option<Value> {
-    json_candidates(generated_text)
-        .into_iter()
-        .find(|value| value.get("tool_calls").is_some_and(is_tool_call_array))
-}
-
-fn parse_tool_calls_json(generated_text: &str) -> Option<Value> {
-    for value in json_candidates(generated_text) {
-        if is_tool_call_array(&value) {
-            return Some(value);
-        }
-        if let Some(tool_calls) = value
-            .get("tool_calls")
-            .filter(|value| is_tool_call_array(value))
-        {
-            return Some(tool_calls.clone());
-        }
-        if is_tool_call_value(&value) {
-            return Some(Value::Array(vec![value]));
-        }
-    }
-    None
-}
-
-fn is_tool_call_array(value: &Value) -> bool {
-    value
-        .as_array()
-        .is_some_and(|items| !items.is_empty() && items.iter().all(is_tool_call_value))
-}
-
-fn is_tool_call_value(value: &Value) -> bool {
-    let direct_name = value.get("name").and_then(|name| name.as_str()).is_some();
-    let direct_arguments = value.get("arguments").is_some();
-    let function_name = value
-        .get("function")
-        .and_then(|function| function.get("name"))
-        .and_then(|name| name.as_str())
-        .is_some();
-
-    (direct_name && direct_arguments) || function_name
-}
-
-fn json_candidates(text: &str) -> Vec<Value> {
-    let mut candidates = Vec::new();
-    let trimmed = text.trim();
-    if let Ok(value) = serde_json::from_str::<Value>(trimmed) {
-        candidates.push(value);
-    }
-
-    for (open, close) in [('{', '}'), ('[', ']')] {
-        let starts = text.match_indices(open).map(|(idx, _)| idx);
-        for start in starts {
-            let mut depth = 0i32;
-            let mut in_string = false;
-            let mut escaped = false;
-            for (offset, ch) in text[start..].char_indices() {
-                if escaped {
-                    escaped = false;
-                    continue;
-                }
-                if ch == '\\' && in_string {
-                    escaped = true;
-                    continue;
-                }
-                if ch == '"' {
-                    in_string = !in_string;
-                    continue;
-                }
-                if in_string {
-                    continue;
-                }
-                if ch == open {
-                    depth += 1;
-                } else if ch == close {
-                    depth -= 1;
-                    if depth == 0 {
-                        let end = start + offset + ch.len_utf8();
-                        if let Ok(value) = serde_json::from_str::<Value>(&text[start..end]) {
-                            candidates.push(value);
-                        }
-                        break;
-                    }
-                }
-            }
-        }
-    }
-
-    candidates
-}
-
-fn append_text(content: &mut Vec<MessageContent>, value: Option<&Value>) {
-    if let Some(text) = value.and_then(|value| value.as_str()) {
-        if !text.is_empty() {
-            content.push(MessageContent::text(text));
-        }
-    }
-}
-
-fn append_tool_calls(content: &mut Vec<MessageContent>, value: Option<&Value>) {
-    let Some(tool_calls) = value.and_then(|value| value.as_array()) else {
-        return;
-    };
-
-    for tool_call in tool_calls {
-        content.push(tool_call_content(tool_call));
-    }
-}
-
-fn tool_call_content(tool_call: &Value) -> MessageContent {
-    let id = tool_call
-        .get("id")
-        .and_then(|id| id.as_str())
-        .map(ToString::to_string)
-        .unwrap_or_else(|| Uuid::new_v4().to_string());
-
-    let function = tool_call.get("function").unwrap_or(tool_call);
-    let name = function
-        .get("name")
-        .and_then(|name| name.as_str())
-        .unwrap_or_default()
-        .to_string();
-
-    if !is_valid_function_name(&name) {
-        return MessageContent::tool_request(
-            id,
-            Err(ErrorData {
-                code: ErrorCode::INVALID_REQUEST,
-                message: Cow::from(format!(
-                    "The provided function name '{}' had invalid characters, it must match this regex [a-zA-Z0-9_-]+",
-                    name
-                )),
-                data: None,
-            }),
-        );
-    }
-
-    let arguments = function
-        .get("arguments")
-        .or_else(|| tool_call.get("arguments"))
-        .cloned()
-        .unwrap_or_else(|| json!({}));
-
-    let raw_arguments = arguments.to_string();
-    let parsed = match arguments {
-        Value::String(arguments) if arguments.trim().is_empty() => Ok(json!({})),
-        Value::String(arguments) => safely_parse_json(&arguments),
-        Value::Object(_) => Ok(arguments),
-        Value::Null => Ok(json!({})),
-        other => Ok(other),
-    };
-
-    match parsed {
-        Ok(params) => MessageContent::tool_request(
-            id,
-            Ok(CallToolRequestParams::new(name).with_arguments(object(params))),
-        ),
-        Err(error) => {
-            let message = format!(
-                "Could not interpret tool use parameters for id {}: {}. Raw arguments: '{}'",
-                id, error, raw_arguments
-            );
-            MessageContent::tool_request(
-                id,
-                Err(ErrorData {
-                    code: ErrorCode::INVALID_PARAMS,
-                    message: Cow::from(message),
-                    data: None,
-                }),
-            )
-        }
-    }
-}
-
-fn parse_xml_tool_calls(content: &str) -> (Option<String>, Vec<MessageContent>) {
-    let function_re = regex::Regex::new(r"<function=([^>]+)>([\s\S]*?)</function>").unwrap();
-    let param_re = regex::Regex::new(r"<parameter=([^>]+)>([\s\S]*?)</parameter>").unwrap();
-
-    let prefix = content
-        .find("<function=")
-        .and_then(|idx| content.get(..idx))
-        .map(str::trim)
-        .filter(|text| !text.is_empty())
-        .map(ToString::to_string);
-
-    let mut tool_calls = Vec::new();
-    for func_cap in function_re.captures_iter(content) {
-        let function_name = func_cap[1].trim().to_string();
-        let function_body = &func_cap[2];
-        let mut arguments = serde_json::Map::new();
-        for param_cap in param_re.captures_iter(function_body) {
-            arguments.insert(
-                param_cap[1].trim().to_string(),
-                Value::String(param_cap[2].trim().to_string()),
-            );
-        }
-        tool_calls.push(tool_call_content(&json!({
-            "function": {
-                "name": function_name,
-                "arguments": arguments,
-            }
-        })));
-    }
-
-    (prefix, tool_calls)
-}
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    fn tool_count(message: &Message) -> usize {
-        message
-            .content
-            .iter()
-            .filter(|content| matches!(content, MessageContent::ToolRequest(_)))
-            .count()
-    }
-
-    #[test]
-    fn parses_openai_message_tool_calls() {
-        let text = r#"{"content":null,"tool_calls":[{"id":"call_1","type":"function","function":{"name":"developer__shell","arguments":"{\"command\":\"pwd\"}"}}]}"#;
-        let message = message_from_native_tool_text(text, "msg").unwrap().unwrap();
-        assert_eq!(tool_count(&message), 1);
-    }
-
-    #[test]
-    fn parses_top_level_tool_calls() {
-        let text = r#"{"tool_calls":[{"name":"developer__shell","arguments":{"command":"pwd"}}]}"#;
-        let message = message_from_native_tool_text(text, "msg").unwrap().unwrap();
-        assert_eq!(tool_count(&message), 1);
-    }
-
-    #[test]
-    fn parses_top_level_tool_call_array() {
-        let text = r#"[{"name":"developer__shell","arguments":{"command":"pwd"}}]"#;
-        let message = message_from_native_tool_text(text, "msg").unwrap().unwrap();
-        assert_eq!(tool_count(&message), 1);
-    }
-
-    #[test]
-    fn parses_top_level_tool_call_object_with_arguments() {
-        let text = r#"{"name":"developer__shell","arguments":{"command":"pwd"}}"#;
-        let message = message_from_native_tool_text(text, "msg").unwrap().unwrap();
-        assert_eq!(tool_count(&message), 1);
-    }
-
-    #[test]
-    fn ignores_plain_json_objects_with_name_fields() {
-        let text = r#"{"name":"Alice","age":30}"#;
-        assert!(message_from_native_tool_text(text, "msg")
-            .unwrap()
-            .is_none());
-    }
-
-    #[test]
-    fn ignores_plain_json_arrays() {
-        let text = r#"["a","b"]"#;
-        assert!(message_from_native_tool_text(text, "msg")
-            .unwrap()
-            .is_none());
-    }
-
-    #[test]
-    fn ignores_non_tool_call_arrays_in_tool_calls_field() {
-        let text = r#"{"tool_calls":["a","b"]}"#;
-        assert!(message_from_native_tool_text(text, "msg")
-            .unwrap()
-            .is_none());
-    }
-
-    #[test]
-    fn parses_xml_tool_calls() {
-        let text = r#"<function=developer__shell><parameter=command>pwd</parameter></function>"#;
-        let message = message_from_native_tool_text(text, "msg").unwrap().unwrap();
-        assert_eq!(tool_count(&message), 1);
-    }
-}

crates/goose-local-inference/src/paths.rs

@@ -1,88 +0,0 @@
-use etcetera::{choose_app_strategy, AppStrategy, AppStrategyArgs};
-use std::path::PathBuf;
-
-pub struct Paths;
-
-impl Paths {
-    fn get_dir(dir_type: DirType) -> PathBuf {
-        if let Ok(test_root) = std::env::var("GOOSE_PATH_ROOT") {
-            let base = PathBuf::from(test_root);
-            match dir_type {
-                DirType::Config => base.join("config"),
-                DirType::Data => base.join("data"),
-                DirType::State => base.join("state"),
-                DirType::Plugins => base.join(".agents").join("plugins"),
-                DirType::Agents => base.join(".agents").join("agents"),
-                DirType::AgentsHome => base.join(".agents"),
-            }
-        } else {
-            // NOTE: "Block" is kept here for backwards compatibility with existing
-            // user config/data directories (e.g. ~/Library/Application Support/Block/goose/).
-            // Changing this would orphan existing installations.
-            let strategy = choose_app_strategy(AppStrategyArgs {
-                top_level_domain: "Block".to_string(),
-                author: "Block".to_string(),
-                app_name: "goose".to_string(),
-            })
-            .expect("goose requires a home dir");
-
-            match dir_type {
-                DirType::Config => strategy.config_dir(),
-                DirType::Data => strategy.data_dir(),
-                DirType::State => strategy.state_dir().unwrap_or(strategy.data_dir()),
-                DirType::Plugins => strategy.home_dir().join(".agents").join("plugins"),
-                DirType::Agents => strategy.home_dir().join(".agents").join("agents"),
-                DirType::AgentsHome => strategy.home_dir().join(".agents"),
-            }
-        }
-    }
-
-    pub fn config_dir() -> PathBuf {
-        Self::get_dir(DirType::Config)
-    }
-
-    pub fn data_dir() -> PathBuf {
-        Self::get_dir(DirType::Data)
-    }
-
-    pub fn state_dir() -> PathBuf {
-        Self::get_dir(DirType::State)
-    }
-
-    pub fn plugins_dir() -> PathBuf {
-        Self::get_dir(DirType::Plugins)
-    }
-
-    pub fn agents_dir() -> PathBuf {
-        Self::get_dir(DirType::Agents)
-    }
-
-    pub fn agents_home_dir() -> PathBuf {
-        Self::get_dir(DirType::AgentsHome)
-    }
-
-    pub fn in_agents_home_dir(subpath: &str) -> PathBuf {
-        Self::agents_home_dir().join(subpath)
-    }
-
-    pub fn in_state_dir(subpath: &str) -> PathBuf {
-        Self::state_dir().join(subpath)
-    }
-
-    pub fn in_config_dir(subpath: &str) -> PathBuf {
-        Self::config_dir().join(subpath)
-    }
-
-    pub fn in_data_dir(subpath: &str) -> PathBuf {
-        Self::data_dir().join(subpath)
-    }
-}
-
-enum DirType {
-    Config,
-    Data,
-    State,
-    Plugins,
-    Agents,
-    AgentsHome,
-}