[ { "path": ".cargo/config.toml", "content": "[target.x86_64-apple-darwin]\nrustflags = [\"-C\", \"link-arg=-undefined\", \"-C\", \"link-arg=dynamic_lookup\"]\n\n[target.aarch64-apple-darwin]\nrustflags = [\"-C\", \"link-arg=-undefined\", \"-C\", \"link-arg=dynamic_lookup\"]\n\n[target.x86_64-unknown-linux-musl]\nrustflags = [\"-C\", \"target-feature=-crt-static\"]\n" }, { "path": ".editorconfig", "content": "; https://editorconfig.org/\n\nroot = true\n\n[*]\ninsert_final_newline = true\ncharset = utf-8\ntrim_trailing_whitespace = true\nindent_style = space\nindent_size = 2\ntab_width = 8\n\n[{Makefile,**/Makefile}]\nindent_style = tab\nindent_size = 8\n" }, { "path": ".gitattributes", "content": "* text=auto eol=lf\n**/*.lock linguist-generated=true\n*.avanterules linguist-language=jinja\nsyntax/jinja.vim linguist-vendored\n" }, { "path": ".github/ISSUE_TEMPLATE/bug_report.yml", "content": "name: 🐛 Bug Report\ndescription: Create a bug report to help us improve Avante\ntitle: 'bug: '\nlabels: ['bug']\nbody:\n - type: markdown\n id: issue-already-exists\n attributes:\n value: |\n Please search to see if an issue already exists for the bug you encountered.\n See [Searching Issues and Pull Requests](https://docs.github.com/en/search-github/searching-on-github/searching-issues-and-pull-requests) for how to use the GitHub search bar and filters.\n - type: textarea\n id: describe-the-bug\n validations:\n required: true\n attributes:\n label: Describe the bug\n description: Please provide a clear and concise description about the problem you ran into.\n placeholder: This happened when I ...\n - type: textarea\n id: to-reproduce\n validations:\n required: false\n attributes:\n label: To reproduce\n description: |\n Please provide a code sample or a code snippet to reproduce said problem. If you have code snippets, error messages, or a stack trace please also provide them here.\n\n **IMPORTANT**: make sure to use [code tags](https://docs.github.com/en/get-started/writing-on-github/working-with-advanced-formatting/creating-and-highlighting-code-blocks#syntax-highlighting) to correctly format your code. Screenshots are helpful but don't use them for code snippets as they don't allow others to copy-and-paste your code.\n\n placeholder: |\n Give a minimal config to reproduce the issue.\n - type: textarea\n id: expected-behavior\n validations:\n required: false\n attributes:\n label: Expected behavior\n description: 'A clear and concise description of what you would expect to happen.'\n - type: textarea\n id: how-to-install\n validations:\n required: true\n attributes:\n label: Installation method\n description: |\n Please share your installation method with us.\n value: |\n Use lazy.nvim:\n ```lua\n {\n \"yetone/avante.nvim\",\n event = \"VeryLazy\",\n lazy = false,\n version = false, -- set this if you want to always pull the latest change\n opts = {\n -- add any opts here\n },\n -- if you want to build from source then do `make BUILD_FROM_SOURCE=true`\n build = \"make\",\n -- build = \"powershell -ExecutionPolicy Bypass -File Build.ps1 -BuildFromSource false\" -- for windows\n dependencies = {\n \"nvim-lua/plenary.nvim\",\n \"MunifTanjim/nui.nvim\",\n },\n }\n ```\n - type: textarea\n id: environment-info\n attributes:\n label: Environment\n description: |\n Please share your environment with us, including your neovim version using `nvim -v` and `uname -a`.\n placeholder: |\n neovim version: ...\n distribution (if any): ...\n platform: ...\n validations:\n required: true\n - type: textarea\n attributes:\n label: Repro\n description: Minimal `init.lua` to reproduce this issue. Save as `repro.lua` and run with `nvim -u repro.lua`\n value: |\n vim.env.LAZY_STDPATH = \".repro\"\n load(vim.fn.system(\"curl -s https://raw.githubusercontent.com/folke/lazy.nvim/main/bootstrap.lua\"))()\n\n require(\"lazy.minit\").repro({\n spec = {\n -- add any other plugins here\n },\n })\n render: lua\n validations:\n required: false\n" }, { "path": ".github/ISSUE_TEMPLATE/config.yml", "content": "blank_issues_enabled: true\nversion: 2.1\n" }, { "path": ".github/ISSUE_TEMPLATE/feature_request.yml", "content": "name: 🚀 Feature Request\ndescription: Submit a proposal/request for new Avante feature.\ntitle: 'feature: '\nlabels: ['new-feature', 'enhancement']\nbody:\n - type: textarea\n id: feature-request\n validations:\n required: true\n attributes:\n label: Feature request\n description: |\n A clear and concise description of the feature request.\n placeholder: |\n I would like it if...\n - type: textarea\n id: motivation\n validations:\n required: false\n attributes:\n label: Motivation\n description: |\n Please outline the motivation for this feature request. Is your feature request related to a problem? e.g., I'm always frustrated when [...].\n If this is related to another issue, please link here too.\n If you have a current workaround, please also provide it here.\n placeholder: |\n This feature would solve ...\n - type: textarea\n id: other\n attributes:\n label: Other\n description: |\n Is there any way that you could help, e.g. by submitting a PR?\n placeholder: |\n I would love to contribute ...\n" }, { "path": ".github/workflows/close-stale-issues-and-prs.yaml", "content": "name: 'Close stale issues and PRs'\non:\n schedule:\n - cron: '30 1 * * *'\n\npermissions:\n contents: write # only for delete-branch option\n issues: write\n pull-requests: write\n\njobs:\n stale:\n runs-on: ubuntu-latest\n steps:\n - uses: actions/stale@v9\n with:\n stale-issue-message: 'This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.'\n stale-pr-message: 'This PR is stale because it has been open 14 days with no activity. Remove stale label or comment or this will be closed in 10 days.'\n close-issue-message: 'This issue was closed because it has been stalled for 5 days with no activity.'\n close-pr-message: 'This PR was closed because it has been stalled for 10 days with no activity.'\n days-before-issue-stale: 30\n days-before-pr-stale: 14\n days-before-issue-close: 5\n days-before-pr-close: 10\n" }, { "path": ".github/workflows/lua.yaml", "content": "name: Lua CI\n\non:\n push:\n branches:\n - main\n paths:\n - \"**/*.lua\"\n - .github/workflows/lua.yaml\n pull_request:\n branches:\n - main\n paths:\n - \"**/*.lua\"\n - .github/workflows/lua.yaml\n\njobs:\n # reference from: https://github.com/nvim-lua/plenary.nvim/blob/2d9b06177a975543726ce5c73fca176cedbffe9d/.github/workflows/default.yml#L6C3-L43C20\n run_tests:\n name: unit tests\n runs-on: ${{ matrix.os }}\n strategy:\n fail-fast: false\n matrix:\n include:\n - os: ubuntu-22.04\n rev: v0.10.0\n steps:\n - uses: actions/checkout@v6\n\n - id: todays-date\n run: echo \"date=$(date +%F)\" >> \"$GITHUB_OUTPUT\"\n\n - name: Restore cache for today's nightly.\n id: cache-neovim\n uses: actions/cache@v4\n with:\n path: _neovim\n key: ${{ runner.os }}-${{ matrix.rev }}-${{ steps.todays-date.outputs.date }}\n\n - name: Download neovim ${{ matrix.rev }}\n env:\n GH_TOKEN: ${{ github.token }}\n NEOVIM_VERSION: ${{ matrix.rev }}\n if: steps.cache-neovim.outputs.cache-hit != 'true'\n run: |\n mkdir -p _neovim\n gh release download \\\n --output - \\\n --pattern nvim-linux64.tar.gz \\\n --repo neovim/neovim \\\n \"$NEOVIM_VERSION\" | tar xvz --strip-components 1 --directory _neovim\n\n - name: Prepare\n run: |\n sudo apt-get update\n sudo apt-get install -y ripgrep\n sudo apt-get install -y silversearcher-ag\n echo \"${PWD}/_neovim/bin\" >> \"$GITHUB_PATH\"\n echo VIM=\"${PWD}/_neovim/share/nvim/runtime\" >> \"$GITHUB_ENV\"\n\n - name: Run tests\n run: |\n nvim --version\n make luatest\n\n typecheck:\n name: Typecheck\n runs-on: ubuntu-latest\n strategy:\n matrix:\n nvim_version: [ stable ]\n luals_version: [ 3.13.6 ]\n steps:\n - uses: actions/checkout@v6\n\n - uses: rhysd/action-setup-vim@v1\n with:\n neovim: true\n version: ${{ matrix.nvim_version }}\n\n - name: Typecheck\n env:\n VIMRUNTIME: /home/runner/nvim-${{ matrix.nvim_version }}/share/nvim/runtime\n run: |\n make lua-typecheck\n" }, { "path": ".github/workflows/pre-commit.yaml", "content": "name: pre-commit\n\non:\n pull_request:\n types: [labeled, opened, reopened, synchronize]\n push:\n branches: [main, test-me-*]\n\njobs:\n pre-commit:\n if: \"github.event.action != 'labeled' || github.event.label.name == 'pre-commit ci run'\"\n runs-on: ubuntu-latest\n steps:\n - uses: actions/checkout@v3\n - run: gh pr edit ${{ github.event.number }} --remove-label 'pre-commit ci run'\n if: github.event.action == 'labeled' && github.event.label.name == 'pre-commit ci run'\n env:\n GH_TOKEN: ${{ github.token }}\n - uses: actions/setup-python@v3\n with:\n python-version: '3.11'\n - name: Install uv\n uses: astral-sh/setup-uv@v5\n - run: |\n uv venv\n source .venv/bin/activate\n uv pip install -r py/rag-service/requirements.txt\n - uses: leafo/gh-actions-lua@v11\n - uses: leafo/gh-actions-luarocks@v5\n - run: luarocks install luacheck\n - name: Install stylua from crates.io\n uses: baptiste0928/cargo-install@v3\n with:\n crate: stylua\n args: --features lua54\n - uses: pre-commit/action@v3.0.1\n - uses: pre-commit-ci/lite-action@v1.1.0\n if: always()\n" }, { "path": ".github/workflows/release.yaml", "content": "name: Release\n\non:\n push:\n tags: [v\\d+\\.\\d+\\.\\d+]\n\npermissions:\n contents: write\n packages: write\n\nenv:\n CARGO_TERM_COLOR: always\n\njobs:\n create-release:\n permissions:\n contents: write\n runs-on: ubuntu-24.04\n outputs:\n release_id: ${{ steps.create-release.outputs.id }}\n release_upload_url: ${{ steps.create-release.outputs.upload_url }}\n release_body: \"${{ steps.tag.outputs.message }}\"\n\n steps:\n - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # ratchet:actions/checkout@v4\n\n - name: Get version\n id: get_version\n uses: battila7/get-version-action@d97fbc34ceb64d1f5d95f4dfd6dce33521ccccf5 # ratchet:battila7/get-version-action@v2\n\n - name: Get tag message\n id: tag\n run: |\n git fetch --depth=1 origin +refs/tags/*:refs/tags/*\n echo \"message<> $GITHUB_OUTPUT\n echo \"$(git tag -l --format='%(contents)' ${{ steps.get_version.outputs.version }})\" >> $GITHUB_OUTPUT\n echo \"EOF\" >> $GITHUB_OUTPUT\n\n - name: Create Release\n id: create-release\n uses: ncipollo/release-action@2c591bcc8ecdcd2db72b97d6147f871fcd833ba5 # ratchet:ncipollo/release-action@v1\n with:\n draft: true\n name: \"avante-libs ${{ steps.get_version.outputs.version }}\"\n tag: ${{ steps.get_version.outputs.version }}\n body: \"${{ steps.tag.outputs.message }}\"\n\n releases-matrix:\n needs: [create-release]\n strategy:\n fail-fast: false\n matrix:\n feature: [lua51, luajit]\n config:\n - os: ubuntu-24.04-arm\n os_name: linux\n arch: aarch64\n rust_target: aarch64-unknown-linux-gnu\n docker_platform: linux/aarch64\n container: quay.io/pypa/manylinux2014_aarch64\n - os: ubuntu-latest\n os_name: linux\n arch: x86_64\n rust_target: x86_64-unknown-linux-gnu\n docker_platform: linux/amd64\n container: quay.io/pypa/manylinux2014_x86_64 # for glibc 2.17\n - os: macos-13\n os_name: darwin\n arch: x86_64\n rust_target: x86_64-apple-darwin\n - os: macos-latest\n os_name: darwin\n arch: aarch64\n rust_target: aarch64-apple-darwin\n - os: windows-latest\n os_name: windows\n arch: x86_64\n rust_target: x86_64-pc-windows-msvc\n - os: windows-latest\n os_name: windows\n arch: aarch64\n rust_target: aarch64-pc-windows-msvc\n\n runs-on: ${{ matrix.config.os }}\n\n steps:\n - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # ratchet:actions/checkout@v4\n - uses: Swatinem/rust-cache@82a92a6e8fbeee089604da2575dc567ae9ddeaab # ratchet:Swatinem/rust-cache@v2\n if: ${{ matrix.config.container == null }}\n - uses: dtolnay/rust-toolchain@7b1c307e0dcbda6122208f10795a713336a9b35a # ratchet:dtolnay/rust-toolchain@master\n if: ${{ matrix.config.container == null }}\n with:\n targets: ${{ matrix.config.rust_target }}\n toolchain: \"1.85.0\"\n - name: Build all crates\n if: ${{ matrix.config.container == null }}\n run: |\n cargo build --release --features ${{ matrix.feature }}\n\n - name: Build all crates with glibc 2.17 # for glibc 2.17\n if: ${{ matrix.config.container != null }}\n run: |\n # sudo apt-get install -y qemu qemu-user-static\n docker run \\\n --rm \\\n -v $(pwd):/workspace \\\n -w /workspace \\\n --platform ${{ matrix.config.docker_platform }} \\\n ${{ matrix.config.container }} \\\n bash -c \"yum install -y perl-IPC-Cmd openssl-devel && curl --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --profile minimal && . /root/.cargo/env && cargo build --release --features ${{ matrix.feature }}\"\n\n - name: Handle binaries\n if: ${{ matrix.config.os_name != 'windows' }}\n shell: bash\n run: |\n mkdir -p results\n if [ \"${{ matrix.config.os_name }}\" == \"linux\" ]; then\n EXT=\"so\"\n else\n EXT=\"dylib\"\n fi\n cp target/release/libavante_templates.$EXT results/avante_templates.$EXT\n cp target/release/libavante_tokenizers.$EXT results/avante_tokenizers.$EXT\n cp target/release/libavante_repo_map.$EXT results/avante_repo_map.$EXT\n cp target/release/libavante_html2md.$EXT results/avante_html2md.$EXT\n\n cd results\n tar zcvf avante_lib-${{ matrix.config.os_name }}-${{ matrix.config.arch }}-${{ matrix.feature }}.tar.gz *.${EXT}\n\n - name: Handle binaries (Windows)\n if: ${{ matrix.config.os_name == 'windows' }}\n shell: pwsh\n run: |\n New-Item -ItemType Directory -Force -Path results\n\n Copy-Item -Path \"target\\release\\avante_templates.dll\" -Destination \"results\\avante_templates.dll\"\n Copy-Item -Path \"target\\release\\avante_tokenizers.dll\" -Destination \"results\\avante_tokenizers.dll\"\n Copy-Item -Path \"target\\release\\avante_repo_map.dll\" -Destination \"results\\avante_repo_map.dll\"\n Copy-Item -Path \"target\\release\\avante_html2md.dll\" -Destination \"results\\avante_html2md.dll\"\n\n Set-Location -Path results\n\n $dllFiles = Get-ChildItem -Filter \"*.dll\" | Select-Object -ExpandProperty Name\n Compress-Archive -Path $dllFiles -DestinationPath \"avante_lib-${{ matrix.config.os_name }}-${{ matrix.config.arch }}-${{ matrix.feature }}.zip\"\n\n - name: Upload Release Asset\n uses: shogo82148/actions-upload-release-asset@8482bd769644976d847e96fb4b9354228885e7b4 # ratchet:shogo82148/actions-upload-release-asset@v1\n if: ${{ matrix.config.os_name != 'windows' }}\n env:\n GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n ASSET_NAME: avante_lib-${{ matrix.config.os_name }}-${{ matrix.config.arch }}-${{ matrix.feature }}.tar.gz\n with:\n upload_url: ${{ needs.create-release.outputs.release_upload_url }}\n asset_path: ./results/avante_lib-${{ matrix.config.os_name }}-${{ matrix.config.arch }}-${{ matrix.feature }}.tar.gz\n - name: Upload Release Asset (Windows)\n uses: shogo82148/actions-upload-release-asset@8482bd769644976d847e96fb4b9354228885e7b4 # ratchet:shogo82148/actions-upload-release-asset@v1\n if: ${{ matrix.config.os_name == 'windows' }}\n env:\n GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n ASSET_NAME: avante_lib-${{ matrix.config.os_name }}-${{ matrix.config.arch }}-${{ matrix.feature }}.zip\n with:\n upload_url: ${{ needs.create-release.outputs.release_upload_url }}\n asset_path: ./results/avante_lib-${{ matrix.config.os_name }}-${{ matrix.config.arch }}-${{ matrix.feature }}.zip\n\n publish-release:\n permissions:\n contents: write\n runs-on: ubuntu-24.04\n needs: [create-release, releases-matrix]\n\n steps:\n - name: publish release\n id: publish-release\n uses: actions/github-script@d7906e4ad0b1822421a7e6a35d5ca353c962f410 # ratchet:actions/github-script@v6\n env:\n release_id: ${{ needs.create-release.outputs.release_id }}\n with:\n script: |\n github.rest.repos.updateRelease({\n owner: context.repo.owner,\n repo: context.repo.repo,\n release_id: process.env.release_id,\n draft: false,\n prerelease: false\n })\n" }, { "path": ".github/workflows/rust.yaml", "content": "name: Rust CI\n\non:\n push:\n branches:\n - main\n paths:\n - \"crates/**/*\"\n - \"Cargo.lock\"\n - \"Cargo.toml\"\n pull_request:\n branches:\n - main\n paths:\n - \"crates/**/*\"\n - \"Cargo.lock\"\n - \"Cargo.toml\"\n\njobs:\n tests:\n name: Run Rust tests\n runs-on: ubuntu-latest\n steps:\n - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # ratchet:actions/checkout@v4\n - uses: Swatinem/rust-cache@82a92a6e8fbeee089604da2575dc567ae9ddeaab # ratchet:Swatinem/rust-cache@v2\n - uses: dtolnay/rust-toolchain@7b1c307e0dcbda6122208f10795a713336a9b35a # ratchet:dtolnay/rust-toolchain@master\n with:\n toolchain: stable\n components: clippy, rustfmt\n - name: Run rust tests\n run: cargo test --features luajit\n" }, { "path": ".gitignore", "content": "*.so\n# Lua compiled files\n*.lua~\n*.luac\n\n.venv\n__pycache__/\n\n# Neovim plugin specific files\nplugin/packer_compiled.lua\n\n# OS generated files\n.DS_Store\nThumbs.db\n\n# Editor/IDE generated files\n*.swp\n*.swo\n*~\n.vscode/\n.idea/\n\n# Dependency manager generated directories\n/lua_modules/\n/.luarocks/\n\n# Log files\n*.log\n\n# Temporary files\ntmp/\ntemp/\n\n# Environment variable files (if you use .env file to store API keys)\n.env\n.envrc\n\n# If you use any build tools, you might need to ignore build output directories\nbuild/\ndist/\n\n# If you use any test frameworks, you might need to ignore test coverage reports\ncoverage/\n\n# If you use documentation generation tools, you might need to ignore generated docs\ndoc/\n\n# If you have any personal configuration files, you should ignore them too\nconfig.personal.lua\n\ntarget\n" }, { "path": ".luacheckrc", "content": "-- Rerun tests only if their modification time changed.\ncache = true\n\n-- Glorious list of warnings: https://luacheck.readthedocs.io/en/stable/warnings.html\nignore = {\n '211',\n '631',\n '212', -- Unused argument, In the case of callback function, _arg_name is easier to understand than _, so this option is set to off.\n '411', -- Redefining a local variable.\n '412', -- Redefining an argument.\n '422', -- Shadowing an argument\n '431', -- Shadowing a variable\n '122', -- Indirectly setting a readonly global\n}\n\n-- Global objects defined by the C code\nread_globals = {\n 'vim',\n 'Snacks',\n}\n" }, { "path": ".pre-commit-config.yaml", "content": "repos:\n- repo: https://github.com/pre-commit/pre-commit-hooks\n rev: v5.0.0\n hooks:\n - id: check-yaml\n - id: end-of-file-fixer\n - id: trailing-whitespace\n - id: check-ast # 检查Python语法错误\n - id: debug-statements # 检查是否有debug语句\n - id: check-added-large-files\n - id: check-merge-conflict\n- repo: https://github.com/JohnnyMorganz/StyLua\n rev: v2.0.2\n hooks:\n - id: stylua-system # or stylua-system / stylua-github\n files: \\.lua$\n- repo: https://github.com/Calinou/pre-commit-luacheck\n rev: v1.0.0\n hooks:\n - id: luacheck\n- repo: https://github.com/doublify/pre-commit-rust\n rev: v1.0\n hooks:\n - id: fmt\n files: \\.rs$\n - id: cargo-check\n args: ['--features', 'luajit']\n files: \\.rs$\n- repo: https://github.com/astral-sh/ruff-pre-commit\n rev: v0.9.9\n hooks:\n # 运行 Ruff linter\n - id: ruff\n args: [--fix]\n # 运行 Ruff formatter\n - id: ruff-format\n- repo: https://github.com/RobertCraigie/pyright-python\n rev: v1.1.395\n hooks:\n - id: pyright\n additional_dependencies:\n - \"types-setuptools\"\n - \"types-requests\"\n" }, { "path": "Build.ps1", "content": "param (\n [string]$Version = \"luajit\",\n [string]$BuildFromSource = \"false\"\n)\n\n$Build = [System.Convert]::ToBoolean($BuildFromSource)\n\n$ErrorActionPreference = \"Stop\"\n$ProgressPreference = \"SilentlyContinue\"\n\n$BuildDir = \"build\"\n$REPO_OWNER = \"yetone\"\n$REPO_NAME = \"avante.nvim\"\n\nfunction Build-FromSource($feature) {\n if (-not (Test-Path $BuildDir)) {\n New-Item -ItemType Directory -Path $BuildDir | Out-Null\n }\n\n cargo build --release --features=$feature\n\n $SCRIPT_DIR = $PSScriptRoot\n $targetTokenizerFile = \"avante_tokenizers.dll\"\n $targetTemplatesFile = \"avante_templates.dll\"\n $targetRepoMapFile = \"avante_repo_map.dll\"\n Copy-Item (Join-Path $SCRIPT_DIR \"target\\release\\avante_tokenizers.dll\") (Join-Path $BuildDir $targetTokenizerFile)\n Copy-Item (Join-Path $SCRIPT_DIR \"target\\release\\avante_templates.dll\") (Join-Path $BuildDir $targetTemplatesFile)\n Copy-Item (Join-Path $SCRIPT_DIR \"target\\release\\avante_repo_map.dll\") (Join-Path $BuildDir $targetRepoMapFile)\n\n Remove-Item -Recurse -Force \"target\"\n}\n\nfunction Test-Command($cmdname) {\n return $null -ne (Get-Command $cmdname -ErrorAction SilentlyContinue)\n}\n\nfunction Test-GHAuth {\n try {\n $null = gh api user\n return $true\n } catch {\n return $false\n }\n}\n\nfunction Download-Prebuilt($feature, $tag) {\n\n $SCRIPT_DIR = $PSScriptRoot\n # Set the target directory to clone the artifact\n $TARGET_DIR = Join-Path $SCRIPT_DIR \"build\"\n\n # Set the platform to Windows\n $PLATFORM = \"windows\"\n $ARCH = \"x86_64\"\n if ($env:PROCESSOR_ARCHITECTURE -eq \"ARM64\") {\n $ARCH = \"aarch64\"\n }\n\n # Set the Lua version (lua51 or luajit)\n $LUA_VERSION = if ($feature) { $feature } else { \"luajit\" }\n\n # Set the artifact name pattern\n $ARTIFACT_NAME_PATTERN = \"avante_lib-$PLATFORM-$ARCH-$LUA_VERSION\"\n\n $TempFile = Get-Item ([System.IO.Path]::GetTempFilename()) | Rename-Item -NewName { $_.Name + \".zip\" } -PassThru\n\n if ((Test-Command \"gh\") -and (Test-GHAuth)) {\n write-host \"Using GitHub CLI to download artifacts...\"\n gh release download $latestTag --repo \"$REPO_OWNER/$REPO_NAME\" --pattern \"*$ARTIFACT_NAME_PATTERN*\" --output $TempFile --clobber\n } else {\n # Get the artifact download URL\n $RELEASE = Invoke-RestMethod -Uri \"https://api.github.com/repos/$REPO_OWNER/$REPO_NAME/releases/tags/$tag\"\n $ARTIFACT_URL = $RELEASE.assets | Where-Object { $_.name -like \"*$ARTIFACT_NAME_PATTERN*\" } | Select-Object -ExpandProperty browser_download_url\n\n # Download and extract the artifact\n Invoke-WebRequest -Uri $ARTIFACT_URL -OutFile $TempFile\n }\n\n # Create target directory if it doesn't exist\n if (-not (Test-Path $TARGET_DIR)) {\n New-Item -ItemType Directory -Path $TARGET_DIR | Out-Null\n }\n Expand-Archive -Path $TempFile -DestinationPath $TARGET_DIR -Force\n Remove-Item $TempFile\n}\n\nfunction Main {\n Set-Location $PSScriptRoot\n if ($Build) {\n Write-Host \"Building for $Version...\"\n Build-FromSource $Version\n } else {\n $latestTag = git describe --tags --abbrev=0 2>&1 | Where-Object { $_ -ne $null }\n $builtTag = if (Test-Path \"build/.tag\") {\n Get-Content \"build/.tag\"\n } else {\n $null\n }\n\n function Save-Tag($tag) {\n $tag | Set-Content \"build/.tag\"\n }\n\n if ($latestTag -eq $builtTag -and $latestTag) {\n Write-Host \"Local build is up to date. No download needed.\"\n } elseif ($latestTag -ne $builtTag -and $latestTag) {\n Write-Host \"Downloading prebuilt binaries $latestTag for $Version...\"\n Download-Prebuilt $Version $latestTag\n Save-Tag $latestTag\n } else {\n cargo build --release --features=$Version\n Get-ChildItem -Path \"target/release/avante_*.dll\" | ForEach-Object {\n Copy-Item $_.FullName \"build/$($_.Name)\"\n }\n Save-Tag $latestTag\n }\n }\n Write-Host \"Completed!\"\n}\n\n# Run the main function\nMain\n" }, { "path": "Cargo.toml", "content": "[workspace]\nmembers = [\"crates/*\"]\nresolver = \"2\"\n\n[workspace.package]\nedition = \"2021\"\nrust-version = \"1.80\"\nlicense = \"Apache-2.0\"\nversion = \"0.1.0\"\n\n[workspace.dependencies]\navante-tokenizers = { path = \"crates/avante-tokenizers\" }\navante-templates = { path = \"crates/avante-templates\" }\navante-repo-map = { path = \"crates/avante-repo-map\" }\navante-html2md = { path = \"crates/avante-html2md\" }\nminijinja = { version = \"2.4.0\", features = [\n \"loader\",\n \"json\",\n \"fuel\",\n \"unicode\",\n \"speedups\",\n \"custom_syntax\",\n \"loop_controls\",\n] }\nmlua = { version = \"0.10.0\", features = [\"module\", \"serialize\"] }\ntiktoken-rs = { version = \"0.6.0\" }\ntokenizers = { version = \"0.20.0\", features = [\n \"esaxx_fast\",\n \"http\",\n \"unstable_wasm\",\n \"onig\",\n], default-features = false }\nserde = { version = \"1.0.209\", features = [\"derive\"] }\n\n[workspace.lints.rust]\nunsafe_code = \"warn\"\nunreachable_pub = \"warn\"\n\n[workspace.lints.clippy]\npedantic = { level = \"warn\", priority = -2 }\n# Allowed pedantic lints\nchar_lit_as_u8 = \"allow\"\ncollapsible_else_if = \"allow\"\ncollapsible_if = \"allow\"\nimplicit_hasher = \"allow\"\nmap_unwrap_or = \"allow\"\nmatch_same_arms = \"allow\"\nmissing_errors_doc = \"allow\"\nmissing_panics_doc = \"allow\"\nmodule_name_repetitions = \"allow\"\nmust_use_candidate = \"allow\"\nsimilar_names = \"allow\"\ntoo_many_lines = \"allow\"\ntoo_many_arguments = \"allow\"\n# Disallowed restriction lints\nprint_stdout = \"warn\"\nprint_stderr = \"warn\"\ndbg_macro = \"warn\"\nempty_drop = \"warn\"\nempty_structs_with_brackets = \"warn\"\nexit = \"warn\"\nget_unwrap = \"warn\"\nrc_buffer = \"warn\"\nrc_mutex = \"warn\"\nrest_pat_in_fully_bound_structs = \"warn\"\n" }, { "path": "LICENSE", "content": " Apache License\n Version 2.0, January 2004\n http://www.apache.org/licenses/\n\n TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION\n\n 1. Definitions.\n\n \"License\" shall mean the terms and conditions for use, reproduction,\n and distribution as defined by Sections 1 through 9 of this document.\n\n \"Licensor\" shall mean the copyright owner or entity authorized by\n the copyright owner that is granting the License.\n\n \"Legal Entity\" shall mean the union of the acting entity and all\n other entities that control, are controlled by, or are under common\n control with that entity. For the purposes of this definition,\n \"control\" means (i) the power, direct or indirect, to cause the\n direction or management of such entity, whether by contract or\n otherwise, or (ii) ownership of fifty percent (50%) or more of the\n outstanding shares, or (iii) beneficial ownership of such entity.\n\n \"You\" (or \"Your\") shall mean an individual or Legal Entity\n exercising permissions granted by this License.\n\n \"Source\" form shall mean the preferred form for making modifications,\n including but not limited to software source code, documentation\n source, and configuration files.\n\n \"Object\" form shall mean any form resulting from mechanical\n transformation or translation of a Source form, including but\n not limited to compiled object code, generated documentation,\n and conversions to other media types.\n\n \"Work\" shall mean the work of authorship, whether in Source or\n Object form, made available under the License, as indicated by a\n copyright notice that is included in or attached to the work\n (an example is provided in the Appendix below).\n\n \"Derivative Works\" shall mean any work, whether in Source or Object\n form, that is based on (or derived from) the Work and for which the\n editorial revisions, annotations, elaborations, or other modifications\n represent, as a whole, an original work of authorship. For the purposes\n of this License, Derivative Works shall not include works that remain\n separable from, or merely link (or bind by name) to the interfaces of,\n the Work and Derivative Works thereof.\n\n \"Contribution\" shall mean any work of authorship, including\n the original version of the Work and any modifications or additions\n to that Work or Derivative Works thereof, that is intentionally\n submitted to Licensor for inclusion in the Work by the copyright owner\n or by an individual or Legal Entity authorized to submit on behalf of\n the copyright owner. For the purposes of this definition, \"submitted\"\n means any form of electronic, verbal, or written communication sent\n to the Licensor or its representatives, including but not limited to\n communication on electronic mailing lists, source code control systems,\n and issue tracking systems that are managed by, or on behalf of, the\n Licensor for the purpose of discussing and improving the Work, but\n excluding communication that is conspicuously marked or otherwise\n designated in writing by the copyright owner as \"Not a Contribution.\"\n\n \"Contributor\" shall mean Licensor and any individual or Legal Entity\n on behalf of whom a Contribution has been received by Licensor and\n subsequently incorporated within the Work.\n\n 2. Grant of Copyright License. Subject to the terms and conditions of\n this License, each Contributor hereby grants to You a perpetual,\n worldwide, non-exclusive, no-charge, royalty-free, irrevocable\n copyright license to reproduce, prepare Derivative Works of,\n publicly display, publicly perform, sublicense, and distribute the\n Work and such Derivative Works in Source or Object form.\n\n 3. Grant of Patent License. Subject to the terms and conditions of\n this License, each Contributor hereby grants to You a perpetual,\n worldwide, non-exclusive, no-charge, royalty-free, irrevocable\n (except as stated in this section) patent license to make, have made,\n use, offer to sell, sell, import, and otherwise transfer the Work,\n where such license applies only to those patent claims licensable\n by such Contributor that are necessarily infringed by their\n Contribution(s) alone or by combination of their Contribution(s)\n with the Work to which such Contribution(s) was submitted. If You\n institute patent litigation against any entity (including a\n cross-claim or counterclaim in a lawsuit) alleging that the Work\n or a Contribution incorporated within the Work constitutes direct\n or contributory patent infringement, then any patent licenses\n granted to You under this License for that Work shall terminate\n as of the date such litigation is filed.\n\n 4. Redistribution. You may reproduce and distribute copies of the\n Work or Derivative Works thereof in any medium, with or without\n modifications, and in Source or Object form, provided that You\n meet the following conditions:\n\n (a) You must give any other recipients of the Work or\n Derivative Works a copy of this License; and\n\n (b) You must cause any modified files to carry prominent notices\n stating that You changed the files; and\n\n (c) You must retain, in the Source form of any Derivative Works\n that You distribute, all copyright, patent, trademark, and\n attribution notices from the Source form of the Work,\n excluding those notices that do not pertain to any part of\n the Derivative Works; and\n\n (d) If the Work includes a \"NOTICE\" text file as part of its\n distribution, then any Derivative Works that You distribute must\n include a readable copy of the attribution notices contained\n within such NOTICE file, excluding those notices that do not\n pertain to any part of the Derivative Works, in at least one\n of the following places: within a NOTICE text file distributed\n as part of the Derivative Works; within the Source form or\n documentation, if provided along with the Derivative Works; or,\n within a display generated by the Derivative Works, if and\n wherever such third-party notices normally appear. The contents\n of the NOTICE file are for informational purposes only and\n do not modify the License. You may add Your own attribution\n notices within Derivative Works that You distribute, alongside\n or as an addendum to the NOTICE text from the Work, provided\n that such additional attribution notices cannot be construed\n as modifying the License.\n\n You may add Your own copyright statement to Your modifications and\n may provide additional or different license terms and conditions\n for use, reproduction, or distribution of Your modifications, or\n for any such Derivative Works as a whole, provided Your use,\n reproduction, and distribution of the Work otherwise complies with\n the conditions stated in this License.\n\n 5. Submission of Contributions. Unless You explicitly state otherwise,\n any Contribution intentionally submitted for inclusion in the Work\n by You to the Licensor shall be under the terms and conditions of\n this License, without any additional terms or conditions.\n Notwithstanding the above, nothing herein shall supersede or modify\n the terms of any separate license agreement you may have executed\n with Licensor regarding such Contributions.\n\n 6. Trademarks. This License does not grant permission to use the trade\n names, trademarks, service marks, or product names of the Licensor,\n except as required for reasonable and customary use in describing the\n origin of the Work and reproducing the content of the NOTICE file.\n\n 7. Disclaimer of Warranty. Unless required by applicable law or\n agreed to in writing, Licensor provides the Work (and each\n Contributor provides its Contributions) on an \"AS IS\" BASIS,\n WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or\n implied, including, without limitation, any warranties or conditions\n of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A\n PARTICULAR PURPOSE. You are solely responsible for determining the\n appropriateness of using or redistributing the Work and assume any\n risks associated with Your exercise of permissions under this License.\n\n 8. Limitation of Liability. In no event and under no legal theory,\n whether in tort (including negligence), contract, or otherwise,\n unless required by applicable law (such as deliberate and grossly\n negligent acts) or agreed to in writing, shall any Contributor be\n liable to You for damages, including any direct, indirect, special,\n incidental, or consequential damages of any character arising as a\n result of this License or out of the use or inability to use the\n Work (including but not limited to damages for loss of goodwill,\n work stoppage, computer failure or malfunction, or any and all\n other commercial damages or losses), even if such Contributor\n has been advised of the possibility of such damages.\n\n 9. Accepting Warranty or Additional Liability. While redistributing\n the Work or Derivative Works thereof, You may choose to offer,\n and charge a fee for, acceptance of support, warranty, indemnity,\n or other liability obligations and/or rights consistent with this\n License. However, in accepting such obligations, You may act only\n on Your own behalf and on Your sole responsibility, not on behalf\n of any other Contributor, and only if You agree to indemnify,\n defend, and hold each Contributor harmless for any liability\n incurred by, or claims asserted against, such Contributor by reason\n of your accepting any such warranty or additional liability.\n\n END OF TERMS AND CONDITIONS\n\n APPENDIX: How to apply the Apache License to your work.\n\n To apply the Apache License to your work, attach the following\n boilerplate notice, with the fields enclosed by brackets \"[]\"\n replaced with your own identifying information. (Don't include\n the brackets!) The text should be enclosed in the appropriate\n comment syntax for the file format. We also recommend that a\n file or class name and description of purpose be included on the\n same \"printed page\" as the copyright notice for easier\n identification within third-party archives.\n\n Copyright [yyyy] [name of copyright owner]\n\n Licensed under the Apache License, Version 2.0 (the \"License\");\n you may not use this file except in compliance with the License.\n You may obtain a copy of the License at\n\n http://www.apache.org/licenses/LICENSE-2.0\n\n Unless required by applicable law or agreed to in writing, software\n distributed under the License is distributed on an \"AS IS\" BASIS,\n WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n See the License for the specific language governing permissions and\n limitations under the License.\n" }, { "path": "Makefile", "content": "UNAME := $(shell uname)\nARCH := $(shell uname -m)\n\nifeq ($(UNAME), Linux)\n\tOS := linux\n\tEXT := so\nelse ifeq ($(UNAME), Darwin)\n\tOS := macOS\n\tEXT := dylib\nelse\n\t$(error Unsupported operating system: $(UNAME))\nendif\n\nLUA_VERSIONS := luajit lua51\n\nBUILD_DIR := build\nBUILD_FROM_SOURCE ?= false\nTARGET_LIBRARY ?= all\n\nRAG_SERVICE_VERSION ?= 0.0.11\nRAG_SERVICE_IMAGE := quay.io/yetoneful/avante-rag-service:$(RAG_SERVICE_VERSION)\n\nall: luajit\n\ndefine make_definitions\nifeq ($(BUILD_FROM_SOURCE),true)\nifeq ($(TARGET_LIBRARY), all)\n$1: $(BUILD_DIR)/libAvanteTokenizers-$1.$(EXT) $(BUILD_DIR)/libAvanteTemplates-$1.$(EXT) $(BUILD_DIR)/libAvanteRepoMap-$1.$(EXT) $(BUILD_DIR)/libAvanteHtml2md-$1.$(EXT)\nelse ifeq ($(TARGET_LIBRARY), tokenizers)\n$1: $(BUILD_DIR)/libAvanteTokenizers-$1.$(EXT)\nelse ifeq ($(TARGET_LIBRARY), templates)\n$1: $(BUILD_DIR)/libAvanteTemplates-$1.$(EXT)\nelse ifeq ($(TARGET_LIBRARY), repo-map)\n$1: $(BUILD_DIR)/libAvanteRepoMap-$1.$(EXT)\nelse ifeq ($(TARGET_LIBRARY), html2md)\n$1: $(BUILD_DIR)/libAvanteHtml2md-$1.$(EXT)\nelse\n\t$$(error TARGET_LIBRARY must be one of all, tokenizers, templates, repo-map, html2md)\nendif\nelse\n$1:\n\tLUA_VERSION=$1 bash ./build.sh\nendif\nendef\n\n$(foreach lua_version,$(LUA_VERSIONS),$(eval $(call make_definitions,$(lua_version))))\n\ndefine build_package\n$1-$2:\n\tcargo build --release --features=$1 -p avante-$2\n\tcp target/release/libavante_$(shell echo $2 | tr - _).$(EXT) $(BUILD_DIR)/avante_$(shell echo $2 | tr - _).$(EXT)\nendef\n\ndefine build_targets\n$(BUILD_DIR)/libAvanteTokenizers-$1.$(EXT): $(BUILD_DIR) $1-tokenizers\n$(BUILD_DIR)/libAvanteTemplates-$1.$(EXT): $(BUILD_DIR) $1-templates\n$(BUILD_DIR)/libAvanteRepoMap-$1.$(EXT): $(BUILD_DIR) $1-repo-map\n$(BUILD_DIR)/libAvanteHtml2md-$1.$(EXT): $(BUILD_DIR) $1-html2md\nendef\n\n$(foreach lua_version,$(LUA_VERSIONS),$(eval $(call build_package,$(lua_version),tokenizers)))\n$(foreach lua_version,$(LUA_VERSIONS),$(eval $(call build_package,$(lua_version),templates)))\n$(foreach lua_version,$(LUA_VERSIONS),$(eval $(call build_package,$(lua_version),repo-map)))\n$(foreach lua_version,$(LUA_VERSIONS),$(eval $(call build_package,$(lua_version),html2md)))\n$(foreach lua_version,$(LUA_VERSIONS),$(eval $(call build_targets,$(lua_version))))\n\n$(BUILD_DIR):\n\t@mkdir -p $(BUILD_DIR)\n\nclean:\n\t@rm -rf $(BUILD_DIR)\n\nluacheck:\n\t@luacheck `find \\( -path './target' -prune \\) -o -name \"*.lua\" -print` --codes\n\nluastylecheck:\n\t@stylua --check lua/ plugin/ tests/\n\nstylefix:\n\t@stylua lua/ plugin/\n\n.PHONY: ruststylecheck\nruststylecheck:\n\t@rustup component add rustfmt 2> /dev/null\n\t@cargo fmt --all -- --check\n\n.PHONY: rustlint\nrustlint:\n\t@rustup component add clippy 2> /dev/null\n\t@cargo clippy -F luajit --all -- -F clippy::dbg-macro -D warnings\n\n.PHONY: rusttest\nrusttest:\n\t@cargo test --features luajit\n\n.PHONY: luatest\nluatest:\n\t@./scripts/run-luatest.sh\n\n.PHONY: lint\nlint: luacheck luastylecheck ruststylecheck rustlint\n\n.PHONY: lua-typecheck\nlua-typecheck:\n\t@./scripts/lua-typecheck.sh\n\n.PHONY: build-image\nbuild-image:\n\tdocker build --platform=linux/amd64 -t $(RAG_SERVICE_IMAGE) -f py/rag-service/Dockerfile py/rag-service\n\n.PHONY: push-image\npush-image: build-image\n\tdocker push $(RAG_SERVICE_IMAGE)\n" }, { "path": "README.md", "content": "
\n \"logo\"\n

avante.nvim

\n
\n\n

\n \"Neovim:\n \"Lua\n \"Rust\n \"pre-commit\n \"Discord\"\n \n

\n\n**avante.nvim** is a Neovim plugin designed to emulate the behaviour of the [Cursor](https://www.cursor.com) AI IDE. It provides users with AI-driven code suggestions and the ability to apply these recommendations directly to their source files with minimal effort.\n\n[查看中文版](README_zh.md)\n\n> [!NOTE]\n>\n> 🥰 This project is undergoing rapid iterations, and many exciting features will be added successively. Stay tuned!\n\n\n\n\n\n## Sponsorship ❤️\n\nIf you like this project, please consider supporting me on Patreon, as it helps me to continue maintaining and improving it:\n\n[Sponsor me](https://patreon.com/yetone)\n\n## Features\n\n- **AI-Powered Code Assistance**: Interact with AI to ask questions about your current code file and receive intelligent suggestions for improvement or modification.\n- **One-Click Application**: Quickly apply the AI's suggested changes to your source code with a single command, streamlining the editing process and saving time.\n- **Project-Specific Instruction Files**: Customize AI behavior by adding a markdown file (`avante.md` by default) in the project root. This file is automatically referenced during workspace changes. You can also configure a custom file name for tailored project instructions.\n\n## Avante Zen Mode\n\nDue to the prevalence of claude code, it is clear that this is an era of Coding Agent CLIs. As a result, there are many arguments like: in the Vibe Coding era, editors are no longer needed; you only need to use the CLI in the terminal. But have people realized that for more than half a century, Terminal-based Editors have solved and standardized the biggest problem with Terminal-based applications — that is, the awkward TUI interactions! No matter how much these Coding Agent CLIs optimize their UI/UX, their UI/UX will always be a subset of Terminal-based Editors (Vim, Emacs)! They cannot achieve Vim’s elegant action + text objects abstraction (imagine how you usually edit large multi-line prompts in an Agent CLI), nor can they leverage thousands of mature Vim/Neovim plugins to help optimize TUI UI/UX—such as easymotions and so on. Moreover, when they want to view or modify code, they often have to jump into other applications which forcibly interrupts the UI/UX experience.\n\nTherefore, Avante’s Zen Mode was born! It looks like a Vibe Coding Agent CLI but it is completely Neovim underneath. So you can use your muscle-memory Vim operations and those rich and mature Neovim plugins on it. At the same time, by leveraging [ACP](https://github.com/yetone/avante.nvim#acp-support) it has all capabilities of claude code / gemini-cli / codex! Why not enjoy both?\n\nNow all you need to do is alias this command to avante; then every time you simply type avante just like using claude code and enter Avante’s Zen Mode!\n\n```bash\nalias avante='nvim -c \"lua vim.defer_fn(function()require(\\\"avante.api\\\").zen_mode()end, 100)\"'\n```\n\nThe effect is as follows:\n\n\"Avante\n\n## Project instructions with avante.md\n\n
\n\n\nThe `avante.md` file allows you to provide project-specific context and instructions to the ai. this file should be placed in your project root and will be automatically referenced during all interactions with avante.\n\n\n### Best practices for avante.md\n\nto get the most out of your project instruction file, consider following this structure:\n\n#### Your role\n\ndefine the ai's persona and expertise level for your project:\n\n```markdown\n### your role\n\nyou are an expert senior software engineer specializing in [technology stack]. you have deep knowledge of [specific frameworks/tools] and understand best practices for [domain/industry]. you write clean, maintainable, and well-documented code. you prioritize code quality, performance, and security in all your recommendations.\n```\n\n#### Your mission\n\nclearly describe what the ai should focus on and how it should help:\n\n```markdown\n### your mission\n\nyour primary goal is to help build and maintain [project description]. you should:\n\n- provide code suggestions that follow our established patterns and conventions\n- help debug issues by analyzing code and suggesting solutions\n- assist with refactoring to improve code quality and maintainability\n- suggest optimizations for performance and scalability\n- ensure all code follows our security guidelines\n- help write comprehensive tests for new features\n```\n\n#### Additional sections to consider\n\n- **project context**: brief description of the project, its goals, and target users\n- **technology stack**: list of technologies, frameworks, and tools used\n- **coding standards**: specific conventions, style guides, and patterns to follow\n- **architecture guidelines**: how components should interact and be organized\n- **testing requirements**: testing strategies and coverage expectations\n- **security considerations**: specific security requirements or constraints\n\n### example avante.md\n\n```markdown\n# project instructions for myapp\n\n## your role\n\nyou are an expert full-stack developer specializing in react, node.js, and typescript. you understand modern web development practices and have experience with our tech stack.\n\n## your mission\n\nhelp build a scalable e-commerce platform by:\n\n- writing type-safe typescript code\n- following react best practices and hooks patterns\n- implementing restful apis with proper error handling\n- ensuring responsive design with tailwind css\n- writing comprehensive unit and integration tests\n\n## project context\n\nmyapp is a modern e-commerce platform targeting small businesses. we prioritize performance, accessibility, and user experience.\n\n## technology stack\n\n- frontend: react 18, typescript, tailwind css, vite\n- backend: node.js, express, prisma, postgresql\n- testing: jest, react testing library, playwright\n- deployment: docker, aws\n\n## coding standards\n\n- use functional components with hooks\n- prefer composition over inheritance\n- write self-documenting code with clear variable names\n- add jsdoc comments for complex functions\n- follow the existing folder structure and naming conventions\n```\n\n
\n\n## Installation\n\nFor building binary if you wish to build from source, then `cargo` is required. Otherwise `curl` and `tar` will be used to get prebuilt binary from GitHub.\n\n
\n\n lazy.nvim (recommended)\n\n```lua\n{\n \"yetone/avante.nvim\",\n -- if you want to build from source then do `make BUILD_FROM_SOURCE=true`\n -- ⚠️ must add this setting! ! !\n build = vim.fn.has(\"win32\") ~= 0\n and \"powershell -ExecutionPolicy Bypass -File Build.ps1 -BuildFromSource false\"\n or \"make\",\n event = \"VeryLazy\",\n version = false, -- Never set this value to \"*\"! Never!\n ---@module 'avante'\n ---@type avante.Config\n opts = {\n -- add any opts here\n -- this file can contain specific instructions for your project\n instructions_file = \"avante.md\",\n -- for example\n provider = \"claude\",\n providers = {\n claude = {\n endpoint = \"https://api.anthropic.com\",\n model = \"claude-sonnet-4-20250514\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 20480,\n },\n },\n moonshot = {\n endpoint = \"https://api.moonshot.ai/v1\",\n model = \"kimi-k2-0711-preview\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 32768,\n },\n },\n },\n },\n dependencies = {\n \"nvim-lua/plenary.nvim\",\n \"MunifTanjim/nui.nvim\",\n --- The below dependencies are optional,\n \"nvim-mini/mini.pick\", -- for file_selector provider mini.pick\n \"nvim-telescope/telescope.nvim\", -- for file_selector provider telescope\n \"hrsh7th/nvim-cmp\", -- autocompletion for avante commands and mentions\n \"ibhagwan/fzf-lua\", -- for file_selector provider fzf\n \"stevearc/dressing.nvim\", -- for input provider dressing\n \"folke/snacks.nvim\", -- for input provider snacks\n \"nvim-tree/nvim-web-devicons\", -- or echasnovski/mini.icons\n \"zbirenbaum/copilot.lua\", -- for providers='copilot'\n {\n -- support for image pasting\n \"HakonHarnes/img-clip.nvim\",\n event = \"VeryLazy\",\n opts = {\n -- recommended settings\n default = {\n embed_image_as_base64 = false,\n prompt_for_file_name = false,\n drag_and_drop = {\n insert_mode = true,\n },\n -- required for Windows users\n use_absolute_path = true,\n },\n },\n },\n {\n -- Make sure to set this up properly if you have lazy=true\n 'MeanderingProgrammer/render-markdown.nvim',\n opts = {\n file_types = { \"markdown\", \"Avante\" },\n },\n ft = { \"markdown\", \"Avante\" },\n },\n },\n}\n```\n\n
\n\n
\n\n vim-plug\n\n```vim\n\ncall plug#begin()\n\n\" Deps\nPlug 'nvim-lua/plenary.nvim'\nPlug 'MunifTanjim/nui.nvim'\nPlug 'MeanderingProgrammer/render-markdown.nvim'\n\n\" Optional deps\nPlug 'hrsh7th/nvim-cmp'\nPlug 'nvim-tree/nvim-web-devicons' \"or Plug 'echasnovski/mini.icons'\nPlug 'HakonHarnes/img-clip.nvim'\nPlug 'zbirenbaum/copilot.lua'\nPlug 'stevearc/dressing.nvim' \" for enhanced input UI\nPlug 'folke/snacks.nvim' \" for modern input UI\n\n\" Yay, pass source=true if you want to build from source\nPlug 'yetone/avante.nvim', { 'branch': 'main', 'do': 'make' }\n\ncall plug#end()\n\nautocmd! User avante.nvim\nlua << EOF\nrequire('avante').setup({})\nEOF\n```\n\n
\n\n
\n\n mini.deps\n\n```lua\nlocal add, later, now = MiniDeps.add, MiniDeps.later, MiniDeps.now\n\nadd({\n source = 'yetone/avante.nvim',\n monitor = 'main',\n depends = {\n 'nvim-lua/plenary.nvim',\n 'MunifTanjim/nui.nvim',\n 'echasnovski/mini.icons'\n },\n hooks = { post_checkout = function() vim.cmd('make') end }\n})\n--- optional\nadd({ source = 'hrsh7th/nvim-cmp' })\nadd({ source = 'zbirenbaum/copilot.lua' })\nadd({ source = 'HakonHarnes/img-clip.nvim' })\nadd({ source = 'MeanderingProgrammer/render-markdown.nvim' })\n\nlater(function() require('render-markdown').setup({...}) end)\nlater(function()\n require('img-clip').setup({...}) -- config img-clip\n require(\"copilot\").setup({...}) -- setup copilot to your liking\n require(\"avante\").setup({...}) -- config for avante.nvim\nend)\n```\n\n
\n\n
\n\n Packer\n\n```vim\n\n -- Required plugins\n use 'nvim-lua/plenary.nvim'\n use 'MunifTanjim/nui.nvim'\n use 'MeanderingProgrammer/render-markdown.nvim'\n\n -- Optional dependencies\n use 'hrsh7th/nvim-cmp'\n use 'nvim-tree/nvim-web-devicons' -- or use 'echasnovski/mini.icons'\n use 'HakonHarnes/img-clip.nvim'\n use 'zbirenbaum/copilot.lua'\n use 'stevearc/dressing.nvim' -- for enhanced input UI\n use 'folke/snacks.nvim' -- for modern input UI\n\n -- Avante.nvim with build process\n use {\n 'yetone/avante.nvim',\n branch = 'main',\n run = 'make',\n config = function()\n require('avante').setup()\n end\n }\n```\n\n
\n\n
\n\n Home Manager\n\n```nix\nprograms.neovim = {\n plugins = [\n {\n plugin = pkgs.vimPlugins.avante-nvim;\n type = \"lua\";\n config = ''\n require(\"avante_lib\").load()\n require(\"avante\").setup()\n '' # or builtins.readFile ./plugins/avante.lua;\n }\n ];\n};\n```\n\n
\n\n
\n\n Nixvim\n\n```nix\n plugins.avante.enable = true;\n plugins.avante.settings = {\n # setup options here\n };\n```\n\n
\n\n
\n\n Lua\n\n```lua\n-- deps:\nrequire('cmp').setup ({\n -- use recommended settings from above\n})\nrequire('img-clip').setup ({\n -- use recommended settings from above\n})\nrequire('copilot').setup ({\n -- use recommended settings from above\n})\nrequire('render-markdown').setup ({\n -- use recommended settings from above\n})\nrequire('avante').setup({\n -- Example: Using snacks.nvim as input provider\n input = {\n provider = \"snacks\", -- \"native\" | \"dressing\" | \"snacks\"\n provider_opts = {\n -- Snacks input configuration\n title = \"Avante Input\",\n icon = \" \",\n placeholder = \"Enter your API key...\",\n },\n },\n -- Your other config here!\n})\n```\n\n
\n\n> [!IMPORTANT]\n>\n> `avante.nvim` is currently only compatible with Neovim 0.10.1 or later. Please ensure that your Neovim version meets these requirements before proceeding.\n\n> [!NOTE]\n>\n> When loading the plugin synchronously, we recommend `require`ing it sometime after your colorscheme.\n\n> [!NOTE]\n>\n> Recommended **Neovim** options:\n>\n> ```lua\n> -- views can only be fully collapsed with the global statusline\n> vim.opt.laststatus = 3\n> ```\n\n> [!TIP]\n>\n> Any rendering plugins that support markdown should work with Avante as long as you add the supported filetype `Avante`. See and [this comment](https://github.com/yetone/avante.nvim/issues/175#issuecomment-2313749363) for more information.\n\n### Default setup configuration\n\n_See [config.lua#L9](./lua/avante/config.lua) for the full config_\n\n
\nDefault configuration\n\n```lua\n{\n ---@alias Provider \"claude\" | \"openai\" | \"azure\" | \"gemini\" | \"cohere\" | \"copilot\" | string\n ---@type Provider\n provider = \"claude\", -- The provider used in Aider mode or in the planning phase of Cursor Planning Mode\n ---@alias Mode \"agentic\" | \"legacy\"\n ---@type Mode\n mode = \"agentic\", -- The default mode for interaction. \"agentic\" uses tools to automatically generate code, \"legacy\" uses the old planning method to generate code.\n -- WARNING: Since auto-suggestions are a high-frequency operation and therefore expensive,\n -- currently designating it as `copilot` provider is dangerous because: https://github.com/yetone/avante.nvim/issues/1048\n -- Of course, you can reduce the request frequency by increasing `suggestion.debounce`.\n auto_suggestions_provider = \"claude\",\n providers = {\n claude = {\n endpoint = \"https://api.anthropic.com\",\n auth_type = \"api\" -- Set to \"max\" to sign in with Claude Pro/Max subscription\n model = \"claude-3-5-sonnet-20241022\",\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 4096,\n },\n },\n },\n ---Specify the special dual_boost mode\n ---1. enabled: Whether to enable dual_boost mode. Default to false.\n ---2. first_provider: The first provider to generate response. Default to \"openai\".\n ---3. second_provider: The second provider to generate response. Default to \"claude\".\n ---4. prompt: The prompt to generate response based on the two reference outputs.\n ---5. timeout: Timeout in milliseconds. Default to 60000.\n ---How it works:\n --- When dual_boost is enabled, avante will generate two responses from the first_provider and second_provider respectively. Then use the response from the first_provider as provider1_output and the response from the second_provider as provider2_output. Finally, avante will generate a response based on the prompt and the two reference outputs, with the default Provider as normal.\n ---Note: This is an experimental feature and may not work as expected.\n dual_boost = {\n enabled = false,\n first_provider = \"openai\",\n second_provider = \"claude\",\n prompt = \"Based on the two reference outputs below, generate a response that incorporates elements from both but reflects your own judgment and unique perspective. Do not provide any explanation, just give the response directly. Reference Output 1: [{{provider1_output}}], Reference Output 2: [{{provider2_output}}]\",\n timeout = 60000, -- Timeout in milliseconds\n },\n behaviour = {\n auto_suggestions = false, -- Experimental stage\n auto_set_highlight_group = true,\n auto_set_keymaps = true,\n auto_apply_diff_after_generation = false,\n support_paste_from_clipboard = false,\n minimize_diff = true, -- Whether to remove unchanged lines when applying a code block\n enable_token_counting = true, -- Whether to enable token counting. Default to true.\n auto_add_current_file = true, -- Whether to automatically add the current file when opening a new chat. Default to true.\n auto_approve_tool_permissions = true, -- Default: auto-approve all tools (no prompts)\n -- Examples:\n -- auto_approve_tool_permissions = false, -- Show permission prompts for all tools\n -- auto_approve_tool_permissions = {\"bash\", \"str_replace\"}, -- Auto-approve specific tools only\n ---@type \"popup\" | \"inline_buttons\"\n confirmation_ui_style = \"inline_buttons\",\n --- Whether to automatically open files and navigate to lines when ACP agent makes edits\n ---@type boolean\n acp_follow_agent_locations = true,\n },\n prompt_logger = { -- logs prompts to disk (timestamped, for replay/debugging)\n enabled = true, -- toggle logging entirely\n log_dir = vim.fn.stdpath(\"cache\") .. \"/avante_prompts\", -- directory where logs are saved\n fortune_cookie_on_success = false, -- shows a random fortune after each logged prompt (requires `fortune` installed)\n next_prompt = {\n normal = \"\", -- load the next (newer) prompt log in normal mode\n insert = \"\",\n },\n prev_prompt = {\n normal = \"\", -- load the previous (older) prompt log in normal mode\n insert = \"\",\n },\n },\n mappings = {\n --- @class AvanteConflictMappings\n diff = {\n ours = \"co\",\n theirs = \"ct\",\n all_theirs = \"ca\",\n both = \"cb\",\n cursor = \"cc\",\n next = \"]x\",\n prev = \"[x\",\n },\n suggestion = {\n accept = \"\",\n next = \"\",\n prev = \"\",\n dismiss = \"\",\n },\n jump = {\n next = \"]]\",\n prev = \"[[\",\n },\n submit = {\n normal = \"\",\n insert = \"\",\n },\n cancel = {\n normal = { \"\", \"\", \"q\" },\n insert = { \"\" },\n },\n sidebar = {\n apply_all = \"A\",\n apply_cursor = \"a\",\n retry_user_request = \"r\",\n edit_user_request = \"e\",\n switch_windows = \"\",\n reverse_switch_windows = \"\",\n remove_file = \"d\",\n add_file = \"@\",\n close = { \"\", \"q\" },\n close_from_input = nil, -- e.g., { normal = \"\", insert = \"\" }\n },\n },\n selection = {\n enabled = true,\n hint_display = \"delayed\",\n },\n windows = {\n ---@type \"right\" | \"left\" | \"top\" | \"bottom\"\n position = \"right\", -- the position of the sidebar\n wrap = true, -- similar to vim.o.wrap\n width = 30, -- default % based on available width\n sidebar_header = {\n enabled = true, -- true, false to enable/disable the header\n align = \"center\", -- left, center, right for title\n rounded = true,\n },\n spinner = {\n editing = { \"⡀\", \"⠄\", \"⠂\", \"⠁\", \"⠈\", \"⠐\", \"⠠\", \"⢀\", \"⣀\", \"⢄\", \"⢂\", \"⢁\", \"⢈\", \"⢐\", \"⢠\", \"⣠\", \"⢤\", \"⢢\", \"⢡\", \"⢨\", \"⢰\", \"⣰\", \"⢴\", \"⢲\", \"⢱\", \"⢸\", \"⣸\", \"⢼\", \"⢺\", \"⢹\", \"⣹\", \"⢽\", \"⢻\", \"⣻\", \"⢿\", \"⣿\" },\n generating = { \"·\", \"✢\", \"✳\", \"∗\", \"✻\", \"✽\" }, -- Spinner characters for the 'generating' state\n thinking = { \"🤯\", \"🙄\" }, -- Spinner characters for the 'thinking' state\n },\n input = {\n prefix = \"> \",\n height = 8, -- Height of the input window in vertical layout\n },\n edit = {\n border = \"rounded\",\n start_insert = true, -- Start insert mode when opening the edit window\n },\n ask = {\n floating = false, -- Open the 'AvanteAsk' prompt in a floating window\n start_insert = true, -- Start insert mode when opening the ask window\n border = \"rounded\",\n ---@type \"ours\" | \"theirs\"\n focus_on_apply = \"ours\", -- which diff to focus after applying\n },\n },\n highlights = {\n ---@type AvanteConflictHighlights\n diff = {\n current = \"DiffText\",\n incoming = \"DiffAdd\",\n },\n },\n --- @class AvanteConflictUserConfig\n diff = {\n autojump = true,\n ---@type string | fun(): any\n list_opener = \"copen\",\n --- Override the 'timeoutlen' setting while hovering over a diff (see :help timeoutlen).\n --- Helps to avoid entering operator-pending mode with diff mappings starting with `c`.\n --- Disable by setting to -1.\n override_timeoutlen = 500,\n },\n suggestion = {\n debounce = 600,\n throttle = 600,\n },\n}\n```\n\n
\n\n## Blink.cmp users\n\nFor blink cmp users (nvim-cmp alternative) view below instruction for configuration\nThis is achieved by emulating nvim-cmp using blink.compat\nor you can use [Kaiser-Yang/blink-cmp-avante](https://github.com/Kaiser-Yang/blink-cmp-avante).\n\n
\n Lua\n\n```lua\n selector = {\n --- @alias avante.SelectorProvider \"native\" | \"fzf_lua\" | \"mini_pick\" | \"snacks\" | \"telescope\" | fun(selector: avante.ui.Selector): nil\n --- @type avante.SelectorProvider\n provider = \"fzf\",\n -- Options override for custom providers\n provider_opts = {},\n }\n```\n\nTo create a customized selector provider, you can specify a customized function to launch a picker to select items and pass the selected items to the `on_select` callback.\n\n```lua\n selector = {\n ---@param selector avante.ui.Selector\n provider = function(selector)\n local items = selector.items ---@type avante.ui.SelectorItem[]\n local title = selector.title ---@type string\n local on_select = selector.on_select ---@type fun(selected_item_ids: string[]|nil): nil\n\n --- your customized picker logic here\n end,\n }\n```\n\n### Input Provider Configuration\n\nAvante.nvim supports multiple input providers for user input (like API key entry). You can configure which provider to use:\n\n
\n Native Input Provider (Default)\n\n```lua\n{\n input = {\n provider = \"native\", -- Uses vim.ui.input\n provider_opts = {},\n }\n}\n```\n\n
\n\n
\n Dressing.nvim Input Provider\n\nFor enhanced input UI with better styling and features:\n\n```lua\n{\n input = {\n provider = \"dressing\",\n provider_opts = {},\n }\n}\n```\n\nYou'll need to install dressing.nvim:\n\n```lua\n-- With lazy.nvim\n{ \"stevearc/dressing.nvim\" }\n```\n\n
\n\n
\n Snacks.nvim Input Provider (Recommended)\n\nFor modern, feature-rich input UI:\n\n```lua\n{\n input = {\n provider = \"snacks\",\n provider_opts = {\n -- Additional snacks.input options\n title = \"Avante Input\",\n icon = \" \",\n },\n }\n}\n```\n\nYou'll need to install snacks.nvim:\n\n```lua\n-- With lazy.nvim\n{ \"folke/snacks.nvim\" }\n```\n\n
\n\n
\n Custom Input Provider\n\nTo create a customized input provider, you can specify a function:\n\n```lua\n{\n input = {\n ---@param input avante.ui.Input\n provider = function(input)\n local title = input.title ---@type string\n local default = input.default ---@type string\n local conceal = input.conceal ---@type boolean\n local on_submit = input.on_submit ---@type fun(result: string|nil): nil\n\n --- your customized input logic here\n end,\n }\n}\n```\n\n
\n\nChoose a selector other that native, the default as that currently has an issue\nFor lazyvim users copy the full config for blink.cmp from the website or extend the options\n\n```lua\n compat = {\n \"avante_commands\",\n \"avante_mentions\",\n \"avante_files\",\n }\n```\n\nFor other users just add a custom provider\n\n### Available Completion Sources\n\nAvante.nvim provides several completion sources that can be integrated with blink.cmp:\n\n#### Mentions (`@` trigger)\n\nMentions allow you to quickly reference specific features or add files to the chat context:\n\n- `@codebase` - Enable project context and repository mapping\n- `@diagnostics` - Enable diagnostics information\n- `@file` - Open file selector to add files to chat context\n- `@quickfix` - Add files from quickfix list to chat context\n- `@buffers` - Add open buffers to chat context\n\n#### Slash Commands (`/` trigger)\n\nBuilt-in slash commands for common operations:\n\n- `/help` - Show help message with available commands\n- `/init` - Initialize AGENTS.md based on current project\n- `/clear` - Clear chat history\n- `/new` - Start a new chat\n- `/compact` - Compact history messages to save tokens\n- `/lines - ` - Ask about specific lines\n- `/commit` - Generate commit message for changes\n\n#### Shortcuts (`#` trigger)\n\nShortcuts provide quick access to predefined prompt templates. You can customize these in your config:\n\n```lua\n{\n shortcuts = {\n {\n name = \"refactor\",\n description = \"Refactor code with best practices\",\n details = \"Automatically refactor code to improve readability, maintainability, and follow best practices while preserving functionality\",\n prompt = \"Please refactor this code following best practices, improving readability and maintainability while preserving functionality.\"\n },\n {\n name = \"test\",\n description = \"Generate unit tests\",\n details = \"Create comprehensive unit tests covering edge cases, error scenarios, and various input conditions\",\n prompt = \"Please generate comprehensive unit tests for this code, covering edge cases and error scenarios.\"\n },\n -- Add more custom shortcuts...\n }\n}\n```\n\nWhen you type `#refactor` in the input, it will automatically be replaced with the corresponding prompt text.\n\n### Configuration Example\n\nHere's a complete blink.cmp configuration example with all Avante sources:\n\n```lua\n default = {\n ...\n \"avante_commands\",\n \"avante_mentions\",\n \"avante_shortcuts\",\n \"avante_files\",\n }\n```\n\n```lua\n providers = {\n avante_commands = {\n name = \"avante_commands\",\n module = \"blink.compat.source\",\n score_offset = 90, -- show at a higher priority than lsp\n opts = {},\n },\n avante_files = {\n name = \"avante_files\",\n module = \"blink.compat.source\",\n score_offset = 100, -- show at a higher priority than lsp\n opts = {},\n },\n avante_mentions = {\n name = \"avante_mentions\",\n module = \"blink.compat.source\",\n score_offset = 1000, -- show at a higher priority than lsp\n opts = {},\n },\n avante_shortcuts = {\n name = \"avante_shortcuts\",\n module = \"blink.compat.source\",\n score_offset = 1000, -- show at a higher priority than lsp\n opts = {},\n }\n ...\n }\n```\n\n
\n\n## Usage\n\n### Using Claude Pro/Max Subscription\nTo login with your Claude subscription, set the **auth_type** of the Claude provider entry in your config to \"max\", re-open Neovim then the authentication process will start in your browser. Once logged in and authorized, a code will show that needs to be copied into the prompt in Neovim, which should then give access to use your subscription with Avante.\n\nYou may need to run `AvanteSwitchProvider claude` to initiate the authentication if you previously had a different provider selected.\n\n```lua\n-- Providers = { ...\n\n claude = {\n -- ...\n auth_type = \"max\",\n },\n\n```\n\n### Basic Functionality\n\nGiven its early stage, `avante.nvim` currently supports the following basic functionalities:\n\n> [!IMPORTANT]\n>\n> For most consistency between neovim session, it is recommended to set the environment variables in your shell file.\n> By default, `Avante` will prompt you at startup to input the API key for the provider you have selected.\n>\n> **Scoped API Keys (Recommended for Isolation)**\n>\n> Avante now supports scoped API keys, allowing you to isolate API keys specifically for Avante without affecting other applications. Simply prefix any API key with `AVANTE_`:\n>\n> ```sh\n> # Scoped keys (recommended)\n> export AVANTE_ANTHROPIC_API_KEY=your-claude-api-key\n> export AVANTE_OPENAI_API_KEY=your-openai-api-key\n> export AVANTE_AZURE_OPENAI_API_KEY=your-azure-api-key\n> export AVANTE_GEMINI_API_KEY=your-gemini-api-key\n> export AVANTE_CO_API_KEY=your-cohere-api-key\n> export AVANTE_AIHUBMIX_API_KEY=your-aihubmix-api-key\n> export AVANTE_MOONSHOT_API_KEY=your-moonshot-api-key\n> ```\n>\n> **Global API Keys (Legacy)**\n>\n> You can still use the traditional global API keys if you prefer:\n>\n> For Claude:\n>\n> ```sh\n> export ANTHROPIC_API_KEY=your-api-key\n> ```\n>\n> For OpenAI:\n>\n> ```sh\n> export OPENAI_API_KEY=your-api-key\n> ```\n>\n> For Azure OpenAI:\n>\n> ```sh\n> export AZURE_OPENAI_API_KEY=your-api-key\n> ```\n>\n> For Amazon Bedrock:\n>\n> You can specify the `BEDROCK_KEYS` environment variable to set credentials. When this variable is not specified, bedrock will use the default AWS credentials chain (see below).\n>\n> ```sh\n> export BEDROCK_KEYS=aws_access_key_id,aws_secret_access_key,aws_region[,aws_session_token]\n> ```\n>\n> Note: The aws_session_token is optional and only needed when using temporary AWS credentials\n>\n> Alternatively Bedrock tries to resolve AWS credentials using the [Default Credentials Provider Chain](https://docs.aws.amazon.com/cli/v1/userguide/cli-chap-authentication.html).\n> This means you can have credentials e.g. configured via the AWS CLI, stored in your ~/.aws/profile, use AWS SSO etc.\n> In this case `aws_region` and optionally `aws_profile` should be specified via the bedrock config, e.g.:\n>\n> ```lua\n> bedrock = {\n> model = \"us.anthropic.claude-3-5-sonnet-20241022-v2:0\",\n> aws_profile = \"bedrock\",\n> aws_region = \"us-east-1\",\n> },\n> ```\n>\n> Note: Bedrock requires the [AWS CLI](https://aws.amazon.com/cli/) to be installed on your system.\n\n1. Open a code file in Neovim.\n2. Use the `:AvanteAsk` command to query the AI about the code.\n3. Review the AI's suggestions.\n4. Apply the recommended changes directly to your code with a simple command or key binding.\n\n**Note**: The plugin is still under active development, and both its functionality and interface are subject to significant changes. Expect some rough edges and instability as the project evolves.\n\n## Key Bindings\n\nThe following key bindings are available for use with `avante.nvim`:\n\n| Key Binding | Description |\n| ----------------------------------------- | -------------------------------------- |\n| **Sidebar** | |\n| ]p | next prompt |\n| [p | previous prompt |\n| A | apply all |\n| a | apply cursor |\n| r | retry user request |\n| e | edit user request |\n| <Tab> | switch windows |\n| <S-Tab> | reverse switch windows |\n| d | remove file |\n| @ | add file |\n| q | close sidebar |\n| Leaderaa | show sidebar |\n| Leaderat | toggle sidebar visibility |\n| Leaderar | refresh sidebar |\n| Leaderaf | switch sidebar focus |\n| **Suggestion** | |\n| Leadera? | select model |\n| Leaderan | new ask |\n| Leaderae | edit selected blocks |\n| LeaderaS | stop current AI request |\n| Leaderah | select between chat histories |\n| <M-l> | accept suggestion |\n| <M-]> | next suggestion |\n| <M-[> | previous suggestion |\n| <C-]> | dismiss suggestion |\n| Leaderad | toggle debug mode |\n| Leaderas | toggle suggestion display |\n| LeaderaR | toggle repomap |\n| **Files** | |\n| Leaderac | add current buffer to selected files |\n| LeaderaB | add all buffer files to selected files |\n| **Diff** | |\n| co | choose ours |\n| ct | choose theirs |\n| ca | choose all theirs |\n| cb | choose both |\n| cc | choose cursor |\n| ]x | move to next conflict |\n| [x | move to previous conflict |\n| **Confirm** | |\n| Ctrlwf | focus confirm window |\n| c | confirm code |\n| r | confirm response |\n| i | confirm input |\n\n> [!NOTE]\n>\n> If you are using `lazy.nvim`, then all keymap here will be safely set, meaning if `aa` is already binded, then avante.nvim won't bind this mapping.\n> In this case, user will be responsible for setting up their own. See [notes on keymaps](https://github.com/yetone/avante.nvim/wiki#keymaps-and-api-i-guess) for more details.\n\n### Neotree shortcut\n\nIn the neotree sidebar, you can also add a new keyboard shortcut to quickly add `file/folder` to `Avante Selected Files`.\n\n
\nNeotree configuration\n\n```lua\nreturn {\n {\n 'nvim-neo-tree/neo-tree.nvim',\n config = function()\n require('neo-tree').setup({\n filesystem = {\n commands = {\n avante_add_files = function(state)\n local node = state.tree:get_node()\n local filepath = node:get_id()\n local relative_path = require('avante.utils').relative_path(filepath)\n\n local sidebar = require('avante').get()\n\n local open = sidebar:is_open()\n -- ensure avante sidebar is open\n if not open then\n require('avante.api').ask()\n sidebar = require('avante').get()\n end\n\n sidebar.file_selector:add_selected_file(relative_path)\n\n -- remove neo tree buffer\n if not open then\n sidebar.file_selector:remove_selected_file('neo-tree filesystem [1]')\n end\n end,\n },\n window = {\n mappings = {\n ['oa'] = 'avante_add_files',\n },\n },\n },\n })\n end,\n },\n}\n```\n\n
\n\n## Commands\n\n| Command | Description | Examples |\n| ---------------------------------- | ----------------------------------------------------------------------------------------------------------- | --------------------------------------------------- |\n| `:AvanteAsk [question] [position]` | Ask AI about your code. Optional `position` set window position and `ask` enable/disable direct asking mode | `:AvanteAsk position=right Refactor this code here` |\n| `:AvanteBuild` | Build dependencies for the project | |\n| `:AvanteChat` | Start a chat session with AI about your codebase. Default is `ask`=false | |\n| `:AvanteChatNew` | Start a new chat session. The current chat can be re-opened with the chat session selector | |\n| `:AvanteHistory` | Opens a picker for your previous chat sessions | |\n| `:AvanteClear` | Clear the chat history for your current chat session | |\n| `:AvanteEdit` | Edit the selected code blocks | |\n| `:AvanteFocus` | Switch focus to/from the sidebar | |\n| `:AvanteRefresh` | Refresh all Avante windows | |\n| `:AvanteStop` | Stop the current AI request | |\n| `:AvanteSwitchProvider` | Switch AI provider (e.g. openai) | |\n| `:AvanteShowRepoMap` | Show repo map for project's structure | |\n| `:AvanteToggle` | Toggle the Avante sidebar | |\n| `:AvanteModels` | Show model list | |\n| `:AvanteSwitchSelectorProvider` | Switch avante selector provider (e.g. native, telescope, fzf_lua, mini_pick, snacks) | |\n\n## Highlight Groups\n\n| Highlight Group | Description | Notes |\n| --------------------------- | --------------------------------------------- | -------------------------------------------- |\n| AvanteTitle | Title | |\n| AvanteReversedTitle | Used for rounded border | |\n| AvanteSubtitle | Selected code title | |\n| AvanteReversedSubtitle | Used for rounded border | |\n| AvanteThirdTitle | Prompt title | |\n| AvanteReversedThirdTitle | Used for rounded border | |\n| AvanteConflictCurrent | Current conflict highlight | Default to `Config.highlights.diff.current` |\n| AvanteConflictIncoming | Incoming conflict highlight | Default to `Config.highlights.diff.incoming` |\n| AvanteConflictCurrentLabel | Current conflict label highlight | Default to shade of `AvanteConflictCurrent` |\n| AvanteConflictIncomingLabel | Incoming conflict label highlight | Default to shade of `AvanteConflictIncoming` |\n| AvantePopupHint | Usage hints in popup menus | |\n| AvanteInlineHint | The end-of-line hint displayed in visual mode | |\n| AvantePromptInput | The body highlight of the prompt input | |\n| AvantePromptInputBorder | The border highlight of the prompt input | Default to `NormalFloat` |\n\nSee [highlights.lua](./lua/avante/highlights.lua) for more information\n\n## Fast Apply\n\nFast Apply is a feature that enables instant code edits with high accuracy by leveraging specialized models. It replicates Cursor's instant apply functionality, allowing for seamless code modifications without the typical delays associated with traditional code generation.\n\n### Purpose and Benefits\n\nFast Apply addresses the common pain point of slow code application in AI-assisted development. Instead of waiting for a full language model to process and apply changes, Fast Apply uses a specialized \"apply model\" that can quickly and accurately merge code edits with 96-98% accuracy at speeds of 2500-4500+ tokens per second.\n\nKey benefits:\n\n- **Instant application**: Code changes are applied immediately without noticeable delays\n- **High accuracy**: Specialized models achieve 96-98% accuracy for code edits\n- **Seamless workflow**: Maintains the natural flow of development without interruptions\n- **Large context support**: Handles up to 16k tokens for both input and output\n\n### Configuration\n\nTo enable Fast Apply, you need to:\n\n1. **Enable Fast Apply in your configuration**:\n\n ```lua\n behaviour = {\n enable_fastapply = true, -- Enable Fast Apply feature\n },\n -- ... other configuration\n ```\n\n2. **Get your Morph API key**:\n Go to [morphllm.com](https://morphllm.com/api-keys) and create an account and get the API key.\n\n3. **Set your Morph API key**:\n\n ```bash\n export MORPH_API_KEY=\"your-api-key\"\n ```\n\n4. **Change Morph model**:\n ```lua\n providers = {\n morph = {\n model = \"morph-v3-large\",\n },\n }\n ```\n\n### Model Options\n\nMorph provides different models optimized for different use cases:\n\n| Model | Speed | Accuracy | Context Limit |\n| ---------------- | ----------------- | -------- | ------------- |\n| `morph-v3-fast` | 4500+ tok/sec | 96% | 16k tokens |\n| `morph-v3-large` | 2500+ tok/sec | 98% | 16k tokens |\n| `auto` | 2500-4500 tok/sec | 98% | 16k tokens |\n\n### How It Works\n\nWhen Fast Apply is enabled and a Morph provider is configured, avante.nvim will:\n\n1. Use the `edit_file` tool for code modifications instead of traditional tools\n2. Send the original code, edit instructions, and update snippet to the Morph API\n3. Receive the fully merged code back from the specialized apply model\n4. Apply the changes directly to your files with high accuracy\n\nThe process uses a specialized prompt format that includes:\n\n- ``: Clear description of what changes to make\n- ``: The original code content\n- ``: The specific changes using truncation markers (`// ... existing code ...`)\n\nThis approach ensures that the apply model can quickly and accurately merge your changes without the overhead of full code generation.\n\n## Ollama\n\nOllama is a first-class provider for avante.nvim. To start using it you need to set `provider = \"ollama\"`\nin the configuration, set the `model` field in `ollama` to the model you want to use. Ollama is disabled\nby default, you need to provide an implementation for its `is_env_set` method to properly enable it.\nFor example:\n\n```lua\nprovider = \"ollama\",\nproviders = {\n ollama = {\n model = \"qwq:32b\",\n is_env_set = require(\"avante.providers.ollama\").check_endpoint_alive,\n },\n}\n```\n\n## ACP Support\n\nAvante.nvim now supports the [Agent Client Protocol (ACP)](https://agentclientprotocol.com/overview/introduction), enabling seamless integration with AI agents that follow this standardized communication protocol. ACP provides a unified way for AI agents to interact with development environments, offering enhanced capabilities for code editing, file operations, and tool execution.\n\n### What is ACP?\n\nThe Agent Client Protocol (ACP) is a standardized protocol that enables AI agents to communicate with development tools and environments. It provides:\n\n- **Standardized Communication**: A unified JSON-RPC based protocol for agent-client interactions\n- **Tool Integration**: Support for various development tools like file operations, code execution, and search\n- **Session Management**: Persistent sessions that maintain context across interactions\n- **Permission System**: Granular control over what agents can access and modify\n\n### Enabling ACP\n\nTo use ACP-compatible agents with Avante.nvim, you need to configure an ACP provider. Here are the currently supported ACP agents:\n\n#### Gemini CLI with ACP\n```lua\n{\n provider = \"gemini-cli\",\n -- other configuration options...\n}\n```\n\n#### Claude Code with ACP\n```lua\n{\n provider = \"claude-code\",\n -- other configuration options...\n}\n```\n\n#### Goose with ACP\n```lua\n{\n provider = \"goose\",\n -- other configuration options...\n}\n```\n\n#### Codex with ACP\n```lua\n{\n provider = \"codex\",\n -- other configuration options...\n}\n```\n\n#### Kimi CLI with ACP\n```lua\n{\n provider = \"kimi-cli\",\n -- other configuration options...\n}\n```\n\n### ACP Configuration\n\nACP providers are configured in the `acp_providers` section of your configuration:\n\n```lua\n{\n acp_providers = {\n [\"gemini-cli\"] = {\n command = \"gemini\",\n args = { \"--experimental-acp\" },\n env = {\n NODE_NO_WARNINGS = \"1\",\n GEMINI_API_KEY = os.getenv(\"GEMINI_API_KEY\"),\n },\n },\n [\"claude-code\"] = {\n command = \"npx\",\n args = { \"@zed-industries/claude-code-acp\" },\n env = {\n NODE_NO_WARNINGS = \"1\",\n ANTHROPIC_API_KEY = os.getenv(\"ANTHROPIC_API_KEY\"),\n },\n },\n [\"goose\"] = {\n command = \"goose\",\n args = { \"acp\" },\n },\n [\"codex\"] = {\n command = \"npx\",\n args = { \"@zed-industries/codex-acp\" },\n env = {\n NODE_NO_WARNINGS = \"1\",\n OPENAI_API_KEY = os.getenv(\"OPENAI_API_KEY\"),\n },\n },\n },\n -- other configuration options...\n}\n```\n\n### Prerequisites\n\nBefore using ACP agents, ensure you have the required tools installed:\n\n- **For Gemini CLI**: Install the `gemini` CLI tool and set your `GEMINI_API_KEY`\n- **For Claude Code**: Install the `acp-claude-code` package via npm and set your `ANTHROPIC_API_KEY`\n\n### ACP vs Traditional Providers\n\nACP providers offer several advantages over traditional API-based providers:\n\n- **Enhanced Tool Access**: Agents can directly interact with your file system, run commands, and access development tools\n- **Persistent Context**: Sessions maintain state across multiple interactions\n- **Fine-grained Permissions**: Control exactly what agents can access and modify\n- **Standardized Protocol**: Compatible with any ACP-compliant agent\n\n## Custom providers\n\nAvante provides a set of default providers, but users can also create their own providers.\n\nFor more information, see [Custom Providers](https://github.com/yetone/avante.nvim/wiki/Custom-providers)\n\n## RAG Service\n\nAvante provides a RAG service, which is a tool for obtaining the required context for the AI to generate the codes. By default, it is not enabled. You can enable it this way:\n\n```lua\n rag_service = { -- RAG Service configuration\n enabled = false, -- Enables the RAG service\n host_mount = os.getenv(\"HOME\"), -- Host mount path for the rag service (Docker will mount this path)\n runner = \"docker\", -- Runner for the RAG service (can use docker or nix)\n llm = { -- Language Model (LLM) configuration for RAG service\n provider = \"openai\", -- LLM provider\n endpoint = \"https://api.openai.com/v1\", -- LLM API endpoint\n api_key = \"OPENAI_API_KEY\", -- Environment variable name for the LLM API key\n model = \"gpt-4o-mini\", -- LLM model name\n extra = nil, -- Additional configuration options for LLM\n },\n embed = { -- Embedding model configuration for RAG service\n provider = \"openai\", -- Embedding provider\n endpoint = \"https://api.openai.com/v1\", -- Embedding API endpoint\n api_key = \"OPENAI_API_KEY\", -- Environment variable name for the embedding API key\n model = \"text-embedding-3-large\", -- Embedding model name\n extra = nil, -- Additional configuration options for the embedding model\n },\n docker_extra_args = \"\", -- Extra arguments to pass to the docker command\n },\n```\n\nThe RAG Service can currently configure the LLM and embedding models separately. In the `llm` and `embed` configuration blocks, you can set the following fields:\n\n- `provider`: Model provider (e.g., \"openai\", \"ollama\", \"dashscope\", and \"openrouter\")\n- `endpoint`: API endpoint\n- `api_key`: Environment variable name for the API key\n- `model`: Model name\n- `extra`: Additional configuration options\n\nFor detailed configuration of different model providers, you can check [here](./py/rag-service/README.md).\n\nAdditionally, RAG Service also depends on Docker! (For macOS users, OrbStack is recommended as a Docker alternative).\n\n`host_mount` is the path that will be mounted to the container, and the default is the home directory. The mount is required\nfor the RAG service to access the files in the host machine. It is up to the user to decide if you want to mount the whole\n`/` directory, just the project directory, or the home directory. If you plan using avante and RAG event for projects\nstored outside your home directory, you will need to set the `host_mount` to the root directory of your file system.\n\nThe mount will be read only.\n\nAfter changing the rag_service configuration, you need to manually delete the rag_service container to ensure the new configuration is used: `docker rm -fv avante-rag-service`\n\n## Web Search Engines\n\nAvante's tools include some web search engines, currently support:\n\n- [Tavily](https://tavily.com/)\n- [SerpApi - Search API](https://serpapi.com/)\n- Google's [Programmable Search Engine](https://developers.google.com/custom-search/v1/overview)\n- [Kagi](https://help.kagi.com/kagi/api/search.html)\n- [Brave Search](https://api-dashboard.search.brave.com/app/documentation/web-search/get-started)\n- [SearXNG](https://searxng.github.io/searxng/)\n\nThe default is Tavily, and can be changed through configuring `Config.web_search_engine.provider`:\n\n```lua\nweb_search_engine = {\n provider = \"tavily\", -- tavily, serpapi, google, kagi, brave, or searxng\n proxy = nil, -- proxy support, e.g., http://127.0.0.1:7890\n}\n```\n\nEnvironment variables required for providers:\n\n- Tavily: `TAVILY_API_KEY`\n- SerpApi: `SERPAPI_API_KEY`\n- Google:\n - `GOOGLE_SEARCH_API_KEY` as the [API key](https://developers.google.com/custom-search/v1/overview)\n - `GOOGLE_SEARCH_ENGINE_ID` as the [search engine](https://programmablesearchengine.google.com) ID\n- Kagi: `KAGI_API_KEY` as the [API Token](https://kagi.com/settings?p=api)\n- Brave Search: `BRAVE_API_KEY` as the [API key](https://api-dashboard.search.brave.com/app/keys)\n- SearXNG: `SEARXNG_API_URL` as the [API URL](https://docs.searxng.org/dev/search_api.html)\n\n## Disable Tools\n\nAvante enables tools by default, but some LLM models do not support tools. You can disable tools by setting `disable_tools = true` for the provider. For example:\n\n```lua\nproviders = {\n claude = {\n endpoint = \"https://api.anthropic.com\",\n model = \"claude-sonnet-4-20250514\",\n timeout = 30000, -- Timeout in milliseconds\n disable_tools = true, -- disable tools!\n extra_request_body = {\n temperature = 0,\n max_tokens = 4096,\n }\n }\n}\n```\n\nIn case you want to ban some tools to avoid its usage (like Claude 3.7 overusing the python tool) you can disable just specific tools\n\n```lua\n{\n disabled_tools = { \"python\" },\n}\n```\n\nTool list\n\n> rag_search, python, git_diff, git_commit, glob, search_keyword, read_file_toplevel_symbols,\n> read_file, create_file, move_path, copy_path, delete_path, create_dir, bash, web_search, fetch\n\n## Custom Tools\n\nAvante allows you to define custom tools that can be used by the AI during code generation and analysis. These tools can execute shell commands, run scripts, or perform any custom logic you need.\n\n### Example: Go Test Runner\n\n
\nHere's an example of a custom tool that runs Go unit tests:\n\n```lua\n{\n custom_tools = {\n {\n name = \"run_go_tests\", -- Unique name for the tool\n description = \"Run Go unit tests and return results\", -- Description shown to AI\n command = \"go test -v ./...\", -- Shell command to execute\n param = { -- Input parameters (optional)\n type = \"table\",\n fields = {\n {\n name = \"target\",\n description = \"Package or directory to test (e.g. './pkg/...' or './internal/pkg')\",\n type = \"string\",\n optional = true,\n },\n },\n },\n returns = { -- Expected return values\n {\n name = \"result\",\n description = \"Result of the fetch\",\n type = \"string\",\n },\n {\n name = \"error\",\n description = \"Error message if the fetch was not successful\",\n type = \"string\",\n optional = true,\n },\n },\n func = function(params, on_log, on_complete) -- Custom function to execute\n local target = params.target or \"./...\"\n return vim.system({ \"go\", \"test\", \"-v\", target }, { text = true }):wait().stdout\n end,\n },\n },\n}\n```\n\n
\n\n## MCP\n\nNow you can integrate MCP functionality for Avante through `mcphub.nvim`. For detailed documentation, please refer to [mcphub.nvim](https://ravitemer.github.io/mcphub.nvim/extensions/avante.html)\n\n## Custom prompts\n\nBy default, `avante.nvim` provides three different modes to interact with: `planning`, `editing`, and `suggesting`, followed with three different prompts per mode.\n\n- `planning`: Used with `require(\"avante\").toggle()` on sidebar\n- `editing`: Used with `require(\"avante\").edit()` on selection codeblock\n- `suggesting`: Used with `require(\"avante\").get_suggestion():suggest()` on Tab flow.\n- `cursor-planning`: Used with `require(\"avante\").toggle()` on Tab flow, but only when cursor planning mode is enabled.\n\nUsers can customize the system prompts via `Config.system_prompt` or `Config.override_prompt_dir`.\n\n`Config.system_prompt` allows you to set a global system prompt. We recommend calling this in a custom Autocmds depending on your need:\n\n```lua\nvim.api.nvim_create_autocmd(\"User\", {\n pattern = \"ToggleMyPrompt\",\n callback = function() require(\"avante.config\").override({system_prompt = \"MY CUSTOM SYSTEM PROMPT\"}) end,\n})\n\nvim.keymap.set(\"n\", \"am\", function() vim.api.nvim_exec_autocmds(\"User\", { pattern = \"ToggleMyPrompt\" }) end, { desc = \"avante: toggle my prompt\" })\n```\n\n`Config.override_prompt_dir` allows you to specify a directory containing your own custom prompt templates, which will override the built-in templates. This is useful if you want to maintain a set of custom prompts outside of your Neovim configuration. It can be a string representing the directory path, or a function that returns a string representing the directory path.\n\n```lua\n-- Example: Override with prompts from a specific directory\nrequire(\"avante\").setup({\n override_prompt_dir = vim.fn.expand(\"~/.config/nvim/avante_prompts\"),\n})\n\n-- Example: Override with prompts from a function (dynamic directory)\nrequire(\"avante\").setup({\n override_prompt_dir = function()\n -- Your logic to determine the prompt directory\n return vim.fn.expand(\"~/.config/nvim/my_dynamic_prompts\")\n end,\n})\n```\n\n> [!WARNING]\n>\n> If you customize `base.avanterules`, please ensure that `{% block custom_prompt %}{% endblock %}` and `{% block extra_prompt %}{% endblock %}` exist, otherwise the entire plugin may become unusable.\n> If you are unsure about the specific reasons or what you are doing, please do not override the built-in prompts. The built-in prompts work very well.\n\nIf you wish to custom prompts for each mode, `avante.nvim` will check for project root based on the given buffer whether it contains\nthe following patterns: `*.{mode}.avanterules`.\n\nThe rules for root hierarchy:\n\n- lsp workspace folders\n- lsp root_dir\n- root pattern of filename of the current buffer\n- root pattern of cwd\n\nYou can also configure custom directories for your `avanterules` files using the `rules` option:\n\n```lua\nrequire('avante').setup({\n rules = {\n project_dir = '.avante/rules', -- relative to project root, can also be an absolute path\n global_dir = '~/.config/avante/rules', -- absolute path\n },\n})\n```\n\nThe loading priority is as follows:\n\n1. `rules.project_dir`\n2. `rules.global_dir`\n3. Project root\n\n
\n\n Example folder structure for custom prompt\n\nIf you have the following structure:\n\n```bash\n.\n├── .git/\n├── typescript.planning.avanterules\n├── snippets.editing.avanterules\n├── suggesting.avanterules\n└── src/\n\n```\n\n- `typescript.planning.avanterules` will be used for `planning` mode\n- `snippets.editing.avanterules` will be used for `editing` mode\n- `suggesting.avanterules` will be used for `suggesting` mode.\n\n
\n\n> [!important]\n>\n> `*.avanterules` is a jinja template file, in which will be rendered using [minijinja](https://github.com/mitsuhiko/minijinja). See [templates](https://github.com/yetone/avante.nvim/blob/main/lua/avante/templates) for example on how to extend current templates.\n\n## Integration\n\nAvante.nvim can be extended to work with other plugins by using its extension modules. Below is an example of integrating Avante with [`nvim-tree`](https://github.com/nvim-tree/nvim-tree.lua), allowing you to select or deselect files directly from the NvimTree UI:\n\n```lua\n{\n \"yetone/avante.nvim\",\n event = \"VeryLazy\",\n keys = {\n {\n \"a+\",\n function()\n local tree_ext = require(\"avante.extensions.nvim_tree\")\n tree_ext.add_file()\n end,\n desc = \"Select file in NvimTree\",\n ft = \"NvimTree\",\n },\n {\n \"a-\",\n function()\n local tree_ext = require(\"avante.extensions.nvim_tree\")\n tree_ext.remove_file()\n end,\n desc = \"Deselect file in NvimTree\",\n ft = \"NvimTree\",\n },\n },\n opts = {\n --- other configurations\n selector = {\n exclude_auto_select = { \"NvimTree\" },\n },\n },\n}\n```\n\n## TODOs\n\n- [x] Chat with current file\n- [x] Apply diff patch\n- [x] Chat with the selected block\n- [x] Slash commands\n- [x] Edit the selected block\n- [x] Smart Tab (Cursor Flow)\n- [x] Chat with project (You can use `@codebase` to chat with the whole project)\n- [x] Chat with selected files\n- [x] Tool use\n- [x] MCP\n- [x] ACP\n- [ ] Better codebase indexing\n\n## Roadmap\n\n- **Enhanced AI Interactions**: Improve the depth of AI analysis and recommendations for more complex coding scenarios.\n- **LSP + Tree-sitter + LLM Integration**: Integrate with LSP and Tree-sitter and LLM to provide more accurate and powerful code suggestions and analysis.\n\n## FAQ\n\n### How to disable agentic mode?\n\nAvante.nvim provides two interaction modes:\n\n- **`agentic`** (default): Uses AI tools to automatically generate and apply code changes\n- **`legacy`**: Uses the traditional planning method without automatic tool execution\n\nTo disable agentic mode and switch to legacy mode, update your configuration:\n\n```lua\n{\n mode = \"legacy\", -- Switch from \"agentic\" to \"legacy\"\n -- ... your other configuration options\n}\n```\n\n**What's the difference?**\n\n- **Agentic mode**: AI can automatically execute tools like file operations, bash commands, web searches, etc. to complete complex tasks\n- **Legacy mode**: AI provides suggestions and plans but requires manual approval for all actions\n\n**When should you use legacy mode?**\n\n- If you prefer more control over what actions the AI takes\n- If you're concerned about security with automatic tool execution\n- If you want to manually review each step before applying changes\n- If you're working in a sensitive environment where automatic code changes aren't desired\n\nYou can also disable specific tools while keeping agentic mode enabled by configuring `disabled_tools`:\n\n```lua\n{\n mode = \"agentic\",\n disabled_tools = { \"bash\", \"python\" }, -- Disable specific tools\n -- ... your other configuration options\n}\n```\n\n## Contributing\n\nContributions to avante.nvim are welcome! If you're interested in helping out, please feel free to submit pull requests or open issues. Before contributing, ensure that your code has been thoroughly tested.\n\nSee [wiki](https://github.com/yetone/avante.nvim/wiki) for more recipes and tricks.\n\n## Acknowledgments\n\nWe would like to express our heartfelt gratitude to the contributors of the following open-source projects, whose code has provided invaluable inspiration and reference for the development of avante.nvim:\n\n| Nvim Plugin | License | Functionality | Location |\n| --------------------------------------------------------------------- | ------------------ | ----------------------------- | -------------------------------------------------------------------------------------------------------------------------------------- |\n| [git-conflict.nvim](https://github.com/akinsho/git-conflict.nvim) | No License | Diff comparison functionality | [lua/avante/diff.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/diff.lua) |\n| [ChatGPT.nvim](https://github.com/jackMort/ChatGPT.nvim) | Apache 2.0 License | Calculation of tokens count | [lua/avante/utils/tokens.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/utils/tokens.lua) |\n| [img-clip.nvim](https://github.com/HakonHarnes/img-clip.nvim) | MIT License | Clipboard image support | [lua/avante/clipboard.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/clipboard.lua) |\n| [copilot.lua](https://github.com/zbirenbaum/copilot.lua) | MIT License | Copilot support | [lua/avante/providers/copilot.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/providers/copilot.lua) |\n| [jinja.vim](https://github.com/HiPhish/jinja.vim) | MIT License | Template filetype support | [syntax/jinja.vim](https://github.com/yetone/avante.nvim/blob/main/syntax/jinja.vim) |\n| [codecompanion.nvim](https://github.com/olimorris/codecompanion.nvim) | MIT License | Secrets logic support | [lua/avante/providers/init.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/providers/init.lua) |\n| [aider](https://github.com/paul-gauthier/aider) | Apache 2.0 License | Planning mode user prompt | [lua/avante/templates/planning.avanterules](https://github.com/yetone/avante.nvim/blob/main/lua/avante/templates/planning.avanterules) |\n\nThe high quality and ingenuity of these projects' source code have been immensely beneficial throughout our development process. We extend our sincere thanks and respect to the authors and contributors of these projects. It is the selfless dedication of the open-source community that drives projects like avante.nvim forward.\n\n## Business Sponsors\n\n\n \n \n \n \n
\n \n \"Meshy
\n Meshy AI\n
 
\n
The #1 AI 3D Model Generator for Creators
\n \n 		\n \n \"BabelTower
\n BabelTower API\n
 
\n
No account needed, use any model instantly
\n \n
\n\n## License\n\navante.nvim is licensed under the Apache 2.0 License. For more details, please refer to the [LICENSE](./LICENSE) file.\n\n# Star History\n\n

\n \n \n \n \"NebulaGraph\n \n \n

\n" }, { "path": "README_zh.md", "content": "
\n \"logo\"\n

avante.nvim

\n
\n\n

\n \"Neovim:\n \"Lua\n \"Rust\n \"pre-commit\n \"Discord\"\n \n

\n\n**avante.nvim** 是一个 Neovim 插件，旨在模拟 [Cursor](https://www.cursor.com) AI IDE 的行为。它为用户提供 AI 驱动的代码建议，并能够轻松地将这些建议直接应用到源文件中。\n\n[View in English](README.md)\n\n> [!NOTE]\n>\n> 🥰 该项目正在快速迭代中，许多令人兴奋的功能将陆续添加。敬请期待！\n\n\n\n\n\n## 赞助 ❤️\n\n如果您喜欢这个项目，请考虑在 Patreon 上支持我，因为这有助于我继续维护和改进它：\n\n[赞助我](https://patreon.com/yetone)\n\n## 功能\n\n- **AI 驱动的代码辅助**：与 AI 互动，询问有关当前代码文件的问题，并接收智能建议以进行改进或修改。\n- **一键应用**：通过单个命令快速将 AI 的建议更改应用到源代码中，简化编辑过程并节省时间。\n\n## 安装\n\n如果您希望从源代码构建二进制文件，则需要 `cargo`。否则，将使用 `curl` 和 `tar` 从 GitHub 获取预构建的二进制文件。\n\n
\n\n lazy.nvim (推荐)\n\n```lua\n{\n \"yetone/avante.nvim\",\n -- 如果您想从源代码构建，请执行 `make BUILD_FROM_SOURCE=true`\n -- ⚠️ 一定要加上这一行配置！！！！！\n build = vim.fn.has(\"win32\") ~= 0\n and \"powershell -ExecutionPolicy Bypass -File Build.ps1 -BuildFromSource false\"\n or \"make\",\n event = \"VeryLazy\",\n version = false, -- 永远不要将此值设置为 \"*\"！永远不要！\n ---@module 'avante'\n ---@type avante.Config\n opts = {\n -- 在此处添加任何选项\n -- 例如\n provider = \"claude\",\n providers = {\n claude = {\n endpoint = \"https://api.anthropic.com\",\n model = \"claude-sonnet-4-20250514\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 20480,\n },\n },\n moonshot = {\n endpoint = \"https://api.moonshot.ai/v1\",\n model = \"kimi-k2-0711-preview\",\n timeout = 30000, -- 超时时间（毫秒）\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 32768,\n },\n },\n },\n },\n dependencies = {\n \"nvim-lua/plenary.nvim\",\n \"MunifTanjim/nui.nvim\",\n --- 以下依赖项是可选的，\n \"echasnovski/mini.pick\", -- 用于文件选择器提供者 mini.pick\n \"nvim-telescope/telescope.nvim\", -- 用于文件选择器提供者 telescope\n \"hrsh7th/nvim-cmp\", -- avante 命令和提及的自动完成\n \"ibhagwan/fzf-lua\", -- 用于文件选择器提供者 fzf\n \"nvim-tree/nvim-web-devicons\", -- 或 echasnovski/mini.icons\n \"zbirenbaum/copilot.lua\", -- 用于 providers='copilot'\n {\n -- 支持图像粘贴\n \"HakonHarnes/img-clip.nvim\",\n event = \"VeryLazy\",\n opts = {\n -- 推荐设置\n default = {\n embed_image_as_base64 = false,\n prompt_for_file_name = false,\n drag_and_drop = {\n insert_mode = true,\n },\n -- Windows 用户必需\n use_absolute_path = true,\n },\n },\n },\n {\n -- 如果您有 lazy=true，请确保正确设置\n 'MeanderingProgrammer/render-markdown.nvim',\n opts = {\n file_types = { \"markdown\", \"Avante\" },\n },\n ft = { \"markdown\", \"Avante\" },\n },\n },\n}\n```\n\n
\n\n
\n\n vim-plug\n\n```vim\n\n\" 依赖项\nPlug 'nvim-lua/plenary.nvim'\nPlug 'MunifTanjim/nui.nvim'\nPlug 'MeanderingProgrammer/render-markdown.nvim'\n\n\" 可选依赖项\nPlug 'hrsh7th/nvim-cmp'\nPlug 'nvim-tree/nvim-web-devicons' \"或 Plug 'echasnovski/mini.icons'\nPlug 'HakonHarnes/img-clip.nvim'\nPlug 'zbirenbaum/copilot.lua'\n\n\" Yay，如果您想从源代码构建，请传递 source=true\nPlug 'yetone/avante.nvim', { 'branch': 'main', 'do': 'make' }\nautocmd! User avante.nvim lua << EOF\nrequire('avante').setup()\nEOF\n```\n\n
\n\n
\n\n mini.deps\n\n```lua\nlocal add, later, now = MiniDeps.add, MiniDeps.later, MiniDeps.now\n\nadd({\n source = 'yetone/avante.nvim',\n monitor = 'main',\n depends = {\n 'nvim-lua/plenary.nvim',\n 'MunifTanjim/nui.nvim',\n 'echasnovski/mini.icons'\n },\n hooks = { post_checkout = function() vim.cmd('make') end }\n})\n--- 可选\nadd({ source = 'hrsh7th/nvim-cmp' })\nadd({ source = 'zbirenbaum/copilot.lua' })\nadd({ source = 'HakonHarnes/img-clip.nvim' })\nadd({ source = 'MeanderingProgrammer/render-markdown.nvim' })\n\nlater(function() require('render-markdown').setup({...}) end)\nlater(function()\n require('img-clip').setup({...}) -- 配置 img-clip\n require(\"copilot\").setup({...}) -- 根据您的喜好设置 copilot\n require(\"avante\").setup({...}) -- 配置 avante.nvim\nend)\n```\n\n
\n\n
\n\n Packer\n\n```vim\n\n -- 必需插件\n use 'nvim-lua/plenary.nvim'\n use 'MunifTanjim/nui.nvim'\n use 'MeanderingProgrammer/render-markdown.nvim'\n\n -- 可选依赖项\n use 'hrsh7th/nvim-cmp'\n use 'nvim-tree/nvim-web-devicons' -- 或使用 'echasnovski/mini.icons'\n use 'HakonHarnes/img-clip.nvim'\n use 'zbirenbaum/copilot.lua'\n\n -- Avante.nvim 带有构建过程\n use {\n 'yetone/avante.nvim',\n branch = 'main',\n run = 'make',\n config = function()\n require('avante').setup()\n end\n }\n```\n\n
\n\n
\n\n Home Manager\n\n```nix\nprograms.neovim = {\n plugins = [\n {\n plugin = pkgs.vimPlugins.avante-nvim;\n type = \"lua\";\n config = ''\n require(\"avante_lib\").load()\n require(\"avante\").setup()\n '' # 或 builtins.readFile ./plugins/avante.lua;\n }\n ];\n};\n```\n\n
\n\n
\n\n Nixvim\n\n```nix\n plugins.avante.enable = true;\n plugins.avante.settings = {\n # 在此处设置选项\n };\n```\n\n
\n\n
\n\n Lua\n\n```lua\n-- 依赖项：\nrequire('cmp').setup ({\n -- 使用上面的推荐设置\n})\nrequire('img-clip').setup ({\n -- 使用上面的推荐设置\n})\nrequire('copilot').setup ({\n -- 使用上面的推荐设置\n})\nrequire('render-markdown').setup ({\n -- 使用上面的推荐设置\n})\nrequire('avante').setup ({\n -- 在此处配置！\n})\n```\n\n
\n\n> [!IMPORTANT]\n>\n> `avante.nvim` 目前仅兼容 Neovim 0.10.1 或更高版本。请确保您的 Neovim 版本符合这些要求后再继续。\n\n> [!NOTE]\n>\n> 在同步加载插件时，我们建议在您的配色方案之后的某个时间 `require` 它。\n\n> [!NOTE]\n>\n> 推荐的 **Neovim** 选项：\n>\n> ```lua\n> -- 视图只能通过全局状态栏完全折叠\n> vim.opt.laststatus = 3\n> ```\n\n> [!TIP]\n>\n> 任何支持 markdown 的渲染插件都可以与 Avante 一起使用，只要您添加支持的文件类型 `Avante`。有关更多信息，请参见 和 [此评论](https://github.com/yetone/avante.nvim/issues/175#issuecomment-2313749363)。\n\n### 默认设置配置\n\n_请参见 [config.lua#L9](./lua/avante/config.lua) 以获取完整配置_\n\n
\n默认配置\n\n```lua\n{\n ---@alias Provider \"claude\" | \"openai\" | \"azure\" | \"gemini\" | \"cohere\" | \"copilot\" | string\n provider = \"claude\", -- 在 Aider 模式或 Cursor 规划模式的规划阶段使用的提供者\n -- 警告：由于自动建议是高频操作，因此成本较高，\n -- 目前将其指定为 `copilot` 提供者是危险的，因为：https://github.com/yetone/avante.nvim/issues/1048\n -- 当然，您可以通过增加 `suggestion.debounce` 来减少请求频率。\n auto_suggestions_provider = \"claude\",\n providers = {\n claude = {\n endpoint = \"https://api.anthropic.com\",\n model = \"claude-3-5-sonnet-20241022\",\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 4096,\n },\n },\n moonshot = {\n endpoint = \"https://api.moonshot.ai/v1\",\n model = \"kimi-k2-0711-preview\",\n timeout = 30000, -- 超时时间（毫秒）\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 32768,\n },\n },\n },\n ---指定特殊的 dual_boost 模式\n ---1. enabled: 是否启用 dual_boost 模式。默认为 false。\n ---2. first_provider: 第一个提供者用于生成响应。默认为 \"openai\"。\n ---3. second_provider: 第二个提供者用于生成响应。默认为 \"claude\"。\n ---4. prompt: 用于根据两个参考输出生成响应的提示。\n ---5. timeout: 超时时间（毫秒）。默认为 60000。\n ---工作原理：\n --- 启用 dual_boost 后，avante 将分别从 first_provider 和 second_provider 生成两个响应。然后使用 first_provider 的响应作为 provider1_output，second_provider 的响应作为 provider2_output。最后，avante 将根据提示和两个参考输出生成响应，默认提供者与正常情况相同。\n ---注意：这是一个实验性功能，可能无法按预期工作。\n dual_boost = {\n enabled = false,\n first_provider = \"openai\",\n second_provider = \"claude\",\n prompt = \"根据以下两个参考输出，生成一个结合两者元素但反映您自己判断和独特视角的响应。不要提供任何解释，只需直接给出响应。参考输出 1: [{{provider1_output}}], 参考输出 2: [{{provider2_output}}]\",\n timeout = 60000, -- 超时时间（毫秒）\n },\n behaviour = {\n auto_suggestions = false, -- 实验阶段\n auto_set_highlight_group = true,\n auto_set_keymaps = true,\n auto_apply_diff_after_generation = false,\n support_paste_from_clipboard = false,\n minimize_diff = true, -- 是否在应用代码块时删除未更改的行\n enable_token_counting = true, -- 是否启用令牌计数。默认为 true。\n auto_add_current_file = true, -- 打开新聊天时是否自动添加当前文件。默认为 true。\n enable_cursor_planning_mode = false, -- 是否启用 Cursor 规划模式。默认为 false。\n enable_claude_text_editor_tool_mode = false, -- 是否启用 Claude 文本编辑器工具模式。\n ---@type \"popup\" | \"inline_buttons\"\n confirmation_ui_style = \"inline_buttons\",\n },\n mappings = {\n --- @class AvanteConflictMappings\n diff = {\n ours = \"co\",\n theirs = \"ct\",\n all_theirs = \"ca\",\n both = \"cb\",\n cursor = \"cc\",\n next = \"]x\",\n prev = \"[x\",\n },\n suggestion = {\n accept = \"\",\n next = \"\",\n prev = \"\",\n dismiss = \"\",\n },\n jump = {\n next = \"]]\",\n prev = \"[[\",\n },\n submit = {\n normal = \"\",\n insert = \"\",\n },\n cancel = {\n normal = { \"\", \"\", \"q\" },\n insert = { \"\" },\n },\n sidebar = {\n apply_all = \"A\",\n apply_cursor = \"a\",\n retry_user_request = \"r\",\n edit_user_request = \"e\",\n switch_windows = \"\",\n reverse_switch_windows = \"\",\n remove_file = \"d\",\n add_file = \"@\",\n close = { \"\", \"q\" },\n close_from_input = nil, -- 例如，{ normal = \"\", insert = \"\" }\n },\n },\n selection = {\n enabled = true,\n hint_display = \"delayed\",\n },\n windows = {\n ---@type \"right\" | \"left\" | \"top\" | \"bottom\"\n position = \"right\", -- 侧边栏的位置\n wrap = true, -- 类似于 vim.o.wrap\n width = 30, -- 默认基于可用宽度的百分比\n sidebar_header = {\n enabled = true, -- true, false 启用/禁用标题\n align = \"center\", -- left, center, right 用于标题\n rounded = true,\n },\n spinner = {\n editing = { \"⡀\", \"⠄\", \"⠂\", \"⠁\", \"⠈\", \"⠐\", \"⠠\", \"⢀\", \"⣀\", \"⢄\", \"⢂\", \"⢁\", \"⢈\", \"⢐\", \"⢠\", \"⣠\", \"⢤\", \"⢢\", \"⢡\", \"⢨\", \"⢰\", \"⣰\", \"⢴\", \"⢲\", \"⢱\", \"⢸\", \"⣸\", \"⢼\", \"⢺\", \"⢹\", \"⣹\", \"⢽\", \"⢻\", \"⣻\", \"⢿\", \"⣿\" },\n generating = { \"·\", \"✢\", \"✳\", \"∗\", \"✻\", \"✽\" }, -- '生成中' 状态的旋转字符\n thinking = { \"🤯\", \"🙄\" }, -- '思考中' 状态的旋转字符\n },\n input = {\n prefix = \"> \",\n height = 8, -- 垂直布局中输入窗口的高度\n },\n edit = {\n border = \"rounded\",\n start_insert = true, -- 打开编辑窗口时开始插入模式\n },\n ask = {\n floating = false, -- 在浮动窗口中打开 'AvanteAsk' 提示\n start_insert = true, -- 打开询问窗口时开始插入模式\n border = \"rounded\",\n ---@type \"ours\" | \"theirs\"\n focus_on_apply = \"ours\", -- 应用后聚焦的差异\n },\n },\n highlights = {\n ---@type AvanteConflictHighlights\n diff = {\n current = \"DiffText\",\n incoming = \"DiffAdd\",\n },\n },\n --- @class AvanteConflictUserConfig\n diff = {\n autojump = true,\n ---@type string | fun(): any\n list_opener = \"copen\",\n --- 覆盖悬停在差异上时的 'timeoutlen' 设置（请参阅 :help timeoutlen）。\n --- 有助于避免进入以 `c` 开头的差异映射的操作员挂起模式。\n --- 通过设置为 -1 禁用。\n override_timeoutlen = 500,\n },\n suggestion = {\n debounce = 600,\n throttle = 600,\n },\n}\n```\n\n
\n\n## Blink.cmp 用户\n\n对于 blink cmp 用户（nvim-cmp 替代品），请查看以下配置说明\n这是通过使用 blink.compat 模拟 nvim-cmp 实现的\n或者您可以使用 [Kaiser-Yang/blink-cmp-avante](https://github.com/Kaiser-Yang/blink-cmp-avante)。\n\n
\n Lua\n\n```lua\n selector = {\n --- @alias avante.SelectorProvider \"native\" | \"fzf_lua\" | \"mini_pick\" | \"snacks\" | \"telescope\" | fun(selector: avante.ui.Selector): nil\n provider = \"fzf\",\n -- 自定义提供者的选项覆盖\n provider_opts = {},\n }\n```\n\n要创建自定义选择器，您可以指定一个自定义函数来启动选择器以选择项目，并将选定的项目传递给 `on_select` 回调。\n\n```lua\n selector = {\n ---@param selector avante.ui.Selector\n provider = function(selector)\n local items = selector.items ---@type avante.ui.SelectorItem[]\n local title = selector.title ---@type string\n local on_select = selector.on_select ---@type fun(selected_item_ids: string[]|nil): nil\n\n --- 在这里添加您的自定义选择器逻辑\n end,\n }\n```\n\n选择 native 以外的选择器，默认情况下目前存在问题\n对于 lazyvim 用户，请从网站复制 blink.cmp 的完整配置或扩展选项\n\n```lua\n compat = {\n \"avante_commands\",\n \"avante_mentions\",\n \"avante_files\",\n }\n```\n\n对于其他用户，只需添加自定义提供者\n\n### 可用的补全项\n\nAvante.nvim 提供了多个可以与 blink.cmp 集成的补全项：\n\n#### 提及功能 (`@` 触发器)\n提及功能允许您快速引用特定功能或将文件添加到聊天上下文：\n\n- `@codebase` - 启用项目上下文和仓库映射\n- `@diagnostics` - 启用诊断信息\n- `@file` - 打开文件选择器以将文件添加到聊天上下文\n- `@quickfix` - 将快速修复列表中的文件添加到聊天上下文\n- `@buffers` - 将打开的缓冲区添加到聊天上下文\n\n#### 斜杠命令 (`/` 触发器)\n内置斜杠命令用于常见操作：\n\n- `/help` - 显示可用命令的帮助信息\n- `/init` - 基于当前项目初始化 AGENTS.md\n- `/clear` - 清除聊天历史\n- `/new` - 开始新聊天\n- `/compact` - 压缩历史消息以节省令牌\n- `/lines - ` - 询问特定行的问题\n- `/commit` - 为更改生成提交消息\n\n#### 快捷方式 (`#` 触发器)\n快捷方式提供对预定义提示模板的快速访问。您可以在配置中自定义这些：\n\n```lua\n{\n shortcuts = {\n {\n name = \"refactor\",\n description = \"使用最佳实践重构代码\",\n details = \"自动重构代码以提高可读性、可维护性，并遵循最佳实践，同时保持功能不变\",\n prompt = \"请按照最佳实践重构此代码，提高可读性和可维护性，同时保持功能不变。\"\n },\n {\n name = \"test\",\n description = \"生成单元测试\",\n details = \"创建全面的单元测试，涵盖边界情况、错误场景和各种输入条件\",\n prompt = \"请为此代码生成全面的单元测试，涵盖边界情况和错误场景。\"\n },\n -- 添加更多自定义快捷方式...\n }\n}\n```\n\n当您在输入中键入 `#refactor` 时，它将自动替换为相应的提示文本。\n\n### 配置示例\n\n以下是包含所有 Avante 源的完整 blink.cmp 配置示例：\n\n```lua\n default = {\n ...\n \"avante_commands\",\n \"avante_mentions\",\n \"avante_shortcuts\",\n \"avante_files\",\n }\n```\n\n```lua\n providers = {\n avante_commands = {\n name = \"avante_commands\",\n module = \"blink.compat.source\",\n score_offset = 90, -- 显示优先级高于 lsp\n opts = {},\n },\n avante_files = {\n name = \"avante_files\",\n module = \"blink.compat.source\",\n score_offset = 100, -- 显示优先级高于 lsp\n opts = {},\n },\n avante_mentions = {\n name = \"avante_mentions\",\n module = \"blink.compat.source\",\n score_offset = 1000, -- 显示优先级高于 lsp\n opts = {},\n },\n avante_shortcuts = {\n name = \"avante_shortcuts\",\n module = \"blink.compat.source\",\n score_offset = 1000, -- 显示优先级高于 lsp\n opts = {},\n }\n ...\n }\n```\n\n
\n\n## 用法\n\n鉴于其早期阶段，`avante.nvim` 目前支持以下基本功能：\n\n> [!IMPORTANT]\n>\n> 为了在 neovim 会话之间保持一致性，建议在 shell 文件中设置环境变量。\n> 默认情况下，`Avante` 会在启动时提示您输入所选提供者的 API 密钥。\n>\n> **作用域 API 密钥（推荐用于隔离）**\n>\n> Avante 现在支持作用域 API 密钥，允许您专门为 Avante 隔离 API 密钥，而不影响其他应用程序。只需在任何 API 密钥前加上 `AVANTE_` 前缀：\n>\n> ```sh\n> # 作用域密钥（推荐）\n> export AVANTE_ANTHROPIC_API_KEY=your-claude-api-key\n> export AVANTE_OPENAI_API_KEY=your-openai-api-key\n> export AVANTE_AZURE_OPENAI_API_KEY=your-azure-api-key\n> export AVANTE_GEMINI_API_KEY=your-gemini-api-key\n> export AVANTE_CO_API_KEY=your-cohere-api-key\n> export AVANTE_AIHUBMIX_API_KEY=your-aihubmix-api-key\n> export AVANTE_MOONSHOT_API_KEY=your-moonshot-api-key\n> ```\n>\n> **全局 API 密钥（传统方式）**\n>\n> 如果您愿意，仍然可以使用传统的全局 API 密钥：\n>\n> 对于 Claude：\n>\n> ```sh\n> export ANTHROPIC_API_KEY=your-api-key\n> ```\n>\n> 对于 OpenAI：\n>\n> ```sh\n> export OPENAI_API_KEY=your-api-key\n> ```\n>\n> 对于 Azure OpenAI：\n>\n> ```sh\n> export AZURE_OPENAI_API_KEY=your-api-key\n> ```\n>\n> 对于 Amazon Bedrock：\n>\n> ```sh\n> export BEDROCK_KEYS=aws_access_key_id,aws_secret_access_key,aws_region[,aws_session_token]\n>\n> ```\n>\n> 注意：aws_session_token 是可选的，仅在使用临时 AWS 凭证时需要\n\n1. 在 Neovim 中打开代码文件。\n2. 使用 `:AvanteAsk` 命令查询 AI 关于代码的问题。\n3. 查看 AI 的建议。\n4. 通过简单的命令或按键绑定将推荐的更改直接应用到代码中。\n\n**注意**：该插件仍在积极开发中，其功能和界面可能会发生重大变化。随着项目的发展，预计会有一些粗糙的边缘和不稳定性。\n\n## 键绑定\n\n以下键绑定可用于 `avante.nvim`：\n\n| 键绑定 | 描述 |\n| ----------------------------------------- | ----------------------------- |\n| Leaderaa | 显示侧边栏 |\n| Leaderat | 切换侧边栏可见性 |\n| Leaderar | 刷新侧边栏 |\n| Leaderaf | 切换侧边栏焦点 |\n| Leadera? | 选择模型 |\n| Leaderae | 编辑选定的块 |\n| LeaderaS | 停止当前 AI 请求 |\n| co | 选择我们的 |\n| ct | 选择他们的 |\n| ca | 选择所有他们的 |\n| c0 | 选择无 |\n| cb | 选择两者 |\n| cc | 选择光标 |\n| ]x | 移动到上一个冲突 |\n| [x | 移动到下一个冲突 |\n| [[ | 跳转到上一个代码块 (结果窗口) |\n| ]] | 跳转到下一个代码块 (结果窗口) |\n\n> [!NOTE]\n>\n> 如果您使用 `lazy.nvim`，那么此处的所有键映射都将安全设置，这意味着如果 `aa` 已经绑定，则 avante.nvim 不会绑定此映射。\n> 在这种情况下，用户将负责设置自己的。有关更多详细信息，请参见 [关于键映射的说明](https://github.com/yetone/avante.nvim/wiki#keymaps-and-api-i-guess)。\n\n### Neotree 快捷方式\n\n在 neotree 侧边栏中，您还可以添加新的键盘快捷方式，以快速将 `file/folder` 添加到 `Avante Selected Files`。\n\n
\nNeotree 配置\n\n```lua\nreturn {\n {\n 'nvim-neo-tree/neo-tree.nvim',\n config = function()\n require('neo-tree').setup({\n filesystem = {\n commands = {\n avante_add_files = function(state)\n local node = state.tree:get_node()\n local filepath = node:get_id()\n local relative_path = require('avante.utils').relative_path(filepath)\n\n local sidebar = require('avante').get()\n\n local open = sidebar:is_open()\n -- 确保 avante 侧边栏已打开\n if not open then\n require('avante.api').ask()\n sidebar = require('avante').get()\n end\n\n sidebar.file_selector:add_selected_file(relative_path)\n\n -- 删除 neo tree 缓冲区\n if not open then\n sidebar.file_selector:remove_selected_file('neo-tree filesystem [1]')\n end\n end,\n },\n window = {\n mappings = {\n ['oa'] = 'avante_add_files',\n },\n },\n },\n })\n end,\n },\n}\n```\n\n
\n\n## 命令\n\n| 命令 | 描述 | 示例 |\n| ---------------------------------- | ---------------------------------------------------------------------------------------- | --------------------------------------------------- |\n| `:AvanteAsk [question] [position]` | 询问 AI 关于您的代码的问题。可选的 `position` 设置窗口位置和 `ask` 启用/禁用直接询问模式 | `:AvanteAsk position=right Refactor this code here` |\n| `:AvanteBuild` | 构建项目的依赖项 | |\n| `:AvanteChat` | 启动与 AI 的聊天会话，讨论您的代码库。默认情况下 `ask`=false | |\n| `:AvanteClear` | 清除聊天记录 | |\n| `:AvanteEdit` | 编辑选定的代码块 | |\n| `:AvanteFocus` | 切换焦点到/从侧边栏 | |\n| `:AvanteRefresh` | 刷新所有 Avante 窗口 | |\n| `:AvanteStop` | 停止当前 AI 请求 | |\n| `:AvanteSwitchProvider` | 切换 AI 提供者（例如 openai） | |\n| `:AvanteShowRepoMap` | 显示项目结构的 repo map | |\n| `:AvanteToggle` | 切换 Avante 侧边栏 | |\n| `:AvanteModels` | 显示模型列表 | |\n\n## 高亮组\n\n| 高亮组 | 描述 | 备注 |\n| --------------------------- | -------------------------- | ------------------------------------------ |\n| AvanteTitle | 标题 | |\n| AvanteReversedTitle | 用于圆角边框 | |\n| AvanteSubtitle | 选定代码标题 | |\n| AvanteReversedSubtitle | 用于圆角边框 | |\n| AvanteThirdTitle | 提示标题 | |\n| AvanteReversedThirdTitle | 用于圆角边框 | |\n| AvanteConflictCurrent | 当前冲突高亮 | 默认值为 `Config.highlights.diff.current` |\n| AvanteConflictIncoming | 即将到来的冲突高亮 | 默认值为 `Config.highlights.diff.incoming` |\n| AvanteConflictCurrentLabel | 当前冲突标签高亮 | 默认值为 `AvanteConflictCurrent` 的阴影 |\n| AvanteConflictIncomingLabel | 即将到来的冲突标签高亮 | 默认值为 `AvanteConflictIncoming` 的阴影 |\n| AvantePopupHint | 弹出菜单中的使用提示 | |\n| AvanteInlineHint | 在可视模式下显示的行尾提示 | |\n\n有关更多信息，请参见 [highlights.lua](./lua/avante/highlights.lua)\n\n## Ollama\n\nollama 是 avante.nvim 的一流提供者。要开始使用它，您需要在配置中设置 `provider = \"ollama\"`，并将 `ollama` 中的 `model` 字段设置为您想要使用的模型。Ollama 默认是禁用的，您需要为其 `is_env_set` 方法提供一个实现来正确地启用它。例如：\n\n```lua\nprovider = \"ollama\",\nproviders = {\n ollama = {\n model = \"qwq:32b\",\n is_env_set = require(\"avante.providers.ollama\").check_endpoint_alive,\n },\n}\n```\n\n## 自定义提供者\n\nAvante 提供了一组默认提供者，但用户也可以创建自己的提供者。\n\n有关更多信息，请参见 [自定义提供者](https://github.com/yetone/avante.nvim/wiki/Custom-providers)\n\n## Cursor 规划模式\n\n因为 avante.nvim 一直使用 Aider 的方法进行规划应用，但其提示对模型要求很高，需要像 claude-3.5-sonnet 或 gpt-4o 这样的模型才能正常工作。\n\n因此，我采用了 Cursor 的方法来实现规划应用。有关实现的详细信息，请参阅 [cursor-planning-mode.md](./cursor-planning-mode.md)\n\n## RAG 服务\n\nAvante 提供了一个 RAG 服务，这是一个用于获取 AI 生成代码所需上下文的工具。默认情况下，它未启用。您可以通过以下方式启用它：\n\n```lua\n rag_service = { -- RAG 服务配置\n enabled = false, -- 启用 RAG 服务\n host_mount = os.getenv(\"HOME\"), -- RAG 服务的主机挂载路径 (Docker 将挂载此路径)\n runner = \"docker\", -- RAG 服务的运行器 (可以使用 docker 或 nix)\n llm = { -- RAG 服务使用的语言模型 (LLM) 配置\n provider = \"openai\", -- LLM 提供者\n endpoint = \"https://api.openai.com/v1\", -- LLM API 端点\n api_key = \"OPENAI_API_KEY\", -- LLM API 密钥的环境变量名称\n model = \"gpt-4o-mini\", -- LLM 模型名称\n extra = nil, -- LLM 的额外配置选项\n },\n embed = { -- RAG 服务使用的嵌入模型配置\n provider = \"openai\", -- 嵌入提供者\n endpoint = \"https://api.openai.com/v1\", -- 嵌入 API 端点\n api_key = \"OPENAI_API_KEY\", -- 嵌入 API 密钥的环境变量名称\n model = \"text-embedding-3-large\", -- 嵌入模型名称\n extra = nil, -- 嵌入模型的额外配置选项\n },\n docker_extra_args = \"\", -- 传递给 docker 命令的额外参数\n },\n```\n\nRAG 服务可以单独设置llm模型和嵌入模型。在 `llm` 和 `embed` 配置块中，您可以设置以下字段：\n\n- `provider`: 模型提供者（例如 \"openai\", \"ollama\", \"dashscope\"以及\"openrouter\"）\n- `endpoint`: API 端点\n- `api_key`: API 密钥的环境变量名称\n- `model`: 模型名称\n- `extra`: 额外的配置选项\n\n有关不同模型提供商的详细配置，你可以在[这里](./py/rag-service/README.md)查看。\n\n此外，RAG 服务还依赖于 Docker！（对于 macOS 用户，推荐使用 OrbStack 作为 Docker 的替代品）。\n\n`host_mount` 是将挂载到容器的路径，默认是主目录。挂载是 RAG 服务访问主机机器中文件所必需的。用户可以决定是否要挂载整个 `/` 目录、仅项目目录或主目录。如果您计划使用 avante 和 RAG 事件处理存储在主目录之外的项目，您需要将 `host_mount` 设置为文件系统的根目录。\n\n挂载将是只读的。\n\n更改 rag_service 配置后，您需要手动删除 rag_service 容器以确保使用新配置：`docker rm -fv avante-rag-service`\n\n## Web 搜索引擎\n\nAvante 的工具包括一些 Web 搜索引擎，目前支持：\n\n- [Tavily](https://tavily.com/)\n- [SerpApi - Search API](https://serpapi.com/)\n- Google's [Programmable Search Engine](https://developers.google.com/custom-search/v1/overview)\n- [Kagi](https://help.kagi.com/kagi/api/search.html)\n- [Brave Search](https://api-dashboard.search.brave.com/app/documentation/web-search/get-started)\n- [SearXNG](https://searxng.github.io/searxng/)\n\n默认是 Tavily，可以通过配置 `Config.web_search_engine.provider` 进行更改：\n\n```lua\nweb_search_engine = {\n provider = \"tavily\", -- tavily, serpapi, google, kagi, brave 或 searxng\n proxy = nil, -- proxy support, e.g., http://127.0.0.1:7890\n}\n```\n\n提供者所需的环境变量：\n\n- Tavily: `TAVILY_API_KEY`\n- SerpApi: `SERPAPI_API_KEY`\n- Google:\n - `GOOGLE_SEARCH_API_KEY` 作为 [API 密钥](https://developers.google.com/custom-search/v1/overview)\n - `GOOGLE_SEARCH_ENGINE_ID` 作为 [搜索引擎](https://programmablesearchengine.google.com) ID\n- Kagi: `KAGI_API_KEY` 作为 [API 令牌](https://kagi.com/settings?p=api)\n- Brave Search: `BRAVE_API_KEY` 作为 [API 密钥](https://api-dashboard.search.brave.com/app/keys)\n- SearXNG: `SEARXNG_API_URL` 作为 [API URL](https://docs.searxng.org/dev/search_api.html)\n\n## 禁用工具\n\nAvante 默认启用工具，但某些 LLM 模型不支持工具。您可以通过为提供者设置 `disable_tools = true` 来禁用工具。例如：\n\n```lua\n{\n claude = {\n endpoint = \"https://api.anthropic.com\",\n model = \"claude-3-5-sonnet-20241022\",\n timeout = 30000, -- 超时时间（毫秒）\n temperature = 0,\n max_tokens = 4096,\n disable_tools = true, -- 禁用工具！\n },\n}\n```\n\n如果您想禁止某些工具以避免其使用（例如 Claude 3.7 过度使用 python 工具），您可以仅禁用特定工具\n\n```lua\n{\n disabled_tools = { \"python\" },\n}\n```\n\n工具列表\n\n> rag_search, python, git_diff, git_commit, glob, search_keyword, read_file_toplevel_symbols,\n> read_file, create_file, move_path, copy_path, delete_path, create_dir, bash, web_search, fetch\n\n## 自定义工具\n\nAvante 允许您定义自定义工具，AI 可以在代码生成和分析期间使用这些工具。这些工具可以执行 shell 命令、运行脚本或执行您需要的任何自定义逻辑。\n\n### 示例：Go 测试运行器\n\n
\n以下是一个运行 Go 单元测试的自定义工具示例：\n\n```lua\n{\n custom_tools = {\n {\n name = \"run_go_tests\", -- 工具的唯一名称\n description = \"运行 Go 单元测试并返回结果\", -- 显示给 AI 的描述\n command = \"go test -v ./...\", -- 要执行的 shell 命令\n param = { -- 输入参数（可选）\n type = \"table\",\n fields = {\n {\n name = \"target\",\n description = \"要测试的包或目录（例如 './pkg/...' 或 './internal/pkg'）\",\n type = \"string\",\n optional = true,\n },\n },\n },\n returns = { -- 预期返回值\n {\n name = \"result\",\n description = \"获取的结果\",\n type = \"string\",\n },\n {\n name = \"error\",\n description = \"如果获取不成功的错误消息\",\n type = \"string\",\n optional = true,\n },\n },\n func = function(params, on_log, on_complete) -- 要执行的自定义函数\n local target = params.target or \"./...\"\n return vim.system({ \"go\", \"test\", \"-v\", target }, { text = true }):wait().stdout\n end,\n },\n },\n}\n```\n\n
\n\n## MCP\n\n现在您可以通过 `mcphub.nvim` 为 Avante 集成 MCP 功能。有关详细文档，请参阅 [mcphub.nvim](https://ravitemer.github.io/mcphub.nvim/extensions/avante.html)\n\n## Claude 文本编辑器工具模式\n\nAvante 利用 [Claude 文本编辑器工具](https://docs.anthropic.com/en/docs/build-with-claude/tool-use/text-editor-tool) 提供更优雅的代码编辑体验。您现在可以通过在 `behaviour` 配置中将 `enable_claude_text_editor_tool_mode` 设置为 `true` 来启用此功能：\n\n```lua\n{\n behaviour = {\n enable_claude_text_editor_tool_mode = true,\n },\n}\n```\n\n> [!NOTE]\n> 要启用 **Claude 文本编辑器工具模式**，您必须使用 `claude-3-5-sonnet-*` 或 `claude-3-7-sonnet-*` 模型与 `claude` 提供者！此功能不支持任何其他模型！\n\n## 自定义提示\n\n默认情况下，`avante.nvim` 提供三种不同的模式进行交互：`planning`、`editing` 和 `suggesting`，每种模式都有三种不同的提示。\n\n- `planning`：与侧边栏上的 `require(\"avante\").toggle()` 一起使用\n- `editing`：与选定代码块上的 `require(\"avante\").edit()` 一起使用\n- `suggesting`：与 Tab 流上的 `require(\"avante\").get_suggestion():suggest()` 一起使用。\n- `cursor-planning`：与 Tab 流上的 `require(\"avante\").toggle()` 一起使用，但仅在启用 cursor 规划模式时。\n\n用户可以通过 `Config.system_prompt` 或 `Config.override_prompt_dir` 自定义系统提示。\n\n`Config.system_prompt` 允许您设置全局系统提示。我们建议根据您的需要在自定义 Autocmds 中调用此方法：\n\n```lua\nvim.api.nvim_create_autocmd(\"User\", {\n pattern = \"ToggleMyPrompt\",\n callback = function() require(\"avante.config\").override({system_prompt = \"MY CUSTOM SYSTEM PROMPT\"}) end,\n})\n\nvim.keymap.set(\"n\", \"am\", function() vim.api.nvim_exec_autocmds(\"User\", { pattern = \"ToggleMyPrompt\" }) end, { desc = \"avante: toggle my prompt\" })\n```\n\n`Config.override_prompt_dir` 允许您指定一个目录，其中包含您自己的自定义提示模板，这将覆盖内置模板。如果您想在 Neovim 配置之外维护一组自定义提示，这将非常有用。它可以是一个表示目录路径的字符串，也可以是一个返回表示目录路径的字符串的函数。\n\n```lua\n-- 示例：使用特定目录中的提示进行覆盖\nrequire(\"avante\").setup({\n override_prompt_dir = vim.fn.expand(\"~/.config/nvim/avante_prompts\"),\n})\n\n-- 示例：使用函数（动态目录）中的提示进行覆盖\nrequire(\"avante\").setup({\n override_prompt_dir = function()\n -- 确定提示目录的逻辑\n return vim.fn.expand(\"~/.config/nvim/my_dynamic_prompts\")\n end,\n})\n```\n\n> [!WARNING]\n>\n> 如果您自定 `base.avanterules`，请一定要确保 `{% block custom_prompt %}{% endblock %}` 和 `{% block extra_prompt %}{% endblock %}` 存在，否则可能会导致整个插件无法使用。\n> 如果您不清楚具体原因或者您不知道自己在干什么，请不要覆盖内置 prompt。内置 prompt 工作得非常好。\n\n如果希望为每种模式自定义提示，`avante.nvim` 将根据给定缓冲区的项目根目录检查是否包含以下模式：`*.{mode}.avanterules`。\n\n根目录层次结构的规则：\n\n- lsp 工作区文件夹\n- lsp root_dir\n- 当前缓冲区的文件名的根模式\n- cwd 的根模式\n\n您还可以使用 `rules` 选项为您的 `avanterules` 文件配置自定义目录：\n\n```lua\nrequire('avante').setup({\n rules = {\n project_dir = '.avante/rules', -- 相对于项目根目录，也可以是绝对路径\n global_dir = '~/.config/avante/rules', -- 绝对路径\n },\n})\n```\n\n加载优先级如下：\n\n1. `rules.project_dir`\n2. `rules.global_dir`\n3. 项目根目录\n\n
\n\n 自定义提示的示例文件夹结构\n\n如果您有以下结构：\n\n```bash\n.\n├── .git/\n├── typescript.planning.avanterules\n├── snippets.editing.avanterules\n├── suggesting.avanterules\n└── src/\n\n```\n\n- `typescript.planning.avanterules` 将用于 `planning` 模式\n- `snippets.editing.avanterules` 将用于 `editing` 模式\n- `suggesting.avanterules` 将用于 `suggesting` 模式。\n\n
\n\n> [!important]\n>\n> `*.avanterules` 是一个 jinja 模板文件，将使用 [minijinja](https://github.com/mitsuhiko/minijinja) 渲染。有关如何扩展当前模板的示例，请参见 [templates](https://github.com/yetone/avante.nvim/blob/main/lua/avante/templates)。\n\n## 集成\n\nAvante.nvim 可以通过其扩展模块与其他插件协同工作。下面是一个将 Avante 与 nvim-tree 集成的示例，允许你直接从 NvimTree UI 中选择或取消选择文件：\n\n```lua\n{\n \"yetone/avante.nvim\",\n event = \"VeryLazy\",\n keys = {\n {\n \"a+\",\n function()\n local tree_ext = require(\"avante.extensions.nvim_tree\")\n tree_ext.add_file()\n end,\n desc = \"Select file in NvimTree\",\n ft = \"NvimTree\",\n },\n {\n \"a-\",\n function()\n local tree_ext = require(\"avante.extensions.nvim_tree\")\n tree_ext.remove_file()\n end,\n desc = \"Deselect file in NvimTree\",\n ft = \"NvimTree\",\n },\n },\n opts = {\n --- 其他配置\n selector = {\n exclude_auto_select = { \"NvimTree\" },\n },\n },\n}\n```\n\n## TODOs\n\n- [x] 与当前文件聊天\n- [x] 应用差异补丁\n- [x] 与选定的块聊天\n- [x] 斜杠命令\n- [x] 编辑选定的块\n- [x] 智能 Tab（Cursor 流）\n- [x] 与项目聊天（您可以使用 `@codebase` 与整个项目聊天）\n- [x] 与选定文件聊天\n- [x] 工具使用\n- [x] MCP\n- [ ] 更好的代码库索引\n\n## 路线图\n\n- **增强的 AI 交互**：提高 AI 分析和建议的深度，以应对更复杂的编码场景。\n- **LSP + Tree-sitter + LLM 集成**：与 LSP 和 Tree-sitter 以及 LLM 集成，以提供更准确和强大的代码建议和分析。\n\n## 贡献\n\n欢迎为 avante.nvim 做出贡献！如果您有兴趣提供帮助，请随时提交拉取请求或打开问题。在贡献之前，请确保您的代码已经过彻底测试。\n\n有关更多配方和技巧，请参见 [wiki](https://github.com/yetone/avante.nvim/wiki)。\n\n## 致谢\n\n我们要向以下开源项目的贡献者表示衷心的感谢，他们的代码为 avante.nvim 的开发提供了宝贵的灵感和参考：\n\n| Nvim 插件 | 许可证 | 功能 | 位置 |\n| --------------------------------------------------------------------- | ----------------- | ---------------- | -------------------------------------------------------------------------------------------------------------------------------------- |\n| [git-conflict.nvim](https://github.com/akinsho/git-conflict.nvim) | 无许可证 | 差异比较功能 | [lua/avante/diff.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/diff.lua) |\n| [ChatGPT.nvim](https://github.com/jackMort/ChatGPT.nvim) | Apache 2.0 许可证 | 令牌计数的计算 | [lua/avante/utils/tokens.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/utils/tokens.lua) |\n| [img-clip.nvim](https://github.com/HakonHarnes/img-clip.nvim) | MIT 许可证 | 剪贴板图像支持 | [lua/avante/clipboard.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/clipboard.lua) |\n| [copilot.lua](https://github.com/zbirenbaum/copilot.lua) | MIT 许可证 | Copilot 支持 | [lua/avante/providers/copilot.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/providers/copilot.lua) |\n| [jinja.vim](https://github.com/HiPhish/jinja.vim) | MIT 许可证 | 模板文件类型支持 | [syntax/jinja.vim](https://github.com/yetone/avante.nvim/blob/main/syntax/jinja.vim) |\n| [codecompanion.nvim](https://github.com/olimorris/codecompanion.nvim) | MIT 许可证 | Secrets 逻辑支持 | [lua/avante/providers/init.lua](https://github.com/yetone/avante.nvim/blob/main/lua/avante/providers/init.lua) |\n| [aider](https://github.com/paul-gauthier/aider) | Apache 2.0 许可证 | 规划模式用户提示 | [lua/avante/templates/planning.avanterules](https://github.com/yetone/avante.nvim/blob/main/lua/avante/templates/planning.avanterules) |\n\n这些项目的源代码的高质量和独创性在我们的开发过程中提供了极大的帮助。我们向这些项目的作者和贡献者表示诚挚的感谢和敬意。正是开源社区的无私奉献推动了像 avante.nvim 这样的项目向前发展。\n\n## 商业赞助商\n\n\n \n \n \n \n
\n \n \"Meshy
\n Meshy AI\n
 
\n
为创作者提供的 #1 AI 3D 模型生成器
\n \n 		\n \n \"BabelTower
\n BabelTower API\n
 
\n
无需帐户，立即使用任何模型
\n \n
\n\n## 许可证\n\navante.nvim 根据 Apache 2.0 许可证授权。有关更多详细信息，请参阅 [LICENSE](./LICENSE) 文件。\n\n# Star 历史\n\n

\n \n \n \n \"NebulaGraph\n \n \n

\n" }, { "path": "autoload/avante.vim", "content": "function avante#build(...) abort\n let l:source = get(a:, 1, v:false)\n return join([luaeval(\"require('avante_lib').load()\") ,luaeval(\"require('avante.api').build(_A)\", l:source)], \"\\n\")\nendfunction\n" }, { "path": "build.sh", "content": "#!/usr/bin/env bash\n\nset -e\n\nREPO_OWNER=\"yetone\"\nREPO_NAME=\"avante.nvim\"\n\nSCRIPT_DIR=\"$(cd -- \"$(dirname -- \"${BASH_SOURCE[0]}\")\" &>/dev/null && pwd)\"\n\n# Set the target directory to clone the artifact\nTARGET_DIR=\"${SCRIPT_DIR}/build\"\n\n# Get the artifact download URL based on the platform and Lua version\ncase \"$(uname -s)\" in\nLinux*)\n PLATFORM=\"linux\"\n LIB_EXT=\"so\"\n ;;\nDarwin*)\n PLATFORM=\"darwin\"\n LIB_EXT=\"dylib\"\n ;;\nCYGWIN* | MINGW* | MSYS*)\n PLATFORM=\"windows\"\n LIB_EXT=\"dll\"\n ;;\n*)\n echo \"Unsupported platform\"\n exit 1\n ;;\nesac\n\n# Get the architecture (x86_64 or aarch64)\ncase \"$(uname -m)\" in\nx86_64)\n ARCH=\"x86_64\"\n ;;\naarch64)\n ARCH=\"aarch64\"\n ;;\narm64)\n ARCH=\"aarch64\"\n ;;\n*)\n echo \"Unsupported architecture\"\n exit 1\n ;;\nesac\n\n# Set the Lua version (lua54 or luajit)\nLUA_VERSION=\"${LUA_VERSION:-luajit}\"\n\n# Set the artifact name pattern\nARTIFACT_NAME_PATTERN=\"avante_lib-$PLATFORM-$ARCH-$LUA_VERSION\"\n\ntest_command() {\n command -v \"$1\" >/dev/null 2>&1\n}\n\ntest_gh_auth() {\n if gh api user >/dev/null 2>&1; then\n return 0\n else\n return 1\n fi\n}\n\nfetch_remote_tags() {\n git ls-remote --tags origin | cut -f2 | sed 's|refs/tags/||' | while read tag; do\n if ! git rev-parse \"$tag\" >/dev/null 2>&1; then\n git fetch origin \"refs/tags/$tag:refs/tags/$tag\"\n fi\n done\n}\n\nif [ ! -d \"$TARGET_DIR\" ]; then\n mkdir -p \"$TARGET_DIR\"\nfi\n\nfetch_remote_tags\nlatest_tag=\"$(git describe --tags --abbrev=0 || true)\" # will be empty in clone repos\nbuilt_tag=\"$(cat build/.tag 2>/dev/null || true)\"\n\nsave_tag() {\n echo \"$latest_tag\" > build/.tag\n}\n\nif [[ \"$latest_tag\" = \"$built_tag\" && -n \"$latest_tag\" ]]; then\n echo \"Local build is up to date $latest_tag. No download needed.\"\nelif [[ \"$latest_tag\" != \"$built_tag\" && -n \"$latest_tag\" ]]; then\n echo \"Local build is out of date $built_tag. Downloading latest $latest_tag.\"\n if test_command \"gh\" && test_gh_auth; then\n gh release download \"$latest_tag\" --repo \"github.com/$REPO_OWNER/$REPO_NAME\" --pattern \"*$ARTIFACT_NAME_PATTERN*\" --clobber --output - | tar -zxv -C \"$TARGET_DIR\"\n save_tag\n else\n # Get the artifact download URL\n ARTIFACT_URL=$(curl -s \"https://api.github.com/repos/$REPO_OWNER/$REPO_NAME/releases/tags/$latest_tag\" | grep \"browser_download_url\" | cut -d '\"' -f 4 | grep $ARTIFACT_NAME_PATTERN)\n\n set -x\n\n mkdir -p \"$TARGET_DIR\"\n\n curl -L \"$ARTIFACT_URL\" | tar -zxv -C \"$TARGET_DIR\"\n save_tag\n fi\nelse\n echo \"No latest tag found. Building from source.\"\n cargo build --release --features=$LUA_VERSION\n for f in target/release/lib*.$LIB_EXT; do\n cp \"$f\" \"build/$(echo $f | sed 's#.*/lib##')\"\n done\nfi\n" }, { "path": "crates/avante-html2md/Cargo.toml", "content": "[lib]\ncrate-type = [\"cdylib\"]\n\n[package]\nname = \"avante-html2md\"\nedition.workspace = true\nrust-version.workspace = true\nlicense.workspace = true\nversion.workspace = true\n\n[dependencies]\nhtmd = \"0.1.6\"\n#html2md = \"0.2.15\"\nhtml2md = { git = \"https://gitlab.com/Kanedias/html2md.git\", rev = \"850ccf756a87fedebcea707c5c981c3103019238\" }\nmlua.workspace = true\nreqwest = { version = \"0.12.12\", features = [\"blocking\", \"native-tls-vendored\"] }\n\n[lints]\nworkspace = true\n\n[features]\nlua51 = [\"mlua/lua51\"]\nlua52 = [\"mlua/lua52\"]\nlua53 = [\"mlua/lua53\"]\nlua54 = [\"mlua/lua54\"]\nluajit = [\"mlua/luajit\"]\n" }, { "path": "crates/avante-html2md/src/lib.rs", "content": "use htmd::HtmlToMarkdown;\nuse mlua::prelude::*;\nuse std::error::Error;\n\n#[derive(Debug)]\nenum MyError {\n HtmlToMd(String),\n Request(String),\n}\n\nimpl std::fmt::Display for MyError {\n fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n match self {\n MyError::HtmlToMd(e) => write!(f, \"HTML to Markdown error: {e}\"),\n MyError::Request(e) => write!(f, \"Request error: {e}\"),\n }\n }\n}\n\nimpl Error for MyError {}\n\nfn do_html2md(html: &str) -> Result {\n let converter = HtmlToMarkdown::builder()\n .skip_tags(vec![\"script\", \"style\", \"header\", \"footer\"])\n .build();\n let md = converter\n .convert(html)\n .map_err(|e| MyError::HtmlToMd(e.to_string()))?;\n Ok(md)\n}\n\nfn do_fetch_md(url: &str) -> Result {\n let mut headers = reqwest::header::HeaderMap::new();\n headers.insert(\n reqwest::header::USER_AGENT,\n reqwest::header::HeaderValue::from_static(\"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36\"),\n );\n let client = reqwest::blocking::Client::builder()\n .default_headers(headers)\n .build()\n .map_err(|e| MyError::Request(e.to_string()))?;\n let response = client\n .get(url)\n .send()\n .map_err(|e| MyError::Request(e.to_string()))?;\n let body = response\n .text()\n .map_err(|e| MyError::Request(e.to_string()))?;\n let html = body.trim().to_string();\n let md = do_html2md(&html)?;\n Ok(md)\n}\n\n#[mlua::lua_module]\nfn avante_html2md(lua: &Lua) -> LuaResult {\n let exports = lua.create_table()?;\n exports.set(\n \"fetch_md\",\n lua.create_function(move |_, url: String| -> LuaResult {\n do_fetch_md(&url).map_err(|e| mlua::Error::RuntimeError(e.to_string()))\n })?,\n )?;\n exports.set(\n \"html2md\",\n lua.create_function(move |_, html: String| -> LuaResult {\n do_html2md(&html).map_err(|e| mlua::Error::RuntimeError(e.to_string()))\n })?,\n )?;\n Ok(exports)\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_fetch_md() {\n let md = do_fetch_md(\"https://github.com/yetone/avante.nvim\").unwrap();\n println!(\"{md}\");\n }\n}\n" }, { "path": "crates/avante-repo-map/Cargo.toml", "content": "[lib]\ncrate-type = [\"cdylib\"]\n\n[package]\nname = \"avante-repo-map\"\nedition.workspace = true\nrust-version.workspace = true\nlicense.workspace = true\nversion.workspace = true\n\n[build-dependencies]\ncc=\"*\"\n\n[dependencies]\nmlua = { workspace = true }\nminijinja = { workspace = true }\nserde = { workspace = true, features = [\"derive\"] }\ntree-sitter = \"0.23\"\ntree-sitter-language = \"0.1\"\ntree-sitter-rust = \"0.23\"\ntree-sitter-php = \"0.23.11\"\ntree-sitter-python = \"0.23\"\ntree-sitter-java = \"0.23.5\"\ntree-sitter-javascript = \"0.23\"\ntree-sitter-typescript = \"0.23\"\ntree-sitter-go = \"0.23\"\ntree-sitter-c = \"0.23\"\ntree-sitter-cpp = \"0.23\"\ntree-sitter-lua = \"0.2\"\ntree-sitter-ruby = \"0.23\"\ntree-sitter-zig = \"1.0.2\"\ntree-sitter-scala = \"0.23\"\ntree-sitter-swift = \"0.7.0\"\ntree-sitter-elixir = \"0.3.1\"\ntree-sitter-c-sharp = \"0.23\"\n\n[lints]\nworkspace = true\n\n[features]\nlua51 = [\"mlua/lua51\"]\nlua52 = [\"mlua/lua52\"]\nlua53 = [\"mlua/lua53\"]\nlua54 = [\"mlua/lua54\"]\nluajit = [\"mlua/luajit\"]\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-c-defs.scm", "content": ";; Capture extern functions, variables, public classes, and methods\n(function_definition\n (storage_class_specifier) @extern\n) @function\n(struct_specifier) @struct\n(struct_specifier\n body: (field_declaration_list\n (field_declaration\n declarator: (field_identifier))? @class_variable\n )\n)\n(declaration\n (storage_class_specifier) @extern\n) @variable\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-c-sharp-defs.scm", "content": "(class_declaration\n name: (identifier) @class\n (parameter_list)? @method) ;; Primary constructor\n\n(record_declaration\n name: (identifier) @class\n (parameter_list)? @method) ;; Primary constructor\n\n(interface_declaration\n name: (identifier) @class)\n\n(method_declaration) @method\n\n(constructor_declaration) @method\n\n(property_declaration) @class_variable\n\n(field_declaration\n (variable_declaration\n (variable_declarator))) @class_variable\n\n(enum_declaration\n body: (enum_member_declaration_list\n (enum_member_declaration) @enum_item))\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-cpp-defs.scm", "content": ";; Capture functions, variables, nammespaces, classes, methods, and enums\n(namespace_definition) @namespace\n(function_definition) @function\n(class_specifier) @class\n(class_specifier\n body: (field_declaration_list\n (declaration\n declarator: (function_declarator))? @method\n (field_declaration\n declarator: (function_declarator))? @method\n (function_definition)? @method\n (function_declarator)? @method\n (field_declaration\n declarator: (field_identifier))? @class_variable\n )\n)\n(struct_specifier) @struct\n(struct_specifier\n body: (field_declaration_list\n (declaration\n declarator: (function_declarator))? @method\n (field_declaration\n declarator: (function_declarator))? @method\n (function_definition)? @method\n (function_declarator)? @method\n (field_declaration\n declarator: (field_identifier))? @class_variable\n )\n)\n((declaration type: (_))) @variable\n(enumerator_list ((enumerator) @enum_item))\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-elixir-defs.scm", "content": "; * modules and protocols\n(call\n target: (identifier) @ignore\n (arguments (alias) @class)\n (#match? @ignore \"^(defmodule|defprotocol)$\"))\n\n; * functions\n(call\n target: (identifier) @ignore\n (arguments\n [\n ; zero-arity functions with no parentheses\n (identifier) @method\n ; regular function clause\n (call target: (identifier) @method)\n ; function clause with a guard clause\n (binary_operator\n left: (call target: (identifier) @method)\n operator: \"when\")\n ])\n (#match? @ignore \"^(def|defdelegate|defguard|defn)$\"))\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-go-defs.scm", "content": ";; Capture top-level functions and struct definitions\n(source_file\n (var_declaration\n (var_spec) @variable\n )\n)\n(source_file\n (const_declaration\n (const_spec) @variable\n )\n)\n(source_file\n (function_declaration) @function\n)\n(source_file\n (type_declaration\n (type_spec (struct_type)) @class\n )\n)\n(source_file\n (type_declaration\n (type_spec\n (struct_type\n (field_declaration_list\n (field_declaration) @class_variable)))\n )\n)\n(source_file\n (method_declaration) @method\n)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-java-defs.scm", "content": ";; Capture exported functions, arrow functions, variables, classes, and method definitions\n\n(class_declaration\n name: (identifier) @class)\n\n(interface_declaration\n name: (identifier) @class)\n\n(enum_declaration\n name: (identifier) @enum)\n\n(enum_constant\n name: (identifier) @enum_item)\n\n(class_body\n (field_declaration) @class_variable)\n\n(class_body\n (constructor_declaration) @method)\n\n(class_body\n (method_declaration) @method)\n\n(interface_body\n (method_declaration) @method)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-javascript-defs.scm", "content": ";; Capture exported functions, arrow functions, variables, classes, and method definitions\n(export_statement\n declaration: (lexical_declaration\n (variable_declarator) @variable\n )\n)\n(export_statement\n declaration: (function_declaration) @function\n)\n(export_statement\n declaration: (class_declaration\n body: (class_body\n (field_definition) @class_variable\n )\n )\n)\n(export_statement\n declaration: (class_declaration\n body: (class_body\n (method_definition) @method\n )\n )\n)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-lua-defs.scm", "content": ";; Capture function and method definitions\n(variable_list) @variable\n(function_declaration) @function\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-php-defs.scm", "content": ";; Capture exported functions, arrow functions, variables, classes, and method definitions\n\n(class_declaration) @class\n(interface_declaration) @class\n\n(function_definition) @function\n\n(assignment_expression) @assignment\n\n(const_declaration\n (const_element\n (name) @variable))\n\n(_\n body: (declaration_list\n (property_declaration) @class_variable))\n\n(_\n body: (declaration_list\n (method_declaration) @method))\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-python-defs.scm", "content": ";; Capture top-level functions, class, and method definitions\n(module\n (expression_statement\n (assignment) @assignment\n )\n)\n(module\n (function_definition) @function\n)\n(module\n (class_definition\n body: (block\n (expression_statement\n (assignment) @class_assignment\n )\n )\n )\n)\n(module\n (class_definition\n body: (block\n (function_definition) @method\n )\n )\n)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-ruby-defs.scm", "content": ";; Capture top-level methods, class definitions, and methods within classes\n\n(class\n (body_statement\n (call)? @class_call\n (assignment)? @class_assignment\n (method)? @method\n )\n) @class\n\n(program\n (method) @function\n)\n(program\n (assignment) @assignment\n)\n\n(module) @module\n\n(module\n (body_statement\n (call)? @class_call\n (assignment)? @class_assignment\n (method)? @method\n )\n)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-rust-defs.scm", "content": ";; Capture public functions, structs, methods, and variable definitions\n(function_item) @function\n(impl_item\n body: (declaration_list\n (function_item) @method\n )\n)\n(struct_item) @class\n(struct_item\n body: (field_declaration_list\n (field_declaration) @class_variable\n )\n)\n(enum_item\n body: (enum_variant_list\n (enum_variant) @enum_item\n )\n)\n(const_item) @variable\n(static_item) @variable\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-scala-defs.scm", "content": "(class_definition\n name: (identifier) @class)\n\n(object_definition\n name: (identifier) @class)\n\n(trait_definition\n name: (identifier) @class)\n\n(simple_enum_case\n name: (identifier) @enum_item)\n\n(full_enum_case\n name: (identifier) @enum_item)\n\n(template_body\n (function_definition) @method\n)\n\n(template_body\n (function_declaration) @method\n)\n\n(template_body\n (val_definition) @class_variable\n)\n\n(template_body\n (val_declaration) @class_variable\n)\n\n\n(template_body\n (var_definition) @class_variable\n)\n\n(template_body\n (var_declaration) @class_variable\n)\n\n(compilation_unit\n (function_definition) @function\n)\n\n(compilation_unit\n (val_definition) @variable\n)\n\n(compilation_unit\n (var_definition) @variable\n)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-swift-defs.scm", "content": "(property_declaration) @variable\n\n(function_declaration) @function\n\n\n(class_declaration\n\t_?\n\t[\n\t \"struct\"\n\t \"class\"\n\t]) @class\n\n(class_declaration\n\t_?\n\t \"enum\"\n\t) @enum\n\n(class_body\n (property_declaration) @class_variable)\n\n(class_body\n (function_declaration) @method)\n\n(class_body\n (init_declaration) @method)\n\n(protocol_declaration\n body: (protocol_body\n (protocol_function_declaration) @function))\n\n(protocol_declaration\n body: (protocol_body\n (protocol_property_declaration) @class_variable))\n\n(class_declaration\n\tbody: (enum_class_body\n (enum_entry) @enum_item))\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-typescript-defs.scm", "content": ";; Capture exported functions, arrow functions, variables, classes, and method definitions\n(export_statement\n declaration: (lexical_declaration\n (variable_declarator) @variable\n )\n)\n(export_statement\n declaration: (function_declaration) @function\n)\n(export_statement\n declaration: (class_declaration\n body: (class_body\n (public_field_definition) @class_variable\n )\n )\n)\n(interface_declaration\n body: (interface_body\n (property_signature) @class_variable\n )\n)\n(type_alias_declaration\n value: (object_type\n (property_signature) @class_variable\n )\n)\n(export_statement\n declaration: (class_declaration\n body: (class_body\n (method_definition) @method\n )\n )\n)\n" }, { "path": "crates/avante-repo-map/queries/tree-sitter-zig-defs.scm", "content": " ;; Capture functions, structs, methods, variable definitions, and unions in Zig\n(variable_declaration (identifier)\n (struct_declaration\n (container_field) @class_variable))\n\n(variable_declaration (identifier)\n (struct_declaration\n (function_declaration\n name: (identifier) @method)))\n\n(variable_declaration (identifier)\n (enum_declaration\n (container_field\n type: (identifier) @enum_item)))\n\n(variable_declaration (identifier)\n (union_declaration\n (container_field\n name: (identifier) @union_item)))\n\n(source_file (function_declaration) @function)\n\n(source_file (variable_declaration (identifier) @variable))\n" }, { "path": "crates/avante-repo-map/src/lib.rs", "content": "#![allow(clippy::unnecessary_map_or)]\n\nuse mlua::prelude::*;\nuse std::cell::RefCell;\nuse std::collections::BTreeMap;\nuse tree_sitter::{Node, Parser, Query, QueryCursor};\nuse tree_sitter_language::LanguageFn;\n\n#[derive(Debug, Clone)]\npub struct Func {\n pub name: String,\n pub params: String,\n pub return_type: String,\n pub accessibility_modifier: Option,\n}\n\n#[derive(Debug, Clone)]\npub struct Class {\n pub type_name: String,\n pub name: String,\n pub methods: Vec,\n pub properties: Vec,\n pub visibility_modifier: Option,\n}\n\n#[derive(Debug, Clone)]\npub struct Enum {\n pub name: String,\n pub items: Vec,\n}\n\n#[derive(Debug, Clone)]\npub struct Union {\n pub name: String,\n pub items: Vec,\n}\n\n#[derive(Debug, Clone)]\npub struct Variable {\n pub name: String,\n pub value_type: String,\n}\n\n#[derive(Debug, Clone)]\npub enum Definition {\n Func(Func),\n Class(Class),\n Module(Class),\n Enum(Enum),\n Variable(Variable),\n Union(Union),\n // TODO: Namespace support\n}\n\nfn get_ts_language(language: &str) -> Option {\n match language {\n \"rust\" => Some(tree_sitter_rust::LANGUAGE),\n \"python\" => Some(tree_sitter_python::LANGUAGE),\n \"php\" => Some(tree_sitter_php::LANGUAGE_PHP),\n \"java\" => Some(tree_sitter_java::LANGUAGE),\n \"javascript\" => Some(tree_sitter_javascript::LANGUAGE),\n \"typescript\" => Some(tree_sitter_typescript::LANGUAGE_TSX),\n \"go\" => Some(tree_sitter_go::LANGUAGE),\n \"c\" => Some(tree_sitter_c::LANGUAGE),\n \"cpp\" => Some(tree_sitter_cpp::LANGUAGE),\n \"lua\" => Some(tree_sitter_lua::LANGUAGE),\n \"ruby\" => Some(tree_sitter_ruby::LANGUAGE),\n \"zig\" => Some(tree_sitter_zig::LANGUAGE),\n \"scala\" => Some(tree_sitter_scala::LANGUAGE),\n \"swift\" => Some(tree_sitter_swift::LANGUAGE),\n \"elixir\" => Some(tree_sitter_elixir::LANGUAGE),\n \"csharp\" => Some(tree_sitter_c_sharp::LANGUAGE),\n _ => None,\n }\n}\n\nconst C_QUERY: &str = include_str!(\"../queries/tree-sitter-c-defs.scm\");\nconst CPP_QUERY: &str = include_str!(\"../queries/tree-sitter-cpp-defs.scm\");\nconst GO_QUERY: &str = include_str!(\"../queries/tree-sitter-go-defs.scm\");\nconst JAVA_QUERY: &str = include_str!(\"../queries/tree-sitter-java-defs.scm\");\nconst JAVASCRIPT_QUERY: &str = include_str!(\"../queries/tree-sitter-javascript-defs.scm\");\nconst LUA_QUERY: &str = include_str!(\"../queries/tree-sitter-lua-defs.scm\");\nconst PYTHON_QUERY: &str = include_str!(\"../queries/tree-sitter-python-defs.scm\");\nconst PHP_QUERY: &str = include_str!(\"../queries/tree-sitter-php-defs.scm\");\nconst RUST_QUERY: &str = include_str!(\"../queries/tree-sitter-rust-defs.scm\");\nconst ZIG_QUERY: &str = include_str!(\"../queries/tree-sitter-zig-defs.scm\");\nconst TYPESCRIPT_QUERY: &str = include_str!(\"../queries/tree-sitter-typescript-defs.scm\");\nconst RUBY_QUERY: &str = include_str!(\"../queries/tree-sitter-ruby-defs.scm\");\nconst SCALA_QUERY: &str = include_str!(\"../queries/tree-sitter-scala-defs.scm\");\nconst SWIFT_QUERY: &str = include_str!(\"../queries/tree-sitter-swift-defs.scm\");\nconst ELIXIR_QUERY: &str = include_str!(\"../queries/tree-sitter-elixir-defs.scm\");\nconst CSHARP_QUERY: &str = include_str!(\"../queries/tree-sitter-c-sharp-defs.scm\");\n\nfn get_definitions_query(language: &str) -> Result {\n let ts_language = get_ts_language(language);\n if ts_language.is_none() {\n return Err(format!(\"Unsupported language: {language}\"));\n }\n let ts_language = ts_language.unwrap();\n let contents = match language {\n \"c\" => C_QUERY,\n \"cpp\" => CPP_QUERY,\n \"go\" => GO_QUERY,\n \"java\" => JAVA_QUERY,\n \"javascript\" => JAVASCRIPT_QUERY,\n \"lua\" => LUA_QUERY,\n \"php\" => PHP_QUERY,\n \"python\" => PYTHON_QUERY,\n \"rust\" => RUST_QUERY,\n \"zig\" => ZIG_QUERY,\n \"typescript\" => TYPESCRIPT_QUERY,\n \"ruby\" => RUBY_QUERY,\n \"scala\" => SCALA_QUERY,\n \"swift\" => SWIFT_QUERY,\n \"elixir\" => ELIXIR_QUERY,\n \"csharp\" => CSHARP_QUERY,\n _ => return Err(format!(\"Unsupported language: {language}\")),\n };\n let query = Query::new(&ts_language.into(), contents)\n .unwrap_or_else(|e| panic!(\"Failed to parse query for {language}: {e}\"));\n Ok(query)\n}\n\nfn get_closest_ancestor_name(node: &Node, source: &str) -> String {\n let mut parent = node.parent();\n while let Some(parent_node) = parent {\n let name_node = parent_node.child_by_field_name(\"name\");\n if let Some(name_node) = name_node {\n return get_node_text(&name_node, source.as_bytes()).to_string();\n }\n parent = parent_node.parent();\n }\n String::new()\n}\n\nfn find_ancestor_by_type<'a>(node: &'a Node, parent_type: &str) -> Option> {\n let mut parent = node.parent();\n while let Some(parent_node) = parent {\n if parent_node.kind() == parent_type {\n return Some(parent_node);\n }\n parent = parent_node.parent();\n }\n None\n}\n\nfn find_first_ancestor_by_types<'a>(\n node: &'a Node,\n possible_parent_types: &[&str],\n) -> Option> {\n let mut parent = node.parent();\n while let Some(parent_node) = parent {\n if possible_parent_types.contains(&parent_node.kind()) {\n return Some(parent_node);\n }\n parent = parent_node.parent();\n }\n None\n}\n\nfn find_descendant_by_type<'a>(node: &'a Node, child_type: &str) -> Option> {\n let mut cursor = node.walk();\n for i in 0..node.descendant_count() {\n cursor.goto_descendant(i);\n let node = cursor.node();\n if node.kind() == child_type {\n return Some(node);\n }\n }\n None\n}\n\nfn ruby_method_is_private<'a>(node: &'a Node, source: &'a [u8]) -> bool {\n let mut prev_sibling = node.prev_sibling();\n while let Some(prev_sibling_node) = prev_sibling {\n if prev_sibling_node.kind() == \"identifier\" {\n let text = prev_sibling_node.utf8_text(source).unwrap_or_default();\n if text == \"private\" {\n return true;\n } else if text == \"public\" || text == \"protected\" {\n return false;\n }\n } else if prev_sibling_node.kind() == \"class\" || prev_sibling_node.kind() == \"module\" {\n return false;\n }\n prev_sibling = prev_sibling_node.prev_sibling();\n }\n false\n}\n\nfn find_child_by_type<'a>(node: &'a Node, child_type: &str) -> Option> {\n node.children(&mut node.walk())\n .find(|child| child.kind() == child_type)\n}\n\n// Zig-specific function to find the parent variable declaration\nfn zig_find_parent_variable_declaration_name<'a>(\n node: &'a Node,\n source: &'a [u8],\n) -> Option {\n let vardec = find_ancestor_by_type(node, \"variable_declaration\");\n if let Some(vardec) = vardec {\n // Find the identifier child node, which represents the class name\n let identifier_node = find_child_by_type(&vardec, \"identifier\");\n if let Some(identifier_node) = identifier_node {\n return Some(get_node_text(&identifier_node, source));\n }\n }\n None\n}\n\nfn zig_is_declaration_public<'a>(node: &'a Node, declaration_type: &str, source: &'a [u8]) -> bool {\n let declaration = find_ancestor_by_type(node, declaration_type);\n if let Some(declaration) = declaration {\n let declaration_text = get_node_text(&declaration, source);\n return declaration_text.starts_with(\"pub\");\n }\n false\n}\n\nfn zig_is_variable_declaration_public<'a>(node: &'a Node, source: &'a [u8]) -> bool {\n zig_is_declaration_public(node, \"variable_declaration\", source)\n}\n\nfn zig_is_function_declaration_public<'a>(node: &'a Node, source: &'a [u8]) -> bool {\n zig_is_declaration_public(node, \"function_declaration\", source)\n}\n\nfn zig_find_type_in_parent<'a>(node: &'a Node, source: &'a [u8]) -> Option {\n // First go to the parent and then get the child_by_field_name \"type\"\n if let Some(parent) = node.parent() {\n if let Some(type_node) = parent.child_by_field_name(\"type\") {\n return Some(get_node_text(&type_node, source));\n }\n }\n None\n}\n\nfn csharp_is_primary_constructor(node: &Node) -> bool {\n node.kind() == \"parameter_list\"\n && node.parent().map_or(false, |n| {\n n.kind() == \"class_declaration\" || n.kind() == \"record_declaration\"\n })\n}\n\nfn csharp_find_parent_type_node<'a>(node: &'a Node) -> Option> {\n find_first_ancestor_by_types(node, &[\"class_declaration\", \"record_declaration\"])\n}\n\nfn ex_find_parent_module_declaration_name<'a>(node: &'a Node, source: &'a [u8]) -> Option {\n let mut parent = node.parent();\n while let Some(parent_node) = parent {\n if parent_node.kind() == \"call\" {\n let text = get_node_text(&parent_node, source);\n if text.starts_with(\"defmodule \") {\n let arguments_node = find_child_by_type(&parent_node, \"arguments\");\n if let Some(arguments_node) = arguments_node {\n return Some(get_node_text(&arguments_node, source));\n }\n }\n }\n parent = parent_node.parent();\n }\n None\n}\n\nfn ruby_find_parent_module_declaration_name<'a>(\n node: &'a Node,\n source: &'a [u8],\n) -> Option {\n let mut path_parts = Vec::new();\n let mut current = Some(*node);\n\n while let Some(current_node) = current {\n if current_node.kind() == \"module\" || current_node.kind() == \"class\" {\n if let Some(name_node) = current_node.child_by_field_name(\"name\") {\n path_parts.push(get_node_text(&name_node, source));\n }\n }\n current = current_node.parent();\n }\n\n if path_parts.is_empty() {\n None\n } else {\n path_parts.reverse();\n Some(path_parts.join(\"::\"))\n }\n}\n\nfn get_node_text<'a>(node: &'a Node, source: &'a [u8]) -> String {\n node.utf8_text(source).unwrap_or_default().to_string()\n}\n\nfn get_node_type<'a>(node: &'a Node, source: &'a [u8]) -> String {\n let predefined_type_node = find_descendant_by_type(node, \"predefined_type\");\n if let Some(type_node) = predefined_type_node {\n return type_node.utf8_text(source).unwrap().to_string();\n }\n let value_type_node = node.child_by_field_name(\"type\");\n value_type_node\n .map(|n| n.utf8_text(source).unwrap().to_string())\n .unwrap_or_default()\n}\n\nfn is_first_letter_uppercase(name: &str) -> bool {\n if name.is_empty() {\n return false;\n }\n name.chars().next().unwrap().is_uppercase()\n}\n\n// Given a language, parse the given source code and return exported definitions\nfn extract_definitions(language: &str, source: &str) -> Result, String> {\n let ts_language = get_ts_language(language);\n\n if ts_language.is_none() {\n return Ok(vec![]);\n }\n\n let ts_language = ts_language.unwrap();\n\n let mut definitions = Vec::new();\n let mut parser = Parser::new();\n parser\n .set_language(&ts_language.into())\n .unwrap_or_else(|_| panic!(\"Failed to set language for {language}\"));\n let tree = parser\n .parse(source, None)\n .unwrap_or_else(|| panic!(\"Failed to parse source code for {language}\"));\n let root_node = tree.root_node();\n\n let query = get_definitions_query(language)?;\n let mut query_cursor = QueryCursor::new();\n let captures = query_cursor.captures(&query, root_node, source.as_bytes());\n\n let mut class_def_map: BTreeMap> = BTreeMap::new();\n let mut enum_def_map: BTreeMap> = BTreeMap::new();\n let mut union_def_map: BTreeMap> = BTreeMap::new();\n\n let ensure_class_def =\n |language: &str, name: &str, class_def_map: &mut BTreeMap>| {\n let mut type_name = \"class\";\n if language == \"elixir\" {\n type_name = \"module\";\n }\n class_def_map.entry(name.to_string()).or_insert_with(|| {\n RefCell::new(Class {\n type_name: type_name.to_string(),\n name: name.to_string(),\n methods: vec![],\n properties: vec![],\n visibility_modifier: None,\n })\n });\n };\n\n let ensure_module_def = |name: &str, class_def_map: &mut BTreeMap>| {\n class_def_map.entry(name.to_string()).or_insert_with(|| {\n RefCell::new(Class {\n name: name.to_string(),\n type_name: \"module\".to_string(),\n methods: vec![],\n properties: vec![],\n visibility_modifier: None,\n })\n });\n };\n\n let ensure_enum_def = |name: &str, enum_def_map: &mut BTreeMap>| {\n enum_def_map.entry(name.to_string()).or_insert_with(|| {\n RefCell::new(Enum {\n name: name.to_string(),\n items: vec![],\n })\n });\n };\n\n let ensure_union_def = |name: &str, union_def_map: &mut BTreeMap>| {\n union_def_map.entry(name.to_string()).or_insert_with(|| {\n RefCell::new(Union {\n name: name.to_string(),\n items: vec![],\n })\n });\n };\n\n // Sometimes, multiple queries capture the same node with the same capture name.\n // We need to ensure that we only add the node to the definition map once.\n let mut captured_nodes: BTreeMap> = BTreeMap::new();\n\n for (m, _) in captures {\n for capture in m.captures {\n let capture_name = &query.capture_names()[capture.index as usize];\n let node = capture.node;\n let node_text = node.utf8_text(source.as_bytes()).unwrap();\n\n let node_id = node.id();\n if captured_nodes\n .get(*capture_name)\n .map_or(false, |v| v.contains(&node_id))\n {\n continue;\n }\n captured_nodes\n .entry(String::from(*capture_name))\n .or_default()\n .push(node_id);\n\n let name = match language {\n \"cpp\" => {\n if *capture_name == \"class\" {\n node.child_by_field_name(\"name\")\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(node_text)\n .to_string()\n } else {\n let ident = find_descendant_by_type(&node, \"field_identifier\")\n .or_else(|| find_descendant_by_type(&node, \"operator_name\"))\n .or_else(|| find_descendant_by_type(&node, \"identifier\"))\n .map(|n| n.utf8_text(source.as_bytes()).unwrap());\n if let Some(ident) = ident {\n let scope = node\n .child_by_field_name(\"declarator\")\n .and_then(|n| n.child_by_field_name(\"declarator\"))\n .and_then(|n| n.child_by_field_name(\"scope\"));\n\n if let Some(scope_node) = scope {\n format!(\n \"{}::{}\",\n scope_node.utf8_text(source.as_bytes()).unwrap(),\n ident\n )\n } else {\n ident.to_string()\n }\n } else {\n node_text.to_string()\n }\n }\n }\n \"scala\" => node\n .child_by_field_name(\"name\")\n .or_else(|| node.child_by_field_name(\"pattern\"))\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(node_text)\n .to_string(),\n \"csharp\" => {\n let mut identifier = node;\n // Handle primary constructors (they are direct children of *_declaration)\n if *capture_name == \"method\" && csharp_is_primary_constructor(&node) {\n identifier = node.parent().unwrap_or(node);\n } else if *capture_name == \"class_variable\" {\n identifier =\n find_descendant_by_type(&node, \"variable_declarator\").unwrap_or(node);\n }\n\n identifier\n .child_by_field_name(\"name\")\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(node_text)\n .to_string()\n }\n \"ruby\" => {\n let name = node\n .child_by_field_name(\"name\")\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(node_text)\n .to_string();\n if *capture_name == \"class\" || *capture_name == \"module\" {\n ruby_find_parent_module_declaration_name(&node, source.as_bytes())\n .unwrap_or(name)\n } else {\n name\n }\n }\n _ => node\n .child_by_field_name(\"name\")\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(node_text)\n .to_string(),\n };\n\n match *capture_name {\n \"class\" => {\n if !name.is_empty() {\n if language == \"go\" && !is_first_letter_uppercase(&name) {\n continue;\n }\n ensure_class_def(language, &name, &mut class_def_map);\n let visibility_modifier_node =\n find_child_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n let class_def = class_def_map.get_mut(&name).unwrap();\n class_def.borrow_mut().visibility_modifier =\n if visibility_modifier.is_empty() {\n None\n } else {\n Some(visibility_modifier.to_string())\n };\n }\n }\n \"module\" => {\n if !name.is_empty() {\n ensure_module_def(&name, &mut class_def_map);\n }\n }\n \"enum_item\" => {\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n if language == \"zig\"\n && !zig_is_variable_declaration_public(&node, source.as_bytes())\n {\n continue;\n }\n let mut enum_name = get_closest_ancestor_name(&node, source);\n if language == \"zig\" {\n enum_name =\n zig_find_parent_variable_declaration_name(&node, source.as_bytes())\n .unwrap_or_default();\n }\n if language == \"scala\" {\n if let Some(enum_node) = find_ancestor_by_type(&node, \"enum_definition\") {\n if let Some(name_node) = enum_node.child_by_field_name(\"name\") {\n enum_name =\n name_node.utf8_text(source.as_bytes()).unwrap().to_string();\n }\n }\n }\n if !enum_name.is_empty()\n && language == \"go\"\n && !is_first_letter_uppercase(&enum_name)\n {\n continue;\n }\n ensure_enum_def(&enum_name, &mut enum_def_map);\n let enum_def = enum_def_map.get_mut(&enum_name).unwrap();\n let enum_type_node = find_descendant_by_type(&node, \"type_identifier\");\n let enum_type = enum_type_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n let variable = Variable {\n name: name.to_string(),\n value_type: enum_type.to_string(),\n };\n enum_def.borrow_mut().items.push(variable);\n }\n \"union_item\" => {\n if language != \"zig\" {\n continue;\n }\n if !zig_is_variable_declaration_public(&node, source.as_bytes()) {\n continue;\n }\n let union_name =\n zig_find_parent_variable_declaration_name(&node, source.as_bytes())\n .unwrap_or_default();\n ensure_union_def(&union_name, &mut union_def_map);\n let union_def = union_def_map.get_mut(&union_name).unwrap();\n let union_type_node = find_descendant_by_type(&node, \"type_identifier\");\n let union_type = union_type_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n let variable = Variable {\n name: name.to_string(),\n value_type: union_type.to_string(),\n };\n union_def.borrow_mut().items.push(variable);\n }\n \"method\" => {\n // TODO: C++: Skip private/protected class/struct methods\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n if language == \"swift\" {\n if visibility_modifier.contains(\"private\") {\n continue;\n }\n }\n if language == \"java\" {\n let modifier_node = find_descendant_by_type(&node, \"modifiers\");\n if modifier_node.is_some() {\n let modifier_text =\n modifier_node.unwrap().utf8_text(source.as_bytes()).unwrap();\n if modifier_text.contains(\"private\") {\n continue;\n }\n }\n }\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n if language == \"zig\"\n && !(zig_is_function_declaration_public(&node, source.as_bytes())\n && zig_is_variable_declaration_public(&node, source.as_bytes()))\n {\n continue;\n }\n if language == \"cpp\"\n && find_descendant_by_type(&node, \"destructor_name\").is_some()\n {\n continue;\n }\n\n if !name.is_empty() && language == \"go\" && !is_first_letter_uppercase(&name) {\n continue;\n }\n\n if language == \"csharp\" {\n let csharp_visibility = find_descendant_by_type(&node, \"modifier\");\n if csharp_visibility.is_none() && !csharp_is_primary_constructor(&node) {\n continue;\n }\n if csharp_visibility.is_some() {\n let csharp_visibility_text = csharp_visibility\n .unwrap()\n .utf8_text(source.as_bytes())\n .unwrap();\n if csharp_visibility_text == \"private\" {\n continue;\n }\n }\n }\n\n let mut params_node = node\n .child_by_field_name(\"parameters\")\n .or_else(|| find_descendant_by_type(&node, \"parameter_list\"));\n\n let zig_function_node = find_ancestor_by_type(&node, \"function_declaration\");\n if language == \"zig\" {\n params_node = zig_function_node\n .as_ref()\n .and_then(|n| find_child_by_type(n, \"parameters\"));\n }\n let ex_function_node = find_ancestor_by_type(&node, \"call\");\n if language == \"elixir\" {\n params_node = ex_function_node\n .as_ref()\n .and_then(|n| find_child_by_type(n, \"arguments\"));\n }\n\n let params = params_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"()\");\n let mut return_type_node = match language {\n \"cpp\" => node.child_by_field_name(\"type\"),\n \"csharp\" => node.child_by_field_name(\"returns\"),\n _ => node.child_by_field_name(\"return_type\"),\n };\n if language == \"cpp\" {\n let class_specifier_node = find_ancestor_by_type(&node, \"class_specifier\");\n let type_identifier_node =\n class_specifier_node.and_then(|n| n.child_by_field_name(\"name\"));\n\n if let Some(type_identifier_node) = type_identifier_node {\n let type_identifier_text =\n type_identifier_node.utf8_text(source.as_bytes()).unwrap();\n if name == type_identifier_text {\n return_type_node = Some(type_identifier_node);\n }\n }\n }\n if language == \"csharp\" {\n let type_specifier_node = csharp_find_parent_type_node(&node);\n let type_identifier_node =\n type_specifier_node.and_then(|n| n.child_by_field_name(\"name\"));\n\n if let Some(type_identifier_node) = type_identifier_node {\n let type_identifier_text =\n type_identifier_node.utf8_text(source.as_bytes()).unwrap();\n if name == type_identifier_text {\n return_type_node = Some(type_identifier_node);\n }\n }\n }\n if return_type_node.is_none() {\n return_type_node = node.child_by_field_name(\"result\");\n }\n let mut return_type = \"void\".to_string();\n if language == \"elixir\" {\n return_type = String::new();\n }\n if return_type_node.is_some() {\n return_type = get_node_type(&return_type_node.unwrap(), source.as_bytes());\n if return_type.is_empty() {\n return_type = return_type_node\n .unwrap()\n .utf8_text(source.as_bytes())\n .unwrap_or(\"void\")\n .to_string();\n }\n }\n\n let impl_item_node = find_ancestor_by_type(&node, \"impl_item\");\n let receiver_node = node.child_by_field_name(\"receiver\");\n let class_name = if language == \"zig\" {\n zig_find_parent_variable_declaration_name(&node, source.as_bytes())\n .unwrap_or_default()\n } else if language == \"elixir\" {\n ex_find_parent_module_declaration_name(&node, source.as_bytes())\n .unwrap_or_default()\n } else if language == \"cpp\" {\n find_ancestor_by_type(&node, \"class_specifier\")\n .or_else(|| find_ancestor_by_type(&node, \"struct_specifier\"))\n .and_then(|n| n.child_by_field_name(\"name\"))\n .and_then(|n| n.utf8_text(source.as_bytes()).ok())\n .unwrap_or(\"\")\n .to_string()\n } else if language == \"csharp\" {\n csharp_find_parent_type_node(&node)\n .and_then(|n| n.child_by_field_name(\"name\"))\n .and_then(|n| n.utf8_text(source.as_bytes()).ok())\n .unwrap_or(\"\")\n .to_string()\n } else if language == \"ruby\" {\n ruby_find_parent_module_declaration_name(&node, source.as_bytes())\n .unwrap_or_default()\n } else if let Some(impl_item) = impl_item_node {\n let impl_type_node = impl_item.child_by_field_name(\"type\");\n impl_type_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\")\n .to_string()\n } else if let Some(receiver) = receiver_node {\n let type_identifier_node =\n find_descendant_by_type(&receiver, \"type_identifier\");\n type_identifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\")\n .to_string()\n } else {\n get_closest_ancestor_name(&node, source).to_string()\n };\n\n if language == \"go\" && !is_first_letter_uppercase(&class_name) {\n continue;\n }\n\n ensure_class_def(language, &class_name, &mut class_def_map);\n let class_def = class_def_map.get_mut(&class_name).unwrap();\n\n let accessibility_modifier_node =\n find_descendant_by_type(&node, \"accessibility_modifier\");\n let accessibility_modifier = if language == \"ruby\" {\n if ruby_method_is_private(&node, source.as_bytes()) {\n \"private\"\n } else {\n \"\"\n }\n } else {\n accessibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\")\n };\n\n let func = Func {\n name: name.to_string(),\n params: params.to_string(),\n return_type: return_type.to_string(),\n accessibility_modifier: if accessibility_modifier.is_empty() {\n None\n } else {\n Some(accessibility_modifier.to_string())\n },\n };\n class_def.borrow_mut().methods.push(func);\n }\n \"class_assignment\" => {\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n if language == \"swift\" || language == \"java\" {\n if visibility_modifier.contains(\"private\") {\n continue;\n }\n }\n if language == \"java\" {\n let modifier_node = find_descendant_by_type(&node, \"modifiers\");\n if modifier_node.is_some() {\n let modifier_text =\n modifier_node.unwrap().utf8_text(source.as_bytes()).unwrap();\n if modifier_text.contains(\"private\") {\n continue;\n }\n }\n }\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n let left_node = node.child_by_field_name(\"left\");\n let left = left_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n let value_type = get_node_type(&node, source.as_bytes());\n let mut class_name = get_closest_ancestor_name(&node, source);\n if !class_name.is_empty() {\n if language == \"ruby\" {\n if let Some(namespaced_name) =\n ruby_find_parent_module_declaration_name(&node, source.as_bytes())\n {\n class_name = namespaced_name;\n }\n } else if language == \"go\" && !is_first_letter_uppercase(&class_name) {\n continue;\n }\n }\n if class_name.is_empty() {\n continue;\n }\n ensure_class_def(language, &class_name, &mut class_def_map);\n let class_def = class_def_map.get_mut(&class_name).unwrap();\n let variable = Variable {\n name: left.to_string(),\n value_type: value_type.to_string(),\n };\n class_def.borrow_mut().properties.push(variable);\n }\n \"class_variable\" => {\n // TODO: C++: Skip private/protected class/struct variables\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n\n if language == \"swift\" || language == \"java\" {\n if visibility_modifier.contains(\"private\") {\n continue;\n }\n }\n\n if language == \"java\" {\n let modifier_node = find_descendant_by_type(&node, \"modifiers\");\n if modifier_node.is_some() {\n let modifier_text =\n modifier_node.unwrap().utf8_text(source.as_bytes()).unwrap();\n if modifier_text.contains(\"private\") {\n continue;\n }\n }\n }\n\n let value_type = get_node_type(&node, source.as_bytes());\n\n if language == \"zig\" {\n // when top level class is not public, skip\n if !zig_is_variable_declaration_public(&node, source.as_bytes()) {\n continue;\n }\n }\n\n let mut class_name = get_closest_ancestor_name(&node, source);\n if language == \"cpp\" {\n class_name = find_ancestor_by_type(&node, \"class_specifier\")\n .or_else(|| find_ancestor_by_type(&node, \"struct_specifier\"))\n .and_then(|n| n.child_by_field_name(\"name\"))\n .and_then(|n| n.utf8_text(source.as_bytes()).ok())\n .unwrap_or(\"\")\n .to_string();\n }\n\n if language == \"csharp\" {\n let csharp_visibility = find_descendant_by_type(&node, \"modifier\");\n if csharp_visibility.is_none() {\n continue;\n }\n let csharp_visibility_text = csharp_visibility\n .unwrap()\n .utf8_text(source.as_bytes())\n .unwrap();\n if csharp_visibility_text == \"private\" {\n continue;\n }\n }\n\n if language == \"zig\" {\n class_name =\n zig_find_parent_variable_declaration_name(&node, source.as_bytes())\n .unwrap_or_default();\n }\n if !class_name.is_empty()\n && language == \"go\"\n && !is_first_letter_uppercase(&class_name)\n {\n continue;\n }\n if class_name.is_empty() {\n continue;\n }\n if !name.is_empty() && language == \"go\" && !is_first_letter_uppercase(&name) {\n continue;\n }\n ensure_class_def(language, &class_name, &mut class_def_map);\n let class_def = class_def_map.get_mut(&class_name).unwrap();\n let variable = Variable {\n name: name.to_string(),\n value_type: value_type.to_string(),\n };\n class_def.borrow_mut().properties.push(variable);\n }\n \"function\" | \"arrow_function\" => {\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n\n if language == \"swift\" || language == \"java\" {\n if visibility_modifier.contains(\"private\") {\n continue;\n }\n\n if node.parent().is_some() {\n continue;\n }\n }\n\n if language == \"java\" {\n let modifier_node = find_descendant_by_type(&node, \"modifiers\");\n if modifier_node.is_some() {\n let modifier_text =\n modifier_node.unwrap().utf8_text(source.as_bytes()).unwrap();\n if modifier_text.contains(\"private\") {\n continue;\n }\n }\n }\n\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n\n if language == \"zig\" {\n let variable_declaration_text =\n node.utf8_text(source.as_bytes()).unwrap_or(\"\");\n if !variable_declaration_text.contains(\"pub\") {\n continue;\n }\n }\n\n if !name.is_empty() && language == \"go\" && !is_first_letter_uppercase(&name) {\n continue;\n }\n let impl_item_node = find_ancestor_by_type(&node, \"impl_item\");\n if impl_item_node.is_some() {\n continue;\n }\n let class_specifier_node = find_ancestor_by_type(&node, \"class_specifier\");\n if class_specifier_node.is_some() {\n continue;\n }\n let struct_specifier_node = find_ancestor_by_type(&node, \"struct_specifier\");\n if struct_specifier_node.is_some() {\n continue;\n }\n let function_node = find_ancestor_by_type(&node, \"function_declaration\")\n .or_else(|| find_ancestor_by_type(&node, \"function_definition\"));\n if function_node.is_some() {\n continue;\n }\n let params_node = node\n .child_by_field_name(\"parameters\")\n .or_else(|| find_descendant_by_type(&node, \"parameter_list\"));\n let params = params_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"()\");\n\n let mut return_type = \"void\".to_string();\n let return_type_node = match language {\n \"cpp\" => node.child_by_field_name(\"type\"),\n _ => node\n .child_by_field_name(\"return_type\")\n .or_else(|| node.child_by_field_name(\"result\")),\n };\n if return_type_node.is_some() {\n return_type = get_node_type(&return_type_node.unwrap(), source.as_bytes());\n if return_type.is_empty() {\n return_type = return_type_node\n .unwrap()\n .utf8_text(source.as_bytes())\n .unwrap_or(\"void\")\n .to_string();\n }\n }\n\n let accessibility_modifier_node =\n find_descendant_by_type(&node, \"accessibility_modifier\");\n let accessibility_modifier = accessibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n\n let func = Func {\n name: name.to_string(),\n params: params.to_string(),\n return_type: return_type.to_string(),\n accessibility_modifier: if accessibility_modifier.is_empty() {\n None\n } else {\n Some(accessibility_modifier.to_string())\n },\n };\n definitions.push(Definition::Func(func));\n }\n \"assignment\" => {\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n if language == \"swift\" || language == \"java\" {\n if visibility_modifier.contains(\"private\") {\n continue;\n }\n }\n if language == \"java\" {\n let modifier_node = find_descendant_by_type(&node, \"modifiers\");\n if modifier_node.is_some() {\n let modifier_text =\n modifier_node.unwrap().utf8_text(source.as_bytes()).unwrap();\n if modifier_text.contains(\"private\") {\n continue;\n }\n }\n }\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n let impl_item_node = find_ancestor_by_type(&node, \"impl_item\")\n .or_else(|| find_ancestor_by_type(&node, \"class_declaration\"))\n .or_else(|| find_ancestor_by_type(&node, \"class_definition\"));\n if impl_item_node.is_some() {\n continue;\n }\n let function_node = find_ancestor_by_type(&node, \"function_declaration\")\n .or_else(|| find_ancestor_by_type(&node, \"function_definition\"));\n if function_node.is_some() {\n continue;\n }\n let left_node = node.child_by_field_name(\"left\");\n let left = left_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n if !left.is_empty() && language == \"go\" && !is_first_letter_uppercase(left) {\n continue;\n }\n\n let value_type = get_node_type(&node, source.as_bytes());\n let variable = Variable {\n name: left.to_string(),\n value_type: value_type.to_string(),\n };\n definitions.push(Definition::Variable(variable));\n }\n \"variable\" => {\n let visibility_modifier_node =\n find_descendant_by_type(&node, \"visibility_modifier\");\n let visibility_modifier = visibility_modifier_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"\");\n\n if language == \"swift\" {\n if visibility_modifier.contains(\"private\") {\n continue;\n }\n }\n\n if language == \"java\" {\n let modifier_node = find_descendant_by_type(&node, \"modifiers\");\n if modifier_node.is_some() {\n let modifier_text =\n modifier_node.unwrap().utf8_text(source.as_bytes()).unwrap();\n if modifier_text.contains(\"private\") {\n continue;\n }\n }\n }\n\n if language == \"rust\" && !visibility_modifier.contains(\"pub\") {\n continue;\n }\n\n if language == \"zig\"\n && !zig_is_variable_declaration_public(&node, source.as_bytes())\n {\n continue;\n }\n\n let impl_item_node = find_ancestor_by_type(&node, \"impl_item\")\n .or_else(|| find_ancestor_by_type(&node, \"class_declaration\"))\n .or_else(|| find_ancestor_by_type(&node, \"class_definition\"));\n if impl_item_node.is_some() {\n continue;\n }\n let function_node = find_ancestor_by_type(&node, \"function_declaration\")\n .or_else(|| find_ancestor_by_type(&node, \"function_definition\"));\n if function_node.is_some() {\n continue;\n }\n let value_node = node.child_by_field_name(\"value\");\n if value_node.is_some() {\n let value_type = value_node.unwrap().kind();\n if value_type == \"arrow_function\" {\n let params_node = value_node.unwrap().child_by_field_name(\"parameters\");\n let params = params_node\n .map(|n| n.utf8_text(source.as_bytes()).unwrap())\n .unwrap_or(\"()\");\n let mut return_type = \"void\".to_string();\n let return_type_node =\n value_node.unwrap().child_by_field_name(\"return_type\");\n if return_type_node.is_some() {\n return_type =\n get_node_type(&return_type_node.unwrap(), source.as_bytes());\n }\n let func = Func {\n name: name.to_string(),\n params: params.to_string(),\n return_type,\n accessibility_modifier: None,\n };\n definitions.push(Definition::Func(func));\n continue;\n }\n }\n\n let mut value_type = get_node_type(&node, source.as_bytes());\n if language == \"zig\" {\n if let Some(zig_type) = zig_find_type_in_parent(&node, source.as_bytes()) {\n value_type = zig_type;\n } else {\n continue;\n };\n }\n if !name.is_empty() && language == \"go\" && !is_first_letter_uppercase(&name) {\n continue;\n }\n let variable = Variable {\n name: name.to_string(),\n value_type: value_type.to_string(),\n };\n definitions.push(Definition::Variable(variable));\n }\n _ => {}\n }\n }\n }\n\n for (_, def) in class_def_map {\n let class_def = def.into_inner();\n if language == \"rust\" {\n if let Some(visibility_modifier) = &class_def.visibility_modifier {\n if visibility_modifier.contains(\"pub\") {\n definitions.push(Definition::Class(class_def));\n }\n }\n } else {\n definitions.push(Definition::Class(class_def));\n }\n }\n\n for (_, def) in enum_def_map {\n definitions.push(Definition::Enum(def.into_inner()));\n }\n for (_, def) in union_def_map {\n definitions.push(Definition::Union(def.into_inner()));\n }\n\n Ok(definitions)\n}\n\nfn stringify_function(func: &Func) -> String {\n let mut res = format!(\"func {}\", func.name);\n if func.params.is_empty() {\n res = format!(\"{res}()\");\n } else {\n res = format!(\"{res}{}\", func.params);\n }\n if !func.return_type.is_empty() {\n res = format!(\"{res} -> {}\", func.return_type);\n }\n if let Some(modifier) = &func.accessibility_modifier {\n res = format!(\"{modifier} {res}\");\n }\n format!(\"{res};\")\n}\n\nfn stringify_variable(variable: &Variable) -> String {\n let mut res = format!(\"var {}\", variable.name);\n if !variable.value_type.is_empty() {\n res = format!(\"{res}:{}\", variable.value_type);\n }\n format!(\"{res};\")\n}\n\nfn stringify_enum_item(item: &Variable) -> String {\n let mut res = item.name.clone();\n if !item.value_type.is_empty() {\n res = format!(\"{res}:{}\", item.value_type);\n }\n format!(\"{res};\")\n}\n\nfn stringify_union_item(item: &Variable) -> String {\n let mut res = item.name.clone();\n if !item.value_type.is_empty() {\n res = format!(\"{res}:{}\", item.value_type);\n }\n format!(\"{res};\")\n}\n\nfn stringify_class(class: &Class) -> String {\n let mut res = format!(\"{} {}{{\", class.type_name, class.name);\n for method in &class.methods {\n let method_str = stringify_function(method);\n res = format!(\"{res}{method_str}\");\n }\n for property in &class.properties {\n let property_str = stringify_variable(property);\n res = format!(\"{res}{property_str}\");\n }\n format!(\"{res}}};\")\n}\n\nfn stringify_enum(enum_def: &Enum) -> String {\n let mut res = format!(\"enum {}{{\", enum_def.name);\n for item in &enum_def.items {\n let item_str = stringify_enum_item(item);\n res = format!(\"{res}{item_str}\");\n }\n format!(\"{res}}};\")\n}\nfn stringify_union(union_def: &Union) -> String {\n let mut res = format!(\"union {}{{\", union_def.name);\n for item in &union_def.items {\n let item_str = stringify_union_item(item);\n res = format!(\"{res}{item_str}\");\n }\n format!(\"{res}}};\")\n}\n\nfn stringify_definitions(definitions: &Vec) -> String {\n let mut res = String::new();\n for definition in definitions {\n match definition {\n Definition::Class(class) => res = format!(\"{res}{}\", stringify_class(class)),\n Definition::Module(module) => res = format!(\"{res}{}\", stringify_class(module)),\n Definition::Enum(enum_def) => res = format!(\"{res}{}\", stringify_enum(enum_def)),\n Definition::Union(union_def) => res = format!(\"{res}{}\", stringify_union(union_def)),\n Definition::Func(func) => res = format!(\"{res}{}\", stringify_function(func)),\n Definition::Variable(variable) => {\n let variable_str = stringify_variable(variable);\n res = format!(\"{res}{variable_str}\");\n }\n }\n }\n res\n}\n\npub fn get_definitions_string(language: &str, source: &str) -> LuaResult {\n let definitions =\n extract_definitions(language, source).map_err(|e| LuaError::RuntimeError(e.to_string()))?;\n let stringified = stringify_definitions(&definitions);\n Ok(stringified)\n}\n\n#[mlua::lua_module]\nfn avante_repo_map(lua: &Lua) -> LuaResult {\n let exports = lua.create_table()?;\n exports.set(\n \"stringify_definitions\",\n lua.create_function(move |_, (language, source): (String, String)| {\n get_definitions_string(language.as_str(), source.as_str())\n })?,\n )?;\n Ok(exports)\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_rust() {\n let source = r#\"\n // This is a test comment\n pub const TEST_CONST: u32 = 1;\n pub static TEST_STATIC: u32 = 2;\n const INNER_TEST_CONST: u32 = 3;\n static INNER_TEST_STATIC: u32 = 4;\n pub(crate) struct TestStruct {\n pub test_field: String,\n inner_test_field: String,\n }\n impl TestStruct {\n pub fn test_method(&self, a: u32, b: u32) -> u32 {\n a + b\n }\n fn inner_test_method(&self, a: u32, b: u32) -> u32 {\n a + b\n }\n }\n struct InnerTestStruct {\n pub test_field: String,\n inner_test_field: String,\n }\n impl InnerTestStruct {\n pub fn test_method(&self, a: u32, b: u32) -> u32 {\n a + b\n }\n fn inner_test_method(&self, a: u32, b: u32) -> u32 {\n a + b\n }\n }\n pub enum TestEnum {\n TestEnumField1,\n TestEnumField2,\n }\n enum InnerTestEnum {\n InnerTestEnumField1,\n InnerTestEnumField2,\n }\n pub fn test_fn(a: u32, b: u32) -> u32 {\n let inner_var_in_func = 1;\n struct InnerStructInFunc {\n c: u32,\n }\n a + b + c\n }\n fn inner_test_fn(a: u32, b: u32) -> u32 {\n a + b\n }\n \"#;\n let definitions = extract_definitions(\"rust\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var TEST_CONST:u32;var TEST_STATIC:u32;func test_fn(a: u32, b: u32) -> u32;class TestStruct{func test_method(&self, a: u32, b: u32) -> u32;var test_field:String;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_zig() {\n let source = r#\"\n // This is a test comment\n pub const TEST_CONST: u32 = 1;\n pub var TEST_VAR: u32 = 2;\n const INNER_TEST_CONST: u32 = 3;\n var INNER_TEST_VAR: u32 = 4;\n pub const TestStruct = struct {\n test_field: []const u8,\n test_field2: u64,\n\n pub fn test_method(_: *TestStruct, a: u32, b: u32) u32 {\n return a + b;\n }\n\n fn inner_test_method(_: *TestStruct, a: u32, b: u32) u32 {\n return a + b;\n }\n };\n const InnerTestStruct = struct {\n test_field: []const u8,\n test_field2: u64,\n\n pub fn test_method(_: *InnerTestStruct, a: u32, b: u32) u32 {\n return a + b;\n }\n\n fn inner_test_method(_: *InnerTestStruct, a: u32, b: u32) u32 {\n return a + b;\n }\n };\n pub const TestEnum = enum {\n TestEnumField1,\n TestEnumField2,\n };\n const InnerTestEnum = enum {\n InnerTestEnumField1,\n InnerTestEnumField2,\n };\n\n pub const TestUnion = union {\n TestUnionField1: u32,\n TestUnionField2: u64,\n };\n\n const InnerTestUnion = union {\n InnerTestUnionField1: u32,\n InnerTestUnionField2: u64,\n };\n\n pub fn test_fn(a: u32, b: u32) u32 {\n const inner_var_in_func = 1;\n const InnerStructInFunc = struct {\n c: u32,\n };\n _ = InnerStructInFunc;\n return a + b + inner_var_in_func;\n }\n fn inner_test_fn(a: u32, b: u32) u32 {\n return a + b;\n }\n \"#;\n\n let definitions = extract_definitions(\"zig\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var TEST_CONST:u32;var TEST_VAR:u32;func test_fn() -> void;class TestStruct{func test_method(_: *TestStruct, a: u32, b: u32) -> void;var test_field:[]const u8;var test_field2:u64;};enum TestEnum{TestEnumField1;TestEnumField2;};union TestUnion{TestUnionField1;TestUnionField2;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_go() {\n let source = r#\"\n // This is a test comment\n package main\n import \"fmt\"\n const TestConst string = \"test\"\n const innerTestConst string = \"test\"\n var TestVar string\n var innerTestVar string\n type TestStruct struct {\n TestField string\n innerTestField string\n }\n func (t *TestStruct) TestMethod(a int, b int) (int, error) {\n var InnerVarInFunc int = 1\n type InnerStructInFunc struct {\n C int\n }\n return a + b, nil\n }\n func (t *TestStruct) innerTestMethod(a int, b int) (int, error) {\n return a + b, nil\n }\n type innerTestStruct struct {\n innerTestField string\n }\n func (t *innerTestStruct) testMethod(a int, b int) (int, error) {\n return a + b, nil\n }\n func (t *innerTestStruct) innerTestMethod(a int, b int) (int, error) {\n return a + b, nil\n }\n func TestFunc(a int, b int) (int, error) {\n return a + b, nil\n }\n func innerTestFunc(a int, b int) (int, error) {\n return a + b, nil\n }\n \"#;\n let definitions = extract_definitions(\"go\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var TestConst:string;var TestVar:string;func TestFunc(a int, b int) -> (int, error);class TestStruct{func TestMethod(a int, b int) -> (int, error);var TestField:string;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_python() {\n let source = r#\"\n # This is a test comment\n test_var: str = \"test\"\n class TestClass:\n def __init__(self, a, b):\n self.a = a\n self.b = b\n def test_method(self, a: int, b: int) -> int:\n inner_var_in_method: int = 1\n return a + b\n def test_func(a: int, b: int) -> int:\n inner_var_in_func: str = \"test\"\n class InnerClassInFunc:\n def __init__(self, a, b):\n self.a = a\n self.b = b\n def test_method(self, a: int, b: int) -> int:\n return a + b\n def inne_func_in_func(a: int, b: int) -> int:\n return a + b\n return a + b\n \"#;\n let definitions = extract_definitions(\"python\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var test_var:str;func test_func(a: int, b: int) -> int;class TestClass{func __init__(self, a, b) -> void;func test_method(self, a: int, b: int) -> int;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_typescript() {\n let source = r#\"\n // This is a test comment\n export const testVar: string = \"test\";\n const innerTestVar: string = \"test\";\n export class TestClass {\n a: number;\n b: number;\n constructor(a: number, b: number) {\n this.a = a;\n this.b = b;\n }\n testMethod(a: number, b: number): number {\n const innerConstInMethod: number = 1;\n function innerFuncInMethod(a: number, b: number): number {\n return a + b;\n }\n return a + b;\n }\n }\n class InnerTestClass {\n a: number;\n b: number;\n }\n export function testFunc(a: number, b: number) {\n const innerConstInFunc: number = 1;\n function innerFuncInFunc(a: number, b: number): number {\n return a + b;\n }\n return a + b;\n }\n export const testFunc2 = (a: number, b: number) => {\n return a + b;\n }\n export const testFunc3 = (a: number, b: number): number => {\n return a + b;\n }\n function innerTestFunc(a: number, b: number) {\n return a + b;\n }\n \"#;\n let definitions = extract_definitions(\"typescript\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var testVar:string;func testFunc(a: number, b: number) -> void;func testFunc2(a: number, b: number) -> void;func testFunc3(a: number, b: number) -> number;class TestClass{func constructor(a: number, b: number) -> void;func testMethod(a: number, b: number) -> number;var a:number;var b:number;};\"\n;\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_javascript() {\n let source = r#\"\n // This is a test comment\n export const testVar = \"test\";\n const innerTestVar = \"test\";\n export class TestClass {\n constructor(a, b) {\n this.a = a;\n this.b = b;\n }\n testMethod(a, b) {\n const innerConstInMethod = 1;\n function innerFuncInMethod(a, b) {\n return a + b;\n }\n return a + b;\n }\n }\n class InnerTestClass {\n constructor(a, b) {\n this.a = a;\n this.b = b;\n }\n }\n export const testFunc = function(a, b) {\n const innerConstInFunc = 1;\n function innerFuncInFunc(a, b) {\n return a + b;\n }\n return a + b;\n }\n export const testFunc2 = (a, b) => {\n return a + b;\n }\n export const testFunc3 = (a, b) => a + b;\n function innerTestFunc(a, b) {\n return a + b;\n }\n \"#;\n let definitions = extract_definitions(\"javascript\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var testVar;var testFunc;func testFunc2(a, b) -> void;func testFunc3(a, b) -> void;class TestClass{func constructor(a, b) -> void;func testMethod(a, b) -> void;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_ruby() {\n let source = r#\"\n # This is a test comment\n test_var = \"test\"\n def test_func(a, b)\n inner_var_in_func = \"test\"\n class InnerClassInFunc\n attr_accessor :a, :b\n def initialize(a, b)\n @a = a\n @b = b\n end\n def test_method(a, b)\n return a + b\n end\n end\n return a + b\n end\n class TestClass\n attr_accessor :a, :b\n def initialize(a, b)\n @a = a\n @b = b\n end\n def test_method(a, b)\n inner_var_in_method = 1\n def inner_func_in_method(a, b)\n return a + b\n end\n return a + b\n end\n end\n \"#;\n let definitions = extract_definitions(\"ruby\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n // FIXME:\n let expected = \"var test_var;func test_func(a, b) -> void;class InnerClassInFunc{func initialize(a, b) -> void;func test_method(a, b) -> void;};class TestClass{func initialize(a, b) -> void;func test_method(a, b) -> void;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_ruby2() {\n let source = r#\"\n # frozen_string_literal: true\n\n require('jwt')\n\n top_level_var = 1\n\n def top_level_func\n inner_var_in_func = 2\n end\n\n module A\n module B\n @module_var = :foo\n\n def module_method\n @module_var\n end\n\n class C < Base\n TEST_CONST = 1\n @class_var = :bar\n attr_accessor :a, :b\n\n def initialize(a, b)\n @a = a\n @b = b\n super\n end\n\n def bar\n inner_var_in_method = 1\n true\n end\n\n private\n\n def baz(request, params)\n auth_header = request.headers['Authorization']\n parts = auth_header.try(:split, /\\s+/)\n JWT.decode(parts.last)\n end\n end\n end\n end\n \"#;\n let definitions = extract_definitions(\"ruby\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var top_level_var;func top_level_func() -> void;module A{};module A::B{func module_method() -> void;var @module_var;};class A::B::C{func initialize(a, b) -> void;func bar() -> void;private func baz(request, params) -> void;var TEST_CONST;var @class_var;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_lua() {\n let source = r#\"\n -- This is a test comment\n local test_var = \"test\"\n function test_func(a, b)\n local inner_var_in_func = 1\n function inner_func_in_func(a, b)\n return a + b\n end\n return a + b\n end\n \"#;\n let definitions = extract_definitions(\"lua\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var test_var;func test_func(a, b) -> void;\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_c() {\n let source = r#\"\n #include \n\n int test_var = 2;\n extern int extern_test_var;\n\n int TestFunc(bool b) { return b ? 42 : -1; }\n extern void ExternTestFunc();\n\n struct Foo {\n int a;\n int b;\n };\n\n typedef int my_int;\n \"#;\n let definitions = extract_definitions(\"c\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var extern int extern_test_var;:int;var extern void ExternTestFunc();:void;class Foo{var int a;:int;var int b;:int;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_cpp() {\n let source = r#\"\n // This is a test comment\n #include \n\n namespace {\n constexpr int TEST_CONSTEXPR = 1;\n const int TEST_CONST = 1;\n }; // namespace\n\n int test_var = 2;\n\n int TestFunc(bool b) { return b ? 42 : -1; }\n\n template class TestClass {\n public:\n TestClass();\n TestClass(T a, T b);\n ~TestClass();\n bool operator==(const TestClass &other);\n T testMethod(T x, T y) { return x + y; }\n T c;\n\n private:\n void privateMethod();\n T a = 0;\n T b;\n };\n\n struct TestStruct {\n public:\n TestStruct(int a, int b);\n ~TestStruct();\n bool operator==(const TestStruct &other);\n int testMethod(int x, int y) { return x + y; }\n static int c;\n\n private:\n int a = 0;\n int b;\n };\n\n bool TestStruct::operator==(const TestStruct &other) { return true; }\n\n int TestStruct::c = 0;\n\n int testFunction(int a, int b) { return a + b; }\n\n namespace TestNamespace {\n class InnerClass {\n public:\n bool innerMethod(int a) const;\n };\n bool InnerClass::innerMethod(int a) const { return doSomething(a * 2); }\n } // namespace TestNamespace\n\n enum TestEnum { ENUM_VALUE_1, ENUM_VALUE_2 };\n \"#;\n let definitions = extract_definitions(\"cpp\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{}\", stringified);\n let expected = \"var TEST_CONSTEXPR:int;var TEST_CONST:int;var test_var:int;func TestFunc(bool b) -> int;func TestStruct::operator==(const TestStruct &other) -> bool;var TestStruct::c:int;func testFunction(int a, int b) -> int;func InnerClass::innerMethod(int a) -> bool;class InnerClass{func innerMethod(int a) -> bool;};class TestClass{func TestClass() -> TestClass;func operator==(const TestClass &other) -> bool;func testMethod(T x, T y) -> T;func privateMethod() -> void;func TestClass(T a, T b) -> TestClass;var c:T;var a:T;var b:T;};class TestStruct{func TestStruct(int a, int b) -> void;func operator==(const TestStruct &other) -> bool;func testMethod(int x, int y) -> int;var c:int;var a:int;var b:int;};enum TestEnum{ENUM_VALUE_1;ENUM_VALUE_2;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_scala() {\n let source = r#\"\n object Main {\n def main(args: Array[String]): Unit = {\n println(\"Hello, World!\")\n }\n }\n\n class TestClass {\n val testVal: String = \"test\"\n var testVar = 42\n\n def testMethod(a: Int, b: Int): Int = {\n a + b\n }\n }\n\n // braceless syntax is also supported\n trait TestTrait:\n def abstractMethod(x: Int): Int\n def concreteMethod(y: Int): Int = y * 2\n\n case class TestCaseClass(name: String, age: Int)\n\n enum TestEnum {\n case First, Second, Third\n }\n\n val foo: TestClass = ???\n \"#;\n\n let definitions = extract_definitions(\"scala\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var foo:TestClass;class Main{func main(args: Array[String]) -> Unit;};class TestCaseClass{};class TestClass{func testMethod(a: Int, b: Int) -> Int;var testVal:String;var testVar;};class TestTrait{func abstractMethod(x: Int) -> Int;func concreteMethod(y: Int) -> Int;};enum TestEnum{First;Second;Third;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_elixir() {\n let source = r#\"\n defmodule TestModule do\n @moduledoc \"\"\"\n This is a test module\n \"\"\"\n\n @test_const \"test\"\n @other_const 123\n\n def test_func(a, b) do\n a + b\n end\n\n defp private_func(x) do\n x * 2\n end\n\n defmacro test_macro(expr) do\n quote do\n unquote(expr)\n end\n end\n end\n\n defmodule AnotherModule do\n def another_func() do\n :ok\n end\n end\n \"#;\n let definitions = extract_definitions(\"elixir\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected =\n \"module AnotherModule{func another_func();};module TestModule{func test_func(a, b);};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_csharp() {\n let source = r#\"\n using System;\n\n namespace TestNamespace;\n\n public class TestClass(TestDependency m)\n {\n\n private int PrivateTestProperty { get; set; }\n\n private int _privateTestField;\n\n public int TestProperty { get; set; }\n\n public string TestField;\n\n public TestClass()\n {\n TestProperty = 0;\n }\n\n\n public void TestMethod(int a, int b)\n {\n var innerVarInMethod = 1;\n return a + b;\n }\n\n public int TestMethod(int a, int b, int c) => a + b + c;\n\n private void PrivateMethod()\n {\n return;\n }\n\n public class MyInnerClass(InnerClassDependency m) {}\n\n public record MyInnerRecord(int a);\n }\n\n public record TestRecord(int a, int b);\n\n public enum TestEnum { Value1, Value2 }\n \"#;\n\n let definitions = extract_definitions(\"csharp\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"class MyInnerClass{func MyInnerClass(InnerClassDependency m) -> MyInnerClass;};class MyInnerRecord{func MyInnerRecord(int a) -> MyInnerRecord;};class TestClass{func TestClass(TestDependency m) -> TestClass;func TestClass() -> TestClass;func TestMethod(int a, int b) -> void;func TestMethod(int a, int b, int c) -> int;var TestProperty:int;var TestField:string;};class TestRecord{func TestRecord(int a, int b) -> TestRecord;};enum TestEnum{Value1;Value2;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_swift() {\n let source = r#\"\n import Foundation\n\n private var myVariable = 0\n public var myPublicVariable = 0\n\n struct MyStruct {\n public var myPublicVariable = 0\n private var myPrivateVariable = 0\n\n func myPublicMethod(with parameter: Int) -> {\n }\n\n private func myPrivateMethod(with parameter: Int) -> {\n }\n }\n\n class MyClass {\n public var myPublicVariable = 0\n private var myPrivateVariable = 0\n\n init(myParameter: Int, myOtherParameter: Int) {\n }\n\n func myPublicMethod(with parameter: Int) -> {\n }\n\n private func myPrivateMethod(with parameter: Int) -> {\n }\n\n func myMethod() {\n print(\"Hello, world!\")\n }\n }\n \"#;\n\n let definitions = extract_definitions(\"swift\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"var myPublicVariable;class MyClass{func init() -> void;func myPublicMethod() -> void;func myMethod() -> void;var myPublicVariable;};class MyStruct{func myPublicMethod() -> void;var myPublicVariable;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_php() {\n let source = r#\"\n \n \"#;\n\n let definitions = extract_definitions(\"php\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"class MyClass{func myPublicMethod($parameter) -> void;func myPrivateMethod($parameter) -> void;func myMethod() -> void;var public $myPublicVariable = 0;;var private $myPrivateVariable = 0;;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_java() {\n let source = r#\"\n public class MyClass {\n public void myPublicMethod(String parameter) {\n System.out.println(\"Hello, world!\");\n }\n\n private void myPrivateMethod(String parameter) {\n System.out.println(\"Hello, world!\");\n }\n\n void myMethod() {\n System.out.println(\"Hello, world!\");\n }\n }\n \"#;\n\n let definitions = extract_definitions(\"java\", source).unwrap();\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected =\n \"class MyClass{func myPublicMethod(String parameter) -> void;func myMethod() -> void;};\";\n assert_eq!(stringified, expected);\n }\n\n #[test]\n fn test_unsupported_language() {\n let source = \"print('Hello, world!')\";\n let definitions = extract_definitions(\"unknown\", source).unwrap();\n\n let stringified = stringify_definitions(&definitions);\n println!(\"{stringified}\");\n let expected = \"\";\n assert_eq!(stringified, expected);\n }\n}\n" }, { "path": "crates/avante-templates/Cargo.toml", "content": "[lib]\ncrate-type = [\"cdylib\"]\n\n[package]\nname = \"avante-templates\"\nedition.workspace = true\nrust-version.workspace = true\nlicense.workspace = true\nversion.workspace = true\n\n[dependencies]\nmlua = { workspace = true }\nminijinja = { workspace = true }\nserde = { workspace = true, features = [\"derive\"] }\n\n[lints]\nworkspace = true\n\n[features]\nlua51 = [\"mlua/lua51\"]\nlua52 = [\"mlua/lua52\"]\nlua53 = [\"mlua/lua53\"]\nlua54 = [\"mlua/lua54\"]\nluajit = [\"mlua/luajit\"]\n" }, { "path": "crates/avante-templates/src/lib.rs", "content": "use minijinja::{context, Environment};\nuse mlua::prelude::*;\nuse serde::{Deserialize, Serialize};\nuse std::path::Path;\nuse std::sync::{Arc, Mutex};\n\nstruct State<'a> {\n environment: Mutex>>,\n}\n\nimpl State<'_> {\n fn new() -> Self {\n State {\n environment: Mutex::new(None),\n }\n }\n}\n\n#[derive(Debug, Serialize, Deserialize)]\nstruct SelectedCode {\n path: String,\n content: Option,\n file_type: String,\n}\n\n#[derive(Debug, Serialize, Deserialize)]\nstruct SelectedFile {\n path: String,\n content: Option,\n file_type: String,\n}\n\n#[derive(Debug, Serialize, Deserialize)]\nstruct TemplateContext {\n ask: bool,\n code_lang: String,\n selected_files: Option>,\n selected_code: Option,\n recently_viewed_files: Option>,\n relevant_files: Option>,\n project_context: Option,\n diagnostics: Option,\n system_info: Option,\n model_name: Option,\n memory: Option,\n todos: Option,\n enable_fastapply: Option,\n use_react_prompt: Option,\n}\n\n// Given the file name registered after add, the context table in Lua, resulted in a formatted\n// Lua string\n#[allow(clippy::needless_pass_by_value)]\nfn render(state: &State, template: &str, context: TemplateContext) -> LuaResult {\n let environment = state.environment.lock().unwrap();\n match environment.as_ref() {\n Some(environment) => {\n let jinja_template = environment\n .get_template(template)\n .map_err(LuaError::external)\n .unwrap();\n\n Ok(jinja_template\n .render(context! {\n ask => context.ask,\n code_lang => context.code_lang,\n selected_files => context.selected_files,\n selected_code => context.selected_code,\n recently_viewed_files => context.recently_viewed_files,\n relevant_files => context.relevant_files,\n project_context => context.project_context,\n diagnostics => context.diagnostics,\n system_info => context.system_info,\n model_name => context.model_name,\n memory => context.memory,\n todos => context.todos,\n enable_fastapply => context.enable_fastapply,\n use_react_prompt => context.use_react_prompt,\n })\n .map_err(LuaError::external)\n .unwrap())\n }\n None => Err(LuaError::RuntimeError(\n \"Environment not initialized\".to_string(),\n )),\n }\n}\n\nfn initialize(state: &State, cache_directory: String, project_directory: String) {\n let mut environment_mutex = state.environment.lock().unwrap();\n let mut env = Environment::new();\n\n // Create a custom loader that searches both cache and project directories\n let cache_dir = cache_directory.clone();\n let project_dir = project_directory.clone();\n\n env.set_loader(\n move |name: &str| -> Result, minijinja::Error> {\n // First try the cache directory (for built-in templates)\n let cache_path = Path::new(&cache_dir).join(name);\n if cache_path.exists() {\n match std::fs::read_to_string(&cache_path) {\n Ok(content) => return Ok(Some(content)),\n Err(_) => {} // Continue to try project directory\n }\n }\n\n // Then try the project directory (for custom includes)\n let project_path = Path::new(&project_dir).join(name);\n if project_path.exists() {\n match std::fs::read_to_string(&project_path) {\n Ok(content) => return Ok(Some(content)),\n Err(_) => {} // File not found or read error\n }\n }\n\n // Template not found in either directory\n Ok(None)\n },\n );\n\n *environment_mutex = Some(env);\n}\n\n#[mlua::lua_module]\nfn avante_templates(lua: &Lua) -> LuaResult {\n let core = State::new();\n let state = Arc::new(core);\n let state_clone = Arc::clone(&state);\n\n let exports = lua.create_table()?;\n exports.set(\n \"initialize\",\n lua.create_function(\n move |_, (cache_directory, project_directory): (String, String)| {\n initialize(&state, cache_directory, project_directory);\n Ok(())\n },\n )?,\n )?;\n exports.set(\n \"render\",\n lua.create_function_mut(move |lua, (template, context): (String, LuaValue)| {\n let ctx = lua.from_value(context)?;\n render(&state_clone, template.as_str(), ctx)\n })?,\n )?;\n Ok(exports)\n}\n" }, { "path": "crates/avante-tokenizers/Cargo.toml", "content": "[lib]\ncrate-type = [\"cdylib\"]\n\n[package]\nname = \"avante-tokenizers\"\nedition = { workspace = true }\nversion = { workspace = true }\nrust-version = { workspace = true }\nlicense = { workspace = true }\n\n[lints]\nworkspace = true\n\n[dependencies]\ndirs = \"5.0.1\"\nregex = \"1.11.1\"\nhf-hub = { git = \"https://github.com/yetone/hf-hub\", branch='main', features = [\"default\", \"ureq\"] }\nureq = { version = \"2.10.1\", features = [\"json\", \"socks-proxy\"] }\nmlua = { workspace = true }\ntiktoken-rs = { workspace = true }\ntokenizers = { workspace = true }\n\n[features]\nlua51 = [\"mlua/lua51\"]\nlua52 = [\"mlua/lua52\"]\nlua53 = [\"mlua/lua53\"]\nlua54 = [\"mlua/lua54\"]\nluajit = [\"mlua/luajit\"]\n" }, { "path": "crates/avante-tokenizers/README.md", "content": "A simple crate to unify hf/tokenizers and tiktoken-rs\n" }, { "path": "crates/avante-tokenizers/src/lib.rs", "content": "use hf_hub::{api::sync::ApiBuilder, Repo, RepoType};\nuse mlua::prelude::*;\nuse regex::Regex;\nuse std::path::PathBuf;\nuse std::sync::{Arc, Mutex};\nuse tiktoken_rs::{get_bpe_from_model, CoreBPE};\nuse tokenizers::Tokenizer;\n\nstruct Tiktoken {\n bpe: CoreBPE,\n}\n\nimpl Tiktoken {\n fn new(model: &str) -> Self {\n let bpe = get_bpe_from_model(model).unwrap();\n Self { bpe }\n }\n\n fn encode(&self, text: &str) -> (Vec, usize, usize) {\n let tokens = self.bpe.encode_with_special_tokens(text);\n let num_tokens = tokens.len();\n let num_chars = text.chars().count();\n (tokens, num_tokens, num_chars)\n }\n}\n\nstruct HuggingFaceTokenizer {\n tokenizer: Tokenizer,\n}\n\nfn is_valid_url(url: &str) -> bool {\n let url_regex = Regex::new(r\"^https?://[^\\s/$.?#].[^\\s]*$\").unwrap();\n url_regex.is_match(url)\n}\n\nimpl HuggingFaceTokenizer {\n fn new(model: &str) -> Self {\n let tokenizer_path = if is_valid_url(model) {\n Self::get_cached_tokenizer(model)\n } else {\n // Use existing HuggingFace Hub logic for model names\n let identifier = model.to_string();\n let api = ApiBuilder::new().with_progress(false).build().unwrap();\n let repo = Repo::new(identifier, RepoType::Model);\n let api = api.repo(repo);\n api.get(\"tokenizer.json\").unwrap()\n };\n\n let tokenizer = Tokenizer::from_file(tokenizer_path).unwrap();\n Self { tokenizer }\n }\n\n fn encode(&self, text: &str) -> (Vec, usize, usize) {\n let encoding = self.tokenizer.encode(text, false).unwrap();\n let tokens = encoding.get_ids().to_vec();\n let num_tokens = tokens.len();\n let num_chars = encoding.get_offsets().last().unwrap().1;\n (tokens, num_tokens, num_chars)\n }\n\n fn get_cached_tokenizer(url: &str) -> PathBuf {\n let cache_dir = dirs::home_dir()\n .map(|h| h.join(\".cache\").join(\"avante\"))\n .unwrap();\n std::fs::create_dir_all(&cache_dir).unwrap();\n\n // Extract filename from URL\n let filename = url.split('/').last().unwrap();\n\n let cached_path = cache_dir.join(filename);\n\n if !cached_path.exists() {\n let response = ureq::get(url).call().unwrap();\n let mut file = std::fs::File::create(&cached_path).unwrap();\n let mut reader = response.into_reader();\n std::io::copy(&mut reader, &mut file).unwrap();\n }\n cached_path\n }\n}\n\nenum TokenizerType {\n Tiktoken(Tiktoken),\n HuggingFace(Box),\n}\n\nstruct State {\n tokenizer: Mutex>,\n}\n\nimpl State {\n fn new() -> Self {\n State {\n tokenizer: Mutex::new(None),\n }\n }\n}\n\nfn encode(state: &State, text: &str) -> LuaResult<(Vec, usize, usize)> {\n let tokenizer = state.tokenizer.lock().unwrap();\n match tokenizer.as_ref() {\n Some(TokenizerType::Tiktoken(tokenizer)) => Ok(tokenizer.encode(text)),\n Some(TokenizerType::HuggingFace(tokenizer)) => Ok(tokenizer.encode(text)),\n None => Err(LuaError::RuntimeError(\n \"Tokenizer not initialized\".to_string(),\n )),\n }\n}\n\nfn from_pretrained(state: &State, model: &str) {\n let mut tokenizer_mutex = state.tokenizer.lock().unwrap();\n *tokenizer_mutex = Some(match model {\n \"gpt-4o\" => TokenizerType::Tiktoken(Tiktoken::new(model)),\n _ => TokenizerType::HuggingFace(Box::new(HuggingFaceTokenizer::new(model))),\n });\n}\n\n#[mlua::lua_module]\nfn avante_tokenizers(lua: &Lua) -> LuaResult {\n let core = State::new();\n let state = Arc::new(core);\n let state_clone = Arc::clone(&state);\n\n let exports = lua.create_table()?;\n exports.set(\n \"from_pretrained\",\n lua.create_function(move |_, model: String| {\n from_pretrained(&state, model.as_str());\n Ok(())\n })?,\n )?;\n exports.set(\n \"encode\",\n lua.create_function(move |_, text: String| encode(&state_clone, text.as_str()))?,\n )?;\n Ok(exports)\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_tiktoken() {\n let model = \"gpt-4o\";\n let source = \"Hello, world!\";\n let tokenizer = Tiktoken::new(model);\n let (tokens, num_tokens, num_chars) = tokenizer.encode(source);\n assert_eq!(tokens, vec![13225, 11, 2375, 0]);\n assert_eq!(num_tokens, 4);\n assert_eq!(num_chars, source.chars().count());\n }\n\n #[test]\n fn test_hf() {\n let model = \"gpt2\";\n let source = \"Hello, world!\";\n let tokenizer = HuggingFaceTokenizer::new(model);\n let (tokens, num_tokens, num_chars) = tokenizer.encode(source);\n assert_eq!(tokens, vec![15496, 11, 995, 0]);\n assert_eq!(num_tokens, 4);\n assert_eq!(num_chars, source.chars().count());\n }\n\n #[test]\n fn test_roundtrip() {\n let state = State::new();\n let source = \"Hello, world!\";\n let model = \"gpt2\";\n\n from_pretrained(&state, model);\n let (tokens, num_tokens, num_chars) = encode(&state, \"Hello, world!\").unwrap();\n assert_eq!(tokens, vec![15496, 11, 995, 0]);\n assert_eq!(num_tokens, 4);\n assert_eq!(num_chars, source.chars().count());\n }\n\n // For example: https://storage.googleapis.com/cohere-public/tokenizers/command-r-08-2024.json\n // Disable testing on GitHub Actions to avoid rate limiting and file size limits\n #[test]\n fn test_public_url() {\n if std::env::var(\"GITHUB_ACTIONS\").is_ok() {\n return;\n }\n let state = State::new();\n let source = \"Hello, world!\";\n let model =\n \"https://storage.googleapis.com/cohere-public/tokenizers/command-r-08-2024.json\";\n\n from_pretrained(&state, model);\n let (tokens, num_tokens, num_chars) = encode(&state, \"Hello, world!\").unwrap();\n assert_eq!(tokens, vec![28339, 19, 3845, 8]);\n assert_eq!(num_tokens, 4);\n assert_eq!(num_chars, source.chars().count());\n }\n}\n" }, { "path": "lua/avante/api.lua", "content": "local Config = require(\"avante.config\")\nlocal Utils = require(\"avante.utils\")\nlocal PromptInput = require(\"avante.ui.prompt_input\")\n\n---@class avante.ApiToggle\n---@operator call(): boolean\n---@field debug ToggleBind.wrap\n---@field hint ToggleBind.wrap\n\n---@class avante.Api\n---@field toggle avante.ApiToggle\nlocal M = {}\n\n---@param target_provider avante.SelectorProvider\nfunction M.switch_selector_provider(target_provider)\n require(\"avante.config\").override({\n selector = {\n provider = target_provider,\n },\n })\nend\n\n---@param target_provider avante.InputProvider\nfunction M.switch_input_provider(target_provider)\n require(\"avante.config\").override({\n input = {\n provider = target_provider,\n },\n })\nend\n\n---@param target avante.ProviderName\nfunction M.switch_provider(target) require(\"avante.providers\").refresh(target) end\n\n---@param path string\nlocal function to_windows_path(path)\n local winpath = path:gsub(\"/\", \"\\\\\")\n\n if winpath:match(\"^%a:\") then winpath = winpath:sub(1, 2):upper() .. winpath:sub(3) end\n\n winpath = winpath:gsub(\"\\\\$\", \"\")\n\n return winpath\nend\n\n---@param opts? {source: boolean}\nfunction M.build(opts)\n opts = opts or { source = true }\n local dirname = Utils.trim(string.sub(debug.getinfo(1).source, 2, #\"/init.lua\" * -1), { suffix = \"/\" })\n local git_root = vim.fs.find(\".git\", { path = dirname, upward = true })[1]\n local build_directory = git_root and vim.fn.fnamemodify(git_root, \":h\") or (dirname .. \"/../../\")\n\n if opts.source and not vim.fn.executable(\"cargo\") then\n error(\"Building avante.nvim requires cargo to be installed.\", 2)\n end\n\n ---@type string[]\n local cmd\n local os_name = Utils.get_os_name()\n\n if vim.tbl_contains({ \"linux\", \"darwin\" }, os_name) then\n cmd = {\n \"sh\",\n \"-c\",\n string.format(\"make BUILD_FROM_SOURCE=%s -C %s\", opts.source == true and \"true\" or \"false\", build_directory),\n }\n elseif os_name == \"windows\" then\n build_directory = to_windows_path(build_directory)\n cmd = {\n \"powershell\",\n \"-ExecutionPolicy\",\n \"Bypass\",\n \"-File\",\n string.format(\"%s\\\\Build.ps1\", build_directory),\n \"-WorkingDirectory\",\n build_directory,\n \"-BuildFromSource\",\n string.format(\"%s\", opts.source == true and \"true\" or \"false\"),\n }\n else\n error(\"Unsupported operating system: \" .. os_name, 2)\n end\n\n ---@type integer\n local pid\n local exit_code = { 0 }\n\n local ok, job_or_err = pcall(vim.system, cmd, { text = true }, function(obj)\n local stderr = obj.stderr and vim.split(obj.stderr, \"\\n\") or {}\n local stdout = obj.stdout and vim.split(obj.stdout, \"\\n\") or {}\n if vim.tbl_contains(exit_code, obj.code) then\n local output = stdout\n if #output == 0 then\n table.insert(output, \"\")\n Utils.debug(\"build output:\", output)\n else\n Utils.debug(\"build error:\", stderr)\n end\n end\n end)\n if not ok then Utils.error(\"Failed to build the command: \" .. cmd .. \"\\n\" .. job_or_err, { once = true }) end\n pid = job_or_err.pid\n return pid\nend\n\n---@class AskOptions\n---@field question? string optional questions\n---@field win? table windows options similar to |nvim_open_win()|\n---@field ask? boolean\n---@field floating? boolean whether to open a floating input to enter the question\n---@field new_chat? boolean whether to open a new chat\n---@field without_selection? boolean whether to open a new chat without selection\n---@field sidebar_pre_render? fun(sidebar: avante.Sidebar)\n---@field sidebar_post_render? fun(sidebar: avante.Sidebar)\n---@field project_root? string optional project root\n---@field show_logo? boolean whether to show the logo\n\nfunction M.full_view_ask()\n M.ask({\n show_logo = true,\n sidebar_post_render = function(sidebar)\n sidebar:toggle_code_window()\n -- vim.wo[sidebar.containers.result.winid].number = true\n -- vim.wo[sidebar.containers.result.winid].relativenumber = true\n end,\n })\nend\n\nM.zen_mode = M.full_view_ask\n\n---@param opts? AskOptions\nfunction M.ask(opts)\n opts = opts or {}\n Config.ask_opts = opts\n if type(opts) == \"string\" then\n Utils.warn(\"passing 'ask' as string is deprecated, do {question = '...'} instead\", { once = true })\n opts = { question = opts }\n end\n\n local has_question = opts.question ~= nil and opts.question ~= \"\"\n local new_chat = opts.new_chat == true\n\n if Utils.is_sidebar_buffer(0) and not has_question and not new_chat then\n require(\"avante\").close_sidebar()\n return false\n end\n\n opts = vim.tbl_extend(\"force\", { selection = Utils.get_visual_selection_and_range() }, opts)\n\n ---@param input string | nil\n local function ask(input)\n if input == nil or input == \"\" then input = opts.question end\n local sidebar = require(\"avante\").get()\n if sidebar and sidebar:is_open() and sidebar.code.bufnr ~= vim.api.nvim_get_current_buf() then\n sidebar:close({ goto_code_win = false })\n end\n require(\"avante\").open_sidebar(opts)\n sidebar = require(\"avante\").get()\n if new_chat then sidebar:new_chat() end\n if opts.without_selection then\n sidebar.code.selection = nil\n sidebar.file_selector:reset()\n if sidebar.containers.selected_files then sidebar.containers.selected_files:unmount() end\n end\n if input == nil or input == \"\" then return true end\n vim.api.nvim_exec_autocmds(\"User\", { pattern = \"AvanteInputSubmitted\", data = { request = input } })\n return true\n end\n\n if opts.floating == true or (Config.windows.ask.floating == true and not has_question and opts.floating == nil) then\n local prompt_input = PromptInput:new({\n submit_callback = function(input) ask(input) end,\n close_on_submit = true,\n win_opts = {\n border = Config.windows.ask.border,\n title = { { \"Avante Ask\", \"FloatTitle\" } },\n },\n start_insert = Config.windows.ask.start_insert,\n default_value = opts.question,\n })\n prompt_input:open()\n return true\n end\n\n return ask()\nend\n\n---@param request? string\n---@param line1? integer\n---@param line2? integer\nfunction M.edit(request, line1, line2)\n local _, selection = require(\"avante\").get()\n if not selection then require(\"avante\")._init(vim.api.nvim_get_current_tabpage()) end\n _, selection = require(\"avante\").get()\n if not selection then return end\n selection:create_editing_input(request, line1, line2)\n if request ~= nil and request ~= \"\" then\n vim.api.nvim_exec_autocmds(\"User\", { pattern = \"AvanteEditSubmitted\", data = { request = request } })\n end\nend\n\n---@return avante.Suggestion | nil\nfunction M.get_suggestion()\n local _, _, suggestion = require(\"avante\").get()\n return suggestion\nend\n\n---@param opts? AskOptions\nfunction M.refresh(opts)\n opts = opts or {}\n local sidebar = require(\"avante\").get()\n if not sidebar then return end\n if not sidebar:is_open() then return end\n local curbuf = vim.api.nvim_get_current_buf()\n\n local focused = sidebar.containers.result.bufnr == curbuf or sidebar.containers.input.bufnr == curbuf\n if focused or not sidebar:is_open() then return end\n local listed = vim.api.nvim_get_option_value(\"buflisted\", { buf = curbuf })\n\n if Utils.is_sidebar_buffer(curbuf) or not listed then return end\n\n local curwin = vim.api.nvim_get_current_win()\n\n sidebar:close()\n sidebar.code.winid = curwin\n sidebar.code.bufnr = curbuf\n sidebar:render(opts)\nend\n\n---@param opts? AskOptions\nfunction M.focus(opts)\n opts = opts or {}\n local sidebar = require(\"avante\").get()\n if not sidebar then return end\n\n local curbuf = vim.api.nvim_get_current_buf()\n local curwin = vim.api.nvim_get_current_win()\n\n if sidebar:is_open() then\n if curbuf == sidebar.containers.input.bufnr then\n if sidebar.code.winid and sidebar.code.winid ~= curwin then vim.api.nvim_set_current_win(sidebar.code.winid) end\n elseif curbuf == sidebar.containers.result.bufnr then\n if sidebar.code.winid and sidebar.code.winid ~= curwin then vim.api.nvim_set_current_win(sidebar.code.winid) end\n else\n if sidebar.containers.input.winid and sidebar.containers.input.winid ~= curwin then\n vim.api.nvim_set_current_win(sidebar.containers.input.winid)\n end\n end\n else\n if sidebar.code.winid then vim.api.nvim_set_current_win(sidebar.code.winid) end\n ---@cast opts SidebarOpenOptions\n sidebar:open(opts)\n if sidebar.containers.input.winid then vim.api.nvim_set_current_win(sidebar.containers.input.winid) end\n end\nend\n\nfunction M.select_model() require(\"avante.model_selector\").open() end\n\nfunction M.select_history()\n local buf = vim.api.nvim_get_current_buf()\n require(\"avante.history_selector\").open(buf, function(filename)\n vim.api.nvim_buf_call(buf, function()\n if not require(\"avante\").is_sidebar_open() then require(\"avante\").open_sidebar({}) end\n local Path = require(\"avante.path\")\n Path.history.save_latest_filename(buf, filename)\n local sidebar = require(\"avante\").get()\n sidebar:update_content_with_history()\n sidebar:create_todos_container()\n sidebar:initialize_token_count()\n vim.schedule(function() sidebar:focus_input() end)\n end)\n end)\nend\n\nfunction M.add_buffer_files()\n local sidebar = require(\"avante\").get()\n if not sidebar then\n require(\"avante.api\").ask()\n sidebar = require(\"avante\").get()\n end\n if not sidebar:is_open() then sidebar:open({}) end\n sidebar.file_selector:add_buffer_files()\nend\n\nfunction M.add_selected_file(filepath)\n local rel_path = Utils.uniform_path(filepath)\n\n local sidebar = require(\"avante\").get()\n if not sidebar then\n require(\"avante.api\").ask()\n sidebar = require(\"avante\").get()\n end\n if not sidebar:is_open() then sidebar:open({}) end\n sidebar.file_selector:add_selected_file(rel_path)\nend\n\nfunction M.remove_selected_file(filepath)\n ---@diagnostic disable-next-line: undefined-field\n local stat = vim.uv.fs_stat(filepath)\n local files\n if stat and stat.type == \"directory\" then\n files = Utils.scan_directory({ directory = filepath, add_dirs = true })\n else\n files = { filepath }\n end\n\n local sidebar = require(\"avante\").get()\n if not sidebar then\n require(\"avante.api\").ask()\n sidebar = require(\"avante\").get()\n end\n if not sidebar:is_open() then sidebar:open({}) end\n\n for _, file in ipairs(files) do\n local rel_path = Utils.uniform_path(file)\n sidebar.file_selector:remove_selected_file(rel_path)\n end\nend\n\nfunction M.stop() require(\"avante.llm\").cancel_inflight_request() end\n\nreturn setmetatable(M, {\n __index = function(t, k)\n local module = require(\"avante\")\n ---@class AvailableApi: ApiCaller\n ---@field api? boolean\n local has = module[k]\n if type(has) ~= \"table\" or not has.api then\n Utils.warn(k .. \" is not a valid avante's API method\", { once = true })\n return\n end\n t[k] = has\n return t[k]\n end,\n}) --[[@as avante.Api]]\n" }, { "path": "lua/avante/auth/pkce.lua", "content": "local M = {}\nlocal uv = vim.uv\n\n-- Generates sha256 bytes with vim.fn\nlocal function sha256_bytes(data)\n -- vim.fn.sha256 returns hex string (64 chars)\n local hex = vim.fn.sha256(data)\n -- vim.text.hexdecode returns raw bytes\n return vim.text.hexdecode(hex)\nend\n\nlocal function windows_random_bytes(n)\n local ps = [[\n $bytes = New-Object byte[] (]] .. n .. [[);\n [System.Security.Cryptography.RandomNumberGenerator]::Create().GetBytes($bytes);\n [Convert]::ToBase64String($bytes)\n ]]\n\n local result = vim.system({ \"powershell\", \"-NoProfile\", \"-Command\", ps }):wait()\n if result.code ~= 0 then return nil, result.stderr end\n\n local decoded = vim.base64.decode(vim.trim(result.stdout))\n if not decoded or #decoded ~= n then return nil, \"failed to decode bytes\" end\n\n return decoded, nil\nend\n\n-- Reads random bytes from urandom\nlocal function random_bytes_urandom(n)\n local path = \"/dev/urandom\"\n local fd, open_err = uv.fs_open(path, \"r\", 438) -- 0666; ignored on most systems for read\n if not fd then return nil, (\"uv.fs_open(%s) failed: %s\"):format(path, tostring(open_err)) end\n\n local chunk, read_err = uv.fs_read(fd, n, 0)\n uv.fs_close(fd)\n\n if not chunk then return nil, (\"uv.fs_read(%s) failed: %s\"):format(path, tostring(read_err)) end\n if #chunk ~= n then return nil, (\"short read from %s: wanted %d got %d\"):format(path, n, #chunk) end\n return chunk, nil\nend\n\n---Generates a random N number of bytes using crypto lib over ffi, falling back to urandom\n---@param n integer number of bytes to generate\n---@return string|nil bytes string of bytes generated, or nil if all methods fail\n---@return string|nil error error message if generation failed\nlocal function get_random_bytes(n)\n if type(uv.random) == \"function\" then\n local ok, err_or_bytes, maybe_bytes = pcall(uv.random, n)\n if ok then\n if type(err_or_bytes) == \"string\" and maybe_bytes == nil then\n if #err_or_bytes == n then return err_or_bytes end\n else\n local err = err_or_bytes\n local bytes = maybe_bytes\n if err == 0 and type(bytes) == \"string\" and #bytes == n then return bytes end\n end\n end\n end\n\n -- Fallback\n if vim.uv.os_uname().sysname ~= \"Windows_NT\" then\n local bytes, err = random_bytes_urandom(n)\n if err ~= nil or #bytes ~= n then\n return nil, err or \"Failed to generate random bytes using urandom\"\n else\n return bytes, nil\n end\n else\n local bytes, err = windows_random_bytes(n)\n if err ~= nil or #bytes ~= n then\n return nil, err or \"Failed to generate random bytes using powershell\"\n else\n return bytes, nil\n end\n end\nend\n\n--- URL-safe base64\n--- @param data string value to base64 encode\n--- @return string base64String base64 encoded string\nlocal function base64url_encode(data)\n local b64 = vim.base64.encode(data)\n local b64_string, _ = b64:gsub(\"+\", \"-\"):gsub(\"/\", \"_\"):gsub(\"=\", \"\")\n return b64_string\nend\n\n-- Generate code_verifier (43-128 characters)\n--- @return string|nil verifier String representing pkce verifier or nil if generation fails\n--- @return string|nil error error message if generation failed\nfunction M.generate_verifier()\n local bytes, err = get_random_bytes(32) -- 256 bits\n if bytes then return base64url_encode(bytes), nil end\n\n return nil, err or \"Failed to generate random bytes\"\nend\n\n-- Generate code_challenge (S256 method)\n---@return string|nil challenge String representing pkce challenge or nil if generation fails\n---@return string|nil error error message if generation failed\nfunction M.generate_challenge(verifier) return base64url_encode(sha256_bytes(verifier)), nil end\n\nreturn M\n" }, { "path": "lua/avante/clipboard.lua", "content": "---NOTE: this module is inspired by https://github.com/HakonHarnes/img-clip.nvim/tree/main\n---@see https://github.com/ekickx/clipboard-image.nvim/blob/main/lua/clipboard-image/paste.lua\n\nlocal Path = require(\"plenary.path\")\nlocal Utils = require(\"avante.utils\")\nlocal Config = require(\"avante.config\")\n---@module \"img-clip\"\nlocal ImgClip = nil\n\n---@class AvanteClipboard\n---@field get_base64_content fun(filepath: string): string | nil\n---\n---@class avante.Clipboard: AvanteClipboard\nlocal M = {}\n\n---@type Path\nlocal paste_directory = nil\n\n---@return Path\nlocal function get_paste_directory()\n if paste_directory then return paste_directory end\n paste_directory = Path:new(Config.history.storage_path):joinpath(\"pasted_images\")\n return paste_directory\nend\n\nM.support_paste_image = Config.support_paste_image\n\nfunction M.setup()\n get_paste_directory()\n\n if not paste_directory:exists() then paste_directory:mkdir({ parent = true }) end\n\n if M.support_paste_image() and ImgClip == nil then ImgClip = require(\"img-clip\") end\nend\n\n---@param line? string\nfunction M.paste_image(line)\n line = line or nil\n if not Config.support_paste_image() then return false end\n\n local opts = {\n dir_path = paste_directory:absolute(),\n prompt_for_file_name = false,\n filetypes = {\n AvanteInput = Config.img_paste,\n },\n }\n\n if vim.fn.has(\"wsl\") > 0 or vim.fn.has(\"win32\") > 0 then opts.use_absolute_path = true end\n\n ---@diagnostic disable-next-line: need-check-nil, undefined-field\n return ImgClip.paste_image(opts, line)\nend\n\n---@param filepath string\nfunction M.get_base64_content(filepath)\n local os_mapping = Utils.get_os_name()\n ---@type vim.SystemCompleted\n local output\n if os_mapping == \"darwin\" or os_mapping == \"linux\" then\n output = Utils.shell_run((\"cat %s | base64 | tr -d '\\n'\"):format(filepath))\n else\n output =\n Utils.shell_run((\"([Convert]::ToBase64String([IO.File]::ReadAllBytes('%s')) -replace '`r`n')\"):format(filepath))\n end\n if output.code == 0 then\n return output.stdout\n else\n error(\"Failed to convert image to base64\")\n end\nend\n\nreturn M\n" }, { "path": "lua/avante/config.lua", "content": "---NOTE: user will be merged with defaults and\n---we add a default var_accessor for this table to config values.\n\n---@alias WebSearchEngineProviderResponseBodyFormatter fun(body: table): (string, string?)\n---@alias avante.InputProvider \"native\" | \"dressing\" | \"snacks\" | fun(input: avante.ui.Input): nil\n\nlocal Utils = require(\"avante.utils\")\n\nlocal function copilot_use_response_api(opts)\n local model = opts and opts.model\n return type(model) == \"string\" and model:match(\"gpt%-%d+%.?%d*%-codex\") ~= nil\nend\n\n---@class avante.file_selector.IParams\n---@field public title string\n---@field public filepaths string[]\n---@field public handler fun(filepaths: string[]|nil): nil\n\n---@class avante.file_selector.opts.IGetFilepathsParams\n---@field public cwd string\n---@field public selected_filepaths string[]\n\n---@class avante.CoreConfig: avante.Config\nlocal M = {}\n\n--- Default configuration for project-specific instruction file\nM.instructions_file = \"avante.md\"\n---@class avante.Config\nM._defaults = {\n debug = false,\n ---@alias avante.Mode \"agentic\" | \"legacy\"\n ---@type avante.Mode\n mode = \"agentic\",\n ---@alias avante.ProviderName \"claude\" | \"openai\" | \"azure\" | \"gemini\" | \"vertex\" | \"cohere\" | \"copilot\" | \"bedrock\" | \"ollama\" | \"watsonx_code_assistant\" | \"mistral\" | string\n ---@type avante.ProviderName\n provider = \"claude\",\n -- WARNING: Since auto-suggestions are a high-frequency operation and therefore expensive,\n -- currently designating it as `copilot` provider is dangerous because: https://github.com/yetone/avante.nvim/issues/1048\n -- Of course, you can reduce the request frequency by increasing `suggestion.debounce`.\n auto_suggestions_provider = nil,\n memory_summary_provider = nil,\n ---@alias Tokenizer \"tiktoken\" | \"hf\"\n ---@type Tokenizer\n -- Used for counting tokens and encoding text.\n -- By default, we will use tiktoken.\n -- For most providers that we support we will determine this automatically.\n -- If you wish to use a given implementation, then you can override it here.\n tokenizer = \"tiktoken\",\n ---@type string | fun(): string | nil\n system_prompt = nil,\n ---@type string | fun(): string | nil\n override_prompt_dir = nil,\n rules = {\n project_dir = nil, ---@type string | nil (could be relative dirpath)\n global_dir = nil, ---@type string | nil (absolute dirpath)\n },\n rag_service = { -- RAG service configuration\n enabled = false, -- Enables the RAG service\n host_mount = os.getenv(\"HOME\"), -- Host mount path for the RAG service (Docker will mount this path)\n runner = \"docker\", -- The runner for the RAG service (can use docker or nix)\n -- The image to use to run the rag service if runner is docker\n image = \"quay.io/yetoneful/avante-rag-service:0.0.11\",\n llm = { -- Configuration for the Language Model (LLM) used by the RAG service\n provider = \"openai\", -- The LLM provider\n endpoint = \"https://api.openai.com/v1\", -- The LLM API endpoint\n api_key = \"OPENAI_API_KEY\", -- The environment variable name for the LLM API key\n model = \"gpt-4o-mini\", -- The LLM model name\n extra = nil, -- Extra configuration options for the LLM\n },\n embed = { -- Configuration for the Embedding model used by the RAG service\n provider = \"openai\", -- The embedding provider\n endpoint = \"https://api.openai.com/v1\", -- The embedding API endpoint\n api_key = \"OPENAI_API_KEY\", -- The environment variable name for the embedding API key\n model = \"text-embedding-3-large\", -- The embedding model name\n extra = nil, -- Extra configuration options for the embedding model\n },\n docker_extra_args = \"\", -- Extra arguments to pass to the docker command\n },\n web_search_engine = {\n provider = \"tavily\",\n proxy = nil,\n providers = {\n tavily = {\n api_key_name = \"TAVILY_API_KEY\",\n extra_request_body = {\n include_answer = \"basic\",\n },\n ---@type WebSearchEngineProviderResponseBodyFormatter\n format_response_body = function(body) return body.answer, nil end,\n },\n serpapi = {\n api_key_name = \"SERPAPI_API_KEY\",\n extra_request_body = {\n engine = \"google\",\n google_domain = \"google.com\",\n },\n ---@type WebSearchEngineProviderResponseBodyFormatter\n format_response_body = function(body)\n if body.answer_box ~= nil and body.answer_box.result ~= nil then return body.answer_box.result, nil end\n if body.organic_results ~= nil then\n local jsn = vim\n .iter(body.organic_results)\n :map(\n function(result)\n return {\n title = result.title,\n link = result.link,\n snippet = result.snippet,\n date = result.date,\n }\n end\n )\n :take(10)\n :totable()\n return vim.json.encode(jsn), nil\n end\n return \"\", nil\n end,\n },\n searchapi = {\n api_key_name = \"SEARCHAPI_API_KEY\",\n extra_request_body = {\n engine = \"google\",\n },\n ---@type WebSearchEngineProviderResponseBodyFormatter\n format_response_body = function(body)\n if body.answer_box ~= nil then return body.answer_box.result, nil end\n if body.organic_results ~= nil then\n local jsn = vim\n .iter(body.organic_results)\n :map(\n function(result)\n return {\n title = result.title,\n link = result.link,\n snippet = result.snippet,\n date = result.date,\n }\n end\n )\n :take(10)\n :totable()\n return vim.json.encode(jsn), nil\n end\n return \"\", nil\n end,\n },\n google = {\n api_key_name = \"GOOGLE_SEARCH_API_KEY\",\n engine_id_name = \"GOOGLE_SEARCH_ENGINE_ID\",\n extra_request_body = {},\n ---@type WebSearchEngineProviderResponseBodyFormatter\n format_response_body = function(body)\n if body.items ~= nil then\n local jsn = vim\n .iter(body.items)\n :map(\n function(result)\n return {\n title = result.title,\n link = result.link,\n snippet = result.snippet,\n }\n end\n )\n :take(10)\n :totable()\n return vim.json.encode(jsn), nil\n end\n return \"\", nil\n end,\n },\n kagi = {\n api_key_name = \"KAGI_API_KEY\",\n extra_request_body = {\n limit = \"10\",\n },\n ---@type WebSearchEngineProviderResponseBodyFormatter\n format_response_body = function(body)\n if body.data ~= nil then\n local jsn = vim\n .iter(body.data)\n -- search results only\n :filter(function(result) return result.t == 0 end)\n :map(\n function(result)\n return {\n title = result.title,\n url = result.url,\n snippet = result.snippet,\n }\n end\n )\n :take(10)\n :totable()\n return vim.json.encode(jsn), nil\n end\n return \"\", nil\n end,\n },\n brave = {\n api_key_name = \"BRAVE_API_KEY\",\n extra_request_body = {\n count = \"10\",\n result_filter = \"web\",\n },\n format_response_body = function(body)\n if body.web == nil then return \"\", nil end\n\n local jsn = vim.iter(body.web.results):map(\n function(result)\n return {\n title = result.title,\n url = result.url,\n snippet = result.description,\n }\n end\n )\n\n return vim.json.encode(jsn), nil\n end,\n },\n searxng = {\n api_url_name = \"SEARXNG_API_URL\",\n extra_request_body = {\n format = \"json\",\n },\n ---@type WebSearchEngineProviderResponseBodyFormatter\n format_response_body = function(body)\n if body.results == nil then return \"\", nil end\n\n local jsn = vim.iter(body.results):map(\n function(result)\n return {\n title = result.title,\n url = result.url,\n snippet = result.content,\n }\n end\n )\n\n return vim.json.encode(jsn), nil\n end,\n },\n },\n },\n acp_providers = {\n [\"gemini-cli\"] = {\n command = \"gemini\",\n args = { \"--experimental-acp\" },\n env = {\n NODE_NO_WARNINGS = \"1\",\n GEMINI_API_KEY = os.getenv(\"GEMINI_API_KEY\"),\n },\n auth_method = \"gemini-api-key\",\n },\n [\"claude-code\"] = {\n command = \"npx\",\n args = { \"-y\", \"-g\", \"@zed-industries/claude-code-acp\" },\n env = {\n NODE_NO_WARNINGS = \"1\",\n ANTHROPIC_API_KEY = os.getenv(\"ANTHROPIC_API_KEY\"),\n ANTHROPIC_BASE_URL = os.getenv(\"ANTHROPIC_BASE_URL\"),\n ACP_PATH_TO_CLAUDE_CODE_EXECUTABLE = vim.fn.exepath(\"claude\"),\n ACP_PERMISSION_MODE = \"bypassPermissions\",\n },\n },\n [\"goose\"] = {\n command = \"goose\",\n args = { \"acp\" },\n },\n [\"codex\"] = {\n command = \"npx\",\n args = { \"-y\", \"-g\", \"@zed-industries/codex-acp\" },\n env = {\n NODE_NO_WARNINGS = \"1\",\n HOME = os.getenv(\"HOME\"),\n PATH = os.getenv(\"PATH\"),\n OPENAI_API_KEY = os.getenv(\"OPENAI_API_KEY\"),\n },\n },\n [\"opencode\"] = {\n command = \"opencode\",\n args = { \"acp\" },\n },\n [\"kimi-cli\"] = {\n command = \"kimi\",\n args = { \"--acp\" },\n },\n },\n ---To add support for custom provider, follow the format below\n ---See https://github.com/yetone/avante.nvim/wiki#custom-providers for more details\n ---@type {[string]: AvanteProvider}\n providers = {\n ---@type AvanteSupportedProvider\n openai = {\n endpoint = \"https://api.openai.com/v1\",\n model = \"gpt-4o\",\n timeout = 30000, -- Timeout in milliseconds, increase this for reasoning models\n context_window = 128000, -- Number of tokens to send to the model for context\n use_response_api = copilot_use_response_api, -- Automatically switch to Response API for GPT-5 Codex models\n support_previous_response_id = true, -- OpenAI Response API supports previous_response_id for stateful conversations\n -- NOTE: Response API automatically manages conversation state using previous_response_id for tool calling\n extra_request_body = {\n temperature = 0.75,\n max_completion_tokens = 16384, -- Increase this to include reasoning tokens (for reasoning models). For Response API, will be converted to max_output_tokens\n reasoning_effort = \"medium\", -- low|medium|high, only used for reasoning models. For Response API, this will be converted to reasoning.effort\n -- background = false, -- Response API only: set to true to start a background task\n -- NOTE: previous_response_id is automatically managed by the provider for tool calling - don't set manually\n },\n },\n ---@type AvanteSupportedProvider\n copilot = {\n endpoint = \"https://api.githubcopilot.com\",\n model = \"gpt-4o-2024-11-20\",\n proxy = nil, -- [protocol://]host[:port] Use this proxy\n allow_insecure = false, -- Allow insecure server connections\n timeout = 30000, -- Timeout in milliseconds\n context_window = 64000, -- Number of tokens to send to the model for context\n use_response_api = copilot_use_response_api, -- Automatically switch to Response API for GPT-5 Codex models\n support_previous_response_id = false, -- Copilot doesn't support previous_response_id, must send full history\n -- NOTE: Copilot doesn't support previous_response_id, always sends full conversation history including tool_calls\n -- NOTE: Response API doesn't support some parameters like top_p, frequency_penalty, presence_penalty\n extra_request_body = {\n -- temperature is not supported by Response API for reasoning models\n max_tokens = 20480,\n },\n },\n ---@type AvanteAzureProvider\n azure = {\n endpoint = \"\", -- example: \"https://.openai.azure.com\"\n deployment = \"\", -- Azure deployment name (e.g., \"gpt-4o\", \"my-gpt-4o-deployment\")\n api_version = \"2024-12-01-preview\",\n timeout = 30000, -- Timeout in milliseconds, increase this for reasoning models\n extra_request_body = {\n temperature = 0.75,\n max_completion_tokens = 16384, -- Increase this toinclude reasoning tokens (for reasoning models); but too large default value will not fit for some models (e.g. gpt-5-chat supports at most 16384 completion tokens)\n reasoning_effort = \"medium\", -- low|medium|high, only used for reasoning models\n },\n },\n ---@type AvanteAnthropicProvider\n claude = {\n endpoint = \"https://api.anthropic.com\",\n auth_type = \"api\",\n model = \"claude-sonnet-4-5-20250929\",\n timeout = 30000, -- Timeout in milliseconds\n context_window = 200000,\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 64000,\n },\n },\n ---@type AvanteSupportedProvider\n bedrock = {\n model = \"us.anthropic.claude-3-7-sonnet-20250219-v1:0\",\n model_names = {\n \"anthropic.claude-3-5-sonnet-20241022-v2:0\",\n \"us.anthropic.claude-3-7-sonnet-20250219-v1:0\",\n \"us.anthropic.claude-opus-4-20250514-v1:0\",\n \"us.anthropic.claude-opus-4-1-20250805-v1:0\",\n \"us.anthropic.claude-sonnet-4-20250514-v1:0\",\n },\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 20480,\n },\n aws_region = \"\", -- AWS region to use for authentication and bedrock API\n aws_profile = \"\", -- AWS profile to use for authentication, if unspecified uses default credentials chain\n },\n ---@type AvanteSupportedProvider\n gemini = {\n endpoint = \"https://generativelanguage.googleapis.com/v1beta/models\",\n model = \"gemini-2.0-flash\",\n timeout = 30000, -- Timeout in milliseconds\n context_window = 1048576,\n use_ReAct_prompt = true,\n extra_request_body = {\n generationConfig = {\n temperature = 0.75,\n },\n },\n },\n ---@type AvanteSupportedProvider\n vertex = {\n endpoint = \"https://aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/LOCATION/publishers/google/models\",\n model = \"gemini-1.5-flash-002\",\n timeout = 30000, -- Timeout in milliseconds\n context_window = 1048576,\n use_ReAct_prompt = true,\n extra_request_body = {\n generationConfig = {\n temperature = 0.75,\n },\n },\n },\n ---@type AvanteSupportedProvider\n cohere = {\n endpoint = \"https://api.cohere.com/v2\",\n model = \"command-r-plus-08-2024\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 20480,\n },\n },\n ---@type AvanteSupportedProvider\n ollama = {\n endpoint = \"http://127.0.0.1:11434\",\n timeout = 30000, -- Timeout in milliseconds\n use_ReAct_prompt = true,\n extra_request_body = {\n options = {\n temperature = 0.75,\n num_ctx = 20480,\n keep_alive = \"5m\",\n },\n },\n },\n ---@type AvanteSupportedProvider\n watsonx_code_assistant = {\n endpoint = \"https://api.dataplatform.cloud.ibm.com/v2/wca/core/chat/text/generation\",\n model = \"granite-8b-code-instruct\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n -- Additional watsonx-specific parameters can be added here\n },\n },\n\n ---@type AvanteSupportedProvider\n vertex_claude = {\n endpoint = \"https://LOCATION-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/LOCATION/publishers/anthropic/models\",\n model = \"claude-3-5-sonnet-v2@20241022\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 20480,\n },\n },\n ---@type AvanteSupportedProvider\n [\"claude-haiku\"] = {\n __inherited_from = \"claude\",\n model = \"claude-3-5-haiku-20241022\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 8192,\n },\n },\n ---@type AvanteSupportedProvider\n [\"claude-opus\"] = {\n __inherited_from = \"claude\",\n model = \"claude-3-opus-20240229\",\n timeout = 30000, -- Timeout in milliseconds\n extra_request_body = {\n temperature = 0.75,\n max_tokens = 20480,\n },\n },\n [\"openai-gpt-4o-mini\"] = {\n __inherited_from = \"openai\",\n model = \"gpt-4o-mini\",\n },\n aihubmix = {\n __inherited_from = \"openai\",\n endpoint = \"https://aihubmix.com/v1\",\n model = \"gpt-4o-2024-11-20\",\n api_key_name = \"AIHUBMIX_API_KEY\",\n },\n [\"aihubmix-claude\"] = {\n __inherited_from = \"claude\",\n endpoint = \"https://aihubmix.com\",\n model = \"claude-3-7-sonnet-20250219\",\n api_key_name = \"AIHUBMIX_API_KEY\",\n },\n morph = {\n __inherited_from = \"openai\",\n endpoint = \"https://api.morphllm.com/v1\",\n model = \"auto\",\n api_key_name = \"MORPH_API_KEY\",\n },\n moonshot = {\n __inherited_from = \"openai\",\n endpoint = \"https://api.moonshot.ai/v1\",\n model = \"kimi-k2-0711-preview\",\n api_key_name = \"MOONSHOT_API_KEY\",\n },\n xai = {\n __inherited_from = \"openai\",\n endpoint = \"https://api.x.ai/v1\",\n model = \"grok-code-fast-1\",\n api_key_name = \"XAI_API_KEY\",\n },\n glm = {\n __inherited_from = \"openai\",\n endpoint = \"https://open.bigmodel.cn/api/coding/paas/v4\",\n model = \"GLM-4.7\",\n api_key_name = \"GLM_API_KEY\",\n },\n qwen = {\n __inherited_from = \"openai\",\n endpoint = \"https://dashscope.aliyuncs.com/compatible-mode/v1\",\n model = \"qwen3-coder-plus\",\n api_key_name = \"DASHSCOPE_API_KEY\",\n },\n mistral = {\n __inherited_from = \"openai\",\n endpoint = \"https://api.mistral.ai/v1\",\n model = \"mistral-large-latest\",\n api_key_name = \"MISTRAL_API_KEY\",\n extra_request_body = {\n max_tokens = 4096, -- to avoid using the unsupported max_completion_tokens\n },\n },\n },\n ---Specify the special dual_boost mode\n ---1. enabled: Whether to enable dual_boost mode. Default to false.\n ---2. first_provider: The first provider to generate response. Default to \"openai\".\n ---3. second_provider: The second provider to generate response. Default to \"claude\".\n ---4. prompt: The prompt to generate response based on the two reference outputs.\n ---5. timeout: Timeout in milliseconds. Default to 60000.\n ---How it works:\n --- When dual_boost is enabled, avante will generate two responses from the first_provider and second_provider respectively. Then use the response from the first_provider as provider1_output and the response from the second_provider as provider2_output. Finally, avante will generate a response based on the prompt and the two reference outputs, with the default Provider as normal.\n ---Note: This is an experimental feature and may not work as expected.\n dual_boost = {\n enabled = false,\n first_provider = \"openai\",\n second_provider = \"claude\",\n prompt = \"Based on the two reference outputs below, generate a response that incorporates elements from both but reflects your own judgment and unique perspective. Do not provide any explanation, just give the response directly. Reference Output 1: [{{provider1_output}}], Reference Output 2: [{{provider2_output}}]\",\n timeout = 60000, -- Timeout in milliseconds\n },\n ---Specify the behaviour of avante.nvim\n ---1. auto_focus_sidebar : Whether to automatically focus the sidebar when opening avante.nvim. Default to true.\n ---2. auto_suggestions = false, -- Whether to enable auto suggestions. Default to false.\n ---3. auto_apply_diff_after_generation: Whether to automatically apply diff after LLM response.\n --- This would simulate similar behaviour to cursor. Default to false.\n ---4. auto_set_keymaps : Whether to automatically set the keymap for the current line. Default to true.\n --- Note that avante will safely set these keymap. See https://github.com/yetone/avante.nvim/wiki#keymaps-and-api-i-guess for more details.\n ---5. auto_set_highlight_group : Whether to automatically set the highlight group for the current line. Default to true.\n ---6. jump_result_buffer_on_finish = false, -- Whether to automatically jump to the result buffer after generation\n ---7. support_paste_from_clipboard : Whether to support pasting image from clipboard. This will be determined automatically based whether img-clip is available or not.\n ---8. minimize_diff : Whether to remove unchanged lines when applying a code block\n ---9. enable_token_counting : Whether to enable token counting. Default to true.\n ---10. auto_add_current_file : Whether to automatically add the current file when opening a new chat. Default to true.\n behaviour = {\n auto_focus_sidebar = true,\n auto_suggestions = false, -- Experimental stage\n auto_suggestions_respect_ignore = false,\n auto_set_highlight_group = true,\n auto_set_keymaps = true,\n auto_apply_diff_after_generation = false,\n jump_result_buffer_on_finish = false,\n support_paste_from_clipboard = false,\n minimize_diff = true,\n enable_token_counting = true,\n use_cwd_as_project_root = false,\n auto_focus_on_diff_view = false,\n ---@type boolean | string[] -- true: auto-approve all tools, false: normal prompts, string[]: auto-approve specific tools by name\n auto_approve_tool_permissions = true, -- Default: auto-approve all tools (no prompts)\n auto_check_diagnostics = true,\n allow_access_to_git_ignored_files = false,\n enable_fastapply = false,\n include_generated_by_commit_line = false, -- Controls if 'Generated-by: ' line is added to git commit message\n auto_add_current_file = true, -- Whether to automatically add the current file when opening a new chat\n --- popup is the original yes,all,no in a floating window\n --- inline_buttons is the new inline buttons in the sidebar\n ---@type \"popup\" | \"inline_buttons\"\n confirmation_ui_style = \"inline_buttons\",\n --- Whether to automatically open files and navigate to lines when ACP agent makes edits\n ---@type boolean\n acp_follow_agent_locations = true,\n },\n prompt_logger = { -- logs prompts to disk (timestamped, for replay/debugging)\n enabled = true, -- toggle logging entirely\n log_dir = vim.fn.stdpath(\"cache\"), -- directory where logs are saved\n max_entries = 100, -- the uplimit of entries that can be sotred\n next_prompt = {\n normal = \"\", -- load the next (newer) prompt log in normal mode\n insert = \"\",\n },\n prev_prompt = {\n normal = \"\", -- load the previous (older) prompt log in normal mode\n insert = \"\",\n },\n },\n history = {\n max_tokens = 4096,\n carried_entry_count = nil,\n storage_path = Utils.join_paths(vim.fn.stdpath(\"state\"), \"avante\"),\n paste = {\n extension = \"png\",\n filename = \"pasted-%Y-%m-%d-%H-%M-%S\",\n },\n },\n highlights = {\n diff = {\n current = nil,\n incoming = nil,\n },\n },\n img_paste = {\n url_encode_path = true,\n template = \"\\nimage: $FILE_PATH\\n\",\n },\n mappings = {\n ---@class AvanteConflictMappings\n diff = {\n ours = \"co\",\n theirs = \"ct\",\n all_theirs = \"ca\",\n both = \"cb\",\n cursor = \"cc\",\n next = \"]x\",\n prev = \"[x\",\n },\n suggestion = {\n accept = \"\",\n next = \"\",\n prev = \"\",\n dismiss = \"\",\n },\n jump = {\n next = \"]]\",\n prev = \"[[\",\n },\n submit = {\n normal = \"\",\n insert = \"\",\n },\n cancel = {\n normal = { \"\", \"\", \"q\" },\n insert = { \"\" },\n },\n -- NOTE: The following will be safely set by avante.nvim\n ask = \"aa\",\n new_ask = \"an\",\n zen_mode = \"az\",\n edit = \"ae\",\n refresh = \"ar\",\n focus = \"af\",\n stop = \"aS\",\n toggle = {\n default = \"at\",\n debug = \"ad\",\n selection = \"aC\",\n suggestion = \"as\",\n repomap = \"aR\",\n },\n sidebar = {\n expand_tool_use = \"\",\n next_prompt = \"]p\",\n prev_prompt = \"[p\",\n apply_all = \"A\",\n apply_cursor = \"a\",\n retry_user_request = \"r\",\n edit_user_request = \"e\",\n switch_windows = \"\",\n reverse_switch_windows = \"\",\n toggle_code_window = \"x\",\n remove_file = \"d\",\n add_file = \"@\",\n close = { \"q\" },\n ---@alias AvanteCloseFromInput { normal: string | nil, insert: string | nil }\n ---@type AvanteCloseFromInput | nil\n close_from_input = nil, -- e.g., { normal = \"\", insert = \"\" }\n ---@alias AvanteToggleCodeWindowFromInput { normal: string | nil, insert: string | nil }\n ---@type AvanteToggleCodeWindowFromInput | nil\n toggle_code_window_from_input = nil, -- e.g., { normal = \"x\", insert = \"\" }\n },\n files = {\n add_current = \"ac\", -- Add current buffer to selected files\n add_all_buffers = \"aB\", -- Add all buffer files to selected files\n },\n select_model = \"a?\", -- Select model command\n select_history = \"ah\", -- Select history command\n confirm = {\n focus_window = \"f\",\n code = \"c\",\n resp = \"r\",\n input = \"i\",\n },\n },\n windows = {\n ---@alias AvantePosition \"right\" | \"left\" | \"top\" | \"bottom\" | \"smart\"\n ---@type AvantePosition\n position = \"right\",\n fillchars = \"eob: \",\n wrap = true, -- similar to vim.o.wrap\n width = 30, -- default % based on available width in vertical layout\n height = 30, -- default % based on available height in horizontal layout\n sidebar_header = {\n enabled = true, -- true, false to enable/disable the header\n align = \"center\", -- left, center, right for title\n rounded = true,\n include_model = false,\n },\n spinner = {\n editing = {\n \"⡀\",\n \"⠄\",\n \"⠂\",\n \"⠁\",\n \"⠈\",\n \"⠐\",\n \"⠠\",\n \"⢀\",\n \"⣀\",\n \"⢄\",\n \"⢂\",\n \"⢁\",\n \"⢈\",\n \"⢐\",\n \"⢠\",\n \"⣠\",\n \"⢤\",\n \"⢢\",\n \"⢡\",\n \"⢨\",\n \"⢰\",\n \"⣰\",\n \"⢴\",\n \"⢲\",\n \"⢱\",\n \"⢸\",\n \"⣸\",\n \"⢼\",\n \"⢺\",\n \"⢹\",\n \"⣹\",\n \"⢽\",\n \"⢻\",\n \"⣻\",\n \"⢿\",\n \"⣿\",\n },\n generating = { \"·\", \"✢\", \"✳\", \"∗\", \"✻\", \"✽\" },\n thinking = { \"🤯\", \"🙄\" },\n },\n input = {\n prefix = \"> \",\n height = 8, -- Height of the input window in vertical layout\n },\n selected_files = {\n height = 6, -- Maximum height of the selected files window\n },\n edit = {\n border = { \" \", \" \", \" \", \" \", \" \", \" \", \" \", \" \" },\n start_insert = true, -- Start insert mode when opening the edit window\n },\n ask = {\n floating = false, -- Open the 'AvanteAsk' prompt in a floating window\n border = { \" \", \" \", \" \", \" \", \" \", \" \", \" \", \" \" },\n start_insert = true, -- Start insert mode when opening the ask window\n ---@alias AvanteInitialDiff \"ours\" | \"theirs\"\n ---@type AvanteInitialDiff\n focus_on_apply = \"ours\", -- which diff to focus after applying\n },\n },\n --- @class AvanteConflictConfig\n diff = {\n autojump = true,\n --- Override the 'timeoutlen' setting while hovering over a diff (see :help timeoutlen).\n --- Helps to avoid entering operator-pending mode with diff mappings starting with `c`.\n --- Disable by setting to -1.\n override_timeoutlen = 500,\n },\n --- Allows selecting code or other data in a buffer and ask LLM questions about it or\n --- to perform edits/transformations.\n --- @class AvanteSelectionConfig\n --- @field enabled boolean\n --- @field hint_display \"delayed\" | \"immediate\" | \"none\" When to show key map hints.\n selection = {\n enabled = true,\n hint_display = \"delayed\",\n },\n --- @class AvanteRepoMapConfig\n repo_map = {\n ignore_patterns = { \"%.git\", \"%.worktree\", \"__pycache__\", \"node_modules\" }, -- ignore files matching these\n negate_patterns = {}, -- negate ignore files matching these.\n },\n --- @class AvanteFileSelectorConfig\n file_selector = {\n provider = nil,\n -- Options override for custom providers\n provider_opts = {},\n },\n selector = {\n ---@alias avante.SelectorProvider \"native\" | \"fzf_lua\" | \"mini_pick\" | \"snacks\" | \"telescope\" | fun(selector: avante.ui.Selector): nil\n ---@type avante.SelectorProvider\n provider = \"native\",\n provider_opts = {},\n exclude_auto_select = {}, -- List of items to exclude from auto selection\n },\n input = {\n provider = \"native\",\n provider_opts = {},\n },\n suggestion = {\n debounce = 600,\n throttle = 600,\n },\n disabled_tools = {}, ---@type string[]\n ---@type AvanteLLMToolPublic[] | fun(): AvanteLLMToolPublic[]\n custom_tools = {},\n ---@type AvanteSlashCommand[]\n slash_commands = {},\n ---@type AvanteShortcut[]\n shortcuts = {},\n ---@type AskOptions\n ask_opts = {},\n}\n\n---@type avante.Config\n---@diagnostic disable-next-line: missing-fields\nM._options = {}\n\nlocal function get_config_dir_path() return Utils.join_paths(vim.fn.expand(\"~\"), \".config\", \"avante.nvim\") end\nlocal function get_config_file_path() return Utils.join_paths(get_config_dir_path(), \"config.json\") end\n\n--- Function to save the last used model\n---@param model_name string\nfunction M.save_last_model(model_name, provider_name)\n local config_dir = get_config_dir_path()\n local storage_path = get_config_file_path()\n\n if not Utils.path_exists(config_dir) then vim.fn.mkdir(config_dir, \"p\") end\n\n local Providers = require(\"avante.providers\")\n local provider = Providers[provider_name]\n local provider_model = provider and provider.model\n\n local file = io.open(storage_path, \"w\")\n if file then\n file:write(\n vim.json.encode({ last_model = model_name, last_provider = provider_name, provider_model = provider_model })\n )\n file:close()\n end\nend\n\n--- Retrieves names of the last used model and provider. May remove saved config if it is deemed invalid\n---@param known_providers table\n---@return string|nil Model name\n---@return string|nil Provider name\nfunction M.get_last_used_model(known_providers)\n local storage_path = get_config_file_path()\n local file = io.open(storage_path, \"r\")\n if file then\n local content = file:read(\"*a\")\n file:close()\n\n if not content or content == \"\" then\n Utils.warn(\"Last used model file is empty: \" .. storage_path)\n -- Remove to not have repeated warnings\n os.remove(storage_path)\n end\n\n local success, data = pcall(vim.json.decode, content)\n if not success or not data or not data.last_model or data.last_model == \"\" or data.last_provider == \"\" then\n Utils.warn(\"Invalid or corrupt JSON in last used model file: \" .. storage_path)\n -- Rename instead of deleting so user can examine contents\n os.rename(storage_path, storage_path .. \".bad\")\n return\n end\n\n if data.last_provider then\n local provider = known_providers[data.last_provider]\n if not provider then\n Utils.warn(\n \"Provider \" .. data.last_provider .. \" is no longer a valid provider, falling back to default configuration\"\n )\n os.remove(storage_path)\n return\n end\n if data.provider_model and provider.model and provider.model ~= data.provider_model then\n return provider.model, data.last_provider\n end\n end\n\n return data.last_model, data.last_provider\n end\nend\n\n---Applies given model and provider to the config\n---@param config avante.Config\n---@param model_name string\n---@param provider_name? string\nlocal function apply_model_selection(config, model_name, provider_name)\n local provider_list = config.providers or {}\n local current_provider_name = config.provider\n if config.acp_providers[current_provider_name] then return end\n\n local target_provider_name = provider_name or current_provider_name\n local target_provider = provider_list[target_provider_name]\n\n if not target_provider then return end\n\n local current_provider_data = provider_list[current_provider_name]\n local current_model_name = current_provider_data and current_provider_data.model\n\n if target_provider_name ~= current_provider_name or model_name ~= current_model_name then\n config.provider = target_provider_name\n target_provider.model = model_name\n if not target_provider.model_names then target_provider.model_names = {} end\n for _, model_name_ in ipairs({ model_name, current_model_name }) do\n if not vim.tbl_contains(target_provider.model_names, model_name_) then\n table.insert(target_provider.model_names, model_name_)\n end\n end\n if not config.windows.sidebar_header.include_model then\n Utils.info(string.format(\"Using previously selected model: %s/%s\", target_provider_name, model_name))\n end\n end\nend\n\n---@param opts table|nil -- Optional table parameter for configuration settings\nfunction M.setup(opts)\n opts = opts or {} -- Ensure `opts` is defined with a default table\n if vim.fn.has(\"nvim-0.11\") == 1 then\n vim.validate(\"opts\", opts, \"table\", true)\n else\n vim.validate({ opts = { opts, \"table\", true } })\n end\n\n opts = opts or {}\n\n local migration_url = \"https://github.com/yetone/avante.nvim/wiki/Provider-configuration-migration-guide\"\n\n if opts.providers ~= nil then\n for k, v in pairs(opts.providers) do\n local extra_request_body\n if type(v) == \"table\" then\n if M._defaults.providers[k] ~= nil then\n extra_request_body = M._defaults.providers[k].extra_request_body\n elseif v.__inherited_from ~= nil then\n if M._defaults.providers[v.__inherited_from] ~= nil then\n extra_request_body = M._defaults.providers[v.__inherited_from].extra_request_body\n end\n end\n end\n if extra_request_body ~= nil then\n for k_, v_ in pairs(v) do\n if extra_request_body[k_] ~= nil then\n opts.providers[k].extra_request_body = opts.providers[k].extra_request_body or {}\n opts.providers[k].extra_request_body[k_] = v_\n Utils.warn(\n string.format(\n \"[DEPRECATED] The configuration of `providers.%s.%s` should be placed in `providers.%s.extra_request_body.%s`; for detailed migration instructions, please visit: %s\",\n k,\n k_,\n k,\n k_,\n migration_url\n ),\n { title = \"Avante\" }\n )\n end\n end\n end\n end\n end\n\n for k, v in pairs(opts) do\n if M._defaults.providers[k] ~= nil then\n opts.providers = opts.providers or {}\n opts.providers[k] = v\n Utils.warn(\n string.format(\n \"[DEPRECATED] The configuration of `%s` should be placed in `providers.%s`. For detailed migration instructions, please visit: %s\",\n k,\n k,\n migration_url\n ),\n { title = \"Avante\" }\n )\n local extra_request_body = M._defaults.providers[k].extra_request_body\n if type(v) == \"table\" and extra_request_body ~= nil then\n for k_, v_ in pairs(v) do\n if extra_request_body[k_] ~= nil then\n opts.providers[k].extra_request_body = opts.providers[k].extra_request_body or {}\n opts.providers[k].extra_request_body[k_] = v_\n Utils.warn(\n string.format(\n \"[DEPRECATED] The configuration of `%s.%s` should be placed in `providers.%s.extra_request_body.%s`; for detailed migration instructions, please visit: %s\",\n k,\n k_,\n k,\n k_,\n migration_url\n ),\n { title = \"Avante\" }\n )\n end\n end\n end\n end\n if k == \"vendors\" and v ~= nil then\n for k2, v2 in pairs(v) do\n opts.providers = opts.providers or {}\n opts.providers[k2] = v2\n Utils.warn(\n string.format(\n \"[DEPRECATED] The configuration of `vendors.%s` should be placed in `providers.%s`. For detailed migration instructions, please visit: %s\",\n k2,\n k2,\n migration_url\n ),\n { title = \"Avante\" }\n )\n if\n type(v2) == \"table\"\n and v2.__inherited_from ~= nil\n and M._defaults.providers[v2.__inherited_from] ~= nil\n then\n local extra_request_body = M._defaults.providers[v2.__inherited_from].extra_request_body\n if extra_request_body ~= nil then\n for k2_, v2_ in pairs(v2) do\n if extra_request_body[k2_] ~= nil then\n opts.providers[k2].extra_request_body = opts.providers[k2].extra_request_body or {}\n opts.providers[k2].extra_request_body[k2_] = v2_\n Utils.warn(\n string.format(\n \"[DEPRECATED] The configuration of `vendors.%s.%s` should be placed in `providers.%s.extra_request_body.%s`; for detailed migration instructions, please visit: %s\",\n k2,\n k2_,\n k2,\n k2_,\n migration_url\n ),\n { title = \"Avante\" }\n )\n end\n end\n end\n end\n end\n end\n end\n\n local merged = vim.tbl_deep_extend(\n \"force\",\n M._defaults,\n opts,\n ---@type avante.Config\n {\n behaviour = {\n support_paste_from_clipboard = M.support_paste_image(),\n },\n }\n )\n\n local last_model, last_provider = M.get_last_used_model(merged.providers or {})\n if last_model then apply_model_selection(merged, last_model, last_provider) end\n\n M._options = merged\n\n ---@diagnostic disable-next-line: undefined-field\n if M._options.disable_tools ~= nil then\n Utils.warn(\n \"`disable_tools` is provider-scoped, not globally scoped. Therefore, you cannot set `disable_tools` at the top level. It should be set under a provider, for example: `openai.disable_tools = true`\",\n { title = \"Avante\" }\n )\n end\n\n if type(M._options.disabled_tools) == \"boolean\" then\n Utils.warn(\n '`disabled_tools` must be a list, not a boolean. Please change it to `disabled_tools = { \"tool1\", \"tool2\" }`. Note the difference between `disabled_tools` and `disable_tools`.',\n { title = \"Avante\" }\n )\n end\n\n if vim.fn.has(\"nvim-0.11\") == 1 then\n vim.validate(\"provider\", M._options.provider, \"string\", false)\n else\n vim.validate({ provider = { M._options.provider, \"string\", false } })\n end\n\n for k, v in pairs(M._options.providers) do\n M._options.providers[k] = type(v) == \"function\" and v() or v\n end\nend\n\n---@param opts table\nfunction M.override(opts)\n if vim.fn.has(\"nvim-0.11\") == 1 then\n vim.validate(\"opts\", opts, \"table\", true)\n else\n vim.validate({ opts = { opts, \"table\", true } })\n end\n\n M._options = vim.tbl_deep_extend(\"force\", M._options, opts or {})\n\n for k, v in pairs(M._options.providers) do\n M._options.providers[k] = type(v) == \"function\" and v() or v\n end\nend\n\nM = setmetatable(M, {\n __index = function(_, k)\n if M._options[k] then return M._options[k] end\n end,\n})\n\nfunction M.support_paste_image() return Utils.has(\"img-clip.nvim\") or Utils.has(\"img-clip\") end\n\nfunction M.get_window_width() return math.ceil(vim.o.columns * (M.windows.width / 100)) end\n\n---get supported providers\n---@param provider_name avante.ProviderName\nfunction M.get_provider_config(provider_name)\n local found = false\n local config = {}\n\n if M.providers[provider_name] ~= nil then\n found = true\n config = vim.tbl_deep_extend(\"force\", config, vim.deepcopy(M.providers[provider_name], true))\n end\n\n if not found then error(\"Failed to find provider: \" .. provider_name, 2) end\n\n return config\nend\n\nreturn M\n" }, { "path": "lua/avante/diff.lua", "content": "local api = vim.api\n\nlocal Config = require(\"avante.config\")\nlocal Utils = require(\"avante.utils\")\nlocal Highlights = require(\"avante.highlights\")\n\nlocal H = {}\nlocal M = {}\n\n-----------------------------------------------------------------------------//\n-- REFERENCES:\n-----------------------------------------------------------------------------//\n-- Detecting the state of a git repository based on files in the .git directory.\n-- https://stackoverflow.com/questions/49774200/how-to-tell-if-my-git-repo-is-in-a-conflict\n-- git diff commands to git a list of conflicted files\n-- https://stackoverflow.com/questions/3065650/whats-the-simplest-way-to-list-conflicted-files-in-git\n-- how to show a full path for files in a git diff command\n-- https://stackoverflow.com/questions/10459374/making-git-diff-stat-show-full-file-path\n-- Advanced merging\n-- https://git-scm.com/book/en/v2/Git-Tools-Advanced-Merging\n\n-----------------------------------------------------------------------------//\n-- Types\n-----------------------------------------------------------------------------//\n\n---@alias ConflictSide \"'ours'\"|\"'theirs'\"|\"'all_theirs'\"|\"'both'\"|\"'cursor'\"|\"'base'\"|\"'none'\"\n\n--- @class AvanteConflictHighlights\n--- @field current string\n--- @field incoming string\n\n---@class RangeMark\n---@field label integer\n---@field content string\n\n--- @class PositionMarks\n--- @field current RangeMark\n--- @field incoming RangeMark\n\n--- @class Range\n--- @field range_start integer\n--- @field range_end integer\n--- @field content_start integer\n--- @field content_end integer\n\n--- @class ConflictPosition\n--- @field incoming Range\n--- @field middle Range\n--- @field current Range\n--- @field marks PositionMarks\n\n--- @class ConflictBufferCache\n--- @field lines table map of conflicted line numbers\n--- @field positions ConflictPosition[]\n--- @field tick integer\n--- @field bufnr integer\n\n-----------------------------------------------------------------------------//\n-- Constants\n-----------------------------------------------------------------------------//\n---@enum AvanteConflictSides\nlocal SIDES = {\n OURS = \"ours\",\n THEIRS = \"theirs\",\n ALL_THEIRS = \"all_theirs\",\n BOTH = \"both\",\n NONE = \"none\",\n CURSOR = \"cursor\",\n}\n\n-- A mapping between the internal names and the display names\nlocal name_map = {\n ours = \"current\",\n theirs = \"incoming\",\n both = \"both\",\n none = \"none\",\n cursor = \"cursor\",\n}\n\nlocal CURRENT_HL = Highlights.CURRENT\nlocal INCOMING_HL = Highlights.INCOMING\nlocal CURRENT_LABEL_HL = Highlights.CURRENT_LABEL\nlocal INCOMING_LABEL_HL = Highlights.INCOMING_LABEL\nlocal PRIORITY = (vim.hl or vim.highlight).priorities.user\nlocal NAMESPACE = api.nvim_create_namespace(\"avante-conflict\")\nlocal KEYBINDING_NAMESPACE = api.nvim_create_namespace(\"avante-conflict-keybinding\")\nlocal AUGROUP_NAME = \"avante_conflicts\"\n\nlocal conflict_start = \"^<<<<<<<\"\nlocal conflict_middle = \"^=======\"\nlocal conflict_end = \"^>>>>>>>\"\n\n-----------------------------------------------------------------------------//\n\n--- @return table\nlocal function create_visited_buffers()\n return setmetatable({}, {\n __index = function(t, k)\n if type(k) == \"number\" then return t[api.nvim_buf_get_name(k)] end\n end,\n })\nend\n\n--- A list of buffers that have conflicts in them. This is derived from\n--- git using the diff command, and updated at intervals\nlocal visited_buffers = create_visited_buffers()\n\n-----------------------------------------------------------------------------//\n\n---Add the positions to the buffer in our in memory buffer list\n---positions are keyed by a list of range start and end for each mark\n---@param buf integer\n---@param positions ConflictPosition[]\nlocal function update_visited_buffers(buf, positions)\n if not buf or not api.nvim_buf_is_valid(buf) then return end\n local name = api.nvim_buf_get_name(buf)\n -- If this buffer is not in the list\n if not visited_buffers[name] then return end\n visited_buffers[name].bufnr = buf\n visited_buffers[name].tick = vim.b[buf].changedtick\n visited_buffers[name].positions = positions\nend\n\nfunction M.add_visited_buffer(bufnr)\n local name = api.nvim_buf_get_name(bufnr)\n visited_buffers[name] = visited_buffers[name] or {}\nend\n\n---Set an extmark for each section of the git conflict\n---@param bufnr integer\n---@param hl string\n---@param range_start integer\n---@param range_end integer\n---@return integer? extmark_id\nlocal function hl_range(bufnr, hl, range_start, range_end)\n if not range_start or not range_end then return end\n return api.nvim_buf_set_extmark(bufnr, NAMESPACE, range_start, 0, {\n hl_group = hl,\n hl_eol = true,\n hl_mode = \"combine\",\n end_row = range_end,\n priority = PRIORITY,\n })\nend\n\n---Add highlights and additional data to each section heading of the conflict marker\n---These works by covering the underlying text with an extmark that contains the same information\n---with some extra detail appended.\n---TODO: ideally this could be done by using virtual text at the EOL and highlighting the\n---background but this doesn't work and currently this is done by filling the rest of the line with\n---empty space and overlaying the line content\n---@param bufnr integer\n---@param hl_group string\n---@param label string\n---@param lnum integer\n---@return integer extmark id\nlocal function draw_section_label(bufnr, hl_group, label, lnum)\n local remaining_space = api.nvim_win_get_width(0) - api.nvim_strwidth(label)\n return api.nvim_buf_set_extmark(bufnr, NAMESPACE, lnum, 0, {\n hl_group = hl_group,\n virt_text = { { label .. string.rep(\" \", remaining_space), hl_group } },\n virt_text_pos = \"overlay\",\n priority = PRIORITY,\n })\nend\n\n---Highlight each part of a git conflict i.e. the incoming changes vs the current/HEAD changes\n---TODO: should extmarks be ephemeral? or is it less expensive to save them and only re-apply\n---them when a buffer changes since otherwise we have to reparse the whole buffer constantly\n---@param bufnr integer\n---@param positions table\n---@param lines string[]\nlocal function highlight_conflicts(bufnr, positions, lines)\n M.clear(bufnr)\n\n for _, position in ipairs(positions) do\n local current_start = position.current.range_start\n local current_end = position.current.range_end\n local incoming_start = position.incoming.range_start\n local incoming_end = position.incoming.range_end\n -- Add one since the index access in lines is 1 based\n local current_label = lines[current_start + 1] .. \" (Current changes)\"\n local incoming_label = lines[incoming_end + 1] .. \" (Incoming changes)\"\n\n local curr_label_id = draw_section_label(bufnr, CURRENT_LABEL_HL, current_label, current_start)\n local curr_id = hl_range(bufnr, CURRENT_HL, current_start, current_end + 1)\n local inc_id = hl_range(bufnr, INCOMING_HL, incoming_start, incoming_end + 1)\n local inc_label_id = draw_section_label(bufnr, INCOMING_LABEL_HL, incoming_label, incoming_end)\n\n position.marks = {\n current = { label = curr_label_id, content = curr_id },\n incoming = { label = inc_label_id, content = inc_id },\n }\n end\nend\n\n---Iterate through the buffer line by line checking there is a matching conflict marker\n---when we find a starting mark we collect the position details and add it to a list of positions\n---@param lines string[]\n---@return boolean\n---@return ConflictPosition[]\nlocal function detect_conflicts(lines)\n local positions = {}\n local position, has_middle = nil, false\n for index, line in ipairs(lines) do\n local lnum = index - 1\n if line:match(conflict_start) then\n position = {\n current = { range_start = lnum, content_start = lnum + 1 },\n middle = {},\n incoming = {},\n }\n end\n if position ~= nil and line:match(conflict_middle) then\n has_middle = true\n position.current.range_end = lnum - 1\n position.current.content_end = lnum - 1\n position.middle.range_start = lnum\n position.middle.range_end = lnum + 1\n position.incoming.range_start = lnum + 1\n position.incoming.content_start = lnum + 1\n end\n if position ~= nil and has_middle and line:match(conflict_end) then\n position.incoming.range_end = lnum\n position.incoming.content_end = lnum - 1\n positions[#positions + 1] = position\n\n position, has_middle = nil, false\n end\n end\n return #positions > 0, positions\nend\n\n---Helper function to find a conflict position based on a comparator function\n---@param bufnr integer\n---@param comparator fun(string, integer): boolean\n---@param opts table?\n---@return ConflictPosition?\nlocal function find_position(bufnr, comparator, opts)\n local match = visited_buffers[bufnr]\n if not match then return end\n local line = Utils.get_cursor_pos()\n line = line - 1 -- Convert to 0-based for position comparison\n\n if opts and opts.reverse then\n for i = #match.positions, 1, -1 do\n local position = match.positions[i]\n if comparator(line, position) then return position end\n end\n return nil\n end\n\n for _, position in ipairs(match.positions) do\n if comparator(line, position) then return position end\n end\n return nil\nend\n\n---Retrieves a conflict marker position by checking the visited buffers for a supported range\n---@param bufnr integer\n---@return ConflictPosition?\nlocal function get_current_position(bufnr)\n return find_position(\n bufnr,\n function(line, position) return position.current.range_start <= line and position.incoming.range_end >= line end\n )\nend\n\n---@param position ConflictPosition?\n---@param side ConflictSide\nlocal function set_cursor(position, side)\n if not position then return end\n local target = side == SIDES.OURS and position.current or position.incoming\n api.nvim_win_set_cursor(0, { target.range_start + 1, 0 })\n vim.cmd(\"normal! zz\")\nend\n\nlocal show_keybinding_hint_extmark_id = nil\nlocal function register_cursor_move_events(bufnr)\n local function show_keybinding_hint(lnum)\n if show_keybinding_hint_extmark_id then\n api.nvim_buf_del_extmark(bufnr, KEYBINDING_NAMESPACE, show_keybinding_hint_extmark_id)\n end\n\n local hint = string.format(\n \"[<%s>: OURS, <%s>: THEIRS, <%s>: CURSOR, <%s>: ALL THEIRS, <%s>: PREV, <%s>: NEXT]\",\n Config.mappings.diff.ours,\n Config.mappings.diff.theirs,\n Config.mappings.diff.cursor,\n Config.mappings.diff.all_theirs,\n Config.mappings.diff.prev,\n Config.mappings.diff.next\n )\n\n show_keybinding_hint_extmark_id = api.nvim_buf_set_extmark(bufnr, KEYBINDING_NAMESPACE, lnum - 1, -1, {\n hl_group = \"AvanteInlineHint\",\n virt_text = { { hint, \"AvanteInlineHint\" } },\n virt_text_pos = \"right_align\",\n priority = PRIORITY,\n })\n end\n\n api.nvim_create_autocmd({ \"CursorMoved\", \"CursorMovedI\", \"WinLeave\" }, {\n buffer = bufnr,\n callback = function(event)\n local position = get_current_position(bufnr)\n if (event.event == \"CursorMoved\" or event.event == \"CursorMovedI\") and position then\n show_keybinding_hint(position.current.range_start + 1)\n M.override_timeoutlen(bufnr)\n else\n api.nvim_buf_clear_namespace(bufnr, KEYBINDING_NAMESPACE, 0, -1)\n M.restore_timeoutlen(bufnr)\n end\n end,\n })\nend\n\n---Get the conflict marker positions for a buffer if any and update the buffers state\n---@param bufnr integer\n---@param range_start? integer\n---@param range_end? integer\nlocal function parse_buffer(bufnr, range_start, range_end)\n local lines = Utils.get_buf_lines(range_start or 0, range_end or -1, bufnr)\n local prev_conflicts = visited_buffers[bufnr].positions ~= nil and #visited_buffers[bufnr].positions > 0\n local has_conflict, positions = detect_conflicts(lines)\n\n update_visited_buffers(bufnr, positions)\n if has_conflict then\n register_cursor_move_events(bufnr)\n highlight_conflicts(bufnr, positions, lines)\n else\n M.clear(bufnr)\n end\n if prev_conflicts ~= has_conflict or not vim.b[bufnr].avante_conflict_mappings_set then\n local pattern = has_conflict and \"AvanteConflictDetected\" or \"AvanteConflictResolved\"\n api.nvim_exec_autocmds(\"User\", { pattern = pattern })\n end\nend\n\n---Process a buffer if the changed tick has changed\n---@param bufnr integer\n---@param range_start integer?\n---@param range_end integer?\nfunction M.process(bufnr, range_start, range_end)\n bufnr = bufnr\n if visited_buffers[bufnr] and visited_buffers[bufnr].tick == vim.b[bufnr].changedtick then return end\n parse_buffer(bufnr, range_start, range_end)\nend\n\n-----------------------------------------------------------------------------//\n-- Mappings\n-----------------------------------------------------------------------------//\n\n---@param bufnr integer given buffer id\nfunction H.setup_buffer_mappings(bufnr)\n ---@param desc string\n local function opts(desc) return { silent = true, buffer = bufnr, desc = \"avante(conflict): \" .. desc } end\n\n vim.keymap.set({ \"n\", \"v\" }, Config.mappings.diff.ours, function() M.choose(\"ours\") end, opts(\"choose ours\"))\n vim.keymap.set({ \"n\", \"v\" }, Config.mappings.diff.both, function() M.choose(\"both\") end, opts(\"choose both\"))\n vim.keymap.set({ \"n\", \"v\" }, Config.mappings.diff.theirs, function() M.choose(\"theirs\") end, opts(\"choose theirs\"))\n vim.keymap.set(\n { \"n\", \"v\" },\n Config.mappings.diff.all_theirs,\n function() M.choose(\"all_theirs\") end,\n opts(\"choose all theirs\")\n )\n vim.keymap.set(\"n\", Config.mappings.diff.cursor, function() M.choose(\"cursor\") end, opts(\"choose under cursor\"))\n vim.keymap.set(\"n\", Config.mappings.diff.prev, function() M.find_prev(\"ours\") end, opts(\"previous conflict\"))\n vim.keymap.set(\"n\", Config.mappings.diff.next, function() M.find_next(\"ours\") end, opts(\"next conflict\"))\n\n vim.b[bufnr].avante_conflict_mappings_set = true\nend\n\n---@param bufnr integer\nfunction H.clear_buffer_mappings(bufnr)\n if not bufnr or not vim.b[bufnr].avante_conflict_mappings_set then return end\n\n for _, diff_mapping in pairs(Config.mappings.diff) do\n pcall(vim.api.nvim_buf_del_keymap, bufnr, \"n\", diff_mapping)\n end\n\n vim.b[bufnr].avante_conflict_mappings_set = false\n M.restore_timeoutlen(bufnr)\nend\n\n---@param bufnr integer\nfunction M.override_timeoutlen(bufnr)\n if vim.b[bufnr].avante_original_timeoutlen then return end\n if Config.diff.override_timeoutlen > 0 then\n vim.b[bufnr].avante_original_timeoutlen = vim.o.timeoutlen\n vim.o.timeoutlen = Config.diff.override_timeoutlen\n end\nend\n\n---@param bufnr integer\nfunction M.restore_timeoutlen(bufnr)\n if vim.b[bufnr].avante_original_timeoutlen then\n vim.o.timeoutlen = vim.b[bufnr].avante_original_timeoutlen\n vim.b[bufnr].avante_original_timeoutlen = nil\n end\nend\n\nM.augroup = api.nvim_create_augroup(AUGROUP_NAME, { clear = true })\n\nfunction M.setup()\n local previous_inlay_enabled = nil\n\n api.nvim_create_autocmd(\"User\", {\n group = M.augroup,\n pattern = \"AvanteConflictDetected\",\n callback = function(ev)\n vim.diagnostic.enable(false, { bufnr = ev.buf })\n if vim.lsp.inlay_hint then\n previous_inlay_enabled = vim.lsp.inlay_hint.is_enabled({ bufnr = ev.buf })\n vim.lsp.inlay_hint.enable(false, { bufnr = ev.buf })\n end\n H.setup_buffer_mappings(ev.buf)\n end,\n })\n\n api.nvim_create_autocmd(\"User\", {\n group = M.augroup,\n pattern = \"AvanteConflictResolved\",\n callback = function(ev)\n vim.diagnostic.enable(true, { bufnr = ev.buf })\n if vim.lsp.inlay_hint and previous_inlay_enabled ~= nil then\n vim.lsp.inlay_hint.enable(previous_inlay_enabled, { bufnr = ev.buf })\n previous_inlay_enabled = nil\n end\n H.clear_buffer_mappings(ev.buf)\n end,\n })\n\n api.nvim_set_decoration_provider(NAMESPACE, {\n on_win = function(_, _, bufnr, _, _)\n if visited_buffers[bufnr] then M.process(bufnr) end\n end,\n })\nend\n\n--- Add additional metadata to a quickfix entry if we have already visited the buffer and have that\n--- information\n---@param item table\n---@param items table[]\n---@param visited_buf ConflictBufferCache\nlocal function quickfix_items_from_positions(item, items, visited_buf)\n if vim.tbl_isempty(visited_buf.positions) then return end\n for _, pos in ipairs(visited_buf.positions) do\n for key, value in pairs(pos) do\n if vim.tbl_contains({ name_map.ours, name_map.theirs, name_map.base }, key) and not vim.tbl_isempty(value) then\n local lnum = value.range_start + 1\n local next_item = vim.deepcopy(item)\n next_item.text = string.format(\"%s change\", key, lnum)\n next_item.lnum = lnum\n next_item.col = 0\n table.insert(items, next_item)\n end\n end\n end\nend\n\n--- Convert the conflicts detected via get conflicted files into a list of quickfix entries.\n---@param callback fun(files: table)\nfunction M.conflicts_to_qf_items(callback)\n local items = {}\n for filename, visited_buf in pairs(visited_buffers) do\n local item = {\n filename = filename,\n pattern = conflict_start,\n text = \"git conflict\",\n type = \"E\",\n valid = 1,\n }\n\n if visited_buf and next(visited_buf) then\n quickfix_items_from_positions(item, items, visited_buf)\n else\n table.insert(items, item)\n end\n end\n\n callback(items)\nend\n\n---@param bufnr integer?\nfunction M.clear(bufnr)\n if bufnr and not api.nvim_buf_is_valid(bufnr) then return end\n bufnr = bufnr or 0\n api.nvim_buf_clear_namespace(bufnr, NAMESPACE, 0, -1)\n api.nvim_buf_clear_namespace(bufnr, KEYBINDING_NAMESPACE, 0, -1)\nend\n\n---@param side ConflictSide\nfunction M.find_next(side)\n local pos = find_position(\n 0,\n function(line, position) return position.current.range_start > line and position.incoming.range_end > line end\n )\n set_cursor(pos, side)\nend\n\n---@param side ConflictSide\nfunction M.find_prev(side)\n local pos = find_position(\n 0,\n function(line, position) return position.current.range_start <= line and position.incoming.range_end <= line end,\n { reverse = true }\n )\n set_cursor(pos, side)\nend\n\n---Select the changes to keep\n---@param side ConflictSide\nfunction M.choose(side)\n local bufnr = api.nvim_get_current_buf()\n if vim.fn.mode() == \"v\" or vim.fn.mode() == \"V\" or vim.fn.mode() == \"\u0016\" then\n vim.cmd(\"noautocmd stopinsert\")\n -- have to defer so that the < and > marks are set\n vim.defer_fn(function()\n local start = api.nvim_buf_get_mark(0, \"<\")[1]\n local finish = api.nvim_buf_get_mark(0, \">\")[1]\n local position = find_position(bufnr, function(_, pos)\n local left = pos.current.range_start >= start - 1\n local right = pos.incoming.range_end <= finish + 1\n return left and right\n end)\n while position ~= nil do\n M.process_position(bufnr, side, position, false)\n position = find_position(bufnr, function(_, pos)\n local left = pos.current.range_start >= start - 1\n local right = pos.incoming.range_end <= finish + 1\n return left and right\n end)\n end\n end, 50)\n if Config.diff.autojump then\n M.find_next(side)\n vim.cmd([[normal! zz]])\n end\n return\n end\n local position = get_current_position(bufnr)\n if not position then return end\n if side == SIDES.ALL_THEIRS then\n ---@diagnostic disable-next-line: unused-local\n local pos = find_position(bufnr, function(line, pos) return true end)\n while pos ~= nil do\n M.process_position(bufnr, \"theirs\", pos, false)\n ---@diagnostic disable-next-line: unused-local\n pos = find_position(bufnr, function(line, pos_) return true end)\n end\n else\n M.process_position(bufnr, side, position, true)\n end\nend\n\n---@param side ConflictSide\n---@param position ConflictPosition\n---@param enable_autojump boolean\nfunction M.process_position(bufnr, side, position, enable_autojump)\n local lines = {}\n if vim.tbl_contains({ SIDES.OURS, SIDES.THEIRS }, side) then\n local data = position[name_map[side]]\n lines = Utils.get_buf_lines(data.content_start, data.content_end + 1)\n elseif side == SIDES.BOTH then\n local first = Utils.get_buf_lines(position.current.content_start, position.current.content_end + 1)\n local second = Utils.get_buf_lines(position.incoming.content_start, position.incoming.content_end + 1)\n lines = vim.list_extend(first, second)\n elseif side == SIDES.NONE then\n lines = {}\n elseif side == SIDES.CURSOR then\n local cursor_line = Utils.get_cursor_pos()\n for _, pos in ipairs({ SIDES.OURS, SIDES.THEIRS }) do\n local data = position[name_map[pos]] or {}\n if data.range_start and data.range_start + 1 <= cursor_line and data.range_end + 1 >= cursor_line then\n side = pos\n lines = Utils.get_buf_lines(data.content_start, data.content_end + 1)\n break\n end\n end\n if side == SIDES.CURSOR then return end\n else\n return\n end\n\n local pos_start = position.current.range_start < 0 and 0 or position.current.range_start\n local pos_end = position.incoming.range_end + 1\n\n api.nvim_buf_set_lines(0, pos_start, pos_end, false, lines)\n api.nvim_buf_del_extmark(0, NAMESPACE, position.marks.incoming.label)\n api.nvim_buf_del_extmark(0, NAMESPACE, position.marks.current.label)\n parse_buffer(bufnr)\n if enable_autojump and Config.diff.autojump then\n M.find_next(side)\n vim.cmd([[normal! zz]])\n end\nend\n\nfunction M.conflict_count(bufnr)\n if bufnr and not api.nvim_buf_is_valid(bufnr) then return 0 end\n bufnr = bufnr or 0\n\n local name = api.nvim_buf_get_name(bufnr)\n if not visited_buffers[name] then return 0 end\n\n return #visited_buffers[name].positions\nend\n\nreturn M\n" }, { "path": "lua/avante/extensions/init.lua", "content": "---@class avante.extensions\nlocal M = {}\n\nsetmetatable(M, {\n __index = function(t, k)\n ---@diagnostic disable-next-line: no-unknown\n t[k] = require(\"avante.extensions.\" .. k)\n return t[k]\n end,\n})\n" }, { "path": "lua/avante/extensions/nvim_tree.lua", "content": "local Api = require(\"avante.api\")\n\n--- @class avante.extensions.nvim_tree\nlocal M = {}\n\n--- Adds the currently selected file in NvimTree to the selection via Api.add_selected_file.\n-- Notifies the user if not invoked within NvimTree or if errors occur.\n--- @return nil\nfunction M.add_file()\n if vim.bo.filetype ~= \"NvimTree\" then\n vim.notify(\"This action can only be used inside NvimTree.\", vim.log.levels.WARN)\n return\n end\n\n local ok, nvim_tree_api = pcall(require, \"nvim-tree.api\")\n if not ok then\n vim.notify(\"nvim-tree needed\", vim.log.levels.ERROR)\n return\n end\n\n local success, node = pcall(function() return nvim_tree_api.tree.get_node_under_cursor() end)\n if not success then\n vim.notify(\"Error getting node: \" .. tostring(node), vim.log.levels.ERROR)\n return\n end\n\n local filepath = node.absolute_path\n Api.add_selected_file(filepath)\nend\n\n--- Removes the currently selected file in NvimTree from the selection via Api.remove_selected_file.\n-- Notifies the user if not invoked within NvimTree or if errors occur.\n--- @return nil\nfunction M.remove_file()\n if vim.bo.filetype ~= \"NvimTree\" then\n vim.notify(\"This action can only be used inside NvimTree.\", vim.log.levels.WARN)\n return\n end\n\n local ok, nvim_tree_api = pcall(require, \"nvim-tree.api\")\n if not ok then\n vim.notify(\"nvim-tree needed\", vim.log.levels.ERROR)\n return\n end\n\n local success, node = pcall(function() return nvim_tree_api.tree.get_node_under_cursor() end)\n if not success then\n vim.notify(\"Error getting node: \" .. tostring(node), vim.log.levels.ERROR)\n return\n end\n\n local filepath = node.absolute_path\n Api.remove_selected_file(filepath)\nend\n\nreturn M\n" }, { "path": "lua/avante/file_selector.lua", "content": "local Utils = require(\"avante.utils\")\nlocal Config = require(\"avante.config\")\nlocal Selector = require(\"avante.ui.selector\")\n\nlocal PROMPT_TITLE = \"(Avante) Add a file\"\n\n--- @class FileSelector\nlocal FileSelector = {}\n\n--- @class FileSelector\n--- @field id integer\n--- @field selected_filepaths string[] Absolute paths\n--- @field event_handlers table\n\n---@alias FileSelectorHandler fun(self: FileSelector, on_select: fun(filepaths: string[] | nil)): nil\n\nlocal function has_scheme(path) return path:find(\"^(?!term://)%w+://\") ~= nil end\n\nfunction FileSelector:process_directory(absolute_path)\n if absolute_path:sub(-1) == Utils.path_sep then absolute_path = absolute_path:sub(1, -2) end\n local files = Utils.scan_directory({ directory = absolute_path, add_dirs = false })\n\n for _, file in ipairs(files) do\n local abs_path = Utils.to_absolute_path(file)\n if not vim.tbl_contains(self.selected_filepaths, abs_path) then table.insert(self.selected_filepaths, abs_path) end\n end\n self:emit(\"update\")\nend\n\n---@param selected_paths string[] | nil\n---@return nil\nfunction FileSelector:handle_path_selection(selected_paths)\n if not selected_paths then return end\n\n for _, selected_path in ipairs(selected_paths) do\n local absolute_path = Utils.to_absolute_path(selected_path)\n if vim.fn.isdirectory(absolute_path) == 1 then\n self:process_directory(absolute_path)\n else\n local abs_path = Utils.to_absolute_path(selected_path)\n if Config.file_selector.provider == \"native\" then\n table.insert(self.selected_filepaths, abs_path)\n else\n if not vim.tbl_contains(self.selected_filepaths, abs_path) then\n table.insert(self.selected_filepaths, abs_path)\n end\n end\n end\n end\n self:emit(\"update\")\nend\n\n---Scans a given directory and produces a list of files/directories with absolute paths\n---@param excluded_paths_set? table Optional set of absolute paths to exclude\n---@return { path: string, is_dir: boolean }[]\nlocal function get_project_filepaths(excluded_paths_set)\n excluded_paths_set = excluded_paths_set or {}\n local project_root = Utils.get_project_root()\n local files = Utils.scan_directory({ directory = project_root, add_dirs = true })\n return vim\n .iter(files)\n :filter(function(path) return not excluded_paths_set[path] end)\n :map(function(path)\n local is_dir = vim.fn.isdirectory(path) == 1\n return { path = path, is_dir = is_dir }\n end)\n :totable()\nend\n\n---@param id integer\n---@return FileSelector\nfunction FileSelector:new(id)\n return setmetatable({\n id = id,\n selected_filepaths = {},\n event_handlers = {},\n }, { __index = self })\nend\n\nfunction FileSelector:reset()\n self.selected_filepaths = {}\n self.event_handlers = {}\n self:emit(\"update\")\nend\n\nfunction FileSelector:add_selected_file(filepath)\n if not filepath or filepath == \"\" or has_scheme(filepath) then return end\n if filepath:match(\"^oil:\") then filepath = filepath:gsub(\"^oil:\", \"\") end\n local absolute_path = Utils.to_absolute_path(filepath)\n if vim.fn.isdirectory(absolute_path) == 1 then\n self:process_directory(absolute_path)\n return\n end\n -- Avoid duplicates\n if not vim.tbl_contains(self.selected_filepaths, absolute_path) then\n table.insert(self.selected_filepaths, absolute_path)\n self:emit(\"update\")\n end\nend\n\nfunction FileSelector:add_current_buffer()\n local current_buf = vim.api.nvim_get_current_buf()\n local filepath = vim.api.nvim_buf_get_name(current_buf)\n if filepath and filepath ~= \"\" and not has_scheme(filepath) then\n local absolute_path = Utils.to_absolute_path(filepath)\n for i, path in ipairs(self.selected_filepaths) do\n if path == absolute_path then\n table.remove(self.selected_filepaths, i)\n self:emit(\"update\")\n return true\n end\n end\n self:add_selected_file(absolute_path)\n return true\n end\n return false\nend\n\nfunction FileSelector:on(event, callback)\n local handlers = self.event_handlers[event]\n if not handlers then\n handlers = {}\n self.event_handlers[event] = handlers\n end\n\n table.insert(handlers, callback)\nend\n\nfunction FileSelector:emit(event, ...)\n local handlers = self.event_handlers[event]\n if not handlers then return end\n\n for _, handler in ipairs(handlers) do\n handler(...)\n end\nend\n\nfunction FileSelector:off(event, callback)\n if not callback then\n self.event_handlers[event] = {}\n return\n end\n local handlers = self.event_handlers[event]\n if not handlers then return end\n\n for i, handler in ipairs(handlers) do\n if handler == callback then\n table.remove(handlers, i)\n break\n end\n end\nend\n\nfunction FileSelector:open() self:show_selector_ui() end\n\nfunction FileSelector:get_filepaths()\n if type(Config.file_selector.provider_opts.get_filepaths) == \"function\" then\n ---@type avante.file_selector.opts.IGetFilepathsParams\n local params = {\n cwd = Utils.get_project_root(),\n selected_filepaths = self.selected_filepaths,\n }\n return Config.file_selector.provider_opts.get_filepaths(params)\n end\n\n local selected_filepaths_set = {}\n for _, abs_path in ipairs(self.selected_filepaths) do\n selected_filepaths_set[abs_path] = true\n end\n\n local project_root = Utils.get_project_root()\n local file_info = get_project_filepaths(selected_filepaths_set)\n\n table.sort(file_info, function(a, b)\n -- Sort alphabetically with directories being first\n if a.is_dir and not b.is_dir then\n return true\n elseif not a.is_dir and b.is_dir then\n return false\n else\n return a.path < b.path\n end\n end)\n\n return vim\n .iter(file_info)\n :map(function(info)\n local rel_path = Utils.make_relative_path(info.path, project_root)\n if info.is_dir then rel_path = rel_path .. \"/\" end\n return rel_path\n end)\n :totable()\nend\n\n---@return nil\nfunction FileSelector:show_selector_ui()\n local function handler(selected_paths) self:handle_path_selection(selected_paths) end\n\n vim.schedule(function()\n if Config.file_selector.provider ~= nil then\n Utils.warn(\"config.file_selector is deprecated, please use config.selector instead!\")\n if type(Config.file_selector.provider) == \"function\" then\n local title = string.format(\"%s:\", PROMPT_TITLE) ---@type string\n local filepaths = self:get_filepaths() ---@type string[]\n local params = { title = title, filepaths = filepaths, handler = handler } ---@type avante.file_selector.IParams\n Config.file_selector.provider(params)\n else\n ---@type avante.SelectorProvider\n local provider = \"native\"\n if Config.file_selector.provider == \"native\" then\n provider = \"native\"\n elseif Config.file_selector.provider == \"fzf\" then\n provider = \"fzf_lua\"\n elseif Config.file_selector.provider == \"mini.pick\" then\n provider = \"mini_pick\"\n elseif Config.file_selector.provider == \"snacks\" then\n provider = \"snacks\"\n elseif Config.file_selector.provider == \"telescope\" then\n provider = \"telescope\"\n elseif type(Config.file_selector.provider) == \"function\" then\n provider = Config.file_selector.provider\n end\n ---@cast provider avante.SelectorProvider\n local selector = Selector:new({\n provider = provider,\n title = PROMPT_TITLE,\n items = vim.tbl_map(function(filepath) return { id = filepath, title = filepath } end, self:get_filepaths()),\n default_item_id = self.selected_filepaths[1],\n selected_item_ids = self.selected_filepaths,\n provider_opts = Config.file_selector.provider_opts,\n on_select = function(item_ids) self:handle_path_selection(item_ids) end,\n get_preview_content = function(item_id)\n local content = Utils.read_file_from_buf_or_disk(item_id)\n local filetype = Utils.get_filetype(item_id)\n return table.concat(content or {}, \"\\n\"), filetype\n end,\n })\n selector:open()\n end\n else\n local selector = Selector:new({\n provider = Config.selector.provider,\n title = PROMPT_TITLE,\n items = vim.tbl_map(function(filepath) return { id = filepath, title = filepath } end, self:get_filepaths()),\n default_item_id = self.selected_filepaths[1],\n selected_item_ids = self.selected_filepaths,\n provider_opts = Config.selector.provider_opts,\n on_select = function(item_ids) self:handle_path_selection(item_ids) end,\n get_preview_content = function(item_id)\n local content = Utils.read_file_from_buf_or_disk(item_id)\n local filetype = Utils.get_filetype(item_id)\n return table.concat(content or {}, \"\\n\"), filetype\n end,\n })\n selector:open()\n end\n end)\n\n -- unlist the current buffer as vim.ui.select will be listed\n local winid = vim.api.nvim_get_current_win()\n local bufnr = vim.api.nvim_win_get_buf(winid)\n vim.api.nvim_set_option_value(\"buflisted\", false, { buf = bufnr })\n vim.api.nvim_set_option_value(\"bufhidden\", \"wipe\", { buf = bufnr })\nend\n\n---@param idx integer\n---@return boolean\nfunction FileSelector:remove_selected_filepaths_with_index(idx)\n if idx > 0 and idx <= #self.selected_filepaths then\n table.remove(self.selected_filepaths, idx)\n self:emit(\"update\")\n return true\n end\n return false\nend\n\nfunction FileSelector:remove_selected_file(rel_path)\n local abs_path = Utils.to_absolute_path(rel_path)\n local idx = Utils.tbl_indexof(self.selected_filepaths, abs_path)\n if idx then self:remove_selected_filepaths_with_index(idx) end\nend\n\n---@return { path: string, content: string, file_type: string }[]\nfunction FileSelector:get_selected_files_contents()\n local contents = {}\n for _, filepath in ipairs(self.selected_filepaths) do\n local lines, error = Utils.read_file_from_buf_or_disk(filepath)\n lines = lines or {}\n local filetype = Utils.get_filetype(filepath)\n if error ~= nil then\n Utils.error(\"error reading file: \" .. error)\n else\n local content = table.concat(lines, \"\\n\")\n table.insert(contents, { path = filepath, content = content, file_type = filetype })\n end\n end\n return contents\nend\n\nfunction FileSelector:get_selected_filepaths() return vim.deepcopy(self.selected_filepaths) end\n\n---@return nil\nfunction FileSelector:add_quickfix_files()\n local quickfix_files = vim\n .iter(vim.fn.getqflist({ items = 0 }).items)\n :filter(function(item) return item.bufnr ~= 0 end)\n :map(function(item) return Utils.to_absolute_path(vim.api.nvim_buf_get_name(item.bufnr)) end)\n :totable()\n for _, filepath in ipairs(quickfix_files) do\n self:add_selected_file(filepath)\n end\nend\n\n---@return nil\nfunction FileSelector:add_buffer_files()\n local buffers = vim.api.nvim_list_bufs()\n for _, bufnr in ipairs(buffers) do\n -- Skip invalid or unlisted buffers\n if vim.api.nvim_buf_is_valid(bufnr) and vim.bo[bufnr].buflisted then\n local filepath = vim.api.nvim_buf_get_name(bufnr)\n -- Skip empty paths and special buffers (like terminals)\n if filepath ~= \"\" and not has_scheme(filepath) then\n local absolute_path = Utils.to_absolute_path(filepath)\n self:add_selected_file(absolute_path)\n end\n end\n end\nend\n\nreturn FileSelector\n" }, { "path": "lua/avante/health.lua", "content": "local M = {}\nlocal H = require(\"vim.health\")\nlocal Utils = require(\"avante.utils\")\nlocal Config = require(\"avante.config\")\n\nfunction M.check()\n H.start(\"avante.nvim\")\n\n -- Required dependencies with their module names\n local required_plugins = {\n [\"plenary.nvim\"] = {\n path = \"nvim-lua/plenary.nvim\",\n module = \"plenary\",\n },\n [\"nui.nvim\"] = {\n path = \"MunifTanjim/nui.nvim\",\n module = \"nui.popup\",\n },\n }\n\n for name, plugin in pairs(required_plugins) do\n if Utils.has(name) or Utils.has(plugin.module) then\n H.ok(string.format(\"Found required plugin: %s\", plugin.path))\n else\n H.error(string.format(\"Missing required plugin: %s\", plugin.path))\n end\n end\n\n -- Optional dependencies\n if Utils.icons_enabled() then\n H.ok(\"Found icons plugin (nvim-web-devicons or mini.icons)\")\n else\n H.warn(\"No icons plugin found (nvim-web-devicons or mini.icons). Icons will not be displayed\")\n end\n\n -- Check input UI provider\n local input_provider = Config.input and Config.input.provider or \"native\"\n if input_provider == \"dressing\" then\n if Utils.has(\"dressing.nvim\") or Utils.has(\"dressing\") then\n H.ok(\"Found configured input provider: dressing.nvim\")\n else\n H.error(\"Input provider is set to 'dressing' but dressing.nvim is not installed\")\n end\n elseif input_provider == \"snacks\" then\n if Utils.has(\"snacks.nvim\") or Utils.has(\"snacks\") then\n H.ok(\"Found configured input provider: snacks.nvim\")\n else\n H.error(\"Input provider is set to 'snacks' but snacks.nvim is not installed\")\n end\n else\n H.ok(\"Using native input provider (no additional dependencies required)\")\n end\n\n -- Check Copilot if configured\n if Config.provider and Config.provider == \"copilot\" then\n if Utils.has(\"copilot.lua\") or Utils.has(\"copilot.vim\") or Utils.has(\"copilot\") then\n H.ok(\"Found Copilot plugin\")\n else\n H.error(\"Copilot provider is configured but neither copilot.lua nor copilot.vim is installed\")\n end\n end\n\n -- Check TreeSitter dependencies\n M.check_treesitter()\nend\n\n-- Check TreeSitter functionality and parsers\nfunction M.check_treesitter()\n H.start(\"TreeSitter Dependencies\")\n\n -- List of important parsers for avante.nvim\n local essential_parsers = {\n \"markdown\",\n }\n\n local missing_parsers = {} ---@type string[]\n\n for _, parser_name in ipairs(essential_parsers) do\n local loaded_parser = vim.treesitter.language.add(parser_name)\n if not loaded_parser then missing_parsers[#missing_parsers + 1] = parser_name end\n end\n\n if #missing_parsers == 0 then\n H.ok(\"All essential TreeSitter parsers are installed\")\n else\n H.warn(\n string.format(\n \"Missing recommended parsers: %s. Install with :TSInstall %s\",\n table.concat(missing_parsers, \", \"),\n table.concat(missing_parsers, \" \")\n )\n )\n end\n\n -- Check TreeSitter highlight\n local _, highlighter = pcall(require, \"vim.treesitter.highlighter\")\n if not highlighter then\n H.warn(\"TreeSitter highlighter not available. Syntax highlighting might be limited\")\n else\n H.ok(\"TreeSitter highlighter is available\")\n end\nend\n\nreturn M\n" }, { "path": "lua/avante/highlights.lua", "content": "local api = vim.api\n\nlocal Config = require(\"avante.config\")\nlocal Utils = require(\"avante.utils\")\nlocal bit = require(\"bit\")\nlocal rshift, band = bit.rshift, bit.band\n\nlocal Highlights = {\n TITLE = { name = \"AvanteTitle\", fg = \"#1e222a\", bg = \"#98c379\" },\n REVERSED_TITLE = { name = \"AvanteReversedTitle\", fg = \"#98c379\", bg_link = \"NormalFloat\" },\n SUBTITLE = { name = \"AvanteSubtitle\", fg = \"#1e222a\", bg = \"#56b6c2\" },\n REVERSED_SUBTITLE = { name = \"AvanteReversedSubtitle\", fg = \"#56b6c2\", bg_link = \"NormalFloat\" },\n THIRD_TITLE = { name = \"AvanteThirdTitle\", fg = \"#ABB2BF\", bg = \"#353B45\" },\n REVERSED_THIRD_TITLE = { name = \"AvanteReversedThirdTitle\", fg = \"#353B45\", bg_link = \"NormalFloat\" },\n SUGGESTION = { name = \"AvanteSuggestion\", link = \"Comment\" },\n ANNOTATION = { name = \"AvanteAnnotation\", link = \"Comment\" },\n POPUP_HINT = { name = \"AvantePopupHint\", link = \"NormalFloat\" },\n INLINE_HINT = { name = \"AvanteInlineHint\", link = \"Keyword\" },\n TO_BE_DELETED = { name = \"AvanteToBeDeleted\", bg = \"#ffcccc\", strikethrough = true },\n TO_BE_DELETED_WITHOUT_STRIKETHROUGH = { name = \"AvanteToBeDeletedWOStrikethrough\", bg = \"#562C30\" },\n CONFIRM_TITLE = { name = \"AvanteConfirmTitle\", fg = \"#1e222a\", bg = \"#e06c75\" },\n BUTTON_DEFAULT = { name = \"AvanteButtonDefault\", fg = \"#1e222a\", bg = \"#ABB2BF\" },\n BUTTON_DEFAULT_HOVER = { name = \"AvanteButtonDefaultHover\", fg = \"#1e222a\", bg = \"#a9cf8a\" },\n BUTTON_PRIMARY = { name = \"AvanteButtonPrimary\", fg = \"#1e222a\", bg = \"#ABB2BF\" },\n BUTTON_PRIMARY_HOVER = { name = \"AvanteButtonPrimaryHover\", fg = \"#1e222a\", bg = \"#56b6c2\" },\n BUTTON_DANGER = { name = \"AvanteButtonDanger\", fg = \"#1e222a\", bg = \"#ABB2BF\" },\n BUTTON_DANGER_HOVER = { name = \"AvanteButtonDangerHover\", fg = \"#1e222a\", bg = \"#e06c75\" },\n AVANTE_PROMPT_INPUT = { name = \"AvantePromptInput\" },\n AVANTE_PROMPT_INPUT_BORDER = { name = \"AvantePromptInputBorder\", link = \"NormalFloat\" },\n AVANTE_SIDEBAR_WIN_SEPARATOR = {\n name = \"AvanteSidebarWinSeparator\",\n fg_link_bg = \"NormalFloat\",\n bg_link = \"NormalFloat\",\n },\n AVANTE_SIDEBAR_WIN_HORIZONTAL_SEPARATOR = {\n name = \"AvanteSidebarWinHorizontalSeparator\",\n fg_link = \"WinSeparator\",\n bg_link = \"NormalFloat\",\n },\n AVANTE_SIDEBAR_NORMAL = { name = \"AvanteSidebarNormal\", link = \"NormalFloat\" },\n AVANTE_COMMENT_FG = { name = \"AvanteCommentFg\", fg_link = \"Comment\" },\n AVANTE_REVERSED_NORMAL = { name = \"AvanteReversedNormal\", fg_link_bg = \"Normal\", bg_link_fg = \"Normal\" },\n AVANTE_STATE_SPINNER_GENERATING = { name = \"AvanteStateSpinnerGenerating\", fg = \"#1e222a\", bg = \"#ab9df2\" },\n AVANTE_STATE_SPINNER_TOOL_CALLING = { name = \"AvanteStateSpinnerToolCalling\", fg = \"#1e222a\", bg = \"#56b6c2\" },\n AVANTE_STATE_SPINNER_FAILED = { name = \"AvanteStateSpinnerFailed\", fg = \"#1e222a\", bg = \"#e06c75\" },\n AVANTE_STATE_SPINNER_SUCCEEDED = { name = \"AvanteStateSpinnerSucceeded\", fg = \"#1e222a\", bg = \"#98c379\" },\n AVANTE_STATE_SPINNER_SEARCHING = { name = \"AvanteStateSpinnerSearching\", fg = \"#1e222a\", bg = \"#c678dd\" },\n AVANTE_STATE_SPINNER_THINKING = { name = \"AvanteStateSpinnerThinking\", fg = \"#1e222a\", bg = \"#c678dd\" },\n AVANTE_STATE_SPINNER_COMPACTING = { name = \"AvanteStateSpinnerCompacting\", fg = \"#1e222a\", bg = \"#c678dd\" },\n AVANTE_TASK_RUNNING = { name = \"AvanteTaskRunning\", fg = \"#c678dd\", bg_link = \"Normal\" },\n AVANTE_TASK_COMPLETED = { name = \"AvanteTaskCompleted\", fg = \"#98c379\", bg_link = \"Normal\" },\n AVANTE_TASK_FAILED = { name = \"AvanteTaskFailed\", fg = \"#e06c75\", bg_link = \"Normal\" },\n AVANTE_THINKING = { name = \"AvanteThinking\", fg = \"#c678dd\", bg_link = \"Normal\" },\n -- Gradient logo highlights\n AVANTE_LOGO_LINE_1 = { name = \"AvanteLogoLine1\", fg = \"#f5f5f5\" },\n AVANTE_LOGO_LINE_2 = { name = \"AvanteLogoLine2\", fg = \"#e8e8e8\" },\n AVANTE_LOGO_LINE_3 = { name = \"AvanteLogoLine3\", fg = \"#dbdbdb\" },\n AVANTE_LOGO_LINE_4 = { name = \"AvanteLogoLine4\", fg = \"#cfcfcf\" },\n AVANTE_LOGO_LINE_5 = { name = \"AvanteLogoLine5\", fg = \"#c2c2c2\" },\n AVANTE_LOGO_LINE_6 = { name = \"AvanteLogoLine6\", fg = \"#b5b5b5\" },\n AVANTE_LOGO_LINE_7 = { name = \"AvanteLogoLine7\", fg = \"#a8a8a8\" },\n AVANTE_LOGO_LINE_8 = { name = \"AvanteLogoLine8\", fg = \"#9b9b9b\" },\n AVANTE_LOGO_LINE_9 = { name = \"AvanteLogoLine9\", fg = \"#8e8e8e\" },\n AVANTE_LOGO_LINE_10 = { name = \"AvanteLogoLine10\", fg = \"#818181\" },\n AVANTE_LOGO_LINE_11 = { name = \"AvanteLogoLine11\", fg = \"#747474\" },\n AVANTE_LOGO_LINE_12 = { name = \"AvanteLogoLine12\", fg = \"#676767\" },\n AVANTE_LOGO_LINE_13 = { name = \"AvanteLogoLine13\", fg = \"#5a5a5a\" },\n AVANTE_LOGO_LINE_14 = { name = \"AvanteLogoLine14\", fg = \"#4d4d4d\" },\n}\n\nHighlights.conflict = {\n CURRENT = { name = \"AvanteConflictCurrent\", bg = \"#562C30\", bold = true },\n CURRENT_LABEL = { name = \"AvanteConflictCurrentLabel\", shade_link = \"AvanteConflictCurrent\", shade = 30 },\n INCOMING = { name = \"AvanteConflictIncoming\", bg = 3229523, bold = true }, -- #314753\n INCOMING_LABEL = { name = \"AvanteConflictIncomingLabel\", shade_link = \"AvanteConflictIncoming\", shade = 30 },\n}\n\n--- helper\nlocal H = {}\n\nlocal M = {}\n\nlocal function has_set_colors(hl_group) return next(Utils.get_hl(hl_group)) ~= nil end\n\nlocal first_setup = true\nlocal already_set_highlights = {}\n\nfunction M.setup()\n if Config.behaviour.auto_set_highlight_group then\n vim\n .iter(Highlights)\n :filter(function(k, _)\n -- return all uppercase key with underscore or fully uppercase key\n return k:match(\"^%u+_\") or k:match(\"^%u+$\")\n end)\n :each(function(_, hl)\n if first_setup and has_set_colors(hl.name) then already_set_highlights[hl.name] = true end\n if not already_set_highlights[hl.name] then\n local bg = hl.bg\n local fg = hl.fg\n if hl.bg_link ~= nil then bg = Utils.get_hl(hl.bg_link).bg end\n if hl.fg_link ~= nil then fg = Utils.get_hl(hl.fg_link).fg end\n if hl.bg_link_fg ~= nil then bg = Utils.get_hl(hl.bg_link_fg).fg end\n if hl.fg_link_bg ~= nil then fg = Utils.get_hl(hl.fg_link_bg).bg end\n api.nvim_set_hl(\n 0,\n hl.name,\n { fg = fg or nil, bg = bg or nil, link = hl.link or nil, strikethrough = hl.strikethrough }\n )\n end\n end)\n end\n\n if first_setup then\n vim.iter(Highlights.conflict):each(function(_, hl)\n if hl.name and has_set_colors(hl.name) then already_set_highlights[hl.name] = true end\n end)\n end\n first_setup = false\n\n M.setup_conflict_highlights()\nend\n\nfunction M.setup_conflict_highlights()\n local custom_hls = Config.highlights.diff\n\n ---@return number | nil\n local function get_bg(hl_name) return Utils.get_hl(hl_name).bg end\n\n local function get_bold(hl_name) return Utils.get_hl(hl_name).bold end\n\n vim.iter(Highlights.conflict):each(function(key, hl)\n --- set none shade linked highlights first\n if hl.shade_link ~= nil and hl.shade ~= nil then return end\n\n if already_set_highlights[hl.name] then return end\n\n local bg = hl.bg\n local bold = hl.bold\n\n local custom_hl_name = custom_hls[key:lower()]\n if custom_hl_name ~= nil then\n bg = get_bg(custom_hl_name) or hl.bg\n bold = get_bold(custom_hl_name) or hl.bold\n end\n\n api.nvim_set_hl(0, hl.name, { bg = bg, default = true, bold = bold })\n end)\n\n vim.iter(Highlights.conflict):each(function(key, hl)\n --- only set shade linked highlights\n if hl.shade_link == nil or hl.shade == nil then return end\n\n if already_set_highlights[hl.name] then return end\n\n local bg\n local bold = hl.bold\n\n local custom_hl_name = custom_hls[key:lower()]\n if custom_hl_name ~= nil then\n bg = get_bg(custom_hl_name)\n bold = get_bold(custom_hl_name) or hl.bold\n else\n local link_bg = get_bg(hl.shade_link)\n if link_bg == nil then\n Utils.warn(string.format(\"highlights %s don't have bg, use fallback\", hl.shade_link))\n link_bg = 3229523\n end\n bg = H.shade_color(link_bg, hl.shade)\n end\n\n api.nvim_set_hl(0, hl.name, { bg = bg, default = true, bold = bold })\n end)\nend\n\nsetmetatable(M, {\n __index = function(t, k)\n if Highlights[k] ~= nil then\n return Highlights[k].name\n elseif Highlights.conflict[k] ~= nil then\n return Highlights.conflict[k].name\n end\n return t[k]\n end,\n})\n\n--- Returns a table containing the RGB values encoded inside 24 least\n--- significant bits of the number @rgb_24bit\n---\n---@param rgb_24bit number 24-bit RGB value\n---@return {r: integer, g: integer, b: integer} with keys 'r', 'g', 'b' in [0,255]\nfunction H.decode_24bit_rgb(rgb_24bit)\n if vim.fn.has(\"nvim-0.11\") == 1 then\n vim.validate(\"rgb_24bit\", rgb_24bit, \"number\", true)\n else\n vim.validate({ rgb_24bit = { rgb_24bit, \"number\", true } })\n end\n local r = band(rshift(rgb_24bit, 16), 255)\n local g = band(rshift(rgb_24bit, 8), 255)\n local b = band(rgb_24bit, 255)\n return { r = r, g = g, b = b }\nend\n\n---@param attr integer\n---@param percent integer\nfunction H.alter(attr, percent) return math.floor(attr * (100 + percent) / 100) end\n\n---@source https://stackoverflow.com/q/5560248\n---@see https://stackoverflow.com/a/37797380\n---Lighten a specified hex color\n---@param color number\n---@param percent number\n---@return string\nfunction H.shade_color(color, percent)\n percent = vim.opt.background:get() == \"light\" and percent / 5 or percent\n local rgb = H.decode_24bit_rgb(color)\n if not rgb.r or not rgb.g or not rgb.b then return \"NONE\" end\n local r, g, b = H.alter(rgb.r, percent), H.alter(rgb.g, percent), H.alter(rgb.b, percent)\n r, g, b = math.min(r, 255), math.min(g, 255), math.min(b, 255)\n return string.format(\"#%02x%02x%02x\", r, g, b)\nend\n\nreturn M\n" }, { "path": "lua/avante/history/helpers.lua", "content": "local Utils = require(\"avante.utils\")\n\nlocal M = {}\n\n---If message is a text message return the text.\n---@param message avante.HistoryMessage\n---@return string | nil\nfunction M.get_text_data(message)\n local content = message.message.content\n if type(content) == \"table\" then\n assert(#content == 1, \"more than one entry in message content\")\n local item = content[1]\n if type(item) == \"string\" then\n return item\n elseif type(item) == \"table\" and item.type == \"text\" then\n return item.content\n end\n elseif type(content) == \"string\" then\n return content\n end\nend\n\n---If message is a \"tool use\" message returns information about the tool invocation.\n---@param message avante.HistoryMessage\n---@return AvanteLLMToolUse | nil\nfunction M.get_tool_use_data(message)\n local content = message.message.content\n if type(content) == \"table\" then\n assert(#content == 1, \"more than one entry in message content\")\n local item = content[1]\n if item.type == \"tool_use\" then\n ---@cast item AvanteLLMToolUse\n return item\n end\n end\nend\n\n---If message is a \"tool result\" message returns results of the tool invocation.\n---@param message avante.HistoryMessage\n---@return AvanteLLMToolResult | nil\nfunction M.get_tool_result_data(message)\n local content = message.message.content\n if type(content) == \"table\" then\n assert(#content == 1, \"more than one entry in message content\")\n local item = content[1]\n if item.type == \"tool_result\" then\n ---@cast item AvanteLLMToolResult\n return item\n end\n end\nend\n\n---Attempts to locate result of a tool execution given tool invocation ID\n---@param id string\n---@param messages avante.HistoryMessage[]\n---@return AvanteLLMToolResult | nil\nfunction M.get_tool_result(id, messages)\n for idx = #messages, 1, -1 do\n local msg = messages[idx]\n local result = M.get_tool_result_data(msg)\n if result and result.tool_use_id == id then return result end\n end\nend\n\n---Given a tool invocation ID locate corresponding tool use message\n---@param id string\n---@param messages avante.HistoryMessage[]\n---@return avante.HistoryMessage | nil\nfunction M.get_tool_use_message(id, messages)\n for idx = #messages, 1, -1 do\n local msg = messages[idx]\n local use = M.get_tool_use_data(msg)\n if use and use.id == id then return msg end\n end\nend\n\n---Given a tool invocation ID locate corresponding tool result message\n---@param id string\n---@param messages avante.HistoryMessage[]\n---@return avante.HistoryMessage | nil\nfunction M.get_tool_result_message(id, messages)\n for idx = #messages, 1, -1 do\n local msg = messages[idx]\n local result = M.get_tool_result_data(msg)\n if result and result.tool_use_id == id then return msg end\n end\nend\n\n---@param message avante.HistoryMessage\n---@return boolean\nfunction M.is_thinking_message(message)\n local content = message.message.content\n return type(content) == \"table\" and (content[1].type == \"thinking\" or content[1].type == \"redacted_thinking\")\nend\n\n---@param message avante.HistoryMessage\n---@return boolean\nfunction M.is_tool_result_message(message) return M.get_tool_result_data(message) ~= nil end\n\n---@param message avante.HistoryMessage\n---@return boolean\nfunction M.is_tool_use_message(message) return M.get_tool_use_data(message) ~= nil end\n\nreturn M\n" }, { "path": "lua/avante/history/init.lua", "content": "local Helpers = require(\"avante.history.helpers\")\nlocal Message = require(\"avante.history.message\")\nlocal Utils = require(\"avante.utils\")\n\nlocal M = {}\n\nM.Helpers = Helpers\nM.Message = Message\n\n---@param history avante.ChatHistory\n---@return avante.HistoryMessage[]\nfunction M.get_history_messages(history)\n if history.messages then return history.messages end\n local messages = {}\n for _, entry in ipairs(history.entries or {}) do\n if entry.request and entry.request ~= \"\" then\n local message = Message:new(\"user\", entry.request, {\n timestamp = entry.timestamp,\n is_user_submission = true,\n visible = entry.visible,\n selected_filepaths = entry.selected_filepaths,\n selected_code = entry.selected_code,\n })\n table.insert(messages, message)\n end\n if entry.response and entry.response ~= \"\" then\n local message = Message:new(\"assistant\", entry.response, {\n timestamp = entry.timestamp,\n visible = entry.visible,\n })\n table.insert(messages, message)\n end\n end\n history.messages = messages\n return messages\nend\n\n---Represents information about tool use: invocation, result, affected file (for \"view\" or \"edit\" tools).\n---@class HistoryToolInfo\n---@field kind \"edit\" | \"view\" | \"other\"\n---@field use AvanteLLMToolUse\n---@field result? AvanteLLMToolResult\n---@field result_message? avante.HistoryMessage Complete result message\n---@field path? string Uniform (normalized) path of the affected file\n\n---@class HistoryFileInfo\n---@field last_tool_id? string ID of the tool with most up-to-date state of the file\n---@field edit_tool_id? string ID of the last tool done edit on the file\n\n---Collects information about all uses of tools in the history: their invocations, results, and affected files.\n---@param messages avante.HistoryMessage[]\n---@return table\n---@return table\nlocal function collect_tool_info(messages)\n ---@type table Maps tool ID to tool information\n local tools = {}\n ---@type table Maps file path to file information\n local files = {}\n\n -- Collect invocations of all tools, and also build a list of viewed or edited files.\n for _, message in ipairs(messages) do\n local use = Helpers.get_tool_use_data(message)\n if use then\n if use.name == \"view\" or Utils.is_edit_tool_use(use) then\n if use.input.path then\n local path = Utils.uniform_path(use.input.path)\n if use.id then tools[use.id] = { kind = use.name == \"view\" and \"view\" or \"edit\", use = use, path = path } end\n end\n else\n if use.id then tools[use.id] = { kind = \"other\", use = use } end\n end\n goto continue\n end\n\n local result = Helpers.get_tool_result_data(message)\n if result then\n -- We assume that \"result\" entries always come after corresponding \"use\" entries.\n local info = tools[result.tool_use_id]\n if info then\n info.result = result\n info.result_message = message\n if info.path then\n local f = files[info.path]\n if not f then\n f = {}\n files[info.path] = f\n end\n f.last_tool_id = result.tool_use_id\n if info.kind == \"edit\" and not (result.is_error or result.is_user_declined) then\n f.edit_tool_id = result.tool_use_id\n end\n end\n end\n end\n\n ::continue::\n end\n\n return tools, files\nend\n\n---Converts a tool invocation (use + result) into a simple request/response pair of text messages\n---@param tool_info HistoryToolInfo\n---@return avante.HistoryMessage[]\nlocal function convert_tool_to_text(tool_info)\n return {\n Message:new_assistant_synthetic(\n string.format(\"Tool use %s(%s)\", tool_info.use.name, vim.json.encode(tool_info.use.input))\n ),\n Message:new_user_synthetic({\n type = \"text\",\n text = string.format(\n \"Tool use [%s] is successful: %s\",\n tool_info.use.name,\n tostring(not tool_info.result.is_error)\n ),\n }),\n }\nend\n\n---Generates a fake file \"content\" telling LLM to look further for up-to-date data\n---@param path string\n---@return string\nlocal function stale_view_content(path)\n return string.format(\"The file %s has been updated. Please use the latest `view` tool result!\", path)\nend\n\n---Updates the result of \"view\" tool invocation with latest contents of a buffer or file,\n---or a stub message if this result will be superseded by another one.\n---@param tool_info HistoryToolInfo\n---@param stale_view boolean\nlocal function update_view_result(tool_info, stale_view)\n local use = tool_info.use\n local result = tool_info.result\n\n if stale_view then\n result.content = stale_view_content(tool_info.path)\n else\n local view_result, view_error = require(\"avante.llm_tools.view\").func(\n { path = tool_info.path, start_line = use.input.start_line, end_line = use.input.end_line },\n {}\n )\n result.content = view_error and (\"Error: \" .. view_error) or view_result\n result.is_error = view_error ~= nil\n end\nend\n\n---Generates synthetic \"view\" tool invocation to tell LLM to refresh its view of a file after editing\n---@param tool_use AvanteLLMToolUse\n---@param path any\n---@param stale_view any\n---@return avante.HistoryMessage[]\nlocal function generate_view_messages(tool_use, path, stale_view)\n local view_result, view_error\n if stale_view then\n view_result = stale_view_content(path)\n else\n view_result, view_error = require(\"avante.llm_tools.view\").func({ path = path }, {})\n end\n\n if view_error then view_result = \"Error: \" .. view_error end\n\n local view_tool_use_id = Utils.generate_call_tool_id()\n local view_tool_name = \"view\"\n local view_tool_input = { path = path }\n\n if tool_use.name == \"str_replace_editor\" and tool_use.input.command == \"str_replace\" then\n view_tool_name = \"str_replace_editor\"\n view_tool_input.command = \"view\"\n elseif tool_use.name == \"str_replace_based_edit_tool\" and tool_use.input.command == \"str_replace\" then\n view_tool_name = \"str_replace_based_edit_tool\"\n view_tool_input.command = \"view\"\n end\n\n return {\n Message:new_assistant_synthetic(string.format(\"Viewing file %s to get the latest content\", path)),\n Message:new_assistant_synthetic({\n type = \"tool_use\",\n id = view_tool_use_id,\n name = view_tool_name,\n input = view_tool_input,\n }),\n Message:new_user_synthetic({\n type = \"tool_result\",\n tool_use_id = view_tool_use_id,\n content = view_result,\n is_error = view_error ~= nil,\n is_user_declined = false,\n }),\n }\nend\n\n---Generates \"diagnostic\" for a file after it has been edited to help catching errors\n---@param path string\n---@return avante.HistoryMessage[]\nlocal function generate_diagnostic_messages(path)\n local get_diagnostics_tool_use_id = Utils.generate_call_tool_id()\n local diagnostics = Utils.lsp.get_diagnostics_from_filepath(path)\n return {\n Message:new_assistant_synthetic(\n string.format(\"The file %s has been modified, let me check if there are any errors in the changes.\", path)\n ),\n Message:new_assistant_synthetic({\n type = \"tool_use\",\n id = get_diagnostics_tool_use_id,\n name = \"get_diagnostics\",\n input = { path = path },\n }),\n Message:new_user_synthetic({\n type = \"tool_result\",\n tool_use_id = get_diagnostics_tool_use_id,\n content = vim.json.encode(diagnostics),\n is_error = false,\n is_user_declined = false,\n }),\n }\nend\n\n---Iterate through history messages and generate a new list containing updated history\n---that has up-to-date file contents and potentially updated diagnostic for modified\n---files.\n---@param messages avante.HistoryMessage[]\n---@param tools HistoryToolInfo[]\n---@param files HistoryFileInfo[]\n---@param add_diagnostic boolean Whether to generate and add diagnostic info to \"edit\" invocations\n---@param tools_to_text integer Number of tool invocations to be converted to simple text\n---@return avante.HistoryMessage[]\nlocal function refresh_history(messages, tools, files, add_diagnostic, tools_to_text)\n ---@type avante.HistoryMessage[]\n local updated_messages = {}\n local tool_count = 0\n\n for _, message in ipairs(messages) do\n local use = Helpers.get_tool_use_data(message)\n if use then\n -- This is a tool invocation message. We will be handling both use and result together.\n local tool_info = tools[use.id]\n if not tool_info then goto continue end\n if not tool_info.result then goto continue end\n\n if tool_count < tools_to_text then\n local text_msgs = convert_tool_to_text(tool_info)\n Utils.debug(\"Converted\", use.name, \"invocation to\", #text_msgs, \"messages\")\n updated_messages = vim.list_extend(updated_messages, text_msgs)\n else\n table.insert(updated_messages, message)\n table.insert(updated_messages, tool_info.result_message)\n tool_count = tool_count + 1\n\n if tool_info.kind == \"view\" then\n local path = tool_info.path\n assert(path, \"encountered 'view' tool invocation without path\")\n update_view_result(tool_info, use.id ~= files[tool_info.path].last_tool_id)\n end\n end\n\n if tool_info.kind == \"edit\" then\n local path = tool_info.path\n assert(path, \"encountered 'edit' tool invocation without path\")\n local file_info = files[path]\n\n -- If this is the last operation for this file, generate synthetic \"view\"\n -- invocation to provide the up-to-date file contents.\n if not tool_info.result.is_error then\n local view_msgs = generate_view_messages(use, path, use.id == file_info.last_tool_id)\n Utils.debug(\"Added\", #view_msgs, \"'view' tool messages for\", path)\n updated_messages = vim.list_extend(updated_messages, view_msgs)\n tool_count = tool_count + 1\n end\n\n if add_diagnostic and use.id == file_info.edit_tool_id then\n local diag_msgs = generate_diagnostic_messages(path)\n Utils.debug(\"Added\", #diag_msgs, \"'diagnostics' tool messages for\", path)\n updated_messages = vim.list_extend(updated_messages, diag_msgs)\n tool_count = tool_count + 1\n end\n end\n elseif not Helpers.get_tool_result_data(message) then\n -- Skip the tool result messages (since we process them together with their \"use\"s.\n -- All other (non-tool-related) messages we simply keep.\n table.insert(updated_messages, message)\n end\n\n ::continue::\n end\n\n return updated_messages\nend\n\n---Analyzes the history looking for tool invocations, drops incomplete invocations,\n---and updates complete ones with the latest data available.\n---@param messages avante.HistoryMessage[]\n---@param max_tool_use integer | nil Maximum number of tool invocations to keep\n---@param add_diagnostic boolean Mix in LSP diagnostic info for affected files\n---@return avante.HistoryMessage[]\nM.update_tool_invocation_history = function(messages, max_tool_use, add_diagnostic)\n local tools, files = collect_tool_info(messages)\n\n -- Figure number of tool invocations that should be converted to simple \"text\"\n -- messages to reduce prompt costs.\n local tools_to_text = 0\n if max_tool_use then\n local n_edits = vim.iter(files):fold(\n 0,\n ---@param count integer\n ---@param file_info HistoryFileInfo\n function(count, file_info)\n if file_info.edit_tool_id then count = count + 1 end\n return count\n end\n )\n -- Each valid \"edit\" invocation will result in synthetic \"view\" and also\n -- in \"diagnostic\" if it is requested by the caller.\n local expected = #tools + n_edits + (add_diagnostic and n_edits or 0)\n tools_to_text = expected - max_tool_use\n end\n\n return refresh_history(messages, tools, files, add_diagnostic, tools_to_text)\nend\n\n---Scans message history backwards, looking for tool invocations that have not been executed yet\n---@param messages avante.HistoryMessage[]\n---@return AvantePartialLLMToolUse[]\n---@return avante.HistoryMessage[]\nfunction M.get_pending_tools(messages)\n local last_turn_id = nil\n if #messages > 0 then last_turn_id = messages[#messages].turn_id end\n\n local pending_tool_uses = {} ---@type AvantePartialLLMToolUse[]\n local pending_tool_uses_messages = {} ---@type avante.HistoryMessage[]\n local tool_result_seen = {}\n\n for idx = #messages, 1, -1 do\n local message = messages[idx]\n\n if last_turn_id and message.turn_id ~= last_turn_id then break end\n\n local use = Helpers.get_tool_use_data(message)\n if use then\n if not tool_result_seen[use.id] then\n local partial_tool_use = {\n name = use.name,\n id = use.id,\n input = use.input,\n state = message.state,\n }\n table.insert(pending_tool_uses, 1, partial_tool_use)\n table.insert(pending_tool_uses_messages, 1, message)\n end\n goto continue\n end\n\n local result = Helpers.get_tool_result_data(message)\n if result then tool_result_seen[result.tool_use_id] = true end\n\n ::continue::\n end\n\n return pending_tool_uses, pending_tool_uses_messages\nend\n\nreturn M\n" }, { "path": "lua/avante/history/message.lua", "content": "local Utils = require(\"avante.utils\")\n\n---@class avante.HistoryMessage\nlocal M = {}\nM.__index = M\n\n---@class avante.HistoryMessage.Opts\n---@field uuid? string\n---@field turn_id? string\n---@field state? avante.HistoryMessageState\n---@field displayed_content? string\n---@field original_content? AvanteLLMMessageContent\n---@field selected_code? AvanteSelectedCode\n---@field selected_filepaths? string[]\n---@field is_calling? boolean\n---@field is_dummy? boolean\n---@field is_user_submission? boolean\n---@field just_for_display? boolean\n---@field visible? boolean\n---\n---@param role \"user\" | \"assistant\"\n---@param content AvanteLLMMessageContentItem\n---@param opts? avante.HistoryMessage.Opts\n---@return avante.HistoryMessage\nfunction M:new(role, content, opts)\n ---@type AvanteLLMMessage\n local message = { role = role, content = type(content) == \"string\" and content or { content } }\n local obj = {\n message = message,\n uuid = Utils.uuid(),\n state = \"generated\",\n timestamp = Utils.get_timestamp(),\n is_user_submission = false,\n visible = true,\n }\n obj = vim.tbl_extend(\"force\", obj, opts or {})\n return setmetatable(obj, M)\nend\n\n---Creates a new instance of synthetic (dummy) history message\n---@param role \"assistant\" | \"user\"\n---@param item AvanteLLMMessageContentItem\n---@return avante.HistoryMessage\nfunction M:new_synthetic(role, item) return M:new(role, item, { is_dummy = true }) end\n\n---Creates a new instance of synthetic (dummy) history message attributed to the assistant\n---@param item AvanteLLMMessageContentItem\n---@return avante.HistoryMessage\nfunction M:new_assistant_synthetic(item) return M:new_synthetic(\"assistant\", item) end\n\n---Creates a new instance of synthetic (dummy) history message attributed to the user\n---@param item AvanteLLMMessageContentItem\n---@return avante.HistoryMessage\nfunction M:new_user_synthetic(item) return M:new_synthetic(\"user\", item) end\n\n---Updates content of a message as long as it is a simple text (or empty).\n---@param new_content string\nfunction M:update_content(new_content)\n assert(type(self.message.content) == \"string\", \"can only update content of simple string messages\")\n self.message.content = new_content\nend\n\nreturn M\n" }, { "path": "lua/avante/history/render.lua", "content": "local Helpers = require(\"avante.history.helpers\")\nlocal Line = require(\"avante.ui.line\")\nlocal Utils = require(\"avante.utils\")\nlocal Highlights = require(\"avante.highlights\")\n\nlocal M = {}\n\n---@diagnostic disable-next-line: deprecated\nlocal islist = vim.islist or vim.tbl_islist\n\n---Converts text into format suitable for UI\n---@param text string\n---@param decoration string | nil\n---@param filter? fun(text_line: string, idx: integer, len: integer): boolean\n---@return avante.ui.Line[]\nlocal function text_to_lines(text, decoration, filter)\n local text_lines = vim.split(text, \"\\n\")\n local lines = {}\n for idx, text_line in ipairs(text_lines) do\n if filter and not filter(text_line, idx, #text_lines) then goto continue end\n if decoration then\n table.insert(lines, Line:new({ { decoration }, { text_line } }))\n else\n table.insert(lines, Line:new({ { text_line } }))\n end\n ::continue::\n end\n return lines\nend\n\n---Converts text into format suitable for UI\n---@param text string\n---@param decoration string | nil\n---@param truncate boolean | nil\n---@return avante.ui.Line[]\nlocal function text_to_truncated_lines(text, decoration, truncate)\n local text_lines = vim.split(text, \"\\n\")\n local lines = {}\n for _, text_line in ipairs(text_lines) do\n if truncate and #lines > 3 then\n table.insert(\n lines,\n Line:new({\n { decoration },\n {\n string.format(\"... (Result truncated, remaining %d lines not shown)\", #text_lines - #lines + 1),\n Highlights.AVANTE_COMMENT_FG,\n },\n })\n )\n break\n end\n table.insert(lines, Line:new({ { decoration }, { text_line } }))\n end\n return lines\nend\n\n---@param lines avante.ui.Line[]\n---@param decoration string | nil\n---@param truncate boolean | nil\n---@return avante.ui.Line[]\nlocal function lines_to_truncated_lines(lines, decoration, truncate)\n local truncated_lines = {}\n for idx, line in ipairs(lines) do\n if truncate and #truncated_lines > 3 then\n table.insert(\n truncated_lines,\n Line:new({\n { decoration },\n {\n string.format(\"... (Result truncated, remaining %d lines not shown)\", #lines - idx + 1),\n Highlights.AVANTE_COMMENT_FG,\n },\n })\n )\n break\n end\n table.insert(truncated_lines, line)\n end\n return truncated_lines\nend\n\n---Converts \"thinking\" item into format suitable for UI\n---@param item AvanteLLMMessageContentItem\n---@return avante.ui.Line[]\nlocal function thinking_to_lines(item)\n local text = item.thinking or item.data or \"\"\n local text_lines = vim.split(text, \"\\n\")\n --- trim prefix empty lines\n while #text_lines > 0 and text_lines[1] == \"\" do\n table.remove(text_lines, 1)\n end\n --- trim suffix empty lines\n while #text_lines > 0 and text_lines[#text_lines] == \"\" do\n table.remove(text_lines, #text_lines)\n end\n local ui_lines = {}\n table.insert(ui_lines, Line:new({ { Utils.icon(\"🤔 \") .. \"Thought content:\" } }))\n table.insert(ui_lines, Line:new({ { \"\" } }))\n for _, text_line in ipairs(text_lines) do\n table.insert(ui_lines, Line:new({ { \"> \" .. text_line } }))\n end\n return ui_lines\nend\n\n---Converts logs generated by a tool during execution into format suitable for UI\n---@param tool_name string\n---@param logs string[]\n---@return avante.ui.Line[]\nfunction M.tool_logs_to_lines(tool_name, logs)\n local ui_lines = {}\n local num_logs = #logs\n\n for log_idx = 1, num_logs do\n local log_lines = vim.split(logs[log_idx]:gsub(\"^%[\" .. tool_name .. \"%]: \", \"\", 1), \"\\n\")\n local num_lines = #log_lines\n\n for line_idx = 1, num_lines do\n local decoration = \"│ \"\n table.insert(ui_lines, Line:new({ { decoration }, { \" \" .. log_lines[line_idx] } }))\n end\n end\n return ui_lines\nend\n\nlocal STATE_TO_HL = {\n generating = \"AvanteStateSpinnerToolCalling\",\n failed = \"AvanteStateSpinnerFailed\",\n succeeded = \"AvanteStateSpinnerSucceeded\",\n}\n\nfunction M.get_diff_lines(old_str, new_str, decoration, truncate)\n local lines = {}\n local line_count = 0\n local old_lines = vim.split(old_str, \"\\n\")\n local new_lines = vim.split(new_str, \"\\n\")\n ---@diagnostic disable-next-line: assign-type-mismatch, missing-fields\n local patch = vim.diff(old_str, new_str, { ---@type integer[][]\n algorithm = \"histogram\",\n result_type = \"indices\",\n ctxlen = vim.o.scrolloff,\n })\n local prev_start_a = 0\n local truncated_lines = 0\n for _, hunk in ipairs(patch) do\n local start_a, count_a, start_b, count_b = unpack(hunk)\n local no_change_lines = vim.list_slice(old_lines, prev_start_a, start_a - 1)\n if truncate then\n local last_three_no_change_lines = vim.list_slice(no_change_lines, #no_change_lines - 3)\n truncated_lines = truncated_lines + #no_change_lines - #last_three_no_change_lines\n if #no_change_lines > 4 then\n table.insert(lines, Line:new({ { decoration }, { \"...\", Highlights.AVANTE_COMMENT_FG } }))\n end\n no_change_lines = last_three_no_change_lines\n end\n for idx, line in ipairs(no_change_lines) do\n if truncate and line_count > 10 then\n truncated_lines = truncated_lines + #no_change_lines - idx\n break\n end\n line_count = line_count + 1\n table.insert(lines, Line:new({ { decoration }, { line } }))\n end\n prev_start_a = start_a + count_a\n if count_a > 0 then\n local delete_lines = vim.list_slice(old_lines, start_a, start_a + count_a - 1)\n for idx, line in ipairs(delete_lines) do\n if truncate and line_count > 10 then\n truncated_lines = truncated_lines + #delete_lines - idx\n break\n end\n line_count = line_count + 1\n table.insert(lines, Line:new({ { decoration }, { line, Highlights.TO_BE_DELETED_WITHOUT_STRIKETHROUGH } }))\n end\n end\n if count_b > 0 then\n local create_lines = vim.list_slice(new_lines, start_b, start_b + count_b - 1)\n for idx, line in ipairs(create_lines) do\n if truncate and line_count > 10 then\n truncated_lines = truncated_lines + #create_lines - idx\n break\n end\n line_count = line_count + 1\n table.insert(lines, Line:new({ { decoration }, { line, Highlights.INCOMING } }))\n end\n end\n end\n if prev_start_a < #old_lines then\n -- Append remaining old_lines\n local no_change_lines = vim.list_slice(old_lines, prev_start_a, #old_lines)\n local first_three_no_change_lines = vim.list_slice(no_change_lines, 1, 3)\n for idx, line in ipairs(first_three_no_change_lines) do\n if truncate and line_count > 10 then\n truncated_lines = truncated_lines + #first_three_no_change_lines - idx\n break\n end\n line_count = line_count + 1\n table.insert(lines, Line:new({ { decoration }, { line } }))\n end\n end\n if truncate and truncated_lines > 0 then\n table.insert(\n lines,\n Line:new({\n { decoration },\n {\n string.format(\"... (Result truncated, remaining %d lines not shown)\", truncated_lines),\n Highlights.AVANTE_COMMENT_FG,\n },\n })\n )\n end\n return lines\nend\n\n---@param content any\n---@param decoration string | nil\n---@param truncate boolean | nil\nfunction M.get_content_lines(content, decoration, truncate)\n local lines = {}\n local content_obj = content\n if type(content) == \"string\" then\n local ok, content_obj_ = pcall(vim.json.decode, content)\n if ok then content_obj = content_obj_ end\n end\n if type(content_obj) == \"table\" then\n if islist(content_obj) then\n local all_lines = {}\n for _, content_item in ipairs(content_obj) do\n if type(content_item) == \"string\" then\n local lines_ = text_to_lines(content_item, decoration)\n all_lines = vim.list_extend(all_lines, lines_)\n end\n end\n local lines_ = lines_to_truncated_lines(all_lines, decoration, truncate)\n lines = vim.list_extend(lines, lines_)\n end\n if type(content_obj.content) == \"string\" then\n local lines_ = text_to_truncated_lines(content_obj.content, decoration, truncate)\n lines = vim.list_extend(lines, lines_)\n end\n if islist(content_obj.content) then\n local all_lines = {}\n for _, content_item in ipairs(content_obj.content) do\n if type(content_item) == \"string\" then\n local lines_ = text_to_lines(content_item, decoration)\n all_lines = vim.list_extend(all_lines, lines_)\n end\n end\n local lines_ = lines_to_truncated_lines(all_lines, decoration, truncate)\n lines = vim.list_extend(lines, lines_)\n end\n if islist(content_obj.matches) then\n local all_lines = {}\n for _, content_item in ipairs(content_obj.matches) do\n if type(content_item) == \"string\" then\n local lines_ = text_to_lines(content_item, decoration)\n all_lines = vim.list_extend(all_lines, lines_)\n end\n end\n local lines_ = lines_to_truncated_lines(all_lines, decoration, truncate)\n lines = vim.list_extend(lines, lines_)\n end\n end\n if type(content_obj) == \"string\" then\n local lines_ = text_to_lines(content_obj, decoration)\n local line_count = 0\n for idx, line in ipairs(lines_) do\n if truncate and line_count > 3 then\n table.insert(\n lines,\n Line:new({\n { decoration },\n {\n string.format(\"... (Result truncated, remaining %d lines not shown)\", #lines_ - idx + 1),\n Highlights.AVANTE_COMMENT_FG,\n },\n })\n )\n break\n end\n line_count = line_count + 1\n table.insert(lines, line)\n end\n end\n if type(content_obj) == \"number\" then\n table.insert(lines, Line:new({ { decoration }, { tostring(content_obj) } }))\n end\n if islist(content) then\n for _, content_item in ipairs(content) do\n local line_count = 0\n if content_item.type == \"content\" then\n if content_item.content.type == \"text\" then\n local lines_ = text_to_lines(content_item.content.text, decoration, function(text_line, idx, len)\n if idx == 1 and text_line:match(\"^%s*```%s*$\") then return false end\n if idx == len and text_line:match(\"^%s*```%s*$\") then return false end\n return true\n end)\n for idx, line in ipairs(lines_) do\n if truncate and line_count > 3 then\n table.insert(\n lines,\n Line:new({\n { decoration },\n {\n string.format(\"... (Result truncated, remaining %d lines not shown)\", #lines_ - idx + 1),\n Highlights.AVANTE_COMMENT_FG,\n },\n })\n )\n break\n end\n line_count = line_count + 1\n table.insert(lines, line)\n end\n end\n elseif\n content_item.type == \"diff\"\n and content_item.oldText ~= nil\n and content_item.newText ~= nil\n and content_item.oldText ~= vim.NIL\n and content_item.newText ~= vim.NIL\n then\n local relative_path = Utils.relative_path(content_item.path)\n table.insert(lines, Line:new({ { decoration }, { \"Path: \" .. relative_path } }))\n local lines_ = M.get_diff_lines(content_item.oldText, content_item.newText, decoration, truncate)\n lines = vim.list_extend(lines, lines_)\n end\n end\n end\n return lines\nend\n\n---@param message avante.HistoryMessage\n---@return string tool_name\n---@return string | nil error\nfunction M.get_tool_display_name(message)\n local content = message.message.content\n if type(content) ~= \"table\" then return \"\", \"expected message content to be a table\" end\n\n ---@cast content AvanteLLMMessageContentItem[]\n\n if not islist(content) then return \"\", \"expected message content to be a list\" end\n\n local item = message.message.content[1]\n\n local native_tool_name = item.name\n if native_tool_name == \"other\" and message.acp_tool_call then\n native_tool_name = message.acp_tool_call.title or \"Other\"\n end\n if message.acp_tool_call and message.acp_tool_call.title then native_tool_name = message.acp_tool_call.title end\n local tool_name = native_tool_name\n if message.displayed_tool_name then\n tool_name = message.displayed_tool_name\n else\n local param\n if item.input and type(item.input) == \"table\" then\n local path\n if type(item.input.path) == \"string\" then path = item.input.path end\n if type(item.input.rel_path) == \"string\" then path = item.input.rel_path end\n if type(item.input.filepath) == \"string\" then path = item.input.filepath end\n if type(item.input.file_path) == \"string\" then path = item.input.file_path end\n if type(item.input.query) == \"string\" then param = item.input.query end\n if type(item.input.pattern) == \"string\" then param = item.input.pattern end\n if type(item.input.command) == \"string\" then\n param = item.input.command\n local pieces = vim.split(param, \"\\n\")\n if #pieces > 1 then param = pieces[1] .. \"...\" end\n end\n if native_tool_name == \"execute\" and not param then\n if message.acp_tool_call and message.acp_tool_call.title then param = message.acp_tool_call.title end\n end\n if not param and path then\n local relative_path = Utils.relative_path(path)\n param = relative_path\n end\n end\n if not param and message.acp_tool_call then\n if message.acp_tool_call.locations then\n for _, location in ipairs(message.acp_tool_call.locations) do\n if location.path then\n local relative_path = Utils.relative_path(location.path)\n param = relative_path\n break\n end\n end\n end\n end\n if\n not param\n and message.acp_tool_call\n and message.acp_tool_call.rawInput\n and message.acp_tool_call.rawInput.command\n then\n param = message.acp_tool_call.rawInput.command\n pcall(function()\n local project_root = Utils.root.get()\n param = param:gsub(project_root .. \"/?\", \"\")\n end)\n end\n if param then tool_name = native_tool_name .. \"(\" .. vim.inspect(param) .. \")\" end\n end\n\n ---@cast tool_name string\n\n return tool_name, nil\nend\n\n---Converts a tool invocation into format suitable for UI\n---@param item AvanteLLMMessageContentItem\n---@param message avante.HistoryMessage\n---@param messages avante.HistoryMessage[]\n---@param expanded boolean | nil\n---@return avante.ui.Line[]\nlocal function tool_to_lines(item, message, messages, expanded)\n -- local logs = message.tool_use_logs\n local lines = {}\n\n local tool_name, error = M.get_tool_display_name(message)\n if error then\n table.insert(lines, Line:new({ { \"❌ \" }, { error } }))\n return lines\n end\n\n local rest_input_text_lines = {}\n\n local result = Helpers.get_tool_result(item.id, messages)\n local state\n if not result then\n state = \"generating\"\n elseif result.is_error then\n state = \"failed\"\n else\n state = \"succeeded\"\n end\n table.insert(\n lines,\n Line:new({\n { \"╭─ \" },\n { \" \" .. tool_name .. \" \", STATE_TO_HL[state] },\n { \" \" .. state },\n })\n )\n -- if logs then vim.list_extend(lines, tool_logs_to_lines(item.name, logs)) end\n local decoration = \"│ \"\n if rest_input_text_lines and #rest_input_text_lines > 0 then\n local lines_ = text_to_lines(table.concat(rest_input_text_lines, \"\\n\"), decoration)\n local line_count = 0\n for idx, line in ipairs(lines_) do\n if not expanded and line_count > 3 then\n table.insert(\n lines,\n Line:new({\n { decoration },\n {\n string.format(\"... (Input truncated, remaining %d lines not shown)\", #lines_ - idx + 1),\n Highlights.AVANTE_COMMENT_FG,\n },\n })\n )\n break\n end\n line_count = line_count + 1\n table.insert(lines, line)\n end\n table.insert(lines, Line:new({ { decoration }, { \"\" } }))\n end\n local add_diff_lines = false\n if item.input and type(item.input) == \"table\" then\n if type(item.input.old_str) == \"string\" and type(item.input.new_str) == \"string\" then\n local diff_lines = M.get_diff_lines(item.input.old_str, item.input.new_str, decoration, not expanded)\n add_diff_lines = true\n vim.list_extend(lines, diff_lines)\n end\n end\n if\n not add_diff_lines\n and message.acp_tool_call\n and message.acp_tool_call.rawInput\n and message.acp_tool_call.rawInput.oldString\n then\n local diff_lines = M.get_diff_lines(\n message.acp_tool_call.rawInput.oldString,\n message.acp_tool_call.rawInput.newString,\n decoration,\n not expanded\n )\n vim.list_extend(lines, diff_lines)\n end\n if\n message.acp_tool_call\n and message.acp_tool_call.rawOutput\n and message.acp_tool_call.rawOutput.metadata\n and message.acp_tool_call.rawOutput.metadata.preview\n then\n local preview = message.acp_tool_call.rawOutput.metadata.preview\n if preview then\n local content_lines = M.get_content_lines(preview, decoration, not expanded)\n vim.list_extend(lines, content_lines)\n end\n else\n if message.acp_tool_call and message.acp_tool_call.content then\n local content = message.acp_tool_call.content\n if content then\n local content_lines = M.get_content_lines(content, decoration, not expanded)\n vim.list_extend(lines, content_lines)\n end\n else\n if result and result.content then\n local result_content = result.content\n if result_content then\n local content_lines = M.get_content_lines(result_content, decoration, not expanded)\n vim.list_extend(lines, content_lines)\n end\n end\n end\n end\n if #lines <= 1 then\n if state == \"generating\" then\n table.insert(lines, Line:new({ { decoration }, { \"...\", Highlights.AVANTE_COMMENT_FG } }))\n else\n table.insert(lines, Line:new({ { decoration }, { \"completed\" } }))\n end\n end\n --- remove last empty lines\n while #lines > 0 and lines[#lines].sections[2] and lines[#lines].sections[2][1] == \"\" do\n table.remove(lines, #lines)\n end\n local last_line = lines[#lines]\n last_line.sections[1][1] = \"╰─ \"\n return lines\nend\n\n---Converts a message item into representation suitable for UI\n---@param item AvanteLLMMessageContentItem\n---@param message avante.HistoryMessage\n---@param messages avante.HistoryMessage[]\n---@param expanded boolean | nil\n---@return avante.ui.Line[]\nlocal function message_content_item_to_lines(item, message, messages, expanded)\n if type(item) == \"string\" then\n return text_to_lines(item)\n elseif type(item) == \"table\" then\n if item.type == \"thinking\" or item.type == \"redacted_thinking\" then\n return thinking_to_lines(item)\n elseif item.type == \"text\" then\n return text_to_lines(item.text)\n elseif item.type == \"image\" then\n return { Line:new({ { \"![image](\" .. item.source.media_type .. \": \" .. item.source.data .. \")\" } }) }\n elseif item.type == \"tool_use\" and item.name then\n local ok, llm_tool = pcall(require, \"avante.llm_tools.\" .. item.name)\n if ok then\n local tool_result_message = Helpers.get_tool_result_message(item.id, messages)\n ---@cast llm_tool AvanteLLMTool\n if llm_tool.on_render then\n return llm_tool.on_render(item.input, {\n logs = message.tool_use_logs,\n state = message.state,\n store = message.tool_use_store,\n result_message = tool_result_message,\n })\n end\n end\n\n local lines = tool_to_lines(item, message, messages, expanded)\n if message.tool_use_log_lines then lines = vim.list_extend(lines, message.tool_use_log_lines) end\n return lines\n end\n end\n return {}\nend\n\n---Converts a message into representation suitable for UI\n---@param message avante.HistoryMessage\n---@param messages avante.HistoryMessage[]\n---@param expanded boolean | nil\n---@return avante.ui.Line[]\nfunction M.message_to_lines(message, messages, expanded)\n if message.displayed_content then return text_to_lines(message.displayed_content) end\n local content = message.message.content\n if type(content) == \"string\" then return text_to_lines(content) end\n if islist(content) then\n local lines = {}\n for _, item in ipairs(content) do\n local item_lines = message_content_item_to_lines(item, message, messages, expanded)\n lines = vim.list_extend(lines, item_lines)\n end\n return lines\n end\n return {}\nend\n\n---Converts a message item into text representation\n---@param item AvanteLLMMessageContentItem\n---@param message avante.HistoryMessage\n---@param messages avante.HistoryMessage[]\n---@return string\nlocal function message_content_item_to_text(item, message, messages)\n local lines = message_content_item_to_lines(item, message, messages)\n return vim.iter(lines):map(function(line) return tostring(line) end):join(\"\\n\")\nend\n\n---Converts a message into text representation\n---@param message avante.HistoryMessage\n---@param messages avante.HistoryMessage[]\n---@return string\nfunction M.message_to_text(message, messages)\n local content = message.message.content\n if type(content) == \"string\" then return content end\n if islist(content) then\n return vim\n .iter(content)\n :map(function(item) return message_content_item_to_text(item, message, messages) end)\n :filter(function(text) return text ~= \"\" end)\n :join(\"\\n\")\n end\n return \"\"\nend\n\nreturn M\n" }, { "path": "lua/avante/history_selector.lua", "content": "local History = require(\"avante.history\")\nlocal Utils = require(\"avante.utils\")\nlocal Path = require(\"avante.path\")\nlocal Config = require(\"avante.config\")\nlocal Selector = require(\"avante.ui.selector\")\n\n---@class avante.HistorySelector\nlocal M = {}\n\n---@param history avante.ChatHistory\n---@return table?\nlocal function to_selector_item(history)\n local messages = History.get_history_messages(history)\n local timestamp = #messages > 0 and messages[#messages].timestamp or history.timestamp\n local name = history.title .. \" - \" .. timestamp .. \" (\" .. #messages .. \")\"\n name = name:gsub(\"\\n\", \"\\\\n\")\n return {\n name = name,\n filename = history.filename,\n }\nend\n\n---@param bufnr integer\n---@param cb fun(filename: string)\nfunction M.open(bufnr, cb)\n local selector_items = {}\n\n local histories = Path.history.list(bufnr)\n\n for _, history in ipairs(histories) do\n table.insert(selector_items, to_selector_item(history))\n end\n\n if #selector_items == 0 then\n Utils.warn(\"No history items found.\")\n return\n end\n\n local current_selector -- To be able to close it from the keymap\n\n current_selector = Selector:new({\n provider = Config.selector.provider, -- This should be 'native' for the current setup\n title = \"Avante History (Select, then choose action)\", -- Updated title\n items = vim\n .iter(selector_items)\n :map(\n function(item)\n return {\n id = item.filename,\n title = item.name,\n }\n end\n )\n :totable(),\n on_select = function(item_ids)\n if not item_ids then return end\n if #item_ids == 0 then return end\n cb(item_ids[1])\n end,\n get_preview_content = function(item_id)\n local history = Path.history.load(vim.api.nvim_get_current_buf(), item_id)\n local Sidebar = require(\"avante.sidebar\")\n local content = Sidebar.render_history_content(history)\n return content, \"markdown\"\n end,\n on_delete_item = function(item_id_to_delete)\n if not item_id_to_delete then\n Utils.warn(\"No item ID provided for deletion.\")\n return\n end\n Path.history.delete(bufnr, item_id_to_delete) -- bufnr from M.open's scope\n end,\n on_open = function() M.open(bufnr, cb) end,\n })\n current_selector:open()\nend\n\nreturn M\n" }, { "path": "lua/avante/html2md.lua", "content": "---@class AvanteHtml2Md\n---@field fetch_md fun(url: string): string\nlocal _html2md_lib = nil\n\nlocal M = {}\n\n---@return AvanteHtml2Md|nil\nfunction M._init_html2md_lib()\n if _html2md_lib ~= nil then return _html2md_lib end\n\n local ok, core = pcall(require, \"avante_html2md\")\n if not ok then return nil end\n\n _html2md_lib = core\n return _html2md_lib\nend\n\nfunction M.setup() vim.defer_fn(M._init_html2md_lib, 1000) end\n\nfunction M.fetch_md(url)\n local html2md_lib = M._init_html2md_lib()\n if not html2md_lib then return nil, \"Failed to load avante_html2md\" end\n\n local ok, res = pcall(html2md_lib.fetch_md, url)\n if not ok then return nil, res end\n return res, nil\nend\n\nreturn M\n" }, { "path": "lua/avante/init.lua", "content": "local api = vim.api\n\nlocal Utils = require(\"avante.utils\")\nlocal Sidebar = require(\"avante.sidebar\")\nlocal Selection = require(\"avante.selection\")\nlocal Suggestion = require(\"avante.suggestion\")\nlocal Config = require(\"avante.config\")\nlocal Diff = require(\"avante.diff\")\nlocal RagService = require(\"avante.rag_service\")\n\n---@class Avante\nlocal M = {\n ---@type avante.Sidebar[] we use this to track chat command across tabs\n sidebars = {},\n ---@type avante.Selection[]\n selections = {},\n ---@type avante.Suggestion[]\n suggestions = {},\n ---@type {sidebar?: avante.Sidebar, selection?: avante.Selection, suggestion?: avante.Suggestion}\n current = { sidebar = nil, selection = nil, suggestion = nil },\n ---@type table Global ACP client registry for cleanup on exit\n acp_clients = {},\n}\n\nM.did_setup = false\n\n-- ACP Client Management Functions\n---Register an ACP client for cleanup on exit\n---@param client_id string Unique identifier for the client\n---@param client any ACP client instance\nfunction M.register_acp_client(client_id, client)\n M.acp_clients[client_id] = client\n Utils.debug(\"Registered ACP client: \" .. client_id)\nend\n\n---Unregister an ACP client\n---@param client_id string Unique identifier for the client\nfunction M.unregister_acp_client(client_id)\n M.acp_clients[client_id] = nil\n Utils.debug(\"Unregistered ACP client: \" .. client_id)\nend\n\n---Cleanup all registered ACP clients\nfunction M.cleanup_all_acp_clients()\n Utils.debug(\"Cleaning up all ACP clients...\")\n for client_id, client in pairs(M.acp_clients) do\n if client and client.stop then\n Utils.debug(\"Stopping ACP client: \" .. client_id)\n pcall(function() client:stop() end)\n end\n end\n M.acp_clients = {}\n Utils.debug(\"All ACP clients cleaned up\")\nend\n\nlocal H = {}\n\nfunction H.load_path()\n local ok, LazyConfig = pcall(require, \"lazy.core.config\")\n\n if ok then\n Utils.debug(\"LazyConfig loaded\")\n local name = \"avante.nvim\"\n local function load_path() require(\"avante_lib\").load() end\n\n if LazyConfig.plugins[name] and LazyConfig.plugins[name]._.loaded then\n vim.schedule(load_path)\n else\n api.nvim_create_autocmd(\"User\", {\n pattern = \"LazyLoad\",\n callback = function(event)\n if event.data == name then\n load_path()\n return true\n end\n end,\n })\n end\n\n api.nvim_create_autocmd(\"User\", {\n pattern = \"VeryLazy\",\n callback = load_path,\n })\n else\n require(\"avante_lib\").load()\n end\nend\n\nfunction H.keymaps()\n vim.keymap.set({ \"n\", \"v\" }, \"(AvanteAsk)\", function() require(\"avante.api\").ask() end, { noremap = true })\n vim.keymap.set(\n { \"n\", \"v\" },\n \"(AvanteAskNew)\",\n function() require(\"avante.api\").ask({ new_chat = true }) end,\n { noremap = true }\n )\n vim.keymap.set(\n { \"n\", \"v\" },\n \"(AvanteChat)\",\n function() require(\"avante.api\").ask({ ask = false }) end,\n { noremap = true }\n )\n vim.keymap.set(\"v\", \"(AvanteEdit)\", function() require(\"avante.api\").edit() end, { noremap = true })\n vim.keymap.set(\"n\", \"(AvanteRefresh)\", function() require(\"avante.api\").refresh() end, { noremap = true })\n vim.keymap.set(\"n\", \"(AvanteFocus)\", function() require(\"avante.api\").focus() end, { noremap = true })\n vim.keymap.set(\"n\", \"(AvanteBuild)\", function() require(\"avante.api\").build() end, { noremap = true })\n vim.keymap.set(\"n\", \"(AvanteToggle)\", function() M.toggle() end, { noremap = true })\n vim.keymap.set(\"n\", \"(AvanteToggleDebug)\", function() M.toggle.debug() end)\n vim.keymap.set(\"n\", \"(AvanteToggleSelection)\", function() M.toggle.selection() end)\n vim.keymap.set(\"n\", \"(AvanteToggleSuggestion)\", function() M.toggle.suggestion() end)\n\n vim.keymap.set({ \"n\", \"v\" }, \"(AvanteConflictOurs)\", function() Diff.choose(\"ours\") end)\n vim.keymap.set({ \"n\", \"v\" }, \"(AvanteConflictBoth)\", function() Diff.choose(\"both\") end)\n vim.keymap.set({ \"n\", \"v\" }, \"(AvanteConflictTheirs)\", function() Diff.choose(\"theirs\") end)\n vim.keymap.set({ \"n\", \"v\" }, \"(AvanteConflictAllTheirs)\", function() Diff.choose(\"all_theirs\") end)\n vim.keymap.set({ \"n\", \"v\" }, \"(AvanteConflictCursor)\", function() Diff.choose(\"cursor\") end)\n vim.keymap.set(\"n\", \"(AvanteConflictNextConflict)\", function() Diff.find_next(\"ours\") end)\n vim.keymap.set(\"n\", \"(AvanteConflictPrevConflict)\", function() Diff.find_prev(\"ours\") end)\n vim.keymap.set(\"n\", \"(AvanteSelectModel)\", function() require(\"avante.api\").select_model() end)\n\n if Config.behaviour.auto_set_keymaps then\n Utils.safe_keymap_set(\n { \"n\", \"v\" },\n Config.mappings.ask,\n function() require(\"avante.api\").ask() end,\n { desc = \"avante: ask\" }\n )\n Utils.safe_keymap_set(\n { \"n\", \"v\" },\n Config.mappings.zen_mode,\n function() require(\"avante.api\").zen_mode() end,\n { desc = \"avante: toggle Zen Mode\" }\n )\n Utils.safe_keymap_set(\n { \"n\", \"v\" },\n Config.mappings.new_ask,\n function() require(\"avante.api\").ask({ new_chat = true }) end,\n { desc = \"avante: create new ask\" }\n )\n Utils.safe_keymap_set(\n \"v\",\n Config.mappings.edit,\n function() require(\"avante.api\").edit() end,\n { desc = \"avante: edit\" }\n )\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.stop,\n function() require(\"avante.api\").stop() end,\n { desc = \"avante: stop\" }\n )\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.refresh,\n function() require(\"avante.api\").refresh() end,\n { desc = \"avante: refresh\" }\n )\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.focus,\n function() require(\"avante.api\").focus() end,\n { desc = \"avante: focus\" }\n )\n\n Utils.safe_keymap_set(\"n\", Config.mappings.toggle.default, function() M.toggle() end, { desc = \"avante: toggle\" })\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.toggle.debug,\n function() M.toggle.debug() end,\n { desc = \"avante: toggle debug\" }\n )\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.toggle.selection,\n function() M.toggle.hint() end,\n { desc = \"avante: toggle selection\" }\n )\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.toggle.suggestion,\n function() M.toggle.suggestion() end,\n { desc = \"avante: toggle suggestion\" }\n )\n Utils.safe_keymap_set(\"n\", Config.mappings.toggle.repomap, function() require(\"avante.repo_map\").show() end, {\n desc = \"avante: display repo map\",\n noremap = true,\n silent = true,\n })\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.select_model,\n function() require(\"avante.api\").select_model() end,\n { desc = \"avante: select model\" }\n )\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.select_history,\n function() require(\"avante.api\").select_history() end,\n { desc = \"avante: select history\" }\n )\n\n Utils.safe_keymap_set(\n \"n\",\n Config.mappings.files.add_all_buffers,\n function() require(\"avante.api\").add_buffer_files() end,\n { desc = \"avante: add all open buffers\" }\n )\n end\n\n if Config.behaviour.auto_suggestions then\n Utils.safe_keymap_set(\"i\", Config.mappings.suggestion.accept, function()\n local _, _, sg = M.get()\n sg:accept()\n end, {\n desc = \"avante: accept suggestion\",\n noremap = true,\n silent = true,\n })\n\n Utils.safe_keymap_set(\"i\", Config.mappings.suggestion.dismiss, function()\n local _, _, sg = M.get()\n if sg:is_visible() then sg:dismiss() end\n end, {\n desc = \"avante: dismiss suggestion\",\n noremap = true,\n silent = true,\n })\n\n Utils.safe_keymap_set(\"i\", Config.mappings.suggestion.next, function()\n local _, _, sg = M.get()\n sg:next()\n end, {\n desc = \"avante: next suggestion\",\n noremap = true,\n silent = true,\n })\n\n Utils.safe_keymap_set(\"i\", Config.mappings.suggestion.prev, function()\n local _, _, sg = M.get()\n sg:prev()\n end, {\n desc = \"avante: previous suggestion\",\n noremap = true,\n silent = true,\n })\n end\nend\n\n---@class ApiCaller\n---@operator call(...): any\n\nfunction H.api(fun)\n return setmetatable({ api = true }, {\n __call = function(...) return fun(...) end,\n }) --[[@as ApiCaller]]\nend\n\nfunction H.signs() vim.fn.sign_define(\"AvanteInputPromptSign\", { text = Config.windows.input.prefix }) end\n\nH.augroup = api.nvim_create_augroup(\"avante_autocmds\", { clear = true })\n\nfunction H.autocmds()\n api.nvim_create_autocmd(\"TabEnter\", {\n group = H.augroup,\n pattern = \"*\",\n once = true,\n callback = function(ev)\n local tab = tonumber(ev.file)\n M._init(tab or api.nvim_get_current_tabpage())\n if Config.selection.enabled and not M.current.selection.did_setup then M.current.selection:setup_autocmds() end\n end,\n })\n\n api.nvim_create_autocmd(\"VimResized\", {\n group = H.augroup,\n callback = function()\n local sidebar = M.get()\n if not sidebar then return end\n if not sidebar:is_open() then return end\n sidebar:resize()\n end,\n })\n\n api.nvim_create_autocmd(\"QuitPre\", {\n group = H.augroup,\n callback = function()\n local current_buf = vim.api.nvim_get_current_buf()\n if Utils.is_sidebar_buffer(current_buf) then return end\n\n local non_sidebar_wins = 0\n local sidebar_wins = {}\n for _, win in ipairs(vim.api.nvim_list_wins()) do\n if vim.api.nvim_win_is_valid(win) then\n local win_buf = vim.api.nvim_win_get_buf(win)\n if Utils.is_sidebar_buffer(win_buf) then\n table.insert(sidebar_wins, win)\n else\n non_sidebar_wins = non_sidebar_wins + 1\n end\n end\n end\n\n if non_sidebar_wins <= 1 then\n for _, win in ipairs(sidebar_wins) do\n pcall(vim.api.nvim_win_close, win, false)\n end\n end\n end,\n nested = true,\n })\n\n api.nvim_create_autocmd(\"TabClosed\", {\n group = H.augroup,\n pattern = \"*\",\n callback = function(ev)\n local tab = tonumber(ev.file)\n local s = M.sidebars[tab]\n local sl = M.selections[tab]\n if s then s:reset() end\n if sl then sl:delete_autocmds() end\n if tab ~= nil then M.sidebars[tab] = nil end\n end,\n })\n\n -- Fix Issue #2749: Cleanup ACP processes on Neovim exit\n api.nvim_create_autocmd(\"VimLeavePre\", {\n group = H.augroup,\n desc = \"Cleanup all ACP processes before Neovim exits\",\n callback = function()\n Utils.debug(\"VimLeavePre: Starting ACP cleanup...\")\n -- Cancel any inflight requests first\n local ok, Llm = pcall(require, \"avante.llm\")\n if ok then pcall(function() Llm.cancel_inflight_request() end) end\n -- Cleanup all registered ACP clients\n M.cleanup_all_acp_clients()\n Utils.debug(\"VimLeavePre: ACP cleanup completed\")\n end,\n })\n\n vim.schedule(function()\n M._init(api.nvim_get_current_tabpage())\n if Config.selection.enabled then M.current.selection:setup_autocmds() end\n end)\n\n local function setup_colors()\n Utils.debug(\"Setting up avante colors\")\n require(\"avante.highlights\").setup()\n end\n\n api.nvim_create_autocmd(\"ColorSchemePre\", {\n group = H.augroup,\n callback = function()\n vim.schedule(function() setup_colors() end)\n end,\n })\n\n api.nvim_create_autocmd(\"ColorScheme\", {\n group = H.augroup,\n callback = function()\n vim.schedule(function() setup_colors() end)\n end,\n })\n\n -- automatically setup Avante filetype to markdown\n vim.treesitter.language.register(\"markdown\", \"Avante\")\n\n vim.filetype.add({\n extension = {\n [\"avanterules\"] = \"jinja\",\n },\n pattern = {\n [\"%.avanterules%.[%w_.-]+\"] = \"jinja\",\n },\n })\nend\n\n---@param current boolean? false to disable setting current, otherwise use this to track across tabs.\n---@return avante.Sidebar, avante.Selection, avante.Suggestion\nfunction M.get(current)\n local tab = api.nvim_get_current_tabpage()\n local sidebar = M.sidebars[tab]\n local selection = M.selections[tab]\n local suggestion = M.suggestions[tab]\n if current ~= false then\n M.current.sidebar = sidebar\n M.current.selection = selection\n M.current.suggestion = suggestion\n end\n return sidebar, selection, suggestion\nend\n\n---@param id integer\nfunction M._init(id)\n local sidebar = M.sidebars[id]\n local selection = M.selections[id]\n local suggestion = M.suggestions[id]\n\n if not sidebar then\n sidebar = Sidebar:new(id)\n M.sidebars[id] = sidebar\n end\n if not selection then\n selection = Selection:new(id)\n M.selections[id] = selection\n end\n if not suggestion then\n suggestion = Suggestion:new(id)\n M.suggestions[id] = suggestion\n end\n M.current = { sidebar = sidebar, selection = selection, suggestion = suggestion }\n return M\nend\n\nM.toggle = { api = true }\n\n---@param opts? AskOptions\nfunction M.toggle_sidebar(opts)\n opts = opts or {}\n if opts.ask == nil then opts.ask = true end\n\n local sidebar = M.get()\n if not sidebar then\n M._init(api.nvim_get_current_tabpage())\n ---@cast opts SidebarOpenOptions\n M.current.sidebar:open(opts)\n return true\n end\n\n return sidebar:toggle(opts)\nend\n\nfunction M.is_sidebar_open()\n local sidebar = M.get()\n if not sidebar then return false end\n return sidebar:is_open()\nend\n\n---@param opts? AskOptions\nfunction M.open_sidebar(opts)\n opts = opts or {}\n if opts.ask == nil then opts.ask = true end\n local sidebar = M.get()\n if not sidebar then M._init(api.nvim_get_current_tabpage()) end\n ---@cast opts SidebarOpenOptions\n M.current.sidebar:open(opts)\nend\n\nfunction M.close_sidebar()\n local sidebar = M.get()\n if not sidebar then return end\n sidebar:close()\nend\n\nM.toggle.debug = H.api(Utils.toggle_wrap({\n name = \"debug\",\n get = function() return Config.debug end,\n set = function(state) Config.override({ debug = state }) end,\n}))\n\nM.toggle.selection = H.api(Utils.toggle_wrap({\n name = \"selection\",\n get = function() return Config.selection.enabled end,\n set = function(state) Config.override({ selection = { enabled = state } }) end,\n}))\n\nM.toggle.suggestion = H.api(Utils.toggle_wrap({\n name = \"suggestion\",\n get = function() return Config.behaviour.auto_suggestions end,\n set = function(state)\n Config.override({ behaviour = { auto_suggestions = state } })\n local _, _, sg = M.get()\n if state ~= false then\n if sg then sg:setup_autocmds() end\n H.keymaps()\n else\n if sg then sg:delete_autocmds() end\n end\n end,\n}))\n\nsetmetatable(M.toggle, {\n __index = M.toggle,\n __call = function() M.toggle_sidebar() end,\n})\n\nM.slash_commands_id = nil\n\n---@param opts? avante.Config\nfunction M.setup(opts)\n ---PERF: we can still allow running require(\"avante\").setup() multiple times to override config if users wish to\n ---but most of the other functionality will only be called once from lazy.nvim\n Config.setup(opts)\n\n if M.did_setup then return end\n\n H.load_path()\n\n require(\"avante.html2md\").setup()\n require(\"avante.repo_map\").setup()\n require(\"avante.path\").setup()\n require(\"avante.highlights\").setup()\n require(\"avante.diff\").setup()\n require(\"avante.providers\").setup()\n require(\"avante.clipboard\").setup()\n\n -- setup helpers\n H.autocmds()\n H.keymaps()\n H.signs()\n\n M.did_setup = true\n\n local function run_rag_service()\n local started_at = os.time()\n local add_resource_with_delay\n local function add_resource()\n local is_ready = RagService.is_ready()\n if not is_ready then\n local elapsed = os.time() - started_at\n if elapsed > 1000 * 60 * 15 then\n Utils.warn(\"Rag Service is not ready, giving up\")\n return\n end\n add_resource_with_delay()\n return\n end\n vim.defer_fn(function()\n Utils.info(\"Adding project root to Rag Service ...\")\n local uri = \"file://\" .. Utils.get_project_root()\n if uri:sub(-1) ~= \"/\" then uri = uri .. \"/\" end\n RagService.add_resource(uri)\n end, 5000)\n end\n add_resource_with_delay = function()\n vim.defer_fn(function() add_resource() end, 5000)\n end\n vim.schedule(function()\n Utils.info(\"Starting Rag Service ...\")\n RagService.launch_rag_service(add_resource_with_delay)\n end)\n end\n\n if Config.rag_service.enabled then run_rag_service() end\n\n local has_cmp, cmp = pcall(require, \"cmp\")\n if has_cmp then\n M.slash_commands_id = cmp.register_source(\"avante_commands\", require(\"cmp_avante.commands\"):new())\n\n cmp.register_source(\"avante_mentions\", require(\"cmp_avante.mentions\"):new(Utils.get_chat_mentions))\n\n cmp.register_source(\"avante_prompt_mentions\", require(\"cmp_avante.mentions\"):new(Utils.get_mentions))\n\n cmp.register_source(\"avante_shortcuts\", require(\"cmp_avante.shortcuts\"):new())\n\n cmp.setup.filetype({ \"AvanteInput\" }, {\n enabled = true,\n sources = {\n { name = \"avante_commands\" },\n { name = \"avante_mentions\" },\n { name = \"avante_shortcuts\" },\n { name = \"avante_files\" },\n },\n })\n\n cmp.setup.filetype({ \"AvantePromptInput\" }, {\n enabled = true,\n sources = {\n { name = \"avante_prompt_mentions\" },\n },\n })\n end\nend\n\nreturn M\n" }, { "path": "lua/avante/libs/ReAct_parser.lua", "content": "local M = {}\n\n-- Helper function to parse a parameter tag like value\n-- Returns {name = string, value = string, next_pos = number} or nil if incomplete\nlocal function parse_parameter(text, start_pos)\n local i = start_pos\n local len = #text\n\n -- Skip whitespace\n while i <= len and string.match(string.sub(text, i, i), \"%s\") do\n i = i + 1\n end\n\n if i > len or string.sub(text, i, i) ~= \"<\" then return nil end\n\n -- Find parameter name\n local param_name_start = i + 1\n local param_name_end = string.find(text, \">\", param_name_start)\n\n if not param_name_end then\n return nil -- Incomplete parameter tag\n end\n\n local param_name = string.sub(text, param_name_start, param_name_end - 1)\n i = param_name_end + 1\n\n -- Find parameter value (everything until closing tag)\n local param_close_tag = \"\"\n local param_value_start = i\n local param_close_pos = string.find(text, param_close_tag, i, true)\n\n if not param_close_pos then\n -- Incomplete parameter value, return what we have\n local param_value = string.sub(text, param_value_start)\n return {\n name = param_name,\n value = param_value,\n next_pos = len + 1,\n }\n end\n\n local param_value = string.sub(text, param_value_start, param_close_pos - 1)\n i = param_close_pos + #param_close_tag\n\n return {\n name = param_name,\n value = param_value,\n next_pos = i,\n }\nend\n\n-- Helper function to parse tool use content starting after \n-- Returns {content = ToolUseContent, next_pos = number} or nil if incomplete\nlocal function parse_tool_use(text, start_pos)\n local i = start_pos\n local len = #text\n\n -- Skip whitespace\n while i <= len and string.match(string.sub(text, i, i), \"%s\") do\n i = i + 1\n end\n\n if i > len then\n return nil -- No content after \n end\n\n -- Check if we have opening tag for tool name\n if string.sub(text, i, i) ~= \"<\" then\n return nil -- Invalid format\n end\n\n -- Find tool name\n local tool_name_start = i + 1\n local tool_name_end = string.find(text, \">\", tool_name_start)\n\n if not tool_name_end then\n return nil -- Incomplete tool name tag\n end\n\n local tool_name = string.sub(text, tool_name_start, tool_name_end - 1)\n i = tool_name_end + 1\n\n -- Parse tool parameters\n local tool_input = {}\n local partial = false\n\n -- Look for tool closing tag or \n local tool_close_tag = \"\"\n local tool_use_close_tag = \"\"\n\n while i <= len do\n -- Skip whitespace before checking for closing tags\n while i <= len and string.match(string.sub(text, i, i), \"%s\") do\n i = i + 1\n end\n\n if i > len then\n partial = true\n break\n end\n\n -- Check for tool closing tag first\n local tool_close_pos = string.find(text, tool_close_tag, i, true)\n local tool_use_close_pos = string.find(text, tool_use_close_tag, i, true)\n\n if tool_close_pos and tool_close_pos == i then\n -- Found tool closing tag\n i = tool_close_pos + #tool_close_tag\n\n -- Skip whitespace\n while i <= len and string.match(string.sub(text, i, i), \"%s\") do\n i = i + 1\n end\n\n -- Check for \n if i <= len and string.find(text, tool_use_close_tag, i, true) == i then\n i = i + #tool_use_close_tag\n partial = false\n else\n partial = true\n end\n break\n elseif tool_use_close_pos and tool_use_close_pos == i then\n -- Found without tool closing tag (malformed, but handle it)\n i = tool_use_close_pos + #tool_use_close_tag\n partial = false\n break\n else\n -- Parse parameter tag\n local param_result = parse_parameter(text, i)\n if param_result then\n tool_input[param_result.name] = param_result.value\n i = param_result.next_pos\n else\n -- Incomplete parameter, mark as partial\n partial = true\n break\n end\n end\n end\n\n -- If we reached end of text without proper closing, it's partial\n if i > len then partial = true end\n\n return {\n content = {\n type = \"tool_use\",\n tool_name = tool_name,\n tool_input = tool_input,\n partial = partial,\n },\n next_pos = i,\n }\nend\n\n--- Parse the text into a list of TextContent and ToolUseContent\n--- The text is a string.\n--- For example:\n--- parse(\"Hello, world!\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world!\",\n--- partial = false,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txtfoo\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {\n--- path = \"path/to/file.txt\",\n--- content = \"foo\",\n--- },\n--- partial = false,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txtfooI am another tool.path/to/file.txtbarhello\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {\n--- path = \"path/to/file.txt\",\n--- content = \"foo\",\n--- },\n--- partial = false,\n--- },\n--- {\n--- type = \"text\",\n--- text = \"I am another tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {\n--- path = \"path/to/file.txt\",\n--- content = \"bar\",\n--- },\n--- partial = false,\n--- },\n--- {\n--- type = \"text\",\n--- text = \"hello\",\n--- partial = false,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {},\n--- partial = true,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txt\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {\n--- path = \"path/to/file.txt\",\n--- },\n--- partial = true,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txtfoo bar\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {\n--- path = \"path/to/file.txt\",\n--- content = \"foo bar\",\n--- },\n--- partial = true,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txtfoo bar\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.path/to/file.txtfoo bar\",\n--- partial = false,\n--- }\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txt\")\n--- returns\n--- {\n--- {\n--- type = \"text\",\n--- text = \"Hello, world! I am a tool.\",\n--- partial = false,\n--- },\n--- {\n--- type = \"tool_use\",\n--- tool_name = \"write\",\n--- tool_input = {\n--- path = \"path/to/file.txt\",\n--- content = \"\",\n--- },\n--- partial = false,\n--- },\n--- }\n---\n--- parse(\"Hello, world! I am a tool.path/to/file.txt\n \n )}\n \n=======\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
\n
{logsLoading && }
\n
{\n if (el) {\n el.scrollIntoView({ behavior: 'smooth', block: 'end' });\n }\n }} />\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n+++++++ REPLACE\n]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(diff, fixed_diff)\n end)\n\n it(\"should not break normal multiple diff\", function()\n local diff = [[------- SEARCH\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
{logsLoading && }
\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n=======\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
\n
{logsLoading && }
\n
{\n if (el) {\n el.scrollIntoView({ behavior: 'smooth', block: 'end' });\n }\n }} />\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n+++++++ REPLACE\n\n------- SEARCH\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n=======\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl aaa\">\n
\n+++++++ REPLACE\n]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(diff, fixed_diff)\n end)\n\n it(\"should fix duplicated REPLACE delimiters\", function()\n local diff = [[------- SEARCH\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
{logsLoading && }
\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n------- REPLACE\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
\n
{logsLoading && }
\n
{\n if (el) {\n el.scrollIntoView({ behavior: 'smooth', block: 'end' });\n }\n }} />\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n------- REPLACE\n]]\n\n local expected_diff = [[------- SEARCH\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
{logsLoading && }
\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n=======\n setShowLogs(false)} title=\"Project PRD Logs\" size=\"xl\">\n
\n
\n
\n {logs.split('\\n').join('\\n\\n')}\n
\n
{logsLoading && }
\n
{\n if (el) {\n el.scrollIntoView({ behavior: 'smooth', block: 'end' });\n }\n }} />\n
\n
\n {logs.length > 0 && (\n
\n setShowLogs(false)}\n className=\"bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors\"\n >\n Close\n \n
\n )}\n \n+++++++ REPLACE\n]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(expected_diff, fixed_diff)\n end)\n\n it(\"should fix the delimiter is on the same line as the content\", function()\n local diff = [[------- // Fetch initial stages when project changes\n useEffect(() => {\n if (!subscribedProject) return;\n\n const fetchStages = async () => {\n try {\n const response = await fetch(`/api/projects/${subscribedProject}/stages`);\n if (response.ok) {\n const stagesData = await response.json();\n setStages(stagesData);\n }\n } catch (error) {\n console.error('Failed to fetch stages:', error);\n }\n };\n\n fetchStages();\n }, [subscribedProject, forceUpdateCounter]);\n======= // Fetch initial stages when project changes\n useEffect(() => {\n if (!subscribedProject) return;\n\n const fetchStages = async () => {\n try {\n // Use the correct API endpoint for stages by project UUID\n const response = await fetch(`/api/stages?project_uuid=${subscribedProject}`);\n if (response.ok) {\n const stagesData = await response.json();\n setStages(stagesData);\n }\n } catch (error) {\n console.error('Failed to fetch stages:', error);\n }\n };\n\n fetchStages();\n }, [subscribedProject, forceUpdateCounter]);\n+++++++ REPLACE\n]]\n\n local expected_diff = [[------- SEARCH\n // Fetch initial stages when project changes\n useEffect(() => {\n if (!subscribedProject) return;\n\n const fetchStages = async () => {\n try {\n const response = await fetch(`/api/projects/${subscribedProject}/stages`);\n if (response.ok) {\n const stagesData = await response.json();\n setStages(stagesData);\n }\n } catch (error) {\n console.error('Failed to fetch stages:', error);\n }\n };\n\n fetchStages();\n }, [subscribedProject, forceUpdateCounter]);\n=======\n // Fetch initial stages when project changes\n useEffect(() => {\n if (!subscribedProject) return;\n\n const fetchStages = async () => {\n try {\n // Use the correct API endpoint for stages by project UUID\n const response = await fetch(`/api/stages?project_uuid=${subscribedProject}`);\n if (response.ok) {\n const stagesData = await response.json();\n setStages(stagesData);\n }\n } catch (error) {\n console.error('Failed to fetch stages:', error);\n }\n };\n\n fetchStages();\n }, [subscribedProject, forceUpdateCounter]);\n+++++++ REPLACE\n]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(expected_diff, fixed_diff)\n end)\n\n it(\"should fix unified diff\", function()\n local diff = [[--- lua/avante/sidebar.lua\n+++ lua/avante/sidebar.lua\n@@ -3099,7 +3099,7 @@\n function Sidebar:create_todos_container()\n local history = Path.history.load(self.code.bufnr)\n if not history or not history.todos or #history.todos == 0 then\n- if self.containers.todos then self.containers.todos:unmount() end\n+ if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end\n self.containers.todos = nil\n self:adjust_layout()\n return\n@@ -3121,7 +3121,7 @@\n }),\n position = \"bottom\",\n size = {\n- height = 3,\n+ height = math.min(3, math.max(1, vim.o.lines - 5)),\n },\n })\n self.containers.todos:mount()\n@@ -3151,11 +3151,15 @@\n self:render_header(\n self.containers.todos.winid,\n todos_buf,\n- Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n+ Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n Highlights.SUBTITLE,\n Highlights.REVERSED_SUBTITLE\n )\n- self:adjust_layout()\n+\n+ local ok, err = pcall(function()\n+ self:adjust_layout()\n+ end)\n+ if not ok then Utils.debug(\"Failed to adjust layout after todos creation:\", err) end\n end\n\n function Sidebar:adjust_layout()\n]]\n\n local expected_diff = [[------- SEARCH\nfunction Sidebar:create_todos_container()\n local history = Path.history.load(self.code.bufnr)\n if not history or not history.todos or #history.todos == 0 then\n if self.containers.todos then self.containers.todos:unmount() end\n self.containers.todos = nil\n self:adjust_layout()\n return\n=======\nfunction Sidebar:create_todos_container()\n local history = Path.history.load(self.code.bufnr)\n if not history or not history.todos or #history.todos == 0 then\n if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end\n self.containers.todos = nil\n self:adjust_layout()\n return\n+++++++ REPLACE\n\n------- SEARCH\n}),\n position = \"bottom\",\n size = {\n height = 3,\n },\n })\n self.containers.todos:mount()\n=======\n}),\n position = \"bottom\",\n size = {\n height = math.min(3, math.max(1, vim.o.lines - 5)),\n },\n })\n self.containers.todos:mount()\n+++++++ REPLACE\n\n------- SEARCH\nself:render_header(\n self.containers.todos.winid,\n todos_buf,\n Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n Highlights.SUBTITLE,\n Highlights.REVERSED_SUBTITLE\n )\n self:adjust_layout()\nend\nfunction Sidebar:adjust_layout()\n=======\nself:render_header(\n self.containers.todos.winid,\n todos_buf,\n Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n Highlights.SUBTITLE,\n Highlights.REVERSED_SUBTITLE\n )\n\n local ok, err = pcall(function()\n self:adjust_layout()\n end)\n if not ok then Utils.debug(\"Failed to adjust layout after todos creation:\", err) end\nend\nfunction Sidebar:adjust_layout()\n+++++++ REPLACE]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(expected_diff, fixed_diff)\n end)\n\n it(\"should fix unified diff 2\", function()\n local diff = [[\n@@ -3099,7 +3099,7 @@\n function Sidebar:create_todos_container()\n local history = Path.history.load(self.code.bufnr)\n if not history or not history.todos or #history.todos == 0 then\n- if self.containers.todos then self.containers.todos:unmount() end\n+ if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end\n self.containers.todos = nil\n self:adjust_layout()\n return\n@@ -3121,7 +3121,7 @@\n }),\n position = \"bottom\",\n size = {\n- height = 3,\n+ height = math.min(3, math.max(1, vim.o.lines - 5)),\n },\n })\n self.containers.todos:mount()\n@@ -3151,11 +3151,15 @@\n self:render_header(\n self.containers.todos.winid,\n todos_buf,\n- Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n+ Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n Highlights.SUBTITLE,\n Highlights.REVERSED_SUBTITLE\n )\n- self:adjust_layout()\n+\n+ local ok, err = pcall(function()\n+ self:adjust_layout()\n+ end)\n+ if not ok then Utils.debug(\"Failed to adjust layout after todos creation:\", err) end\n end\n\n function Sidebar:adjust_layout()\n]]\n local expected_diff = [[------- SEARCH\nfunction Sidebar:create_todos_container()\n local history = Path.history.load(self.code.bufnr)\n if not history or not history.todos or #history.todos == 0 then\n if self.containers.todos then self.containers.todos:unmount() end\n self.containers.todos = nil\n self:adjust_layout()\n return\n=======\nfunction Sidebar:create_todos_container()\n local history = Path.history.load(self.code.bufnr)\n if not history or not history.todos or #history.todos == 0 then\n if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end\n self.containers.todos = nil\n self:adjust_layout()\n return\n+++++++ REPLACE\n\n------- SEARCH\n}),\n position = \"bottom\",\n size = {\n height = 3,\n },\n })\n self.containers.todos:mount()\n=======\n}),\n position = \"bottom\",\n size = {\n height = math.min(3, math.max(1, vim.o.lines - 5)),\n },\n })\n self.containers.todos:mount()\n+++++++ REPLACE\n\n------- SEARCH\nself:render_header(\n self.containers.todos.winid,\n todos_buf,\n Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n Highlights.SUBTITLE,\n Highlights.REVERSED_SUBTITLE\n )\n self:adjust_layout()\nend\nfunction Sidebar:adjust_layout()\n=======\nself:render_header(\n self.containers.todos.winid,\n todos_buf,\n Utils.icon(\" \") .. \"Todos\" .. \" (\" .. done_count .. \"/\" .. total_count .. \")\",\n Highlights.SUBTITLE,\n Highlights.REVERSED_SUBTITLE\n )\n\n local ok, err = pcall(function()\n self:adjust_layout()\n end)\n if not ok then Utils.debug(\"Failed to adjust layout after todos creation:\", err) end\nend\nfunction Sidebar:adjust_layout()\n+++++++ REPLACE]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(expected_diff, fixed_diff)\n end)\n\n it(\"should fix duplicated replace blocks\", function()\n local diff = [[------- SEARCH\n useEffect(() => {\n if (!isExpanded || !textContentRef.current) {\n setShowFixedCollapseButton(false);\n return;\n }\n\n const observer = new IntersectionObserver(\n ([entry]) => {\n setShowFixedCollapseButton(!entry.isIntersecting);\n },\n {\n root: null,\n rootMargin: '0px',\n threshold: 1.0,\n }\n );\n\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n observer.observe(collapseButton);\n }\n\n return () => {\n if (collapseButton) {\n observer.unobserve(collapseButton);\n }\n };\n }, [isExpanded, textContentRef.current]);\n=======\n useEffect(() => {\n if (!isExpanded || !textContentRef.current) {\n setShowFixedCollapseButton(false);\n return;\n }\n\n // Check initial visibility of the collapse button\n const checkInitialVisibility = () => {\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n const rect = collapseButton.getBoundingClientRect();\n const isVisible = rect.top >= 0 && rect.bottom <= window.innerHeight;\n setShowFixedCollapseButton(!isVisible);\n }\n };\n\n // Small delay to ensure DOM is updated after expansion\n const timeoutId = setTimeout(checkInitialVisibility, 100);\n\n const observer = new IntersectionObserver(\n ([entry]) => {\n setShowFixedCollapseButton(!entry.isIntersecting);\n },\n {\n root: null,\n rootMargin: '0px',\n threshold: [0, 1.0], // Check both when it starts to leave and when fully visible\n }\n );\n\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n observer.observe(collapseButton);\n }\n\n return () => {\n clearTimeout(timeoutId);\n if (collapseButton) {\n observer.unobserve(collapseButton);\n }\n };\n }, [isExpanded, textContentRef.current]);\n=======\n useEffect(() => {\n if (!isExpanded || !textContentRef.current) {\n setShowFixedCollapseButton(false);\n return;\n }\n\n // Check initial visibility of the collapse button\n const checkInitialVisibility = () => {\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n const rect = collapseButton.getBoundingClientRect();\n const isVisible = rect.top >= 0 && rect.bottom <= window.innerHeight;\n setShowFixedCollapseButton(!isVisible);\n }\n };\n\n // Small delay to ensure DOM is updated after expansion\n const timeoutId = setTimeout(checkInitialVisibility, 100);\n\n const observer = new IntersectionObserver(\n ([entry]) => {\n setShowFixedCollapseButton(!entry.isIntersecting);\n },\n {\n root: null,\n rootMargin: '0px',\n threshold: [0, 1.0], // Check both when it starts to leave and when fully visible\n }\n );\n\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n observer.observe(collapseButton);\n }\n\n return () => {\n clearTimeout(timeoutId);\n if (collapseButton) {\n observer.unobserve(collapseButton);\n }\n };\n }, [isExpanded, textContentRef.current]);\n+++++++ REPLACE\n]]\n\n local expected_diff = [[------- SEARCH\n useEffect(() => {\n if (!isExpanded || !textContentRef.current) {\n setShowFixedCollapseButton(false);\n return;\n }\n\n const observer = new IntersectionObserver(\n ([entry]) => {\n setShowFixedCollapseButton(!entry.isIntersecting);\n },\n {\n root: null,\n rootMargin: '0px',\n threshold: 1.0,\n }\n );\n\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n observer.observe(collapseButton);\n }\n\n return () => {\n if (collapseButton) {\n observer.unobserve(collapseButton);\n }\n };\n }, [isExpanded, textContentRef.current]);\n=======\n useEffect(() => {\n if (!isExpanded || !textContentRef.current) {\n setShowFixedCollapseButton(false);\n return;\n }\n\n // Check initial visibility of the collapse button\n const checkInitialVisibility = () => {\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n const rect = collapseButton.getBoundingClientRect();\n const isVisible = rect.top >= 0 && rect.bottom <= window.innerHeight;\n setShowFixedCollapseButton(!isVisible);\n }\n };\n\n // Small delay to ensure DOM is updated after expansion\n const timeoutId = setTimeout(checkInitialVisibility, 100);\n\n const observer = new IntersectionObserver(\n ([entry]) => {\n setShowFixedCollapseButton(!entry.isIntersecting);\n },\n {\n root: null,\n rootMargin: '0px',\n threshold: [0, 1.0], // Check both when it starts to leave and when fully visible\n }\n );\n\n const collapseButton = collapseButtonRef.current;\n if (collapseButton) {\n observer.observe(collapseButton);\n }\n\n return () => {\n clearTimeout(timeoutId);\n if (collapseButton) {\n observer.unobserve(collapseButton);\n }\n };\n }, [isExpanded, textContentRef.current]);\n+++++++ REPLACE]]\n\n local fixed_diff = Utils.fix_diff(diff)\n assert.equals(expected_diff, fixed_diff)\n end)\nend)\n" }, { "path": "tests/utils/get_parent_path_spec.lua", "content": "local utils = require(\"avante.utils\")\n\ndescribe(\"get_parent_path\", function()\n -- Define path separator for our tests, using the same logic as in the utils module\n local path_sep = jit.os:find(\"Windows\") ~= nil and \"\\\\\" or \"/\"\n\n it(\"should return the parent directory of a file path\", function()\n local filepath = \"foo\" .. path_sep .. \"bar\" .. path_sep .. \"baz.txt\"\n local expected = \"foo\" .. path_sep .. \"bar\"\n assert.are.equal(expected, utils.get_parent_path(filepath))\n end)\n\n it(\"should return the parent directory of a directory path\", function()\n local dirpath = \"foo\" .. path_sep .. \"bar\" .. path_sep .. \"baz\"\n local expected = \"foo\" .. path_sep .. \"bar\"\n assert.are.equal(expected, utils.get_parent_path(dirpath))\n end)\n\n it(\"should handle trailing separators\", function()\n local dirpath = \"foo\" .. path_sep .. \"bar\" .. path_sep .. \"baz\" .. path_sep\n local expected = \"foo\" .. path_sep .. \"bar\"\n assert.are.equal(expected, utils.get_parent_path(dirpath))\n end)\n\n it(\"should return '.' for a single file or directory\", function()\n assert.are.equal(\".\", utils.get_parent_path(\"foo.txt\"))\n assert.are.equal(\".\", utils.get_parent_path(\"dir\"))\n end)\n\n it(\"should handle paths with multiple levels\", function()\n local filepath = \"a\" .. path_sep .. \"b\" .. path_sep .. \"c\" .. path_sep .. \"d\" .. path_sep .. \"file.txt\"\n local expected = \"a\" .. path_sep .. \"b\" .. path_sep .. \"c\" .. path_sep .. \"d\"\n assert.are.equal(expected, utils.get_parent_path(filepath))\n end)\n\n it(\"should return empty string for root directory\", function()\n -- Root directory on Unix-like systems\n if path_sep == \"/\" then\n assert.are.equal(\"/\", utils.get_parent_path(\"/foo\"))\n else\n -- Windows uses drive letters, so parent of \"C:\\foo\" is \"C:\"\n local winpath = \"C:\" .. path_sep .. \"foo\"\n assert.are.equal(\"C:\", utils.get_parent_path(winpath))\n end\n end)\n\n it(\"should return empty string for an empty string\", function() assert.are.equal(\"\", utils.get_parent_path(\"\")) end)\n\n it(\"should throw an error for nil input\", function()\n assert.has_error(function() utils.get_parent_path(nil) end, \"filepath cannot be nil\")\n end)\n\n it(\"should handle paths with spaces\", function()\n local filepath = \"path with spaces\" .. path_sep .. \"file name.txt\"\n local expected = \"path with spaces\"\n assert.are.equal(expected, utils.get_parent_path(filepath))\n end)\n\n it(\"should handle special characters in paths\", function()\n local filepath = \"folder-name!\" .. path_sep .. \"file_#$%&.txt\"\n local expected = \"folder-name!\"\n assert.are.equal(expected, utils.get_parent_path(filepath))\n end)\n\n it(\"should handle absolute paths\", function()\n if path_sep == \"/\" then\n -- Unix-like paths\n local filepath = path_sep .. \"home\" .. path_sep .. \"user\" .. path_sep .. \"file.txt\"\n local expected = path_sep .. \"home\" .. path_sep .. \"user\"\n assert.are.equal(expected, utils.get_parent_path(filepath))\n\n -- Root directory edge case\n assert.are.equal(\"\", utils.get_parent_path(path_sep))\n else\n -- Windows paths\n local filepath = \"C:\" .. path_sep .. \"Users\" .. path_sep .. \"user\" .. path_sep .. \"file.txt\"\n local expected = \"C:\" .. path_sep .. \"Users\" .. path_sep .. \"user\"\n assert.are.equal(expected, utils.get_parent_path(filepath))\n end\n end)\nend)\n" }, { "path": "tests/utils/init_spec.lua", "content": "local Utils = require(\"avante.utils\")\n\ndescribe(\"Utils\", function()\n describe(\"trim\", function()\n it(\"should trim prefix\", function() assert.equals(\"test\", Utils.trim(\"prefix_test\", { prefix = \"prefix_\" })) end)\n\n it(\"should trim suffix\", function() assert.equals(\"test\", Utils.trim(\"test_suffix\", { suffix = \"_suffix\" })) end)\n\n it(\n \"should trim both prefix and suffix\",\n function() assert.equals(\"test\", Utils.trim(\"prefix_test_suffix\", { prefix = \"prefix_\", suffix = \"_suffix\" })) end\n )\n\n it(\n \"should return original string if no match\",\n function() assert.equals(\"test\", Utils.trim(\"test\", { prefix = \"xxx\", suffix = \"yyy\" })) end\n )\n end)\n\n describe(\"url_join\", function()\n it(\"should join url parts correctly\", function()\n assert.equals(\"http://example.com/path\", Utils.url_join(\"http://example.com\", \"path\"))\n assert.equals(\"http://example.com/path\", Utils.url_join(\"http://example.com/\", \"/path\"))\n assert.equals(\"http://example.com/path/to\", Utils.url_join(\"http://example.com\", \"path\", \"to\"))\n assert.equals(\"http://example.com/path\", Utils.url_join(\"http://example.com/\", \"/path/\"))\n end)\n\n it(\"should handle empty parts\", function()\n assert.equals(\"http://example.com\", Utils.url_join(\"http://example.com\", \"\"))\n assert.equals(\"http://example.com\", Utils.url_join(\"http://example.com\", nil))\n end)\n end)\n\n describe(\"is_type\", function()\n it(\"should check basic types correctly\", function()\n assert.is_true(Utils.is_type(\"string\", \"test\"))\n assert.is_true(Utils.is_type(\"number\", 123))\n assert.is_true(Utils.is_type(\"boolean\", true))\n assert.is_true(Utils.is_type(\"table\", {}))\n assert.is_true(Utils.is_type(\"function\", function() end))\n assert.is_true(Utils.is_type(\"nil\", nil))\n end)\n\n it(\"should check list type correctly\", function()\n assert.is_true(Utils.is_type(\"list\", { 1, 2, 3 }))\n assert.is_false(Utils.is_type(\"list\", { a = 1, b = 2 }))\n end)\n\n it(\"should check map type correctly\", function()\n assert.is_true(Utils.is_type(\"map\", { a = 1, b = 2 }))\n assert.is_false(Utils.is_type(\"map\", { 1, 2, 3 }))\n end)\n end)\n\n describe(\"get_indentation\", function()\n it(\"should get correct indentation\", function()\n assert.equals(\" \", Utils.get_indentation(\" test\"))\n assert.equals(\"\\t\", Utils.get_indentation(\"\\ttest\"))\n assert.equals(\"\", Utils.get_indentation(\"test\"))\n end)\n\n it(\"should handle empty or nil input\", function()\n assert.equals(\"\", Utils.get_indentation(\"\"))\n assert.equals(\"\", Utils.get_indentation(nil))\n end)\n end)\n\n describe(\"trime_space\", function()\n it(\"should remove indentation correctly\", function()\n assert.equals(\"test\", Utils.trim_space(\" test\"))\n assert.equals(\"test\", Utils.trim_space(\"\\ttest\"))\n assert.equals(\"test\", Utils.trim_space(\"test\"))\n end)\n\n it(\"should handle empty or nil input\", function()\n assert.equals(\"\", Utils.trim_space(\"\"))\n assert.equals(nil, Utils.trim_space(nil))\n end)\n end)\n\n describe(\"is_first_letter_uppercase\", function()\n it(\"should detect uppercase first letter\", function()\n assert.is_true(Utils.is_first_letter_uppercase(\"Test\"))\n assert.is_true(Utils.is_first_letter_uppercase(\"ABC\"))\n end)\n\n it(\"should detect lowercase first letter\", function()\n assert.is_false(Utils.is_first_letter_uppercase(\"test\"))\n assert.is_false(Utils.is_first_letter_uppercase(\"abc\"))\n end)\n end)\n\n describe(\"extract_mentions\", function()\n it(\"should extract @codebase mention\", function()\n local result = Utils.extract_mentions(\"test @codebase\")\n assert.equals(\"test \", result.new_content)\n assert.is_true(result.enable_project_context)\n assert.is_false(result.enable_diagnostics)\n end)\n\n it(\"should extract @diagnostics mention\", function()\n local result = Utils.extract_mentions(\"test @diagnostics\")\n assert.equals(\"test @diagnostics\", result.new_content)\n assert.is_false(result.enable_project_context)\n assert.is_true(result.enable_diagnostics)\n end)\n\n it(\"should handle multiple mentions\", function()\n local result = Utils.extract_mentions(\"test @codebase @diagnostics\")\n assert.equals(\"test @diagnostics\", result.new_content)\n assert.is_true(result.enable_project_context)\n assert.is_true(result.enable_diagnostics)\n end)\n end)\n\n describe(\"get_mentions\", function()\n it(\"should return valid mentions\", function()\n local mentions = Utils.get_mentions()\n assert.equals(\"codebase\", mentions[1].command)\n assert.equals(\"diagnostics\", mentions[2].command)\n end)\n end)\n\n describe(\"trim_think_content\", function()\n it(\"should remove think content\", function()\n local input = \"this should be removed Hello World\"\n assert.equals(\" Hello World\", Utils.trim_think_content(input))\n end)\n\n it(\"The think tag that is not in the prefix should not be deleted.\", function()\n local input = \"Hello this should not be removed World\"\n assert.equals(\"Hello this should not be removed World\", Utils.trim_think_content(input))\n end)\n\n it(\"should handle multiple think blocks\", function()\n local input = \"firstmiddlesecond\"\n assert.equals(\"middlesecond\", Utils.trim_think_content(input))\n end)\n\n it(\"should handle empty think blocks\", function()\n local input = \"testtest\"\n assert.equals(\"testtest\", Utils.trim_think_content(input))\n end)\n\n it(\"should handle empty think blocks\", function()\n local input = \"testtest\"\n assert.equals(\"testtest\", Utils.trim_think_content(input))\n end)\n\n it(\"should handle input without think blocks\", function()\n local input = \"just normal text\"\n assert.equals(\"just normal text\", Utils.trim_think_content(input))\n end)\n end)\n\n describe(\"debounce\", function()\n it(\"should debounce function calls\", function()\n local count = 0\n local debounced = Utils.debounce(function() count = count + 1 end, 100)\n\n -- Call multiple times in quick succession\n debounced()\n debounced()\n debounced()\n\n -- Should not have executed yet\n assert.equals(0, count)\n\n -- Wait for debounce timeout\n vim.wait(200, function() return false end)\n\n -- Should have executed once\n assert.equals(1, count)\n end)\n\n it(\"should cancel previous timer on new calls\", function()\n local count = 0\n local debounced = Utils.debounce(function(c) count = c end, 100)\n\n -- First call\n debounced(1)\n\n -- Wait partial time\n vim.wait(50, function() return false end)\n\n -- Second call should cancel first\n debounced(233)\n\n -- Count should still be 0\n assert.equals(0, count)\n\n -- Wait for timeout\n vim.wait(200, function() return false end)\n\n -- Should only execute the latest once\n assert.equals(233, count)\n end)\n\n it(\"should pass arguments correctly\", function()\n local result\n local debounced = Utils.debounce(function(x, y) result = x + y end, 100)\n\n debounced(2, 3)\n\n -- Wait for timeout\n vim.wait(200, function() return false end)\n\n assert.equals(5, result)\n end)\n end)\n\n describe(\"fuzzy_match\", function()\n it(\"should match exact lines\", function()\n local lines = { \"test\", \"test2\", \"test3\", \"test4\" }\n local start_line, end_line = Utils.fuzzy_match(lines, { \"test2\", \"test3\" })\n assert.equals(2, start_line)\n assert.equals(3, end_line)\n end)\n\n it(\"should match lines with suffix\", function()\n local lines = { \"test\", \"test2\", \"test3\", \"test4\" }\n local start_line, end_line = Utils.fuzzy_match(lines, { \"test2 \\t\", \"test3\" })\n assert.equals(2, start_line)\n assert.equals(3, end_line)\n end)\n\n it(\"should match lines with space\", function()\n local lines = { \"test\", \"test2\", \"test3\", \"test4\" }\n local start_line, end_line = Utils.fuzzy_match(lines, { \"test2 \", \" test3\" })\n assert.equals(2, start_line)\n assert.equals(3, end_line)\n end)\n end)\nend)\n" }, { "path": "tests/utils/join_paths_spec.lua", "content": "local assert = require(\"luassert\")\nlocal utils = require(\"avante.utils\")\n\ndescribe(\"join_paths\", function()\n it(\"should join multiple path segments with proper separator\", function()\n local result = utils.join_paths(\"path\", \"to\", \"file.lua\")\n assert.equals(\"path\" .. utils.path_sep .. \"to\" .. utils.path_sep .. \"file.lua\", result)\n end)\n\n it(\"should handle empty path segments\", function()\n local result = utils.join_paths(\"\", \"to\", \"file.lua\")\n assert.equals(\"to\" .. utils.path_sep .. \"file.lua\", result)\n end)\n\n it(\"should handle nil path segments\", function()\n local result = utils.join_paths(nil, \"to\", \"file.lua\")\n assert.equals(\"to\" .. utils.path_sep .. \"file.lua\", result)\n end)\n\n it(\"should handle empty path segments\", function()\n local result = utils.join_paths(\"path\", \"\", \"file.lua\")\n assert.equals(\"path\" .. utils.path_sep .. \"file.lua\", result)\n end)\n\n it(\"should use absolute path when encountered\", function()\n local absolute_path = utils.is_win() and \"C:\\\\absolute\\\\path\" or \"/absolute/path\"\n local result = utils.join_paths(\"relative\", \"path\", absolute_path)\n assert.equals(absolute_path, result)\n end)\n\n it(\"should handle paths with trailing separators\", function()\n local path_with_sep = \"path\" .. utils.path_sep\n local result = utils.join_paths(path_with_sep, \"file.lua\")\n assert.equals(\"path\" .. utils.path_sep .. \"file.lua\", result)\n end)\n\n it(\"should handle no paths provided\", function()\n local result = utils.join_paths()\n assert.equals(\".\", result)\n end)\n\n it(\"should return first path when only one path provided\", function()\n local result = utils.join_paths(\"path\")\n assert.equals(\"path\", result)\n end)\n\n it(\"should handle path with mixed separators\", function()\n -- This test is more relevant on Windows where both / and \\ are valid separators\n local mixed_path = utils.is_win() and \"path\\\\to/file\" or \"path/to/file\"\n local result = utils.join_paths(\"base\", mixed_path)\n -- The function should use utils.path_sep for joining\n assert.equals(\"base\" .. utils.path_sep .. mixed_path, result)\n end)\nend)\n" }, { "path": "tests/utils/make_relative_path_spec.lua", "content": "local assert = require(\"luassert\")\nlocal utils = require(\"avante.utils\")\n\ndescribe(\"make_relative_path\", function()\n it(\"should remove base directory from filepath\", function()\n local test_filepath = \"/path/to/project/src/file.lua\"\n local test_base_dir = \"/path/to/project\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\"src/file.lua\", result)\n end)\n\n it(\"should handle trailing dot-slash in base_dir\", function()\n local test_filepath = \"/path/to/project/src/file.lua\"\n local test_base_dir = \"/path/to/project/.\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\"src/file.lua\", result)\n end)\n\n it(\"should handle trailing dot-slash in filepath\", function()\n local test_filepath = \"/path/to/project/src/.\"\n local test_base_dir = \"/path/to/project\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\"src\", result)\n end)\n\n it(\"should handle both having trailing dot-slash\", function()\n local test_filepath = \"/path/to/project/src/.\"\n local test_base_dir = \"/path/to/project/.\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\"src\", result)\n end)\n\n it(\"should return the filepath when base_dir is not a prefix\", function()\n local test_filepath = \"/path/to/project/src/file.lua\"\n local test_base_dir = \"/different/path\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\"/path/to/project/src/file.lua\", result)\n end)\n\n it(\"should handle identical paths\", function()\n local test_filepath = \"/path/to/project\"\n local test_base_dir = \"/path/to/project\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\".\", result)\n end)\n\n it(\"should handle empty strings\", function()\n local result = utils.make_relative_path(\"\", \"\")\n assert.equals(\".\", result)\n end)\n\n it(\"should preserve trailing slash in filepath\", function()\n local test_filepath = \"/path/to/project/src/\"\n local test_base_dir = \"/path/to/project\"\n local result = utils.make_relative_path(test_filepath, test_base_dir)\n assert.equals(\"src/\", result)\n end)\nend)\n" }, { "path": "tests/utils/streaming_json_parser_spec.lua", "content": "local StreamingJSONParser = require(\"avante.utils.streaming_json_parser\")\n\ndescribe(\"StreamingJSONParser\", function()\n local parser\n\n before_each(function() parser = StreamingJSONParser:new() end)\n\n describe(\"initialization\", function()\n it(\"should create a new parser with empty state\", function()\n assert.is_not_nil(parser)\n assert.equals(\"\", parser.buffer)\n assert.is_not_nil(parser.state)\n assert.is_false(parser.state.inString)\n assert.is_false(parser.state.escaping)\n assert.is_table(parser.state.stack)\n assert.equals(0, #parser.state.stack)\n assert.is_nil(parser.state.result)\n assert.is_nil(parser.state.currentKey)\n assert.is_nil(parser.state.current)\n assert.is_table(parser.state.parentKeys)\n end)\n end)\n\n describe(\"parse\", function()\n it(\"should parse a complete simple JSON object\", function()\n local result, complete = parser:parse('{\"key\": \"value\"}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals(\"value\", result.key)\n end)\n\n it(\"should parse breaklines\", function()\n local result, complete = parser:parse('{\"key\": \"value\\nv\"}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals(\"value\\nv\", result.key)\n end)\n\n it(\"should parse a complete simple JSON array\", function()\n local result, complete = parser:parse(\"[1, 2, 3]\")\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals(1, result[1])\n assert.equals(2, result[2])\n assert.equals(3, result[3])\n end)\n\n it(\"should handle streaming JSON in multiple chunks\", function()\n local result1, complete1 = parser:parse('{\"name\": \"John')\n assert.is_false(complete1)\n assert.is_table(result1)\n assert.equals(\"John\", result1.name)\n\n local result2, complete2 = parser:parse('\", \"age\": 30}')\n assert.is_true(complete2)\n assert.is_table(result2)\n assert.equals(\"John\", result2.name)\n assert.equals(30, result2.age)\n end)\n\n it(\"should handle streaming string field\", function()\n local result1, complete1 = parser:parse('{\"name\": {\"first\": \"John')\n assert.is_false(complete1)\n assert.is_table(result1)\n assert.equals(\"John\", result1.name.first)\n end)\n\n it(\"should parse nested objects\", function()\n local json = [[{\n \"person\": {\n \"name\": \"John\",\n \"age\": 30,\n \"address\": {\n \"city\": \"New York\",\n \"zip\": \"10001\"\n }\n }\n }]]\n\n local result, complete = parser:parse(json)\n assert.is_true(complete)\n assert.is_table(result)\n assert.is_table(result.person)\n assert.equals(\"John\", result.person.name)\n assert.equals(30, result.person.age)\n assert.is_table(result.person.address)\n assert.equals(\"New York\", result.person.address.city)\n assert.equals(\"10001\", result.person.address.zip)\n end)\n\n it(\"should parse nested arrays\", function()\n local json = [[{\n \"matrix\": [\n [1, 2, 3],\n [4, 5, 6],\n [7, 8, 9]\n ]\n }]]\n\n local result, complete = parser:parse(json)\n assert.is_true(complete)\n assert.is_table(result)\n assert.is_table(result.matrix)\n assert.equals(3, #result.matrix)\n assert.equals(1, result.matrix[1][1])\n assert.equals(5, result.matrix[2][2])\n assert.equals(9, result.matrix[3][3])\n end)\n\n it(\"should handle boolean values\", function()\n local result, complete = parser:parse('{\"success\": true, \"failed\": false}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.is_true(result.success)\n assert.is_false(result.failed)\n end)\n\n it(\"should handle null values\", function()\n local result, complete = parser:parse('{\"value\": null}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.is_nil(result.value)\n end)\n\n it(\"should handle escaped characters in strings\", function()\n local result, complete = parser:parse('{\"text\": \"line1\\\\nline2\\\\t\\\\\"quoted\\\\\"\"}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals('line1\\nline2\\t\"quoted\"', result.text)\n end)\n\n it(\"should handle numbers correctly\", function()\n local result, complete = parser:parse('{\"integer\": 42, \"float\": 3.14, \"negative\": -10, \"exponent\": 1.2e3}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals(42, result.integer)\n assert.equals(3.14, result.float)\n assert.equals(-10, result.negative)\n assert.equals(1200, result.exponent)\n end)\n\n it(\"should handle streaming complex JSON\", function()\n local chunks = {\n '{\"data\": [{\"id\": 1, \"info\": {\"name\":',\n ' \"Product A\", \"active\": true}}, {\"id\": 2, ',\n '\"info\": {\"name\": \"Product B\", \"active\": false',\n '}}], \"total\": 2}',\n }\n\n local complete = false\n local result\n\n for _, chunk in ipairs(chunks) do\n result, complete = parser:parse(chunk)\n end\n\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals(2, #result.data)\n assert.equals(1, result.data[1].id)\n assert.equals(\"Product A\", result.data[1].info.name)\n assert.is_true(result.data[1].info.active)\n assert.equals(2, result.data[2].id)\n assert.equals(\"Product B\", result.data[2].info.name)\n assert.is_false(result.data[2].info.active)\n assert.equals(2, result.total)\n end)\n\n it(\"should reset the parser state correctly\", function()\n parser:parse('{\"key\": \"value\"}')\n parser:reset()\n\n assert.equals(\"\", parser.buffer)\n assert.is_false(parser.state.inString)\n assert.is_false(parser.state.escaping)\n assert.is_table(parser.state.stack)\n assert.equals(0, #parser.state.stack)\n assert.is_nil(parser.state.result)\n assert.is_nil(parser.state.currentKey)\n assert.is_nil(parser.state.current)\n assert.is_table(parser.state.parentKeys)\n end)\n\n it(\"should return partial results for incomplete JSON\", function()\n parser:reset()\n local result, complete = parser:parse('{\"stream\": [1, 2,')\n assert.is_false(complete)\n assert.is_table(result)\n assert.is_table(result.stream)\n assert.equals(1, result.stream[1])\n assert.equals(2, result.stream[2])\n\n -- We need exactly one item in the stack (the array)\n assert.equals(2, #parser.state.stack)\n end)\n\n it(\"should handle whitespace correctly\", function()\n parser:reset()\n local result, complete = parser:parse('{\"key1\": \"value1\", \"key2\": 42}')\n assert.is_true(complete)\n assert.is_table(result)\n assert.equals(\"value1\", result.key1)\n assert.equals(42, result.key2)\n end)\n\n it(\"should provide access to partial results during streaming\", function()\n parser:parse('{\"name\": \"John\", \"items\": [')\n\n local partial = parser:getCurrentPartial()\n assert.is_table(partial)\n assert.equals(\"John\", partial.name)\n assert.is_table(partial.items)\n\n parser:parse(\"1, 2]\")\n local result, complete = parser:parse(\"}\")\n\n assert.is_true(complete)\n assert.equals(\"John\", result.name)\n assert.equals(1, result.items[1])\n assert.equals(2, result.items[2])\n end)\n end)\nend)\n" } ]