mirror of
https://github.com/QwenLM/qwen-code.git
synced 2025-12-19 09:33:53 +00:00
Sync upstream Gemini-CLI v0.8.2 (#838)
This commit is contained in:
7
.github/CODEOWNERS
vendored
7
.github/CODEOWNERS
vendored
@@ -1,7 +0,0 @@
|
|||||||
# By default, require reviews from the release approvers for all files.
|
|
||||||
* @google-gemini/gemini-cli-askmode-approvers
|
|
||||||
|
|
||||||
# The following files don't need reviews from the release approvers.
|
|
||||||
# These patterns override the rule above.
|
|
||||||
**/*.md
|
|
||||||
/docs/
|
|
||||||
31
.github/scripts/pr-triage.sh
vendored
31
.github/scripts/pr-triage.sh
vendored
@@ -19,24 +19,10 @@ process_pr() {
|
|||||||
local PR_NUMBER=$1
|
local PR_NUMBER=$1
|
||||||
echo "🔄 Processing PR #${PR_NUMBER}"
|
echo "🔄 Processing PR #${PR_NUMBER}"
|
||||||
|
|
||||||
# Get PR body with error handling
|
# Get closing issue number with error handling
|
||||||
local PR_BODY
|
local ISSUE_NUMBER
|
||||||
if ! PR_BODY=$(gh pr view "${PR_NUMBER}" --repo "${GITHUB_REPOSITORY}" --json body -q .body 2>/dev/null); then
|
if ! ISSUE_NUMBER=$(gh pr view "${PR_NUMBER}" --repo "${GITHUB_REPOSITORY}" --json closingIssuesReferences -q '.closingIssuesReferences.nodes[0].number' 2>/dev/null); then
|
||||||
echo " ⚠️ Could not fetch PR #${PR_NUMBER} details"
|
echo " ⚠️ Could not fetch closing issue for PR #${PR_NUMBER}"
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Look for issue references using multiple patterns
|
|
||||||
local ISSUE_NUMBER=""
|
|
||||||
|
|
||||||
# Pattern 1: Direct reference like #123
|
|
||||||
if [[ -z "${ISSUE_NUMBER}" ]]; then
|
|
||||||
ISSUE_NUMBER=$(echo "${PR_BODY}" | grep -oE '#[0-9]+' | head -1 | sed 's/#//' 2>/dev/null || echo "")
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Pattern 2: Closes/Fixes/Resolves patterns (case-insensitive)
|
|
||||||
if [[ -z "${ISSUE_NUMBER}" ]]; then
|
|
||||||
ISSUE_NUMBER=$(echo "${PR_BODY}" | grep -iE '(closes?|fixes?|resolves?) #[0-9]+' | grep -oE '#[0-9]+' | head -1 | sed 's/#//' 2>/dev/null || echo "")
|
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [[ -z "${ISSUE_NUMBER}" ]]; then
|
if [[ -z "${ISSUE_NUMBER}" ]]; then
|
||||||
@@ -110,14 +96,7 @@ process_pr() {
|
|||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [[ -n "${LABELS_TO_REMOVE}" ]]; then
|
if [[ -z "${LABELS_TO_ADD}" ]]; then
|
||||||
echo "➖ Removing labels: ${LABELS_TO_REMOVE}"
|
|
||||||
if ! gh pr edit "${PR_NUMBER}" --repo "${GITHUB_REPOSITORY}" --remove-label "${LABELS_TO_REMOVE}" 2>/dev/null; then
|
|
||||||
echo " ⚠️ Failed to remove some labels"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ -z "${LABELS_TO_ADD}" ]] && [[ -z "${LABELS_TO_REMOVE}" ]]; then
|
|
||||||
echo "✅ Labels already synchronized"
|
echo "✅ Labels already synchronized"
|
||||||
fi
|
fi
|
||||||
echo "needs_comment=false" >> "${GITHUB_OUTPUT}"
|
echo "needs_comment=false" >> "${GITHUB_OUTPUT}"
|
||||||
|
|||||||
221
.github/workflows/ci.yml
vendored
221
.github/workflows/ci.yml
vendored
@@ -12,6 +12,13 @@ on:
|
|||||||
- 'main'
|
- 'main'
|
||||||
- 'release/**'
|
- 'release/**'
|
||||||
merge_group:
|
merge_group:
|
||||||
|
workflow_dispatch:
|
||||||
|
inputs:
|
||||||
|
branch_ref:
|
||||||
|
description: 'Branch to run on'
|
||||||
|
required: true
|
||||||
|
default: 'main'
|
||||||
|
type: 'string'
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: '${{ github.workflow }}-${{ github.head_ref || github.ref }}'
|
group: '${{ github.workflow }}-${{ github.head_ref || github.ref }}'
|
||||||
@@ -33,222 +40,48 @@ env:
|
|||||||
YAMLLINT_VERSION: '1.35.1'
|
YAMLLINT_VERSION: '1.35.1'
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
#
|
lint:
|
||||||
# Lint: GitHub Actions
|
name: 'Lint'
|
||||||
#
|
|
||||||
lint_github_actions:
|
|
||||||
name: 'Lint (GitHub Actions)'
|
|
||||||
runs-on: 'ubuntu-latest'
|
runs-on: 'ubuntu-latest'
|
||||||
steps:
|
steps:
|
||||||
- name: 'Checkout'
|
- name: 'Checkout'
|
||||||
uses: 'actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8' # ratchet:actions/checkout@v5
|
uses: 'actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8' # ratchet:actions/checkout@v5
|
||||||
with:
|
with:
|
||||||
fetch-depth: 1
|
ref: '${{ github.event.inputs.branch_ref || github.ref }}'
|
||||||
|
fetch-depth: 0
|
||||||
- name: 'Install shellcheck' # Actionlint uses shellcheck
|
|
||||||
run: |-
|
|
||||||
mkdir -p "${RUNNER_TEMP}/shellcheck"
|
|
||||||
curl -sSLo "${RUNNER_TEMP}/.shellcheck.txz" "https://github.com/koalaman/shellcheck/releases/download/v${SHELLCHECK_VERSION}/shellcheck-v${SHELLCHECK_VERSION}.linux.x86_64.tar.xz"
|
|
||||||
tar -xf "${RUNNER_TEMP}/.shellcheck.txz" -C "${RUNNER_TEMP}/shellcheck" --strip-components=1
|
|
||||||
echo "${RUNNER_TEMP}/shellcheck" >> "${GITHUB_PATH}"
|
|
||||||
|
|
||||||
- name: 'Install actionlint'
|
|
||||||
run: |-
|
|
||||||
mkdir -p "${RUNNER_TEMP}/actionlint"
|
|
||||||
curl -sSLo "${RUNNER_TEMP}/.actionlint.tgz" "https://github.com/rhysd/actionlint/releases/download/v${ACTIONLINT_VERSION}/actionlint_${ACTIONLINT_VERSION}_linux_amd64.tar.gz"
|
|
||||||
tar -xzf "${RUNNER_TEMP}/.actionlint.tgz" -C "${RUNNER_TEMP}/actionlint"
|
|
||||||
echo "${RUNNER_TEMP}/actionlint" >> "${GITHUB_PATH}"
|
|
||||||
|
|
||||||
# For actionlint, we specifically ignore shellcheck rules that are
|
|
||||||
# annoying or unhelpful. See the shellcheck action for a description.
|
|
||||||
- name: 'Run actionlint'
|
|
||||||
run: |-
|
|
||||||
actionlint \
|
|
||||||
-color \
|
|
||||||
-format '{{range $err := .}}::error file={{$err.Filepath}},line={{$err.Line}},col={{$err.Column}}::{{$err.Filepath}}@{{$err.Line}} {{$err.Message}}%0A```%0A{{replace $err.Snippet "\\n" "%0A"}}%0A```\n{{end}}' \
|
|
||||||
-ignore 'SC2002:' \
|
|
||||||
-ignore 'SC2016:' \
|
|
||||||
-ignore 'SC2129:' \
|
|
||||||
-ignore 'label ".+" is unknown'
|
|
||||||
|
|
||||||
- name: 'Run ratchet'
|
|
||||||
uses: 'sethvargo/ratchet@8b4ca256dbed184350608a3023620f267f0a5253' # ratchet:sethvargo/ratchet@v0.11.4
|
|
||||||
with:
|
|
||||||
files: |-
|
|
||||||
.github/workflows/*.yml
|
|
||||||
.github/actions/**/*.yml
|
|
||||||
|
|
||||||
#
|
|
||||||
# Lint: Javascript
|
|
||||||
#
|
|
||||||
lint_javascript:
|
|
||||||
name: 'Lint (Javascript)'
|
|
||||||
runs-on: 'ubuntu-latest'
|
|
||||||
steps:
|
|
||||||
- name: 'Checkout'
|
|
||||||
uses: 'actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8' # ratchet:actions/checkout@v5
|
|
||||||
with:
|
|
||||||
fetch-depth: 1
|
|
||||||
|
|
||||||
- name: 'Set up Node.js'
|
- name: 'Set up Node.js'
|
||||||
uses: 'actions/setup-node@49933ea5288caeca8642d1e84afbd3f7d6820020' # ratchet:actions/setup-node@v4.4.0
|
uses: 'actions/setup-node@49933ea5288caeca8642d1e84afbd3f7d6820020' # ratchet:actions/setup-node@v4.4.0
|
||||||
with:
|
with:
|
||||||
node-version-file: '.nvmrc'
|
node-version-file: '.nvmrc'
|
||||||
cache: 'npm'
|
cache: 'npm'
|
||||||
cache-dependency-path: 'package-lock.json'
|
|
||||||
registry-url: 'https://registry.npmjs.org/'
|
|
||||||
|
|
||||||
- name: 'Configure npm for rate limiting'
|
|
||||||
run: |-
|
|
||||||
npm config set fetch-retry-mintimeout 20000
|
|
||||||
npm config set fetch-retry-maxtimeout 120000
|
|
||||||
npm config set fetch-retries 5
|
|
||||||
npm config set fetch-timeout 300000
|
|
||||||
|
|
||||||
- name: 'Install dependencies'
|
- name: 'Install dependencies'
|
||||||
run: |-
|
run: 'npm ci'
|
||||||
npm ci --prefer-offline --no-audit --progress=false
|
|
||||||
|
|
||||||
- name: 'Run formatter check'
|
- name: 'Check lockfile'
|
||||||
run: |-
|
run: 'npm run check:lockfile'
|
||||||
npm run format
|
|
||||||
git diff --exit-code
|
|
||||||
|
|
||||||
- name: 'Run linter'
|
- name: 'Install linters'
|
||||||
run: |-
|
run: 'node scripts/lint.js --setup'
|
||||||
npm run lint:ci
|
|
||||||
|
|
||||||
- name: 'Run linter on integration tests'
|
- name: 'Run ESLint'
|
||||||
run: |-
|
run: 'node scripts/lint.js --eslint'
|
||||||
npx eslint integration-tests --max-warnings 0
|
|
||||||
|
|
||||||
- name: 'Run formatter on integration tests'
|
- name: 'Run actionlint'
|
||||||
run: |-
|
run: 'node scripts/lint.js --actionlint'
|
||||||
npx prettier --check integration-tests
|
|
||||||
git diff --exit-code
|
|
||||||
|
|
||||||
- name: 'Build project'
|
|
||||||
run: |-
|
|
||||||
npm run build
|
|
||||||
|
|
||||||
- name: 'Run type check'
|
|
||||||
run: |-
|
|
||||||
npm run typecheck
|
|
||||||
|
|
||||||
#
|
|
||||||
# Lint: Shell
|
|
||||||
#
|
|
||||||
lint_shell:
|
|
||||||
name: 'Lint (Shell)'
|
|
||||||
runs-on: 'ubuntu-latest'
|
|
||||||
steps:
|
|
||||||
- name: 'Checkout'
|
|
||||||
uses: 'actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8' # ratchet:actions/checkout@v5
|
|
||||||
with:
|
|
||||||
fetch-depth: 1
|
|
||||||
|
|
||||||
- name: 'Install shellcheck'
|
|
||||||
run: |-
|
|
||||||
mkdir -p "${RUNNER_TEMP}/shellcheck"
|
|
||||||
curl -sSLo "${RUNNER_TEMP}/.shellcheck.txz" "https://github.com/koalaman/shellcheck/releases/download/v${SHELLCHECK_VERSION}/shellcheck-v${SHELLCHECK_VERSION}.linux.x86_64.tar.xz"
|
|
||||||
tar -xf "${RUNNER_TEMP}/.shellcheck.txz" -C "${RUNNER_TEMP}/shellcheck" --strip-components=1
|
|
||||||
echo "${RUNNER_TEMP}/shellcheck" >> "${GITHUB_PATH}"
|
|
||||||
|
|
||||||
- name: 'Install shellcheck problem matcher'
|
|
||||||
run: |-
|
|
||||||
cat > "${RUNNER_TEMP}/shellcheck/problem-matcher-lint-shell.json" <<"EOF"
|
|
||||||
{
|
|
||||||
"problemMatcher": [
|
|
||||||
{
|
|
||||||
"owner": "lint_shell",
|
|
||||||
"pattern": [
|
|
||||||
{
|
|
||||||
"regexp": "^(.*):(\\\\d+):(\\\\d+):\\\\s+(?:fatal\\\\s+)?(warning|error):\\\\s+(.*)$",
|
|
||||||
"file": 1,
|
|
||||||
"line": 2,
|
|
||||||
"column": 3,
|
|
||||||
"severity": 4,
|
|
||||||
"message": 5
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
EOF
|
|
||||||
echo "::add-matcher::${RUNNER_TEMP}/shellcheck/problem-matcher-lint-shell.json"
|
|
||||||
|
|
||||||
# Note that only warning and error severity show up in the github files
|
|
||||||
# page. So we replace 'style' and 'note' with 'warning' to make it show
|
|
||||||
# up.
|
|
||||||
#
|
|
||||||
# We also try and find all bash scripts even if they don't have an
|
|
||||||
# explicit extension.
|
|
||||||
#
|
|
||||||
# We explicitly ignore the following rules:
|
|
||||||
#
|
|
||||||
# - SC2002: This rule suggests using "cmd < file" instead of "cat | cmd".
|
|
||||||
# While < is more efficient, pipes are much more readable and expected.
|
|
||||||
#
|
|
||||||
# - SC2129: This rule suggests grouping multiple writes to a file in
|
|
||||||
# braces like "{ cmd1; cmd2; } >> file". This is unexpected and less
|
|
||||||
# readable.
|
|
||||||
#
|
|
||||||
# - SC2310: This is an optional warning that only appears with "set -e"
|
|
||||||
# and when a command is used as a conditional.
|
|
||||||
- name: 'Run shellcheck'
|
- name: 'Run shellcheck'
|
||||||
run: |-
|
run: 'node scripts/lint.js --shellcheck'
|
||||||
git ls-files | grep -E '^([^.]+|.*\.(sh|zsh|bash))$' | grep -v '^integration-tests/terminal-bench/ci-tasks/' | xargs file --mime-type \
|
|
||||||
| grep "text/x-shellscript" | awk '{ print substr($1, 1, length($1)-1) }' \
|
|
||||||
| xargs shellcheck \
|
|
||||||
--check-sourced \
|
|
||||||
--enable=all \
|
|
||||||
--exclude=SC2002,SC2129,SC2310 \
|
|
||||||
--severity=style \
|
|
||||||
--format=gcc \
|
|
||||||
--color=never | sed -e 's/note:/warning:/g' -e 's/style:/warning:/g'
|
|
||||||
|
|
||||||
#
|
|
||||||
# Lint: YAML
|
|
||||||
#
|
|
||||||
lint_yaml:
|
|
||||||
name: 'Lint (YAML)'
|
|
||||||
runs-on: 'ubuntu-latest'
|
|
||||||
steps:
|
|
||||||
- name: 'Checkout'
|
|
||||||
uses: 'actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8' # ratchet:actions/checkout@v5
|
|
||||||
with:
|
|
||||||
fetch-depth: 1
|
|
||||||
|
|
||||||
- name: 'Setup Python'
|
|
||||||
uses: 'actions/setup-python@a26af69be951a213d495a4c3e4e4022e16d87065' # ratchet:actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: '3'
|
|
||||||
|
|
||||||
- name: 'Install yamllint'
|
|
||||||
run: |-
|
|
||||||
pip install --user "yamllint==${YAMLLINT_VERSION}"
|
|
||||||
|
|
||||||
- name: 'Run yamllint'
|
- name: 'Run yamllint'
|
||||||
run: |-
|
run: 'node scripts/lint.js --yamllint'
|
||||||
git ls-files | grep -E '\.(yaml|yml)' | grep -v '^integration-tests/terminal-bench/ci-tasks/' | xargs yamllint --format github
|
|
||||||
|
|
||||||
#
|
- name: 'Run Prettier'
|
||||||
# Lint: All
|
run: 'node scripts/lint.js --prettier'
|
||||||
#
|
|
||||||
# This is a virtual job that other jobs depend on to wait for all linters to
|
- name: 'Run sensitive keyword linter'
|
||||||
# finish. It's also used to ensure linting happens on CI via required
|
run: 'node scripts/lint.js --sensitive-keywords'
|
||||||
# workflows.
|
|
||||||
lint:
|
|
||||||
name: 'Lint'
|
|
||||||
needs:
|
|
||||||
- 'lint_github_actions'
|
|
||||||
- 'lint_javascript'
|
|
||||||
- 'lint_shell'
|
|
||||||
- 'lint_yaml'
|
|
||||||
runs-on: 'ubuntu-latest'
|
|
||||||
steps:
|
|
||||||
- run: |-
|
|
||||||
echo 'All linters finished!'
|
|
||||||
|
|
||||||
#
|
#
|
||||||
# Test: Node
|
# Test: Node
|
||||||
|
|||||||
@@ -50,7 +50,8 @@ jobs:
|
|||||||
|
|
||||||
// Search for open issues already assigned to the commenter in this repo
|
// Search for open issues already assigned to the commenter in this repo
|
||||||
const { data: assignedIssues } = await github.rest.search.issuesAndPullRequests({
|
const { data: assignedIssues } = await github.rest.search.issuesAndPullRequests({
|
||||||
q: `is:issue repo:${owner}/${repo} assignee:${commenter} is:open`
|
q: `is:issue repo:${owner}/${repo} assignee:${commenter} is:open`,
|
||||||
|
advanced_search: true
|
||||||
});
|
});
|
||||||
|
|
||||||
if (assignedIssues.total_count >= MAX_ISSUES_ASSIGNED) {
|
if (assignedIssues.total_count >= MAX_ISSUES_ASSIGNED) {
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ on:
|
|||||||
type: 'number'
|
type: 'number'
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: '${{ github.workflow }}-${{ github.event.issue.number }}'
|
group: '${{ github.workflow }}-${{ github.event.issue.number || github.event.inputs.issue_number }}'
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
defaults:
|
defaults:
|
||||||
@@ -29,6 +29,7 @@ permissions:
|
|||||||
issues: 'write'
|
issues: 'write'
|
||||||
statuses: 'write'
|
statuses: 'write'
|
||||||
packages: 'read'
|
packages: 'read'
|
||||||
|
actions: 'write' # Required for cancelling a workflow run
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
triage-issue:
|
triage-issue:
|
||||||
@@ -62,9 +63,9 @@ jobs:
|
|||||||
## Role
|
## Role
|
||||||
|
|
||||||
You are an issue triage assistant. Analyze the current GitHub issue
|
You are an issue triage assistant. Analyze the current GitHub issue
|
||||||
and identify the most appropriate existing labels. Use the available
|
and identify the most appropriate existing labels by only using the provided data. Use the available
|
||||||
tools to gather information; do not ask for information to be
|
tools to gather information; do not ask for information to be
|
||||||
provided. Do not remove labels titled help wanted or good first issue.
|
provided. Do not remove the following labels titled maintainer, help wanted or good first issue.
|
||||||
|
|
||||||
## Steps
|
## Steps
|
||||||
|
|
||||||
@@ -92,23 +93,29 @@ jobs:
|
|||||||
|
|
||||||
Categorization Guidelines:
|
Categorization Guidelines:
|
||||||
P0: Critical / Blocker
|
P0: Critical / Blocker
|
||||||
- A P0 bug is a catastrophic failure that demands immediate attention. It represents a complete showstopper for a significant portion of users or for the development process itself.
|
- A P0 bug is a catastrophic failure that demands immediate attention.
|
||||||
|
- To be a P0 it means almost all users are running into this issue and it is blocking users from being able to use the product.
|
||||||
|
- You would see this in the form of many comments from different developers on the bug.
|
||||||
|
- It represents a complete showstopper for a significant portion of users or for the development process itself.
|
||||||
Impact:
|
Impact:
|
||||||
- Blocks development or testing for the entire team.
|
- Blocks development or testing for the entire team.
|
||||||
- Major security vulnerability that could compromise user data or system integrity.
|
- Major security vulnerability that could compromise user data or system integrity.
|
||||||
- Causes data loss or corruption with no workaround.
|
- Causes data loss or corruption with no workaround.
|
||||||
- Crashes the application or makes a core feature completely unusable for all or most users in a production environment. Will it cause severe quality degration? Is it preventing contributors from contributing to the repository or is it a release blocker?
|
- Crashes the application or makes a core feature completely unusable for all or most users in a production environment. Will it cause severe quality degration? Is it preventing contributors from contributing to the repository or is it a release blocker?
|
||||||
Qualifier: Is the main function of the software broken?
|
Qualifier: Is the main function of the software broken?
|
||||||
Example: The gemini auth login command fails with an unrecoverable error, preventing any user from authenticating and using the rest of the CLI.
|
Example: The qwen auth login command fails with an unrecoverable error, preventing any user from authenticating and using the rest of the CLI.
|
||||||
P1: High
|
P1: High
|
||||||
- A P1 bug is a serious issue that significantly degrades the user experience or impacts a core feature. While not a complete blocker, it's a major problem that needs a fast resolution. Feature requests are almost never P1.
|
- A P1 bug is a serious issue that significantly degrades the user experience or impacts a core feature.
|
||||||
|
- While not a complete blocker, it's a major problem that needs a fast resolution. Feature requests are almost never P1.
|
||||||
|
- Once again this would be affecting many users.
|
||||||
|
- You would see this in the form of comments from different developers on the bug.
|
||||||
Impact:
|
Impact:
|
||||||
- A core feature is broken or behaving incorrectly for a large number of users or large number of use cases.
|
- A core feature is broken or behaving incorrectly for a large number of users or large number of use cases.
|
||||||
- Review the bug details and comments to try figure out if this issue affects a large set of use cases or if it's a narrow set of use cases.
|
- Review the bug details and comments to try figure out if this issue affects a large set of use cases or if it's a narrow set of use cases.
|
||||||
- Severe performance degradation making the application frustratingly slow.
|
- Severe performance degradation making the application frustratingly slow.
|
||||||
- No straightforward workaround exists, or the workaround is difficult and non-obvious.
|
- No straightforward workaround exists, or the workaround is difficult and non-obvious.
|
||||||
Qualifier: Is a key feature unusable or giving very wrong results?
|
Qualifier: Is a key feature unusable or giving very wrong results?
|
||||||
Example: The gemini -p "..." command consistently returns a malformed JSON response or an empty result, making the CLI's primary generation feature unreliable.
|
Example: Qwen Code enters a loop when making read-many-files tool call. I am unable to break out of the loop and qwen doesn't follow instructions subsequently.
|
||||||
P2: Medium
|
P2: Medium
|
||||||
- A P2 bug is a moderately impactful issue. It's a noticeable problem but doesn't prevent the use of the software's main functionality.
|
- A P2 bug is a moderately impactful issue. It's a noticeable problem but doesn't prevent the use of the software's main functionality.
|
||||||
Impact:
|
Impact:
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ jobs:
|
|||||||
|
|
||||||
echo '🏷️ Finding issues that need triage...'
|
echo '🏷️ Finding issues that need triage...'
|
||||||
NEED_TRIAGE_ISSUES="$(gh issue list --repo "${GITHUB_REPOSITORY}" \
|
NEED_TRIAGE_ISSUES="$(gh issue list --repo "${GITHUB_REPOSITORY}" \
|
||||||
--search 'is:open is:issue label:"status/needs-triage"' --json number,title,body)"
|
--search "is:open is:issue label:\"status/need-triage\"" --limit 1000 --json number,title,body)"
|
||||||
|
|
||||||
echo '🔄 Merging and deduplicating issues...'
|
echo '🔄 Merging and deduplicating issues...'
|
||||||
ISSUES="$(echo "${NO_LABEL_ISSUES}" "${NEED_TRIAGE_ISSUES}" | jq -c -s 'add | unique_by(.number)')"
|
ISSUES="$(echo "${NO_LABEL_ISSUES}" "${NEED_TRIAGE_ISSUES}" | jq -c -s 'add | unique_by(.number)')"
|
||||||
@@ -111,7 +111,7 @@ jobs:
|
|||||||
- Do not include any explanation or additional text, just the JSON
|
- Do not include any explanation or additional text, just the JSON
|
||||||
- Only use labels that already exist in the repository.
|
- Only use labels that already exist in the repository.
|
||||||
- Do not add comments or modify the issue content.
|
- Do not add comments or modify the issue content.
|
||||||
- Do not remove labels titled help wanted or good first issue.
|
- Do not remove the following labels maintainer, help wanted or good first issue.
|
||||||
- Triage only the current issue.
|
- Triage only the current issue.
|
||||||
- Identify only one category/ label
|
- Identify only one category/ label
|
||||||
- Identify only one type/ label (Do not apply type/duplicate or type/parent-issue)
|
- Identify only one type/ label (Do not apply type/duplicate or type/parent-issue)
|
||||||
@@ -119,7 +119,10 @@ jobs:
|
|||||||
- Once you categorize the issue if it needs information bump down the priority by 1 eg.. a p0 would become a p1 a p1 would become a p2. P2 and P3 can stay as is in this scenario.
|
- Once you categorize the issue if it needs information bump down the priority by 1 eg.. a p0 would become a p1 a p1 would become a p2. P2 and P3 can stay as is in this scenario.
|
||||||
Categorization Guidelines:
|
Categorization Guidelines:
|
||||||
P0: Critical / Blocker
|
P0: Critical / Blocker
|
||||||
- A P0 bug is a catastrophic failure that demands immediate attention. It represents a complete showstopper for a significant portion of users or for the development process itself.
|
- A P0 bug is a catastrophic failure that demands immediate attention.
|
||||||
|
- To be a P0 it means almost all users are running into this issue and it is blocking users from being able to use the product.
|
||||||
|
- You would see this in the form of many comments from different developers on the bug.
|
||||||
|
- It represents a complete showstopper for a significant portion of users or for the development process itself.
|
||||||
Impact:
|
Impact:
|
||||||
- Blocks development or testing for the entire team.
|
- Blocks development or testing for the entire team.
|
||||||
- Major security vulnerability that could compromise user data or system integrity.
|
- Major security vulnerability that could compromise user data or system integrity.
|
||||||
@@ -129,15 +132,17 @@ jobs:
|
|||||||
Qualifier: Is the main function of the software broken?
|
Qualifier: Is the main function of the software broken?
|
||||||
Example: The gemini auth login command fails with an unrecoverable error, preventing any user from authenticating and using the rest of the CLI.
|
Example: The gemini auth login command fails with an unrecoverable error, preventing any user from authenticating and using the rest of the CLI.
|
||||||
P1: High
|
P1: High
|
||||||
- A P1 bug is a serious issue that significantly degrades the user experience or impacts a core feature. While not a complete blocker, it's a major problem that needs a fast resolution.
|
- A P1 bug is a serious issue that significantly degrades the user experience or impacts a core feature.
|
||||||
- Feature requests are almost never P1.
|
- While not a complete blocker, it's a major problem that needs a fast resolution. Feature requests are almost never P1.
|
||||||
|
- Once again this would be affecting many users.
|
||||||
|
- You would see this in the form of comments from different developers on the bug.
|
||||||
Impact:
|
Impact:
|
||||||
- A core feature is broken or behaving incorrectly for a large number of users or large number of use cases.
|
- A core feature is broken or behaving incorrectly for a large number of users or large number of use cases.
|
||||||
- Review the bug details and comments to try figure out if this issue affects a large set of use cases or if it's a narrow set of use cases.
|
- Review the bug details and comments to try figure out if this issue affects a large set of use cases or if it's a narrow set of use cases.
|
||||||
- Severe performance degradation making the application frustratingly slow.
|
- Severe performance degradation making the application frustratingly slow.
|
||||||
- No straightforward workaround exists, or the workaround is difficult and non-obvious.
|
- No straightforward workaround exists, or the workaround is difficult and non-obvious.
|
||||||
Qualifier: Is a key feature unusable or giving very wrong results?
|
Qualifier: Is a key feature unusable or giving very wrong results?
|
||||||
Example: The gemini -p "..." command consistently returns a malformed JSON response or an empty result, making the CLI's primary generation feature unreliable.
|
Example: Gemini CLI enters a loop when making read-many-files tool call. I am unable to break out of the loop and gemini doesn't follow instructions subsequently.
|
||||||
P2: Medium
|
P2: Medium
|
||||||
- A P2 bug is a moderately impactful issue. It's a noticeable problem but doesn't prevent the use of the software's main functionality.
|
- A P2 bug is a moderately impactful issue. It's a noticeable problem but doesn't prevent the use of the software's main functionality.
|
||||||
Impact:
|
Impact:
|
||||||
|
|||||||
14
.gitignore
vendored
14
.gitignore
vendored
@@ -3,14 +3,21 @@
|
|||||||
.env~
|
.env~
|
||||||
|
|
||||||
# gemini-cli settings
|
# gemini-cli settings
|
||||||
.gemini/
|
# We want to keep the .gemini in the root of the repo and ignore any .gemini
|
||||||
!gemini/config.yaml
|
# in subdirectories. In our root .gemini we want to allow for version control
|
||||||
|
# for subcommands.
|
||||||
|
**/.gemini/
|
||||||
|
!/.gemini/
|
||||||
|
.gemini/*
|
||||||
|
!.gemini/config.yaml
|
||||||
|
!.gemini/commands/
|
||||||
|
|
||||||
# Note: .gemini-clipboard/ is NOT in gitignore so Gemini can access pasted images
|
# Note: .gemini-clipboard/ is NOT in gitignore so Gemini can access pasted images
|
||||||
|
|
||||||
# Dependency directory
|
# Dependency directory
|
||||||
node_modules
|
node_modules
|
||||||
bower_components
|
bower_components
|
||||||
|
package-lock.json
|
||||||
|
|
||||||
# Editors
|
# Editors
|
||||||
.idea
|
.idea
|
||||||
@@ -48,4 +55,5 @@ logs/
|
|||||||
# GHA credentials
|
# GHA credentials
|
||||||
gha-creds-*.json
|
gha-creds-*.json
|
||||||
|
|
||||||
QWEN.md
|
# Log files
|
||||||
|
patch_output.log
|
||||||
|
|||||||
9
.husky/pre-commit
Executable file
9
.husky/pre-commit
Executable file
@@ -0,0 +1,9 @@
|
|||||||
|
npm run pre-commit || {
|
||||||
|
echo ''
|
||||||
|
echo '===================================================='
|
||||||
|
echo 'pre-commit checks failed. in case of emergency, run:'
|
||||||
|
echo ''
|
||||||
|
echo 'git commit --no-verify'
|
||||||
|
echo '===================================================='
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
3
.vscode/settings.json
vendored
3
.vscode/settings.json
vendored
@@ -12,5 +12,6 @@
|
|||||||
},
|
},
|
||||||
"[javascript]": {
|
"[javascript]": {
|
||||||
"editor.defaultFormatter": "esbenp.prettier-vscode"
|
"editor.defaultFormatter": "esbenp.prettier-vscode"
|
||||||
}
|
},
|
||||||
|
"vitest.disableWorkspaceWarning": true
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -86,3 +86,5 @@ ignore:
|
|||||||
- 'thirdparty/'
|
- 'thirdparty/'
|
||||||
- 'third_party/'
|
- 'third_party/'
|
||||||
- 'vendor/'
|
- 'vendor/'
|
||||||
|
- 'node_modules/'
|
||||||
|
- 'integration-tests/terminal-bench/'
|
||||||
|
|||||||
@@ -136,7 +136,7 @@ To start the Gemini CLI from the source code (after building), run the following
|
|||||||
npm start
|
npm start
|
||||||
```
|
```
|
||||||
|
|
||||||
If you'd like to run the source build outside of the gemini-cli folder you can utilize `npm link path/to/gemini-cli/packages/cli` (see: [docs](https://docs.npmjs.com/cli/v9/commands/npm-link)) or `alias gemini="node path/to/gemini-cli/packages/cli"` to run with `gemini`
|
If you'd like to run the source build outside of the gemini-cli folder, you can utilize `npm link path/to/gemini-cli/packages/cli` (see: [docs](https://docs.npmjs.com/cli/v9/commands/npm-link)) or `alias gemini="node path/to/gemini-cli/packages/cli"` to run with `gemini`
|
||||||
|
|
||||||
### Running Tests
|
### Running Tests
|
||||||
|
|
||||||
|
|||||||
@@ -38,10 +38,12 @@ Add the following key to your `settings.json`:
|
|||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
|
"general": {
|
||||||
"checkpointing": {
|
"checkpointing": {
|
||||||
"enabled": true
|
"enabled": true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
## Using the `/restore` Command
|
## Using the `/restore` Command
|
||||||
|
|||||||
@@ -131,13 +131,7 @@ OpenAI-compatible API method if configured:
|
|||||||
|
|
||||||
**Example for headless environments:**
|
**Example for headless environments:**
|
||||||
|
|
||||||
```bash
|
If none of these environment variables are set in a non-interactive session, the CLI will exit with an error.
|
||||||
export OPENAI_API_KEY="your-api-key"
|
|
||||||
export OPENAI_BASE_URL="https://api-inference.modelscope.cn/v1"
|
|
||||||
export OPENAI_MODEL="Qwen/Qwen3-Coder-480B-A35B-Instruct"
|
|
||||||
|
|
||||||
# Run Qwen Code
|
For comprehensive guidance on using Qwen COde programmatically and in
|
||||||
qwen
|
automation workflows, see the [Headless Mode Guide](../headless.md).
|
||||||
```
|
|
||||||
|
|
||||||
If no API key is set in a non-interactive session, the CLI will exit with an error prompting you to configure authentication.
|
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ Slash commands provide meta-level control over the CLI itself.
|
|||||||
### Built-in Commands
|
### Built-in Commands
|
||||||
|
|
||||||
- **`/bug`**
|
- **`/bug`**
|
||||||
- **Description:** File an issue about Qwen Code. By default, the issue is filed within the GitHub repository for Qwen Code. The string you enter after `/bug` will become the headline for the bug being filed. The default `/bug` behavior can be modified using the `bugCommand` setting in your `.qwen/settings.json` files.
|
- **Description:** File an issue about Qwen Code. By default, the issue is filed within the GitHub repository for Qwen Code. The string you enter after `/bug` will become the headline for the bug being filed. The default `/bug` behavior can be modified using the `advanced.bugCommand` setting in your `.qwen/settings.json` files.
|
||||||
|
|
||||||
- **`/chat`**
|
- **`/chat`**
|
||||||
- **Description:** Save and resume conversation history for branching conversation state interactively, or resuming a previous state from a later session.
|
- **Description:** Save and resume conversation history for branching conversation state interactively, or resuming a previous state from a later session.
|
||||||
@@ -30,6 +30,9 @@ Slash commands provide meta-level control over the CLI itself.
|
|||||||
- **`delete`**
|
- **`delete`**
|
||||||
- **Description:** Deletes a saved conversation checkpoint.
|
- **Description:** Deletes a saved conversation checkpoint.
|
||||||
- **Usage:** `/chat delete <tag>`
|
- **Usage:** `/chat delete <tag>`
|
||||||
|
- **`share`**
|
||||||
|
- **Description** Writes the current conversation to a provided Markdown or JSON file.
|
||||||
|
- **Usage** `/chat share file.md` or `/chat share file.json`. If no filename is provided, then the CLI will generate one.
|
||||||
|
|
||||||
- **`/clear`**
|
- **`/clear`**
|
||||||
- **Description:** Clear the terminal screen, including the visible session history and scrollback within the CLI. The underlying session data (for history recall) might be preserved depending on the exact implementation, but the visual display is cleared.
|
- **Description:** Clear the terminal screen, including the visible session history and scrollback within the CLI. The underlying session data (for history recall) might be preserved depending on the exact implementation, but the visual display is cleared.
|
||||||
@@ -162,9 +165,6 @@ Slash commands provide meta-level control over the CLI itself.
|
|||||||
- **`nodesc`** or **`nodescriptions`**:
|
- **`nodesc`** or **`nodescriptions`**:
|
||||||
- **Description:** Hide tool descriptions, showing only the tool names.
|
- **Description:** Hide tool descriptions, showing only the tool names.
|
||||||
|
|
||||||
- **`/privacy`**
|
|
||||||
- **Description:** Display the Privacy Notice and allow users to select whether they consent to the collection of their data for service improvement purposes.
|
|
||||||
|
|
||||||
- **`/quit-confirm`**
|
- **`/quit-confirm`**
|
||||||
- **Description:** Show a confirmation dialog before exiting Qwen Code, allowing you to choose how to handle your current session.
|
- **Description:** Show a confirmation dialog before exiting Qwen Code, allowing you to choose how to handle your current session.
|
||||||
- **Usage:** `/quit-confirm`
|
- **Usage:** `/quit-confirm`
|
||||||
@@ -435,6 +435,16 @@ That's it! You can now run your command in the CLI. First, you might add a file
|
|||||||
|
|
||||||
Qwen Code will then execute the multi-line prompt defined in your TOML file.
|
Qwen Code will then execute the multi-line prompt defined in your TOML file.
|
||||||
|
|
||||||
|
## Input Prompt Shortcuts
|
||||||
|
|
||||||
|
These shortcuts apply directly to the input prompt for text manipulation.
|
||||||
|
|
||||||
|
- **Undo:**
|
||||||
|
- **Keyboard shortcut:** Press **Ctrl+z** to undo the last action in the input prompt.
|
||||||
|
|
||||||
|
- **Redo:**
|
||||||
|
- **Keyboard shortcut:** Press **Ctrl+Shift+Z** to redo the last undone action in the input prompt.
|
||||||
|
|
||||||
## At commands (`@`)
|
## At commands (`@`)
|
||||||
|
|
||||||
At commands are used to include the content of files or directories as part of your prompt to the model. These commands include git-aware filtering.
|
At commands are used to include the content of files or directories as part of your prompt to the model. These commands include git-aware filtering.
|
||||||
@@ -450,7 +460,7 @@ At commands are used to include the content of files or directories as part of y
|
|||||||
- If a path to a directory is provided, the command attempts to read the content of files within that directory and any subdirectories.
|
- If a path to a directory is provided, the command attempts to read the content of files within that directory and any subdirectories.
|
||||||
- Spaces in paths should be escaped with a backslash (e.g., `@My\ Documents/file.txt`).
|
- Spaces in paths should be escaped with a backslash (e.g., `@My\ Documents/file.txt`).
|
||||||
- The command uses the `read_many_files` tool internally. The content is fetched and then inserted into your query before being sent to the model.
|
- The command uses the `read_many_files` tool internally. The content is fetched and then inserted into your query before being sent to the model.
|
||||||
- **Git-aware filtering:** By default, git-ignored files (like `node_modules/`, `dist/`, `.env`, `.git/`) are excluded. This behavior can be changed via the `fileFiltering` settings.
|
- **Git-aware filtering:** By default, git-ignored files (like `node_modules/`, `dist/`, `.env`, `.git/`) are excluded. This behavior can be changed via the `context.fileFiltering` settings.
|
||||||
- **File types:** The command is intended for text-based files. While it might attempt to read any file, binary files or very large files might be skipped or truncated by the underlying `read_many_files` tool to ensure performance and relevance. The tool indicates if files were skipped.
|
- **File types:** The command is intended for text-based files. While it might attempt to read any file, binary files or very large files might be skipped or truncated by the underlying `read_many_files` tool to ensure performance and relevance. The tool indicates if files were skipped.
|
||||||
- **Output:** The CLI will show a tool call message indicating that `read_many_files` was used, along with a message detailing the status and the path(s) that were processed.
|
- **Output:** The CLI will show a tool call message indicating that `read_many_files` was used, along with a message detailing the status and the path(s) that were processed.
|
||||||
|
|
||||||
|
|||||||
670
docs/cli/configuration-v1.md
Normal file
670
docs/cli/configuration-v1.md
Normal file
@@ -0,0 +1,670 @@
|
|||||||
|
# Qwen Code Configuration
|
||||||
|
|
||||||
|
Qwen Code offers several ways to configure its behavior, including environment variables, command-line arguments, and settings files. This document outlines the different configuration methods and available settings.
|
||||||
|
|
||||||
|
## Configuration layers
|
||||||
|
|
||||||
|
Configuration is applied in the following order of precedence (lower numbers are overridden by higher numbers):
|
||||||
|
|
||||||
|
1. **Default values:** Hardcoded defaults within the application.
|
||||||
|
2. **System defaults file:** System-wide default settings that can be overridden by other settings files.
|
||||||
|
3. **User settings file:** Global settings for the current user.
|
||||||
|
4. **Project settings file:** Project-specific settings.
|
||||||
|
5. **System settings file:** System-wide settings that override all other settings files.
|
||||||
|
6. **Environment variables:** System-wide or session-specific variables, potentially loaded from `.env` files.
|
||||||
|
7. **Command-line arguments:** Values passed when launching the CLI.
|
||||||
|
|
||||||
|
## Settings files
|
||||||
|
|
||||||
|
Qwen Code uses JSON settings files for persistent configuration. There are four locations for these files:
|
||||||
|
|
||||||
|
- **System defaults file:**
|
||||||
|
- **Location:** `/etc/qwen-code/system-defaults.json` (Linux), `C:\ProgramData\qwen-code\system-defaults.json` (Windows) or `/Library/Application Support/QwenCode/system-defaults.json` (macOS). The path can be overridden using the `QWEN_CODE_SYSTEM_DEFAULTS_PATH` environment variable.
|
||||||
|
- **Scope:** Provides a base layer of system-wide default settings. These settings have the lowest precedence and are intended to be overridden by user, project, or system override settings.
|
||||||
|
- **User settings file:**
|
||||||
|
- **Location:** `~/.qwen/settings.json` (where `~` is your home directory).
|
||||||
|
- **Scope:** Applies to all Qwen Code sessions for the current user.
|
||||||
|
- **Project settings file:**
|
||||||
|
- **Location:** `.qwen/settings.json` within your project's root directory.
|
||||||
|
- **Scope:** Applies only when running Qwen Code from that specific project. Project settings override user settings.
|
||||||
|
|
||||||
|
- **System settings file:**
|
||||||
|
- **Location:** `/etc/qwen-code/settings.json` (Linux), `C:\ProgramData\qwen-code\settings.json` (Windows) or `/Library/Application Support/QwenCode/settings.json` (macOS). The path can be overridden using the `QWEN_CODE_SYSTEM_SETTINGS_PATH` environment variable.
|
||||||
|
- **Scope:** Applies to all Qwen Code sessions on the system, for all users. System settings override user and project settings. May be useful for system administrators at enterprises to have controls over users' Qwen Code setups.
|
||||||
|
|
||||||
|
**Note on environment variables in settings:** String values within your `settings.json` files can reference environment variables using either `$VAR_NAME` or `${VAR_NAME}` syntax. These variables will be automatically resolved when the settings are loaded. For example, if you have an environment variable `MY_API_TOKEN`, you could use it in `settings.json` like this: `"apiKey": "$MY_API_TOKEN"`.
|
||||||
|
|
||||||
|
### The `.qwen` directory in your project
|
||||||
|
|
||||||
|
In addition to a project settings file, a project's `.qwen` directory can contain other project-specific files related to Qwen Code's operation, such as:
|
||||||
|
|
||||||
|
- [Custom sandbox profiles](#sandboxing) (e.g., `.qwen/sandbox-macos-custom.sb`, `.qwen/sandbox.Dockerfile`).
|
||||||
|
|
||||||
|
### Available settings in `settings.json`:
|
||||||
|
|
||||||
|
- **`contextFileName`** (string or array of strings):
|
||||||
|
- **Description:** Specifies the filename for context files (e.g., `QWEN.md`, `AGENTS.md`). Can be a single filename or a list of accepted filenames.
|
||||||
|
- **Default:** `QWEN.md`
|
||||||
|
- **Example:** `"contextFileName": "AGENTS.md"`
|
||||||
|
|
||||||
|
- **`bugCommand`** (object):
|
||||||
|
- **Description:** Overrides the default URL for the `/bug` command.
|
||||||
|
- **Default:** `"urlTemplate": "https://github.com/QwenLM/qwen-code/issues/new?template=bug_report.yml&title={title}&info={info}"`
|
||||||
|
- **Properties:**
|
||||||
|
- **`urlTemplate`** (string): A URL that can contain `{title}` and `{info}` placeholders.
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"bugCommand": {
|
||||||
|
"urlTemplate": "https://bug.example.com/new?title={title}&info={info}"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`fileFiltering`** (object):
|
||||||
|
- **Description:** Controls git-aware file filtering behavior for @ commands and file discovery tools.
|
||||||
|
- **Default:** `"respectGitIgnore": true, "enableRecursiveFileSearch": true`
|
||||||
|
- **Properties:**
|
||||||
|
- **`respectGitIgnore`** (boolean): Whether to respect .gitignore patterns when discovering files. When set to `true`, git-ignored files (like `node_modules/`, `dist/`, `.env`) are automatically excluded from @ commands and file listing operations.
|
||||||
|
- **`enableRecursiveFileSearch`** (boolean): Whether to enable searching recursively for filenames under the current tree when completing @ prefixes in the prompt.
|
||||||
|
- **`disableFuzzySearch`** (boolean): When `true`, disables the fuzzy search capabilities when searching for files, which can improve performance on projects with a large number of files.
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"fileFiltering": {
|
||||||
|
"respectGitIgnore": true,
|
||||||
|
"enableRecursiveFileSearch": false,
|
||||||
|
"disableFuzzySearch": true
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Troubleshooting File Search Performance
|
||||||
|
|
||||||
|
If you are experiencing performance issues with file searching (e.g., with `@` completions), especially in projects with a very large number of files, here are a few things you can try in order of recommendation:
|
||||||
|
|
||||||
|
1. **Use `.qwenignore`:** Create a `.qwenignore` file in your project root to exclude directories that contain a large number of files that you don't need to reference (e.g., build artifacts, logs, `node_modules`). Reducing the total number of files crawled is the most effective way to improve performance.
|
||||||
|
|
||||||
|
2. **Disable Fuzzy Search:** If ignoring files is not enough, you can disable fuzzy search by setting `disableFuzzySearch` to `true` in your `settings.json` file. This will use a simpler, non-fuzzy matching algorithm, which can be faster.
|
||||||
|
|
||||||
|
3. **Disable Recursive File Search:** As a last resort, you can disable recursive file search entirely by setting `enableRecursiveFileSearch` to `false`. This will be the fastest option as it avoids a recursive crawl of your project. However, it means you will need to type the full path to files when using `@` completions.
|
||||||
|
|
||||||
|
- **`coreTools`** (array of strings):
|
||||||
|
- **Description:** Allows you to specify a list of core tool names that should be made available to the model. This can be used to restrict the set of built-in tools. See [Built-in Tools](../core/tools-api.md#built-in-tools) for a list of core tools. You can also specify command-specific restrictions for tools that support it, like the `ShellTool`. For example, `"coreTools": ["ShellTool(ls -l)"]` will only allow the `ls -l` command to be executed.
|
||||||
|
- **Default:** All tools available for use by the model.
|
||||||
|
- **Example:** `"coreTools": ["ReadFileTool", "GlobTool", "ShellTool(ls)"]`.
|
||||||
|
|
||||||
|
- **`allowedTools`** (array of strings):
|
||||||
|
- **Default:** `undefined`
|
||||||
|
- **Description:** A list of tool names that will bypass the confirmation dialog. This is useful for tools that you trust and use frequently. The match semantics are the same as `coreTools`.
|
||||||
|
- **Example:** `"allowedTools": ["ShellTool(git status)"]`.
|
||||||
|
|
||||||
|
- **`excludeTools`** (array of strings):
|
||||||
|
- **Description:** Allows you to specify a list of core tool names that should be excluded from the model. A tool listed in both `excludeTools` and `coreTools` is excluded. You can also specify command-specific restrictions for tools that support it, like the `ShellTool`. For example, `"excludeTools": ["ShellTool(rm -rf)"]` will block the `rm -rf` command.
|
||||||
|
- **Default**: No tools excluded.
|
||||||
|
- **Example:** `"excludeTools": ["run_shell_command", "findFiles"]`.
|
||||||
|
- **Security Note:** Command-specific restrictions in
|
||||||
|
`excludeTools` for `run_shell_command` are based on simple string matching and can be easily bypassed. This feature is **not a security mechanism** and should not be relied upon to safely execute untrusted code. It is recommended to use `coreTools` to explicitly select commands
|
||||||
|
that can be executed.
|
||||||
|
|
||||||
|
- **`allowMCPServers`** (array of strings):
|
||||||
|
- **Description:** Allows you to specify a list of MCP server names that should be made available to the model. This can be used to restrict the set of MCP servers to connect to. Note that this will be ignored if `--allowed-mcp-server-names` is set.
|
||||||
|
- **Default:** All MCP servers are available for use by the model.
|
||||||
|
- **Example:** `"allowMCPServers": ["myPythonServer"]`.
|
||||||
|
- **Security Note:** This uses simple string matching on MCP server names, which can be modified. If you're a system administrator looking to prevent users from bypassing this, consider configuring the `mcpServers` at the system settings level such that the user will not be able to configure any MCP servers of their own. This should not be used as an airtight security mechanism.
|
||||||
|
|
||||||
|
- **`excludeMCPServers`** (array of strings):
|
||||||
|
- **Description:** Allows you to specify a list of MCP server names that should be excluded from the model. A server listed in both `excludeMCPServers` and `allowMCPServers` is excluded. Note that this will be ignored if `--allowed-mcp-server-names` is set.
|
||||||
|
- **Default**: No MCP servers excluded.
|
||||||
|
- **Example:** `"excludeMCPServers": ["myNodeServer"]`.
|
||||||
|
- **Security Note:** This uses simple string matching on MCP server names, which can be modified. If you're a system administrator looking to prevent users from bypassing this, consider configuring the `mcpServers` at the system settings level such that the user will not be able to configure any MCP servers of their own. This should not be used as an airtight security mechanism.
|
||||||
|
|
||||||
|
- **`autoAccept`** (boolean):
|
||||||
|
- **Description:** Controls whether the CLI automatically accepts and executes tool calls that are considered safe (e.g., read-only operations) without explicit user confirmation. If set to `true`, the CLI will bypass the confirmation prompt for tools deemed safe.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:** `"autoAccept": true`
|
||||||
|
|
||||||
|
- **`theme`** (string):
|
||||||
|
- **Description:** Sets the visual [theme](./themes.md) for Qwen Code.
|
||||||
|
- **Default:** `"Default"`
|
||||||
|
- **Example:** `"theme": "GitHub"`
|
||||||
|
|
||||||
|
- **`vimMode`** (boolean):
|
||||||
|
- **Description:** Enables or disables vim mode for input editing. When enabled, the input area supports vim-style navigation and editing commands with NORMAL and INSERT modes. The vim mode status is displayed in the footer and persists between sessions.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:** `"vimMode": true`
|
||||||
|
|
||||||
|
- **`sandbox`** (boolean or string):
|
||||||
|
- **Description:** Controls whether and how to use sandboxing for tool execution. If set to `true`, Qwen Code uses a pre-built `qwen-code-sandbox` Docker image. For more information, see [Sandboxing](#sandboxing).
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:** `"sandbox": "docker"`
|
||||||
|
|
||||||
|
- **`toolDiscoveryCommand`** (string):
|
||||||
|
- **Description:** **Align with Gemini CLI.** Defines a custom shell command for discovering tools from your project. The shell command must return on `stdout` a JSON array of [function declarations](https://ai.google.dev/gemini-api/docs/function-calling#function-declarations). Tool wrappers are optional.
|
||||||
|
- **Default:** Empty
|
||||||
|
- **Example:** `"toolDiscoveryCommand": "bin/get_tools"`
|
||||||
|
|
||||||
|
- **`toolCallCommand`** (string):
|
||||||
|
- **Description:** **Align with Gemini CLI.** Defines a custom shell command for calling a specific tool that was discovered using `toolDiscoveryCommand`. The shell command must meet the following criteria:
|
||||||
|
- It must take function `name` (exactly as in [function declaration](https://ai.google.dev/gemini-api/docs/function-calling#function-declarations)) as first command line argument.
|
||||||
|
- It must read function arguments as JSON on `stdin`, analogous to [`functionCall.args`](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/inference#functioncall).
|
||||||
|
- It must return function output as JSON on `stdout`, analogous to [`functionResponse.response.content`](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/inference#functionresponse).
|
||||||
|
- **Default:** Empty
|
||||||
|
- **Example:** `"toolCallCommand": "bin/call_tool"`
|
||||||
|
|
||||||
|
- **`mcpServers`** (object):
|
||||||
|
- **Description:** Configures connections to one or more Model-Context Protocol (MCP) servers for discovering and using custom tools. Qwen Code attempts to connect to each configured MCP server to discover available tools. If multiple MCP servers expose a tool with the same name, the tool names will be prefixed with the server alias you defined in the configuration (e.g., `serverAlias__actualToolName`) to avoid conflicts. Note that the system might strip certain schema properties from MCP tool definitions for compatibility. At least one of `command`, `url`, or `httpUrl` must be provided. If multiple are specified, the order of precedence is `httpUrl`, then `url`, then `command`.
|
||||||
|
- **Default:** Empty
|
||||||
|
- **Properties:**
|
||||||
|
- **`<SERVER_NAME>`** (object): The server parameters for the named server.
|
||||||
|
- `command` (string, optional): The command to execute to start the MCP server via standard I/O.
|
||||||
|
- `args` (array of strings, optional): Arguments to pass to the command.
|
||||||
|
- `env` (object, optional): Environment variables to set for the server process.
|
||||||
|
- `cwd` (string, optional): The working directory in which to start the server.
|
||||||
|
- `url` (string, optional): The URL of an MCP server that uses Server-Sent Events (SSE) for communication.
|
||||||
|
- `httpUrl` (string, optional): The URL of an MCP server that uses streamable HTTP for communication.
|
||||||
|
- `headers` (object, optional): A map of HTTP headers to send with requests to `url` or `httpUrl`.
|
||||||
|
- `timeout` (number, optional): Timeout in milliseconds for requests to this MCP server.
|
||||||
|
- `trust` (boolean, optional): Trust this server and bypass all tool call confirmations.
|
||||||
|
- `description` (string, optional): A brief description of the server, which may be used for display purposes.
|
||||||
|
- `includeTools` (array of strings, optional): List of tool names to include from this MCP server. When specified, only the tools listed here will be available from this server (whitelist behavior). If not specified, all tools from the server are enabled by default.
|
||||||
|
- `excludeTools` (array of strings, optional): List of tool names to exclude from this MCP server. Tools listed here will not be available to the model, even if they are exposed by the server. **Note:** `excludeTools` takes precedence over `includeTools` - if a tool is in both lists, it will be excluded.
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"mcpServers": {
|
||||||
|
"myPythonServer": {
|
||||||
|
"command": "python",
|
||||||
|
"args": ["mcp_server.py", "--port", "8080"],
|
||||||
|
"cwd": "./mcp_tools/python",
|
||||||
|
"timeout": 5000,
|
||||||
|
"includeTools": ["safe_tool", "file_reader"],
|
||||||
|
},
|
||||||
|
"myNodeServer": {
|
||||||
|
"command": "node",
|
||||||
|
"args": ["mcp_server.js"],
|
||||||
|
"cwd": "./mcp_tools/node",
|
||||||
|
"excludeTools": ["dangerous_tool", "file_deleter"]
|
||||||
|
},
|
||||||
|
"myDockerServer": {
|
||||||
|
"command": "docker",
|
||||||
|
"args": ["run", "-i", "--rm", "-e", "API_KEY", "ghcr.io/foo/bar"],
|
||||||
|
"env": {
|
||||||
|
"API_KEY": "$MY_API_TOKEN"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"mySseServer": {
|
||||||
|
"url": "http://localhost:8081/events",
|
||||||
|
"headers": {
|
||||||
|
"Authorization": "Bearer $MY_SSE_TOKEN"
|
||||||
|
},
|
||||||
|
"description": "An example SSE-based MCP server."
|
||||||
|
},
|
||||||
|
"myStreamableHttpServer": {
|
||||||
|
"httpUrl": "http://localhost:8082/stream",
|
||||||
|
"headers": {
|
||||||
|
"X-API-Key": "$MY_HTTP_API_KEY"
|
||||||
|
},
|
||||||
|
"description": "An example Streamable HTTP-based MCP server."
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`checkpointing`** (object):
|
||||||
|
- **Description:** Configures the checkpointing feature, which allows you to save and restore conversation and file states. See the [Checkpointing documentation](../checkpointing.md) for more details.
|
||||||
|
- **Default:** `{"enabled": false}`
|
||||||
|
- **Properties:**
|
||||||
|
- **`enabled`** (boolean): When `true`, the `/restore` command is available.
|
||||||
|
|
||||||
|
- **`preferredEditor`** (string):
|
||||||
|
- **Description:** Specifies the preferred editor to use for viewing diffs.
|
||||||
|
- **Default:** `vscode`
|
||||||
|
- **Example:** `"preferredEditor": "vscode"`
|
||||||
|
|
||||||
|
- **`telemetry`** (object)
|
||||||
|
- **Description:** Configures logging and metrics collection for Qwen Code. For more information, see [Telemetry](../telemetry.md).
|
||||||
|
- **Default:** `{"enabled": false, "target": "local", "otlpEndpoint": "http://localhost:4317", "logPrompts": true}`
|
||||||
|
- **Properties:**
|
||||||
|
- **`enabled`** (boolean): Whether or not telemetry is enabled.
|
||||||
|
- **`target`** (string): The destination for collected telemetry. Supported values are `local` and `gcp`.
|
||||||
|
- **`otlpEndpoint`** (string): The endpoint for the OTLP Exporter.
|
||||||
|
- **`logPrompts`** (boolean): Whether or not to include the content of user prompts in the logs.
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"telemetry": {
|
||||||
|
"enabled": true,
|
||||||
|
"target": "local",
|
||||||
|
"otlpEndpoint": "http://localhost:16686",
|
||||||
|
"logPrompts": false
|
||||||
|
}
|
||||||
|
```
|
||||||
|
- **`usageStatisticsEnabled`** (boolean):
|
||||||
|
- **Description:** Enables or disables the collection of usage statistics. See [Usage Statistics](#usage-statistics) for more information.
|
||||||
|
- **Default:** `true`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"usageStatisticsEnabled": false
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`hideTips`** (boolean):
|
||||||
|
- **Description:** Enables or disables helpful tips in the CLI interface.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:**
|
||||||
|
|
||||||
|
```json
|
||||||
|
"hideTips": true
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`hideBanner`** (boolean):
|
||||||
|
- **Description:** Enables or disables the startup banner (ASCII art logo) in the CLI interface.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:**
|
||||||
|
|
||||||
|
```json
|
||||||
|
"hideBanner": true
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`maxSessionTurns`** (number):
|
||||||
|
- **Description:** Sets the maximum number of turns for a session. If the session exceeds this limit, the CLI will stop processing and start a new chat.
|
||||||
|
- **Default:** `-1` (unlimited)
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"maxSessionTurns": 10
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`summarizeToolOutput`** (object):
|
||||||
|
- **Description:** Enables or disables the summarization of tool output. You can specify the token budget for the summarization using the `tokenBudget` setting.
|
||||||
|
- Note: Currently only the `run_shell_command` tool is supported.
|
||||||
|
- **Default:** `{}` (Disabled by default)
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"summarizeToolOutput": {
|
||||||
|
"run_shell_command": {
|
||||||
|
"tokenBudget": 2000
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`excludedProjectEnvVars`** (array of strings):
|
||||||
|
- **Description:** Specifies environment variables that should be excluded from being loaded from project `.env` files. This prevents project-specific environment variables (like `DEBUG=true`) from interfering with the CLI behavior. Variables from `.qwen/.env` files are never excluded.
|
||||||
|
- **Default:** `["DEBUG", "DEBUG_MODE"]`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"excludedProjectEnvVars": ["DEBUG", "DEBUG_MODE", "NODE_ENV"]
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`includeDirectories`** (array of strings):
|
||||||
|
- **Description:** Specifies an array of additional absolute or relative paths to include in the workspace context. Missing directories will be skipped with a warning by default. Paths can use `~` to refer to the user's home directory. This setting can be combined with the `--include-directories` command-line flag.
|
||||||
|
- **Default:** `[]`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"includeDirectories": [
|
||||||
|
"/path/to/another/project",
|
||||||
|
"../shared-library",
|
||||||
|
"~/common-utils"
|
||||||
|
]
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`loadMemoryFromIncludeDirectories`** (boolean):
|
||||||
|
- **Description:** Controls the behavior of the `/memory refresh` command. If set to `true`, `QWEN.md` files should be loaded from all directories that are added. If set to `false`, `QWEN.md` should only be loaded from the current directory.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"loadMemoryFromIncludeDirectories": true
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`tavilyApiKey`** (string):
|
||||||
|
- **Description:** API key for Tavily web search service. Required to enable the `web_search` tool functionality. If not configured, the web search tool will be disabled and skipped.
|
||||||
|
- **Default:** `undefined` (web search disabled)
|
||||||
|
- **Example:** `"tavilyApiKey": "tvly-your-api-key-here"`
|
||||||
|
- **`chatCompression`** (object):
|
||||||
|
- **Description:** Controls the settings for chat history compression, both automatic and
|
||||||
|
when manually invoked through the /compress command.
|
||||||
|
- **Properties:**
|
||||||
|
- **`contextPercentageThreshold`** (number): A value between 0 and 1 that specifies the token threshold for compression as a percentage of the model's total token limit. For example, a value of `0.6` will trigger compression when the chat history exceeds 60% of the token limit.
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"chatCompression": {
|
||||||
|
"contextPercentageThreshold": 0.6
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`showLineNumbers`** (boolean):
|
||||||
|
- **Description:** Controls whether line numbers are displayed in code blocks in the CLI output.
|
||||||
|
- **Default:** `true`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"showLineNumbers": false
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`accessibility`** (object):
|
||||||
|
- **Description:** Configures accessibility features for the CLI.
|
||||||
|
- **Properties:**
|
||||||
|
- **`screenReader`** (boolean): Enables screen reader mode, which adjusts the TUI for better compatibility with screen readers. This can also be enabled with the `--screen-reader` command-line flag, which will take precedence over the setting.
|
||||||
|
- **`disableLoadingPhrases`** (boolean): Disables the display of loading phrases during operations.
|
||||||
|
- **Default:** `{"screenReader": false, "disableLoadingPhrases": false}`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"accessibility": {
|
||||||
|
"screenReader": true,
|
||||||
|
"disableLoadingPhrases": true
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`skipNextSpeakerCheck`** (boolean):
|
||||||
|
- **Description:** Skips the next speaker check after text responses. When enabled, the system bypasses analyzing whether the AI should continue speaking.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"skipNextSpeakerCheck": true
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`skipLoopDetection`** (boolean):
|
||||||
|
- **Description:** Disables all loop detection checks (streaming and LLM-based). Loop detection prevents infinite loops in AI responses but can generate false positives that interrupt legitimate workflows. Enable this option if you experience frequent false positive loop detection interruptions.
|
||||||
|
- **Default:** `false`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"skipLoopDetection": true
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`approvalMode`** (string):
|
||||||
|
- **Description:** Sets the default approval mode for tool usage. Accepted values are:
|
||||||
|
- `plan`: Analyze only, do not modify files or execute commands.
|
||||||
|
- `default`: Require approval before file edits or shell commands run.
|
||||||
|
- `auto-edit`: Automatically approve file edits.
|
||||||
|
- `yolo`: Automatically approve all tool calls.
|
||||||
|
- **Default:** `"default"`
|
||||||
|
- **Example:**
|
||||||
|
```json
|
||||||
|
"approvalMode": "plan"
|
||||||
|
```
|
||||||
|
|
||||||
|
### Example `settings.json`:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"theme": "GitHub",
|
||||||
|
"sandbox": "docker",
|
||||||
|
"toolDiscoveryCommand": "bin/get_tools",
|
||||||
|
"toolCallCommand": "bin/call_tool",
|
||||||
|
"tavilyApiKey": "$TAVILY_API_KEY",
|
||||||
|
"mcpServers": {
|
||||||
|
"mainServer": {
|
||||||
|
"command": "bin/mcp_server.py"
|
||||||
|
},
|
||||||
|
"anotherServer": {
|
||||||
|
"command": "node",
|
||||||
|
"args": ["mcp_server.js", "--verbose"]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"telemetry": {
|
||||||
|
"enabled": true,
|
||||||
|
"target": "local",
|
||||||
|
"otlpEndpoint": "http://localhost:4317",
|
||||||
|
"logPrompts": true
|
||||||
|
},
|
||||||
|
"usageStatisticsEnabled": true,
|
||||||
|
"hideTips": false,
|
||||||
|
"hideBanner": false,
|
||||||
|
"skipNextSpeakerCheck": false,
|
||||||
|
"skipLoopDetection": false,
|
||||||
|
"maxSessionTurns": 10,
|
||||||
|
"summarizeToolOutput": {
|
||||||
|
"run_shell_command": {
|
||||||
|
"tokenBudget": 100
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"excludedProjectEnvVars": ["DEBUG", "DEBUG_MODE", "NODE_ENV"],
|
||||||
|
"includeDirectories": ["path/to/dir1", "~/path/to/dir2", "../path/to/dir3"],
|
||||||
|
"loadMemoryFromIncludeDirectories": true
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## Shell History
|
||||||
|
|
||||||
|
The CLI keeps a history of shell commands you run. To avoid conflicts between different projects, this history is stored in a project-specific directory within your user's home folder.
|
||||||
|
|
||||||
|
- **Location:** `~/.qwen/tmp/<project_hash>/shell_history`
|
||||||
|
- `<project_hash>` is a unique identifier generated from your project's root path.
|
||||||
|
- The history is stored in a file named `shell_history`.
|
||||||
|
|
||||||
|
## Environment Variables & `.env` Files
|
||||||
|
|
||||||
|
Environment variables are a common way to configure applications, especially for sensitive information like API keys or for settings that might change between environments. For authentication setup, see the [Authentication documentation](./authentication.md) which covers all available authentication methods.
|
||||||
|
|
||||||
|
The CLI automatically loads environment variables from an `.env` file. The loading order is:
|
||||||
|
|
||||||
|
1. `.env` file in the current working directory.
|
||||||
|
2. If not found, it searches upwards in parent directories until it finds an `.env` file or reaches the project root (identified by a `.git` folder) or the home directory.
|
||||||
|
3. If still not found, it looks for `~/.env` (in the user's home directory).
|
||||||
|
|
||||||
|
**Environment Variable Exclusion:** Some environment variables (like `DEBUG` and `DEBUG_MODE`) are automatically excluded from project `.env` files by default to prevent interference with the CLI behavior. Variables from `.qwen/.env` files are never excluded. You can customize this behavior using the `excludedProjectEnvVars` setting in your `settings.json` file.
|
||||||
|
|
||||||
|
- **`OPENAI_API_KEY`**:
|
||||||
|
- One of several available [authentication methods](./authentication.md).
|
||||||
|
- Set this in your shell profile (e.g., `~/.bashrc`, `~/.zshrc`) or an `.env` file.
|
||||||
|
- **`OPENAI_BASE_URL`**:
|
||||||
|
- One of several available [authentication methods](./authentication.md).
|
||||||
|
- Set this in your shell profile (e.g., `~/.bashrc`, `~/.zshrc`) or an `.env` file.
|
||||||
|
- **`OPENAI_MODEL`**:
|
||||||
|
- Specifies the default OPENAI model to use.
|
||||||
|
- Overrides the hardcoded default
|
||||||
|
- Example: `export OPENAI_MODEL="qwen3-coder-plus"`
|
||||||
|
- **`GEMINI_SANDBOX`**:
|
||||||
|
- Alternative to the `sandbox` setting in `settings.json`.
|
||||||
|
- Accepts `true`, `false`, `docker`, `podman`, or a custom command string.
|
||||||
|
- **`SEATBELT_PROFILE`** (macOS specific):
|
||||||
|
- Switches the Seatbelt (`sandbox-exec`) profile on macOS.
|
||||||
|
- `permissive-open`: (Default) Restricts writes to the project folder (and a few other folders, see `packages/cli/src/utils/sandbox-macos-permissive-open.sb`) but allows other operations.
|
||||||
|
- `strict`: Uses a strict profile that declines operations by default.
|
||||||
|
- `<profile_name>`: Uses a custom profile. To define a custom profile, create a file named `sandbox-macos-<profile_name>.sb` in your project's `.qwen/` directory (e.g., `my-project/.qwen/sandbox-macos-custom.sb`).
|
||||||
|
- **`DEBUG` or `DEBUG_MODE`** (often used by underlying libraries or the CLI itself):
|
||||||
|
- Set to `true` or `1` to enable verbose debug logging, which can be helpful for troubleshooting.
|
||||||
|
- **Note:** These variables are automatically excluded from project `.env` files by default to prevent interference with the CLI behavior. Use `.qwen/.env` files if you need to set these for Qwen Code specifically.
|
||||||
|
- **`NO_COLOR`**:
|
||||||
|
- Set to any value to disable all color output in the CLI.
|
||||||
|
- **`CLI_TITLE`**:
|
||||||
|
- Set to a string to customize the title of the CLI.
|
||||||
|
- **`CODE_ASSIST_ENDPOINT`**:
|
||||||
|
- Specifies the endpoint for the code assist server.
|
||||||
|
- This is useful for development and testing.
|
||||||
|
- **`TAVILY_API_KEY`**:
|
||||||
|
- Your API key for the Tavily web search service.
|
||||||
|
- Required to enable the `web_search` tool functionality.
|
||||||
|
- If not configured, the web search tool will be disabled and skipped.
|
||||||
|
- Example: `export TAVILY_API_KEY="tvly-your-api-key-here"`
|
||||||
|
|
||||||
|
## Command-Line Arguments
|
||||||
|
|
||||||
|
Arguments passed directly when running the CLI can override other configurations for that specific session.
|
||||||
|
|
||||||
|
- **`--model <model_name>`** (**`-m <model_name>`**):
|
||||||
|
- Specifies the Qwen model to use for this session.
|
||||||
|
- Example: `npm start -- --model qwen3-coder-plus`
|
||||||
|
- **`--prompt <your_prompt>`** (**`-p <your_prompt>`**):
|
||||||
|
- Used to pass a prompt directly to the command. This invokes Qwen Code in a non-interactive mode.
|
||||||
|
- **`--prompt-interactive <your_prompt>`** (**`-i <your_prompt>`**):
|
||||||
|
- Starts an interactive session with the provided prompt as the initial input.
|
||||||
|
- The prompt is processed within the interactive session, not before it.
|
||||||
|
- Cannot be used when piping input from stdin.
|
||||||
|
- Example: `qwen -i "explain this code"`
|
||||||
|
- **`--sandbox`** (**`-s`**):
|
||||||
|
- Enables sandbox mode for this session.
|
||||||
|
- **`--sandbox-image`**:
|
||||||
|
- Sets the sandbox image URI.
|
||||||
|
- **`--debug`** (**`-d`**):
|
||||||
|
- Enables debug mode for this session, providing more verbose output.
|
||||||
|
- **`--all-files`** (**`-a`**):
|
||||||
|
- If set, recursively includes all files within the current directory as context for the prompt.
|
||||||
|
- **`--help`** (or **`-h`**):
|
||||||
|
- Displays help information about command-line arguments.
|
||||||
|
- **`--show-memory-usage`**:
|
||||||
|
- Displays the current memory usage.
|
||||||
|
- **`--yolo`**:
|
||||||
|
- Enables YOLO mode, which automatically approves all tool calls.
|
||||||
|
- **`--approval-mode <mode>`**:
|
||||||
|
- Sets the approval mode for tool calls. Supported modes:
|
||||||
|
- `plan`: Analyze only—do not modify files or execute commands.
|
||||||
|
- `default`: Require approval for file edits or shell commands (default behavior).
|
||||||
|
- `auto-edit`: Automatically approve edit tools (edit, write_file) while prompting for others.
|
||||||
|
- `yolo`: Automatically approve all tool calls (equivalent to `--yolo`).
|
||||||
|
- Cannot be used together with `--yolo`. Use `--approval-mode=yolo` instead of `--yolo` for the new unified approach.
|
||||||
|
- Example: `qwen --approval-mode auto-edit`
|
||||||
|
- **`--allowed-tools <tool1,tool2,...>`**:
|
||||||
|
- A comma-separated list of tool names that will bypass the confirmation dialog.
|
||||||
|
- Example: `qwen --allowed-tools "ShellTool(git status)"`
|
||||||
|
- **`--telemetry`**:
|
||||||
|
- Enables [telemetry](../telemetry.md).
|
||||||
|
- **`--telemetry-target`**:
|
||||||
|
- Sets the telemetry target. See [telemetry](../telemetry.md) for more information.
|
||||||
|
- **`--telemetry-otlp-endpoint`**:
|
||||||
|
- Sets the OTLP endpoint for telemetry. See [telemetry](../telemetry.md) for more information.
|
||||||
|
- **`--telemetry-otlp-protocol`**:
|
||||||
|
- Sets the OTLP protocol for telemetry (`grpc` or `http`). Defaults to `grpc`. See [telemetry](../telemetry.md) for more information.
|
||||||
|
- **`--telemetry-log-prompts`**:
|
||||||
|
- Enables logging of prompts for telemetry. See [telemetry](../telemetry.md) for more information.
|
||||||
|
- **`--checkpointing`**:
|
||||||
|
- Enables [checkpointing](../checkpointing.md).
|
||||||
|
- **`--extensions <extension_name ...>`** (**`-e <extension_name ...>`**):
|
||||||
|
- Specifies a list of extensions to use for the session. If not provided, all available extensions are used.
|
||||||
|
- Use the special term `qwen -e none` to disable all extensions.
|
||||||
|
- Example: `qwen -e my-extension -e my-other-extension`
|
||||||
|
- **`--list-extensions`** (**`-l`**):
|
||||||
|
- Lists all available extensions and exits.
|
||||||
|
- **`--proxy`**:
|
||||||
|
- Sets the proxy for the CLI.
|
||||||
|
- Example: `--proxy http://localhost:7890`.
|
||||||
|
- **`--include-directories <dir1,dir2,...>`**:
|
||||||
|
- Includes additional directories in the workspace for multi-directory support.
|
||||||
|
- Can be specified multiple times or as comma-separated values.
|
||||||
|
- 5 directories can be added at maximum.
|
||||||
|
- Example: `--include-directories /path/to/project1,/path/to/project2` or `--include-directories /path/to/project1 --include-directories /path/to/project2`
|
||||||
|
- **`--screen-reader`**:
|
||||||
|
- Enables screen reader mode for accessibility.
|
||||||
|
- **`--version`**:
|
||||||
|
- Displays the version of the CLI.
|
||||||
|
- **`--openai-logging`**:
|
||||||
|
- Enables logging of OpenAI API calls for debugging and analysis. This flag overrides the `enableOpenAILogging` setting in `settings.json`.
|
||||||
|
- **`--tavily-api-key <api_key>`**:
|
||||||
|
- Sets the Tavily API key for web search functionality for this session.
|
||||||
|
- Example: `qwen --tavily-api-key tvly-your-api-key-here`
|
||||||
|
|
||||||
|
## Context Files (Hierarchical Instructional Context)
|
||||||
|
|
||||||
|
While not strictly configuration for the CLI's _behavior_, context files (defaulting to `QWEN.md` but configurable via the `contextFileName` setting) are crucial for configuring the _instructional context_ (also referred to as "memory"). This powerful feature allows you to give project-specific instructions, coding style guides, or any relevant background information to the AI, making its responses more tailored and accurate to your needs. The CLI includes UI elements, such as an indicator in the footer showing the number of loaded context files, to keep you informed about the active context.
|
||||||
|
|
||||||
|
- **Purpose:** These Markdown files contain instructions, guidelines, or context that you want the Qwen model to be aware of during your interactions. The system is designed to manage this instructional context hierarchically.
|
||||||
|
|
||||||
|
### Example Context File Content (e.g., `QWEN.md`)
|
||||||
|
|
||||||
|
Here's a conceptual example of what a context file at the root of a TypeScript project might contain:
|
||||||
|
|
||||||
|
```markdown
|
||||||
|
# Project: My Awesome TypeScript Library
|
||||||
|
|
||||||
|
## General Instructions:
|
||||||
|
|
||||||
|
- When generating new TypeScript code, please follow the existing coding style.
|
||||||
|
- Ensure all new functions and classes have JSDoc comments.
|
||||||
|
- Prefer functional programming paradigms where appropriate.
|
||||||
|
- All code should be compatible with TypeScript 5.0 and Node.js 20+.
|
||||||
|
|
||||||
|
## Coding Style:
|
||||||
|
|
||||||
|
- Use 2 spaces for indentation.
|
||||||
|
- Interface names should be prefixed with `I` (e.g., `IUserService`).
|
||||||
|
- Private class members should be prefixed with an underscore (`_`).
|
||||||
|
- Always use strict equality (`===` and `!==`).
|
||||||
|
|
||||||
|
## Specific Component: `src/api/client.ts`
|
||||||
|
|
||||||
|
- This file handles all outbound API requests.
|
||||||
|
- When adding new API call functions, ensure they include robust error handling and logging.
|
||||||
|
- Use the existing `fetchWithRetry` utility for all GET requests.
|
||||||
|
|
||||||
|
## Regarding Dependencies:
|
||||||
|
|
||||||
|
- Avoid introducing new external dependencies unless absolutely necessary.
|
||||||
|
- If a new dependency is required, please state the reason.
|
||||||
|
```
|
||||||
|
|
||||||
|
This example demonstrates how you can provide general project context, specific coding conventions, and even notes about particular files or components. The more relevant and precise your context files are, the better the AI can assist you. Project-specific context files are highly encouraged to establish conventions and context.
|
||||||
|
|
||||||
|
- **Hierarchical Loading and Precedence:** The CLI implements a sophisticated hierarchical memory system by loading context files (e.g., `QWEN.md`) from several locations. Content from files lower in this list (more specific) typically overrides or supplements content from files higher up (more general). The exact concatenation order and final context can be inspected using the `/memory show` command. The typical loading order is:
|
||||||
|
1. **Global Context File:**
|
||||||
|
- Location: `~/.qwen/<contextFileName>` (e.g., `~/.qwen/QWEN.md` in your user home directory).
|
||||||
|
- Scope: Provides default instructions for all your projects.
|
||||||
|
2. **Project Root & Ancestors Context Files:**
|
||||||
|
- Location: The CLI searches for the configured context file in the current working directory and then in each parent directory up to either the project root (identified by a `.git` folder) or your home directory.
|
||||||
|
- Scope: Provides context relevant to the entire project or a significant portion of it.
|
||||||
|
3. **Sub-directory Context Files (Contextual/Local):**
|
||||||
|
- Location: The CLI also scans for the configured context file in subdirectories _below_ the current working directory (respecting common ignore patterns like `node_modules`, `.git`, etc.). The breadth of this search is limited to 200 directories by default, but can be configured with a `memoryDiscoveryMaxDirs` field in your `settings.json` file.
|
||||||
|
- Scope: Allows for highly specific instructions relevant to a particular component, module, or subsection of your project.
|
||||||
|
- **Concatenation & UI Indication:** The contents of all found context files are concatenated (with separators indicating their origin and path) and provided as part of the system prompt. The CLI footer displays the count of loaded context files, giving you a quick visual cue about the active instructional context.
|
||||||
|
- **Importing Content:** You can modularize your context files by importing other Markdown files using the `@path/to/file.md` syntax. For more details, see the [Memory Import Processor documentation](../core/memport.md).
|
||||||
|
- **Commands for Memory Management:**
|
||||||
|
- Use `/memory refresh` to force a re-scan and reload of all context files from all configured locations. This updates the AI's instructional context.
|
||||||
|
- Use `/memory show` to display the combined instructional context currently loaded, allowing you to verify the hierarchy and content being used by the AI.
|
||||||
|
- See the [Commands documentation](./commands.md#memory) for full details on the `/memory` command and its sub-commands (`show` and `refresh`).
|
||||||
|
|
||||||
|
By understanding and utilizing these configuration layers and the hierarchical nature of context files, you can effectively manage the AI's memory and tailor Qwen Code's responses to your specific needs and projects.
|
||||||
|
|
||||||
|
## Sandboxing
|
||||||
|
|
||||||
|
Qwen Code can execute potentially unsafe operations (like shell commands and file modifications) within a sandboxed environment to protect your system.
|
||||||
|
|
||||||
|
Sandboxing is disabled by default, but you can enable it in a few ways:
|
||||||
|
|
||||||
|
- Using `--sandbox` or `-s` flag.
|
||||||
|
- Setting `GEMINI_SANDBOX` environment variable.
|
||||||
|
- Sandbox is enabled when using `--yolo` or `--approval-mode=yolo` by default.
|
||||||
|
|
||||||
|
By default, it uses a pre-built `qwen-code-sandbox` Docker image.
|
||||||
|
|
||||||
|
For project-specific sandboxing needs, you can create a custom Dockerfile at `.qwen/sandbox.Dockerfile` in your project's root directory. This Dockerfile can be based on the base sandbox image:
|
||||||
|
|
||||||
|
```dockerfile
|
||||||
|
FROM qwen-code-sandbox
|
||||||
|
|
||||||
|
# Add your custom dependencies or configurations here
|
||||||
|
# For example:
|
||||||
|
# RUN apt-get update && apt-get install -y some-package
|
||||||
|
# COPY ./my-config /app/my-config
|
||||||
|
```
|
||||||
|
|
||||||
|
When `.qwen/sandbox.Dockerfile` exists, you can use `BUILD_SANDBOX` environment variable when running Qwen Code to automatically build the custom sandbox image:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
BUILD_SANDBOX=1 qwen -s
|
||||||
|
```
|
||||||
|
|
||||||
|
## Usage Statistics
|
||||||
|
|
||||||
|
To help us improve Qwen Code, we collect anonymized usage statistics. This data helps us understand how the CLI is used, identify common issues, and prioritize new features.
|
||||||
|
|
||||||
|
**What we collect:**
|
||||||
|
|
||||||
|
- **Tool Calls:** We log the names of the tools that are called, whether they succeed or fail, and how long they take to execute. We do not collect the arguments passed to the tools or any data returned by them.
|
||||||
|
- **API Requests:** We log the model used for each request, the duration of the request, and whether it was successful. We do not collect the content of the prompts or responses.
|
||||||
|
- **Session Information:** We collect information about the configuration of the CLI, such as the enabled tools and the approval mode.
|
||||||
|
|
||||||
|
**What we DON'T collect:**
|
||||||
|
|
||||||
|
- **Personally Identifiable Information (PII):** We do not collect any personal information, such as your name, email address, or API keys.
|
||||||
|
- **Prompt and Response Content:** We do not log the content of your prompts or the responses from the model.
|
||||||
|
- **File Content:** We do not log the content of any files that are read or written by the CLI.
|
||||||
|
|
||||||
|
**How to opt out:**
|
||||||
|
|
||||||
|
You can opt out of usage statistics collection at any time by setting the `usageStatisticsEnabled` property to `false` in your `settings.json` file:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"usageStatisticsEnabled": false
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Note: When usage statistics are enabled, events are sent to an Alibaba Cloud RUM collection endpoint.
|
||||||
|
|
||||||
|
- **`enableWelcomeBack`** (boolean):
|
||||||
|
- **Description:** Show welcome back dialog when returning to a project with conversation history.
|
||||||
|
- **Default:** `true`
|
||||||
|
- **Category:** UI
|
||||||
|
- **Requires Restart:** No
|
||||||
|
- **Example:** `"enableWelcomeBack": false`
|
||||||
|
- **Details:** When enabled, Qwen Code will automatically detect if you're returning to a project with a previously generated project summary (`.qwen/PROJECT_SUMMARY.md`) and show a dialog allowing you to continue your previous conversation or start fresh. This feature integrates with the `/chat summary` command and quit confirmation dialog. See the [Welcome Back documentation](./welcome-back.md) for more details.
|
||||||
@@ -1,5 +1,11 @@
|
|||||||
# Qwen Code Configuration
|
# Qwen Code Configuration
|
||||||
|
|
||||||
|
**Note on New Configuration Format**
|
||||||
|
|
||||||
|
The format of the `settings.json` file has been updated to a new, more organized structure. The old format will be migrated automatically.
|
||||||
|
|
||||||
|
For details on the previous format, please see the [v1 Configuration documentation](./configuration-v1.md).
|
||||||
|
|
||||||
Qwen Code offers several ways to configure its behavior, including environment variables, command-line arguments, and settings files. This document outlines the different configuration methods and available settings.
|
Qwen Code offers several ways to configure its behavior, including environment variables, command-line arguments, and settings files. This document outlines the different configuration methods and available settings.
|
||||||
|
|
||||||
## Configuration layers
|
## Configuration layers
|
||||||
@@ -40,119 +46,265 @@ In addition to a project settings file, a project's `.qwen` directory can contai
|
|||||||
|
|
||||||
- [Custom sandbox profiles](#sandboxing) (e.g., `.qwen/sandbox-macos-custom.sb`, `.qwen/sandbox.Dockerfile`).
|
- [Custom sandbox profiles](#sandboxing) (e.g., `.qwen/sandbox-macos-custom.sb`, `.qwen/sandbox.Dockerfile`).
|
||||||
|
|
||||||
### Available settings in `settings.json`:
|
### Available settings in `settings.json`
|
||||||
|
|
||||||
- **`contextFileName`** (string or array of strings):
|
Settings are organized into categories. All settings should be placed within their corresponding top-level category object in your `settings.json` file.
|
||||||
- **Description:** Specifies the filename for context files (e.g., `QWEN.md`, `AGENTS.md`). Can be a single filename or a list of accepted filenames.
|
|
||||||
- **Default:** `QWEN.md`
|
|
||||||
- **Example:** `"contextFileName": "AGENTS.md"`
|
|
||||||
|
|
||||||
- **`bugCommand`** (object):
|
#### `general`
|
||||||
- **Description:** Overrides the default URL for the `/bug` command.
|
|
||||||
- **Default:** `"urlTemplate": "https://github.com/QwenLM/qwen-code/issues/new?template=bug_report.yml&title={title}&info={info}"`
|
|
||||||
- **Properties:**
|
|
||||||
- **`urlTemplate`** (string): A URL that can contain `{title}` and `{info}` placeholders.
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"bugCommand": {
|
|
||||||
"urlTemplate": "https://bug.example.com/new?title={title}&info={info}"
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`fileFiltering`** (object):
|
- **`general.preferredEditor`** (string):
|
||||||
- **Description:** Controls git-aware file filtering behavior for @ commands and file discovery tools.
|
- **Description:** The preferred editor to open files in.
|
||||||
- **Default:** `"respectGitIgnore": true, "enableRecursiveFileSearch": true`
|
|
||||||
- **Properties:**
|
|
||||||
- **`respectGitIgnore`** (boolean): Whether to respect .gitignore patterns when discovering files. When set to `true`, git-ignored files (like `node_modules/`, `dist/`, `.env`) are automatically excluded from @ commands and file listing operations.
|
|
||||||
- **`enableRecursiveFileSearch`** (boolean): Whether to enable searching recursively for filenames under the current tree when completing @ prefixes in the prompt.
|
|
||||||
- **`disableFuzzySearch`** (boolean): When `true`, disables the fuzzy search capabilities when searching for files, which can improve performance on projects with a large number of files.
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"fileFiltering": {
|
|
||||||
"respectGitIgnore": true,
|
|
||||||
"enableRecursiveFileSearch": false,
|
|
||||||
"disableFuzzySearch": true
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
### Troubleshooting File Search Performance
|
|
||||||
|
|
||||||
If you are experiencing performance issues with file searching (e.g., with `@` completions), especially in projects with a very large number of files, here are a few things you can try in order of recommendation:
|
|
||||||
|
|
||||||
1. **Use `.qwenignore`:** Create a `.qwenignore` file in your project root to exclude directories that contain a large number of files that you don't need to reference (e.g., build artifacts, logs, `node_modules`). Reducing the total number of files crawled is the most effective way to improve performance.
|
|
||||||
|
|
||||||
2. **Disable Fuzzy Search:** If ignoring files is not enough, you can disable fuzzy search by setting `disableFuzzySearch` to `true` in your `settings.json` file. This will use a simpler, non-fuzzy matching algorithm, which can be faster.
|
|
||||||
|
|
||||||
3. **Disable Recursive File Search:** As a last resort, you can disable recursive file search entirely by setting `enableRecursiveFileSearch` to `false`. This will be the fastest option as it avoids a recursive crawl of your project. However, it means you will need to type the full path to files when using `@` completions.
|
|
||||||
|
|
||||||
- **`coreTools`** (array of strings):
|
|
||||||
- **Description:** Allows you to specify a list of core tool names that should be made available to the model. This can be used to restrict the set of built-in tools. See [Built-in Tools](../core/tools-api.md#built-in-tools) for a list of core tools. You can also specify command-specific restrictions for tools that support it, like the `ShellTool`. For example, `"coreTools": ["ShellTool(ls -l)"]` will only allow the `ls -l` command to be executed.
|
|
||||||
- **Default:** All tools available for use by the model.
|
|
||||||
- **Example:** `"coreTools": ["ReadFileTool", "GlobTool", "ShellTool(ls)"]`.
|
|
||||||
|
|
||||||
- **`allowedTools`** (array of strings):
|
|
||||||
- **Default:** `undefined`
|
- **Default:** `undefined`
|
||||||
- **Description:** A list of tool names that will bypass the confirmation dialog. This is useful for tools that you trust and use frequently. The match semantics are the same as `coreTools`.
|
|
||||||
- **Example:** `"allowedTools": ["ShellTool(git status)"]`.
|
|
||||||
|
|
||||||
- **`excludeTools`** (array of strings):
|
- **`general.vimMode`** (boolean):
|
||||||
- **Description:** Allows you to specify a list of core tool names that should be excluded from the model. A tool listed in both `excludeTools` and `coreTools` is excluded. You can also specify command-specific restrictions for tools that support it, like the `ShellTool`. For example, `"excludeTools": ["ShellTool(rm -rf)"]` will block the `rm -rf` command.
|
- **Description:** Enable Vim keybindings.
|
||||||
- **Default**: No tools excluded.
|
|
||||||
- **Example:** `"excludeTools": ["run_shell_command", "findFiles"]`.
|
|
||||||
- **Security Note:** Command-specific restrictions in
|
|
||||||
`excludeTools` for `run_shell_command` are based on simple string matching and can be easily bypassed. This feature is **not a security mechanism** and should not be relied upon to safely execute untrusted code. It is recommended to use `coreTools` to explicitly select commands
|
|
||||||
that can be executed.
|
|
||||||
|
|
||||||
- **`allowMCPServers`** (array of strings):
|
|
||||||
- **Description:** Allows you to specify a list of MCP server names that should be made available to the model. This can be used to restrict the set of MCP servers to connect to. Note that this will be ignored if `--allowed-mcp-server-names` is set.
|
|
||||||
- **Default:** All MCP servers are available for use by the model.
|
|
||||||
- **Example:** `"allowMCPServers": ["myPythonServer"]`.
|
|
||||||
- **Security Note:** This uses simple string matching on MCP server names, which can be modified. If you're a system administrator looking to prevent users from bypassing this, consider configuring the `mcpServers` at the system settings level such that the user will not be able to configure any MCP servers of their own. This should not be used as an airtight security mechanism.
|
|
||||||
|
|
||||||
- **`excludeMCPServers`** (array of strings):
|
|
||||||
- **Description:** Allows you to specify a list of MCP server names that should be excluded from the model. A server listed in both `excludeMCPServers` and `allowMCPServers` is excluded. Note that this will be ignored if `--allowed-mcp-server-names` is set.
|
|
||||||
- **Default**: No MCP servers excluded.
|
|
||||||
- **Example:** `"excludeMCPServers": ["myNodeServer"]`.
|
|
||||||
- **Security Note:** This uses simple string matching on MCP server names, which can be modified. If you're a system administrator looking to prevent users from bypassing this, consider configuring the `mcpServers` at the system settings level such that the user will not be able to configure any MCP servers of their own. This should not be used as an airtight security mechanism.
|
|
||||||
|
|
||||||
- **`autoAccept`** (boolean):
|
|
||||||
- **Description:** Controls whether the CLI automatically accepts and executes tool calls that are considered safe (e.g., read-only operations) without explicit user confirmation. If set to `true`, the CLI will bypass the confirmation prompt for tools deemed safe.
|
|
||||||
- **Default:** `false`
|
- **Default:** `false`
|
||||||
- **Example:** `"autoAccept": true`
|
|
||||||
|
|
||||||
- **`theme`** (string):
|
- **`general.disableAutoUpdate`** (boolean):
|
||||||
- **Description:** Sets the visual [theme](./themes.md) for Qwen Code.
|
- **Description:** Disable automatic updates.
|
||||||
- **Default:** `"Default"`
|
|
||||||
- **Example:** `"theme": "GitHub"`
|
|
||||||
|
|
||||||
- **`vimMode`** (boolean):
|
|
||||||
- **Description:** Enables or disables vim mode for input editing. When enabled, the input area supports vim-style navigation and editing commands with NORMAL and INSERT modes. The vim mode status is displayed in the footer and persists between sessions.
|
|
||||||
- **Default:** `false`
|
- **Default:** `false`
|
||||||
- **Example:** `"vimMode": true`
|
|
||||||
|
|
||||||
- **`sandbox`** (boolean or string):
|
- **`general.disableUpdateNag`** (boolean):
|
||||||
- **Description:** Controls whether and how to use sandboxing for tool execution. If set to `true`, Qwen Code uses a pre-built `qwen-code-sandbox` Docker image. For more information, see [Sandboxing](#sandboxing).
|
- **Description:** Disable update notification prompts.
|
||||||
- **Default:** `false`
|
- **Default:** `false`
|
||||||
- **Example:** `"sandbox": "docker"`
|
|
||||||
|
|
||||||
- **`toolDiscoveryCommand`** (string):
|
- **`general.checkpointing.enabled`** (boolean):
|
||||||
- **Description:** **Align with Gemini CLI.** Defines a custom shell command for discovering tools from your project. The shell command must return on `stdout` a JSON array of [function declarations](https://ai.google.dev/gemini-api/docs/function-calling#function-declarations). Tool wrappers are optional.
|
- **Description:** Enable session checkpointing for recovery.
|
||||||
- **Default:** Empty
|
- **Default:** `false`
|
||||||
- **Example:** `"toolDiscoveryCommand": "bin/get_tools"`
|
|
||||||
|
|
||||||
- **`toolCallCommand`** (string):
|
#### `output`
|
||||||
- **Description:** **Align with Gemini CLI.** Defines a custom shell command for calling a specific tool that was discovered using `toolDiscoveryCommand`. The shell command must meet the following criteria:
|
|
||||||
|
- **`output.format`** (string):
|
||||||
|
- **Description:** The format of the CLI output.
|
||||||
|
- **Default:** `"text"`
|
||||||
|
- **Values:** `"text"`, `"json"`
|
||||||
|
|
||||||
|
#### `ui`
|
||||||
|
|
||||||
|
- **`ui.theme`** (string):
|
||||||
|
- **Description:** The color theme for the UI. See [Themes](./themes.md) for available options.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`ui.customThemes`** (object):
|
||||||
|
- **Description:** Custom theme definitions.
|
||||||
|
- **Default:** `{}`
|
||||||
|
|
||||||
|
- **`ui.hideWindowTitle`** (boolean):
|
||||||
|
- **Description:** Hide the window title bar.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.hideTips`** (boolean):
|
||||||
|
- **Description:** Hide helpful tips in the UI.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.hideBanner`** (boolean):
|
||||||
|
- **Description:** Hide the application banner.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.hideFooter`** (boolean):
|
||||||
|
- **Description:** Hide the footer from the UI.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.showMemoryUsage`** (boolean):
|
||||||
|
- **Description:** Display memory usage information in the UI.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.showLineNumbers`** (boolean):
|
||||||
|
- **Description:** Show line numbers in the chat.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.showCitations`** (boolean):
|
||||||
|
- **Description:** Show citations for generated text in the chat.
|
||||||
|
- **Default:** `true`
|
||||||
|
|
||||||
|
- **`enableWelcomeBack`** (boolean):
|
||||||
|
- **Description:** Show welcome back dialog when returning to a project with conversation history.
|
||||||
|
- **Default:** `true`
|
||||||
|
|
||||||
|
- **`ui.accessibility.disableLoadingPhrases`** (boolean):
|
||||||
|
- **Description:** Disable loading phrases for accessibility.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ui.customWittyPhrases`** (array of strings):
|
||||||
|
- **Description:** A list of custom phrases to display during loading states. When provided, the CLI will cycle through these phrases instead of the default ones.
|
||||||
|
- **Default:** `[]`
|
||||||
|
|
||||||
|
#### `ide`
|
||||||
|
|
||||||
|
- **`ide.enabled`** (boolean):
|
||||||
|
- **Description:** Enable IDE integration mode.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`ide.hasSeenNudge`** (boolean):
|
||||||
|
- **Description:** Whether the user has seen the IDE integration nudge.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
#### `privacy`
|
||||||
|
|
||||||
|
- **`privacy.usageStatisticsEnabled`** (boolean):
|
||||||
|
- **Description:** Enable collection of usage statistics.
|
||||||
|
- **Default:** `true`
|
||||||
|
|
||||||
|
#### `model`
|
||||||
|
|
||||||
|
- **`model.name`** (string):
|
||||||
|
- **Description:** The Qwen model to use for conversations.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`model.maxSessionTurns`** (number):
|
||||||
|
- **Description:** Maximum number of user/model/tool turns to keep in a session. -1 means unlimited.
|
||||||
|
- **Default:** `-1`
|
||||||
|
|
||||||
|
- **`model.summarizeToolOutput`** (object):
|
||||||
|
- **Description:** Enables or disables the summarization of tool output. You can specify the token budget for the summarization using the `tokenBudget` setting. Note: Currently only the `run_shell_command` tool is supported. For example `{"run_shell_command": {"tokenBudget": 2000}}`
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`model.chatCompression.contextPercentageThreshold`** (number):
|
||||||
|
- **Description:** Sets the threshold for chat history compression as a percentage of the model's total token limit. This is a value between 0 and 1 that applies to both automatic compression and the manual `/compress` command. For example, a value of `0.6` will trigger compression when the chat history exceeds 60% of the token limit.
|
||||||
|
- **Default:** `0.7`
|
||||||
|
|
||||||
|
- **`model.skipNextSpeakerCheck`** (boolean):
|
||||||
|
- **Description:** Skip the next speaker check.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`model.skipLoopDetection`**(boolean):
|
||||||
|
- **Description:** Disables loop detection checks. Loop detection prevents infinite loops in AI responses but can generate false positives that interrupt legitimate workflows. Enable this option if you experience frequent false positive loop detection interruptions.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
#### `context`
|
||||||
|
|
||||||
|
- **`context.fileName`** (string or array of strings):
|
||||||
|
- **Description:** The name of the context file(s).
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`context.importFormat`** (string):
|
||||||
|
- **Description:** The format to use when importing memory.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`context.discoveryMaxDirs`** (number):
|
||||||
|
- **Description:** Maximum number of directories to search for memory.
|
||||||
|
- **Default:** `200`
|
||||||
|
|
||||||
|
- **`context.includeDirectories`** (array):
|
||||||
|
- **Description:** Additional directories to include in the workspace context. Missing directories will be skipped with a warning.
|
||||||
|
- **Default:** `[]`
|
||||||
|
|
||||||
|
- **`context.loadFromIncludeDirectories`** (boolean):
|
||||||
|
- **Description:** Controls the behavior of the `/memory refresh` command. If set to `true`, `QWEN.md` files should be loaded from all directories that are added. If set to `false`, `QWEN.md` should only be loaded from the current directory.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`context.fileFiltering.respectGitIgnore`** (boolean):
|
||||||
|
- **Description:** Respect .gitignore files when searching.
|
||||||
|
- **Default:** `true`
|
||||||
|
|
||||||
|
- **`context.fileFiltering.respectQwenIgnore`** (boolean):
|
||||||
|
- **Description:** Respect .qwenignore files when searching.
|
||||||
|
- **Default:** `true`
|
||||||
|
|
||||||
|
- **`context.fileFiltering.enableRecursiveFileSearch`** (boolean):
|
||||||
|
- **Description:** Whether to enable searching recursively for filenames under the current tree when completing `@` prefixes in the prompt.
|
||||||
|
- **Default:** `true`
|
||||||
|
|
||||||
|
#### `tools`
|
||||||
|
|
||||||
|
- **`tools.sandbox`** (boolean or string):
|
||||||
|
- **Description:** Sandbox execution environment (can be a boolean or a path string).
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`tools.shell.enableInteractiveShell`** (boolean):
|
||||||
|
|
||||||
|
Use `node-pty` for an interactive shell experience. Fallback to `child_process` still applies. Defaults to `false`.
|
||||||
|
|
||||||
|
- **`tools.core`** (array of strings):
|
||||||
|
- **Description:** This can be used to restrict the set of built-in tools [with an allowlist](./enterprise.md#restricting-tool-access). See [Built-in Tools](../core/tools-api.md#built-in-tools) for a list of core tools. The match semantics are the same as `tools.allowed`.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`tools.exclude`** (array of strings):
|
||||||
|
- **Description:** Tool names to exclude from discovery.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`tools.allowed`** (array of strings):
|
||||||
|
- **Description:** A list of tool names that will bypass the confirmation dialog. This is useful for tools that you trust and use frequently. For example, `["run_shell_command(git)", "run_shell_command(npm test)"]` will skip the confirmation dialog to run any `git` and `npm test` commands. See [Shell Tool command restrictions](../tools/shell.md#command-restrictions) for details on prefix matching, command chaining, etc.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`tools.approvalMode`** (string):
|
||||||
|
- **Description:** Sets the default approval mode for tool usage. Accepted values are:
|
||||||
|
- `plan`: Analyze only, do not modify files or execute commands.
|
||||||
|
- `default`: Require approval before file edits or shell commands run.
|
||||||
|
- `auto-edit`: Automatically approve file edits.
|
||||||
|
- `yolo`: Automatically approve all tool calls.
|
||||||
|
- **Default:** `default`
|
||||||
|
|
||||||
|
- **`tools.discoveryCommand`** (string):
|
||||||
|
- **Description:** Command to run for tool discovery.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`tools.callCommand`** (string):
|
||||||
|
- **Description:** Defines a custom shell command for calling a specific tool that was discovered using `tools.discoveryCommand`. The shell command must meet the following criteria:
|
||||||
- It must take function `name` (exactly as in [function declaration](https://ai.google.dev/gemini-api/docs/function-calling#function-declarations)) as first command line argument.
|
- It must take function `name` (exactly as in [function declaration](https://ai.google.dev/gemini-api/docs/function-calling#function-declarations)) as first command line argument.
|
||||||
- It must read function arguments as JSON on `stdin`, analogous to [`functionCall.args`](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/inference#functioncall).
|
- It must read function arguments as JSON on `stdin`, analogous to [`functionCall.args`](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/inference#functioncall).
|
||||||
- It must return function output as JSON on `stdout`, analogous to [`functionResponse.response.content`](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/inference#functionresponse).
|
- It must return function output as JSON on `stdout`, analogous to [`functionResponse.response.content`](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/inference#functionresponse).
|
||||||
- **Default:** Empty
|
- **Default:** `undefined`
|
||||||
- **Example:** `"toolCallCommand": "bin/call_tool"`
|
|
||||||
|
|
||||||
- **`mcpServers`** (object):
|
#### `mcp`
|
||||||
- **Description:** Configures connections to one or more Model-Context Protocol (MCP) servers for discovering and using custom tools. Qwen Code attempts to connect to each configured MCP server to discover available tools. If multiple MCP servers expose a tool with the same name, the tool names will be prefixed with the server alias you defined in the configuration (e.g., `serverAlias__actualToolName`) to avoid conflicts. Note that the system might strip certain schema properties from MCP tool definitions for compatibility. At least one of `command`, `url`, or `httpUrl` must be provided. If multiple are specified, the order of precedence is `httpUrl`, then `url`, then `command`.
|
|
||||||
- **Default:** Empty
|
- **`mcp.serverCommand`** (string):
|
||||||
- **Properties:**
|
- **Description:** Command to start an MCP server.
|
||||||
- **`<SERVER_NAME>`** (object): The server parameters for the named server.
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`mcp.allowed`** (array of strings):
|
||||||
|
- **Description:** An allowlist of MCP servers to allow.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`mcp.excluded`** (array of strings):
|
||||||
|
- **Description:** A denylist of MCP servers to exclude.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
#### `security`
|
||||||
|
|
||||||
|
- **`security.folderTrust.enabled`** (boolean):
|
||||||
|
- **Description:** Setting to track whether Folder trust is enabled.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`security.auth.selectedType`** (string):
|
||||||
|
- **Description:** The currently selected authentication type.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`security.auth.enforcedType`** (string):
|
||||||
|
- **Description:** The required auth type (useful for enterprises).
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`security.auth.useExternal`** (boolean):
|
||||||
|
- **Description:** Whether to use an external authentication flow.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
#### `advanced`
|
||||||
|
|
||||||
|
- **`advanced.autoConfigureMemory`** (boolean):
|
||||||
|
- **Description:** Automatically configure Node.js memory limits.
|
||||||
|
- **Default:** `false`
|
||||||
|
|
||||||
|
- **`advanced.dnsResolutionOrder`** (string):
|
||||||
|
- **Description:** The DNS resolution order.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`advanced.excludedEnvVars`** (array of strings):
|
||||||
|
- **Description:** Environment variables to exclude from project context.
|
||||||
|
- **Default:** `["DEBUG","DEBUG_MODE"]`
|
||||||
|
|
||||||
|
- **`advanced.bugCommand`** (object):
|
||||||
|
- **Description:** Configuration for the bug report command.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
- **`advanced.tavilyApiKey`** (string):
|
||||||
|
- **Description:** API key for Tavily web search service. Required to enable the `web_search` tool functionality. If not configured, the web search tool will be disabled and skipped.
|
||||||
|
- **Default:** `undefined`
|
||||||
|
|
||||||
|
#### `mcpServers`
|
||||||
|
|
||||||
|
Configures connections to one or more Model-Context Protocol (MCP) servers for discovering and using custom tools. Qwen Code attempts to connect to each configured MCP server to discover available tools. If multiple MCP servers expose a tool with the same name, the tool names will be prefixed with the server alias you defined in the configuration (e.g., `serverAlias__actualToolName`) to avoid conflicts. Note that the system might strip certain schema properties from MCP tool definitions for compatibility. At least one of `command`, `url`, or `httpUrl` must be provided. If multiple are specified, the order of precedence is `httpUrl`, then `url`, then `command`.
|
||||||
|
|
||||||
|
- **`mcpServers.<SERVER_NAME>`** (object): The server parameters for the named server.
|
||||||
- `command` (string, optional): The command to execute to start the MCP server via standard I/O.
|
- `command` (string, optional): The command to execute to start the MCP server via standard I/O.
|
||||||
- `args` (array of strings, optional): Arguments to pass to the command.
|
- `args` (array of strings, optional): Arguments to pass to the command.
|
||||||
- `env` (object, optional): Environment variables to set for the server process.
|
- `env` (object, optional): Environment variables to set for the server process.
|
||||||
@@ -163,226 +315,48 @@ If you are experiencing performance issues with file searching (e.g., with `@` c
|
|||||||
- `timeout` (number, optional): Timeout in milliseconds for requests to this MCP server.
|
- `timeout` (number, optional): Timeout in milliseconds for requests to this MCP server.
|
||||||
- `trust` (boolean, optional): Trust this server and bypass all tool call confirmations.
|
- `trust` (boolean, optional): Trust this server and bypass all tool call confirmations.
|
||||||
- `description` (string, optional): A brief description of the server, which may be used for display purposes.
|
- `description` (string, optional): A brief description of the server, which may be used for display purposes.
|
||||||
- `includeTools` (array of strings, optional): List of tool names to include from this MCP server. When specified, only the tools listed here will be available from this server (whitelist behavior). If not specified, all tools from the server are enabled by default.
|
- `includeTools` (array of strings, optional): List of tool names to include from this MCP server. When specified, only the tools listed here will be available from this server (allowlist behavior). If not specified, all tools from the server are enabled by default.
|
||||||
- `excludeTools` (array of strings, optional): List of tool names to exclude from this MCP server. Tools listed here will not be available to the model, even if they are exposed by the server. **Note:** `excludeTools` takes precedence over `includeTools` - if a tool is in both lists, it will be excluded.
|
- `excludeTools` (array of strings, optional): List of tool names to exclude from this MCP server. Tools listed here will not be available to the model, even if they are exposed by the server. **Note:** `excludeTools` takes precedence over `includeTools` - if a tool is in both lists, it will be excluded.
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"mcpServers": {
|
|
||||||
"myPythonServer": {
|
|
||||||
"command": "python",
|
|
||||||
"args": ["mcp_server.py", "--port", "8080"],
|
|
||||||
"cwd": "./mcp_tools/python",
|
|
||||||
"timeout": 5000,
|
|
||||||
"includeTools": ["safe_tool", "file_reader"],
|
|
||||||
},
|
|
||||||
"myNodeServer": {
|
|
||||||
"command": "node",
|
|
||||||
"args": ["mcp_server.js"],
|
|
||||||
"cwd": "./mcp_tools/node",
|
|
||||||
"excludeTools": ["dangerous_tool", "file_deleter"]
|
|
||||||
},
|
|
||||||
"myDockerServer": {
|
|
||||||
"command": "docker",
|
|
||||||
"args": ["run", "-i", "--rm", "-e", "API_KEY", "ghcr.io/foo/bar"],
|
|
||||||
"env": {
|
|
||||||
"API_KEY": "$MY_API_TOKEN"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"mySseServer": {
|
|
||||||
"url": "http://localhost:8081/events",
|
|
||||||
"headers": {
|
|
||||||
"Authorization": "Bearer $MY_SSE_TOKEN"
|
|
||||||
},
|
|
||||||
"description": "An example SSE-based MCP server."
|
|
||||||
},
|
|
||||||
"myStreamableHttpServer": {
|
|
||||||
"httpUrl": "http://localhost:8082/stream",
|
|
||||||
"headers": {
|
|
||||||
"X-API-Key": "$MY_HTTP_API_KEY"
|
|
||||||
},
|
|
||||||
"description": "An example Streamable HTTP-based MCP server."
|
|
||||||
}
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`checkpointing`** (object):
|
#### `telemetry`
|
||||||
- **Description:** Configures the checkpointing feature, which allows you to save and restore conversation and file states. See the [Checkpointing documentation](../checkpointing.md) for more details.
|
|
||||||
- **Default:** `{"enabled": false}`
|
|
||||||
- **Properties:**
|
|
||||||
- **`enabled`** (boolean): When `true`, the `/restore` command is available.
|
|
||||||
|
|
||||||
- **`preferredEditor`** (string):
|
Configures logging and metrics collection for Qwen Code. For more information, see [Telemetry](../telemetry.md).
|
||||||
- **Description:** Specifies the preferred editor to use for viewing diffs.
|
|
||||||
- **Default:** `vscode`
|
|
||||||
- **Example:** `"preferredEditor": "vscode"`
|
|
||||||
|
|
||||||
- **`telemetry`** (object)
|
|
||||||
- **Description:** Configures logging and metrics collection for Qwen Code. For more information, see [Telemetry](../telemetry.md).
|
|
||||||
- **Default:** `{"enabled": false, "target": "local", "otlpEndpoint": "http://localhost:4317", "logPrompts": true}`
|
|
||||||
- **Properties:**
|
- **Properties:**
|
||||||
- **`enabled`** (boolean): Whether or not telemetry is enabled.
|
- **`enabled`** (boolean): Whether or not telemetry is enabled.
|
||||||
- **`target`** (string): The destination for collected telemetry. Supported values are `local` and `gcp`.
|
- **`target`** (string): The destination for collected telemetry. Supported values are `local` and `gcp`.
|
||||||
- **`otlpEndpoint`** (string): The endpoint for the OTLP Exporter.
|
- **`otlpEndpoint`** (string): The endpoint for the OTLP Exporter.
|
||||||
|
- **`otlpProtocol`** (string): The protocol for the OTLP Exporter (`grpc` or `http`).
|
||||||
- **`logPrompts`** (boolean): Whether or not to include the content of user prompts in the logs.
|
- **`logPrompts`** (boolean): Whether or not to include the content of user prompts in the logs.
|
||||||
- **Example:**
|
- **`outfile`** (string): The file to write telemetry to when `target` is `local`.
|
||||||
```json
|
- **`useCollector`** (boolean): Whether to use an external OTLP collector.
|
||||||
"telemetry": {
|
|
||||||
"enabled": true,
|
|
||||||
"target": "local",
|
|
||||||
"otlpEndpoint": "http://localhost:16686",
|
|
||||||
"logPrompts": false
|
|
||||||
}
|
|
||||||
```
|
|
||||||
- **`usageStatisticsEnabled`** (boolean):
|
|
||||||
- **Description:** Enables or disables the collection of usage statistics. See [Usage Statistics](#usage-statistics) for more information.
|
|
||||||
- **Default:** `true`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"usageStatisticsEnabled": false
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`hideTips`** (boolean):
|
### Example `settings.json`
|
||||||
- **Description:** Enables or disables helpful tips in the CLI interface.
|
|
||||||
- **Default:** `false`
|
|
||||||
- **Example:**
|
|
||||||
|
|
||||||
```json
|
Here is an example of a `settings.json` file with the nested structure, new as of v0.3.0:
|
||||||
"hideTips": true
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`hideBanner`** (boolean):
|
|
||||||
- **Description:** Enables or disables the startup banner (ASCII art logo) in the CLI interface.
|
|
||||||
- **Default:** `false`
|
|
||||||
- **Example:**
|
|
||||||
|
|
||||||
```json
|
|
||||||
"hideBanner": true
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`maxSessionTurns`** (number):
|
|
||||||
- **Description:** Sets the maximum number of turns for a session. If the session exceeds this limit, the CLI will stop processing and start a new chat.
|
|
||||||
- **Default:** `-1` (unlimited)
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"maxSessionTurns": 10
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`summarizeToolOutput`** (object):
|
|
||||||
- **Description:** Enables or disables the summarization of tool output. You can specify the token budget for the summarization using the `tokenBudget` setting.
|
|
||||||
- Note: Currently only the `run_shell_command` tool is supported.
|
|
||||||
- **Default:** `{}` (Disabled by default)
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"summarizeToolOutput": {
|
|
||||||
"run_shell_command": {
|
|
||||||
"tokenBudget": 2000
|
|
||||||
}
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`excludedProjectEnvVars`** (array of strings):
|
|
||||||
- **Description:** Specifies environment variables that should be excluded from being loaded from project `.env` files. This prevents project-specific environment variables (like `DEBUG=true`) from interfering with the CLI behavior. Variables from `.qwen/.env` files are never excluded.
|
|
||||||
- **Default:** `["DEBUG", "DEBUG_MODE"]`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"excludedProjectEnvVars": ["DEBUG", "DEBUG_MODE", "NODE_ENV"]
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`includeDirectories`** (array of strings):
|
|
||||||
- **Description:** Specifies an array of additional absolute or relative paths to include in the workspace context. Missing directories will be skipped with a warning by default. Paths can use `~` to refer to the user's home directory. This setting can be combined with the `--include-directories` command-line flag.
|
|
||||||
- **Default:** `[]`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"includeDirectories": [
|
|
||||||
"/path/to/another/project",
|
|
||||||
"../shared-library",
|
|
||||||
"~/common-utils"
|
|
||||||
]
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`loadMemoryFromIncludeDirectories`** (boolean):
|
|
||||||
- **Description:** Controls the behavior of the `/memory refresh` command. If set to `true`, `QWEN.md` files should be loaded from all directories that are added. If set to `false`, `QWEN.md` should only be loaded from the current directory.
|
|
||||||
- **Default:** `false`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"loadMemoryFromIncludeDirectories": true
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`tavilyApiKey`** (string):
|
|
||||||
- **Description:** API key for Tavily web search service. Required to enable the `web_search` tool functionality. If not configured, the web search tool will be disabled and skipped.
|
|
||||||
- **Default:** `undefined` (web search disabled)
|
|
||||||
- **Example:** `"tavilyApiKey": "tvly-your-api-key-here"`
|
|
||||||
- **`chatCompression`** (object):
|
|
||||||
- **Description:** Controls the settings for chat history compression, both automatic and
|
|
||||||
when manually invoked through the /compress command.
|
|
||||||
- **Properties:**
|
|
||||||
- **`contextPercentageThreshold`** (number): A value between 0 and 1 that specifies the token threshold for compression as a percentage of the model's total token limit. For example, a value of `0.6` will trigger compression when the chat history exceeds 60% of the token limit.
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"chatCompression": {
|
|
||||||
"contextPercentageThreshold": 0.6
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`showLineNumbers`** (boolean):
|
|
||||||
- **Description:** Controls whether line numbers are displayed in code blocks in the CLI output.
|
|
||||||
- **Default:** `true`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"showLineNumbers": false
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`accessibility`** (object):
|
|
||||||
- **Description:** Configures accessibility features for the CLI.
|
|
||||||
- **Properties:**
|
|
||||||
- **`screenReader`** (boolean): Enables screen reader mode, which adjusts the TUI for better compatibility with screen readers. This can also be enabled with the `--screen-reader` command-line flag, which will take precedence over the setting.
|
|
||||||
- **`disableLoadingPhrases`** (boolean): Disables the display of loading phrases during operations.
|
|
||||||
- **Default:** `{"screenReader": false, "disableLoadingPhrases": false}`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"accessibility": {
|
|
||||||
"screenReader": true,
|
|
||||||
"disableLoadingPhrases": true
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`skipNextSpeakerCheck`** (boolean):
|
|
||||||
- **Description:** Skips the next speaker check after text responses. When enabled, the system bypasses analyzing whether the AI should continue speaking.
|
|
||||||
- **Default:** `false`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"skipNextSpeakerCheck": true
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`skipLoopDetection`** (boolean):
|
|
||||||
- **Description:** Disables all loop detection checks (streaming and LLM-based). Loop detection prevents infinite loops in AI responses but can generate false positives that interrupt legitimate workflows. Enable this option if you experience frequent false positive loop detection interruptions.
|
|
||||||
- **Default:** `false`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"skipLoopDetection": true
|
|
||||||
```
|
|
||||||
|
|
||||||
- **`approvalMode`** (string):
|
|
||||||
- **Description:** Sets the default approval mode for tool usage. Accepted values are:
|
|
||||||
- `plan`: Analyze only, do not modify files or execute commands.
|
|
||||||
- `default`: Require approval before file edits or shell commands run.
|
|
||||||
- `auto-edit`: Automatically approve file edits.
|
|
||||||
- `yolo`: Automatically approve all tool calls.
|
|
||||||
- **Default:** `"default"`
|
|
||||||
- **Example:**
|
|
||||||
```json
|
|
||||||
"approvalMode": "plan"
|
|
||||||
```
|
|
||||||
|
|
||||||
### Example `settings.json`:
|
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
|
"general": {
|
||||||
|
"vimMode": true,
|
||||||
|
"preferredEditor": "code"
|
||||||
|
},
|
||||||
|
"ui": {
|
||||||
"theme": "GitHub",
|
"theme": "GitHub",
|
||||||
|
"hideBanner": true,
|
||||||
|
"hideTips": false,
|
||||||
|
"customWittyPhrases": [
|
||||||
|
"You forget a thousand things every day. Make sure this is one of ’em",
|
||||||
|
"Connecting to AGI"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"tools": {
|
||||||
|
"approvalMode": "yolo",
|
||||||
"sandbox": "docker",
|
"sandbox": "docker",
|
||||||
"toolDiscoveryCommand": "bin/get_tools",
|
"discoveryCommand": "bin/get_tools",
|
||||||
"toolCallCommand": "bin/call_tool",
|
"callCommand": "bin/call_tool",
|
||||||
"tavilyApiKey": "$TAVILY_API_KEY",
|
"exclude": ["write_file"]
|
||||||
|
},
|
||||||
"mcpServers": {
|
"mcpServers": {
|
||||||
"mainServer": {
|
"mainServer": {
|
||||||
"command": "bin/mcp_server.py"
|
"command": "bin/mcp_server.py"
|
||||||
@@ -398,20 +372,29 @@ If you are experiencing performance issues with file searching (e.g., with `@` c
|
|||||||
"otlpEndpoint": "http://localhost:4317",
|
"otlpEndpoint": "http://localhost:4317",
|
||||||
"logPrompts": true
|
"logPrompts": true
|
||||||
},
|
},
|
||||||
"usageStatisticsEnabled": true,
|
"privacy": {
|
||||||
"hideTips": false,
|
"usageStatisticsEnabled": true
|
||||||
"hideBanner": false,
|
},
|
||||||
"skipNextSpeakerCheck": false,
|
"model": {
|
||||||
"skipLoopDetection": false,
|
"name": "qwen3-coder-plus",
|
||||||
"maxSessionTurns": 10,
|
"maxSessionTurns": 10,
|
||||||
"summarizeToolOutput": {
|
"summarizeToolOutput": {
|
||||||
"run_shell_command": {
|
"run_shell_command": {
|
||||||
"tokenBudget": 100
|
"tokenBudget": 100
|
||||||
}
|
}
|
||||||
|
}
|
||||||
},
|
},
|
||||||
"excludedProjectEnvVars": ["DEBUG", "DEBUG_MODE", "NODE_ENV"],
|
"context": {
|
||||||
|
"fileName": ["CONTEXT.md", "QWEN.md"],
|
||||||
"includeDirectories": ["path/to/dir1", "~/path/to/dir2", "../path/to/dir3"],
|
"includeDirectories": ["path/to/dir1", "~/path/to/dir2", "../path/to/dir3"],
|
||||||
"loadMemoryFromIncludeDirectories": true
|
"loadFromIncludeDirectories": true,
|
||||||
|
"fileFiltering": {
|
||||||
|
"respectGitIgnore": false
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"advanced": {
|
||||||
|
"excludedEnvVars": ["DEBUG", "DEBUG_MODE", "NODE_ENV"]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -433,7 +416,7 @@ The CLI automatically loads environment variables from an `.env` file. The loadi
|
|||||||
2. If not found, it searches upwards in parent directories until it finds an `.env` file or reaches the project root (identified by a `.git` folder) or the home directory.
|
2. If not found, it searches upwards in parent directories until it finds an `.env` file or reaches the project root (identified by a `.git` folder) or the home directory.
|
||||||
3. If still not found, it looks for `~/.env` (in the user's home directory).
|
3. If still not found, it looks for `~/.env` (in the user's home directory).
|
||||||
|
|
||||||
**Environment Variable Exclusion:** Some environment variables (like `DEBUG` and `DEBUG_MODE`) are automatically excluded from project `.env` files by default to prevent interference with the CLI behavior. Variables from `.qwen/.env` files are never excluded. You can customize this behavior using the `excludedProjectEnvVars` setting in your `settings.json` file.
|
**Environment Variable Exclusion:** Some environment variables (like `DEBUG` and `DEBUG_MODE`) are automatically excluded from project `.env` files by default to prevent interference with the CLI behavior. Variables from `.qwen/.env` files are never excluded. You can customize this behavior using the `advanced.excludedEnvVars` setting in your `settings.json` file.
|
||||||
|
|
||||||
- **`OPENAI_API_KEY`**:
|
- **`OPENAI_API_KEY`**:
|
||||||
- One of several available [authentication methods](./authentication.md).
|
- One of several available [authentication methods](./authentication.md).
|
||||||
@@ -445,6 +428,27 @@ The CLI automatically loads environment variables from an `.env` file. The loadi
|
|||||||
- Specifies the default OPENAI model to use.
|
- Specifies the default OPENAI model to use.
|
||||||
- Overrides the hardcoded default
|
- Overrides the hardcoded default
|
||||||
- Example: `export OPENAI_MODEL="qwen3-coder-plus"`
|
- Example: `export OPENAI_MODEL="qwen3-coder-plus"`
|
||||||
|
- **`GEMINI_TELEMETRY_ENABLED`**:
|
||||||
|
- Set to `true` or `1` to enable telemetry. Any other value is treated as disabling it.
|
||||||
|
- Overrides the `telemetry.enabled` setting.
|
||||||
|
- **`GEMINI_TELEMETRY_TARGET`**:
|
||||||
|
- Sets the telemetry target (`local` or `gcp`).
|
||||||
|
- Overrides the `telemetry.target` setting.
|
||||||
|
- **`GEMINI_TELEMETRY_OTLP_ENDPOINT`**:
|
||||||
|
- Sets the OTLP endpoint for telemetry.
|
||||||
|
- Overrides the `telemetry.otlpEndpoint` setting.
|
||||||
|
- **`GEMINI_TELEMETRY_OTLP_PROTOCOL`**:
|
||||||
|
- Sets the OTLP protocol (`grpc` or `http`).
|
||||||
|
- Overrides the `telemetry.otlpProtocol` setting.
|
||||||
|
- **`GEMINI_TELEMETRY_LOG_PROMPTS`**:
|
||||||
|
- Set to `true` or `1` to enable or disable logging of user prompts. Any other value is treated as disabling it.
|
||||||
|
- Overrides the `telemetry.logPrompts` setting.
|
||||||
|
- **`GEMINI_TELEMETRY_OUTFILE`**:
|
||||||
|
- Sets the file path to write telemetry to when the target is `local`.
|
||||||
|
- Overrides the `telemetry.outfile` setting.
|
||||||
|
- **`GEMINI_TELEMETRY_USE_COLLECTOR`**:
|
||||||
|
- Set to `true` or `1` to enable or disable using an external OTLP collector. Any other value is treated as disabling it.
|
||||||
|
- Overrides the `telemetry.useCollector` setting.
|
||||||
- **`GEMINI_SANDBOX`**:
|
- **`GEMINI_SANDBOX`**:
|
||||||
- Alternative to the `sandbox` setting in `settings.json`.
|
- Alternative to the `sandbox` setting in `settings.json`.
|
||||||
- Accepts `true`, `false`, `docker`, `podman`, or a custom command string.
|
- Accepts `true`, `false`, `docker`, `podman`, or a custom command string.
|
||||||
@@ -460,9 +464,6 @@ The CLI automatically loads environment variables from an `.env` file. The loadi
|
|||||||
- Set to any value to disable all color output in the CLI.
|
- Set to any value to disable all color output in the CLI.
|
||||||
- **`CLI_TITLE`**:
|
- **`CLI_TITLE`**:
|
||||||
- Set to a string to customize the title of the CLI.
|
- Set to a string to customize the title of the CLI.
|
||||||
- **`CODE_ASSIST_ENDPOINT`**:
|
|
||||||
- Specifies the endpoint for the code assist server.
|
|
||||||
- This is useful for development and testing.
|
|
||||||
- **`TAVILY_API_KEY`**:
|
- **`TAVILY_API_KEY`**:
|
||||||
- Your API key for the Tavily web search service.
|
- Your API key for the Tavily web search service.
|
||||||
- Required to enable the `web_search` tool functionality.
|
- Required to enable the `web_search` tool functionality.
|
||||||
@@ -478,11 +479,18 @@ Arguments passed directly when running the CLI can override other configurations
|
|||||||
- Example: `npm start -- --model qwen3-coder-plus`
|
- Example: `npm start -- --model qwen3-coder-plus`
|
||||||
- **`--prompt <your_prompt>`** (**`-p <your_prompt>`**):
|
- **`--prompt <your_prompt>`** (**`-p <your_prompt>`**):
|
||||||
- Used to pass a prompt directly to the command. This invokes Qwen Code in a non-interactive mode.
|
- Used to pass a prompt directly to the command. This invokes Qwen Code in a non-interactive mode.
|
||||||
|
- For scripting examples, use the `--output-format json` flag to get structured output.
|
||||||
- **`--prompt-interactive <your_prompt>`** (**`-i <your_prompt>`**):
|
- **`--prompt-interactive <your_prompt>`** (**`-i <your_prompt>`**):
|
||||||
- Starts an interactive session with the provided prompt as the initial input.
|
- Starts an interactive session with the provided prompt as the initial input.
|
||||||
- The prompt is processed within the interactive session, not before it.
|
- The prompt is processed within the interactive session, not before it.
|
||||||
- Cannot be used when piping input from stdin.
|
- Cannot be used when piping input from stdin.
|
||||||
- Example: `qwen -i "explain this code"`
|
- Example: `qwen -i "explain this code"`
|
||||||
|
- **`--output-format <format>`**:
|
||||||
|
- **Description:** Specifies the format of the CLI output for non-interactive mode.
|
||||||
|
- **Values:**
|
||||||
|
- `text`: (Default) The standard human-readable output.
|
||||||
|
- `json`: A machine-readable JSON output.
|
||||||
|
- **Note:** For structured output and scripting, use the `--output-format json` flag.
|
||||||
- **`--sandbox`** (**`-s`**):
|
- **`--sandbox`** (**`-s`**):
|
||||||
- Enables sandbox mode for this session.
|
- Enables sandbox mode for this session.
|
||||||
- **`--sandbox-image`**:
|
- **`--sandbox-image`**:
|
||||||
@@ -535,7 +543,7 @@ Arguments passed directly when running the CLI can override other configurations
|
|||||||
- 5 directories can be added at maximum.
|
- 5 directories can be added at maximum.
|
||||||
- Example: `--include-directories /path/to/project1,/path/to/project2` or `--include-directories /path/to/project1 --include-directories /path/to/project2`
|
- Example: `--include-directories /path/to/project1,/path/to/project2` or `--include-directories /path/to/project1 --include-directories /path/to/project2`
|
||||||
- **`--screen-reader`**:
|
- **`--screen-reader`**:
|
||||||
- Enables screen reader mode for accessibility.
|
- Enables screen reader mode, which adjusts the TUI for better compatibility with screen readers.
|
||||||
- **`--version`**:
|
- **`--version`**:
|
||||||
- Displays the version of the CLI.
|
- Displays the version of the CLI.
|
||||||
- **`--openai-logging`**:
|
- **`--openai-logging`**:
|
||||||
@@ -546,7 +554,7 @@ Arguments passed directly when running the CLI can override other configurations
|
|||||||
|
|
||||||
## Context Files (Hierarchical Instructional Context)
|
## Context Files (Hierarchical Instructional Context)
|
||||||
|
|
||||||
While not strictly configuration for the CLI's _behavior_, context files (defaulting to `QWEN.md` but configurable via the `contextFileName` setting) are crucial for configuring the _instructional context_ (also referred to as "memory"). This powerful feature allows you to give project-specific instructions, coding style guides, or any relevant background information to the AI, making its responses more tailored and accurate to your needs. The CLI includes UI elements, such as an indicator in the footer showing the number of loaded context files, to keep you informed about the active context.
|
While not strictly configuration for the CLI's _behavior_, context files (defaulting to `QWEN.md` but configurable via the `context.fileName` setting) are crucial for configuring the _instructional context_ (also referred to as "memory"). This powerful feature allows you to give project-specific instructions, coding style guides, or any relevant background information to the AI, making its responses more tailored and accurate to your needs. The CLI includes UI elements, such as an indicator in the footer showing the number of loaded context files, to keep you informed about the active context.
|
||||||
|
|
||||||
- **Purpose:** These Markdown files contain instructions, guidelines, or context that you want the Qwen model to be aware of during your interactions. The system is designed to manage this instructional context hierarchically.
|
- **Purpose:** These Markdown files contain instructions, guidelines, or context that you want the Qwen model to be aware of during your interactions. The system is designed to manage this instructional context hierarchically.
|
||||||
|
|
||||||
@@ -587,13 +595,13 @@ This example demonstrates how you can provide general project context, specific
|
|||||||
|
|
||||||
- **Hierarchical Loading and Precedence:** The CLI implements a sophisticated hierarchical memory system by loading context files (e.g., `QWEN.md`) from several locations. Content from files lower in this list (more specific) typically overrides or supplements content from files higher up (more general). The exact concatenation order and final context can be inspected using the `/memory show` command. The typical loading order is:
|
- **Hierarchical Loading and Precedence:** The CLI implements a sophisticated hierarchical memory system by loading context files (e.g., `QWEN.md`) from several locations. Content from files lower in this list (more specific) typically overrides or supplements content from files higher up (more general). The exact concatenation order and final context can be inspected using the `/memory show` command. The typical loading order is:
|
||||||
1. **Global Context File:**
|
1. **Global Context File:**
|
||||||
- Location: `~/.qwen/<contextFileName>` (e.g., `~/.qwen/QWEN.md` in your user home directory).
|
- Location: `~/.qwen/<configured-context-filename>` (e.g., `~/.qwen/QWEN.md` in your user home directory).
|
||||||
- Scope: Provides default instructions for all your projects.
|
- Scope: Provides default instructions for all your projects.
|
||||||
2. **Project Root & Ancestors Context Files:**
|
2. **Project Root & Ancestors Context Files:**
|
||||||
- Location: The CLI searches for the configured context file in the current working directory and then in each parent directory up to either the project root (identified by a `.git` folder) or your home directory.
|
- Location: The CLI searches for the configured context file in the current working directory and then in each parent directory up to either the project root (identified by a `.git` folder) or your home directory.
|
||||||
- Scope: Provides context relevant to the entire project or a significant portion of it.
|
- Scope: Provides context relevant to the entire project or a significant portion of it.
|
||||||
3. **Sub-directory Context Files (Contextual/Local):**
|
3. **Sub-directory Context Files (Contextual/Local):**
|
||||||
- Location: The CLI also scans for the configured context file in subdirectories _below_ the current working directory (respecting common ignore patterns like `node_modules`, `.git`, etc.). The breadth of this search is limited to 200 directories by default, but can be configured with a `memoryDiscoveryMaxDirs` field in your `settings.json` file.
|
- Location: The CLI also scans for the configured context file in subdirectories _below_ the current working directory (respecting common ignore patterns like `node_modules`, `.git`, etc.). The breadth of this search is limited to 200 directories by default, but can be configured with the `context.discoveryMaxDirs` setting in your `settings.json` file.
|
||||||
- Scope: Allows for highly specific instructions relevant to a particular component, module, or subsection of your project.
|
- Scope: Allows for highly specific instructions relevant to a particular component, module, or subsection of your project.
|
||||||
- **Concatenation & UI Indication:** The contents of all found context files are concatenated (with separators indicating their origin and path) and provided as part of the system prompt. The CLI footer displays the count of loaded context files, giving you a quick visual cue about the active instructional context.
|
- **Concatenation & UI Indication:** The contents of all found context files are concatenated (with separators indicating their origin and path) and provided as part of the system prompt. The CLI footer displays the count of loaded context files, giving you a quick visual cue about the active instructional context.
|
||||||
- **Importing Content:** You can modularize your context files by importing other Markdown files using the `@path/to/file.md` syntax. For more details, see the [Memory Import Processor documentation](../core/memport.md).
|
- **Importing Content:** You can modularize your context files by importing other Markdown files using the `@path/to/file.md` syntax. For more details, see the [Memory Import Processor documentation](../core/memport.md).
|
||||||
@@ -651,20 +659,14 @@ To help us improve Qwen Code, we collect anonymized usage statistics. This data
|
|||||||
|
|
||||||
**How to opt out:**
|
**How to opt out:**
|
||||||
|
|
||||||
You can opt out of usage statistics collection at any time by setting the `usageStatisticsEnabled` property to `false` in your `settings.json` file:
|
You can opt out of usage statistics collection at any time by setting the `usageStatisticsEnabled` property to `false` under the `privacy` category in your `settings.json` file:
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
|
"privacy": {
|
||||||
"usageStatisticsEnabled": false
|
"usageStatisticsEnabled": false
|
||||||
}
|
}
|
||||||
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
Note: When usage statistics are enabled, events are sent to an Alibaba Cloud RUM collection endpoint.
|
Note: When usage statistics are enabled, events are sent to an Alibaba Cloud RUM collection endpoint.
|
||||||
|
|
||||||
- **`enableWelcomeBack`** (boolean):
|
|
||||||
- **Description:** Show welcome back dialog when returning to a project with conversation history.
|
|
||||||
- **Default:** `true`
|
|
||||||
- **Category:** UI
|
|
||||||
- **Requires Restart:** No
|
|
||||||
- **Example:** `"enableWelcomeBack": false`
|
|
||||||
- **Details:** When enabled, Qwen Code will automatically detect if you're returning to a project with a previously generated project summary (`.qwen/PROJECT_SUMMARY.md`) and show a dialog allowing you to continue your previous conversation or start fresh. This feature integrates with the `/chat summary` command and quit confirmation dialog. See the [Welcome Back documentation](./welcome-back.md) for more details.
|
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ Within Qwen Code, `packages/cli` is the frontend for users to send and receive p
|
|||||||
- **[Authentication](./authentication.md):** A guide to setting up authentication with Qwen OAuth and OpenAI-compatible providers.
|
- **[Authentication](./authentication.md):** A guide to setting up authentication with Qwen OAuth and OpenAI-compatible providers.
|
||||||
- **[Commands](./commands.md):** A reference for Qwen Code CLI commands (e.g., `/help`, `/tools`, `/theme`).
|
- **[Commands](./commands.md):** A reference for Qwen Code CLI commands (e.g., `/help`, `/tools`, `/theme`).
|
||||||
- **[Configuration](./configuration.md):** A guide to tailoring Qwen Code CLI behavior using configuration files.
|
- **[Configuration](./configuration.md):** A guide to tailoring Qwen Code CLI behavior using configuration files.
|
||||||
|
- **[Headless Mode](../headless.md):** A comprehensive guide to using Qwen Code programmatically for scripting and automation.
|
||||||
- **[Token Caching](./token-caching.md):** Optimize API costs through token caching.
|
- **[Token Caching](./token-caching.md):** Optimize API costs through token caching.
|
||||||
- **[Themes](./themes.md)**: A guide to customizing the CLI's appearance with different themes.
|
- **[Themes](./themes.md)**: A guide to customizing the CLI's appearance with different themes.
|
||||||
- **[Tutorials](tutorials.md)**: A tutorial showing how to use Qwen Code to automate a development task.
|
- **[Tutorials](tutorials.md)**: A tutorial showing how to use Qwen Code to automate a development task.
|
||||||
@@ -22,8 +23,10 @@ The following example pipes a command to Qwen Code from your terminal:
|
|||||||
echo "What is fine tuning?" | qwen
|
echo "What is fine tuning?" | qwen
|
||||||
```
|
```
|
||||||
|
|
||||||
Qwen Code executes the command and prints the output to your terminal. Note that you can achieve the same behavior by using the `--prompt` or `-p` flag. For example:
|
You can also use the `--prompt` or `-p` flag:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
qwen -p "What is fine tuning?"
|
qwen -p "What is fine tuning?"
|
||||||
```
|
```
|
||||||
|
|
||||||
|
For comprehensive documentation on headless usage, scripting, automation, and advanced examples, see the **[Headless Mode](../headless.md)** guide.
|
||||||
|
|||||||
@@ -46,25 +46,14 @@ Add a `customThemes` block to your user, project, or system `settings.json` file
|
|||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
|
"ui": {
|
||||||
"customThemes": {
|
"customThemes": {
|
||||||
"MyCustomTheme": {
|
"MyCustomTheme": {
|
||||||
"name": "MyCustomTheme",
|
"name": "MyCustomTheme",
|
||||||
"type": "custom",
|
"type": "custom",
|
||||||
"Background": "#181818",
|
"Background": "#181818",
|
||||||
"Foreground": "#F8F8F2",
|
...
|
||||||
"LightBlue": "#82AAFF",
|
}
|
||||||
"AccentBlue": "#61AFEF",
|
|
||||||
"AccentPurple": "#C678DD",
|
|
||||||
"AccentCyan": "#56B6C2",
|
|
||||||
"AccentGreen": "#98C379",
|
|
||||||
"AccentYellow": "#E5C07B",
|
|
||||||
"AccentRed": "#E06C75",
|
|
||||||
"Comment": "#5C6370",
|
|
||||||
"Gray": "#ABB2BF",
|
|
||||||
"DiffAdded": "#A6E3A1",
|
|
||||||
"DiffRemoved": "#F38BA8",
|
|
||||||
"DiffModified": "#89B4FA",
|
|
||||||
"GradientColors": ["#4796E4", "#847ACE", "#C3677F"]
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -115,8 +104,10 @@ To load a theme from a file, set the `theme` property in your `settings.json` to
|
|||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
|
"ui": {
|
||||||
"theme": "/path/to/your/theme.json"
|
"theme": "/path/to/your/theme.json"
|
||||||
}
|
}
|
||||||
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
The theme file must be a valid JSON file that follows the same structure as a custom theme defined in `settings.json`.
|
The theme file must be a valid JSON file that follows the same structure as a custom theme defined in `settings.json`.
|
||||||
@@ -154,7 +145,7 @@ The theme file must be a valid JSON file that follows the same structure as a cu
|
|||||||
### Using Your Custom Theme
|
### Using Your Custom Theme
|
||||||
|
|
||||||
- Select your custom theme using the `/theme` command in Qwen Code. Your custom theme will appear in the theme selection dialog.
|
- Select your custom theme using the `/theme` command in Qwen Code. Your custom theme will appear in the theme selection dialog.
|
||||||
- Or, set it as the default by adding `"theme": "MyCustomTheme"` to your `settings.json`.
|
- Or, set it as the default by adding `"theme": "MyCustomTheme"` to the `ui` object in your `settings.json`.
|
||||||
- Custom themes can be set at the user, project, or system level, and follow the same [configuration precedence](./configuration.md) as other settings.
|
- Custom themes can be set at the user, project, or system level, and follow the same [configuration precedence](./configuration.md) as other settings.
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|||||||
@@ -147,7 +147,7 @@ Processes import statements in context file content.
|
|||||||
- `debugMode` (boolean, optional): Whether to enable debug logging (default: false)
|
- `debugMode` (boolean, optional): Whether to enable debug logging (default: false)
|
||||||
- `importState` (ImportState, optional): State tracking for circular import prevention
|
- `importState` (ImportState, optional): State tracking for circular import prevention
|
||||||
|
|
||||||
**Returns:** Promise<ProcessImportsResult> - Object containing processed content and import tree
|
**Returns:** Promise<ProcessImportsResult> - Object containing processed content and import tree
|
||||||
|
|
||||||
### `ProcessImportsResult`
|
### `ProcessImportsResult`
|
||||||
|
|
||||||
@@ -187,7 +187,7 @@ Finds the project root by searching for a `.git` directory upwards from the give
|
|||||||
|
|
||||||
- `startDir` (string): The directory to start searching from
|
- `startDir` (string): The directory to start searching from
|
||||||
|
|
||||||
**Returns:** Promise<string> - The project root directory (or the start directory if no `.git` is found)
|
**Returns:** Promise<string> - The project root directory (or the start directory if no `.git` is found)
|
||||||
|
|
||||||
## Best Practices
|
## Best Practices
|
||||||
|
|
||||||
|
|||||||
@@ -23,8 +23,8 @@ The Qwen Code core (`packages/core`) features a robust system for defining, regi
|
|||||||
- **Tool Registry (`tool-registry.ts`):** A class (`ToolRegistry`) responsible for:
|
- **Tool Registry (`tool-registry.ts`):** A class (`ToolRegistry`) responsible for:
|
||||||
- **Registering Tools:** Holding a collection of all available built-in tools (e.g., `ReadFileTool`, `ShellTool`).
|
- **Registering Tools:** Holding a collection of all available built-in tools (e.g., `ReadFileTool`, `ShellTool`).
|
||||||
- **Discovering Tools:** It can also discover tools dynamically:
|
- **Discovering Tools:** It can also discover tools dynamically:
|
||||||
- **Command-based Discovery:** If `toolDiscoveryCommand` is configured in settings, this command is executed. It's expected to output JSON describing custom tools, which are then registered as `DiscoveredTool` instances.
|
- **Command-based Discovery:** If `tools.toolDiscoveryCommand` is configured in settings, this command is executed. It's expected to output JSON describing custom tools, which are then registered as `DiscoveredTool` instances.
|
||||||
- **MCP-based Discovery:** If `mcpServerCommand` is configured, the registry can connect to a Model Context Protocol (MCP) server to list and register tools (`DiscoveredMCPTool`).
|
- **MCP-based Discovery:** If `mcp.mcpServerCommand` is configured, the registry can connect to a Model Context Protocol (MCP) server to list and register tools (`DiscoveredMCPTool`).
|
||||||
- **Providing Schemas:** Exposing the `FunctionDeclaration` schemas of all registered tools to the model, so it knows what tools are available and how to use them.
|
- **Providing Schemas:** Exposing the `FunctionDeclaration` schemas of all registered tools to the model, so it knows what tools are available and how to use them.
|
||||||
- **Retrieving Tools:** Allowing the core to get a specific tool by name for execution.
|
- **Retrieving Tools:** Allowing the core to get a specific tool by name for execution.
|
||||||
|
|
||||||
@@ -69,7 +69,7 @@ Each of these tools extends `BaseTool` and implements the required methods for i
|
|||||||
|
|
||||||
While direct programmatic registration of new tools by users isn't explicitly detailed as a primary workflow in the provided files for typical end-users, the architecture supports extension through:
|
While direct programmatic registration of new tools by users isn't explicitly detailed as a primary workflow in the provided files for typical end-users, the architecture supports extension through:
|
||||||
|
|
||||||
- **Command-based Discovery:** Advanced users or project administrators can define a `toolDiscoveryCommand` in `settings.json`. This command, when run by the core, should output a JSON array of `FunctionDeclaration` objects. The core will then make these available as `DiscoveredTool` instances. The corresponding `toolCallCommand` would then be responsible for actually executing these custom tools.
|
- **Command-based Discovery:** Advanced users or project administrators can define a `tools.toolDiscoveryCommand` in `settings.json`. This command, when run by the core, should output a JSON array of `FunctionDeclaration` objects. The core will then make these available as `DiscoveredTool` instances. The corresponding `tools.toolCallCommand` would then be responsible for actually executing these custom tools.
|
||||||
- **MCP Server(s):** For more complex scenarios, one or more MCP servers can be set up and configured via the `mcpServers` setting in `settings.json`. The core can then discover and use tools exposed by these servers. As mentioned, if you have multiple MCP servers, the tool names will be prefixed with the server name from your configuration (e.g., `serverAlias__actualToolName`).
|
- **MCP Server(s):** For more complex scenarios, one or more MCP servers can be set up and configured via the `mcpServers` setting in `settings.json`. The core can then discover and use tools exposed by these servers. As mentioned, if you have multiple MCP servers, the tool names will be prefixed with the server name from your configuration (e.g., `serverAlias__actualToolName`).
|
||||||
|
|
||||||
This tool system provides a flexible and powerful way to augment the model's capabilities, making Qwen Code a versatile assistant for a wide range of tasks.
|
This tool system provides a flexible and powerful way to augment the model's capabilities, making Qwen Code a versatile assistant for a wide range of tasks.
|
||||||
|
|||||||
121
docs/extension-releasing.md
Normal file
121
docs/extension-releasing.md
Normal file
@@ -0,0 +1,121 @@
|
|||||||
|
# Extension Releasing
|
||||||
|
|
||||||
|
There are two primary ways of releasing extensions to users:
|
||||||
|
|
||||||
|
- [Git repository](#releasing-through-a-git-repository)
|
||||||
|
- [Github Releases](#releasing-through-github-releases)
|
||||||
|
|
||||||
|
Git repository releases tend to be the simplest and most flexible approach, while GitHub releases can be more efficient on initial install as they are shipped as single archives instead of requiring a git clone which downloads each file individually. Github releases may also contain platform specific archives if you need to ship platform specific binary files.
|
||||||
|
|
||||||
|
## Releasing through a git repository
|
||||||
|
|
||||||
|
This is the most flexible and simple option. All you need to do us create a publicly accessible git repo (such as a public github repository) and then users can install your extension using `qwen extensions install <your-repo-uri>`, or for a GitHub repository they can use the simplified `qwen extensions install <org>/<repo>` format. They can optionally depend on a specific ref (branch/tag/commit) using the `--ref=<some-ref>` argument, this defaults to the default branch.
|
||||||
|
|
||||||
|
Whenever commits are pushed to the ref that a user depends on, they will be prompted to update the extension. Note that this also allows for easy rollbacks, the HEAD commit is always treated as the latest version regardless of the actual version in the `qwen-extension.json` file.
|
||||||
|
|
||||||
|
### Managing release channels using a git repository
|
||||||
|
|
||||||
|
Users can depend on any ref from your git repo, such as a branch or tag, which allows you to manage multiple release channels.
|
||||||
|
|
||||||
|
For instance, you can maintain a `stable` branch, which users can install this way `qwen extensions install <your-repo-uri> --ref=stable`. Or, you could make this the default by treating your default branch as your stable release branch, and doing development in a different branch (for instance called `dev`). You can maintain as many branches or tags as you like, providing maximum flexibility for you and your users.
|
||||||
|
|
||||||
|
Note that these `ref` arguments can be tags, branches, or even specific commits, which allows users to depend on a specific version of your extension. It is up to you how you want to manage your tags and branches.
|
||||||
|
|
||||||
|
### Example releasing flow using a git repo
|
||||||
|
|
||||||
|
While there are many options for how you want to manage releases using a git flow, we recommend treating your default branch as your "stable" release branch. This means that the default behavior for `qwen extensions install <your-repo-uri>` is to be on the stable release branch.
|
||||||
|
|
||||||
|
Lets say you want to maintain three standard release channels, `stable`, `preview`, and `dev`. You would do all your standard development in the `dev` branch. When you are ready to do a preview release, you merge that branch into your `preview` branch. When you are ready to promote your preview branch to stable, you merge `preview` into your stable branch (which might be your default branch or a different branch).
|
||||||
|
|
||||||
|
You can also cherry pick changes from one branch into another using `git cherry-pick`, but do note that this will result in your branches having a slightly divergent history from each other, unless you force push changes to your branches on each release to restore the history to a clean slate (which may not be possible for the default branch depending on your repository settings). If you plan on doing cherry picks, you may want to avoid having your default branch be the stable branch to avoid force-pushing to the default branch which should generally be avoided.
|
||||||
|
|
||||||
|
## Releasing through Github releases
|
||||||
|
|
||||||
|
Qwen Code extensions can be distributed through [GitHub Releases](https://docs.github.com/en/repositories/releasing-projects-on-github/about-releases). This provides a faster and more reliable initial installation experience for users, as it avoids the need to clone the repository.
|
||||||
|
|
||||||
|
Each release includes at least one archive file, which contains the full contents of the repo at the tag that it was linked to. Releases may also include [pre-built archives](#custom-pre-built-archives) if your extension requires some build step or has platform specific binaries attached to it.
|
||||||
|
|
||||||
|
When checking for updates, qwen code will just look for the latest release on github (you must mark it as such when creating the release), unless the user installed a specific release by passing `--ref=<some-release-tag>`. We do not at this time support opting in to pre-release releases or semver.
|
||||||
|
|
||||||
|
### Custom pre-built archives
|
||||||
|
|
||||||
|
Custom archives must be attached directly to the github release as assets and must be fully self-contained. This means they should include the entire extension, see [archive structure](#archive-structure).
|
||||||
|
|
||||||
|
If your extension is platform-independent, you can provide a single generic asset. In this case, there should be only one asset attached to the release.
|
||||||
|
|
||||||
|
Custom archives may also be used if you want to develop your extension within a larger repository, you can build an archive which has a different layout from the repo itself (for instance it might just be an archive of a subdirectory containing the extension).
|
||||||
|
|
||||||
|
#### Platform specific archives
|
||||||
|
|
||||||
|
To ensure Qwen Code can automatically find the correct release asset for each platform, you must follow this naming convention. The CLI will search for assets in the following order:
|
||||||
|
|
||||||
|
1. **Platform and Architecture-Specific:** `{platform}.{arch}.{name}.{extension}`
|
||||||
|
2. **Platform-Specific:** `{platform}.{name}.{extension}`
|
||||||
|
3. **Generic:** If only one asset is provided, it will be used as a generic fallback.
|
||||||
|
|
||||||
|
- `{name}`: The name of your extension.
|
||||||
|
- `{platform}`: The operating system. Supported values are:
|
||||||
|
- `darwin` (macOS)
|
||||||
|
- `linux`
|
||||||
|
- `win32` (Windows)
|
||||||
|
- `{arch}`: The architecture. Supported values are:
|
||||||
|
- `x64`
|
||||||
|
- `arm64`
|
||||||
|
- `{extension}`: The file extension of the archive (e.g., `.tar.gz` or `.zip`).
|
||||||
|
|
||||||
|
**Examples:**
|
||||||
|
|
||||||
|
- `darwin.arm64.my-tool.tar.gz` (specific to Apple Silicon Macs)
|
||||||
|
- `darwin.my-tool.tar.gz` (for all Macs)
|
||||||
|
- `linux.x64.my-tool.tar.gz`
|
||||||
|
- `win32.my-tool.zip`
|
||||||
|
|
||||||
|
#### Archive structure
|
||||||
|
|
||||||
|
Archives must be fully contained extensions and have all the standard requirements - specifically the `qwen-extension.json` file must be at the root of the archive.
|
||||||
|
|
||||||
|
The rest of the layout should look exactly the same as a typical extension, see [extensions.md](extension.md).
|
||||||
|
|
||||||
|
#### Example GitHub Actions workflow
|
||||||
|
|
||||||
|
Here is an example of a GitHub Actions workflow that builds and releases a Qwen Code extension for multiple platforms:
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
name: Release Extension
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- 'v*'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
release:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v3
|
||||||
|
|
||||||
|
- name: Set up Node.js
|
||||||
|
uses: actions/setup-node@v3
|
||||||
|
with:
|
||||||
|
node-version: '20'
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
run: npm ci
|
||||||
|
|
||||||
|
- name: Build extension
|
||||||
|
run: npm run build
|
||||||
|
|
||||||
|
- name: Create release assets
|
||||||
|
run: |
|
||||||
|
npm run package -- --platform=darwin --arch=arm64
|
||||||
|
npm run package -- --platform=linux --arch=x64
|
||||||
|
npm run package -- --platform=win32 --arch=x64
|
||||||
|
|
||||||
|
- name: Create GitHub Release
|
||||||
|
uses: softprops/action-gh-release@v1
|
||||||
|
with:
|
||||||
|
files: |
|
||||||
|
release/darwin.arm64.my-tool.tar.gz
|
||||||
|
release/linux.arm64.my-tool.tar.gz
|
||||||
|
release/win32.arm64.my-tool.zip
|
||||||
|
```
|
||||||
@@ -1,19 +1,88 @@
|
|||||||
# Qwen Code Extensions
|
# Qwen Code Extensions
|
||||||
|
|
||||||
Qwen Code supports extensions that can be used to configure and extend its functionality.
|
Qwen Code extensions package prompts, MCP servers, and custom commands into a familiar and user-friendly format. With extensions, you can expand the capabilities of Qwen Code and share those capabilities with others. They are designed to be easily installable and shareable.
|
||||||
|
|
||||||
|
## Extension management
|
||||||
|
|
||||||
|
We offer a suite of extension management tools using `qwen extensions` commands.
|
||||||
|
|
||||||
|
Note that these commands are not supported from within the CLI, although you can list installed extensions using the `/extensions list` subcommand.
|
||||||
|
|
||||||
|
Note that all of these commands will only be reflected in active CLI sessions on restart.
|
||||||
|
|
||||||
|
### Installing an extension
|
||||||
|
|
||||||
|
You can install an extension using `qwen extensions install` with either a GitHub URL or a local path`.
|
||||||
|
|
||||||
|
Note that we create a copy of the installed extension, so you will need to run `qwen extensions update` to pull in changes from both locally-defined extensions and those on GitHub.
|
||||||
|
|
||||||
|
```
|
||||||
|
qwen extensions install https://github.com/qwen-cli-extensions/security
|
||||||
|
```
|
||||||
|
|
||||||
|
This will install the Qwen Code Security extension, which offers support for a `/security:analyze` command.
|
||||||
|
|
||||||
|
### Uninstalling an extension
|
||||||
|
|
||||||
|
To uninstall, run `qwen extensions uninstall extension-name`, so, in the case of the install example:
|
||||||
|
|
||||||
|
```
|
||||||
|
qwen extensions uninstall qwen-cli-security
|
||||||
|
```
|
||||||
|
|
||||||
|
### Disabling an extension
|
||||||
|
|
||||||
|
Extensions are, by default, enabled across all workspaces. You can disable an extension entirely or for specific workspace.
|
||||||
|
|
||||||
|
For example, `qwen extensions disable extension-name` will disable the extension at the user level, so it will be disabled everywhere. `qwen extensions disable extension-name --scope=workspace` will only disable the extension in the current workspace.
|
||||||
|
|
||||||
|
### Enabling an extension
|
||||||
|
|
||||||
|
You can enable extensions using `qwen extensions enable extension-name`. You can also enable an extension for a specific workspace using `qwen extensions enable extension-name --scope=workspace` from within that workspace.
|
||||||
|
|
||||||
|
This is useful if you have an extension disabled at the top-level and only enabled in specific places.
|
||||||
|
|
||||||
|
### Updating an extension
|
||||||
|
|
||||||
|
For extensions installed from a local path or a git repository, you can explicitly update to the latest version (as reflected in the `qwen-extension.json` `version` field) with `qwen extensions update extension-name`.
|
||||||
|
|
||||||
|
You can update all extensions with:
|
||||||
|
|
||||||
|
```
|
||||||
|
qwen extensions update --all
|
||||||
|
```
|
||||||
|
|
||||||
|
## Extension creation
|
||||||
|
|
||||||
|
We offer commands to make extension development easier.
|
||||||
|
|
||||||
|
### Create a boilerplate extension
|
||||||
|
|
||||||
|
We offer several example extensions `context`, `custom-commands`, `exclude-tools` and `mcp-server`. You can view these examples [here](https://github.com/QwenLM/qwen-code/tree/main/packages/cli/src/commands/extensions/examples).
|
||||||
|
|
||||||
|
To copy one of these examples into a development directory using the type of your choosing, run:
|
||||||
|
|
||||||
|
```
|
||||||
|
qwen extensions new path/to/directory custom-commands
|
||||||
|
```
|
||||||
|
|
||||||
|
### Link a local extension
|
||||||
|
|
||||||
|
The `qwen extensions link` command will create a symbolic link from the extension installation directory to the development path.
|
||||||
|
|
||||||
|
This is useful so you don't have to run `qwen extensions update` every time you make changes you'd like to test.
|
||||||
|
|
||||||
|
```
|
||||||
|
qwen extensions link path/to/directory
|
||||||
|
```
|
||||||
|
|
||||||
## How it works
|
## How it works
|
||||||
|
|
||||||
On startup, Qwen Code looks for extensions in two locations:
|
On startup, Qwen Code looks for extensions in `<home>/.qwen/extensions`
|
||||||
|
|
||||||
1. `<workspace>/.qwen/extensions`
|
Extensions exist as a directory that contains a `qwen-extension.json` file. For example:
|
||||||
2. `<home>/.qwen/extensions`
|
|
||||||
|
|
||||||
Qwen Code loads all extensions from both locations. If an extension with the same name exists in both locations, the extension in the workspace directory takes precedence.
|
`<home>/.qwen/extensions/my-extension/qwen-extension.json`
|
||||||
|
|
||||||
Within each location, individual extensions exist as a directory that contains a `qwen-extension.json` file. For example:
|
|
||||||
|
|
||||||
`<workspace>/.qwen/extensions/my-extension/qwen-extension.json`
|
|
||||||
|
|
||||||
### `qwen-extension.json`
|
### `qwen-extension.json`
|
||||||
|
|
||||||
@@ -33,19 +102,20 @@ The `qwen-extension.json` file contains the configuration for the extension. The
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
- `name`: The name of the extension. This is used to uniquely identify the extension and for conflict resolution when extension commands have the same name as user or project commands.
|
- `name`: The name of the extension. This is used to uniquely identify the extension and for conflict resolution when extension commands have the same name as user or project commands. The name should be lowercase or numbers and use dashes instead of underscores or spaces. This is how users will refer to your extension in the CLI. Note that we expect this name to match the extension directory name.
|
||||||
- `version`: The version of the extension.
|
- `version`: The version of the extension.
|
||||||
- `mcpServers`: A map of MCP servers to configure. The key is the name of the server, and the value is the server configuration. These servers will be loaded on startup just like MCP servers configured in a [`settings.json` file](./cli/configuration.md). If both an extension and a `settings.json` file configure an MCP server with the same name, the server defined in the `settings.json` file takes precedence.
|
- `mcpServers`: A map of MCP servers to configure. The key is the name of the server, and the value is the server configuration. These servers will be loaded on startup just like MCP servers configured in a [`settings.json` file](./cli/configuration.md). If both an extension and a `settings.json` file configure an MCP server with the same name, the server defined in the `settings.json` file takes precedence.
|
||||||
- `contextFileName`: The name of the file that contains the context for the extension. This will be used to load the context from the workspace. If this property is not used but a `QWEN.md` file is present in your extension directory, then that file will be loaded.
|
- Note that all MCP server configuration options are supported except for `trust`.
|
||||||
- `excludeTools`: An array of tool names to exclude from the model. You can also specify command-specific restrictions for tools that support it, like the `run_shell_command` tool. For example, `"excludeTools": ["run_shell_command(rm -rf)"]` will block the `rm -rf` command.
|
- `contextFileName`: The name of the file that contains the context for the extension. This will be used to load the context from the extension directory. If this property is not used but a `QWEN.md` file is present in your extension directory, then that file will be loaded.
|
||||||
|
- `excludeTools`: An array of tool names to exclude from the model. You can also specify command-specific restrictions for tools that support it, like the `run_shell_command` tool. For example, `"excludeTools": ["run_shell_command(rm -rf)"]` will block the `rm -rf` command. Note that this differs from the MCP server `excludeTools` functionality, which can be listed in the MCP server config.
|
||||||
|
|
||||||
When Qwen Code starts, it loads all the extensions and merges their configurations. If there are any conflicts, the workspace configuration takes precedence.
|
When Qwen Code starts, it loads all the extensions and merges their configurations. If there are any conflicts, the workspace configuration takes precedence.
|
||||||
|
|
||||||
## Extension Commands
|
### Custom commands
|
||||||
|
|
||||||
Extensions can provide [custom commands](./cli/commands.md#custom-commands) by placing TOML files in a `commands/` subdirectory within the extension directory. These commands follow the same format as user and project custom commands and use standard naming conventions.
|
Extensions can provide [custom commands](./cli/commands.md#custom-commands) by placing TOML files in a `commands/` subdirectory within the extension directory. These commands follow the same format as user and project custom commands and use standard naming conventions.
|
||||||
|
|
||||||
### Example
|
**Example**
|
||||||
|
|
||||||
An extension named `gcp` with the following structure:
|
An extension named `gcp` with the following structure:
|
||||||
|
|
||||||
@@ -63,7 +133,7 @@ Would provide these commands:
|
|||||||
- `/deploy` - Shows as `[gcp] Custom command from deploy.toml` in help
|
- `/deploy` - Shows as `[gcp] Custom command from deploy.toml` in help
|
||||||
- `/gcs:sync` - Shows as `[gcp] Custom command from sync.toml` in help
|
- `/gcs:sync` - Shows as `[gcp] Custom command from sync.toml` in help
|
||||||
|
|
||||||
### Conflict Resolution
|
### Conflict resolution
|
||||||
|
|
||||||
Extension commands have the lowest precedence. When a conflict occurs with user or project commands:
|
Extension commands have the lowest precedence. When a conflict occurs with user or project commands:
|
||||||
|
|
||||||
@@ -75,20 +145,7 @@ For example, if both a user and the `gcp` extension define a `deploy` command:
|
|||||||
- `/deploy` - Executes the user's deploy command
|
- `/deploy` - Executes the user's deploy command
|
||||||
- `/gcp.deploy` - Executes the extension's deploy command (marked with `[gcp]` tag)
|
- `/gcp.deploy` - Executes the extension's deploy command (marked with `[gcp]` tag)
|
||||||
|
|
||||||
## Installing Extensions
|
## Variables
|
||||||
|
|
||||||
You can install extensions using the `install` command. This command allows you to install extensions from a Git repository or a local path.
|
|
||||||
|
|
||||||
### Usage
|
|
||||||
|
|
||||||
`qwen extensions install <source> | [options]`
|
|
||||||
|
|
||||||
### Options
|
|
||||||
|
|
||||||
- `source <url> positional argument`: The URL of a Git repository to install the extension from. The repository must contain a `qwen-extension.json` file in its root.
|
|
||||||
- `--path <path>`: The path to a local directory to install as an extension. The directory must contain a `qwen-extension.json` file.
|
|
||||||
|
|
||||||
# Variables
|
|
||||||
|
|
||||||
Qwen Code extensions allow variable substitution in `qwen-extension.json`. This can be useful if e.g., you need the current directory to run an MCP server using `"cwd": "${extensionPath}${/}run.ts"`.
|
Qwen Code extensions allow variable substitution in `qwen-extension.json`. This can be useful if e.g., you need the current directory to run an MCP server using `"cwd": "${extensionPath}${/}run.ts"`.
|
||||||
|
|
||||||
@@ -97,4 +154,5 @@ Qwen Code extensions allow variable substitution in `qwen-extension.json`. This
|
|||||||
| variable | description |
|
| variable | description |
|
||||||
| -------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------- |
|
| -------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------- |
|
||||||
| `${extensionPath}` | The fully-qualified path of the extension in the user's filesystem e.g., '/Users/username/.qwen/extensions/example-extension'. This will not unwrap symlinks. |
|
| `${extensionPath}` | The fully-qualified path of the extension in the user's filesystem e.g., '/Users/username/.qwen/extensions/example-extension'. This will not unwrap symlinks. |
|
||||||
|
| `${workspacePath}` | The fully-qualified path of the current workspace. |
|
||||||
| `${/} or ${pathSeparator}` | The path separator (differs per OS). |
|
| `${/} or ${pathSeparator}` | The path separator (differs per OS). |
|
||||||
|
|||||||
213
docs/getting-started-extensions.md
Normal file
213
docs/getting-started-extensions.md
Normal file
@@ -0,0 +1,213 @@
|
|||||||
|
# Getting Started with Qwen Code Extensions
|
||||||
|
|
||||||
|
This guide will walk you through creating your first Qwen Code extension. You'll learn how to set up a new extension, add a custom tool via an MCP server, create a custom command, and provide context to the model with a `QWEN.md` file.
|
||||||
|
|
||||||
|
## Prerequisites
|
||||||
|
|
||||||
|
Before you start, make sure you have the Qwen Code installed and a basic understanding of Node.js and TypeScript.
|
||||||
|
|
||||||
|
## Step 1: Create a New Extension
|
||||||
|
|
||||||
|
The easiest way to start is by using one of the built-in templates. We'll use the `mcp-server` example as our foundation.
|
||||||
|
|
||||||
|
Run the following command to create a new directory called `my-first-extension` with the template files:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
qwen extensions new my-first-extension mcp-server
|
||||||
|
```
|
||||||
|
|
||||||
|
This will create a new directory with the following structure:
|
||||||
|
|
||||||
|
```
|
||||||
|
my-first-extension/
|
||||||
|
├── example.ts
|
||||||
|
├── qwen-extension.json
|
||||||
|
├── package.json
|
||||||
|
└── tsconfig.json
|
||||||
|
```
|
||||||
|
|
||||||
|
## Step 2: Understand the Extension Files
|
||||||
|
|
||||||
|
Let's look at the key files in your new extension.
|
||||||
|
|
||||||
|
### `qwen-extension.json`
|
||||||
|
|
||||||
|
This is the manifest file for your extension. It tells Qwen Code how to load and use your extension.
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"name": "my-first-extension",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"mcpServers": {
|
||||||
|
"nodeServer": {
|
||||||
|
"command": "node",
|
||||||
|
"args": ["${extensionPath}${/}dist${/}example.js"],
|
||||||
|
"cwd": "${extensionPath}"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- `name`: The unique name for your extension.
|
||||||
|
- `version`: The version of your extension.
|
||||||
|
- `mcpServers`: This section defines one or more Model Context Protocol (MCP) servers. MCP servers are how you can add new tools for the model to use.
|
||||||
|
- `command`, `args`, `cwd`: These fields specify how to start your server. Notice the use of the `${extensionPath}` variable, which Qwen Code replaces with the absolute path to your extension's installation directory. This allows your extension to work regardless of where it's installed.
|
||||||
|
|
||||||
|
### `example.ts`
|
||||||
|
|
||||||
|
This file contains the source code for your MCP server. It's a simple Node.js server that uses the `@modelcontextprotocol/sdk`.
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
||||||
|
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
||||||
|
import { z } from 'zod';
|
||||||
|
|
||||||
|
const server = new McpServer({
|
||||||
|
name: 'prompt-server',
|
||||||
|
version: '1.0.0',
|
||||||
|
});
|
||||||
|
|
||||||
|
// Registers a new tool named 'fetch_posts'
|
||||||
|
server.registerTool(
|
||||||
|
'fetch_posts',
|
||||||
|
{
|
||||||
|
description: 'Fetches a list of posts from a public API.',
|
||||||
|
inputSchema: z.object({}).shape,
|
||||||
|
},
|
||||||
|
async () => {
|
||||||
|
const apiResponse = await fetch(
|
||||||
|
'https://jsonplaceholder.typicode.com/posts',
|
||||||
|
);
|
||||||
|
const posts = await apiResponse.json();
|
||||||
|
const response = { posts: posts.slice(0, 5) };
|
||||||
|
return {
|
||||||
|
content: [
|
||||||
|
{
|
||||||
|
type: 'text',
|
||||||
|
text: JSON.stringify(response),
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
// ... (prompt registration omitted for brevity)
|
||||||
|
|
||||||
|
const transport = new StdioServerTransport();
|
||||||
|
await server.connect(transport);
|
||||||
|
```
|
||||||
|
|
||||||
|
This server defines a single tool called `fetch_posts` that fetches data from a public API.
|
||||||
|
|
||||||
|
### `package.json` and `tsconfig.json`
|
||||||
|
|
||||||
|
These are standard configuration files for a TypeScript project. The `package.json` file defines dependencies and a `build` script, and `tsconfig.json` configures the TypeScript compiler.
|
||||||
|
|
||||||
|
## Step 3: Build and Link Your Extension
|
||||||
|
|
||||||
|
Before you can use the extension, you need to compile the TypeScript code and link the extension to your Qwen Code installation for local development.
|
||||||
|
|
||||||
|
1. **Install dependencies:**
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd my-first-extension
|
||||||
|
npm install
|
||||||
|
```
|
||||||
|
|
||||||
|
2. **Build the server:**
|
||||||
|
|
||||||
|
```bash
|
||||||
|
npm run build
|
||||||
|
```
|
||||||
|
|
||||||
|
This will compile `example.ts` into `dist/example.js`, which is the file referenced in your `qwen-extension.json`.
|
||||||
|
|
||||||
|
3. **Link the extension:**
|
||||||
|
|
||||||
|
The `link` command creates a symbolic link from the Qwen Code extensions directory to your development directory. This means any changes you make will be reflected immediately without needing to reinstall.
|
||||||
|
|
||||||
|
```bash
|
||||||
|
qwen extensions link .
|
||||||
|
```
|
||||||
|
|
||||||
|
Now, restart your Qwen Code session. The new `fetch_posts` tool will be available. You can test it by asking: "fetch posts".
|
||||||
|
|
||||||
|
## Step 4: Add a Custom Command
|
||||||
|
|
||||||
|
Custom commands provide a way to create shortcuts for complex prompts. Let's add a command that searches for a pattern in your code.
|
||||||
|
|
||||||
|
1. Create a `commands` directory and a subdirectory for your command group:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
mkdir -p commands/fs
|
||||||
|
```
|
||||||
|
|
||||||
|
2. Create a file named `commands/fs/grep-code.toml`:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
prompt = """
|
||||||
|
Please summarize the findings for the pattern `{{args}}`.
|
||||||
|
|
||||||
|
Search Results:
|
||||||
|
!{grep -r {{args}} .}
|
||||||
|
"""
|
||||||
|
```
|
||||||
|
|
||||||
|
This command, `/fs:grep-code`, will take an argument, run the `grep` shell command with it, and pipe the results into a prompt for summarization.
|
||||||
|
|
||||||
|
After saving the file, restart the Qwen Code. You can now run `/fs:grep-code "some pattern"` to use your new command.
|
||||||
|
|
||||||
|
## Step 5: Add a Custom `QWEN.md`
|
||||||
|
|
||||||
|
You can provide persistent context to the model by adding a `QWEN.md` file to your extension. This is useful for giving the model instructions on how to behave or information about your extension's tools. Note that you may not always need this for extensions built to expose commands and prompts.
|
||||||
|
|
||||||
|
1. Create a file named `QWEN.md` in the root of your extension directory:
|
||||||
|
|
||||||
|
```markdown
|
||||||
|
# My First Extension Instructions
|
||||||
|
|
||||||
|
You are an expert developer assistant. When the user asks you to fetch posts, use the `fetch_posts` tool. Be concise in your responses.
|
||||||
|
```
|
||||||
|
|
||||||
|
2. Update your `qwen-extension.json` to tell the CLI to load this file:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"name": "my-first-extension",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"contextFileName": "QWEN.md",
|
||||||
|
"mcpServers": {
|
||||||
|
"nodeServer": {
|
||||||
|
"command": "node",
|
||||||
|
"args": ["${extensionPath}${/}dist${/}example.js"],
|
||||||
|
"cwd": "${extensionPath}"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Restart the CLI again. The model will now have the context from your `QWEN.md` file in every session where the extension is active.
|
||||||
|
|
||||||
|
## Step 6: Releasing Your Extension
|
||||||
|
|
||||||
|
Once you are happy with your extension, you can share it with others. The two primary ways of releasing extensions are via a Git repository or through GitHub Releases. Using a public Git repository is the simplest method.
|
||||||
|
|
||||||
|
For detailed instructions on both methods, please refer to the [Extension Releasing Guide](extension-releasing.md).
|
||||||
|
|
||||||
|
## Conclusion
|
||||||
|
|
||||||
|
You've successfully created a Qwen Code extension! You learned how to:
|
||||||
|
|
||||||
|
- Bootstrap a new extension from a template.
|
||||||
|
- Add custom tools with an MCP server.
|
||||||
|
- Create convenient custom commands.
|
||||||
|
- Provide persistent context to the model.
|
||||||
|
- Link your extension for local development.
|
||||||
|
|
||||||
|
From here, you can explore more advanced features and build powerful new capabilities into the Qwen Code.
|
||||||
308
docs/headless.md
Normal file
308
docs/headless.md
Normal file
@@ -0,0 +1,308 @@
|
|||||||
|
# Headless Mode
|
||||||
|
|
||||||
|
Headless mode allows you to run Qwen Code programmatically from command line
|
||||||
|
scripts and automation tools without any interactive UI. This is ideal for
|
||||||
|
scripting, automation, CI/CD pipelines, and building AI-powered tools.
|
||||||
|
|
||||||
|
- [Headless Mode](#headless-mode)
|
||||||
|
- [Overview](#overview)
|
||||||
|
- [Basic Usage](#basic-usage)
|
||||||
|
- [Direct Prompts](#direct-prompts)
|
||||||
|
- [Stdin Input](#stdin-input)
|
||||||
|
- [Combining with File Input](#combining-with-file-input)
|
||||||
|
- [Output Formats](#output-formats)
|
||||||
|
- [Text Output (Default)](#text-output-default)
|
||||||
|
- [JSON Output](#json-output)
|
||||||
|
- [Response Schema](#response-schema)
|
||||||
|
- [Example Usage](#example-usage)
|
||||||
|
- [File Redirection](#file-redirection)
|
||||||
|
- [Configuration Options](#configuration-options)
|
||||||
|
- [Examples](#examples)
|
||||||
|
- [Code review](#code-review)
|
||||||
|
- [Generate commit messages](#generate-commit-messages)
|
||||||
|
- [API documentation](#api-documentation)
|
||||||
|
- [Batch code analysis](#batch-code-analysis)
|
||||||
|
- [Code review](#code-review-1)
|
||||||
|
- [Log analysis](#log-analysis)
|
||||||
|
- [Release notes generation](#release-notes-generation)
|
||||||
|
- [Model and tool usage tracking](#model-and-tool-usage-tracking)
|
||||||
|
- [Resources](#resources)
|
||||||
|
|
||||||
|
## Overview
|
||||||
|
|
||||||
|
The headless mode provides a headless interface to Qwen Code that:
|
||||||
|
|
||||||
|
- Accepts prompts via command line arguments or stdin
|
||||||
|
- Returns structured output (text or JSON)
|
||||||
|
- Supports file redirection and piping
|
||||||
|
- Enables automation and scripting workflows
|
||||||
|
- Provides consistent exit codes for error handling
|
||||||
|
|
||||||
|
## Basic Usage
|
||||||
|
|
||||||
|
### Direct Prompts
|
||||||
|
|
||||||
|
Use the `--prompt` (or `-p`) flag to run in headless mode:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
qwen --prompt "What is machine learning?"
|
||||||
|
```
|
||||||
|
|
||||||
|
### Stdin Input
|
||||||
|
|
||||||
|
Pipe input to Qwen Code from your terminal:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
echo "Explain this code" | qwen
|
||||||
|
```
|
||||||
|
|
||||||
|
### Combining with File Input
|
||||||
|
|
||||||
|
Read from files and process with Qwen Code:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cat README.md | qwen --prompt "Summarize this documentation"
|
||||||
|
```
|
||||||
|
|
||||||
|
## Output Formats
|
||||||
|
|
||||||
|
### Text Output (Default)
|
||||||
|
|
||||||
|
Standard human-readable output:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
qwen -p "What is the capital of France?"
|
||||||
|
```
|
||||||
|
|
||||||
|
Response format:
|
||||||
|
|
||||||
|
```
|
||||||
|
The capital of France is Paris.
|
||||||
|
```
|
||||||
|
|
||||||
|
### JSON Output
|
||||||
|
|
||||||
|
Returns structured data including response, statistics, and metadata. This
|
||||||
|
format is ideal for programmatic processing and automation scripts.
|
||||||
|
|
||||||
|
#### Response Schema
|
||||||
|
|
||||||
|
The JSON output follows this high-level structure:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"response": "string", // The main AI-generated content answering your prompt
|
||||||
|
"stats": {
|
||||||
|
// Usage metrics and performance data
|
||||||
|
"models": {
|
||||||
|
// Per-model API and token usage statistics
|
||||||
|
"[model-name]": {
|
||||||
|
"api": {
|
||||||
|
/* request counts, errors, latency */
|
||||||
|
},
|
||||||
|
"tokens": {
|
||||||
|
/* prompt, response, cached, total counts */
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"tools": {
|
||||||
|
// Tool execution statistics
|
||||||
|
"totalCalls": "number",
|
||||||
|
"totalSuccess": "number",
|
||||||
|
"totalFail": "number",
|
||||||
|
"totalDurationMs": "number",
|
||||||
|
"totalDecisions": {
|
||||||
|
/* accept, reject, modify, auto_accept counts */
|
||||||
|
},
|
||||||
|
"byName": {
|
||||||
|
/* per-tool detailed stats */
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"files": {
|
||||||
|
// File modification statistics
|
||||||
|
"totalLinesAdded": "number",
|
||||||
|
"totalLinesRemoved": "number"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"error": {
|
||||||
|
// Present only when an error occurred
|
||||||
|
"type": "string", // Error type (e.g., "ApiError", "AuthError")
|
||||||
|
"message": "string", // Human-readable error description
|
||||||
|
"code": "number" // Optional error code
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Example Usage
|
||||||
|
|
||||||
|
```bash
|
||||||
|
qwen -p "What is the capital of France?" --output-format json
|
||||||
|
```
|
||||||
|
|
||||||
|
Response:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"response": "The capital of France is Paris.",
|
||||||
|
"stats": {
|
||||||
|
"models": {
|
||||||
|
"qwen3-coder-plus": {
|
||||||
|
"api": {
|
||||||
|
"totalRequests": 2,
|
||||||
|
"totalErrors": 0,
|
||||||
|
"totalLatencyMs": 5053
|
||||||
|
},
|
||||||
|
"tokens": {
|
||||||
|
"prompt": 24939,
|
||||||
|
"candidates": 20,
|
||||||
|
"total": 25113,
|
||||||
|
"cached": 21263,
|
||||||
|
"thoughts": 154,
|
||||||
|
"tool": 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"tools": {
|
||||||
|
"totalCalls": 1,
|
||||||
|
"totalSuccess": 1,
|
||||||
|
"totalFail": 0,
|
||||||
|
"totalDurationMs": 1881,
|
||||||
|
"totalDecisions": {
|
||||||
|
"accept": 0,
|
||||||
|
"reject": 0,
|
||||||
|
"modify": 0,
|
||||||
|
"auto_accept": 1
|
||||||
|
},
|
||||||
|
"byName": {
|
||||||
|
"google_web_search": {
|
||||||
|
"count": 1,
|
||||||
|
"success": 1,
|
||||||
|
"fail": 0,
|
||||||
|
"durationMs": 1881,
|
||||||
|
"decisions": {
|
||||||
|
"accept": 0,
|
||||||
|
"reject": 0,
|
||||||
|
"modify": 0,
|
||||||
|
"auto_accept": 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"files": {
|
||||||
|
"totalLinesAdded": 0,
|
||||||
|
"totalLinesRemoved": 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### File Redirection
|
||||||
|
|
||||||
|
Save output to files or pipe to other commands:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Save to file
|
||||||
|
qwen -p "Explain Docker" > docker-explanation.txt
|
||||||
|
qwen -p "Explain Docker" --output-format json > docker-explanation.json
|
||||||
|
|
||||||
|
# Append to file
|
||||||
|
qwen -p "Add more details" >> docker-explanation.txt
|
||||||
|
|
||||||
|
# Pipe to other tools
|
||||||
|
qwen -p "What is Kubernetes?" --output-format json | jq '.response'
|
||||||
|
qwen -p "Explain microservices" | wc -w
|
||||||
|
qwen -p "List programming languages" | grep -i "python"
|
||||||
|
```
|
||||||
|
|
||||||
|
## Configuration Options
|
||||||
|
|
||||||
|
Key command-line options for headless usage:
|
||||||
|
|
||||||
|
| Option | Description | Example |
|
||||||
|
| ----------------------- | ---------------------------------- | ------------------------------------------------ |
|
||||||
|
| `--prompt`, `-p` | Run in headless mode | `qwen -p "query"` |
|
||||||
|
| `--output-format` | Specify output format (text, json) | `qwen -p "query" --output-format json` |
|
||||||
|
| `--model`, `-m` | Specify the Qwen model | `qwen -p "query" -m qwen3-coder-plus` |
|
||||||
|
| `--debug`, `-d` | Enable debug mode | `qwen -p "query" --debug` |
|
||||||
|
| `--all-files`, `-a` | Include all files in context | `qwen -p "query" --all-files` |
|
||||||
|
| `--include-directories` | Include additional directories | `qwen -p "query" --include-directories src,docs` |
|
||||||
|
| `--yolo`, `-y` | Auto-approve all actions | `qwen -p "query" --yolo` |
|
||||||
|
| `--approval-mode` | Set approval mode | `qwen -p "query" --approval-mode auto_edit` |
|
||||||
|
|
||||||
|
For complete details on all available configuration options, settings files, and environment variables, see the [Configuration Guide](./cli/configuration.md).
|
||||||
|
|
||||||
|
## Examples
|
||||||
|
|
||||||
|
#### Code review
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cat src/auth.py | qwen -p "Review this authentication code for security issues" > security-review.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Generate commit messages
|
||||||
|
|
||||||
|
```bash
|
||||||
|
result=$(git diff --cached | qwen -p "Write a concise commit message for these changes" --output-format json)
|
||||||
|
echo "$result" | jq -r '.response'
|
||||||
|
```
|
||||||
|
|
||||||
|
#### API documentation
|
||||||
|
|
||||||
|
```bash
|
||||||
|
result=$(cat api/routes.js | qwen -p "Generate OpenAPI spec for these routes" --output-format json)
|
||||||
|
echo "$result" | jq -r '.response' > openapi.json
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Batch code analysis
|
||||||
|
|
||||||
|
```bash
|
||||||
|
for file in src/*.py; do
|
||||||
|
echo "Analyzing $file..."
|
||||||
|
result=$(cat "$file" | qwen -p "Find potential bugs and suggest improvements" --output-format json)
|
||||||
|
echo "$result" | jq -r '.response' > "reports/$(basename "$file").analysis"
|
||||||
|
echo "Completed analysis for $(basename "$file")" >> reports/progress.log
|
||||||
|
done
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Code review
|
||||||
|
|
||||||
|
```bash
|
||||||
|
result=$(git diff origin/main...HEAD | qwen -p "Review these changes for bugs, security issues, and code quality" --output-format json)
|
||||||
|
echo "$result" | jq -r '.response' > pr-review.json
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Log analysis
|
||||||
|
|
||||||
|
```bash
|
||||||
|
grep "ERROR" /var/log/app.log | tail -20 | qwen -p "Analyze these errors and suggest root cause and fixes" > error-analysis.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Release notes generation
|
||||||
|
|
||||||
|
```bash
|
||||||
|
result=$(git log --oneline v1.0.0..HEAD | qwen -p "Generate release notes from these commits" --output-format json)
|
||||||
|
response=$(echo "$result" | jq -r '.response')
|
||||||
|
echo "$response"
|
||||||
|
echo "$response" >> CHANGELOG.md
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Model and tool usage tracking
|
||||||
|
|
||||||
|
```bash
|
||||||
|
result=$(qwen -p "Explain this database schema" --include-directories db --output-format json)
|
||||||
|
total_tokens=$(echo "$result" | jq -r '.stats.models // {} | to_entries | map(.value.tokens.total) | add // 0')
|
||||||
|
models_used=$(echo "$result" | jq -r '.stats.models // {} | keys | join(", ") | if . == "" then "none" else . end')
|
||||||
|
tool_calls=$(echo "$result" | jq -r '.stats.tools.totalCalls // 0')
|
||||||
|
tools_used=$(echo "$result" | jq -r '.stats.tools.byName // {} | keys | join(", ") | if . == "" then "none" else . end')
|
||||||
|
echo "$(date): $total_tokens tokens, $tool_calls tool calls ($tools_used) used with models: $models_used" >> usage.log
|
||||||
|
echo "$result" | jq -r '.response' > schema-docs.md
|
||||||
|
echo "Recent usage trends:"
|
||||||
|
tail -5 usage.log
|
||||||
|
```
|
||||||
|
|
||||||
|
## Resources
|
||||||
|
|
||||||
|
- [CLI Configuration](./cli/configuration.md) - Complete configuration guide
|
||||||
|
- [Authentication](./cli/authentication.md) - Setup authentication
|
||||||
|
- [Commands](./cli/commands.md) - Interactive commands reference
|
||||||
|
- [Tutorials](./cli/tutorials.md) - Step-by-step automation guides
|
||||||
184
docs/ide-companion-spec.md
Normal file
184
docs/ide-companion-spec.md
Normal file
@@ -0,0 +1,184 @@
|
|||||||
|
# Qwen Code Companion Plugin: Interface Specification
|
||||||
|
|
||||||
|
> Last Updated: September 15, 2025
|
||||||
|
|
||||||
|
This document defines the contract for building a companion plugin to enable Qwen Code's IDE mode. For VS Code, these features (native diffing, context awareness) are provided by the official extension ([marketplace](https://marketplace.visualstudio.com/items?itemName=qwenlm.qwen-code-vscode-ide-companion)). This specification is for contributors who wish to bring similar functionality to other editors like JetBrains IDEs, Sublime Text, etc.
|
||||||
|
|
||||||
|
## I. The Communication Interface
|
||||||
|
|
||||||
|
Qwen Code and the IDE plugin communicate through a local communication channel.
|
||||||
|
|
||||||
|
### 1. Transport Layer: MCP over HTTP
|
||||||
|
|
||||||
|
The plugin **MUST** run a local HTTP server that implements the **Model Context Protocol (MCP)**.
|
||||||
|
|
||||||
|
- **Protocol:** The server must be a valid MCP server. We recommend using an existing MCP SDK for your language of choice if available.
|
||||||
|
- **Endpoint:** The server should expose a single endpoint (e.g., `/mcp`) for all MCP communication.
|
||||||
|
- **Port:** The server **MUST** listen on a dynamically assigned port (i.e., listen on port `0`).
|
||||||
|
|
||||||
|
### 2. Discovery Mechanism: The Port File
|
||||||
|
|
||||||
|
For Qwen Code to connect, it needs to discover which IDE instance it's running in and what port your server is using. The plugin **MUST** facilitate this by creating a "discovery file."
|
||||||
|
|
||||||
|
- **How the CLI Finds the File:** The CLI determines the Process ID (PID) of the IDE it's running in by traversing the process tree. It then looks for a discovery file that contains this PID in its name.
|
||||||
|
- **File Location:** The file must be created in a specific directory: `os.tmpdir()/qwen/ide/`. Your plugin must create this directory if it doesn't exist.
|
||||||
|
- **File Naming Convention:** The filename is critical and **MUST** follow the pattern:
|
||||||
|
`qwen-code-ide-server-${PID}-${PORT}.json`
|
||||||
|
- `${PID}`: The process ID of the parent IDE process. Your plugin must determine this PID and include it in the filename.
|
||||||
|
- `${PORT}`: The port your MCP server is listening on.
|
||||||
|
- **File Content & Workspace Validation:** The file **MUST** contain a JSON object with the following structure:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"port": 12345,
|
||||||
|
"workspacePath": "/path/to/project1:/path/to/project2",
|
||||||
|
"authToken": "a-very-secret-token",
|
||||||
|
"ideInfo": {
|
||||||
|
"name": "vscode",
|
||||||
|
"displayName": "VS Code"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
- `port` (number, required): The port of the MCP server.
|
||||||
|
- `workspacePath` (string, required): A list of all open workspace root paths, delimited by the OS-specific path separator (`:` for Linux/macOS, `;` for Windows). The CLI uses this path to ensure it's running in the same project folder that's open in the IDE. If the CLI's current working directory is not a sub-directory of `workspacePath`, the connection will be rejected. Your plugin **MUST** provide the correct, absolute path(s) to the root of the open workspace(s).
|
||||||
|
- `authToken` (string, required): A secret token for securing the connection. The CLI will include this token in an `Authorization: Bearer <token>` header on all requests.
|
||||||
|
- `ideInfo` (object, required): Information about the IDE.
|
||||||
|
- `name` (string, required): A short, lowercase identifier for the IDE (e.g., `vscode`, `jetbrains`).
|
||||||
|
- `displayName` (string, required): A user-friendly name for the IDE (e.g., `VS Code`, `JetBrains IDE`).
|
||||||
|
|
||||||
|
- **Authentication:** To secure the connection, the plugin **MUST** generate a unique, secret token and include it in the discovery file. The CLI will then include this token in the `Authorization` header for all requests to the MCP server (e.g., `Authorization: Bearer a-very-secret-token`). Your server **MUST** validate this token on every request and reject any that are unauthorized.
|
||||||
|
- **Tie-Breaking with Environment Variables (Recommended):** For the most reliable experience, your plugin **SHOULD** both create the discovery file and set the `QWEN_CODE_IDE_SERVER_PORT` environment variable in the integrated terminal. The file serves as the primary discovery mechanism, but the environment variable is crucial for tie-breaking. If a user has multiple IDE windows open for the same workspace, the CLI uses the `QWEN_CODE_IDE_SERVER_PORT` variable to identify and connect to the correct window's server.
|
||||||
|
|
||||||
|
## II. The Context Interface
|
||||||
|
|
||||||
|
To enable context awareness, the plugin **MAY** provide the CLI with real-time information about the user's activity in the IDE.
|
||||||
|
|
||||||
|
### `ide/contextUpdate` Notification
|
||||||
|
|
||||||
|
The plugin **MAY** send an `ide/contextUpdate` [notification](https://modelcontextprotocol.io/specification/2025-06-18/basic/index#notifications) to the CLI whenever the user's context changes.
|
||||||
|
|
||||||
|
- **Triggering Events:** This notification should be sent (with a recommended debounce of 50ms) when:
|
||||||
|
- A file is opened, closed, or focused.
|
||||||
|
- The user's cursor position or text selection changes in the active file.
|
||||||
|
- **Payload (`IdeContext`):** The notification parameters **MUST** be an `IdeContext` object:
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
interface IdeContext {
|
||||||
|
workspaceState?: {
|
||||||
|
openFiles?: File[];
|
||||||
|
isTrusted?: boolean;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
interface File {
|
||||||
|
// Absolute path to the file
|
||||||
|
path: string;
|
||||||
|
// Last focused Unix timestamp (for ordering)
|
||||||
|
timestamp: number;
|
||||||
|
// True if this is the currently focused file
|
||||||
|
isActive?: boolean;
|
||||||
|
cursor?: {
|
||||||
|
// 1-based line number
|
||||||
|
line: number;
|
||||||
|
// 1-based character number
|
||||||
|
character: number;
|
||||||
|
};
|
||||||
|
// The text currently selected by the user
|
||||||
|
selectedText?: string;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
**Note:** The `openFiles` list should only include files that exist on disk. Virtual files (e.g., unsaved files without a path, editor settings pages) **MUST** be excluded.
|
||||||
|
|
||||||
|
### How the CLI Uses This Context
|
||||||
|
|
||||||
|
After receiving the `IdeContext` object, the CLI performs several normalization and truncation steps before sending the information to the model.
|
||||||
|
|
||||||
|
- **File Ordering:** The CLI uses the `timestamp` field to determine the most recently used files. It sorts the `openFiles` list based on this value. Therefore, your plugin **MUST** provide an accurate Unix timestamp for when a file was last focused.
|
||||||
|
- **Active File:** The CLI considers only the most recent file (after sorting) to be the "active" file. It will ignore the `isActive` flag on all other files and clear their `cursor` and `selectedText` fields. Your plugin should focus on setting `isActive: true` and providing cursor/selection details only for the currently focused file.
|
||||||
|
- **Truncation:** To manage token limits, the CLI truncates both the file list (to 10 files) and the `selectedText` (to 16KB).
|
||||||
|
|
||||||
|
While the CLI handles the final truncation, it is highly recommended that your plugin also limits the amount of context it sends.
|
||||||
|
|
||||||
|
## III. The Diffing Interface
|
||||||
|
|
||||||
|
To enable interactive code modifications, the plugin **MAY** expose a diffing interface. This allows the CLI to request that the IDE open a diff view, showing proposed changes to a file. The user can then review, edit, and ultimately accept or reject these changes directly within the IDE.
|
||||||
|
|
||||||
|
### `openDiff` Tool
|
||||||
|
|
||||||
|
The plugin **MUST** register an `openDiff` tool on its MCP server.
|
||||||
|
|
||||||
|
- **Description:** This tool instructs the IDE to open a modifiable diff view for a specific file.
|
||||||
|
- **Request (`OpenDiffRequest`):** The tool is invoked via a `tools/call` request. The `arguments` field within the request's `params` **MUST** be an `OpenDiffRequest` object.
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
interface OpenDiffRequest {
|
||||||
|
// The absolute path to the file to be diffed.
|
||||||
|
filePath: string;
|
||||||
|
// The proposed new content for the file.
|
||||||
|
newContent: string;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **Response (`CallToolResult`):** The tool **MUST** immediately return a `CallToolResult` to acknowledge the request and report whether the diff view was successfully opened.
|
||||||
|
- On Success: If the diff view was opened successfully, the response **MUST** contain empty content (i.e., `content: []`).
|
||||||
|
- On Failure: If an error prevented the diff view from opening, the response **MUST** have `isError: true` and include a `TextContent` block in the `content` array describing the error.
|
||||||
|
|
||||||
|
The actual outcome of the diff (acceptance or rejection) is communicated asynchronously via notifications.
|
||||||
|
|
||||||
|
### `closeDiff` Tool
|
||||||
|
|
||||||
|
The plugin **MUST** register a `closeDiff` tool on its MCP server.
|
||||||
|
|
||||||
|
- **Description:** This tool instructs the IDE to close an open diff view for a specific file.
|
||||||
|
- **Request (`CloseDiffRequest`):** The tool is invoked via a `tools/call` request. The `arguments` field within the request's `params` **MUST** be an `CloseDiffRequest` object.
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
interface CloseDiffRequest {
|
||||||
|
// The absolute path to the file whose diff view should be closed.
|
||||||
|
filePath: string;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **Response (`CallToolResult`):** The tool **MUST** return a `CallToolResult`.
|
||||||
|
- On Success: If the diff view was closed successfully, the response **MUST** include a single **TextContent** block in the content array containing the file's final content before closing.
|
||||||
|
- On Failure: If an error prevented the diff view from closing, the response **MUST** have `isError: true` and include a `TextContent` block in the `content` array describing the error.
|
||||||
|
|
||||||
|
### `ide/diffAccepted` Notification
|
||||||
|
|
||||||
|
When the user accepts the changes in a diff view (e.g., by clicking an "Apply" or "Save" button), the plugin **MUST** send an `ide/diffAccepted` notification to the CLI.
|
||||||
|
|
||||||
|
- **Payload:** The notification parameters **MUST** include the file path and the final content of the file. The content may differ from the original `newContent` if the user made manual edits in the diff view.
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
{
|
||||||
|
// The absolute path to the file that was diffed.
|
||||||
|
filePath: string;
|
||||||
|
// The full content of the file after acceptance.
|
||||||
|
content: string;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### `ide/diffRejected` Notification
|
||||||
|
|
||||||
|
When the user rejects the changes (e.g., by closing the diff view without accepting), the plugin **MUST** send an `ide/diffRejected` notification to the CLI.
|
||||||
|
|
||||||
|
- **Payload:** The notification parameters **MUST** include the file path of the rejected diff.
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
{
|
||||||
|
// The absolute path to the file that was diffed.
|
||||||
|
filePath: string;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## IV. The Lifecycle Interface
|
||||||
|
|
||||||
|
The plugin **MUST** manage its resources and the discovery file correctly based on the IDE's lifecycle.
|
||||||
|
|
||||||
|
- **On Activation (IDE startup/plugin enabled):**
|
||||||
|
1. Start the MCP server.
|
||||||
|
2. Create the discovery file.
|
||||||
|
- **On Deactivation (IDE shutdown/plugin disabled):**
|
||||||
|
1. Stop the MCP server.
|
||||||
|
2. Delete the discovery file.
|
||||||
@@ -2,7 +2,7 @@
|
|||||||
|
|
||||||
Qwen Code can integrate with your IDE to provide a more seamless and context-aware experience. This integration allows the CLI to understand your workspace better and enables powerful features like native in-editor diffing.
|
Qwen Code can integrate with your IDE to provide a more seamless and context-aware experience. This integration allows the CLI to understand your workspace better and enables powerful features like native in-editor diffing.
|
||||||
|
|
||||||
Currently, the only supported IDE is [Visual Studio Code](https://code.visualstudio.com/) and other editors that support VS Code extensions.
|
Currently, the only supported IDE is [Visual Studio Code](https://code.visualstudio.com/) and other editors that support VS Code extensions. To build support for other editors, see the [IDE Companion Extension Spec](./ide-companion-spec.md).
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
|
|||||||
@@ -19,7 +19,9 @@ This documentation is organized into the following sections:
|
|||||||
- **[Checkpointing](./checkpointing.md):** Documentation for the checkpointing feature.
|
- **[Checkpointing](./checkpointing.md):** Documentation for the checkpointing feature.
|
||||||
- **[Extensions](./extension.md):** How to extend the CLI with new functionality.
|
- **[Extensions](./extension.md):** How to extend the CLI with new functionality.
|
||||||
- **[IDE Integration](./ide-integration.md):** Connect the CLI to your editor.
|
- **[IDE Integration](./ide-integration.md):** Connect the CLI to your editor.
|
||||||
|
- **[IDE Companion Extension Spec](./ide-companion-spec.md):** Spec for building IDE companion extensions.
|
||||||
- **[Telemetry](./telemetry.md):** Overview of telemetry in the CLI.
|
- **[Telemetry](./telemetry.md):** Overview of telemetry in the CLI.
|
||||||
|
- **[Trusted Folders](./trusted-folders.md):** An overview of the Trusted Folders security feature.
|
||||||
- **Core Details:** Documentation for `packages/core`.
|
- **Core Details:** Documentation for `packages/core`.
|
||||||
- **[Core Introduction](./core/index.md):** Overview of the core component.
|
- **[Core Introduction](./core/index.md):** Overview of the core component.
|
||||||
- **[Tools API](./core/tools-api.md):** Information on how the core manages and exposes tools.
|
- **[Tools API](./core/tools-api.md):** Information on how the core manages and exposes tools.
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ npm run test:e2e
|
|||||||
|
|
||||||
## Running a specific set of tests
|
## Running a specific set of tests
|
||||||
|
|
||||||
To run a subset of test files, you can use `npm run <integration test command> <file_name1> ....` where <integration test command> is either `test:e2e` or `test:integration*` and `<file_name>` is any of the `.test.js` files in the `integration-tests/` directory. For example, the following command runs `list_directory.test.js` and `write_file.test.js`:
|
To run a subset of test files, you can use `npm run <integration test command> <file_name1> ....` where <integration test command> is either `test:e2e` or `test:integration*` and `<file_name>` is any of the `.test.js` files in the `integration-tests/` directory. For example, the following command runs `list_directory.test.js` and `write_file.test.js`:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
npm run test:e2e list_directory write_file
|
npm run test:e2e list_directory write_file
|
||||||
|
|||||||
@@ -16,10 +16,10 @@ Here is a breakdown of the specific automation workflows that run in our reposit
|
|||||||
|
|
||||||
This is the first bot you will interact with when you create an issue. Its job is to perform an initial analysis and apply the correct labels.
|
This is the first bot you will interact with when you create an issue. Its job is to perform an initial analysis and apply the correct labels.
|
||||||
|
|
||||||
- **Workflow File**: `.github/workflows/gemini-automated-issue-triage.yml`
|
- **Workflow File**: `.github/workflows/qwen-automated-issue-triage.yml`
|
||||||
- **When it runs**: Immediately after an issue is created or reopened.
|
- **When it runs**: Immediately after an issue is created or reopened.
|
||||||
- **What it does**:
|
- **What it does**:
|
||||||
- It uses a Gemini model to analyze the issue's title and body against a detailed set of guidelines.
|
- It uses a Qwen model to analyze the issue's title and body against a detailed set of guidelines.
|
||||||
- **Applies one `area/*` label**: Categorizes the issue into a functional area of the project (e.g., `area/ux`, `area/models`, `area/platform`).
|
- **Applies one `area/*` label**: Categorizes the issue into a functional area of the project (e.g., `area/ux`, `area/models`, `area/platform`).
|
||||||
- **Applies one `kind/*` label**: Identifies the type of issue (e.g., `kind/bug`, `kind/enhancement`, `kind/question`).
|
- **Applies one `kind/*` label**: Identifies the type of issue (e.g., `kind/bug`, `kind/enhancement`, `kind/question`).
|
||||||
- **Applies one `priority/*` label**: Assigns a priority from P0 (critical) to P3 (low) based on the described impact.
|
- **Applies one `priority/*` label**: Assigns a priority from P0 (critical) to P3 (low) based on the described impact.
|
||||||
@@ -47,7 +47,7 @@ This workflow ensures that all changes meet our quality standards before they ca
|
|||||||
|
|
||||||
This workflow runs periodically to ensure all open PRs are correctly linked to issues and have consistent labels.
|
This workflow runs periodically to ensure all open PRs are correctly linked to issues and have consistent labels.
|
||||||
|
|
||||||
- **Workflow File**: `.github/workflows/gemini-scheduled-pr-triage.yml`
|
- **Workflow File**: `.github/workflows/qwen-scheduled-pr-triage.yml`
|
||||||
- **When it runs**: Every 15 minutes on all open pull requests.
|
- **When it runs**: Every 15 minutes on all open pull requests.
|
||||||
- **What it does**:
|
- **What it does**:
|
||||||
- **Checks for a linked issue**: The bot scans your PR description for a keyword that links it to an issue (e.g., `Fixes #123`, `Closes #456`).
|
- **Checks for a linked issue**: The bot scans your PR description for a keyword that links it to an issue (e.g., `Fixes #123`, `Closes #456`).
|
||||||
@@ -61,11 +61,11 @@ This workflow runs periodically to ensure all open PRs are correctly linked to i
|
|||||||
|
|
||||||
This is a fallback workflow to ensure that no issue gets missed by the triage process.
|
This is a fallback workflow to ensure that no issue gets missed by the triage process.
|
||||||
|
|
||||||
- **Workflow File**: `.github/workflows/gemini-scheduled-issue-triage.yml`
|
- **Workflow File**: `.github/workflows/qwen-scheduled-issue-triage.yml`
|
||||||
- **When it runs**: Every hour on all open issues.
|
- **When it runs**: Every hour on all open issues.
|
||||||
- **What it does**:
|
- **What it does**:
|
||||||
- It actively seeks out issues that either have no labels at all or still have the `status/need-triage` label.
|
- It actively seeks out issues that either have no labels at all or still have the `status/need-triage` label.
|
||||||
- It then triggers the same powerful Gemini-based analysis as the initial triage bot to apply the correct labels.
|
- It then triggers the same powerful QwenCode-based analysis as the initial triage bot to apply the correct labels.
|
||||||
- **What you should do**:
|
- **What you should do**:
|
||||||
- You typically don't need to do anything. This workflow is a safety net to ensure every issue is eventually categorized, even if the initial triage fails.
|
- You typically don't need to do anything. This workflow is a safety net to ensure every issue is eventually categorized, even if the initial triage fails.
|
||||||
|
|
||||||
|
|||||||
103
docs/mermaid/context.mmd
Normal file
103
docs/mermaid/context.mmd
Normal file
@@ -0,0 +1,103 @@
|
|||||||
|
graph LR
|
||||||
|
%% --- Style Definitions ---
|
||||||
|
classDef new fill:#98fb98,color:#000
|
||||||
|
classDef changed fill:#add8e6,color:#000
|
||||||
|
classDef unchanged fill:#f0f0f0,color:#000
|
||||||
|
|
||||||
|
%% --- Subgraphs ---
|
||||||
|
subgraph "Context Providers"
|
||||||
|
direction TB
|
||||||
|
A["gemini.tsx"]
|
||||||
|
B["AppContainer.tsx"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "Contexts"
|
||||||
|
direction TB
|
||||||
|
CtxSession["SessionContext"]
|
||||||
|
CtxVim["VimModeContext"]
|
||||||
|
CtxSettings["SettingsContext"]
|
||||||
|
CtxApp["AppContext"]
|
||||||
|
CtxConfig["ConfigContext"]
|
||||||
|
CtxUIState["UIStateContext"]
|
||||||
|
CtxUIActions["UIActionsContext"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "Component Consumers"
|
||||||
|
direction TB
|
||||||
|
ConsumerApp["App"]
|
||||||
|
ConsumerAppContainer["AppContainer"]
|
||||||
|
ConsumerAppHeader["AppHeader"]
|
||||||
|
ConsumerDialogManager["DialogManager"]
|
||||||
|
ConsumerHistoryItem["HistoryItemDisplay"]
|
||||||
|
ConsumerComposer["Composer"]
|
||||||
|
ConsumerMainContent["MainContent"]
|
||||||
|
ConsumerNotifications["Notifications"]
|
||||||
|
end
|
||||||
|
|
||||||
|
%% --- Provider -> Context Connections ---
|
||||||
|
A -.-> CtxSession
|
||||||
|
A -.-> CtxVim
|
||||||
|
A -.-> CtxSettings
|
||||||
|
|
||||||
|
B -.-> CtxApp
|
||||||
|
B -.-> CtxConfig
|
||||||
|
B -.-> CtxUIState
|
||||||
|
B -.-> CtxUIActions
|
||||||
|
B -.-> CtxSettings
|
||||||
|
|
||||||
|
%% --- Context -> Consumer Connections ---
|
||||||
|
CtxSession -.-> ConsumerAppContainer
|
||||||
|
CtxSession -.-> ConsumerApp
|
||||||
|
|
||||||
|
CtxVim -.-> ConsumerAppContainer
|
||||||
|
CtxVim -.-> ConsumerComposer
|
||||||
|
CtxVim -.-> ConsumerApp
|
||||||
|
|
||||||
|
CtxSettings -.-> ConsumerAppContainer
|
||||||
|
CtxSettings -.-> ConsumerAppHeader
|
||||||
|
CtxSettings -.-> ConsumerDialogManager
|
||||||
|
CtxSettings -.-> ConsumerApp
|
||||||
|
|
||||||
|
CtxApp -.-> ConsumerAppHeader
|
||||||
|
CtxApp -.-> ConsumerNotifications
|
||||||
|
|
||||||
|
CtxConfig -.-> ConsumerAppHeader
|
||||||
|
CtxConfig -.-> ConsumerHistoryItem
|
||||||
|
CtxConfig -.-> ConsumerComposer
|
||||||
|
CtxConfig -.-> ConsumerDialogManager
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
CtxUIState -.-> ConsumerApp
|
||||||
|
CtxUIState -.-> ConsumerMainContent
|
||||||
|
CtxUIState -.-> ConsumerComposer
|
||||||
|
CtxUIState -.-> ConsumerDialogManager
|
||||||
|
|
||||||
|
CtxUIActions -.-> ConsumerComposer
|
||||||
|
CtxUIActions -.-> ConsumerDialogManager
|
||||||
|
|
||||||
|
%% --- Apply Styles ---
|
||||||
|
%% New Elements (Green)
|
||||||
|
class B,CtxApp,CtxConfig,CtxUIState,CtxUIActions,ConsumerAppHeader,ConsumerDialogManager,ConsumerComposer,ConsumerMainContent,ConsumerNotifications new
|
||||||
|
|
||||||
|
%% Heavily Changed Elements (Blue)
|
||||||
|
class A,ConsumerApp,ConsumerAppContainer,ConsumerHistoryItem changed
|
||||||
|
|
||||||
|
%% Mostly Unchanged Elements (Gray)
|
||||||
|
class CtxSession,CtxVim,CtxSettings unchanged
|
||||||
|
|
||||||
|
%% --- Link Styles ---
|
||||||
|
%% CtxSession (Red)
|
||||||
|
linkStyle 0,8,9 stroke:#e57373,stroke-width:2px
|
||||||
|
%% CtxVim (Orange)
|
||||||
|
linkStyle 1,10,11,12 stroke:#ffb74d,stroke-width:2px
|
||||||
|
%% CtxSettings (Yellow)
|
||||||
|
linkStyle 2,7,13,14,15,16 stroke:#fff176,stroke-width:2px
|
||||||
|
%% CtxApp (Green)
|
||||||
|
linkStyle 3,17,18 stroke:#81c784,stroke-width:2px
|
||||||
|
%% CtxConfig (Blue)
|
||||||
|
linkStyle 4,19,20,21,22 stroke:#64b5f6,stroke-width:2px
|
||||||
|
%% CtxUIState (Indigo)
|
||||||
|
linkStyle 5,23,24,25,26 stroke:#7986cb,stroke-width:2px
|
||||||
|
%% CtxUIActions (Violet)
|
||||||
|
linkStyle 6,27,28 stroke:#ba68c8,stroke-width:2px
|
||||||
64
docs/mermaid/render-path.mmd
Normal file
64
docs/mermaid/render-path.mmd
Normal file
@@ -0,0 +1,64 @@
|
|||||||
|
graph TD
|
||||||
|
%% --- Style Definitions ---
|
||||||
|
classDef new fill:#98fb98,color:#000
|
||||||
|
classDef changed fill:#add8e6,color:#000
|
||||||
|
classDef unchanged fill:#f0f0f0,color:#000
|
||||||
|
classDef dispatcher fill:#f9e79f,color:#000,stroke:#333,stroke-width:1px
|
||||||
|
classDef container fill:#f5f5f5,color:#000,stroke:#ccc
|
||||||
|
|
||||||
|
%% --- Component Tree ---
|
||||||
|
subgraph "Entry Point"
|
||||||
|
A["gemini.tsx"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "State & Logic Wrapper"
|
||||||
|
B["AppContainer.tsx"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "Primary Layout"
|
||||||
|
C["App.tsx"]
|
||||||
|
end
|
||||||
|
|
||||||
|
A -.-> B
|
||||||
|
B -.-> C
|
||||||
|
|
||||||
|
subgraph "UI Containers"
|
||||||
|
direction LR
|
||||||
|
C -.-> D["MainContent"]
|
||||||
|
C -.-> G["Composer"]
|
||||||
|
C -.-> F["DialogManager"]
|
||||||
|
C -.-> E["Notifications"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "MainContent"
|
||||||
|
direction TB
|
||||||
|
D -.-> H["AppHeader"]
|
||||||
|
D -.-> I["HistoryItemDisplay"]:::dispatcher
|
||||||
|
D -.-> L["ShowMoreLines"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "Composer"
|
||||||
|
direction TB
|
||||||
|
G -.-> K_Prompt["InputPrompt"]
|
||||||
|
G -.-> K_Footer["Footer"]
|
||||||
|
end
|
||||||
|
|
||||||
|
subgraph "DialogManager"
|
||||||
|
F -.-> J["Various Dialogs<br>(Auth, Theme, Settings, etc.)"]
|
||||||
|
end
|
||||||
|
|
||||||
|
%% --- Apply Styles ---
|
||||||
|
class B,D,E,F,G,H,J,K_Prompt,L new
|
||||||
|
class A,C,I changed
|
||||||
|
class K_Footer unchanged
|
||||||
|
|
||||||
|
%% --- Link Styles ---
|
||||||
|
%% MainContent Branch (Blue)
|
||||||
|
linkStyle 2,6,7,8 stroke:#64b5f6,stroke-width:2px
|
||||||
|
%% Composer Branch (Green)
|
||||||
|
linkStyle 3,9,10 stroke:#81c784,stroke-width:2px
|
||||||
|
%% DialogManager Branch (Orange)
|
||||||
|
linkStyle 4,11 stroke:#ffb74d,stroke-width:2px
|
||||||
|
%% Notifications Branch (Violet)
|
||||||
|
linkStyle 5 stroke:#ba68c8,stroke-width:2px
|
||||||
|
|
||||||
@@ -55,8 +55,10 @@ qwen -p "run the test suite"
|
|||||||
|
|
||||||
# Configure in settings.json
|
# Configure in settings.json
|
||||||
{
|
{
|
||||||
|
"tools": {
|
||||||
"sandbox": "docker"
|
"sandbox": "docker"
|
||||||
}
|
}
|
||||||
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
## Configuration
|
## Configuration
|
||||||
@@ -65,7 +67,7 @@ qwen -p "run the test suite"
|
|||||||
|
|
||||||
1. **Command flag**: `-s` or `--sandbox`
|
1. **Command flag**: `-s` or `--sandbox`
|
||||||
2. **Environment variable**: `GEMINI_SANDBOX=true|docker|podman|sandbox-exec`
|
2. **Environment variable**: `GEMINI_SANDBOX=true|docker|podman|sandbox-exec`
|
||||||
3. **Settings file**: `"sandbox": true` in `settings.json`
|
3. **Settings file**: `"sandbox": true` in the `tools` object of your `settings.json` file (e.g., `{"tools": {"sandbox": true}}`).
|
||||||
|
|
||||||
### macOS Seatbelt profiles
|
### macOS Seatbelt profiles
|
||||||
|
|
||||||
|
|||||||
68
docs/sidebar.json
Normal file
68
docs/sidebar.json
Normal file
@@ -0,0 +1,68 @@
|
|||||||
|
[
|
||||||
|
{
|
||||||
|
"label": "Overview",
|
||||||
|
"items": [
|
||||||
|
{ "label": "Welcome", "slug": "docs" },
|
||||||
|
{ "label": "Execution and Deployment", "slug": "docs/deployment" },
|
||||||
|
{ "label": "Architecture Overview", "slug": "docs/architecture" }
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "CLI",
|
||||||
|
"items": [
|
||||||
|
{ "label": "Introduction", "slug": "docs/cli" },
|
||||||
|
{ "label": "Authentication", "slug": "docs/cli/authentication" },
|
||||||
|
{ "label": "Commands", "slug": "docs/cli/commands" },
|
||||||
|
{ "label": "Configuration", "slug": "docs/cli/configuration" },
|
||||||
|
{ "label": "Checkpointing", "slug": "docs/checkpointing" },
|
||||||
|
{ "label": "Extensions", "slug": "docs/extension" },
|
||||||
|
{ "label": "Headless Mode", "slug": "docs/headless" },
|
||||||
|
{ "label": "IDE Integration", "slug": "docs/ide-integration" },
|
||||||
|
{
|
||||||
|
"label": "IDE Companion Spec",
|
||||||
|
"slug": "docs/ide-companion-spec"
|
||||||
|
},
|
||||||
|
{ "label": "Telemetry", "slug": "docs/telemetry" },
|
||||||
|
{ "label": "Themes", "slug": "docs/cli/themes" },
|
||||||
|
{ "label": "Token Caching", "slug": "docs/cli/token-caching" },
|
||||||
|
{ "label": "Trusted Folders", "slug": "docs/trusted-folders" },
|
||||||
|
{ "label": "Tutorials", "slug": "docs/cli/tutorials" }
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "Core",
|
||||||
|
"items": [
|
||||||
|
{ "label": "Introduction", "slug": "docs/core" },
|
||||||
|
{ "label": "Tools API", "slug": "docs/core/tools-api" },
|
||||||
|
{ "label": "Memory Import Processor", "slug": "docs/core/memport" }
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "Tools",
|
||||||
|
"items": [
|
||||||
|
{ "label": "Overview", "slug": "docs/tools" },
|
||||||
|
{ "label": "File System", "slug": "docs/tools/file-system" },
|
||||||
|
{ "label": "Multi-File Read", "slug": "docs/tools/multi-file" },
|
||||||
|
{ "label": "Shell", "slug": "docs/tools/shell" },
|
||||||
|
{ "label": "Web Fetch", "slug": "docs/tools/web-fetch" },
|
||||||
|
{ "label": "Web Search", "slug": "docs/tools/web-search" },
|
||||||
|
{ "label": "Memory", "slug": "docs/tools/memory" },
|
||||||
|
{ "label": "MCP Servers", "slug": "docs/tools/mcp-server" },
|
||||||
|
{ "label": "Sandboxing", "slug": "docs/sandbox" }
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "Development",
|
||||||
|
"items": [
|
||||||
|
{ "label": "NPM", "slug": "docs/npm" },
|
||||||
|
{ "label": "Releases", "slug": "docs/releases" }
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"label": "Support",
|
||||||
|
"items": [
|
||||||
|
{ "label": "Troubleshooting", "slug": "docs/troubleshooting" },
|
||||||
|
{ "label": "Terms of Service", "slug": "docs/tos-privacy" }
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
@@ -1,158 +1,206 @@
|
|||||||
# Qwen Code Observability Guide
|
# Observability with OpenTelemetry
|
||||||
|
|
||||||
Telemetry provides data about Qwen Code's performance, health, and usage. By enabling it, you can monitor operations, debug issues, and optimize tool usage through traces, metrics, and structured logs.
|
Learn how to enable and setup OpenTelemetry for Qwen Code.
|
||||||
|
|
||||||
Qwen Code's telemetry system is built on the **[OpenTelemetry] (OTEL)** standard, allowing you to send data to any compatible backend.
|
- [Observability with OpenTelemetry](#observability-with-opentelemetry)
|
||||||
|
- [Key Benefits](#key-benefits)
|
||||||
|
- [OpenTelemetry Integration](#opentelemetry-integration)
|
||||||
|
- [Configuration](#configuration)
|
||||||
|
- [Google Cloud Telemetry](#google-cloud-telemetry)
|
||||||
|
- [Prerequisites](#prerequisites)
|
||||||
|
- [Direct Export (Recommended)](#direct-export-recommended)
|
||||||
|
- [Collector-Based Export (Advanced)](#collector-based-export-advanced)
|
||||||
|
- [Local Telemetry](#local-telemetry)
|
||||||
|
- [File-based Output (Recommended)](#file-based-output-recommended)
|
||||||
|
- [Collector-Based Export (Advanced)](#collector-based-export-advanced-1)
|
||||||
|
- [Logs and Metrics](#logs-and-metrics)
|
||||||
|
- [Logs](#logs)
|
||||||
|
- [Metrics](#metrics)
|
||||||
|
|
||||||
|
## Key Benefits
|
||||||
|
|
||||||
|
- **🔍 Usage Analytics**: Understand interaction patterns and feature adoption
|
||||||
|
across your team
|
||||||
|
- **⚡ Performance Monitoring**: Track response times, token consumption, and
|
||||||
|
resource utilization
|
||||||
|
- **🐛 Real-time Debugging**: Identify bottlenecks, failures, and error patterns
|
||||||
|
as they occur
|
||||||
|
- **📊 Workflow Optimization**: Make informed decisions to improve
|
||||||
|
configurations and processes
|
||||||
|
- **🏢 Enterprise Governance**: Monitor usage across teams, track costs, ensure
|
||||||
|
compliance, and integrate with existing monitoring infrastructure
|
||||||
|
|
||||||
|
## OpenTelemetry Integration
|
||||||
|
|
||||||
|
Built on **[OpenTelemetry]** — the vendor-neutral, industry-standard
|
||||||
|
observability framework — Qwen Code's observability system provides:
|
||||||
|
|
||||||
|
- **Universal Compatibility**: Export to any OpenTelemetry backend (Google
|
||||||
|
Cloud, Jaeger, Prometheus, Datadog, etc.)
|
||||||
|
- **Standardized Data**: Use consistent formats and collection methods across
|
||||||
|
your toolchain
|
||||||
|
- **Future-Proof Integration**: Connect with existing and future observability
|
||||||
|
infrastructure
|
||||||
|
- **No Vendor Lock-in**: Switch between backends without changing your
|
||||||
|
instrumentation
|
||||||
|
|
||||||
[OpenTelemetry]: https://opentelemetry.io/
|
[OpenTelemetry]: https://opentelemetry.io/
|
||||||
|
|
||||||
## Enabling telemetry
|
## Configuration
|
||||||
|
|
||||||
You can enable telemetry in multiple ways. Configuration is primarily managed via the [`.qwen/settings.json` file](./cli/configuration.md) and environment variables, but CLI flags can override these settings for a specific session.
|
All telemetry behavior is controlled through your `.qwen/settings.json` file.
|
||||||
|
These settings can be overridden by environment variables or CLI flags.
|
||||||
|
|
||||||
### Order of precedence
|
| Setting | Environment Variable | CLI Flag | Description | Values | Default |
|
||||||
|
| -------------- | -------------------------------- | -------------------------------------------------------- | ------------------------------------------------- | ----------------- | ----------------------- |
|
||||||
|
| `enabled` | `GEMINI_TELEMETRY_ENABLED` | `--telemetry` / `--no-telemetry` | Enable or disable telemetry | `true`/`false` | `false` |
|
||||||
|
| `target` | `GEMINI_TELEMETRY_TARGET` | `--telemetry-target <local\|gcp>` | Where to send telemetry data | `"gcp"`/`"local"` | `"local"` |
|
||||||
|
| `otlpEndpoint` | `GEMINI_TELEMETRY_OTLP_ENDPOINT` | `--telemetry-otlp-endpoint <URL>` | OTLP collector endpoint | URL string | `http://localhost:4317` |
|
||||||
|
| `otlpProtocol` | `GEMINI_TELEMETRY_OTLP_PROTOCOL` | `--telemetry-otlp-protocol <grpc\|http>` | OTLP transport protocol | `"grpc"`/`"http"` | `"grpc"` |
|
||||||
|
| `outfile` | `GEMINI_TELEMETRY_OUTFILE` | `--telemetry-outfile <path>` | Save telemetry to file (overrides `otlpEndpoint`) | file path | - |
|
||||||
|
| `logPrompts` | `GEMINI_TELEMETRY_LOG_PROMPTS` | `--telemetry-log-prompts` / `--no-telemetry-log-prompts` | Include prompts in telemetry logs | `true`/`false` | `true` |
|
||||||
|
| `useCollector` | `GEMINI_TELEMETRY_USE_COLLECTOR` | - | Use external OTLP collector (advanced) | `true`/`false` | `false` |
|
||||||
|
|
||||||
The following lists the precedence for applying telemetry settings, with items listed higher having greater precedence:
|
**Note on boolean environment variables:** For the boolean settings (`enabled`,
|
||||||
|
`logPrompts`, `useCollector`), setting the corresponding environment variable to
|
||||||
|
`true` or `1` will enable the feature. Any other value will disable it.
|
||||||
|
|
||||||
1. **CLI flags (for `qwen` command):**
|
For detailed information about all configuration options, see the
|
||||||
- `--telemetry` / `--no-telemetry`: Overrides `telemetry.enabled`.
|
[Configuration Guide](./cli/configuration.md).
|
||||||
- `--telemetry-target <local|gcp>`: Overrides `telemetry.target`.
|
|
||||||
- `--telemetry-otlp-endpoint <URL>`: Overrides `telemetry.otlpEndpoint`.
|
|
||||||
- `--telemetry-log-prompts` / `--no-telemetry-log-prompts`: Overrides `telemetry.logPrompts`.
|
|
||||||
- `--telemetry-outfile <path>`: Redirects telemetry output to a file. See [Exporting to a file](#exporting-to-a-file).
|
|
||||||
|
|
||||||
1. **Environment variables:**
|
## Google Cloud Telemetry
|
||||||
- `OTEL_EXPORTER_OTLP_ENDPOINT`: Overrides `telemetry.otlpEndpoint`.
|
|
||||||
|
|
||||||
1. **Workspace settings file (`.qwen/settings.json`):** Values from the `telemetry` object in this project-specific file.
|
### Prerequisites
|
||||||
|
|
||||||
1. **User settings file (`~/.qwen/settings.json`):** Values from the `telemetry` object in this global user file.
|
Before using either method below, complete these steps:
|
||||||
|
|
||||||
1. **Defaults:** applied if not set by any of the above.
|
1. Set your Google Cloud project ID:
|
||||||
- `telemetry.enabled`: `false`
|
- For telemetry in a separate project from inference:
|
||||||
- `telemetry.target`: `local`
|
```bash
|
||||||
- `telemetry.otlpEndpoint`: `http://localhost:4317`
|
export OTLP_GOOGLE_CLOUD_PROJECT="your-telemetry-project-id"
|
||||||
- `telemetry.logPrompts`: `true`
|
```
|
||||||
|
- For telemetry in the same project as inference:
|
||||||
|
```bash
|
||||||
|
export GOOGLE_CLOUD_PROJECT="your-project-id"
|
||||||
|
```
|
||||||
|
|
||||||
**For the `npm run telemetry -- --target=<gcp|local>` script:**
|
2. Authenticate with Google Cloud:
|
||||||
The `--target` argument to this script _only_ overrides the `telemetry.target` for the duration and purpose of that script (i.e., choosing which collector to start). It does not permanently change your `settings.json`. The script will first look at `settings.json` for a `telemetry.target` to use as its default.
|
- If using a user account:
|
||||||
|
```bash
|
||||||
|
gcloud auth application-default login
|
||||||
|
```
|
||||||
|
- If using a service account:
|
||||||
|
```bash
|
||||||
|
export GOOGLE_APPLICATION_CREDENTIALS="/path/to/your/service-account.json"
|
||||||
|
```
|
||||||
|
3. Make sure your account or service account has these IAM roles:
|
||||||
|
- Cloud Trace Agent
|
||||||
|
- Monitoring Metric Writer
|
||||||
|
- Logs Writer
|
||||||
|
|
||||||
### Example settings
|
4. Enable the required Google Cloud APIs (if not already enabled):
|
||||||
|
```bash
|
||||||
|
gcloud services enable \
|
||||||
|
cloudtrace.googleapis.com \
|
||||||
|
monitoring.googleapis.com \
|
||||||
|
logging.googleapis.com \
|
||||||
|
--project="$OTLP_GOOGLE_CLOUD_PROJECT"
|
||||||
|
```
|
||||||
|
|
||||||
The following code can be added to your workspace (`.qwen/settings.json`) or user (`~/.qwen/settings.json`) settings to enable telemetry and send the output to Google Cloud:
|
### Direct Export (Recommended)
|
||||||
|
|
||||||
|
Sends telemetry directly to Google Cloud services. No collector needed.
|
||||||
|
|
||||||
|
1. Enable telemetry in your `.qwen/settings.json`:
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"telemetry": {
|
"telemetry": {
|
||||||
"enabled": true,
|
"enabled": true,
|
||||||
"target": "gcp"
|
"target": "gcp"
|
||||||
},
|
}
|
||||||
"sandbox": false
|
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
2. Run Qwen Code and send prompts.
|
||||||
|
3. View logs and metrics:
|
||||||
|
- Open the Google Cloud Console in your browser after sending prompts:
|
||||||
|
- Logs: https://console.cloud.google.com/logs/
|
||||||
|
- Metrics: https://console.cloud.google.com/monitoring/metrics-explorer
|
||||||
|
- Traces: https://console.cloud.google.com/traces/list
|
||||||
|
|
||||||
### Exporting to a file
|
### Collector-Based Export (Advanced)
|
||||||
|
|
||||||
You can export all telemetry data to a file for local inspection.
|
For custom processing, filtering, or routing, use an OpenTelemetry collector to
|
||||||
|
forward data to Google Cloud.
|
||||||
|
|
||||||
To enable file export, use the `--telemetry-outfile` flag with a path to your desired output file. This must be run using `--telemetry-target=local`.
|
1. Configure your `.qwen/settings.json`:
|
||||||
|
```json
|
||||||
```bash
|
{
|
||||||
# Set your desired output file path
|
"telemetry": {
|
||||||
TELEMETRY_FILE=".qwen/telemetry.log"
|
"enabled": true,
|
||||||
|
"target": "gcp",
|
||||||
# Run Qwen Code with local telemetry
|
"useCollector": true
|
||||||
# NOTE: --telemetry-otlp-endpoint="" is required to override the default
|
}
|
||||||
# OTLP exporter and ensure telemetry is written to the local file.
|
}
|
||||||
qwen --telemetry \
|
|
||||||
--telemetry-target=local \
|
|
||||||
--telemetry-otlp-endpoint="" \
|
|
||||||
--telemetry-outfile="$TELEMETRY_FILE" \
|
|
||||||
--prompt "What is OpenTelemetry?"
|
|
||||||
```
|
```
|
||||||
|
2. Run the automation script:
|
||||||
## Running an OTEL Collector
|
|
||||||
|
|
||||||
An OTEL Collector is a service that receives, processes, and exports telemetry data.
|
|
||||||
The CLI can send data using either the OTLP/gRPC or OTLP/HTTP protocol.
|
|
||||||
You can specify which protocol to use via the `--telemetry-otlp-protocol` flag
|
|
||||||
or the `telemetry.otlpProtocol` setting in your `settings.json` file. See the
|
|
||||||
[configuration docs](./cli/configuration.md#--telemetry-otlp-protocol) for more
|
|
||||||
details.
|
|
||||||
|
|
||||||
Learn more about OTEL exporter standard configuration in [documentation][otel-config-docs].
|
|
||||||
|
|
||||||
[otel-config-docs]: https://opentelemetry.io/docs/languages/sdk-configuration/otlp-exporter/
|
|
||||||
|
|
||||||
### Local
|
|
||||||
|
|
||||||
Use the `npm run telemetry -- --target=local` command to automate the process of setting up a local telemetry pipeline, including configuring the necessary settings in your `.qwen/settings.json` file. The underlying script installs `otelcol-contrib` (the OpenTelemetry Collector) and `jaeger` (The Jaeger UI for viewing traces). To use it:
|
|
||||||
|
|
||||||
1. **Run the command**:
|
|
||||||
Execute the command from the root of the repository:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
npm run telemetry -- --target=local
|
|
||||||
```
|
|
||||||
|
|
||||||
The script will:
|
|
||||||
- Download Jaeger and OTEL if needed.
|
|
||||||
- Start a local Jaeger instance.
|
|
||||||
- Start an OTEL collector configured to receive data from Qwen Code.
|
|
||||||
- Automatically enable telemetry in your workspace settings.
|
|
||||||
- On exit, disable telemetry.
|
|
||||||
|
|
||||||
1. **View traces**:
|
|
||||||
Open your web browser and navigate to **http://localhost:16686** to access the Jaeger UI. Here you can inspect detailed traces of Qwen Code operations.
|
|
||||||
|
|
||||||
1. **Inspect logs and metrics**:
|
|
||||||
The script redirects the OTEL collector output (which includes logs and metrics) to `~/.qwen/tmp/<projectHash>/otel/collector.log`. The script will provide links to view and a command to tail your telemetry data (traces, metrics, logs) locally.
|
|
||||||
|
|
||||||
1. **Stop the services**:
|
|
||||||
Press `Ctrl+C` in the terminal where the script is running to stop the OTEL Collector and Jaeger services.
|
|
||||||
|
|
||||||
### Google Cloud
|
|
||||||
|
|
||||||
Use the `npm run telemetry -- --target=gcp` command to automate setting up a local OpenTelemetry collector that forwards data to your Google Cloud project, including configuring the necessary settings in your `.qwen/settings.json` file. The underlying script installs `otelcol-contrib`. To use it:
|
|
||||||
|
|
||||||
1. **Prerequisites**:
|
|
||||||
- Have a Google Cloud project ID.
|
|
||||||
- Export the `GOOGLE_CLOUD_PROJECT` environment variable to make it available to the OTEL collector.
|
|
||||||
```bash
|
|
||||||
export OTLP_GOOGLE_CLOUD_PROJECT="your-project-id"
|
|
||||||
```
|
|
||||||
- Authenticate with Google Cloud (e.g., run `gcloud auth application-default login` or ensure `GOOGLE_APPLICATION_CREDENTIALS` is set).
|
|
||||||
- Ensure your Google Cloud account/service account has the necessary IAM roles: "Cloud Trace Agent", "Monitoring Metric Writer", and "Logs Writer".
|
|
||||||
|
|
||||||
1. **Run the command**:
|
|
||||||
Execute the command from the root of the repository:
|
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
npm run telemetry -- --target=gcp
|
npm run telemetry -- --target=gcp
|
||||||
```
|
```
|
||||||
|
This will:
|
||||||
|
- Start a local OTEL collector that forwards to Google Cloud
|
||||||
|
- Configure your workspace
|
||||||
|
- Provide links to view traces, metrics, and logs in Google Cloud Console
|
||||||
|
- Save collector logs to `~/.qwen/tmp/<projectHash>/otel/collector-gcp.log`
|
||||||
|
- Stop collector on exit (e.g. `Ctrl+C`)
|
||||||
|
3. Run Qwen Code and send prompts.
|
||||||
|
4. View logs and metrics:
|
||||||
|
- Open the Google Cloud Console in your browser after sending prompts:
|
||||||
|
- Logs: https://console.cloud.google.com/logs/
|
||||||
|
- Metrics: https://console.cloud.google.com/monitoring/metrics-explorer
|
||||||
|
- Traces: https://console.cloud.google.com/traces/list
|
||||||
|
- Open `~/.qwen/tmp/<projectHash>/otel/collector-gcp.log` to view local
|
||||||
|
collector logs.
|
||||||
|
|
||||||
The script will:
|
## Local Telemetry
|
||||||
- Download the `otelcol-contrib` binary if needed.
|
|
||||||
- Start an OTEL collector configured to receive data from Qwen Code and export it to your specified Google Cloud project.
|
|
||||||
- Automatically enable telemetry and disable sandbox mode in your workspace settings (`.qwen/settings.json`).
|
|
||||||
- Provide direct links to view traces, metrics, and logs in your Google Cloud Console.
|
|
||||||
- On exit (Ctrl+C), it will attempt to restore your original telemetry and sandbox settings.
|
|
||||||
|
|
||||||
1. **Run Qwen Code:**
|
For local development and debugging, you can capture telemetry data locally:
|
||||||
In a separate terminal, run your Qwen Code commands. This generates telemetry data that the collector captures.
|
|
||||||
|
|
||||||
1. **View telemetry in Google Cloud**:
|
### File-based Output (Recommended)
|
||||||
Use the links provided by the script to navigate to the Google Cloud Console and view your traces, metrics, and logs.
|
|
||||||
|
|
||||||
1. **Inspect local collector logs**:
|
1. Enable telemetry in your `.qwen/settings.json`:
|
||||||
The script redirects the local OTEL collector output to `~/.qwen/tmp/<projectHash>/otel/collector-gcp.log`. The script provides links to view and command to tail your collector logs locally.
|
```json
|
||||||
|
{
|
||||||
|
"telemetry": {
|
||||||
|
"enabled": true,
|
||||||
|
"target": "local",
|
||||||
|
"otlpEndpoint": "",
|
||||||
|
"outfile": ".qwen/telemetry.log"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
2. Run Qwen Code and send prompts.
|
||||||
|
3. View logs and metrics in the specified file (e.g., `.qwen/telemetry.log`).
|
||||||
|
|
||||||
1. **Stop the service**:
|
### Collector-Based Export (Advanced)
|
||||||
Press `Ctrl+C` in the terminal where the script is running to stop the OTEL Collector.
|
|
||||||
|
|
||||||
## Logs and metric reference
|
1. Run the automation script:
|
||||||
|
```bash
|
||||||
|
npm run telemetry -- --target=local
|
||||||
|
```
|
||||||
|
This will:
|
||||||
|
- Download and start Jaeger and OTEL collector
|
||||||
|
- Configure your workspace for local telemetry
|
||||||
|
- Provide a Jaeger UI at http://localhost:16686
|
||||||
|
- Save logs/metrics to `~/.qwen/tmp/<projectHash>/otel/collector.log`
|
||||||
|
- Stop collector on exit (e.g. `Ctrl+C`)
|
||||||
|
2. Run Qwen Code and send prompts.
|
||||||
|
3. View traces at http://localhost:16686 and logs/metrics in the collector log
|
||||||
|
file.
|
||||||
|
|
||||||
The following section describes the structure of logs and metrics generated for Qwen Code.
|
## Logs and Metrics
|
||||||
|
|
||||||
|
The following section describes the structure of logs and metrics generated for
|
||||||
|
Qwen Code.
|
||||||
|
|
||||||
- A `sessionId` is included as a common attribute on all logs and metrics.
|
- A `sessionId` is included as a common attribute on all logs and metrics.
|
||||||
|
|
||||||
@@ -174,12 +222,14 @@ Logs are timestamped records of specific events. The following events are logged
|
|||||||
- `file_filtering_respect_git_ignore` (boolean)
|
- `file_filtering_respect_git_ignore` (boolean)
|
||||||
- `debug_mode` (boolean)
|
- `debug_mode` (boolean)
|
||||||
- `mcp_servers` (string)
|
- `mcp_servers` (string)
|
||||||
|
- `output_format` (string: "text" or "json")
|
||||||
|
|
||||||
- `qwen-code.user_prompt`: This event occurs when a user submits a prompt.
|
- `qwen-code.user_prompt`: This event occurs when a user submits a prompt.
|
||||||
- **Attributes**:
|
- **Attributes**:
|
||||||
- `prompt_length` (int)
|
- `prompt_length` (int)
|
||||||
- `prompt_id` (string)
|
- `prompt_id` (string)
|
||||||
- `prompt` (string, this attribute is excluded if `log_prompts_enabled` is configured to be `false`)
|
- `prompt` (string, this attribute is excluded if `log_prompts_enabled` is
|
||||||
|
configured to be `false`)
|
||||||
- `auth_type` (string)
|
- `auth_type` (string)
|
||||||
|
|
||||||
- `qwen-code.tool_call`: This event occurs for each function call.
|
- `qwen-code.tool_call`: This event occurs for each function call.
|
||||||
@@ -188,11 +238,27 @@ Logs are timestamped records of specific events. The following events are logged
|
|||||||
- `function_args`
|
- `function_args`
|
||||||
- `duration_ms`
|
- `duration_ms`
|
||||||
- `success` (boolean)
|
- `success` (boolean)
|
||||||
- `decision` (string: "accept", "reject", "auto_accept", or "modify", if applicable)
|
- `decision` (string: "accept", "reject", "auto_accept", or "modify", if
|
||||||
|
applicable)
|
||||||
- `error` (if applicable)
|
- `error` (if applicable)
|
||||||
- `error_type` (if applicable)
|
- `error_type` (if applicable)
|
||||||
|
- `content_length` (int, if applicable)
|
||||||
- `metadata` (if applicable, dictionary of string -> any)
|
- `metadata` (if applicable, dictionary of string -> any)
|
||||||
|
|
||||||
|
- `qwen-code.file_operation`: This event occurs for each file operation.
|
||||||
|
- **Attributes**:
|
||||||
|
- `tool_name` (string)
|
||||||
|
- `operation` (string: "create", "read", "update")
|
||||||
|
- `lines` (int, if applicable)
|
||||||
|
- `mimetype` (string, if applicable)
|
||||||
|
- `extension` (string, if applicable)
|
||||||
|
- `programming_language` (string, if applicable)
|
||||||
|
- `diff_stat` (json string, if applicable): A JSON string with the following members:
|
||||||
|
- `ai_added_lines` (int)
|
||||||
|
- `ai_removed_lines` (int)
|
||||||
|
- `user_added_lines` (int)
|
||||||
|
- `user_removed_lines` (int)
|
||||||
|
|
||||||
- `qwen-code.api_request`: This event occurs when making a request to Qwen API.
|
- `qwen-code.api_request`: This event occurs when making a request to Qwen API.
|
||||||
- **Attributes**:
|
- **Attributes**:
|
||||||
- `model`
|
- `model`
|
||||||
@@ -221,6 +287,19 @@ Logs are timestamped records of specific events. The following events are logged
|
|||||||
- `response_text` (if applicable)
|
- `response_text` (if applicable)
|
||||||
- `auth_type`
|
- `auth_type`
|
||||||
|
|
||||||
|
- `qwen-code.tool_output_truncated`: This event occurs when the output of a tool call is too large and gets truncated.
|
||||||
|
- **Attributes**:
|
||||||
|
- `tool_name` (string)
|
||||||
|
- `original_content_length` (int)
|
||||||
|
- `truncated_content_length` (int)
|
||||||
|
- `threshold` (int)
|
||||||
|
- `lines` (int)
|
||||||
|
- `prompt_id` (string)
|
||||||
|
|
||||||
|
- `qwen-code.malformed_json_response`: This event occurs when a `generateJson` response from Qwen API cannot be parsed as a json.
|
||||||
|
- **Attributes**:
|
||||||
|
- `model`
|
||||||
|
|
||||||
- `qwen-code.flash_fallback`: This event occurs when Qwen Code switches to flash as fallback.
|
- `qwen-code.flash_fallback`: This event occurs when Qwen Code switches to flash as fallback.
|
||||||
- **Attributes**:
|
- **Attributes**:
|
||||||
- `auth_type`
|
- `auth_type`
|
||||||
@@ -230,6 +309,15 @@ Logs are timestamped records of specific events. The following events are logged
|
|||||||
- `command` (string)
|
- `command` (string)
|
||||||
- `subcommand` (string, if applicable)
|
- `subcommand` (string, if applicable)
|
||||||
|
|
||||||
|
- `qwen-code.extension_enable`: This event occurs when an extension is enabled
|
||||||
|
- `qwen-code.extension_install`: This event occurs when an extension is installed
|
||||||
|
- **Attributes**:
|
||||||
|
- `extension_name` (string)
|
||||||
|
- `extension_version` (string)
|
||||||
|
- `extension_source` (string)
|
||||||
|
- `status` (string)
|
||||||
|
- `qwen-code.extension_uninstall`: This event occurs when an extension is uninstalled
|
||||||
|
|
||||||
### Metrics
|
### Metrics
|
||||||
|
|
||||||
Metrics are numerical measurements of behavior over time. The following metrics are collected for Qwen Code (metric names remain `qwen-code.*` for compatibility):
|
Metrics are numerical measurements of behavior over time. The following metrics are collected for Qwen Code (metric names remain `qwen-code.*` for compatibility):
|
||||||
@@ -269,8 +357,8 @@ Metrics are numerical measurements of behavior over time. The following metrics
|
|||||||
- `lines` (Int, if applicable): Number of lines in the file.
|
- `lines` (Int, if applicable): Number of lines in the file.
|
||||||
- `mimetype` (string, if applicable): Mimetype of the file.
|
- `mimetype` (string, if applicable): Mimetype of the file.
|
||||||
- `extension` (string, if applicable): File extension of the file.
|
- `extension` (string, if applicable): File extension of the file.
|
||||||
- `ai_added_lines` (Int, if applicable): Number of lines added/changed by AI.
|
- `model_added_lines` (Int, if applicable): Number of lines added/changed by the model.
|
||||||
- `ai_removed_lines` (Int, if applicable): Number of lines removed/changed by AI.
|
- `model_removed_lines` (Int, if applicable): Number of lines removed/changed by the model.
|
||||||
- `user_added_lines` (Int, if applicable): Number of lines added/changed by user in AI proposed changes.
|
- `user_added_lines` (Int, if applicable): Number of lines added/changed by user in AI proposed changes.
|
||||||
- `user_removed_lines` (Int, if applicable): Number of lines removed/changed by user in AI proposed changes.
|
- `user_removed_lines` (Int, if applicable): Number of lines removed/changed by user in AI proposed changes.
|
||||||
- `programming_language` (string, if applicable): The programming language of the file.
|
- `programming_language` (string, if applicable): The programming language of the file.
|
||||||
|
|||||||
@@ -51,7 +51,30 @@ Qwen Code uses the `mcpServers` configuration in your `settings.json` file to lo
|
|||||||
|
|
||||||
### Configure the MCP server in settings.json
|
### Configure the MCP server in settings.json
|
||||||
|
|
||||||
You can configure MCP servers at the global level in the `~/.qwen/settings.json` file or in your project's root directory, create or open the `.qwen/settings.json` file. Within the file, add the `mcpServers` configuration block.
|
You can configure MCP servers in your `settings.json` file in two main ways: through the top-level `mcpServers` object for specific server definitions, and through the `mcp` object for global settings that control server discovery and execution.
|
||||||
|
|
||||||
|
#### Global MCP Settings (`mcp`)
|
||||||
|
|
||||||
|
The `mcp` object in your `settings.json` allows you to define global rules for all MCP servers.
|
||||||
|
|
||||||
|
- **`mcp.serverCommand`** (string): A global command to start an MCP server.
|
||||||
|
- **`mcp.allowed`** (array of strings): A list of MCP server names to allow. If this is set, only servers from this list (matching the keys in the `mcpServers` object) will be connected to.
|
||||||
|
- **`mcp.excluded`** (array of strings): A list of MCP server names to exclude. Servers in this list will not be connected to.
|
||||||
|
|
||||||
|
**Example:**
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"mcp": {
|
||||||
|
"allowed": ["my-trusted-server"],
|
||||||
|
"excluded": ["experimental-server"]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
#### Server-Specific Configuration (`mcpServers`)
|
||||||
|
|
||||||
|
The `mcpServers` object is where you define each individual MCP server you want the CLI to connect to.
|
||||||
|
|
||||||
### Configuration Structure
|
### Configuration Structure
|
||||||
|
|
||||||
@@ -92,8 +115,10 @@ Each server configuration supports the following properties:
|
|||||||
- **`cwd`** (string): Working directory for Stdio transport
|
- **`cwd`** (string): Working directory for Stdio transport
|
||||||
- **`timeout`** (number): Request timeout in milliseconds (default: 600,000ms = 10 minutes)
|
- **`timeout`** (number): Request timeout in milliseconds (default: 600,000ms = 10 minutes)
|
||||||
- **`trust`** (boolean): When `true`, bypasses all tool call confirmations for this server (default: `false`)
|
- **`trust`** (boolean): When `true`, bypasses all tool call confirmations for this server (default: `false`)
|
||||||
- **`includeTools`** (string[]): List of tool names to include from this MCP server. When specified, only the tools listed here will be available from this server (whitelist behavior). If not specified, all tools from the server are enabled by default.
|
- **`includeTools`** (string[]): List of tool names to include from this MCP server. When specified, only the tools listed here will be available from this server (allowlist behavior). If not specified, all tools from the server are enabled by default.
|
||||||
- **`excludeTools`** (string[]): List of tool names to exclude from this MCP server. Tools listed here will not be available to the model, even if they are exposed by the server. **Note:** `excludeTools` takes precedence over `includeTools` - if a tool is in both lists, it will be excluded.
|
- **`excludeTools`** (string[]): List of tool names to exclude from this MCP server. Tools listed here will not be available to the model, even if they are exposed by the server. **Note:** `excludeTools` takes precedence over `includeTools` - if a tool is in both lists, it will be excluded.
|
||||||
|
- **`targetAudience`** (string): The OAuth Client ID allowlisted on the IAP-protected application you are trying to access. Used with `authProviderType: 'service_account_impersonation'`.
|
||||||
|
- **`targetServiceAccount`** (string): The email address of the Google Cloud Service Account to impersonate. Used with `authProviderType: 'service_account_impersonation'`.
|
||||||
|
|
||||||
### OAuth Support for Remote MCP Servers
|
### OAuth Support for Remote MCP Servers
|
||||||
|
|
||||||
@@ -187,6 +212,9 @@ You can specify the authentication provider type using the `authProviderType` pr
|
|||||||
- **`authProviderType`** (string): Specifies the authentication provider. Can be one of the following:
|
- **`authProviderType`** (string): Specifies the authentication provider. Can be one of the following:
|
||||||
- **`dynamic_discovery`** (default): The CLI will automatically discover the OAuth configuration from the server.
|
- **`dynamic_discovery`** (default): The CLI will automatically discover the OAuth configuration from the server.
|
||||||
- **`google_credentials`**: The CLI will use the Google Application Default Credentials (ADC) to authenticate with the server. When using this provider, you must specify the required scopes.
|
- **`google_credentials`**: The CLI will use the Google Application Default Credentials (ADC) to authenticate with the server. When using this provider, you must specify the required scopes.
|
||||||
|
- **`service_account_impersonation`**: The CLI will impersonate a Google Cloud Service Account to authenticate with the server. This is useful for accessing IAP-protected services (this was specifically designed for Cloud Run services).
|
||||||
|
|
||||||
|
#### Google Credentials
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
@@ -202,6 +230,24 @@ You can specify the authentication provider type using the `authProviderType` pr
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
#### Service Account Impersonation
|
||||||
|
|
||||||
|
To authenticate with a server using Service Account Impersonation, you must set the `authProviderType` to `service_account_impersonation` and provide the following properties:
|
||||||
|
|
||||||
|
- **`targetAudience`** (string): The OAuth Client ID allowslisted on the IAP-protected application you are trying to access.
|
||||||
|
- **`targetServiceAccount`** (string): The email address of the Google Cloud Service Account to impersonate.
|
||||||
|
|
||||||
|
The CLI will use your local Application Default Credentials (ADC) to generate an OIDC ID token for the specified service account and audience. This token will then be used to authenticate with the MCP server.
|
||||||
|
|
||||||
|
#### Setup Instructions
|
||||||
|
|
||||||
|
1. **[Create](https://cloud.google.com/iap/docs/oauth-client-creation) or use an existing OAuth 2.0 client ID.** To use an existing OAuth 2.0 client ID, follow the steps in [How to share OAuth Clients](https://cloud.google.com/iap/docs/sharing-oauth-clients).
|
||||||
|
2. **Add the OAuth ID to the allowlist for [programmatic access](https://cloud.google.com/iap/docs/sharing-oauth-clients#programmatic_access) for the application.** Since Cloud Run is not yet a supported resource type in gcloud iap, you must allowlist the Client ID on the project.
|
||||||
|
3. **Create a service account.** [Documentation](https://cloud.google.com/iam/docs/service-accounts-create#creating), [Cloud Console Link](https://console.cloud.google.com/iam-admin/serviceaccounts)
|
||||||
|
4. **Add both the service account and users to the IAP Policy** in the "Security" tab of the Cloud Run service itself or via gcloud.
|
||||||
|
5. **Grant all users and groups** who will access the MCP Server the necessary permissions to [impersonate the service account](https://cloud.google.com/docs/authentication/use-service-account-impersonation) (i.e., `roles/iam.serviceAccountTokenCreator`).
|
||||||
|
6. **[Enable](https://console.cloud.google.com/apis/library/iamcredentials.googleapis.com) the IAM Credentials API** for your project.
|
||||||
|
|
||||||
### Example Configurations
|
### Example Configurations
|
||||||
|
|
||||||
#### Python MCP Server (Stdio)
|
#### Python MCP Server (Stdio)
|
||||||
@@ -310,6 +356,21 @@ You can specify the authentication provider type using the `authProviderType` pr
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### SSE MCP Server with SA Impersonation
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"mcpServers": {
|
||||||
|
"myIapProtectedServer": {
|
||||||
|
"url": "https://my-iap-service.run.app/sse",
|
||||||
|
"authProviderType": "service_account_impersonation",
|
||||||
|
"targetAudience": "YOUR_IAP_CLIENT_ID.apps.googleusercontent.com",
|
||||||
|
"targetServiceAccount": "your-sa@your-project.iam.gserviceaccount.com"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
## Discovery Process Deep Dive
|
## Discovery Process Deep Dive
|
||||||
|
|
||||||
When Qwen Code starts, it performs MCP server discovery through the following detailed process:
|
When Qwen Code starts, it performs MCP server discovery through the following detailed process:
|
||||||
@@ -667,9 +728,13 @@ await server.connect(transport);
|
|||||||
This can be included in `settings.json` under `mcpServers` with:
|
This can be included in `settings.json` under `mcpServers` with:
|
||||||
|
|
||||||
```json
|
```json
|
||||||
|
{
|
||||||
|
"mcpServers": {
|
||||||
"nodeServer": {
|
"nodeServer": {
|
||||||
"command": "node",
|
"command": "node",
|
||||||
"args": ["filename.ts"],
|
"args": ["filename.ts"]
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|||||||
@@ -4,7 +4,9 @@ This document describes the `run_shell_command` tool for Qwen Code.
|
|||||||
|
|
||||||
## Description
|
## Description
|
||||||
|
|
||||||
Use `run_shell_command` to interact with the underlying system, run scripts, or perform command-line operations. `run_shell_command` executes a given shell command. On Windows, the command will be executed with `cmd.exe /c`. On other platforms, the command will be executed with `bash -c`.
|
Use `run_shell_command` to interact with the underlying system, run scripts, or perform command-line operations. `run_shell_command` executes a given shell command, including interactive commands that require user input (e.g., `vim`, `git rebase -i`) if the `tools.shell.enableInteractiveShell` setting is set to `true`.
|
||||||
|
|
||||||
|
On Windows, commands are executed with `cmd.exe /c`. On other platforms, they are executed with `bash -c`.
|
||||||
|
|
||||||
### Arguments
|
### Arguments
|
||||||
|
|
||||||
@@ -102,10 +104,67 @@ Start multiple background services:
|
|||||||
run_shell_command(command="docker-compose up", description="Start all services", is_background=true)
|
run_shell_command(command="docker-compose up", description="Start all services", is_background=true)
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## Configuration
|
||||||
|
|
||||||
|
You can configure the behavior of the `run_shell_command` tool by modifying your `settings.json` file or by using the `/settings` command in the Qwen Code.
|
||||||
|
|
||||||
|
### Enabling Interactive Commands
|
||||||
|
|
||||||
|
To enable interactive commands, you need to set the `tools.shell.enableInteractiveShell` setting to `true`. This will use `node-pty` for shell command execution, which allows for interactive sessions. If `node-pty` is not available, it will fall back to the `child_process` implementation, which does not support interactive commands.
|
||||||
|
|
||||||
|
**Example `settings.json`:**
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"tools": {
|
||||||
|
"shell": {
|
||||||
|
"enableInteractiveShell": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Showing Color in Output
|
||||||
|
|
||||||
|
To show color in the shell output, you need to set the `tools.shell.showColor` setting to `true`. **Note: This setting only applies when `tools.shell.enableInteractiveShell` is enabled.**
|
||||||
|
|
||||||
|
**Example `settings.json`:**
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"tools": {
|
||||||
|
"shell": {
|
||||||
|
"showColor": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Setting the Pager
|
||||||
|
|
||||||
|
You can set a custom pager for the shell output by setting the `tools.shell.pager` setting. The default pager is `cat`. **Note: This setting only applies when `tools.shell.enableInteractiveShell` is enabled.**
|
||||||
|
|
||||||
|
**Example `settings.json`:**
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"tools": {
|
||||||
|
"shell": {
|
||||||
|
"pager": "less"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## Interactive Commands
|
||||||
|
|
||||||
|
The `run_shell_command` tool now supports interactive commands by integrating a pseudo-terminal (pty). This allows you to run commands that require real-time user input, such as text editors (`vim`, `nano`), terminal-based UIs (`htop`), and interactive version control operations (`git rebase -i`).
|
||||||
|
|
||||||
|
When an interactive command is running, you can send input to it from the Qwen Code. To focus on the interactive shell, press `ctrl+f`. The terminal output, including complex TUIs, will be rendered correctly.
|
||||||
|
|
||||||
## Important notes
|
## Important notes
|
||||||
|
|
||||||
- **Security:** Be cautious when executing commands, especially those constructed from user input, to prevent security vulnerabilities.
|
- **Security:** Be cautious when executing commands, especially those constructed from user input, to prevent security vulnerabilities.
|
||||||
- **Interactive commands:** Avoid commands that require interactive user input, as this can cause the tool to hang. Use non-interactive flags if available (e.g., `npm init -y`).
|
|
||||||
- **Error handling:** Check the `Stderr`, `Error`, and `Exit Code` fields to determine if a command executed successfully.
|
- **Error handling:** Check the `Stderr`, `Error`, and `Exit Code` fields to determine if a command executed successfully.
|
||||||
- **Background processes:** When `is_background=true` or when a command contains `&`, the tool will return immediately and the process will continue to run in the background. The `Background PIDs` field will contain the process ID of the background process.
|
- **Background processes:** When `is_background=true` or when a command contains `&`, the tool will return immediately and the process will continue to run in the background. The `Background PIDs` field will contain the process ID of the background process.
|
||||||
- **Background execution choices:** The `is_background` parameter is required and provides explicit control over execution mode. You can also add `&` to the command for manual background execution, but the `is_background` parameter must still be specified. The parameter provides clearer intent and automatically handles the background execution setup.
|
- **Background execution choices:** The `is_background` parameter is required and provides explicit control over execution mode. You can also add `&` to the command for manual background execution, but the `is_background` parameter must still be specified. The parameter provides clearer intent and automatically handles the background execution setup.
|
||||||
@@ -117,16 +176,16 @@ When `run_shell_command` executes a command, it sets the `QWEN_CODE=1` environme
|
|||||||
|
|
||||||
## Command Restrictions
|
## Command Restrictions
|
||||||
|
|
||||||
You can restrict the commands that can be executed by the `run_shell_command` tool by using the `coreTools` and `excludeTools` settings in your configuration file.
|
You can restrict the commands that can be executed by the `run_shell_command` tool by using the `tools.core` and `tools.exclude` settings in your configuration file.
|
||||||
|
|
||||||
- `coreTools`: To restrict `run_shell_command` to a specific set of commands, add entries to the `coreTools` list in the format `run_shell_command(<command>)`. For example, `"coreTools": ["run_shell_command(git)"]` will only allow `git` commands. Including the generic `run_shell_command` acts as a wildcard, allowing any command not explicitly blocked.
|
- `tools.core`: To restrict `run_shell_command` to a specific set of commands, add entries to the `core` list under the `tools` category in the format `run_shell_command(<command>)`. For example, `"tools": {"core": ["run_shell_command(git)"]}` will only allow `git` commands. Including the generic `run_shell_command` acts as a wildcard, allowing any command not explicitly blocked.
|
||||||
- `excludeTools`: To block specific commands, add entries to the `excludeTools` list in the format `run_shell_command(<command>)`. For example, `"excludeTools": ["run_shell_command(rm)"]` will block `rm` commands.
|
- `tools.exclude`: To block specific commands, add entries to the `exclude` list under the `tools` category in the format `run_shell_command(<command>)`. For example, `"tools": {"exclude": ["run_shell_command(rm)"]}` will block `rm` commands.
|
||||||
|
|
||||||
The validation logic is designed to be secure and flexible:
|
The validation logic is designed to be secure and flexible:
|
||||||
|
|
||||||
1. **Command Chaining Disabled**: The tool automatically splits commands chained with `&&`, `||`, or `;` and validates each part separately. If any part of the chain is disallowed, the entire command is blocked.
|
1. **Command Chaining Disabled**: The tool automatically splits commands chained with `&&`, `||`, or `;` and validates each part separately. If any part of the chain is disallowed, the entire command is blocked.
|
||||||
2. **Prefix Matching**: The tool uses prefix matching. For example, if you allow `git`, you can run `git status` or `git log`.
|
2. **Prefix Matching**: The tool uses prefix matching. For example, if you allow `git`, you can run `git status` or `git log`.
|
||||||
3. **Blocklist Precedence**: The `excludeTools` list is always checked first. If a command matches a blocked prefix, it will be denied, even if it also matches an allowed prefix in `coreTools`.
|
3. **Blocklist Precedence**: The `tools.exclude` list is always checked first. If a command matches a blocked prefix, it will be denied, even if it also matches an allowed prefix in `tools.core`.
|
||||||
|
|
||||||
### Command Restriction Examples
|
### Command Restriction Examples
|
||||||
|
|
||||||
@@ -136,7 +195,9 @@ To allow only `git` and `npm` commands, and block all others:
|
|||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"coreTools": ["run_shell_command(git)", "run_shell_command(npm)"]
|
"tools": {
|
||||||
|
"core": ["run_shell_command(git)", "run_shell_command(npm)"]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -150,8 +211,10 @@ To block `rm` and allow all other commands:
|
|||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"coreTools": ["run_shell_command"],
|
"tools": {
|
||||||
"excludeTools": ["run_shell_command(rm)"]
|
"core": ["run_shell_command"],
|
||||||
|
"exclude": ["run_shell_command(rm)"]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -161,12 +224,14 @@ To block `rm` and allow all other commands:
|
|||||||
|
|
||||||
**Blocklist takes precedence**
|
**Blocklist takes precedence**
|
||||||
|
|
||||||
If a command prefix is in both `coreTools` and `excludeTools`, it will be blocked.
|
If a command prefix is in both `tools.core` and `tools.exclude`, it will be blocked.
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"coreTools": ["run_shell_command(git)"],
|
"tools": {
|
||||||
"excludeTools": ["run_shell_command(git push)"]
|
"core": ["run_shell_command(git)"],
|
||||||
|
"exclude": ["run_shell_command(git push)"]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -175,11 +240,13 @@ If a command prefix is in both `coreTools` and `excludeTools`, it will be blocke
|
|||||||
|
|
||||||
**Block all shell commands**
|
**Block all shell commands**
|
||||||
|
|
||||||
To block all shell commands, add the `run_shell_command` wildcard to `excludeTools`:
|
To block all shell commands, add the `run_shell_command` wildcard to `tools.exclude`:
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"excludeTools": ["run_shell_command"]
|
"tools": {
|
||||||
|
"exclude": ["run_shell_command"]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|||||||
@@ -62,7 +62,7 @@ This guide provides solutions to common issues and debugging tips, including top
|
|||||||
- **DEBUG mode not working from project .env file**
|
- **DEBUG mode not working from project .env file**
|
||||||
- **Issue:** Setting `DEBUG=true` in a project's `.env` file doesn't enable debug mode for the CLI.
|
- **Issue:** Setting `DEBUG=true` in a project's `.env` file doesn't enable debug mode for the CLI.
|
||||||
- **Cause:** The `DEBUG` and `DEBUG_MODE` variables are automatically excluded from project `.env` files to prevent interference with the CLI behavior.
|
- **Cause:** The `DEBUG` and `DEBUG_MODE` variables are automatically excluded from project `.env` files to prevent interference with the CLI behavior.
|
||||||
- **Solution:** Use a `.qwen/.env` file instead, or configure the `excludedProjectEnvVars` setting in your `settings.json` to exclude fewer variables.
|
- **Solution:** Use a `.qwen/.env` file instead, or configure the `advanced.excludedEnvVars` setting in your `settings.json` to exclude fewer variables.
|
||||||
|
|
||||||
## IDE Companion not connecting
|
## IDE Companion not connecting
|
||||||
|
|
||||||
|
|||||||
61
docs/trusted-folders.md
Normal file
61
docs/trusted-folders.md
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
# Trusted Folders
|
||||||
|
|
||||||
|
The Trusted Folders feature is a security setting that gives you control over which projects can use the full capabilities of the Qwen Code. It prevents potentially malicious code from running by asking you to approve a folder before the CLI loads any project-specific configurations from it.
|
||||||
|
|
||||||
|
## Enabling the Feature
|
||||||
|
|
||||||
|
The Trusted Folders feature is **disabled by default**. To use it, you must first enable it in your settings.
|
||||||
|
|
||||||
|
Add the following to your user `settings.json` file:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"security": {
|
||||||
|
"folderTrust": {
|
||||||
|
"enabled": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## How It Works: The Trust Dialog
|
||||||
|
|
||||||
|
Once the feature is enabled, the first time you run the Qwen Code from a folder, a dialog will automatically appear, prompting you to make a choice:
|
||||||
|
|
||||||
|
- **Trust folder**: Grants full trust to the current folder (e.g., `my-project`).
|
||||||
|
- **Trust parent folder**: Grants trust to the parent directory (e.g., `safe-projects`), which automatically trusts all of its subdirectories as well. This is useful if you keep all your safe projects in one place.
|
||||||
|
- **Don't trust**: Marks the folder as untrusted. The CLI will operate in a restricted "safe mode."
|
||||||
|
|
||||||
|
Your choice is saved in a central file (`~/.qwen/trustedFolders.json`), so you will only be asked once per folder.
|
||||||
|
|
||||||
|
## Why Trust Matters: The Impact of an Untrusted Workspace
|
||||||
|
|
||||||
|
When a folder is **untrusted**, the Qwen Code runs in a restricted "safe mode" to protect you. In this mode, the following features are disabled:
|
||||||
|
|
||||||
|
1. **Workspace Settings are Ignored**: The CLI will **not** load the `.qwen/settings.json` file from the project. This prevents the loading of custom tools and other potentially dangerous configurations.
|
||||||
|
|
||||||
|
2. **Environment Variables are Ignored**: The CLI will **not** load any `.env` files from the project.
|
||||||
|
|
||||||
|
3. **Extension Management is Restricted**: You **cannot install, update, or uninstall** extensions.
|
||||||
|
|
||||||
|
4. **Tool Auto-Acceptance is Disabled**: You will always be prompted before any tool is run, even if you have auto-acceptance enabled globally.
|
||||||
|
|
||||||
|
5. **Automatic Memory Loading is Disabled**: The CLI will not automatically load files into context from directories specified in local settings.
|
||||||
|
|
||||||
|
Granting trust to a folder unlocks the full functionality of the Qwen Code for that workspace.
|
||||||
|
|
||||||
|
## Managing Your Trust Settings
|
||||||
|
|
||||||
|
If you need to change a decision or see all your settings, you have a couple of options:
|
||||||
|
|
||||||
|
- **Change the Current Folder's Trust**: Run the `/permissions` command from within the CLI. This will bring up the same interactive dialog, allowing you to change the trust level for the current folder.
|
||||||
|
|
||||||
|
- **View All Trust Rules**: To see a complete list of all your trusted and untrusted folder rules, you can inspect the contents of the `~/.qwen/trustedFolders.json` file in your home directory.
|
||||||
|
|
||||||
|
## The Trust Check Process (Advanced)
|
||||||
|
|
||||||
|
For advanced users, it's helpful to know the exact order of operations for how trust is determined:
|
||||||
|
|
||||||
|
1. **IDE Trust Signal**: If you are using the [IDE Integration](./ide-integration.md), the CLI first asks the IDE if the workspace is trusted. The IDE's response takes highest priority.
|
||||||
|
|
||||||
|
2. **Local Trust File**: If the IDE is not connected, the CLI checks the central `~/.qwen/trustedFolders.json` file.
|
||||||
@@ -4,16 +4,34 @@
|
|||||||
* SPDX-License-Identifier: Apache-2.0
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import esbuild from 'esbuild';
|
|
||||||
import path from 'node:path';
|
import path from 'node:path';
|
||||||
import { fileURLToPath } from 'node:url';
|
import { fileURLToPath } from 'node:url';
|
||||||
import { createRequire } from 'node:module';
|
import { createRequire } from 'node:module';
|
||||||
|
import { writeFileSync } from 'node:fs';
|
||||||
|
|
||||||
|
let esbuild;
|
||||||
|
try {
|
||||||
|
esbuild = (await import('esbuild')).default;
|
||||||
|
} catch (_error) {
|
||||||
|
console.warn('esbuild not available, skipping bundle step');
|
||||||
|
process.exit(0);
|
||||||
|
}
|
||||||
|
|
||||||
const __filename = fileURLToPath(import.meta.url);
|
const __filename = fileURLToPath(import.meta.url);
|
||||||
const __dirname = path.dirname(__filename);
|
const __dirname = path.dirname(__filename);
|
||||||
const require = createRequire(import.meta.url);
|
const require = createRequire(import.meta.url);
|
||||||
const pkg = require(path.resolve(__dirname, 'package.json'));
|
const pkg = require(path.resolve(__dirname, 'package.json'));
|
||||||
|
|
||||||
|
const external = [
|
||||||
|
'@lydell/node-pty',
|
||||||
|
'node-pty',
|
||||||
|
'@lydell/node-pty-darwin-arm64',
|
||||||
|
'@lydell/node-pty-darwin-x64',
|
||||||
|
'@lydell/node-pty-linux-x64',
|
||||||
|
'@lydell/node-pty-win32-arm64',
|
||||||
|
'@lydell/node-pty-win32-x64',
|
||||||
|
];
|
||||||
|
|
||||||
esbuild
|
esbuild
|
||||||
.build({
|
.build({
|
||||||
entryPoints: ['packages/cli/index.ts'],
|
entryPoints: ['packages/cli/index.ts'],
|
||||||
@@ -21,15 +39,7 @@ esbuild
|
|||||||
outfile: 'bundle/gemini.js',
|
outfile: 'bundle/gemini.js',
|
||||||
platform: 'node',
|
platform: 'node',
|
||||||
format: 'esm',
|
format: 'esm',
|
||||||
external: [
|
external,
|
||||||
'@lydell/node-pty',
|
|
||||||
'node-pty',
|
|
||||||
'@lydell/node-pty-darwin-arm64',
|
|
||||||
'@lydell/node-pty-darwin-x64',
|
|
||||||
'@lydell/node-pty-linux-x64',
|
|
||||||
'@lydell/node-pty-win32-arm64',
|
|
||||||
'@lydell/node-pty-win32-x64',
|
|
||||||
],
|
|
||||||
alias: {
|
alias: {
|
||||||
'is-in-ci': path.resolve(
|
'is-in-ci': path.resolve(
|
||||||
__dirname,
|
__dirname,
|
||||||
@@ -43,5 +53,12 @@ esbuild
|
|||||||
js: `import { createRequire } from 'module'; const require = createRequire(import.meta.url); globalThis.__filename = require('url').fileURLToPath(import.meta.url); globalThis.__dirname = require('path').dirname(globalThis.__filename);`,
|
js: `import { createRequire } from 'module'; const require = createRequire(import.meta.url); globalThis.__filename = require('url').fileURLToPath(import.meta.url); globalThis.__dirname = require('path').dirname(globalThis.__filename);`,
|
||||||
},
|
},
|
||||||
loader: { '.node': 'file' },
|
loader: { '.node': 'file' },
|
||||||
|
metafile: true,
|
||||||
|
write: true,
|
||||||
|
})
|
||||||
|
.then(({ metafile }) => {
|
||||||
|
if (process.env.DEV === 'true') {
|
||||||
|
writeFileSync('./bundle/esbuild.json', JSON.stringify(metafile, null, 2));
|
||||||
|
}
|
||||||
})
|
})
|
||||||
.catch(() => process.exit(1));
|
.catch(() => process.exit(1));
|
||||||
|
|||||||
@@ -34,6 +34,7 @@ export default tseslint.config(
|
|||||||
'bundle/**',
|
'bundle/**',
|
||||||
'package/bundle/**',
|
'package/bundle/**',
|
||||||
'.integration-tests/**',
|
'.integration-tests/**',
|
||||||
|
'dist/**',
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
eslint.configs.recommended,
|
eslint.configs.recommended,
|
||||||
|
|||||||
8
hello/QWEN.md
Normal file
8
hello/QWEN.md
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
# Ink Library Screen Reader Guidance
|
||||||
|
|
||||||
|
When building custom components, it's important to keep accessibility in mind. While Ink provides the building blocks, ensuring your components are accessible will make your CLIs usable by a wider audience.
|
||||||
|
|
||||||
|
## General Principles
|
||||||
|
|
||||||
|
Provide screen reader-friendly output: Use the useIsScreenReaderEnabled hook to detect if a screen reader is active. You can then render a more descriptive output for screen reader users.
|
||||||
|
Leverage ARIA props: For components that have a specific role (e.g., a checkbox or a button), use the aria-role, aria-state, and aria-label props on <Box> and <Text> to provide semantic meaning to screen readers.
|
||||||
5
hello/qwen-extension.json
Normal file
5
hello/qwen-extension.json
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
{
|
||||||
|
"name": "context-example",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"contextFileName": "QWEN.md"
|
||||||
|
}
|
||||||
116
integration-tests/context-compress-interactive.test.ts
Normal file
116
integration-tests/context-compress-interactive.test.ts
Normal file
@@ -0,0 +1,116 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { expect, describe, it, beforeEach, afterEach } from 'vitest';
|
||||||
|
import { TestRig, type } from './test-helper.js';
|
||||||
|
|
||||||
|
describe('Interactive Mode', () => {
|
||||||
|
let rig: TestRig;
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
rig = new TestRig();
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(async () => {
|
||||||
|
await rig.cleanup();
|
||||||
|
});
|
||||||
|
|
||||||
|
it.skipIf(process.platform === 'win32')(
|
||||||
|
'should trigger chat compression with /compress command',
|
||||||
|
async () => {
|
||||||
|
await rig.setup('interactive-compress-test');
|
||||||
|
|
||||||
|
const { ptyProcess } = rig.runInteractive();
|
||||||
|
|
||||||
|
let fullOutput = '';
|
||||||
|
ptyProcess.onData((data) => (fullOutput += data));
|
||||||
|
|
||||||
|
const authDialogAppeared = await rig.waitForText(
|
||||||
|
'How would you like to authenticate',
|
||||||
|
5000,
|
||||||
|
);
|
||||||
|
|
||||||
|
// select the second option if auth dialog come's up
|
||||||
|
if (authDialogAppeared) {
|
||||||
|
ptyProcess.write('2');
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for the app to be ready
|
||||||
|
const isReady = await rig.waitForText('Type your message', 15000);
|
||||||
|
expect(
|
||||||
|
isReady,
|
||||||
|
'CLI did not start up in interactive mode correctly',
|
||||||
|
).toBe(true);
|
||||||
|
|
||||||
|
const longPrompt =
|
||||||
|
'Dont do anything except returning a 1000 token long paragragh with the <name of the scientist who discovered theory of relativity> at the end to indicate end of response. This is a moderately long sentence.';
|
||||||
|
|
||||||
|
await type(ptyProcess, longPrompt);
|
||||||
|
await type(ptyProcess, '\r');
|
||||||
|
|
||||||
|
await rig.waitForText('einstein', 25000);
|
||||||
|
|
||||||
|
await type(ptyProcess, '/compress');
|
||||||
|
// A small delay to allow React to re-render the command list.
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, 100));
|
||||||
|
await type(ptyProcess, '\r');
|
||||||
|
|
||||||
|
const foundEvent = await rig.waitForTelemetryEvent(
|
||||||
|
'chat_compression',
|
||||||
|
90000,
|
||||||
|
);
|
||||||
|
expect(foundEvent, 'chat_compression telemetry event was not found').toBe(
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
it.skipIf(process.platform === 'win32')(
|
||||||
|
'should handle compression failure on token inflation',
|
||||||
|
async () => {
|
||||||
|
await rig.setup('interactive-compress-test');
|
||||||
|
|
||||||
|
const { ptyProcess } = rig.runInteractive();
|
||||||
|
|
||||||
|
let fullOutput = '';
|
||||||
|
ptyProcess.onData((data) => (fullOutput += data));
|
||||||
|
|
||||||
|
const authDialogAppeared = await rig.waitForText(
|
||||||
|
'How would you like to authenticate',
|
||||||
|
5000,
|
||||||
|
);
|
||||||
|
|
||||||
|
// select the second option if auth dialog come's up
|
||||||
|
if (authDialogAppeared) {
|
||||||
|
ptyProcess.write('2');
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for the app to be ready
|
||||||
|
const isReady = await rig.waitForText('Type your message', 25000);
|
||||||
|
expect(
|
||||||
|
isReady,
|
||||||
|
'CLI did not start up in interactive mode correctly',
|
||||||
|
).toBe(true);
|
||||||
|
|
||||||
|
await type(ptyProcess, '/compress');
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, 100));
|
||||||
|
await type(ptyProcess, '\r');
|
||||||
|
|
||||||
|
const foundEvent = await rig.waitForTelemetryEvent(
|
||||||
|
'chat_compression',
|
||||||
|
90000,
|
||||||
|
);
|
||||||
|
expect(foundEvent).toBe(true);
|
||||||
|
|
||||||
|
const compressionFailed = await rig.waitForText(
|
||||||
|
'compression was not beneficial',
|
||||||
|
25000,
|
||||||
|
);
|
||||||
|
|
||||||
|
expect(compressionFailed).toBe(true);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
});
|
||||||
129
integration-tests/ctrl-c-exit.test.ts
Normal file
129
integration-tests/ctrl-c-exit.test.ts
Normal file
@@ -0,0 +1,129 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect } from 'vitest';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
import * as fs from 'node:fs';
|
||||||
|
import * as path from 'node:path';
|
||||||
|
|
||||||
|
describe('Ctrl+C exit', () => {
|
||||||
|
// (#9782) Temporarily disabling on windows because it is failing on main and every
|
||||||
|
// PR, which is potentially hiding other failures
|
||||||
|
it.skipIf(process.platform === 'win32')(
|
||||||
|
'should exit gracefully on second Ctrl+C',
|
||||||
|
async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should exit gracefully on second Ctrl+C');
|
||||||
|
|
||||||
|
const { ptyProcess, promise } = rig.runInteractive();
|
||||||
|
|
||||||
|
let output = '';
|
||||||
|
ptyProcess.onData((data) => {
|
||||||
|
output += data;
|
||||||
|
});
|
||||||
|
|
||||||
|
// Wait for the app to be ready by looking for the initial prompt indicator
|
||||||
|
await rig.poll(() => output.includes('▶'), 5000, 100);
|
||||||
|
|
||||||
|
// Send first Ctrl+C
|
||||||
|
ptyProcess.write(String.fromCharCode(3));
|
||||||
|
|
||||||
|
// Wait for the exit prompt
|
||||||
|
await rig.poll(
|
||||||
|
() => output.includes('Press Ctrl+C again to exit'),
|
||||||
|
1500,
|
||||||
|
50,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Send second Ctrl+C
|
||||||
|
ptyProcess.write(String.fromCharCode(3));
|
||||||
|
|
||||||
|
const result = await promise;
|
||||||
|
|
||||||
|
// Expect a graceful exit (code 0)
|
||||||
|
expect(
|
||||||
|
result.exitCode,
|
||||||
|
`Process exited with code ${result.exitCode}. Output: ${result.output}`,
|
||||||
|
).toBe(0);
|
||||||
|
|
||||||
|
// Check that the quitting message is displayed
|
||||||
|
const quittingMessage = 'Agent powering down. Goodbye!';
|
||||||
|
// The regex below is intentionally matching the ESC control character (\x1b)
|
||||||
|
// to strip ANSI color codes from the terminal output.
|
||||||
|
// eslint-disable-next-line no-control-regex
|
||||||
|
const cleanOutput = output.replace(/\x1b\[[0-9;]*m/g, '');
|
||||||
|
expect(cleanOutput).toContain(quittingMessage);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
it.skipIf(process.platform === 'win32')(
|
||||||
|
'should exit gracefully on second Ctrl+C when calling a tool',
|
||||||
|
async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup(
|
||||||
|
'should exit gracefully on second Ctrl+C when calling a tool',
|
||||||
|
);
|
||||||
|
|
||||||
|
const childProcessFile = 'child_process_file.txt';
|
||||||
|
rig.createFile(
|
||||||
|
'wait.js',
|
||||||
|
`setTimeout(() => require('fs').writeFileSync('${childProcessFile}', 'done'), 5000)`,
|
||||||
|
);
|
||||||
|
|
||||||
|
const { ptyProcess, promise } = rig.runInteractive();
|
||||||
|
|
||||||
|
let output = '';
|
||||||
|
ptyProcess.onData((data) => {
|
||||||
|
output += data;
|
||||||
|
});
|
||||||
|
|
||||||
|
// Wait for the app to be ready by looking for the initial prompt indicator
|
||||||
|
await rig.poll(() => output.includes('▶'), 5000, 100);
|
||||||
|
|
||||||
|
ptyProcess.write('use the tool to run "node -e wait.js"\n');
|
||||||
|
|
||||||
|
await rig.poll(() => output.includes('Shell'), 5000, 100);
|
||||||
|
|
||||||
|
// Send first Ctrl+C
|
||||||
|
ptyProcess.write(String.fromCharCode(3));
|
||||||
|
|
||||||
|
// Wait for the exit prompt
|
||||||
|
await rig.poll(
|
||||||
|
() => output.includes('Press Ctrl+C again to exit'),
|
||||||
|
1500,
|
||||||
|
50,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Send second Ctrl+C
|
||||||
|
ptyProcess.write(String.fromCharCode(3));
|
||||||
|
|
||||||
|
const result = await promise;
|
||||||
|
|
||||||
|
// Expect a graceful exit (code 0)
|
||||||
|
expect(
|
||||||
|
result.exitCode,
|
||||||
|
`Process exited with code ${result.exitCode}. Output: ${result.output}`,
|
||||||
|
).toBe(0);
|
||||||
|
|
||||||
|
// Check that the quitting message is displayed
|
||||||
|
const quittingMessage = 'Agent powering down. Goodbye!';
|
||||||
|
// The regex below is intentionally matching the ESC control character (\x1b)
|
||||||
|
// to strip ANSI color codes from the terminal output.
|
||||||
|
// eslint-disable-next-line no-control-regex
|
||||||
|
const cleanOutput = output.replace(/\x1b\[[0-9;]*m/g, '');
|
||||||
|
expect(cleanOutput).toContain(quittingMessage);
|
||||||
|
|
||||||
|
// Check that the child process was terminated and did not create the file.
|
||||||
|
const childProcessFileExists = fs.existsSync(
|
||||||
|
path.join(rig.testDir!, childProcessFile),
|
||||||
|
);
|
||||||
|
expect(
|
||||||
|
childProcessFileExists,
|
||||||
|
'Child process file should not exist',
|
||||||
|
).toBe(false);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
});
|
||||||
@@ -61,4 +61,125 @@ describe('edit', () => {
|
|||||||
console.log('File edited successfully. New content:', newFileContent);
|
console.log('File edited successfully. New content:', newFileContent);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('should handle $ literally when replacing text ending with $', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup(
|
||||||
|
'should handle $ literally when replacing text ending with $',
|
||||||
|
);
|
||||||
|
|
||||||
|
const fileName = 'regex.yml';
|
||||||
|
const originalContent = "| select('match', '^[sv]d[a-z]$')\n";
|
||||||
|
const expectedContent = "| select('match', '^[sv]d[a-z]$') # updated\n";
|
||||||
|
|
||||||
|
rig.createFile(fileName, originalContent);
|
||||||
|
|
||||||
|
const prompt =
|
||||||
|
"Open regex.yml and append ' # updated' after the line containing ^[sv]d[a-z]$ without breaking the $ character.";
|
||||||
|
|
||||||
|
const result = await rig.run(prompt);
|
||||||
|
const foundToolCall = await rig.waitForToolCall('edit');
|
||||||
|
|
||||||
|
if (!foundToolCall) {
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(foundToolCall, 'Expected to find an edit tool call').toBeTruthy();
|
||||||
|
|
||||||
|
validateModelOutput(result, ['regex.yml'], 'Replace $ literal test');
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
expect(newFileContent).toBe(expectedContent);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should fail safely when old_string is not found', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should fail safely when old_string is not found');
|
||||||
|
const fileName = 'no_match.txt';
|
||||||
|
const fileContent = 'hello world';
|
||||||
|
rig.createFile(fileName, fileContent);
|
||||||
|
|
||||||
|
const prompt = `replace "goodbye" with "farewell" in ${fileName}`;
|
||||||
|
await rig.run(prompt);
|
||||||
|
|
||||||
|
await rig.waitForTelemetryReady();
|
||||||
|
const toolLogs = rig.readToolLogs();
|
||||||
|
|
||||||
|
const editAttempt = toolLogs.find((log) => log.toolRequest.name === 'edit');
|
||||||
|
const readAttempt = toolLogs.find(
|
||||||
|
(log) => log.toolRequest.name === 'read_file',
|
||||||
|
);
|
||||||
|
|
||||||
|
// VERIFY: The model must have at least tried to read the file or perform an edit.
|
||||||
|
expect(
|
||||||
|
readAttempt || editAttempt,
|
||||||
|
'Expected model to attempt a read_file or edit',
|
||||||
|
).toBeDefined();
|
||||||
|
|
||||||
|
// If the model tried to edit, that specific attempt must have failed.
|
||||||
|
if (editAttempt) {
|
||||||
|
if (editAttempt.toolRequest.success) {
|
||||||
|
console.error('The edit tool succeeded when it was expected to fail');
|
||||||
|
console.error('Tool call args:', editAttempt.toolRequest.args);
|
||||||
|
}
|
||||||
|
expect(
|
||||||
|
editAttempt.toolRequest.success,
|
||||||
|
'If edit is called, it must fail',
|
||||||
|
).toBe(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
// CRITICAL: The final content of the file must be unchanged.
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
expect(newFileContent).toBe(fileContent);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should insert a multi-line block of text', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should insert a multi-line block of text');
|
||||||
|
const fileName = 'insert_block.js';
|
||||||
|
const originalContent = 'function hello() {\n // INSERT_CODE_HERE\n}';
|
||||||
|
const newBlock = "console.log('hello');\n console.log('world');";
|
||||||
|
const expectedContent = `function hello() {\n ${newBlock}\n}`;
|
||||||
|
rig.createFile(fileName, originalContent);
|
||||||
|
|
||||||
|
const prompt = `In ${fileName}, replace "// INSERT_CODE_HERE" with:\n${newBlock}`;
|
||||||
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
|
const foundToolCall = await rig.waitForToolCall('edit');
|
||||||
|
if (!foundToolCall) {
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(foundToolCall, 'Expected to find an edit tool call').toBeTruthy();
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
|
||||||
|
expect(newFileContent.replace(/\r\n/g, '\n')).toBe(
|
||||||
|
expectedContent.replace(/\r\n/g, '\n'),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should delete a block of text', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should delete a block of text');
|
||||||
|
const fileName = 'delete_block.txt';
|
||||||
|
const blockToDelete =
|
||||||
|
'## DELETE THIS ##\nThis is a block of text to delete.\n## END DELETE ##';
|
||||||
|
const originalContent = `Hello\n${blockToDelete}\nWorld`;
|
||||||
|
// When deleting the block, a newline remains from the original structure (Hello\n + \nWorld)
|
||||||
|
rig.createFile(fileName, originalContent);
|
||||||
|
|
||||||
|
const prompt = `In ${fileName}, delete the entire block from "## DELETE THIS ##" to "## END DELETE ##" including the markers.`;
|
||||||
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
|
const foundToolCall = await rig.waitForToolCall('edit');
|
||||||
|
if (!foundToolCall) {
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(foundToolCall, 'Expected to find an edit tool call').toBeTruthy();
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
|
||||||
|
// Accept either 1 or 2 newlines between Hello and World
|
||||||
|
expect(newFileContent).toMatch(/^Hello\n\n?World$/);
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
52
integration-tests/extensions-install.test.ts
Normal file
52
integration-tests/extensions-install.test.ts
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { expect, test } from 'vitest';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
import { writeFileSync } from 'node:fs';
|
||||||
|
import { join } from 'node:path';
|
||||||
|
|
||||||
|
const extension = `{
|
||||||
|
"name": "test-extension",
|
||||||
|
"version": "0.0.1"
|
||||||
|
}`;
|
||||||
|
|
||||||
|
const extensionUpdate = `{
|
||||||
|
"name": "test-extension",
|
||||||
|
"version": "0.0.2"
|
||||||
|
}`;
|
||||||
|
|
||||||
|
test('installs a local extension, verifies a command, and updates it', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
rig.setup('extension install test');
|
||||||
|
const testServerPath = join(rig.testDir!, 'qwen-extension.json');
|
||||||
|
writeFileSync(testServerPath, extension);
|
||||||
|
try {
|
||||||
|
await rig.runCommand(['extensions', 'uninstall', 'test-extension']);
|
||||||
|
} catch {
|
||||||
|
/* empty */
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await rig.runCommand(
|
||||||
|
['extensions', 'install', `${rig.testDir!}`],
|
||||||
|
{ stdin: 'y\n' },
|
||||||
|
);
|
||||||
|
expect(result).toContain('test-extension');
|
||||||
|
|
||||||
|
const listResult = await rig.runCommand(['extensions', 'list']);
|
||||||
|
expect(listResult).toContain('test-extension');
|
||||||
|
writeFileSync(testServerPath, extensionUpdate);
|
||||||
|
const updateResult = await rig.runCommand([
|
||||||
|
'extensions',
|
||||||
|
'update',
|
||||||
|
`test-extension`,
|
||||||
|
]);
|
||||||
|
expect(updateResult).toContain('0.0.2');
|
||||||
|
|
||||||
|
await rig.runCommand(['extensions', 'uninstall', 'test-extension']);
|
||||||
|
|
||||||
|
await rig.cleanup();
|
||||||
|
});
|
||||||
85
integration-tests/file-system-interactive.test.ts
Normal file
85
integration-tests/file-system-interactive.test.ts
Normal file
@@ -0,0 +1,85 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { expect, describe, it, beforeEach, afterEach } from 'vitest';
|
||||||
|
import { TestRig, type, printDebugInfo } from './test-helper.js';
|
||||||
|
|
||||||
|
describe('Interactive file system', () => {
|
||||||
|
let rig: TestRig;
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
rig = new TestRig();
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(async () => {
|
||||||
|
await rig.cleanup();
|
||||||
|
});
|
||||||
|
|
||||||
|
it.skipIf(process.platform === 'win32')(
|
||||||
|
'should perform a read-then-write sequence',
|
||||||
|
async () => {
|
||||||
|
const fileName = 'version.txt';
|
||||||
|
await rig.setup('interactive-read-then-write');
|
||||||
|
rig.createFile(fileName, '1.0.0');
|
||||||
|
|
||||||
|
const { ptyProcess } = rig.runInteractive();
|
||||||
|
|
||||||
|
const authDialogAppeared = await rig.waitForText(
|
||||||
|
'How would you like to authenticate',
|
||||||
|
5000,
|
||||||
|
);
|
||||||
|
|
||||||
|
// select the second option if auth dialog come's up
|
||||||
|
if (authDialogAppeared) {
|
||||||
|
ptyProcess.write('2');
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for the app to be ready
|
||||||
|
const isReady = await rig.waitForText('Type your message', 15000);
|
||||||
|
expect(
|
||||||
|
isReady,
|
||||||
|
'CLI did not start up in interactive mode correctly',
|
||||||
|
).toBe(true);
|
||||||
|
|
||||||
|
// Step 1: Read the file
|
||||||
|
const readPrompt = `Read the version from ${fileName}`;
|
||||||
|
await type(ptyProcess, readPrompt);
|
||||||
|
await type(ptyProcess, '\r');
|
||||||
|
|
||||||
|
const readCall = await rig.waitForToolCall('read_file', 30000);
|
||||||
|
expect(readCall, 'Expected to find a read_file tool call').toBe(true);
|
||||||
|
|
||||||
|
const containsExpectedVersion = await rig.waitForText('1.0.0', 15000);
|
||||||
|
expect(
|
||||||
|
containsExpectedVersion,
|
||||||
|
'Expected to see version "1.0.0" in output',
|
||||||
|
).toBe(true);
|
||||||
|
|
||||||
|
// Step 2: Write the file
|
||||||
|
const writePrompt = `now change the version to 1.0.1 in the file`;
|
||||||
|
await type(ptyProcess, writePrompt);
|
||||||
|
await type(ptyProcess, '\r');
|
||||||
|
|
||||||
|
const toolCall = await rig.waitForAnyToolCall(
|
||||||
|
['write_file', 'edit'],
|
||||||
|
30000,
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!toolCall) {
|
||||||
|
printDebugInfo(rig, rig._interactiveOutput, {
|
||||||
|
toolCall,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(toolCall, 'Expected to find a write_file or edit tool call').toBe(
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
expect(newFileContent).toBe('1.0.1');
|
||||||
|
},
|
||||||
|
);
|
||||||
|
});
|
||||||
@@ -5,6 +5,8 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
import { describe, it, expect } from 'vitest';
|
import { describe, it, expect } from 'vitest';
|
||||||
|
import { existsSync } from 'node:fs';
|
||||||
|
import * as path from 'node:path';
|
||||||
import { TestRig, printDebugInfo, validateModelOutput } from './test-helper.js';
|
import { TestRig, printDebugInfo, validateModelOutput } from './test-helper.js';
|
||||||
|
|
||||||
describe('file-system', () => {
|
describe('file-system', () => {
|
||||||
@@ -86,4 +88,169 @@ describe('file-system', () => {
|
|||||||
console.log('File written successfully with hello message.');
|
console.log('File written successfully with hello message.');
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('should correctly handle file paths with spaces', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should correctly handle file paths with spaces');
|
||||||
|
const fileName = 'my test file.txt';
|
||||||
|
|
||||||
|
const result = await rig.run(`write "hello" to "${fileName}"`);
|
||||||
|
|
||||||
|
const foundToolCall = await rig.waitForToolCall('write_file');
|
||||||
|
if (!foundToolCall) {
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(
|
||||||
|
foundToolCall,
|
||||||
|
'Expected to find a write_file tool call',
|
||||||
|
).toBeTruthy();
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
expect(newFileContent).toBe('hello');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should perform a read-then-write sequence', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should perform a read-then-write sequence');
|
||||||
|
const fileName = 'version.txt';
|
||||||
|
rig.createFile(fileName, '1.0.0');
|
||||||
|
|
||||||
|
const prompt = `Read the version from ${fileName} and write the next version 1.0.1 back to the file.`;
|
||||||
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
|
await rig.waitForTelemetryReady();
|
||||||
|
const toolLogs = rig.readToolLogs();
|
||||||
|
|
||||||
|
const readCall = toolLogs.find(
|
||||||
|
(log) => log.toolRequest.name === 'read_file',
|
||||||
|
);
|
||||||
|
const writeCall = toolLogs.find(
|
||||||
|
(log) =>
|
||||||
|
log.toolRequest.name === 'write_file' ||
|
||||||
|
log.toolRequest.name === 'replace',
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!readCall || !writeCall) {
|
||||||
|
printDebugInfo(rig, result, { readCall, writeCall });
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(readCall, 'Expected to find a read_file tool call').toBeDefined();
|
||||||
|
expect(
|
||||||
|
writeCall,
|
||||||
|
'Expected to find a write_file or replace tool call',
|
||||||
|
).toBeDefined();
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
expect(newFileContent).toBe('1.0.1');
|
||||||
|
});
|
||||||
|
|
||||||
|
it.skip('should replace multiple instances of a string', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should replace multiple instances of a string');
|
||||||
|
const fileName = 'ambiguous.txt';
|
||||||
|
const fileContent = 'Hey there, \ntest line\ntest line';
|
||||||
|
const expectedContent = 'Hey there, \nnew line\nnew line';
|
||||||
|
rig.createFile(fileName, fileContent);
|
||||||
|
|
||||||
|
const result = await rig.run(
|
||||||
|
`replace "test line" with "new line" in ${fileName}`,
|
||||||
|
);
|
||||||
|
|
||||||
|
const foundToolCall = await rig.waitForAnyToolCall([
|
||||||
|
'replace',
|
||||||
|
'write_file',
|
||||||
|
]);
|
||||||
|
if (!foundToolCall) {
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(
|
||||||
|
foundToolCall,
|
||||||
|
'Expected to find a replace or write_file tool call',
|
||||||
|
).toBeTruthy();
|
||||||
|
|
||||||
|
const toolLogs = rig.readToolLogs();
|
||||||
|
const successfulEdit = toolLogs.some(
|
||||||
|
(log) =>
|
||||||
|
(log.toolRequest.name === 'replace' ||
|
||||||
|
log.toolRequest.name === 'write_file') &&
|
||||||
|
log.toolRequest.success,
|
||||||
|
);
|
||||||
|
if (!successfulEdit) {
|
||||||
|
console.error(
|
||||||
|
'Expected a successful edit tool call, but none was found.',
|
||||||
|
);
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(successfulEdit, 'Expected a successful edit tool call').toBeTruthy();
|
||||||
|
|
||||||
|
const newFileContent = rig.readFile(fileName);
|
||||||
|
expect(newFileContent).toBe(expectedContent);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should fail safely when trying to edit a non-existent file', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup(
|
||||||
|
'should fail safely when trying to edit a non-existent file',
|
||||||
|
);
|
||||||
|
const fileName = 'non_existent.txt';
|
||||||
|
|
||||||
|
const result = await rig.run(`In ${fileName}, replace "a" with "b"`);
|
||||||
|
|
||||||
|
await rig.waitForTelemetryReady();
|
||||||
|
const toolLogs = rig.readToolLogs();
|
||||||
|
|
||||||
|
const readAttempt = toolLogs.find(
|
||||||
|
(log) => log.toolRequest.name === 'read_file',
|
||||||
|
);
|
||||||
|
const writeAttempt = toolLogs.find(
|
||||||
|
(log) => log.toolRequest.name === 'write_file',
|
||||||
|
);
|
||||||
|
const successfulReplace = toolLogs.find(
|
||||||
|
(log) => log.toolRequest.name === 'replace' && log.toolRequest.success,
|
||||||
|
);
|
||||||
|
|
||||||
|
// The model can either investigate (and fail) or do nothing.
|
||||||
|
// If it chose to investigate by reading, that read must have failed.
|
||||||
|
if (readAttempt && readAttempt.toolRequest.success) {
|
||||||
|
console.error(
|
||||||
|
'A read_file attempt succeeded for a non-existent file when it should have failed.',
|
||||||
|
);
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
if (readAttempt) {
|
||||||
|
expect(
|
||||||
|
readAttempt.toolRequest.success,
|
||||||
|
'If model tries to read the file, that attempt must fail',
|
||||||
|
).toBe(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
// CRITICAL: Verify that no matter what the model did, it never successfully
|
||||||
|
// wrote or replaced anything.
|
||||||
|
if (writeAttempt) {
|
||||||
|
console.error(
|
||||||
|
'A write_file attempt was made when no file should be written.',
|
||||||
|
);
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(
|
||||||
|
writeAttempt,
|
||||||
|
'write_file should not have been called',
|
||||||
|
).toBeUndefined();
|
||||||
|
|
||||||
|
if (successfulReplace) {
|
||||||
|
console.error('A successful replace occurred when it should not have.');
|
||||||
|
printDebugInfo(rig, result);
|
||||||
|
}
|
||||||
|
expect(
|
||||||
|
successfulReplace,
|
||||||
|
'A successful replace should not have occurred',
|
||||||
|
).toBeUndefined();
|
||||||
|
|
||||||
|
// Final verification: ensure the file was not created.
|
||||||
|
const filePath = path.join(rig.testDir!, fileName);
|
||||||
|
const fileExists = existsSync(filePath);
|
||||||
|
expect(fileExists, 'The non-existent file should not be created').toBe(
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -22,7 +22,7 @@ import { fileURLToPath } from 'node:url';
|
|||||||
import * as os from 'node:os';
|
import * as os from 'node:os';
|
||||||
|
|
||||||
import {
|
import {
|
||||||
GEMINI_CONFIG_DIR,
|
QWEN_CONFIG_DIR,
|
||||||
DEFAULT_CONTEXT_FILENAME,
|
DEFAULT_CONTEXT_FILENAME,
|
||||||
} from '../packages/core/src/tools/memoryTool.js';
|
} from '../packages/core/src/tools/memoryTool.js';
|
||||||
|
|
||||||
@@ -33,7 +33,7 @@ let runDir = ''; // Make runDir accessible in teardown
|
|||||||
|
|
||||||
const memoryFilePath = join(
|
const memoryFilePath = join(
|
||||||
os.homedir(),
|
os.homedir(),
|
||||||
GEMINI_CONFIG_DIR,
|
QWEN_CONFIG_DIR,
|
||||||
DEFAULT_CONTEXT_FILENAME,
|
DEFAULT_CONTEXT_FILENAME,
|
||||||
);
|
);
|
||||||
let originalMemoryContent: string | null = null;
|
let originalMemoryContent: string | null = null;
|
||||||
|
|||||||
@@ -1,201 +0,0 @@
|
|||||||
/**
|
|
||||||
* @license
|
|
||||||
* Copyright 2025 Google LLC
|
|
||||||
* SPDX-License-Identifier: Apache-2.0
|
|
||||||
*/
|
|
||||||
|
|
||||||
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
|
|
||||||
import * as fs from 'node:fs';
|
|
||||||
import * as os from 'node:os';
|
|
||||||
import * as path from 'node:path';
|
|
||||||
import * as net from 'node:net';
|
|
||||||
import * as child_process from 'node:child_process';
|
|
||||||
import { IdeClient } from '../packages/core/src/ide/ide-client.js';
|
|
||||||
|
|
||||||
import { TestMcpServer } from './test-mcp-server.js';
|
|
||||||
|
|
||||||
describe.skip('IdeClient', () => {
|
|
||||||
it('reads port from file and connects', async () => {
|
|
||||||
const server = new TestMcpServer();
|
|
||||||
const port = await server.start();
|
|
||||||
const pid = process.pid;
|
|
||||||
const portFile = path.join(os.tmpdir(), `qwen-code-ide-server-${pid}.json`);
|
|
||||||
fs.writeFileSync(portFile, JSON.stringify({ port }));
|
|
||||||
process.env['QWEN_CODE_IDE_WORKSPACE_PATH'] = process.cwd();
|
|
||||||
process.env['TERM_PROGRAM'] = 'vscode';
|
|
||||||
|
|
||||||
const ideClient = await IdeClient.getInstance();
|
|
||||||
await ideClient.connect();
|
|
||||||
|
|
||||||
expect(ideClient.getConnectionStatus()).toEqual({
|
|
||||||
status: 'connected',
|
|
||||||
details: undefined,
|
|
||||||
});
|
|
||||||
|
|
||||||
fs.unlinkSync(portFile);
|
|
||||||
await server.stop();
|
|
||||||
delete process.env['QWEN_CODE_IDE_WORKSPACE_PATH'];
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
const getFreePort = (): Promise<number> => {
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
const server = net.createServer();
|
|
||||||
server.unref();
|
|
||||||
server.on('error', reject);
|
|
||||||
server.listen(0, () => {
|
|
||||||
const port = (server.address() as net.AddressInfo).port;
|
|
||||||
server.close(() => {
|
|
||||||
resolve(port);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
describe('IdeClient fallback connection logic', () => {
|
|
||||||
let server: TestMcpServer;
|
|
||||||
let envPort: number;
|
|
||||||
let pid: number;
|
|
||||||
let portFile: string;
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
pid = process.pid;
|
|
||||||
portFile = path.join(os.tmpdir(), `qwen-code-ide-server-${pid}.json`);
|
|
||||||
server = new TestMcpServer();
|
|
||||||
envPort = await server.start();
|
|
||||||
process.env['QWEN_CODE_IDE_SERVER_PORT'] = String(envPort);
|
|
||||||
process.env['TERM_PROGRAM'] = 'vscode';
|
|
||||||
process.env['QWEN_CODE_IDE_WORKSPACE_PATH'] = process.cwd();
|
|
||||||
// Reset instance
|
|
||||||
(IdeClient as unknown as { instance: IdeClient | undefined }).instance =
|
|
||||||
undefined;
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(async () => {
|
|
||||||
await server.stop();
|
|
||||||
delete process.env['QWEN_CODE_IDE_SERVER_PORT'];
|
|
||||||
delete process.env['QWEN_CODE_IDE_WORKSPACE_PATH'];
|
|
||||||
if (fs.existsSync(portFile)) {
|
|
||||||
fs.unlinkSync(portFile);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
it('connects using env var when port file does not exist', async () => {
|
|
||||||
// Ensure port file doesn't exist
|
|
||||||
if (fs.existsSync(portFile)) {
|
|
||||||
fs.unlinkSync(portFile);
|
|
||||||
}
|
|
||||||
|
|
||||||
const ideClient = await IdeClient.getInstance();
|
|
||||||
await ideClient.connect();
|
|
||||||
|
|
||||||
expect(ideClient.getConnectionStatus()).toEqual({
|
|
||||||
status: 'connected',
|
|
||||||
details: undefined,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('falls back to env var when connection with port from file fails', async () => {
|
|
||||||
const filePort = await getFreePort();
|
|
||||||
// Write port file with a port that is not listening
|
|
||||||
fs.writeFileSync(portFile, JSON.stringify({ port: filePort }));
|
|
||||||
|
|
||||||
const ideClient = await IdeClient.getInstance();
|
|
||||||
await ideClient.connect();
|
|
||||||
|
|
||||||
expect(ideClient.getConnectionStatus()).toEqual({
|
|
||||||
status: 'connected',
|
|
||||||
details: undefined,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe.skip('getIdeProcessId', () => {
|
|
||||||
let child: child_process.ChildProcess;
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
if (child) {
|
|
||||||
child.kill();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should return the pid of the parent process', async () => {
|
|
||||||
// We need to spawn a child process that will run the test
|
|
||||||
// so that we can check that getIdeProcessId returns the pid of the parent
|
|
||||||
const parentPid = process.pid;
|
|
||||||
const output = await new Promise<string>((resolve, reject) => {
|
|
||||||
child = child_process.spawn(
|
|
||||||
'node',
|
|
||||||
[
|
|
||||||
'-e',
|
|
||||||
`
|
|
||||||
const { getIdeProcessId } = require('../packages/core/src/ide/process-utils.js');
|
|
||||||
getIdeProcessId().then(pid => console.log(pid));
|
|
||||||
`,
|
|
||||||
],
|
|
||||||
{
|
|
||||||
stdio: ['pipe', 'pipe', 'pipe'],
|
|
||||||
},
|
|
||||||
);
|
|
||||||
|
|
||||||
let out = '';
|
|
||||||
child.stdout?.on('data', (data) => {
|
|
||||||
out += data.toString();
|
|
||||||
});
|
|
||||||
|
|
||||||
child.on('close', (code) => {
|
|
||||||
if (code === 0) {
|
|
||||||
resolve(out.trim());
|
|
||||||
} else {
|
|
||||||
reject(new Error(`Child process exited with code ${code}`));
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(parseInt(output, 10)).toBe(parentPid);
|
|
||||||
}, 10000);
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('IdeClient with proxy', () => {
|
|
||||||
let mcpServer: TestMcpServer;
|
|
||||||
let proxyServer: net.Server;
|
|
||||||
let mcpServerPort: number;
|
|
||||||
let proxyServerPort: number;
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
mcpServer = new TestMcpServer();
|
|
||||||
mcpServerPort = await mcpServer.start();
|
|
||||||
|
|
||||||
proxyServer = net.createServer().listen();
|
|
||||||
proxyServerPort = (proxyServer.address() as net.AddressInfo).port;
|
|
||||||
|
|
||||||
vi.stubEnv('QWEN_CODE_IDE_SERVER_PORT', String(mcpServerPort));
|
|
||||||
vi.stubEnv('TERM_PROGRAM', 'vscode');
|
|
||||||
vi.stubEnv('QWEN_CODE_IDE_WORKSPACE_PATH', process.cwd());
|
|
||||||
|
|
||||||
// Reset instance
|
|
||||||
(IdeClient as unknown as { instance: IdeClient | undefined }).instance =
|
|
||||||
undefined;
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(async () => {
|
|
||||||
(await IdeClient.getInstance()).disconnect();
|
|
||||||
await mcpServer.stop();
|
|
||||||
proxyServer.close();
|
|
||||||
vi.unstubAllEnvs();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should connect to IDE server when HTTP_PROXY, HTTPS_PROXY and NO_PROXY are set', async () => {
|
|
||||||
vi.stubEnv('HTTP_PROXY', `http://localhost:${proxyServerPort}`);
|
|
||||||
vi.stubEnv('HTTPS_PROXY', `http://localhost:${proxyServerPort}`);
|
|
||||||
vi.stubEnv('NO_PROXY', 'example.com,127.0.0.1,::1');
|
|
||||||
|
|
||||||
const ideClient = await IdeClient.getInstance();
|
|
||||||
await ideClient.connect();
|
|
||||||
|
|
||||||
expect(ideClient.getConnectionStatus()).toEqual({
|
|
||||||
status: 'connected',
|
|
||||||
details: undefined,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
89
integration-tests/json-output.test.ts
Normal file
89
integration-tests/json-output.test.ts
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { expect, describe, it, beforeEach, afterEach } from 'vitest';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
|
||||||
|
describe('JSON output', () => {
|
||||||
|
let rig: TestRig;
|
||||||
|
|
||||||
|
beforeEach(async () => {
|
||||||
|
rig = new TestRig();
|
||||||
|
await rig.setup('json-output-test');
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(async () => {
|
||||||
|
await rig.cleanup();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return a valid JSON with response and stats', async () => {
|
||||||
|
const result = await rig.run(
|
||||||
|
'What is the capital of France?',
|
||||||
|
'--output-format',
|
||||||
|
'json',
|
||||||
|
);
|
||||||
|
const parsed = JSON.parse(result);
|
||||||
|
|
||||||
|
expect(parsed).toHaveProperty('response');
|
||||||
|
expect(typeof parsed.response).toBe('string');
|
||||||
|
expect(parsed.response.toLowerCase()).toContain('paris');
|
||||||
|
|
||||||
|
expect(parsed).toHaveProperty('stats');
|
||||||
|
expect(typeof parsed.stats).toBe('object');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return a JSON error for enforced auth mismatch before running', async () => {
|
||||||
|
process.env['GOOGLE_GENAI_USE_GCA'] = 'true';
|
||||||
|
await rig.setup('json-output-auth-mismatch', {
|
||||||
|
settings: {
|
||||||
|
security: { auth: { enforcedType: 'gemini-api-key' } },
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
let thrown: Error | undefined;
|
||||||
|
try {
|
||||||
|
await rig.run('Hello', '--output-format', 'json');
|
||||||
|
expect.fail('Expected process to exit with error');
|
||||||
|
} catch (e) {
|
||||||
|
thrown = e as Error;
|
||||||
|
} finally {
|
||||||
|
delete process.env['GOOGLE_GENAI_USE_GCA'];
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(thrown).toBeDefined();
|
||||||
|
const message = (thrown as Error).message;
|
||||||
|
|
||||||
|
// Use a regex to find the first complete JSON object in the string
|
||||||
|
const jsonMatch = message.match(/{[\s\S]*}/);
|
||||||
|
|
||||||
|
// Fail if no JSON-like text was found
|
||||||
|
expect(
|
||||||
|
jsonMatch,
|
||||||
|
'Expected to find a JSON object in the error output',
|
||||||
|
).toBeTruthy();
|
||||||
|
|
||||||
|
let payload;
|
||||||
|
try {
|
||||||
|
// Parse the matched JSON string
|
||||||
|
payload = JSON.parse(jsonMatch![0]);
|
||||||
|
} catch (parseError) {
|
||||||
|
console.error('Failed to parse the following JSON:', jsonMatch![0]);
|
||||||
|
throw new Error(
|
||||||
|
`Test failed: Could not parse JSON from error message. Details: ${parseError}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(payload.error).toBeDefined();
|
||||||
|
expect(payload.error.type).toBe('Error');
|
||||||
|
expect(payload.error.code).toBe(1);
|
||||||
|
expect(payload.error.message).toContain(
|
||||||
|
'configured auth type is gemini-api-key',
|
||||||
|
);
|
||||||
|
expect(payload.error.message).toContain(
|
||||||
|
'current auth type is oauth-personal',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -29,7 +29,7 @@ describe('list_directory', () => {
|
|||||||
50, // check every 50ms
|
50, // check every 50ms
|
||||||
);
|
);
|
||||||
|
|
||||||
const prompt = `Can you list the files in the current directory. Display them in the style of 'ls'`;
|
const prompt = `Can you list the files in the current directory.`;
|
||||||
|
|
||||||
const result = await rig.run(prompt);
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
|
|||||||
@@ -5,14 +5,26 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This test verifies we can match maximum schema depth errors from Gemini
|
* This test verifies we can provide MCP tools with recursive input schemas
|
||||||
* and then detect and warn about the potential tools that caused the error.
|
* (in JSON, using the $ref keyword) and both the GenAI SDK and the Gemini
|
||||||
|
* API calls succeed. Note that prior to
|
||||||
|
* https://github.com/googleapis/js-genai/commit/36f6350705ecafc47eaea3f3eecbcc69512edab7#diff-fdde9372aec859322b7c5a5efe467e0ad25a57210c7229724586ee90ea4f5a30
|
||||||
|
* the Gemini API call would fail for such tools because the schema was
|
||||||
|
* passed not as a JSON string but using the Gemini API's tool parameter
|
||||||
|
* schema object which has stricter typing and recursion restrictions.
|
||||||
|
* If this test fails, it's likely because either the GenAI SDK or Gemini API
|
||||||
|
* has become more restrictive about the type of tool parameter schemas that
|
||||||
|
* are accepted. If this occurs: Gemini CLI previously attempted to detect
|
||||||
|
* such tools and proactively remove them from the set of tools provided in
|
||||||
|
* the Gemini API call (as FunctionDeclaration objects). It may be appropriate
|
||||||
|
* to resurrect that behavior but note that it's difficult to keep the
|
||||||
|
* GCLI filters in sync with the Gemini API restrictions and behavior.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import { describe, it, beforeAll, expect } from 'vitest';
|
|
||||||
import { TestRig } from './test-helper.js';
|
|
||||||
import { join } from 'node:path';
|
|
||||||
import { writeFileSync } from 'node:fs';
|
import { writeFileSync } from 'node:fs';
|
||||||
|
import { join } from 'node:path';
|
||||||
|
import { beforeAll, describe, expect, it } from 'vitest';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
|
||||||
// Create a minimal MCP server that doesn't require external dependencies
|
// Create a minimal MCP server that doesn't require external dependencies
|
||||||
// This implements the MCP protocol directly using Node.js built-ins
|
// This implements the MCP protocol directly using Node.js built-ins
|
||||||
@@ -180,15 +192,16 @@ describe('mcp server with cyclic tool schema is detected', () => {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should error and suggest disabling the cyclic tool', async () => {
|
it('mcp tool with cyclic schema should be accessible', async () => {
|
||||||
// Just run any command to trigger the schema depth error.
|
const mcp_list_output = await rig.runCommand(['mcp', 'list']);
|
||||||
// If this test starts failing, check `isSchemaDepthError` from
|
|
||||||
// geminiChat.ts to see if it needs to be updated.
|
|
||||||
// Or, possibly it could mean that gemini has fixed the issue.
|
|
||||||
const output = await rig.run('hello');
|
|
||||||
|
|
||||||
expect(output).toMatch(
|
// Verify the cyclic schema server is configured
|
||||||
/Skipping tool 'tool_with_cyclic_schema' from MCP server 'cyclic-schema-server' because it has missing types in its parameter schema/,
|
expect(mcp_list_output).toContain('cyclic-schema-server');
|
||||||
);
|
});
|
||||||
|
|
||||||
|
it('gemini api call should be successful with cyclic mcp tool schema', async () => {
|
||||||
|
// Run any command and verify that we get a non-error response from
|
||||||
|
// the Gemini API.
|
||||||
|
await rig.run('hello');
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
62
integration-tests/mixed-input-crash.test.ts
Normal file
62
integration-tests/mixed-input-crash.test.ts
Normal file
@@ -0,0 +1,62 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect } from 'vitest';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
|
||||||
|
describe('mixed input crash prevention', () => {
|
||||||
|
it('should not crash when using mixed prompt inputs', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
rig.setup('should not crash when using mixed prompt inputs');
|
||||||
|
|
||||||
|
// Test: echo "say '1'." | gemini --prompt-interactive="say '2'." say '3'.
|
||||||
|
const stdinContent = "say '1'.";
|
||||||
|
|
||||||
|
try {
|
||||||
|
await rig.run(
|
||||||
|
{ stdin: stdinContent },
|
||||||
|
'--prompt-interactive',
|
||||||
|
"say '2'.",
|
||||||
|
"say '3'.",
|
||||||
|
);
|
||||||
|
throw new Error('Expected the command to fail, but it succeeded');
|
||||||
|
} catch (error: unknown) {
|
||||||
|
expect(error).toBeInstanceOf(Error);
|
||||||
|
const err = error as Error;
|
||||||
|
|
||||||
|
expect(err.message).toContain('Process exited with code 1');
|
||||||
|
expect(err.message).toContain(
|
||||||
|
'--prompt-interactive flag cannot be used when input is piped',
|
||||||
|
);
|
||||||
|
expect(err.message).not.toContain('setRawMode is not a function');
|
||||||
|
expect(err.message).not.toContain('unexpected critical error');
|
||||||
|
}
|
||||||
|
|
||||||
|
const lastRequest = rig.readLastApiRequest();
|
||||||
|
expect(lastRequest).toBeNull();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should provide clear error message for mixed input', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
rig.setup('should provide clear error message for mixed input');
|
||||||
|
|
||||||
|
try {
|
||||||
|
await rig.run(
|
||||||
|
{ stdin: 'test input' },
|
||||||
|
'--prompt-interactive',
|
||||||
|
'test prompt',
|
||||||
|
);
|
||||||
|
throw new Error('Expected the command to fail, but it succeeded');
|
||||||
|
} catch (error: unknown) {
|
||||||
|
expect(error).toBeInstanceOf(Error);
|
||||||
|
const err = error as Error;
|
||||||
|
|
||||||
|
expect(err.message).toContain(
|
||||||
|
'--prompt-interactive flag cannot be used when input is piped',
|
||||||
|
);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -14,7 +14,7 @@ describe('read_many_files', () => {
|
|||||||
rig.createFile('file1.txt', 'file 1 content');
|
rig.createFile('file1.txt', 'file 1 content');
|
||||||
rig.createFile('file2.txt', 'file 2 content');
|
rig.createFile('file2.txt', 'file 2 content');
|
||||||
|
|
||||||
const prompt = `Please use read_many_files to read file1.txt and file2.txt and show me what's in them`;
|
const prompt = `Use the read_many_files tool to read the contents of file1.txt and file2.txt and then print the contents of each file.`;
|
||||||
|
|
||||||
const result = await rig.run(prompt);
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
@@ -41,11 +41,7 @@ describe('read_many_files', () => {
|
|||||||
'Expected to find either read_many_files or multiple read_file tool calls',
|
'Expected to find either read_many_files or multiple read_file tool calls',
|
||||||
).toBeTruthy();
|
).toBeTruthy();
|
||||||
|
|
||||||
// Validate model output - will throw if no output, warn if missing expected content
|
// Validate model output - will throw if no output
|
||||||
validateModelOutput(
|
validateModelOutput(result, null, 'Read many files test');
|
||||||
result,
|
|
||||||
['file 1 content', 'file 2 content'],
|
|
||||||
'Read many files test',
|
|
||||||
);
|
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -67,4 +67,64 @@ describe('run_shell_command', () => {
|
|||||||
// Validate model output - will throw if no output, warn if missing expected content
|
// Validate model output - will throw if no output, warn if missing expected content
|
||||||
validateModelOutput(result, 'test-stdin', 'Shell command stdin test');
|
validateModelOutput(result, 'test-stdin', 'Shell command stdin test');
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('should propagate environment variables to the child process', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should propagate environment variables');
|
||||||
|
|
||||||
|
const varName = 'GEMINI_CLI_TEST_VAR';
|
||||||
|
const varValue = `test-value-${Math.random().toString(36).substring(7)}`;
|
||||||
|
process.env[varName] = varValue;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const prompt = `Use echo to learn the value of the environment variable named ${varName} and tell me what it is.`;
|
||||||
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
|
const foundToolCall = await rig.waitForToolCall('run_shell_command');
|
||||||
|
|
||||||
|
if (!foundToolCall || !result.includes(varValue)) {
|
||||||
|
printDebugInfo(rig, result, {
|
||||||
|
'Found tool call': foundToolCall,
|
||||||
|
'Contains varValue': result.includes(varValue),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(
|
||||||
|
foundToolCall,
|
||||||
|
'Expected to find a run_shell_command tool call',
|
||||||
|
).toBeTruthy();
|
||||||
|
validateModelOutput(result, varValue, 'Env var propagation test');
|
||||||
|
expect(result).toContain(varValue);
|
||||||
|
} finally {
|
||||||
|
delete process.env[varName];
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should run a platform-specific file listing command', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
await rig.setup('should run platform-specific file listing');
|
||||||
|
const fileName = `test-file-${Math.random().toString(36).substring(7)}.txt`;
|
||||||
|
rig.createFile(fileName, 'test content');
|
||||||
|
|
||||||
|
const prompt = `Run a shell command to list the files in the current directory and tell me what they are.`;
|
||||||
|
const result = await rig.run(prompt);
|
||||||
|
|
||||||
|
const foundToolCall = await rig.waitForToolCall('run_shell_command');
|
||||||
|
|
||||||
|
// Debugging info
|
||||||
|
if (!foundToolCall || !result.includes(fileName)) {
|
||||||
|
printDebugInfo(rig, result, {
|
||||||
|
'Found tool call': foundToolCall,
|
||||||
|
'Contains fileName': result.includes(fileName),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(
|
||||||
|
foundToolCall,
|
||||||
|
'Expected to find a run_shell_command tool call',
|
||||||
|
).toBeTruthy();
|
||||||
|
|
||||||
|
validateModelOutput(result, fileName, 'Platform-specific listing test');
|
||||||
|
expect(result).toContain(fileName);
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -8,7 +8,9 @@ import { describe, it, expect } from 'vitest';
|
|||||||
import { TestRig, printDebugInfo, validateModelOutput } from './test-helper.js';
|
import { TestRig, printDebugInfo, validateModelOutput } from './test-helper.js';
|
||||||
|
|
||||||
describe('save_memory', () => {
|
describe('save_memory', () => {
|
||||||
it('should be able to save to memory', async () => {
|
// Skipped due to flaky model behavior - the model sometimes answers the question
|
||||||
|
// directly without calling the save_memory tool, even when prompted to "remember"
|
||||||
|
it.skip('should be able to save to memory', async () => {
|
||||||
const rig = new TestRig();
|
const rig = new TestRig();
|
||||||
await rig.setup('should be able to save to memory');
|
await rig.setup('should be able to save to memory');
|
||||||
|
|
||||||
|
|||||||
@@ -1,156 +0,0 @@
|
|||||||
/**
|
|
||||||
* @license
|
|
||||||
* Copyright 2025 Google LLC
|
|
||||||
* SPDX-License-Identifier: Apache-2.0
|
|
||||||
*/
|
|
||||||
|
|
||||||
import { describe, it, expect, beforeAll } from 'vitest';
|
|
||||||
import { ShellExecutionService } from '../packages/core/src/services/shellExecutionService.js';
|
|
||||||
import * as fs from 'node:fs/promises';
|
|
||||||
import * as path from 'node:path';
|
|
||||||
import { vi } from 'vitest';
|
|
||||||
|
|
||||||
describe('ShellExecutionService programmatic integration tests', () => {
|
|
||||||
let testDir: string;
|
|
||||||
|
|
||||||
beforeAll(async () => {
|
|
||||||
// Create a dedicated directory for this test suite to avoid conflicts.
|
|
||||||
testDir = path.join(
|
|
||||||
process.env['INTEGRATION_TEST_FILE_DIR']!,
|
|
||||||
'shell-service-tests',
|
|
||||||
);
|
|
||||||
await fs.mkdir(testDir, { recursive: true });
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should execute a simple cross-platform command (echo)', async () => {
|
|
||||||
const command = 'echo "hello from the service"';
|
|
||||||
const onOutputEvent = vi.fn();
|
|
||||||
const abortController = new AbortController();
|
|
||||||
|
|
||||||
const handle = await ShellExecutionService.execute(
|
|
||||||
command,
|
|
||||||
testDir,
|
|
||||||
onOutputEvent,
|
|
||||||
abortController.signal,
|
|
||||||
false,
|
|
||||||
);
|
|
||||||
|
|
||||||
const result = await handle.result;
|
|
||||||
|
|
||||||
expect(result.error).toBeNull();
|
|
||||||
expect(result.exitCode).toBe(0);
|
|
||||||
// Output can vary slightly between shells (e.g., quotes), so check for inclusion.
|
|
||||||
expect(result.output).toContain('hello from the service');
|
|
||||||
});
|
|
||||||
|
|
||||||
it.runIf(process.platform === 'win32')(
|
|
||||||
'should execute "dir" on Windows',
|
|
||||||
async () => {
|
|
||||||
const testFile = 'test-file-windows.txt';
|
|
||||||
await fs.writeFile(path.join(testDir, testFile), 'windows test');
|
|
||||||
|
|
||||||
const command = 'dir';
|
|
||||||
const onOutputEvent = vi.fn();
|
|
||||||
const abortController = new AbortController();
|
|
||||||
|
|
||||||
const handle = await ShellExecutionService.execute(
|
|
||||||
command,
|
|
||||||
testDir,
|
|
||||||
onOutputEvent,
|
|
||||||
abortController.signal,
|
|
||||||
false,
|
|
||||||
);
|
|
||||||
|
|
||||||
const result = await handle.result;
|
|
||||||
|
|
||||||
expect(result.error).toBeNull();
|
|
||||||
expect(result.exitCode).toBe(0);
|
|
||||||
expect(result.output).toContain(testFile);
|
|
||||||
},
|
|
||||||
);
|
|
||||||
|
|
||||||
it.skipIf(process.platform === 'win32')(
|
|
||||||
'should execute "ls -l" on Unix',
|
|
||||||
async () => {
|
|
||||||
const testFile = 'test-file-unix.txt';
|
|
||||||
await fs.writeFile(path.join(testDir, testFile), 'unix test');
|
|
||||||
|
|
||||||
const command = 'ls -l';
|
|
||||||
const onOutputEvent = vi.fn();
|
|
||||||
const abortController = new AbortController();
|
|
||||||
|
|
||||||
const handle = await ShellExecutionService.execute(
|
|
||||||
command,
|
|
||||||
testDir,
|
|
||||||
onOutputEvent,
|
|
||||||
abortController.signal,
|
|
||||||
false,
|
|
||||||
);
|
|
||||||
|
|
||||||
const result = await handle.result;
|
|
||||||
|
|
||||||
expect(result.error).toBeNull();
|
|
||||||
expect(result.exitCode).toBe(0);
|
|
||||||
expect(result.output).toContain(testFile);
|
|
||||||
},
|
|
||||||
);
|
|
||||||
|
|
||||||
it('should abort a running process', async () => {
|
|
||||||
// A command that runs for a bit. 'sleep' on unix, 'timeout' on windows.
|
|
||||||
const command = process.platform === 'win32' ? 'timeout /t 20' : 'sleep 20';
|
|
||||||
const onOutputEvent = vi.fn();
|
|
||||||
const abortController = new AbortController();
|
|
||||||
|
|
||||||
const handle = await ShellExecutionService.execute(
|
|
||||||
command,
|
|
||||||
testDir,
|
|
||||||
onOutputEvent,
|
|
||||||
abortController.signal,
|
|
||||||
false,
|
|
||||||
);
|
|
||||||
|
|
||||||
// Abort shortly after starting
|
|
||||||
setTimeout(() => abortController.abort(), 50);
|
|
||||||
|
|
||||||
const result = await handle.result;
|
|
||||||
|
|
||||||
// For debugging the flaky test.
|
|
||||||
console.log('Abort test result:', result);
|
|
||||||
|
|
||||||
expect(result.aborted).toBe(true);
|
|
||||||
// A clean exit is exitCode 0 and no signal. If the process was truly
|
|
||||||
// aborted, it should not have exited cleanly.
|
|
||||||
const exitedCleanly = result.exitCode === 0 && result.signal === null;
|
|
||||||
expect(exitedCleanly, 'Process should not have exited cleanly').toBe(false);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should propagate environment variables to the child process', async () => {
|
|
||||||
const varName = 'QWEN_CODE_TEST_VAR';
|
|
||||||
const varValue = `test-value`;
|
|
||||||
process.env[varName] = varValue;
|
|
||||||
|
|
||||||
try {
|
|
||||||
const command =
|
|
||||||
process.platform === 'win32' ? `echo %${varName}%` : `echo $${varName}`;
|
|
||||||
const onOutputEvent = vi.fn();
|
|
||||||
const abortController = new AbortController();
|
|
||||||
|
|
||||||
const handle = await ShellExecutionService.execute(
|
|
||||||
command,
|
|
||||||
testDir,
|
|
||||||
onOutputEvent,
|
|
||||||
abortController.signal,
|
|
||||||
false,
|
|
||||||
);
|
|
||||||
|
|
||||||
const result = await handle.result;
|
|
||||||
|
|
||||||
expect(result.error).toBeNull();
|
|
||||||
expect(result.exitCode).toBe(0);
|
|
||||||
expect(result.output).toContain(varValue);
|
|
||||||
} finally {
|
|
||||||
// Clean up the env var to prevent side-effects on other tests.
|
|
||||||
delete process.env[varName];
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -189,6 +189,25 @@ describe('simple-mcp-server', () => {
|
|||||||
const { chmodSync } = await import('node:fs');
|
const { chmodSync } = await import('node:fs');
|
||||||
chmodSync(testServerPath, 0o755);
|
chmodSync(testServerPath, 0o755);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Poll for script for up to 5s
|
||||||
|
const { accessSync, constants } = await import('node:fs');
|
||||||
|
const isReady = await rig.poll(
|
||||||
|
() => {
|
||||||
|
try {
|
||||||
|
accessSync(testServerPath, constants.F_OK);
|
||||||
|
return true;
|
||||||
|
} catch {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
5000, // Max wait 5 seconds
|
||||||
|
100, // Poll every 100ms
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!isReady) {
|
||||||
|
throw new Error('MCP server script was not ready in time.');
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should add two numbers', async () => {
|
it('should add two numbers', async () => {
|
||||||
|
|||||||
26
integration-tests/telemetry.test.ts
Normal file
26
integration-tests/telemetry.test.ts
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect } from 'vitest';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
|
||||||
|
describe('telemetry', () => {
|
||||||
|
it('should emit a metric and a log event', async () => {
|
||||||
|
const rig = new TestRig();
|
||||||
|
rig.setup('should emit a metric and a log event');
|
||||||
|
|
||||||
|
// Run a simple command that should trigger telemetry
|
||||||
|
await rig.run('just saying hi');
|
||||||
|
|
||||||
|
// Verify that a user_prompt event was logged
|
||||||
|
const hasUserPromptEvent = await rig.waitForTelemetryEvent('user_prompt');
|
||||||
|
expect(hasUserPromptEvent).toBe(true);
|
||||||
|
|
||||||
|
// Verify that a cli_command_count metric was emitted
|
||||||
|
const cliCommandCountMetric = rig.readMetric('session.count');
|
||||||
|
expect(cliCommandCountMetric).not.toBeNull();
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -5,13 +5,15 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
import { execSync, spawn } from 'node:child_process';
|
import { execSync, spawn } from 'node:child_process';
|
||||||
import { parse } from 'shell-quote';
|
|
||||||
import { mkdirSync, writeFileSync, readFileSync } from 'node:fs';
|
import { mkdirSync, writeFileSync, readFileSync } from 'node:fs';
|
||||||
import { join, dirname } from 'node:path';
|
import { join, dirname } from 'node:path';
|
||||||
import { fileURLToPath } from 'node:url';
|
import { fileURLToPath } from 'node:url';
|
||||||
import { env } from 'node:process';
|
import { env } from 'node:process';
|
||||||
|
import { DEFAULT_QWEN_MODEL } from '../packages/core/src/config/models.js';
|
||||||
import fs from 'node:fs';
|
import fs from 'node:fs';
|
||||||
import { EOL } from 'node:os';
|
import { EOL } from 'node:os';
|
||||||
|
import * as pty from '@lydell/node-pty';
|
||||||
|
import stripAnsi from 'strip-ansi';
|
||||||
|
|
||||||
const __dirname = dirname(fileURLToPath(import.meta.url));
|
const __dirname = dirname(fileURLToPath(import.meta.url));
|
||||||
|
|
||||||
@@ -112,11 +114,38 @@ export function validateModelOutput(
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Simulates typing a string one character at a time to avoid paste detection.
|
||||||
|
export async function type(ptyProcess: pty.IPty, text: string) {
|
||||||
|
const delay = 5;
|
||||||
|
for (const char of text) {
|
||||||
|
ptyProcess.write(char);
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, delay));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ParsedLog {
|
||||||
|
attributes?: {
|
||||||
|
'event.name'?: string;
|
||||||
|
function_name?: string;
|
||||||
|
function_args?: string;
|
||||||
|
success?: boolean;
|
||||||
|
duration_ms?: number;
|
||||||
|
};
|
||||||
|
scopeMetrics?: {
|
||||||
|
metrics: {
|
||||||
|
descriptor: {
|
||||||
|
name: string;
|
||||||
|
};
|
||||||
|
}[];
|
||||||
|
}[];
|
||||||
|
}
|
||||||
|
|
||||||
export class TestRig {
|
export class TestRig {
|
||||||
bundlePath: string;
|
bundlePath: string;
|
||||||
testDir: string | null;
|
testDir: string | null;
|
||||||
testName?: string;
|
testName?: string;
|
||||||
_lastRunStdout?: string;
|
_lastRunStdout?: string;
|
||||||
|
_interactiveOutput = '';
|
||||||
|
|
||||||
constructor() {
|
constructor() {
|
||||||
this.bundlePath = join(__dirname, '..', 'bundle/gemini.js');
|
this.bundlePath = join(__dirname, '..', 'bundle/gemini.js');
|
||||||
@@ -140,8 +169,8 @@ export class TestRig {
|
|||||||
mkdirSync(this.testDir, { recursive: true });
|
mkdirSync(this.testDir, { recursive: true });
|
||||||
|
|
||||||
// Create a settings file to point the CLI to the local collector
|
// Create a settings file to point the CLI to the local collector
|
||||||
const geminiDir = join(this.testDir, '.qwen');
|
const qwenDir = join(this.testDir, '.qwen');
|
||||||
mkdirSync(geminiDir, { recursive: true });
|
mkdirSync(qwenDir, { recursive: true });
|
||||||
// In sandbox mode, use an absolute path for telemetry inside the container
|
// In sandbox mode, use an absolute path for telemetry inside the container
|
||||||
// The container mounts the test directory at the same path as the host
|
// The container mounts the test directory at the same path as the host
|
||||||
const telemetryPath = join(this.testDir, 'telemetry.log'); // Always use test directory for telemetry
|
const telemetryPath = join(this.testDir, 'telemetry.log'); // Always use test directory for telemetry
|
||||||
@@ -153,12 +182,12 @@ export class TestRig {
|
|||||||
otlpEndpoint: '',
|
otlpEndpoint: '',
|
||||||
outfile: telemetryPath,
|
outfile: telemetryPath,
|
||||||
},
|
},
|
||||||
sandbox:
|
model: DEFAULT_QWEN_MODEL,
|
||||||
env['GEMINI_SANDBOX'] !== 'false' ? env['GEMINI_SANDBOX'] : false,
|
sandbox: env.GEMINI_SANDBOX !== 'false' ? env.GEMINI_SANDBOX : false,
|
||||||
...options.settings, // Allow tests to override/add settings
|
...options.settings, // Allow tests to override/add settings
|
||||||
};
|
};
|
||||||
writeFileSync(
|
writeFileSync(
|
||||||
join(geminiDir, 'settings.json'),
|
join(qwenDir, 'settings.json'),
|
||||||
JSON.stringify(settings, null, 2),
|
JSON.stringify(settings, null, 2),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -178,13 +207,32 @@ export class TestRig {
|
|||||||
execSync('sync', { cwd: this.testDir! });
|
execSync('sync', { cwd: this.testDir! });
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The command and args to use to invoke Qwen Code CLI. Allows us to switch
|
||||||
|
* between using the bundled gemini.js (the default) and using the installed
|
||||||
|
* 'qwen' (used to verify npm bundles).
|
||||||
|
*/
|
||||||
|
private _getCommandAndArgs(extraInitialArgs: string[] = []): {
|
||||||
|
command: string;
|
||||||
|
initialArgs: string[];
|
||||||
|
} {
|
||||||
|
const isNpmReleaseTest =
|
||||||
|
process.env.INTEGRATION_TEST_USE_INSTALLED_GEMINI === 'true';
|
||||||
|
const command = isNpmReleaseTest ? 'qwen' : 'node';
|
||||||
|
const initialArgs = isNpmReleaseTest
|
||||||
|
? extraInitialArgs
|
||||||
|
: [this.bundlePath, ...extraInitialArgs];
|
||||||
|
return { command, initialArgs };
|
||||||
|
}
|
||||||
|
|
||||||
run(
|
run(
|
||||||
promptOrOptions:
|
promptOrOptions:
|
||||||
| string
|
| string
|
||||||
| { prompt?: string; stdin?: string; stdinDoesNotEnd?: boolean },
|
| { prompt?: string; stdin?: string; stdinDoesNotEnd?: boolean },
|
||||||
...args: string[]
|
...args: string[]
|
||||||
): Promise<string> {
|
): Promise<string> {
|
||||||
let command = `node ${this.bundlePath} --yolo`;
|
const { command, initialArgs } = this._getCommandAndArgs(['--yolo']);
|
||||||
|
const commandArgs = [...initialArgs];
|
||||||
const execOptions: {
|
const execOptions: {
|
||||||
cwd: string;
|
cwd: string;
|
||||||
encoding: 'utf-8';
|
encoding: 'utf-8';
|
||||||
@@ -195,27 +243,25 @@ export class TestRig {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if (typeof promptOrOptions === 'string') {
|
if (typeof promptOrOptions === 'string') {
|
||||||
command += ` --prompt ${JSON.stringify(promptOrOptions)}`;
|
commandArgs.push('--prompt', promptOrOptions);
|
||||||
} else if (
|
} else if (
|
||||||
typeof promptOrOptions === 'object' &&
|
typeof promptOrOptions === 'object' &&
|
||||||
promptOrOptions !== null
|
promptOrOptions !== null
|
||||||
) {
|
) {
|
||||||
if (promptOrOptions.prompt) {
|
if (promptOrOptions.prompt) {
|
||||||
command += ` --prompt ${JSON.stringify(promptOrOptions.prompt)}`;
|
commandArgs.push('--prompt', promptOrOptions.prompt);
|
||||||
}
|
}
|
||||||
if (promptOrOptions.stdin) {
|
if (promptOrOptions.stdin) {
|
||||||
execOptions.input = promptOrOptions.stdin;
|
execOptions.input = promptOrOptions.stdin;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
command += ` ${args.join(' ')}`;
|
commandArgs.push(...args);
|
||||||
|
|
||||||
const commandArgs = parse(command);
|
const child = spawn(command, commandArgs, {
|
||||||
const node = commandArgs.shift() as string;
|
|
||||||
|
|
||||||
const child = spawn(node, commandArgs as string[], {
|
|
||||||
cwd: this.testDir!,
|
cwd: this.testDir!,
|
||||||
stdio: 'pipe',
|
stdio: 'pipe',
|
||||||
|
env: process.env,
|
||||||
});
|
});
|
||||||
|
|
||||||
let stdout = '';
|
let stdout = '';
|
||||||
@@ -291,8 +337,15 @@ export class TestRig {
|
|||||||
|
|
||||||
result = filteredLines.join('\n');
|
result = filteredLines.join('\n');
|
||||||
}
|
}
|
||||||
// If we have stderr output, include that also
|
|
||||||
if (stderr) {
|
// Check if this is a JSON output test - if so, don't include stderr
|
||||||
|
// as it would corrupt the JSON
|
||||||
|
const isJsonOutput =
|
||||||
|
commandArgs.includes('--output-format') &&
|
||||||
|
commandArgs.includes('json');
|
||||||
|
|
||||||
|
// If we have stderr output and it's not a JSON test, include that also
|
||||||
|
if (stderr && !isJsonOutput) {
|
||||||
result += `\n\nStdErr:\n${stderr}`;
|
result += `\n\nStdErr:\n${stderr}`;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -306,6 +359,58 @@ export class TestRig {
|
|||||||
return promise;
|
return promise;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
runCommand(
|
||||||
|
args: string[],
|
||||||
|
options: { stdin?: string } = {},
|
||||||
|
): Promise<string> {
|
||||||
|
const { command, initialArgs } = this._getCommandAndArgs();
|
||||||
|
const commandArgs = [...initialArgs, ...args];
|
||||||
|
|
||||||
|
const child = spawn(command, commandArgs, {
|
||||||
|
cwd: this.testDir!,
|
||||||
|
stdio: 'pipe',
|
||||||
|
});
|
||||||
|
|
||||||
|
let stdout = '';
|
||||||
|
let stderr = '';
|
||||||
|
|
||||||
|
if (options.stdin) {
|
||||||
|
child.stdin!.write(options.stdin);
|
||||||
|
child.stdin!.end();
|
||||||
|
}
|
||||||
|
|
||||||
|
child.stdout!.on('data', (data: Buffer) => {
|
||||||
|
stdout += data;
|
||||||
|
if (env.KEEP_OUTPUT === 'true' || env.VERBOSE === 'true') {
|
||||||
|
process.stdout.write(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
child.stderr!.on('data', (data: Buffer) => {
|
||||||
|
stderr += data;
|
||||||
|
if (env.KEEP_OUTPUT === 'true' || env.VERBOSE === 'true') {
|
||||||
|
process.stderr.write(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
const promise = new Promise<string>((resolve, reject) => {
|
||||||
|
child.on('close', (code: number) => {
|
||||||
|
if (code === 0) {
|
||||||
|
this._lastRunStdout = stdout;
|
||||||
|
let result = stdout;
|
||||||
|
if (stderr) {
|
||||||
|
result += `\n\nStdErr:\n${stderr}`;
|
||||||
|
}
|
||||||
|
resolve(result);
|
||||||
|
} else {
|
||||||
|
reject(new Error(`Process exited with code ${code}:\n${stderr}`));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
return promise;
|
||||||
|
}
|
||||||
|
|
||||||
readFile(fileName: string) {
|
readFile(fileName: string) {
|
||||||
const filePath = join(this.testDir!, fileName);
|
const filePath = join(this.testDir!, fileName);
|
||||||
const content = readFileSync(filePath, 'utf-8');
|
const content = readFileSync(filePath, 'utf-8');
|
||||||
@@ -363,37 +468,12 @@ export class TestRig {
|
|||||||
|
|
||||||
return this.poll(
|
return this.poll(
|
||||||
() => {
|
() => {
|
||||||
const logFilePath = join(this.testDir!, 'telemetry.log');
|
const logs = this._readAndParseTelemetryLog();
|
||||||
|
return logs.some(
|
||||||
if (!logFilePath || !fs.existsSync(logFilePath)) {
|
(logData) =>
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const content = readFileSync(logFilePath, 'utf-8');
|
|
||||||
const jsonObjects = content
|
|
||||||
.split(/}\n{/)
|
|
||||||
.map((obj, index, array) => {
|
|
||||||
// Add back the braces we removed during split
|
|
||||||
if (index > 0) obj = '{' + obj;
|
|
||||||
if (index < array.length - 1) obj = obj + '}';
|
|
||||||
return obj.trim();
|
|
||||||
})
|
|
||||||
.filter((obj) => obj);
|
|
||||||
|
|
||||||
for (const jsonStr of jsonObjects) {
|
|
||||||
try {
|
|
||||||
const logData = JSON.parse(jsonStr);
|
|
||||||
if (
|
|
||||||
logData.attributes &&
|
logData.attributes &&
|
||||||
logData.attributes['event.name'] === `gemini_cli.${eventName}`
|
logData.attributes['event.name'] === `qwen-code.${eventName}`,
|
||||||
) {
|
);
|
||||||
return true;
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// ignore
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
},
|
},
|
||||||
timeout,
|
timeout,
|
||||||
100,
|
100,
|
||||||
@@ -566,7 +646,7 @@ export class TestRig {
|
|||||||
}
|
}
|
||||||
} else if (
|
} else if (
|
||||||
obj.attributes &&
|
obj.attributes &&
|
||||||
obj.attributes['event.name'] === 'gemini_cli.tool_call'
|
obj.attributes['event.name'] === 'qwen-code.tool_call'
|
||||||
) {
|
) {
|
||||||
logs.push({
|
logs.push({
|
||||||
timestamp: obj.attributes['event.timestamp'],
|
timestamp: obj.attributes['event.timestamp'],
|
||||||
@@ -590,6 +670,45 @@ export class TestRig {
|
|||||||
return logs;
|
return logs;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private _readAndParseTelemetryLog(): ParsedLog[] {
|
||||||
|
// Telemetry is always written to the test directory
|
||||||
|
const logFilePath = join(this.testDir!, 'telemetry.log');
|
||||||
|
|
||||||
|
if (!logFilePath || !fs.existsSync(logFilePath)) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
|
||||||
|
const content = readFileSync(logFilePath, 'utf-8');
|
||||||
|
|
||||||
|
// Split the content into individual JSON objects
|
||||||
|
// They are separated by "}\n{"
|
||||||
|
const jsonObjects = content
|
||||||
|
.split(/}\n{/)
|
||||||
|
.map((obj, index, array) => {
|
||||||
|
// Add back the braces we removed during split
|
||||||
|
if (index > 0) obj = '{' + obj;
|
||||||
|
if (index < array.length - 1) obj = obj + '}';
|
||||||
|
return obj.trim();
|
||||||
|
})
|
||||||
|
.filter((obj) => obj);
|
||||||
|
|
||||||
|
const logs: ParsedLog[] = [];
|
||||||
|
|
||||||
|
for (const jsonStr of jsonObjects) {
|
||||||
|
try {
|
||||||
|
const logData = JSON.parse(jsonStr);
|
||||||
|
logs.push(logData);
|
||||||
|
} catch (e) {
|
||||||
|
// Skip objects that aren't valid JSON
|
||||||
|
if (env.VERBOSE === 'true') {
|
||||||
|
console.error('Failed to parse telemetry object:', e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return logs;
|
||||||
|
}
|
||||||
|
|
||||||
readToolLogs() {
|
readToolLogs() {
|
||||||
// For Podman, first check if telemetry file exists and has content
|
// For Podman, first check if telemetry file exists and has content
|
||||||
// If not, fall back to parsing from stdout
|
// If not, fall back to parsing from stdout
|
||||||
@@ -619,33 +738,7 @@ export class TestRig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Telemetry is always written to the test directory
|
const parsedLogs = this._readAndParseTelemetryLog();
|
||||||
const logFilePath = join(this.testDir!, 'telemetry.log');
|
|
||||||
|
|
||||||
if (!logFilePath) {
|
|
||||||
console.warn(`TELEMETRY_LOG_FILE environment variable not set`);
|
|
||||||
return [];
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if file exists, if not return empty array (file might not be created yet)
|
|
||||||
if (!fs.existsSync(logFilePath)) {
|
|
||||||
return [];
|
|
||||||
}
|
|
||||||
|
|
||||||
const content = readFileSync(logFilePath, 'utf-8');
|
|
||||||
|
|
||||||
// Split the content into individual JSON objects
|
|
||||||
// They are separated by "}\n{"
|
|
||||||
const jsonObjects = content
|
|
||||||
.split(/}\n{/)
|
|
||||||
.map((obj, index, array) => {
|
|
||||||
// Add back the braces we removed during split
|
|
||||||
if (index > 0) obj = '{' + obj;
|
|
||||||
if (index < array.length - 1) obj = obj + '}';
|
|
||||||
return obj.trim();
|
|
||||||
})
|
|
||||||
.filter((obj) => obj);
|
|
||||||
|
|
||||||
const logs: {
|
const logs: {
|
||||||
toolRequest: {
|
toolRequest: {
|
||||||
name: string;
|
name: string;
|
||||||
@@ -655,9 +748,7 @@ export class TestRig {
|
|||||||
};
|
};
|
||||||
}[] = [];
|
}[] = [];
|
||||||
|
|
||||||
for (const jsonStr of jsonObjects) {
|
for (const logData of parsedLogs) {
|
||||||
try {
|
|
||||||
const logData = JSON.parse(jsonStr);
|
|
||||||
// Look for tool call logs
|
// Look for tool call logs
|
||||||
if (
|
if (
|
||||||
logData.attributes &&
|
logData.attributes &&
|
||||||
@@ -673,50 +764,85 @@ export class TestRig {
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
} catch (e) {
|
|
||||||
// Skip objects that aren't valid JSON
|
|
||||||
if (env['VERBOSE'] === 'true') {
|
|
||||||
console.error('Failed to parse telemetry object:', e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return logs;
|
return logs;
|
||||||
}
|
}
|
||||||
|
|
||||||
readLastApiRequest(): Record<string, unknown> | null {
|
readLastApiRequest(): Record<string, unknown> | null {
|
||||||
// Telemetry is always written to the test directory
|
const logs = this._readAndParseTelemetryLog();
|
||||||
const logFilePath = join(this.testDir!, 'telemetry.log');
|
const apiRequests = logs.filter(
|
||||||
|
(logData) =>
|
||||||
|
logData.attributes &&
|
||||||
|
logData.attributes['event.name'] === 'qwen-code.api_request',
|
||||||
|
);
|
||||||
|
return apiRequests.pop() || null;
|
||||||
|
}
|
||||||
|
|
||||||
if (!logFilePath || !fs.existsSync(logFilePath)) {
|
readMetric(metricName: string): Record<string, unknown> | null {
|
||||||
|
const logs = this._readAndParseTelemetryLog();
|
||||||
|
for (const logData of logs) {
|
||||||
|
if (logData.scopeMetrics) {
|
||||||
|
for (const scopeMetric of logData.scopeMetrics) {
|
||||||
|
for (const metric of scopeMetric.metrics) {
|
||||||
|
if (metric.descriptor.name === `qwen-code.${metricName}`) {
|
||||||
|
return metric;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
const content = readFileSync(logFilePath, 'utf-8');
|
async waitForText(text: string, timeout?: number): Promise<boolean> {
|
||||||
const jsonObjects = content
|
if (!timeout) {
|
||||||
.split(/}\n{/)
|
timeout = this.getDefaultTimeout();
|
||||||
.map((obj, index, array) => {
|
}
|
||||||
if (index > 0) obj = '{' + obj;
|
return this.poll(
|
||||||
if (index < array.length - 1) obj = obj + '}';
|
() =>
|
||||||
return obj.trim();
|
stripAnsi(this._interactiveOutput)
|
||||||
})
|
.toLowerCase()
|
||||||
.filter((obj) => obj);
|
.includes(text.toLowerCase()),
|
||||||
|
timeout,
|
||||||
|
200,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
let lastApiRequest = null;
|
runInteractive(...args: string[]): {
|
||||||
|
ptyProcess: pty.IPty;
|
||||||
|
promise: Promise<{ exitCode: number; signal?: number; output: string }>;
|
||||||
|
} {
|
||||||
|
const { command, initialArgs } = this._getCommandAndArgs(['--yolo']);
|
||||||
|
const commandArgs = [...initialArgs, ...args];
|
||||||
|
|
||||||
for (const jsonStr of jsonObjects) {
|
this._interactiveOutput = ''; // Reset output for the new run
|
||||||
try {
|
|
||||||
const logData = JSON.parse(jsonStr);
|
const ptyProcess = pty.spawn(command, commandArgs, {
|
||||||
if (
|
name: 'xterm-color',
|
||||||
logData.attributes &&
|
cols: 80,
|
||||||
logData.attributes['event.name'] === 'gemini_cli.api_request'
|
rows: 30,
|
||||||
) {
|
cwd: this.testDir!,
|
||||||
lastApiRequest = logData;
|
env: process.env as { [key: string]: string },
|
||||||
|
});
|
||||||
|
|
||||||
|
ptyProcess.onData((data) => {
|
||||||
|
this._interactiveOutput += data;
|
||||||
|
if (env.KEEP_OUTPUT === 'true' || env.VERBOSE === 'true') {
|
||||||
|
process.stdout.write(data);
|
||||||
}
|
}
|
||||||
} catch {
|
});
|
||||||
// ignore
|
|
||||||
}
|
const promise = new Promise<{
|
||||||
}
|
exitCode: number;
|
||||||
return lastApiRequest;
|
signal?: number;
|
||||||
|
output: string;
|
||||||
|
}>((resolve) => {
|
||||||
|
ptyProcess.onExit(({ exitCode, signal }) => {
|
||||||
|
resolve({ exitCode, signal, output: this._interactiveOutput });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
return { ptyProcess, promise };
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
141
integration-tests/utf-bom-encoding.test.ts
Normal file
141
integration-tests/utf-bom-encoding.test.ts
Normal file
@@ -0,0 +1,141 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect, beforeAll, afterAll } from 'vitest';
|
||||||
|
import { writeFileSync, readFileSync } from 'node:fs';
|
||||||
|
import { join, resolve } from 'node:path';
|
||||||
|
import { TestRig } from './test-helper.js';
|
||||||
|
|
||||||
|
// Windows skip (Option A: avoid infra scope)
|
||||||
|
const d = process.platform === 'win32' ? describe.skip : describe;
|
||||||
|
|
||||||
|
// BOM encoders
|
||||||
|
const utf8BOM = (s: string) =>
|
||||||
|
Buffer.concat([Buffer.from([0xef, 0xbb, 0xbf]), Buffer.from(s, 'utf8')]);
|
||||||
|
const utf16LE = (s: string) =>
|
||||||
|
Buffer.concat([Buffer.from([0xff, 0xfe]), Buffer.from(s, 'utf16le')]);
|
||||||
|
const utf16BE = (s: string) => {
|
||||||
|
const bom = Buffer.from([0xfe, 0xff]);
|
||||||
|
const le = Buffer.from(s, 'utf16le');
|
||||||
|
le.swap16();
|
||||||
|
return Buffer.concat([bom, le]);
|
||||||
|
};
|
||||||
|
const utf32LE = (s: string) => {
|
||||||
|
const bom = Buffer.from([0xff, 0xfe, 0x00, 0x00]);
|
||||||
|
const cps = Array.from(s, (c) => c.codePointAt(0)!);
|
||||||
|
const payload = Buffer.alloc(cps.length * 4);
|
||||||
|
cps.forEach((cp, i) => {
|
||||||
|
const o = i * 4;
|
||||||
|
payload[o] = cp & 0xff;
|
||||||
|
payload[o + 1] = (cp >>> 8) & 0xff;
|
||||||
|
payload[o + 2] = (cp >>> 16) & 0xff;
|
||||||
|
payload[o + 3] = (cp >>> 24) & 0xff;
|
||||||
|
});
|
||||||
|
return Buffer.concat([bom, payload]);
|
||||||
|
};
|
||||||
|
const utf32BE = (s: string) => {
|
||||||
|
const bom = Buffer.from([0x00, 0x00, 0xfe, 0xff]);
|
||||||
|
const cps = Array.from(s, (c) => c.codePointAt(0)!);
|
||||||
|
const payload = Buffer.alloc(cps.length * 4);
|
||||||
|
cps.forEach((cp, i) => {
|
||||||
|
const o = i * 4;
|
||||||
|
payload[o] = (cp >>> 24) & 0xff;
|
||||||
|
payload[o + 1] = (cp >>> 16) & 0xff;
|
||||||
|
payload[o + 2] = (cp >>> 8) & 0xff;
|
||||||
|
payload[o + 3] = cp & 0xff;
|
||||||
|
});
|
||||||
|
return Buffer.concat([bom, payload]);
|
||||||
|
};
|
||||||
|
|
||||||
|
let rig: TestRig;
|
||||||
|
let dir: string;
|
||||||
|
|
||||||
|
d('BOM end-to-end integration', () => {
|
||||||
|
beforeAll(async () => {
|
||||||
|
rig = new TestRig();
|
||||||
|
await rig.setup('bom-integration');
|
||||||
|
dir = rig.testDir!;
|
||||||
|
});
|
||||||
|
|
||||||
|
afterAll(async () => {
|
||||||
|
await rig.cleanup();
|
||||||
|
});
|
||||||
|
|
||||||
|
async function runAndAssert(
|
||||||
|
filename: string,
|
||||||
|
content: Buffer,
|
||||||
|
expectedText: string | null,
|
||||||
|
) {
|
||||||
|
writeFileSync(join(dir, filename), content);
|
||||||
|
const prompt = `read the file ${filename} and output its exact contents`;
|
||||||
|
const output = await rig.run(prompt);
|
||||||
|
await rig.waitForToolCall('read_file');
|
||||||
|
const lower = output.toLowerCase();
|
||||||
|
if (expectedText === null) {
|
||||||
|
expect(
|
||||||
|
lower.includes('binary') ||
|
||||||
|
lower.includes('skipped binary file') ||
|
||||||
|
lower.includes('cannot display'),
|
||||||
|
).toBeTruthy();
|
||||||
|
} else {
|
||||||
|
expect(output.includes(expectedText)).toBeTruthy();
|
||||||
|
expect(lower.includes('skipped binary file')).toBeFalsy();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
it('UTF-8 BOM', async () => {
|
||||||
|
await runAndAssert('utf8.txt', utf8BOM('BOM_OK UTF-8'), 'BOM_OK UTF-8');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('UTF-16 LE BOM', async () => {
|
||||||
|
await runAndAssert(
|
||||||
|
'utf16le.txt',
|
||||||
|
utf16LE('BOM_OK UTF-16LE'),
|
||||||
|
'BOM_OK UTF-16LE',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('UTF-16 BE BOM', async () => {
|
||||||
|
await runAndAssert(
|
||||||
|
'utf16be.txt',
|
||||||
|
utf16BE('BOM_OK UTF-16BE'),
|
||||||
|
'BOM_OK UTF-16BE',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('UTF-32 LE BOM', async () => {
|
||||||
|
await runAndAssert(
|
||||||
|
'utf32le.txt',
|
||||||
|
utf32LE('BOM_OK UTF-32LE'),
|
||||||
|
'BOM_OK UTF-32LE',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('UTF-32 BE BOM', async () => {
|
||||||
|
await runAndAssert(
|
||||||
|
'utf32be.txt',
|
||||||
|
utf32BE('BOM_OK UTF-32BE'),
|
||||||
|
'BOM_OK UTF-32BE',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('Can describe a PNG file', async () => {
|
||||||
|
const imagePath = resolve(
|
||||||
|
process.cwd(),
|
||||||
|
'docs/assets/gemini-screenshot.png',
|
||||||
|
);
|
||||||
|
const imageContent = readFileSync(imagePath);
|
||||||
|
const filename = 'gemini-screenshot.png';
|
||||||
|
writeFileSync(join(dir, filename), imageContent);
|
||||||
|
const prompt = `What is shown in the image ${filename}?`;
|
||||||
|
const output = await rig.run(prompt);
|
||||||
|
await rig.waitForToolCall('read_file');
|
||||||
|
const lower = output.toLowerCase();
|
||||||
|
// The response is non-deterministic, so we just check for some
|
||||||
|
// keywords that are very likely to be in the response.
|
||||||
|
expect(lower.includes('gemini')).toBeTruthy();
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -17,6 +17,12 @@ export default defineConfig({
|
|||||||
include: ['**/*.test.ts'],
|
include: ['**/*.test.ts'],
|
||||||
exclude: ['**/terminal-bench/*.test.ts', '**/node_modules/**'],
|
exclude: ['**/terminal-bench/*.test.ts', '**/node_modules/**'],
|
||||||
retry: 2,
|
retry: 2,
|
||||||
fileParallelism: false,
|
fileParallelism: true,
|
||||||
|
poolOptions: {
|
||||||
|
threads: {
|
||||||
|
minThreads: 2,
|
||||||
|
maxThreads: 4,
|
||||||
|
},
|
||||||
|
},
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|||||||
4467
package-lock.json
generated
4467
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
52
package.json
52
package.json
@@ -16,7 +16,7 @@
|
|||||||
"sandboxImageUri": "ghcr.io/qwenlm/qwen-code:0.0.14"
|
"sandboxImageUri": "ghcr.io/qwenlm/qwen-code:0.0.14"
|
||||||
},
|
},
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"start": "node scripts/start.js",
|
"start": "cross-env node scripts/start.js",
|
||||||
"debug": "cross-env DEBUG=1 node --inspect-brk scripts/start.js",
|
"debug": "cross-env DEBUG=1 node --inspect-brk scripts/start.js",
|
||||||
"auth:npm": "npx google-artifactregistry-auth",
|
"auth:npm": "npx google-artifactregistry-auth",
|
||||||
"auth:docker": "gcloud auth configure-docker us-west1-docker.pkg.dev",
|
"auth:docker": "gcloud auth configure-docker us-west1-docker.pkg.dev",
|
||||||
@@ -27,30 +27,40 @@
|
|||||||
"build:vscode": "node scripts/build_vscode_companion.js",
|
"build:vscode": "node scripts/build_vscode_companion.js",
|
||||||
"build:all": "npm run build && npm run build:sandbox && npm run build:vscode",
|
"build:all": "npm run build && npm run build:sandbox && npm run build:vscode",
|
||||||
"build:packages": "npm run build --workspaces",
|
"build:packages": "npm run build --workspaces",
|
||||||
"build:sandbox": "node scripts/build_sandbox.js --skip-npm-install-build",
|
"build:sandbox": "node scripts/build_sandbox.js",
|
||||||
"bundle": "npm run generate && node esbuild.config.js && node scripts/copy_bundle_assets.js",
|
"bundle": "npm run generate && node esbuild.config.js && node scripts/copy_bundle_assets.js",
|
||||||
"test": "npm run test --workspaces --if-present",
|
"test": "npm run test --workspaces --if-present --parallel",
|
||||||
"test:ci": "npm run test:ci --workspaces --if-present && npm run test:scripts",
|
"test:ci": "npm run test:ci --workspaces --if-present --parallel && npm run test:scripts",
|
||||||
"test:scripts": "vitest run --config ./scripts/tests/vitest.config.ts",
|
"test:scripts": "vitest run --config ./scripts/tests/vitest.config.ts",
|
||||||
"test:e2e": "cross-env VERBOSE=true KEEP_OUTPUT=true npm run test:integration:sandbox:none",
|
"test:e2e": "cross-env VERBOSE=true KEEP_OUTPUT=true npm run test:integration:sandbox:none",
|
||||||
"test:integration:all": "npm run test:integration:sandbox:none && npm run test:integration:sandbox:docker && npm run test:integration:sandbox:podman",
|
"test:integration:all": "npm run test:integration:sandbox:none && npm run test:integration:sandbox:docker && npm run test:integration:sandbox:podman",
|
||||||
"test:integration:sandbox:none": "GEMINI_SANDBOX=false vitest run --root ./integration-tests",
|
"test:integration:sandbox:none": "cross-env GEMINI_SANDBOX=false vitest run --root ./integration-tests",
|
||||||
"test:integration:sandbox:docker": "GEMINI_SANDBOX=docker npm run build:sandbox && GEMINI_SANDBOX=docker vitest run --root ./integration-tests",
|
"test:integration:sandbox:docker": "cross-env GEMINI_SANDBOX=docker npm run build:sandbox && GEMINI_SANDBOX=docker vitest run --root ./integration-tests",
|
||||||
"test:integration:sandbox:podman": "GEMINI_SANDBOX=podman vitest run --root ./integration-tests",
|
"test:integration:sandbox:podman": "cross-env GEMINI_SANDBOX=podman vitest run --root ./integration-tests",
|
||||||
"test:terminal-bench": "cross-env VERBOSE=true KEEP_OUTPUT=true vitest run --config ./vitest.terminal-bench.config.ts --root ./integration-tests",
|
"test:terminal-bench": "cross-env VERBOSE=true KEEP_OUTPUT=true vitest run --config ./vitest.terminal-bench.config.ts --root ./integration-tests",
|
||||||
"test:terminal-bench:oracle": "cross-env VERBOSE=true KEEP_OUTPUT=true vitest run --config ./vitest.terminal-bench.config.ts --root ./integration-tests -t 'oracle'",
|
"test:terminal-bench:oracle": "cross-env VERBOSE=true KEEP_OUTPUT=true vitest run --config ./vitest.terminal-bench.config.ts --root ./integration-tests -t 'oracle'",
|
||||||
"test:terminal-bench:qwen": "cross-env VERBOSE=true KEEP_OUTPUT=true vitest run --config ./vitest.terminal-bench.config.ts --root ./integration-tests -t 'qwen'",
|
"test:terminal-bench:qwen": "cross-env VERBOSE=true KEEP_OUTPUT=true vitest run --config ./vitest.terminal-bench.config.ts --root ./integration-tests -t 'qwen'",
|
||||||
"lint": "eslint . --ext .ts,.tsx && eslint integration-tests",
|
"lint": "eslint . --ext .ts,.tsx && eslint integration-tests",
|
||||||
"lint:fix": "eslint . --fix && eslint integration-tests --fix",
|
"lint:fix": "eslint . --fix && eslint integration-tests --fix",
|
||||||
"lint:ci": "eslint . --ext .ts,.tsx --max-warnings 0 && eslint integration-tests --max-warnings 0",
|
"lint:ci": "eslint . --ext .ts,.tsx --max-warnings 0 && eslint integration-tests --max-warnings 0",
|
||||||
|
"lint:all": "node scripts/lint.js",
|
||||||
"format": "prettier --experimental-cli --write .",
|
"format": "prettier --experimental-cli --write .",
|
||||||
"typecheck": "npm run typecheck --workspaces --if-present",
|
"typecheck": "npm run typecheck --workspaces --if-present",
|
||||||
"preflight": "npm run clean && npm ci && npm run format && npm run lint:ci && npm run build && npm run typecheck && npm run test:ci",
|
"preflight": "npm run clean && npm ci && npm run format && npm run lint:ci && npm run build && npm run typecheck && npm run test:ci",
|
||||||
"prepare": "npm run bundle",
|
"prepare": "husky && npm run bundle",
|
||||||
"prepare:package": "node scripts/prepare-package.js",
|
"prepare:package": "node scripts/prepare-package.js",
|
||||||
"release:version": "node scripts/version.js",
|
"release:version": "node scripts/version.js",
|
||||||
"telemetry": "node scripts/telemetry.js",
|
"telemetry": "node scripts/telemetry.js",
|
||||||
"clean": "node scripts/clean.js"
|
"check:lockfile": "node scripts/check-lockfile.js",
|
||||||
|
"clean": "node scripts/clean.js",
|
||||||
|
"pre-commit": "node scripts/pre-commit.js"
|
||||||
|
},
|
||||||
|
"overrides": {
|
||||||
|
"wrap-ansi": "9.0.2",
|
||||||
|
"ansi-regex": "6.2.2",
|
||||||
|
"cliui": {
|
||||||
|
"wrap-ansi": "7.0.0"
|
||||||
|
}
|
||||||
},
|
},
|
||||||
"bin": {
|
"bin": {
|
||||||
"qwen": "bundle/gemini.js"
|
"qwen": "bundle/gemini.js"
|
||||||
@@ -70,7 +80,6 @@
|
|||||||
"@types/uuid": "^10.0.0",
|
"@types/uuid": "^10.0.0",
|
||||||
"@vitest/coverage-v8": "^3.1.1",
|
"@vitest/coverage-v8": "^3.1.1",
|
||||||
"@vitest/eslint-plugin": "^1.3.4",
|
"@vitest/eslint-plugin": "^1.3.4",
|
||||||
"concurrently": "^9.2.0",
|
|
||||||
"cross-env": "^7.0.3",
|
"cross-env": "^7.0.3",
|
||||||
"esbuild": "^0.25.0",
|
"esbuild": "^0.25.0",
|
||||||
"eslint": "^9.24.0",
|
"eslint": "^9.24.0",
|
||||||
@@ -81,23 +90,27 @@
|
|||||||
"eslint-plugin-react-hooks": "^5.2.0",
|
"eslint-plugin-react-hooks": "^5.2.0",
|
||||||
"glob": "^10.4.5",
|
"glob": "^10.4.5",
|
||||||
"globals": "^16.0.0",
|
"globals": "^16.0.0",
|
||||||
|
"google-artifactregistry-auth": "^3.4.0",
|
||||||
|
"husky": "^9.1.7",
|
||||||
"json": "^11.0.0",
|
"json": "^11.0.0",
|
||||||
"lodash": "^4.17.21",
|
"lint-staged": "^16.1.6",
|
||||||
"memfs": "^4.17.2",
|
"memfs": "^4.42.0",
|
||||||
"mnemonist": "^0.40.3",
|
"mnemonist": "^0.40.3",
|
||||||
"mock-fs": "^5.5.0",
|
"mock-fs": "^5.5.0",
|
||||||
"msw": "^2.10.4",
|
"msw": "^2.10.4",
|
||||||
|
"npm-run-all": "^4.1.5",
|
||||||
"prettier": "^3.5.3",
|
"prettier": "^3.5.3",
|
||||||
"react-devtools-core": "^4.28.5",
|
"react-devtools-core": "^4.28.5",
|
||||||
|
"semver": "^7.7.2",
|
||||||
|
"strip-ansi": "^7.1.2",
|
||||||
"tsx": "^4.20.3",
|
"tsx": "^4.20.3",
|
||||||
"typescript-eslint": "^8.30.1",
|
"typescript-eslint": "^8.30.1",
|
||||||
"vitest": "^3.2.4",
|
"vitest": "^3.2.4",
|
||||||
"yargs": "^17.7.2"
|
"yargs": "^17.7.2"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@lvce-editor/ripgrep": "^1.6.0",
|
"@testing-library/dom": "^10.4.1",
|
||||||
"simple-git": "^3.28.0",
|
"simple-git": "^3.28.0"
|
||||||
"strip-ansi": "^7.1.0"
|
|
||||||
},
|
},
|
||||||
"optionalDependencies": {
|
"optionalDependencies": {
|
||||||
"@lydell/node-pty": "1.1.0",
|
"@lydell/node-pty": "1.1.0",
|
||||||
@@ -107,5 +120,14 @@
|
|||||||
"@lydell/node-pty-win32-arm64": "1.1.0",
|
"@lydell/node-pty-win32-arm64": "1.1.0",
|
||||||
"@lydell/node-pty-win32-x64": "1.1.0",
|
"@lydell/node-pty-win32-x64": "1.1.0",
|
||||||
"node-pty": "^1.0.0"
|
"node-pty": "^1.0.0"
|
||||||
|
},
|
||||||
|
"lint-staged": {
|
||||||
|
"*.{js,jsx,ts,tsx}": [
|
||||||
|
"prettier --write",
|
||||||
|
"eslint --fix --max-warnings 0"
|
||||||
|
],
|
||||||
|
"*.{json,md}": [
|
||||||
|
"prettier --write"
|
||||||
|
]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,7 +18,7 @@
|
|||||||
"lint": "eslint . --ext .ts,.tsx",
|
"lint": "eslint . --ext .ts,.tsx",
|
||||||
"format": "prettier --write .",
|
"format": "prettier --write .",
|
||||||
"test": "vitest run",
|
"test": "vitest run",
|
||||||
"test:ci": "vitest run --coverage",
|
"test:ci": "vitest run",
|
||||||
"typecheck": "tsc --noEmit"
|
"typecheck": "tsc --noEmit"
|
||||||
},
|
},
|
||||||
"files": [
|
"files": [
|
||||||
@@ -28,35 +28,38 @@
|
|||||||
"sandboxImageUri": "ghcr.io/qwenlm/qwen-code:0.0.14"
|
"sandboxImageUri": "ghcr.io/qwenlm/qwen-code:0.0.14"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@google/genai": "1.9.0",
|
"@google/genai": "1.16.0",
|
||||||
"@iarna/toml": "^2.2.5",
|
"@iarna/toml": "^2.2.5",
|
||||||
"@qwen-code/qwen-code-core": "file:../core",
|
"@qwen-code/qwen-code-core": "file:../core",
|
||||||
"@modelcontextprotocol/sdk": "^1.15.1",
|
"@modelcontextprotocol/sdk": "^1.15.1",
|
||||||
"@types/update-notifier": "^6.0.8",
|
"@types/update-notifier": "^6.0.8",
|
||||||
|
"ansi-regex": "^6.2.2",
|
||||||
"command-exists": "^1.2.9",
|
"command-exists": "^1.2.9",
|
||||||
|
"comment-json": "^4.2.5",
|
||||||
"diff": "^7.0.0",
|
"diff": "^7.0.0",
|
||||||
"dotenv": "^17.1.0",
|
"dotenv": "^17.1.0",
|
||||||
"fzf": "^0.5.2",
|
"fzf": "^0.5.2",
|
||||||
"glob": "^10.4.1",
|
"glob": "^10.4.5",
|
||||||
"highlight.js": "^11.11.1",
|
"highlight.js": "^11.11.1",
|
||||||
"ink": "^6.2.3",
|
"ink": "^6.2.3",
|
||||||
"ink-gradient": "^3.0.0",
|
"ink-gradient": "^3.0.0",
|
||||||
"ink-link": "^4.1.0",
|
"ink-link": "^4.1.0",
|
||||||
"ink-spinner": "^5.0.0",
|
"ink-spinner": "^5.0.0",
|
||||||
"lodash-es": "^4.17.21",
|
|
||||||
"lowlight": "^3.3.0",
|
"lowlight": "^3.3.0",
|
||||||
"mime-types": "^3.0.1",
|
|
||||||
"open": "^10.1.2",
|
"open": "^10.1.2",
|
||||||
"qrcode-terminal": "^0.12.0",
|
"qrcode-terminal": "^0.12.0",
|
||||||
"react": "^19.1.0",
|
"react": "^19.1.0",
|
||||||
"read-package-up": "^11.0.0",
|
"read-package-up": "^11.0.0",
|
||||||
"simple-git": "^3.28.0",
|
|
||||||
"shell-quote": "^1.8.3",
|
"shell-quote": "^1.8.3",
|
||||||
|
"simple-git": "^3.28.0",
|
||||||
"string-width": "^7.1.0",
|
"string-width": "^7.1.0",
|
||||||
"strip-ansi": "^7.1.0",
|
"strip-ansi": "^7.1.0",
|
||||||
"strip-json-comments": "^3.1.1",
|
"strip-json-comments": "^3.1.1",
|
||||||
|
"tar": "^7.5.1",
|
||||||
"undici": "^7.10.0",
|
"undici": "^7.10.0",
|
||||||
|
"extract-zip": "^2.0.1",
|
||||||
"update-notifier": "^7.3.1",
|
"update-notifier": "^7.3.1",
|
||||||
|
"wrap-ansi": "9.0.2",
|
||||||
"yargs": "^17.7.2",
|
"yargs": "^17.7.2",
|
||||||
"zod": "^3.23.8"
|
"zod": "^3.23.8"
|
||||||
},
|
},
|
||||||
@@ -64,16 +67,18 @@
|
|||||||
"@babel/runtime": "^7.27.6",
|
"@babel/runtime": "^7.27.6",
|
||||||
"@google/gemini-cli-test-utils": "file:../test-utils",
|
"@google/gemini-cli-test-utils": "file:../test-utils",
|
||||||
"@testing-library/react": "^16.3.0",
|
"@testing-library/react": "^16.3.0",
|
||||||
|
"@types/archiver": "^6.0.3",
|
||||||
"@types/command-exists": "^1.2.3",
|
"@types/command-exists": "^1.2.3",
|
||||||
"@types/diff": "^7.0.2",
|
"@types/diff": "^7.0.2",
|
||||||
"@types/dotenv": "^6.1.1",
|
"@types/dotenv": "^6.1.1",
|
||||||
"@types/lodash-es": "^4.17.12",
|
|
||||||
"@types/node": "^20.11.24",
|
"@types/node": "^20.11.24",
|
||||||
"@types/react": "^19.1.8",
|
"@types/react": "^19.1.8",
|
||||||
"@types/react-dom": "^19.1.6",
|
"@types/react-dom": "^19.1.6",
|
||||||
"@types/semver": "^7.7.0",
|
"@types/semver": "^7.7.0",
|
||||||
"@types/shell-quote": "^1.7.5",
|
"@types/shell-quote": "^1.7.5",
|
||||||
|
"@types/tar": "^6.1.13",
|
||||||
"@types/yargs": "^17.0.32",
|
"@types/yargs": "^17.0.32",
|
||||||
|
"archiver": "^7.0.1",
|
||||||
"ink-testing-library": "^4.0.0",
|
"ink-testing-library": "^4.0.0",
|
||||||
"jsdom": "^26.1.0",
|
"jsdom": "^26.1.0",
|
||||||
"pretty-format": "^30.0.2",
|
"pretty-format": "^30.0.2",
|
||||||
|
|||||||
@@ -11,6 +11,8 @@ import { listCommand } from './extensions/list.js';
|
|||||||
import { updateCommand } from './extensions/update.js';
|
import { updateCommand } from './extensions/update.js';
|
||||||
import { disableCommand } from './extensions/disable.js';
|
import { disableCommand } from './extensions/disable.js';
|
||||||
import { enableCommand } from './extensions/enable.js';
|
import { enableCommand } from './extensions/enable.js';
|
||||||
|
import { linkCommand } from './extensions/link.js';
|
||||||
|
import { newCommand } from './extensions/new.js';
|
||||||
|
|
||||||
export const extensionsCommand: CommandModule = {
|
export const extensionsCommand: CommandModule = {
|
||||||
command: 'extensions <command>',
|
command: 'extensions <command>',
|
||||||
@@ -23,6 +25,8 @@ export const extensionsCommand: CommandModule = {
|
|||||||
.command(updateCommand)
|
.command(updateCommand)
|
||||||
.command(disableCommand)
|
.command(disableCommand)
|
||||||
.command(enableCommand)
|
.command(enableCommand)
|
||||||
|
.command(linkCommand)
|
||||||
|
.command(newCommand)
|
||||||
.demandCommand(1, 'You need at least one command before continuing.')
|
.demandCommand(1, 'You need at least one command before continuing.')
|
||||||
.version(false),
|
.version(false),
|
||||||
handler: () => {
|
handler: () => {
|
||||||
|
|||||||
@@ -11,12 +11,16 @@ import { getErrorMessage } from '../../utils/errors.js';
|
|||||||
|
|
||||||
interface DisableArgs {
|
interface DisableArgs {
|
||||||
name: string;
|
name: string;
|
||||||
scope: SettingScope;
|
scope?: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function handleDisable(args: DisableArgs) {
|
export function handleDisable(args: DisableArgs) {
|
||||||
try {
|
try {
|
||||||
disableExtension(args.name, args.scope);
|
if (args.scope?.toLowerCase() === 'workspace') {
|
||||||
|
disableExtension(args.name, SettingScope.Workspace);
|
||||||
|
} else {
|
||||||
|
disableExtension(args.name, SettingScope.User);
|
||||||
|
}
|
||||||
console.log(
|
console.log(
|
||||||
`Extension "${args.name}" successfully disabled for scope "${args.scope}".`,
|
`Extension "${args.name}" successfully disabled for scope "${args.scope}".`,
|
||||||
);
|
);
|
||||||
@@ -39,13 +43,28 @@ export const disableCommand: CommandModule = {
|
|||||||
describe: 'The scope to disable the extenison in.',
|
describe: 'The scope to disable the extenison in.',
|
||||||
type: 'string',
|
type: 'string',
|
||||||
default: SettingScope.User,
|
default: SettingScope.User,
|
||||||
choices: [SettingScope.User, SettingScope.Workspace],
|
|
||||||
})
|
})
|
||||||
.check((_argv) => true),
|
.check((argv) => {
|
||||||
handler: async (argv) => {
|
if (
|
||||||
await handleDisable({
|
argv.scope &&
|
||||||
|
!Object.values(SettingScope)
|
||||||
|
.map((s) => s.toLowerCase())
|
||||||
|
.includes((argv.scope as string).toLowerCase())
|
||||||
|
) {
|
||||||
|
throw new Error(
|
||||||
|
`Invalid scope: ${argv.scope}. Please use one of ${Object.values(
|
||||||
|
SettingScope,
|
||||||
|
)
|
||||||
|
.map((s) => s.toLowerCase())
|
||||||
|
.join(', ')}.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}),
|
||||||
|
handler: (argv) => {
|
||||||
|
handleDisable({
|
||||||
name: argv['name'] as string,
|
name: argv['name'] as string,
|
||||||
scope: argv['scope'] as SettingScope,
|
scope: argv['scope'] as string,
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -11,15 +11,16 @@ import { SettingScope } from '../../config/settings.js';
|
|||||||
|
|
||||||
interface EnableArgs {
|
interface EnableArgs {
|
||||||
name: string;
|
name: string;
|
||||||
scope?: SettingScope;
|
scope?: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function handleEnable(args: EnableArgs) {
|
export function handleEnable(args: EnableArgs) {
|
||||||
try {
|
try {
|
||||||
const scopes = args.scope
|
if (args.scope?.toLowerCase() === 'workspace') {
|
||||||
? [args.scope]
|
enableExtension(args.name, SettingScope.Workspace);
|
||||||
: [SettingScope.User, SettingScope.Workspace];
|
} else {
|
||||||
enableExtension(args.name, scopes);
|
enableExtension(args.name, SettingScope.User);
|
||||||
|
}
|
||||||
if (args.scope) {
|
if (args.scope) {
|
||||||
console.log(
|
console.log(
|
||||||
`Extension "${args.name}" successfully enabled for scope "${args.scope}".`,
|
`Extension "${args.name}" successfully enabled for scope "${args.scope}".`,
|
||||||
@@ -35,7 +36,7 @@ export async function handleEnable(args: EnableArgs) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
export const enableCommand: CommandModule = {
|
export const enableCommand: CommandModule = {
|
||||||
command: 'disable [--scope] <name>',
|
command: 'enable [--scope] <name>',
|
||||||
describe: 'Enables an extension.',
|
describe: 'Enables an extension.',
|
||||||
builder: (yargs) =>
|
builder: (yargs) =>
|
||||||
yargs
|
yargs
|
||||||
@@ -47,13 +48,28 @@ export const enableCommand: CommandModule = {
|
|||||||
describe:
|
describe:
|
||||||
'The scope to enable the extenison in. If not set, will be enabled in all scopes.',
|
'The scope to enable the extenison in. If not set, will be enabled in all scopes.',
|
||||||
type: 'string',
|
type: 'string',
|
||||||
choices: [SettingScope.User, SettingScope.Workspace],
|
|
||||||
})
|
})
|
||||||
.check((_argv) => true),
|
.check((argv) => {
|
||||||
handler: async (argv) => {
|
if (
|
||||||
await handleEnable({
|
argv.scope &&
|
||||||
|
!Object.values(SettingScope)
|
||||||
|
.map((s) => s.toLowerCase())
|
||||||
|
.includes((argv.scope as string).toLowerCase())
|
||||||
|
) {
|
||||||
|
throw new Error(
|
||||||
|
`Invalid scope: ${argv.scope}. Please use one of ${Object.values(
|
||||||
|
SettingScope,
|
||||||
|
)
|
||||||
|
.map((s) => s.toLowerCase())
|
||||||
|
.join(', ')}.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}),
|
||||||
|
handler: (argv) => {
|
||||||
|
handleEnable({
|
||||||
name: argv['name'] as string,
|
name: argv['name'] as string,
|
||||||
scope: argv['scope'] as SettingScope,
|
scope: argv['scope'] as string,
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -0,0 +1,8 @@
|
|||||||
|
# Ink Library Screen Reader Guidance
|
||||||
|
|
||||||
|
When building custom components, it's important to keep accessibility in mind. While Ink provides the building blocks, ensuring your components are accessible will make your CLIs usable by a wider audience.
|
||||||
|
|
||||||
|
## General Principles
|
||||||
|
|
||||||
|
Provide screen reader-friendly output: Use the useIsScreenReaderEnabled hook to detect if a screen reader is active. You can then render a more descriptive output for screen reader users.
|
||||||
|
Leverage ARIA props: For components that have a specific role (e.g., a checkbox or a button), use the aria-role, aria-state, and aria-label props on <Box> and <Text> to provide semantic meaning to screen readers.
|
||||||
@@ -0,0 +1,4 @@
|
|||||||
|
{
|
||||||
|
"name": "context-example",
|
||||||
|
"version": "1.0.0"
|
||||||
|
}
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
prompt = """
|
||||||
|
Please summarize the findings for the pattern `{{args}}`.
|
||||||
|
|
||||||
|
Search Results:
|
||||||
|
!{grep -r {{args}} .}
|
||||||
|
"""
|
||||||
@@ -0,0 +1,4 @@
|
|||||||
|
{
|
||||||
|
"name": "custom-commands",
|
||||||
|
"version": "1.0.0"
|
||||||
|
}
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
{
|
||||||
|
"name": "excludeTools",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"excludeTools": ["run_shell_command(rm -rf)"]
|
||||||
|
}
|
||||||
@@ -0,0 +1,60 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
||||||
|
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
||||||
|
import { z } from 'zod';
|
||||||
|
|
||||||
|
const server = new McpServer({
|
||||||
|
name: 'prompt-server',
|
||||||
|
version: '1.0.0',
|
||||||
|
});
|
||||||
|
|
||||||
|
server.registerTool(
|
||||||
|
'fetch_posts',
|
||||||
|
{
|
||||||
|
description: 'Fetches a list of posts from a public API.',
|
||||||
|
inputSchema: z.object({}).shape,
|
||||||
|
},
|
||||||
|
async () => {
|
||||||
|
const apiResponse = await fetch(
|
||||||
|
'https://jsonplaceholder.typicode.com/posts',
|
||||||
|
);
|
||||||
|
const posts = await apiResponse.json();
|
||||||
|
const response = { posts: posts.slice(0, 5) };
|
||||||
|
return {
|
||||||
|
content: [
|
||||||
|
{
|
||||||
|
type: 'text',
|
||||||
|
text: JSON.stringify(response),
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
server.registerPrompt(
|
||||||
|
'poem-writer',
|
||||||
|
{
|
||||||
|
title: 'Poem Writer',
|
||||||
|
description: 'Write a nice haiku',
|
||||||
|
argsSchema: { title: z.string(), mood: z.string().optional() },
|
||||||
|
},
|
||||||
|
({ title, mood }) => ({
|
||||||
|
messages: [
|
||||||
|
{
|
||||||
|
role: 'user',
|
||||||
|
content: {
|
||||||
|
type: 'text',
|
||||||
|
text: `Write a haiku${mood ? ` with the mood ${mood}` : ''} called ${title}. Note that a haiku is 5 syllables followed by 7 syllables followed by 5 syllables `,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
|
||||||
|
const transport = new StdioServerTransport();
|
||||||
|
await server.connect(transport);
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
{
|
||||||
|
"name": "mcp-server-example",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "Example MCP Server for Gemini CLI Extension",
|
||||||
|
"type": "module",
|
||||||
|
"main": "example.js",
|
||||||
|
"scripts": {
|
||||||
|
"build": "tsc"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"typescript": "~5.4.5",
|
||||||
|
"@types/node": "^20.11.25"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@modelcontextprotocol/sdk": "^1.11.0",
|
||||||
|
"zod": "^3.22.4"
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
{
|
||||||
|
"name": "mcp-server-example",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"mcpServers": {
|
||||||
|
"nodeServer": {
|
||||||
|
"command": "node",
|
||||||
|
"args": ["${extensionPath}${/}dist${/}example.js"],
|
||||||
|
"cwd": "${extensionPath}"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
{
|
||||||
|
"compilerOptions": {
|
||||||
|
"target": "ES2022",
|
||||||
|
"module": "NodeNext",
|
||||||
|
"moduleResolution": "NodeNext",
|
||||||
|
"strict": true,
|
||||||
|
"esModuleInterop": true,
|
||||||
|
"skipLibCheck": true,
|
||||||
|
"forceConsistentCasingInFileNames": true,
|
||||||
|
"outDir": "./dist"
|
||||||
|
},
|
||||||
|
"include": ["example.ts"]
|
||||||
|
}
|
||||||
@@ -4,10 +4,30 @@
|
|||||||
* SPDX-License-Identifier: Apache-2.0
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import { describe, it, expect } from 'vitest';
|
import { describe, it, expect, vi, type MockInstance } from 'vitest';
|
||||||
import { installCommand } from './install.js';
|
import { handleInstall, installCommand } from './install.js';
|
||||||
import yargs from 'yargs';
|
import yargs from 'yargs';
|
||||||
|
|
||||||
|
const mockInstallExtension = vi.hoisted(() => vi.fn());
|
||||||
|
const mockRequestConsentNonInteractive = vi.hoisted(() => vi.fn());
|
||||||
|
const mockStat = vi.hoisted(() => vi.fn());
|
||||||
|
|
||||||
|
vi.mock('../../config/extension.js', () => ({
|
||||||
|
installExtension: mockInstallExtension,
|
||||||
|
requestConsentNonInteractive: mockRequestConsentNonInteractive,
|
||||||
|
}));
|
||||||
|
|
||||||
|
vi.mock('../../utils/errors.js', () => ({
|
||||||
|
getErrorMessage: vi.fn((error: Error) => error.message),
|
||||||
|
}));
|
||||||
|
|
||||||
|
vi.mock('node:fs/promises', () => ({
|
||||||
|
stat: mockStat,
|
||||||
|
default: {
|
||||||
|
stat: mockStat,
|
||||||
|
},
|
||||||
|
}));
|
||||||
|
|
||||||
describe('extensions install command', () => {
|
describe('extensions install command', () => {
|
||||||
it('should fail if no source is provided', () => {
|
it('should fail if no source is provided', () => {
|
||||||
const validationParser = yargs([])
|
const validationParser = yargs([])
|
||||||
@@ -15,17 +35,109 @@ describe('extensions install command', () => {
|
|||||||
.command(installCommand)
|
.command(installCommand)
|
||||||
.fail(false);
|
.fail(false);
|
||||||
expect(() => validationParser.parse('install')).toThrow(
|
expect(() => validationParser.parse('install')).toThrow(
|
||||||
'Either a git URL --source or a --path must be provided.',
|
'Not enough non-option arguments: got 0, need at least 1',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('handleInstall', () => {
|
||||||
|
let consoleLogSpy: MockInstance;
|
||||||
|
let consoleErrorSpy: MockInstance;
|
||||||
|
let processSpy: MockInstance;
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
consoleLogSpy = vi.spyOn(console, 'log');
|
||||||
|
consoleErrorSpy = vi.spyOn(console, 'error');
|
||||||
|
processSpy = vi
|
||||||
|
.spyOn(process, 'exit')
|
||||||
|
.mockImplementation(() => undefined as never);
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
mockInstallExtension.mockClear();
|
||||||
|
mockRequestConsentNonInteractive.mockClear();
|
||||||
|
mockStat.mockClear();
|
||||||
|
vi.resetAllMocks();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should install an extension from a http source', async () => {
|
||||||
|
mockInstallExtension.mockResolvedValue('http-extension');
|
||||||
|
|
||||||
|
await handleInstall({
|
||||||
|
source: 'http://google.com',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension "http-extension" installed successfully and enabled.',
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should fail if both git source and local path are provided', () => {
|
it('should install an extension from a https source', async () => {
|
||||||
const validationParser = yargs([])
|
mockInstallExtension.mockResolvedValue('https-extension');
|
||||||
.locale('en')
|
|
||||||
.command(installCommand)
|
await handleInstall({
|
||||||
.fail(false);
|
source: 'https://google.com',
|
||||||
expect(() =>
|
});
|
||||||
validationParser.parse('install --source some-url --path /some/path'),
|
|
||||||
).toThrow('Arguments source and path are mutually exclusive');
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension "https-extension" installed successfully and enabled.',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should install an extension from a git source', async () => {
|
||||||
|
mockInstallExtension.mockResolvedValue('git-extension');
|
||||||
|
|
||||||
|
await handleInstall({
|
||||||
|
source: 'git@some-url',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension "git-extension" installed successfully and enabled.',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('throws an error from an unknown source', async () => {
|
||||||
|
mockStat.mockRejectedValue(new Error('ENOENT: no such file or directory'));
|
||||||
|
await handleInstall({
|
||||||
|
source: 'test://google.com',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(consoleErrorSpy).toHaveBeenCalledWith('Install source not found.');
|
||||||
|
expect(processSpy).toHaveBeenCalledWith(1);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should install an extension from a sso source', async () => {
|
||||||
|
mockInstallExtension.mockResolvedValue('sso-extension');
|
||||||
|
|
||||||
|
await handleInstall({
|
||||||
|
source: 'sso://google.com',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension "sso-extension" installed successfully and enabled.',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should install an extension from a local path', async () => {
|
||||||
|
mockInstallExtension.mockResolvedValue('local-extension');
|
||||||
|
mockStat.mockResolvedValue({});
|
||||||
|
await handleInstall({
|
||||||
|
source: '/some/path',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension "local-extension" installed successfully and enabled.',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw an error if install extension fails', async () => {
|
||||||
|
mockInstallExtension.mockRejectedValue(
|
||||||
|
new Error('Install extension failed'),
|
||||||
|
);
|
||||||
|
|
||||||
|
await handleInstall({ source: 'git@some-url' });
|
||||||
|
|
||||||
|
expect(consoleErrorSpy).toHaveBeenCalledWith('Install extension failed');
|
||||||
|
expect(processSpy).toHaveBeenCalledWith(1);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -7,26 +7,56 @@
|
|||||||
import type { CommandModule } from 'yargs';
|
import type { CommandModule } from 'yargs';
|
||||||
import {
|
import {
|
||||||
installExtension,
|
installExtension,
|
||||||
type ExtensionInstallMetadata,
|
requestConsentNonInteractive,
|
||||||
} from '../../config/extension.js';
|
} from '../../config/extension.js';
|
||||||
|
import type { ExtensionInstallMetadata } from '@qwen-code/qwen-code-core';
|
||||||
import { getErrorMessage } from '../../utils/errors.js';
|
import { getErrorMessage } from '../../utils/errors.js';
|
||||||
|
import { stat } from 'node:fs/promises';
|
||||||
|
|
||||||
interface InstallArgs {
|
interface InstallArgs {
|
||||||
source?: string;
|
source: string;
|
||||||
path?: string;
|
ref?: string;
|
||||||
|
autoUpdate?: boolean;
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function handleInstall(args: InstallArgs) {
|
export async function handleInstall(args: InstallArgs) {
|
||||||
try {
|
try {
|
||||||
const installMetadata: ExtensionInstallMetadata = {
|
let installMetadata: ExtensionInstallMetadata;
|
||||||
source: (args.source || args.path) as string,
|
const { source } = args;
|
||||||
type: args.source ? 'git' : 'local',
|
if (
|
||||||
|
source.startsWith('http://') ||
|
||||||
|
source.startsWith('https://') ||
|
||||||
|
source.startsWith('git@') ||
|
||||||
|
source.startsWith('sso://')
|
||||||
|
) {
|
||||||
|
installMetadata = {
|
||||||
|
source,
|
||||||
|
type: 'git',
|
||||||
|
ref: args.ref,
|
||||||
|
autoUpdate: args.autoUpdate,
|
||||||
};
|
};
|
||||||
const extensionName = await installExtension(installMetadata);
|
} else {
|
||||||
console.log(
|
if (args.ref || args.autoUpdate) {
|
||||||
`Extension "${extensionName}" installed successfully and enabled.`,
|
throw new Error(
|
||||||
|
'--ref and --auto-update are not applicable for local extensions.',
|
||||||
);
|
);
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
await stat(source);
|
||||||
|
installMetadata = {
|
||||||
|
source,
|
||||||
|
type: 'local',
|
||||||
|
};
|
||||||
|
} catch {
|
||||||
|
throw new Error('Install source not found.');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const name = await installExtension(
|
||||||
|
installMetadata,
|
||||||
|
requestConsentNonInteractive,
|
||||||
|
);
|
||||||
|
console.log(`Extension "${name}" installed successfully and enabled.`);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error(getErrorMessage(error));
|
console.error(getErrorMessage(error));
|
||||||
process.exit(1);
|
process.exit(1);
|
||||||
@@ -34,31 +64,34 @@ export async function handleInstall(args: InstallArgs) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
export const installCommand: CommandModule = {
|
export const installCommand: CommandModule = {
|
||||||
command: 'install [--source | --path ]',
|
command: 'install <source>',
|
||||||
describe: 'Installs an extension from a git repository or a local path.',
|
describe: 'Installs an extension from a git repository URL or a local path.',
|
||||||
builder: (yargs) =>
|
builder: (yargs) =>
|
||||||
yargs
|
yargs
|
||||||
.option('source', {
|
.positional('source', {
|
||||||
describe: 'The git URL of the extension to install.',
|
describe: 'The github URL or local path of the extension to install.',
|
||||||
|
type: 'string',
|
||||||
|
demandOption: true,
|
||||||
|
})
|
||||||
|
.option('ref', {
|
||||||
|
describe: 'The git ref to install from.',
|
||||||
type: 'string',
|
type: 'string',
|
||||||
})
|
})
|
||||||
.option('path', {
|
.option('auto-update', {
|
||||||
describe: 'Path to a local extension directory.',
|
describe: 'Enable auto-update for this extension.',
|
||||||
type: 'string',
|
type: 'boolean',
|
||||||
})
|
})
|
||||||
.conflicts('source', 'path')
|
|
||||||
.check((argv) => {
|
.check((argv) => {
|
||||||
if (!argv.source && !argv.path) {
|
if (!argv.source) {
|
||||||
throw new Error(
|
throw new Error('The source argument must be provided.');
|
||||||
'Either a git URL --source or a --path must be provided.',
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}),
|
}),
|
||||||
handler: async (argv) => {
|
handler: async (argv) => {
|
||||||
await handleInstall({
|
await handleInstall({
|
||||||
source: argv['source'] as string | undefined,
|
source: argv['source'] as string,
|
||||||
path: argv['path'] as string | undefined,
|
ref: argv['ref'] as string | undefined,
|
||||||
|
autoUpdate: argv['auto-update'] as boolean | undefined,
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|||||||
55
packages/cli/src/commands/extensions/link.ts
Normal file
55
packages/cli/src/commands/extensions/link.ts
Normal file
@@ -0,0 +1,55 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { CommandModule } from 'yargs';
|
||||||
|
import {
|
||||||
|
installExtension,
|
||||||
|
requestConsentNonInteractive,
|
||||||
|
} from '../../config/extension.js';
|
||||||
|
import type { ExtensionInstallMetadata } from '@qwen-code/qwen-code-core';
|
||||||
|
|
||||||
|
import { getErrorMessage } from '../../utils/errors.js';
|
||||||
|
|
||||||
|
interface InstallArgs {
|
||||||
|
path: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function handleLink(args: InstallArgs) {
|
||||||
|
try {
|
||||||
|
const installMetadata: ExtensionInstallMetadata = {
|
||||||
|
source: args.path,
|
||||||
|
type: 'link',
|
||||||
|
};
|
||||||
|
const extensionName = await installExtension(
|
||||||
|
installMetadata,
|
||||||
|
requestConsentNonInteractive,
|
||||||
|
);
|
||||||
|
console.log(
|
||||||
|
`Extension "${extensionName}" linked successfully and enabled.`,
|
||||||
|
);
|
||||||
|
} catch (error) {
|
||||||
|
console.error(getErrorMessage(error));
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const linkCommand: CommandModule = {
|
||||||
|
command: 'link <path>',
|
||||||
|
describe:
|
||||||
|
'Links an extension from a local path. Updates made to the local path will always be reflected.',
|
||||||
|
builder: (yargs) =>
|
||||||
|
yargs
|
||||||
|
.positional('path', {
|
||||||
|
describe: 'The name of the extension to link.',
|
||||||
|
type: 'string',
|
||||||
|
})
|
||||||
|
.check((_) => true),
|
||||||
|
handler: async (argv) => {
|
||||||
|
await handleLink({
|
||||||
|
path: argv['path'] as string,
|
||||||
|
});
|
||||||
|
},
|
||||||
|
};
|
||||||
@@ -17,7 +17,7 @@ export async function handleList() {
|
|||||||
}
|
}
|
||||||
console.log(
|
console.log(
|
||||||
extensions
|
extensions
|
||||||
.map((extension, _): string => toOutputString(extension))
|
.map((extension, _): string => toOutputString(extension, process.cwd()))
|
||||||
.join('\n\n'),
|
.join('\n\n'),
|
||||||
);
|
);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
|
|||||||
91
packages/cli/src/commands/extensions/new.test.ts
Normal file
91
packages/cli/src/commands/extensions/new.test.ts
Normal file
@@ -0,0 +1,91 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
||||||
|
import { newCommand } from './new.js';
|
||||||
|
import yargs from 'yargs';
|
||||||
|
import * as fsPromises from 'node:fs/promises';
|
||||||
|
import path from 'node:path';
|
||||||
|
|
||||||
|
vi.mock('node:fs/promises');
|
||||||
|
|
||||||
|
const mockedFs = vi.mocked(fsPromises);
|
||||||
|
|
||||||
|
describe('extensions new command', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
vi.resetAllMocks();
|
||||||
|
|
||||||
|
const fakeFiles = [
|
||||||
|
{ name: 'context', isDirectory: () => true },
|
||||||
|
{ name: 'custom-commands', isDirectory: () => true },
|
||||||
|
{ name: 'mcp-server', isDirectory: () => true },
|
||||||
|
];
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
||||||
|
mockedFs.readdir.mockResolvedValue(fakeFiles as any);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should fail if no path is provided', async () => {
|
||||||
|
const parser = yargs([]).command(newCommand).fail(false).locale('en');
|
||||||
|
await expect(parser.parseAsync('new')).rejects.toThrow(
|
||||||
|
'Not enough non-option arguments: got 0, need at least 1',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create directory when no template is provided', async () => {
|
||||||
|
mockedFs.access.mockRejectedValue(new Error('ENOENT'));
|
||||||
|
mockedFs.mkdir.mockResolvedValue(undefined);
|
||||||
|
|
||||||
|
const parser = yargs([]).command(newCommand).fail(false);
|
||||||
|
|
||||||
|
await parser.parseAsync('new /some/path');
|
||||||
|
|
||||||
|
expect(mockedFs.mkdir).toHaveBeenCalledWith('/some/path', {
|
||||||
|
recursive: true,
|
||||||
|
});
|
||||||
|
expect(mockedFs.cp).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create directory and copy files when path does not exist', async () => {
|
||||||
|
mockedFs.access.mockRejectedValue(new Error('ENOENT'));
|
||||||
|
mockedFs.mkdir.mockResolvedValue(undefined);
|
||||||
|
mockedFs.cp.mockResolvedValue(undefined);
|
||||||
|
|
||||||
|
const parser = yargs([]).command(newCommand).fail(false);
|
||||||
|
|
||||||
|
await parser.parseAsync('new /some/path context');
|
||||||
|
|
||||||
|
expect(mockedFs.mkdir).toHaveBeenCalledWith('/some/path', {
|
||||||
|
recursive: true,
|
||||||
|
});
|
||||||
|
expect(mockedFs.cp).toHaveBeenCalledWith(
|
||||||
|
expect.stringContaining(path.normalize('context/context')),
|
||||||
|
path.normalize('/some/path/context'),
|
||||||
|
{ recursive: true },
|
||||||
|
);
|
||||||
|
expect(mockedFs.cp).toHaveBeenCalledWith(
|
||||||
|
expect.stringContaining(path.normalize('context/custom-commands')),
|
||||||
|
path.normalize('/some/path/custom-commands'),
|
||||||
|
{ recursive: true },
|
||||||
|
);
|
||||||
|
expect(mockedFs.cp).toHaveBeenCalledWith(
|
||||||
|
expect.stringContaining(path.normalize('context/mcp-server')),
|
||||||
|
path.normalize('/some/path/mcp-server'),
|
||||||
|
{ recursive: true },
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw an error if the path already exists', async () => {
|
||||||
|
mockedFs.access.mockResolvedValue(undefined);
|
||||||
|
const parser = yargs([]).command(newCommand).fail(false);
|
||||||
|
|
||||||
|
await expect(parser.parseAsync('new /some/path context')).rejects.toThrow(
|
||||||
|
'Path already exists: /some/path',
|
||||||
|
);
|
||||||
|
|
||||||
|
expect(mockedFs.mkdir).not.toHaveBeenCalled();
|
||||||
|
expect(mockedFs.cp).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
});
|
||||||
109
packages/cli/src/commands/extensions/new.ts
Normal file
109
packages/cli/src/commands/extensions/new.ts
Normal file
@@ -0,0 +1,109 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { access, cp, mkdir, readdir, writeFile } from 'node:fs/promises';
|
||||||
|
import { join, dirname, basename } from 'node:path';
|
||||||
|
import type { CommandModule } from 'yargs';
|
||||||
|
import { fileURLToPath } from 'node:url';
|
||||||
|
import { getErrorMessage } from '../../utils/errors.js';
|
||||||
|
|
||||||
|
interface NewArgs {
|
||||||
|
path: string;
|
||||||
|
template?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
const __filename = fileURLToPath(import.meta.url);
|
||||||
|
const __dirname = dirname(__filename);
|
||||||
|
|
||||||
|
const EXAMPLES_PATH = join(__dirname, 'examples');
|
||||||
|
|
||||||
|
async function pathExists(path: string) {
|
||||||
|
try {
|
||||||
|
await access(path);
|
||||||
|
return true;
|
||||||
|
} catch (_e) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createDirectory(path: string) {
|
||||||
|
if (await pathExists(path)) {
|
||||||
|
throw new Error(`Path already exists: ${path}`);
|
||||||
|
}
|
||||||
|
await mkdir(path, { recursive: true });
|
||||||
|
}
|
||||||
|
|
||||||
|
async function copyDirectory(template: string, path: string) {
|
||||||
|
await createDirectory(path);
|
||||||
|
|
||||||
|
const examplePath = join(EXAMPLES_PATH, template);
|
||||||
|
const entries = await readdir(examplePath, { withFileTypes: true });
|
||||||
|
for (const entry of entries) {
|
||||||
|
const srcPath = join(examplePath, entry.name);
|
||||||
|
const destPath = join(path, entry.name);
|
||||||
|
await cp(srcPath, destPath, { recursive: true });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleNew(args: NewArgs) {
|
||||||
|
try {
|
||||||
|
if (args.template) {
|
||||||
|
await copyDirectory(args.template, args.path);
|
||||||
|
console.log(
|
||||||
|
`Successfully created new extension from template "${args.template}" at ${args.path}.`,
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
await createDirectory(args.path);
|
||||||
|
const extensionName = basename(args.path);
|
||||||
|
const manifest = {
|
||||||
|
name: extensionName,
|
||||||
|
version: '1.0.0',
|
||||||
|
};
|
||||||
|
await writeFile(
|
||||||
|
join(args.path, 'qwen-extension.json'),
|
||||||
|
JSON.stringify(manifest, null, 2),
|
||||||
|
);
|
||||||
|
console.log(`Successfully created new extension at ${args.path}.`);
|
||||||
|
}
|
||||||
|
console.log(
|
||||||
|
`You can install this using "qwen extensions link ${args.path}" to test it out.`,
|
||||||
|
);
|
||||||
|
} catch (error) {
|
||||||
|
console.error(getErrorMessage(error));
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getBoilerplateChoices() {
|
||||||
|
const entries = await readdir(EXAMPLES_PATH, { withFileTypes: true });
|
||||||
|
return entries
|
||||||
|
.filter((entry) => entry.isDirectory())
|
||||||
|
.map((entry) => entry.name);
|
||||||
|
}
|
||||||
|
|
||||||
|
export const newCommand: CommandModule = {
|
||||||
|
command: 'new <path> [template]',
|
||||||
|
describe: 'Create a new extension from a boilerplate example.',
|
||||||
|
builder: async (yargs) => {
|
||||||
|
const choices = await getBoilerplateChoices();
|
||||||
|
return yargs
|
||||||
|
.positional('path', {
|
||||||
|
describe: 'The path to create the extension in.',
|
||||||
|
type: 'string',
|
||||||
|
})
|
||||||
|
.positional('template', {
|
||||||
|
describe: 'The boilerplate template to use.',
|
||||||
|
type: 'string',
|
||||||
|
choices,
|
||||||
|
});
|
||||||
|
},
|
||||||
|
handler: async (args) => {
|
||||||
|
await handleNew({
|
||||||
|
path: args['path'] as string,
|
||||||
|
template: args['template'] as string | undefined,
|
||||||
|
});
|
||||||
|
},
|
||||||
|
};
|
||||||
@@ -11,9 +11,9 @@ import yargs from 'yargs';
|
|||||||
describe('extensions uninstall command', () => {
|
describe('extensions uninstall command', () => {
|
||||||
it('should fail if no source is provided', () => {
|
it('should fail if no source is provided', () => {
|
||||||
const validationParser = yargs([])
|
const validationParser = yargs([])
|
||||||
.locale('en')
|
|
||||||
.command(uninstallCommand)
|
.command(uninstallCommand)
|
||||||
.fail(false);
|
.fail(false)
|
||||||
|
.locale('en');
|
||||||
expect(() => validationParser.parse('uninstall')).toThrow(
|
expect(() => validationParser.parse('uninstall')).toThrow(
|
||||||
'Not enough non-option arguments: got 0, need at least 1',
|
'Not enough non-option arguments: got 0, need at least 1',
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ import { uninstallExtension } from '../../config/extension.js';
|
|||||||
import { getErrorMessage } from '../../utils/errors.js';
|
import { getErrorMessage } from '../../utils/errors.js';
|
||||||
|
|
||||||
interface UninstallArgs {
|
interface UninstallArgs {
|
||||||
name: string;
|
name: string; // can be extension name or source URL.
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function handleUninstall(args: UninstallArgs) {
|
export async function handleUninstall(args: UninstallArgs) {
|
||||||
@@ -28,7 +28,7 @@ export const uninstallCommand: CommandModule = {
|
|||||||
builder: (yargs) =>
|
builder: (yargs) =>
|
||||||
yargs
|
yargs
|
||||||
.positional('name', {
|
.positional('name', {
|
||||||
describe: 'The name of the extension to uninstall.',
|
describe: 'The name or source path of the extension to uninstall.',
|
||||||
type: 'string',
|
type: 'string',
|
||||||
})
|
})
|
||||||
.check((argv) => {
|
.check((argv) => {
|
||||||
|
|||||||
@@ -5,43 +5,147 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
import type { CommandModule } from 'yargs';
|
import type { CommandModule } from 'yargs';
|
||||||
import { updateExtension } from '../../config/extension.js';
|
import {
|
||||||
|
loadExtensions,
|
||||||
|
annotateActiveExtensions,
|
||||||
|
ExtensionStorage,
|
||||||
|
requestConsentNonInteractive,
|
||||||
|
} from '../../config/extension.js';
|
||||||
|
import {
|
||||||
|
updateAllUpdatableExtensions,
|
||||||
|
type ExtensionUpdateInfo,
|
||||||
|
checkForAllExtensionUpdates,
|
||||||
|
updateExtension,
|
||||||
|
} from '../../config/extensions/update.js';
|
||||||
|
import { checkForExtensionUpdate } from '../../config/extensions/github.js';
|
||||||
import { getErrorMessage } from '../../utils/errors.js';
|
import { getErrorMessage } from '../../utils/errors.js';
|
||||||
|
import { ExtensionUpdateState } from '../../ui/state/extensions.js';
|
||||||
|
import { ExtensionEnablementManager } from '../../config/extensions/extensionEnablement.js';
|
||||||
|
|
||||||
interface UpdateArgs {
|
interface UpdateArgs {
|
||||||
name: string;
|
name?: string;
|
||||||
|
all?: boolean;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const updateOutput = (info: ExtensionUpdateInfo) =>
|
||||||
|
`Extension "${info.name}" successfully updated: ${info.originalVersion} → ${info.updatedVersion}.`;
|
||||||
|
|
||||||
export async function handleUpdate(args: UpdateArgs) {
|
export async function handleUpdate(args: UpdateArgs) {
|
||||||
|
const workingDir = process.cwd();
|
||||||
|
const extensionEnablementManager = new ExtensionEnablementManager(
|
||||||
|
ExtensionStorage.getUserExtensionsDir(),
|
||||||
|
// Force enable named extensions, otherwise we will only update the enabled
|
||||||
|
// ones.
|
||||||
|
args.name ? [args.name] : [],
|
||||||
|
);
|
||||||
|
const allExtensions = loadExtensions(extensionEnablementManager);
|
||||||
|
const extensions = annotateActiveExtensions(
|
||||||
|
allExtensions,
|
||||||
|
workingDir,
|
||||||
|
extensionEnablementManager,
|
||||||
|
);
|
||||||
|
if (args.name) {
|
||||||
try {
|
try {
|
||||||
// TODO(chrstnb): we should list extensions if the requested extension is not installed.
|
const extension = extensions.find(
|
||||||
const updatedExtensionInfo = await updateExtension(args.name);
|
(extension) => extension.name === args.name,
|
||||||
if (!updatedExtensionInfo) {
|
);
|
||||||
console.log(`Extension "${args.name}" failed to update.`);
|
if (!extension) {
|
||||||
|
console.log(`Extension "${args.name}" not found.`);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
let updateState: ExtensionUpdateState | undefined;
|
||||||
|
if (!extension.installMetadata) {
|
||||||
|
console.log(
|
||||||
|
`Unable to install extension "${args.name}" due to missing install metadata`,
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
await checkForExtensionUpdate(extension, (newState) => {
|
||||||
|
updateState = newState;
|
||||||
|
});
|
||||||
|
if (updateState !== ExtensionUpdateState.UPDATE_AVAILABLE) {
|
||||||
|
console.log(`Extension "${args.name}" is already up to date.`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// TODO(chrstnb): we should list extensions if the requested extension is not installed.
|
||||||
|
const updatedExtensionInfo = (await updateExtension(
|
||||||
|
extension,
|
||||||
|
workingDir,
|
||||||
|
requestConsentNonInteractive,
|
||||||
|
updateState,
|
||||||
|
() => {},
|
||||||
|
))!;
|
||||||
|
if (
|
||||||
|
updatedExtensionInfo.originalVersion !==
|
||||||
|
updatedExtensionInfo.updatedVersion
|
||||||
|
) {
|
||||||
console.log(
|
console.log(
|
||||||
`Extension "${args.name}" successfully updated: ${updatedExtensionInfo.originalVersion} → ${updatedExtensionInfo.updatedVersion}.`,
|
`Extension "${args.name}" successfully updated: ${updatedExtensionInfo.originalVersion} → ${updatedExtensionInfo.updatedVersion}.`,
|
||||||
);
|
);
|
||||||
|
} else {
|
||||||
|
console.log(`Extension "${args.name}" is already up to date.`);
|
||||||
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error(getErrorMessage(error));
|
console.error(getErrorMessage(error));
|
||||||
process.exit(1);
|
}
|
||||||
|
}
|
||||||
|
if (args.all) {
|
||||||
|
try {
|
||||||
|
const extensionState = new Map();
|
||||||
|
await checkForAllExtensionUpdates(extensions, (action) => {
|
||||||
|
if (action.type === 'SET_STATE') {
|
||||||
|
extensionState.set(action.payload.name, {
|
||||||
|
status: action.payload.state,
|
||||||
|
processed: true, // No need to process as we will force the update.
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
let updateInfos = await updateAllUpdatableExtensions(
|
||||||
|
workingDir,
|
||||||
|
requestConsentNonInteractive,
|
||||||
|
extensions,
|
||||||
|
extensionState,
|
||||||
|
() => {},
|
||||||
|
);
|
||||||
|
updateInfos = updateInfos.filter(
|
||||||
|
(info) => info.originalVersion !== info.updatedVersion,
|
||||||
|
);
|
||||||
|
if (updateInfos.length === 0) {
|
||||||
|
console.log('No extensions to update.');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
console.log(updateInfos.map((info) => updateOutput(info)).join('\n'));
|
||||||
|
} catch (error) {
|
||||||
|
console.error(getErrorMessage(error));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export const updateCommand: CommandModule = {
|
export const updateCommand: CommandModule = {
|
||||||
command: 'update <name>',
|
command: 'update [<name>] [--all]',
|
||||||
describe: 'Updates an extension.',
|
describe:
|
||||||
|
'Updates all extensions or a named extension to the latest version.',
|
||||||
builder: (yargs) =>
|
builder: (yargs) =>
|
||||||
yargs
|
yargs
|
||||||
.positional('name', {
|
.positional('name', {
|
||||||
describe: 'The name of the extension to update.',
|
describe: 'The name of the extension to update.',
|
||||||
type: 'string',
|
type: 'string',
|
||||||
})
|
})
|
||||||
.check((_argv) => true),
|
.option('all', {
|
||||||
|
describe: 'Update all extensions.',
|
||||||
|
type: 'boolean',
|
||||||
|
})
|
||||||
|
.conflicts('name', 'all')
|
||||||
|
.check((argv) => {
|
||||||
|
if (!argv.all && !argv.name) {
|
||||||
|
throw new Error('Either an extension name or --all must be provided');
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}),
|
||||||
handler: async (argv) => {
|
handler: async (argv) => {
|
||||||
await handleUpdate({
|
await handleUpdate({
|
||||||
name: argv['name'] as string,
|
name: argv['name'] as string | undefined,
|
||||||
|
all: argv['all'] as boolean | undefined,
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -13,6 +13,16 @@ vi.mock('fs/promises', () => ({
|
|||||||
writeFile: vi.fn(),
|
writeFile: vi.fn(),
|
||||||
}));
|
}));
|
||||||
|
|
||||||
|
vi.mock('os', () => {
|
||||||
|
const homedir = vi.fn(() => '/home/user');
|
||||||
|
return {
|
||||||
|
default: {
|
||||||
|
homedir,
|
||||||
|
},
|
||||||
|
homedir,
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
vi.mock('../../config/settings.js', async () => {
|
vi.mock('../../config/settings.js', async () => {
|
||||||
const actual = await vi.importActual('../../config/settings.js');
|
const actual = await vi.importActual('../../config/settings.js');
|
||||||
return {
|
return {
|
||||||
@@ -26,15 +36,20 @@ const mockedLoadSettings = loadSettings as vi.Mock;
|
|||||||
describe('mcp add command', () => {
|
describe('mcp add command', () => {
|
||||||
let parser: yargs.Argv;
|
let parser: yargs.Argv;
|
||||||
let mockSetValue: vi.Mock;
|
let mockSetValue: vi.Mock;
|
||||||
|
let mockConsoleError: vi.Mock;
|
||||||
|
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.resetAllMocks();
|
vi.resetAllMocks();
|
||||||
const yargsInstance = yargs([]).command(addCommand);
|
const yargsInstance = yargs([]).command(addCommand);
|
||||||
parser = yargsInstance;
|
parser = yargsInstance;
|
||||||
mockSetValue = vi.fn();
|
mockSetValue = vi.fn();
|
||||||
|
mockConsoleError = vi.fn();
|
||||||
|
vi.spyOn(console, 'error').mockImplementation(mockConsoleError);
|
||||||
mockedLoadSettings.mockReturnValue({
|
mockedLoadSettings.mockReturnValue({
|
||||||
forScope: () => ({ settings: {} }),
|
forScope: () => ({ settings: {} }),
|
||||||
setValue: mockSetValue,
|
setValue: mockSetValue,
|
||||||
|
workspace: { path: '/path/to/project' },
|
||||||
|
user: { path: '/home/user' },
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -119,4 +134,218 @@ describe('mcp add command', () => {
|
|||||||
},
|
},
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
describe('when handling scope and directory', () => {
|
||||||
|
const serverName = 'test-server';
|
||||||
|
const command = 'echo';
|
||||||
|
|
||||||
|
const setupMocks = (cwd: string, workspacePath: string) => {
|
||||||
|
vi.spyOn(process, 'cwd').mockReturnValue(cwd);
|
||||||
|
mockedLoadSettings.mockReturnValue({
|
||||||
|
forScope: () => ({ settings: {} }),
|
||||||
|
setValue: mockSetValue,
|
||||||
|
workspace: { path: workspacePath },
|
||||||
|
user: { path: '/home/user' },
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
describe('when in a project directory', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
setupMocks('/path/to/project', '/path/to/project');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use project scope by default', async () => {
|
||||||
|
await parser.parseAsync(`add ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.Workspace,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use project scope when --scope=project is used', async () => {
|
||||||
|
await parser.parseAsync(`add --scope project ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.Workspace,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use user scope when --scope=user is used', async () => {
|
||||||
|
await parser.parseAsync(`add --scope user ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.User,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('when in a subdirectory of a project', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
setupMocks('/path/to/project/subdir', '/path/to/project');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use project scope by default', async () => {
|
||||||
|
await parser.parseAsync(`add ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.Workspace,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('when in the home directory', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
setupMocks('/home/user', '/home/user');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should show an error by default', async () => {
|
||||||
|
const mockProcessExit = vi
|
||||||
|
.spyOn(process, 'exit')
|
||||||
|
.mockImplementation((() => {
|
||||||
|
throw new Error('process.exit called');
|
||||||
|
}) as (code?: number) => never);
|
||||||
|
|
||||||
|
await expect(
|
||||||
|
parser.parseAsync(`add ${serverName} ${command}`),
|
||||||
|
).rejects.toThrow('process.exit called');
|
||||||
|
|
||||||
|
expect(mockConsoleError).toHaveBeenCalledWith(
|
||||||
|
'Error: Please use --scope user to edit settings in the home directory.',
|
||||||
|
);
|
||||||
|
expect(mockProcessExit).toHaveBeenCalledWith(1);
|
||||||
|
expect(mockSetValue).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should show an error when --scope=project is used explicitly', async () => {
|
||||||
|
const mockProcessExit = vi
|
||||||
|
.spyOn(process, 'exit')
|
||||||
|
.mockImplementation((() => {
|
||||||
|
throw new Error('process.exit called');
|
||||||
|
}) as (code?: number) => never);
|
||||||
|
|
||||||
|
await expect(
|
||||||
|
parser.parseAsync(`add --scope project ${serverName} ${command}`),
|
||||||
|
).rejects.toThrow('process.exit called');
|
||||||
|
|
||||||
|
expect(mockConsoleError).toHaveBeenCalledWith(
|
||||||
|
'Error: Please use --scope user to edit settings in the home directory.',
|
||||||
|
);
|
||||||
|
expect(mockProcessExit).toHaveBeenCalledWith(1);
|
||||||
|
expect(mockSetValue).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use user scope when --scope=user is used', async () => {
|
||||||
|
await parser.parseAsync(`add --scope user ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.User,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
expect(mockConsoleError).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('when in a subdirectory of home (not a project)', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
setupMocks('/home/user/some/dir', '/home/user/some/dir');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use project scope by default', async () => {
|
||||||
|
await parser.parseAsync(`add ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.Workspace,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should write to the WORKSPACE scope, not the USER scope', async () => {
|
||||||
|
await parser.parseAsync(`add my-new-server echo`);
|
||||||
|
|
||||||
|
// We expect setValue to be called once.
|
||||||
|
expect(mockSetValue).toHaveBeenCalledTimes(1);
|
||||||
|
|
||||||
|
// We get the scope that setValue was called with.
|
||||||
|
const calledScope = mockSetValue.mock.calls[0][0];
|
||||||
|
|
||||||
|
// We assert that the scope was Workspace, not User.
|
||||||
|
expect(calledScope).toBe(SettingScope.Workspace);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('when outside of home (not a project)', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
setupMocks('/tmp/foo', '/tmp/foo');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should use project scope by default', async () => {
|
||||||
|
await parser.parseAsync(`add ${serverName} ${command}`);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.Workspace,
|
||||||
|
'mcpServers',
|
||||||
|
expect.any(Object),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('when updating an existing server', () => {
|
||||||
|
const serverName = 'existing-server';
|
||||||
|
const initialCommand = 'echo old';
|
||||||
|
const updatedCommand = 'echo';
|
||||||
|
const updatedArgs = ['new'];
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
mockedLoadSettings.mockReturnValue({
|
||||||
|
forScope: () => ({
|
||||||
|
settings: {
|
||||||
|
mcpServers: {
|
||||||
|
[serverName]: {
|
||||||
|
command: initialCommand,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
setValue: mockSetValue,
|
||||||
|
workspace: { path: '/path/to/project' },
|
||||||
|
user: { path: '/home/user' },
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should update the existing server in the project scope', async () => {
|
||||||
|
await parser.parseAsync(
|
||||||
|
`add ${serverName} ${updatedCommand} ${updatedArgs.join(' ')}`,
|
||||||
|
);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.Workspace,
|
||||||
|
'mcpServers',
|
||||||
|
expect.objectContaining({
|
||||||
|
[serverName]: expect.objectContaining({
|
||||||
|
command: updatedCommand,
|
||||||
|
args: updatedArgs,
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should update the existing server in the user scope', async () => {
|
||||||
|
await parser.parseAsync(
|
||||||
|
`add --scope user ${serverName} ${updatedCommand} ${updatedArgs.join(' ')}`,
|
||||||
|
);
|
||||||
|
expect(mockSetValue).toHaveBeenCalledWith(
|
||||||
|
SettingScope.User,
|
||||||
|
'mcpServers',
|
||||||
|
expect.objectContaining({
|
||||||
|
[serverName]: expect.objectContaining({
|
||||||
|
command: updatedCommand,
|
||||||
|
args: updatedArgs,
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -36,9 +36,19 @@ async function addMcpServer(
|
|||||||
includeTools,
|
includeTools,
|
||||||
excludeTools,
|
excludeTools,
|
||||||
} = options;
|
} = options;
|
||||||
|
|
||||||
|
const settings = loadSettings(process.cwd());
|
||||||
|
const inHome = settings.workspace.path === settings.user.path;
|
||||||
|
|
||||||
|
if (scope === 'project' && inHome) {
|
||||||
|
console.error(
|
||||||
|
'Error: Please use --scope user to edit settings in the home directory.',
|
||||||
|
);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
const settingsScope =
|
const settingsScope =
|
||||||
scope === 'user' ? SettingScope.User : SettingScope.Workspace;
|
scope === 'user' ? SettingScope.User : SettingScope.Workspace;
|
||||||
const settings = loadSettings(process.cwd());
|
|
||||||
|
|
||||||
let newServer: Partial<MCPServerConfig> = {};
|
let newServer: Partial<MCPServerConfig> = {};
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,7 @@
|
|||||||
import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
|
import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
|
||||||
import { listMcpServers } from './list.js';
|
import { listMcpServers } from './list.js';
|
||||||
import { loadSettings } from '../../config/settings.js';
|
import { loadSettings } from '../../config/settings.js';
|
||||||
import { loadExtensions } from '../../config/extension.js';
|
import { ExtensionStorage, loadExtensions } from '../../config/extension.js';
|
||||||
import { createTransport } from '@qwen-code/qwen-code-core';
|
import { createTransport } from '@qwen-code/qwen-code-core';
|
||||||
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
||||||
|
|
||||||
@@ -16,6 +16,9 @@ vi.mock('../../config/settings.js', () => ({
|
|||||||
}));
|
}));
|
||||||
vi.mock('../../config/extension.js', () => ({
|
vi.mock('../../config/extension.js', () => ({
|
||||||
loadExtensions: vi.fn(),
|
loadExtensions: vi.fn(),
|
||||||
|
ExtensionStorage: {
|
||||||
|
getUserExtensionsDir: vi.fn(),
|
||||||
|
},
|
||||||
}));
|
}));
|
||||||
vi.mock('@qwen-code/qwen-code-core', () => ({
|
vi.mock('@qwen-code/qwen-code-core', () => ({
|
||||||
createTransport: vi.fn(),
|
createTransport: vi.fn(),
|
||||||
@@ -29,11 +32,12 @@ vi.mock('@qwen-code/qwen-code-core', () => ({
|
|||||||
getWorkspaceSettingsPath: () => '/tmp/qwen/workspace-settings.json',
|
getWorkspaceSettingsPath: () => '/tmp/qwen/workspace-settings.json',
|
||||||
getProjectTempDir: () => '/test/home/.qwen/tmp/mocked_hash',
|
getProjectTempDir: () => '/test/home/.qwen/tmp/mocked_hash',
|
||||||
})),
|
})),
|
||||||
GEMINI_CONFIG_DIR: '.qwen',
|
QWEN_CONFIG_DIR: '.qwen',
|
||||||
getErrorMessage: (e: unknown) => (e instanceof Error ? e.message : String(e)),
|
getErrorMessage: (e: unknown) => (e instanceof Error ? e.message : String(e)),
|
||||||
}));
|
}));
|
||||||
vi.mock('@modelcontextprotocol/sdk/client/index.js');
|
vi.mock('@modelcontextprotocol/sdk/client/index.js');
|
||||||
|
|
||||||
|
const mockedExtensionStorage = ExtensionStorage as vi.Mock;
|
||||||
const mockedLoadSettings = loadSettings as vi.Mock;
|
const mockedLoadSettings = loadSettings as vi.Mock;
|
||||||
const mockedLoadExtensions = loadExtensions as vi.Mock;
|
const mockedLoadExtensions = loadExtensions as vi.Mock;
|
||||||
const mockedCreateTransport = createTransport as vi.Mock;
|
const mockedCreateTransport = createTransport as vi.Mock;
|
||||||
@@ -69,6 +73,9 @@ describe('mcp list command', () => {
|
|||||||
MockedClient.mockImplementation(() => mockClient);
|
MockedClient.mockImplementation(() => mockClient);
|
||||||
mockedCreateTransport.mockResolvedValue(mockTransport);
|
mockedCreateTransport.mockResolvedValue(mockTransport);
|
||||||
mockedLoadExtensions.mockReturnValue([]);
|
mockedLoadExtensions.mockReturnValue([]);
|
||||||
|
mockedExtensionStorage.getUserExtensionsDir.mockReturnValue(
|
||||||
|
'/mocked/extensions/dir',
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
afterEach(() => {
|
afterEach(() => {
|
||||||
|
|||||||
@@ -10,7 +10,8 @@ import { loadSettings } from '../../config/settings.js';
|
|||||||
import type { MCPServerConfig } from '@qwen-code/qwen-code-core';
|
import type { MCPServerConfig } from '@qwen-code/qwen-code-core';
|
||||||
import { MCPServerStatus, createTransport } from '@qwen-code/qwen-code-core';
|
import { MCPServerStatus, createTransport } from '@qwen-code/qwen-code-core';
|
||||||
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
||||||
import { loadExtensions } from '../../config/extension.js';
|
import { ExtensionStorage, loadExtensions } from '../../config/extension.js';
|
||||||
|
import { ExtensionEnablementManager } from '../../config/extensions/extensionEnablement.js';
|
||||||
|
|
||||||
const COLOR_GREEN = '\u001b[32m';
|
const COLOR_GREEN = '\u001b[32m';
|
||||||
const COLOR_YELLOW = '\u001b[33m';
|
const COLOR_YELLOW = '\u001b[33m';
|
||||||
@@ -20,8 +21,10 @@ const RESET_COLOR = '\u001b[0m';
|
|||||||
async function getMcpServersFromConfig(): Promise<
|
async function getMcpServersFromConfig(): Promise<
|
||||||
Record<string, MCPServerConfig>
|
Record<string, MCPServerConfig>
|
||||||
> {
|
> {
|
||||||
const settings = loadSettings(process.cwd());
|
const settings = loadSettings();
|
||||||
const extensions = loadExtensions(process.cwd());
|
const extensions = loadExtensions(
|
||||||
|
new ExtensionEnablementManager(ExtensionStorage.getUserExtensionsDir()),
|
||||||
|
);
|
||||||
const mcpServers = { ...(settings.merged.mcpServers || {}) };
|
const mcpServers = { ...(settings.merged.mcpServers || {}) };
|
||||||
for (const extension of extensions) {
|
for (const extension of extensions) {
|
||||||
Object.entries(extension.config.mcpServers || {}).forEach(
|
Object.entries(extension.config.mcpServers || {}).forEach(
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ async function removeMcpServer(
|
|||||||
const { scope } = options;
|
const { scope } = options;
|
||||||
const settingsScope =
|
const settingsScope =
|
||||||
scope === 'user' ? SettingScope.User : SettingScope.Workspace;
|
scope === 'user' ? SettingScope.User : SettingScope.Workspace;
|
||||||
const settings = loadSettings(process.cwd());
|
const settings = loadSettings();
|
||||||
|
|
||||||
const existingSettings = settings.forScope(settingsScope).settings;
|
const existingSettings = settings.forScope(settingsScope).settings;
|
||||||
const mcpServers = existingSettings.mcpServers || {};
|
const mcpServers = existingSettings.mcpServers || {};
|
||||||
|
|||||||
@@ -10,18 +10,22 @@ import { validateAuthMethod } from './auth.js';
|
|||||||
|
|
||||||
vi.mock('./settings.js', () => ({
|
vi.mock('./settings.js', () => ({
|
||||||
loadEnvironment: vi.fn(),
|
loadEnvironment: vi.fn(),
|
||||||
|
loadSettings: vi.fn().mockReturnValue({
|
||||||
|
merged: vi.fn().mockReturnValue({}),
|
||||||
|
}),
|
||||||
}));
|
}));
|
||||||
|
|
||||||
describe('validateAuthMethod', () => {
|
describe('validateAuthMethod', () => {
|
||||||
const originalEnv = process.env;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.resetModules();
|
vi.resetModules();
|
||||||
process.env = {};
|
vi.stubEnv('GEMINI_API_KEY', undefined);
|
||||||
|
vi.stubEnv('GOOGLE_CLOUD_PROJECT', undefined);
|
||||||
|
vi.stubEnv('GOOGLE_CLOUD_LOCATION', undefined);
|
||||||
|
vi.stubEnv('GOOGLE_API_KEY', undefined);
|
||||||
});
|
});
|
||||||
|
|
||||||
afterEach(() => {
|
afterEach(() => {
|
||||||
process.env = originalEnv;
|
vi.unstubAllEnvs();
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return null for LOGIN_WITH_GOOGLE', () => {
|
it('should return null for LOGIN_WITH_GOOGLE', () => {
|
||||||
@@ -34,11 +38,12 @@ describe('validateAuthMethod', () => {
|
|||||||
|
|
||||||
describe('USE_GEMINI', () => {
|
describe('USE_GEMINI', () => {
|
||||||
it('should return null if GEMINI_API_KEY is set', () => {
|
it('should return null if GEMINI_API_KEY is set', () => {
|
||||||
process.env['GEMINI_API_KEY'] = 'test-key';
|
vi.stubEnv('GEMINI_API_KEY', 'test-key');
|
||||||
expect(validateAuthMethod(AuthType.USE_GEMINI)).toBeNull();
|
expect(validateAuthMethod(AuthType.USE_GEMINI)).toBeNull();
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return an error message if GEMINI_API_KEY is not set', () => {
|
it('should return an error message if GEMINI_API_KEY is not set', () => {
|
||||||
|
vi.stubEnv('GEMINI_API_KEY', undefined);
|
||||||
expect(validateAuthMethod(AuthType.USE_GEMINI)).toBe(
|
expect(validateAuthMethod(AuthType.USE_GEMINI)).toBe(
|
||||||
'GEMINI_API_KEY environment variable not found. Add that to your environment and try again (no reload needed if using .env)!',
|
'GEMINI_API_KEY environment variable not found. Add that to your environment and try again (no reload needed if using .env)!',
|
||||||
);
|
);
|
||||||
@@ -47,17 +52,19 @@ describe('validateAuthMethod', () => {
|
|||||||
|
|
||||||
describe('USE_VERTEX_AI', () => {
|
describe('USE_VERTEX_AI', () => {
|
||||||
it('should return null if GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION are set', () => {
|
it('should return null if GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION are set', () => {
|
||||||
process.env['GOOGLE_CLOUD_PROJECT'] = 'test-project';
|
vi.stubEnv('GOOGLE_CLOUD_PROJECT', 'test-project');
|
||||||
process.env['GOOGLE_CLOUD_LOCATION'] = 'test-location';
|
vi.stubEnv('GOOGLE_CLOUD_LOCATION', 'test-location');
|
||||||
expect(validateAuthMethod(AuthType.USE_VERTEX_AI)).toBeNull();
|
expect(validateAuthMethod(AuthType.USE_VERTEX_AI)).toBeNull();
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return null if GOOGLE_API_KEY is set', () => {
|
it('should return null if GOOGLE_API_KEY is set', () => {
|
||||||
process.env['GOOGLE_API_KEY'] = 'test-api-key';
|
vi.stubEnv('GOOGLE_API_KEY', 'test-api-key');
|
||||||
expect(validateAuthMethod(AuthType.USE_VERTEX_AI)).toBeNull();
|
expect(validateAuthMethod(AuthType.USE_VERTEX_AI)).toBeNull();
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return an error message if no required environment variables are set', () => {
|
it('should return an error message if no required environment variables are set', () => {
|
||||||
|
vi.stubEnv('GOOGLE_CLOUD_PROJECT', undefined);
|
||||||
|
vi.stubEnv('GOOGLE_CLOUD_LOCATION', undefined);
|
||||||
expect(validateAuthMethod(AuthType.USE_VERTEX_AI)).toBe(
|
expect(validateAuthMethod(AuthType.USE_VERTEX_AI)).toBe(
|
||||||
'When using Vertex AI, you must specify either:\n' +
|
'When using Vertex AI, you must specify either:\n' +
|
||||||
'• GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION environment variables.\n' +
|
'• GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION environment variables.\n' +
|
||||||
|
|||||||
@@ -5,10 +5,10 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
import { AuthType } from '@qwen-code/qwen-code-core';
|
import { AuthType } from '@qwen-code/qwen-code-core';
|
||||||
import { loadEnvironment } from './settings.js';
|
import { loadEnvironment, loadSettings } from './settings.js';
|
||||||
|
|
||||||
export const validateAuthMethod = (authMethod: string): string | null => {
|
export function validateAuthMethod(authMethod: string): string | null {
|
||||||
loadEnvironment();
|
loadEnvironment(loadSettings().merged);
|
||||||
if (
|
if (
|
||||||
authMethod === AuthType.LOGIN_WITH_GOOGLE ||
|
authMethod === AuthType.LOGIN_WITH_GOOGLE ||
|
||||||
authMethod === AuthType.CLOUD_SHELL
|
authMethod === AuthType.CLOUD_SHELL
|
||||||
@@ -53,7 +53,7 @@ export const validateAuthMethod = (authMethod: string): string | null => {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return 'Invalid auth method selected.';
|
return 'Invalid auth method selected.';
|
||||||
};
|
}
|
||||||
|
|
||||||
export const setOpenAIApiKey = (apiKey: string): void => {
|
export const setOpenAIApiKey = (apiKey: string): void => {
|
||||||
process.env['OPENAI_API_KEY'] = apiKey;
|
process.env['OPENAI_API_KEY'] = apiKey;
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -5,16 +5,16 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
import type {
|
import type {
|
||||||
ConfigParameters,
|
|
||||||
FileFilteringOptions,
|
FileFilteringOptions,
|
||||||
MCPServerConfig,
|
MCPServerConfig,
|
||||||
TelemetryTarget,
|
OutputFormat,
|
||||||
} from '@qwen-code/qwen-code-core';
|
} from '@qwen-code/qwen-code-core';
|
||||||
|
import { extensionsCommand } from '../commands/extensions.js';
|
||||||
import {
|
import {
|
||||||
ApprovalMode,
|
ApprovalMode,
|
||||||
Config,
|
Config,
|
||||||
DEFAULT_GEMINI_EMBEDDING_MODEL,
|
DEFAULT_QWEN_MODEL,
|
||||||
DEFAULT_GEMINI_MODEL,
|
DEFAULT_QWEN_EMBEDDING_MODEL,
|
||||||
DEFAULT_MEMORY_FILE_FILTERING_OPTIONS,
|
DEFAULT_MEMORY_FILE_FILTERING_OPTIONS,
|
||||||
EditTool,
|
EditTool,
|
||||||
FileDiscoveryService,
|
FileDiscoveryService,
|
||||||
@@ -23,24 +23,26 @@ import {
|
|||||||
setGeminiMdFilename as setServerGeminiMdFilename,
|
setGeminiMdFilename as setServerGeminiMdFilename,
|
||||||
ShellTool,
|
ShellTool,
|
||||||
WriteFileTool,
|
WriteFileTool,
|
||||||
|
resolveTelemetrySettings,
|
||||||
|
FatalConfigError,
|
||||||
} from '@qwen-code/qwen-code-core';
|
} from '@qwen-code/qwen-code-core';
|
||||||
import * as fs from 'node:fs';
|
|
||||||
import { homedir } from 'node:os';
|
|
||||||
import * as path from 'node:path';
|
|
||||||
import process from 'node:process';
|
|
||||||
import { hideBin } from 'yargs/helpers';
|
|
||||||
import yargs from 'yargs/yargs';
|
|
||||||
import { extensionsCommand } from '../commands/extensions.js';
|
|
||||||
import { mcpCommand } from '../commands/mcp.js';
|
|
||||||
import type { Settings } from './settings.js';
|
import type { Settings } from './settings.js';
|
||||||
|
import yargs, { type Argv } from 'yargs';
|
||||||
|
import { hideBin } from 'yargs/helpers';
|
||||||
|
import * as fs from 'node:fs';
|
||||||
|
import * as path from 'node:path';
|
||||||
|
import { homedir } from 'node:os';
|
||||||
|
|
||||||
import { resolvePath } from '../utils/resolvePath.js';
|
import { resolvePath } from '../utils/resolvePath.js';
|
||||||
import { getCliVersion } from '../utils/version.js';
|
import { getCliVersion } from '../utils/version.js';
|
||||||
import type { Extension } from './extension.js';
|
import type { Extension } from './extension.js';
|
||||||
import { annotateActiveExtensions } from './extension.js';
|
import { annotateActiveExtensions } from './extension.js';
|
||||||
import { loadSandboxConfig } from './sandboxConfig.js';
|
import { loadSandboxConfig } from './sandboxConfig.js';
|
||||||
|
import { appEvents } from '../utils/events.js';
|
||||||
|
import { mcpCommand } from '../commands/mcp.js';
|
||||||
|
|
||||||
import { isWorkspaceTrusted } from './trustedFolders.js';
|
import { isWorkspaceTrusted } from './trustedFolders.js';
|
||||||
|
import type { ExtensionEnablementManager } from './extensions/extensionEnablement.js';
|
||||||
|
|
||||||
// Simple console logger for now - replace with actual logger if available
|
// Simple console logger for now - replace with actual logger if available
|
||||||
const logger = {
|
const logger = {
|
||||||
@@ -86,6 +88,7 @@ function parseApprovalModeValue(value: string): ApprovalMode {
|
|||||||
}
|
}
|
||||||
|
|
||||||
export interface CliArgs {
|
export interface CliArgs {
|
||||||
|
query: string | undefined;
|
||||||
model: string | undefined;
|
model: string | undefined;
|
||||||
sandbox: boolean | string | undefined;
|
sandbox: boolean | string | undefined;
|
||||||
sandboxImage: string | undefined;
|
sandboxImage: string | undefined;
|
||||||
@@ -116,74 +119,18 @@ export interface CliArgs {
|
|||||||
tavilyApiKey: string | undefined;
|
tavilyApiKey: string | undefined;
|
||||||
screenReader: boolean | undefined;
|
screenReader: boolean | undefined;
|
||||||
vlmSwitchMode: string | undefined;
|
vlmSwitchMode: string | undefined;
|
||||||
|
useSmartEdit: boolean | undefined;
|
||||||
|
outputFormat: string | undefined;
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
||||||
const yargsInstance = yargs(hideBin(process.argv))
|
const rawArgv = hideBin(process.argv);
|
||||||
// Set locale to English for consistent output, especially in tests
|
const yargsInstance = yargs(rawArgv)
|
||||||
.locale('en')
|
.locale('en')
|
||||||
.scriptName('qwen')
|
.scriptName('qwen')
|
||||||
.usage(
|
.usage(
|
||||||
'Usage: qwen [options] [command]\n\nQwen Code - Launch an interactive CLI, use -p/--prompt for non-interactive mode',
|
'Usage: qwen [options] [command]\n\nQwen Code - Launch an interactive CLI, use -p/--prompt for non-interactive mode',
|
||||||
)
|
)
|
||||||
.command('$0', 'Launch Qwen Code', (yargsInstance) =>
|
|
||||||
yargsInstance
|
|
||||||
.option('model', {
|
|
||||||
alias: 'm',
|
|
||||||
type: 'string',
|
|
||||||
description: `Model`,
|
|
||||||
default: process.env['GEMINI_MODEL'],
|
|
||||||
})
|
|
||||||
.option('prompt', {
|
|
||||||
alias: 'p',
|
|
||||||
type: 'string',
|
|
||||||
description: 'Prompt. Appended to input on stdin (if any).',
|
|
||||||
})
|
|
||||||
.option('prompt-interactive', {
|
|
||||||
alias: 'i',
|
|
||||||
type: 'string',
|
|
||||||
description:
|
|
||||||
'Execute the provided prompt and continue in interactive mode',
|
|
||||||
})
|
|
||||||
.option('sandbox', {
|
|
||||||
alias: 's',
|
|
||||||
type: 'boolean',
|
|
||||||
description: 'Run in sandbox?',
|
|
||||||
})
|
|
||||||
.option('sandbox-image', {
|
|
||||||
type: 'string',
|
|
||||||
description: 'Sandbox image URI.',
|
|
||||||
})
|
|
||||||
.option('debug', {
|
|
||||||
alias: 'd',
|
|
||||||
type: 'boolean',
|
|
||||||
description: 'Run in debug mode?',
|
|
||||||
default: false,
|
|
||||||
})
|
|
||||||
.option('all-files', {
|
|
||||||
alias: ['a'],
|
|
||||||
type: 'boolean',
|
|
||||||
description: 'Include ALL files in context?',
|
|
||||||
default: false,
|
|
||||||
})
|
|
||||||
.option('show-memory-usage', {
|
|
||||||
type: 'boolean',
|
|
||||||
description: 'Show memory usage in status bar',
|
|
||||||
default: false,
|
|
||||||
})
|
|
||||||
.option('yolo', {
|
|
||||||
alias: 'y',
|
|
||||||
type: 'boolean',
|
|
||||||
description:
|
|
||||||
'Automatically accept all actions (aka YOLO mode, see https://www.youtube.com/watch?v=xvFZjo5PgG0 for more details)?',
|
|
||||||
default: false,
|
|
||||||
})
|
|
||||||
.option('approval-mode', {
|
|
||||||
type: 'string',
|
|
||||||
choices: ['plan', 'default', 'auto-edit', 'yolo'],
|
|
||||||
description:
|
|
||||||
'Set the approval mode: plan (plan only), default (prompt for approval), auto-edit (auto-approve edit tools), yolo (auto-approve all tools)',
|
|
||||||
})
|
|
||||||
.option('telemetry', {
|
.option('telemetry', {
|
||||||
type: 'boolean',
|
type: 'boolean',
|
||||||
description:
|
description:
|
||||||
@@ -215,6 +162,100 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
type: 'string',
|
type: 'string',
|
||||||
description: 'Redirect all telemetry output to the specified file.',
|
description: 'Redirect all telemetry output to the specified file.',
|
||||||
})
|
})
|
||||||
|
.deprecateOption(
|
||||||
|
'telemetry',
|
||||||
|
'Use the "telemetry.enabled" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'telemetry-target',
|
||||||
|
'Use the "telemetry.target" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'telemetry-otlp-endpoint',
|
||||||
|
'Use the "telemetry.otlpEndpoint" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'telemetry-otlp-protocol',
|
||||||
|
'Use the "telemetry.otlpProtocol" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'telemetry-log-prompts',
|
||||||
|
'Use the "telemetry.logPrompts" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'telemetry-outfile',
|
||||||
|
'Use the "telemetry.outfile" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.option('debug', {
|
||||||
|
alias: 'd',
|
||||||
|
type: 'boolean',
|
||||||
|
description: 'Run in debug mode?',
|
||||||
|
default: false,
|
||||||
|
})
|
||||||
|
.option('proxy', {
|
||||||
|
type: 'string',
|
||||||
|
description:
|
||||||
|
'Proxy for gemini client, like schema://user:password@host:port',
|
||||||
|
})
|
||||||
|
.deprecateOption(
|
||||||
|
'proxy',
|
||||||
|
'Use the "proxy" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.command('$0 [query..]', 'Launch Gemini CLI', (yargsInstance: Argv) =>
|
||||||
|
yargsInstance
|
||||||
|
.positional('query', {
|
||||||
|
description:
|
||||||
|
'Positional prompt. Defaults to one-shot; use -i/--prompt-interactive for interactive.',
|
||||||
|
})
|
||||||
|
.option('model', {
|
||||||
|
alias: 'm',
|
||||||
|
type: 'string',
|
||||||
|
description: `Model`,
|
||||||
|
})
|
||||||
|
.option('prompt', {
|
||||||
|
alias: 'p',
|
||||||
|
type: 'string',
|
||||||
|
description: 'Prompt. Appended to input on stdin (if any).',
|
||||||
|
})
|
||||||
|
.option('prompt-interactive', {
|
||||||
|
alias: 'i',
|
||||||
|
type: 'string',
|
||||||
|
description:
|
||||||
|
'Execute the provided prompt and continue in interactive mode',
|
||||||
|
})
|
||||||
|
.option('sandbox', {
|
||||||
|
alias: 's',
|
||||||
|
type: 'boolean',
|
||||||
|
description: 'Run in sandbox?',
|
||||||
|
})
|
||||||
|
.option('sandbox-image', {
|
||||||
|
type: 'string',
|
||||||
|
description: 'Sandbox image URI.',
|
||||||
|
})
|
||||||
|
.option('all-files', {
|
||||||
|
alias: ['a'],
|
||||||
|
type: 'boolean',
|
||||||
|
description: 'Include ALL files in context?',
|
||||||
|
default: false,
|
||||||
|
})
|
||||||
|
.option('show-memory-usage', {
|
||||||
|
type: 'boolean',
|
||||||
|
description: 'Show memory usage in status bar',
|
||||||
|
default: false,
|
||||||
|
})
|
||||||
|
.option('yolo', {
|
||||||
|
alias: 'y',
|
||||||
|
type: 'boolean',
|
||||||
|
description:
|
||||||
|
'Automatically accept all actions (aka YOLO mode, see https://www.youtube.com/watch?v=xvFZjo5PgG0 for more details)?',
|
||||||
|
default: false,
|
||||||
|
})
|
||||||
|
.option('approval-mode', {
|
||||||
|
type: 'string',
|
||||||
|
choices: ['plan', 'default', 'auto-edit', 'yolo'],
|
||||||
|
description:
|
||||||
|
'Set the approval mode: plan (plan only), default (prompt for approval), auto-edit (auto-approve edit tools), yolo (auto-approve all tools)',
|
||||||
|
})
|
||||||
.option('checkpointing', {
|
.option('checkpointing', {
|
||||||
alias: 'c',
|
alias: 'c',
|
||||||
type: 'boolean',
|
type: 'boolean',
|
||||||
@@ -229,11 +270,19 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
type: 'array',
|
type: 'array',
|
||||||
string: true,
|
string: true,
|
||||||
description: 'Allowed MCP server names',
|
description: 'Allowed MCP server names',
|
||||||
|
coerce: (mcpServerNames: string[]) =>
|
||||||
|
// Handle comma-separated values
|
||||||
|
mcpServerNames.flatMap((mcpServerName) =>
|
||||||
|
mcpServerName.split(',').map((m) => m.trim()),
|
||||||
|
),
|
||||||
})
|
})
|
||||||
.option('allowed-tools', {
|
.option('allowed-tools', {
|
||||||
type: 'array',
|
type: 'array',
|
||||||
string: true,
|
string: true,
|
||||||
description: 'Tools that are allowed to run without confirmation',
|
description: 'Tools that are allowed to run without confirmation',
|
||||||
|
coerce: (tools: string[]) =>
|
||||||
|
// Handle comma-separated values
|
||||||
|
tools.flatMap((tool) => tool.split(',').map((t) => t.trim())),
|
||||||
})
|
})
|
||||||
.option('extensions', {
|
.option('extensions', {
|
||||||
alias: 'e',
|
alias: 'e',
|
||||||
@@ -241,17 +290,17 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
string: true,
|
string: true,
|
||||||
description:
|
description:
|
||||||
'A list of extensions to use. If not provided, all extensions are used.',
|
'A list of extensions to use. If not provided, all extensions are used.',
|
||||||
|
coerce: (extensions: string[]) =>
|
||||||
|
// Handle comma-separated values
|
||||||
|
extensions.flatMap((extension) =>
|
||||||
|
extension.split(',').map((e) => e.trim()),
|
||||||
|
),
|
||||||
})
|
})
|
||||||
.option('list-extensions', {
|
.option('list-extensions', {
|
||||||
alias: 'l',
|
alias: 'l',
|
||||||
type: 'boolean',
|
type: 'boolean',
|
||||||
description: 'List all available extensions and exit.',
|
description: 'List all available extensions and exit.',
|
||||||
})
|
})
|
||||||
.option('proxy', {
|
|
||||||
type: 'string',
|
|
||||||
description:
|
|
||||||
'Proxy for qwen client, like schema://user:password@host:port',
|
|
||||||
})
|
|
||||||
.option('include-directories', {
|
.option('include-directories', {
|
||||||
type: 'array',
|
type: 'array',
|
||||||
string: true,
|
string: true,
|
||||||
@@ -281,7 +330,6 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
.option('screen-reader', {
|
.option('screen-reader', {
|
||||||
type: 'boolean',
|
type: 'boolean',
|
||||||
description: 'Enable screen reader mode for accessibility.',
|
description: 'Enable screen reader mode for accessibility.',
|
||||||
default: false,
|
|
||||||
})
|
})
|
||||||
.option('vlm-switch-mode', {
|
.option('vlm-switch-mode', {
|
||||||
type: 'string',
|
type: 'string',
|
||||||
@@ -290,16 +338,54 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
'Default behavior when images are detected in input. Values: once (one-time switch), session (switch for entire session), persist (continue with current model). Overrides settings files.',
|
'Default behavior when images are detected in input. Values: once (one-time switch), session (switch for entire session), persist (continue with current model). Overrides settings files.',
|
||||||
default: process.env['VLM_SWITCH_MODE'],
|
default: process.env['VLM_SWITCH_MODE'],
|
||||||
})
|
})
|
||||||
.check((argv) => {
|
.option('output-format', {
|
||||||
if (argv.prompt && argv['promptInteractive']) {
|
alias: 'o',
|
||||||
throw new Error(
|
type: 'string',
|
||||||
'Cannot use both --prompt (-p) and --prompt-interactive (-i) together',
|
description: 'The format of the CLI output.',
|
||||||
);
|
choices: ['text', 'json'],
|
||||||
|
})
|
||||||
|
.deprecateOption(
|
||||||
|
'show-memory-usage',
|
||||||
|
'Use the "ui.showMemoryUsage" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'sandbox-image',
|
||||||
|
'Use the "tools.sandbox" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'checkpointing',
|
||||||
|
'Use the "general.checkpointing.enabled" setting in settings.json instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'all-files',
|
||||||
|
'Use @ includes in the application instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
.deprecateOption(
|
||||||
|
'prompt',
|
||||||
|
'Use the positional prompt instead. This flag will be removed in a future version.',
|
||||||
|
)
|
||||||
|
// Ensure validation flows through .fail() for clean UX
|
||||||
|
.fail((msg: string, err: Error | undefined, yargs: Argv) => {
|
||||||
|
console.error(msg || err?.message || 'Unknown error');
|
||||||
|
yargs.showHelp();
|
||||||
|
process.exit(1);
|
||||||
|
})
|
||||||
|
.check((argv: { [x: string]: unknown }) => {
|
||||||
|
// The 'query' positional can be a string (for one arg) or string[] (for multiple).
|
||||||
|
// This guard safely checks if any positional argument was provided.
|
||||||
|
const query = argv['query'] as string | string[] | undefined;
|
||||||
|
const hasPositionalQuery = Array.isArray(query)
|
||||||
|
? query.length > 0
|
||||||
|
: !!query;
|
||||||
|
|
||||||
|
if (argv['prompt'] && hasPositionalQuery) {
|
||||||
|
return 'Cannot use both a positional prompt and the --prompt (-p) flag together';
|
||||||
}
|
}
|
||||||
if (argv.yolo && argv['approvalMode']) {
|
if (argv['prompt'] && argv['promptInteractive']) {
|
||||||
throw new Error(
|
return 'Cannot use both --prompt (-p) and --prompt-interactive (-i) together';
|
||||||
'Cannot use both --yolo (-y) and --approval-mode together. Use --approval-mode=yolo instead.',
|
}
|
||||||
);
|
if (argv['yolo'] && argv['approvalMode']) {
|
||||||
|
return 'Cannot use both --yolo (-y) and --approval-mode together. Use --approval-mode=yolo instead.';
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}),
|
}),
|
||||||
@@ -307,7 +393,7 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
// Register MCP subcommands
|
// Register MCP subcommands
|
||||||
.command(mcpCommand);
|
.command(mcpCommand);
|
||||||
|
|
||||||
if (settings?.experimental?.extensionManagement ?? false) {
|
if (settings?.experimental?.extensionManagement ?? true) {
|
||||||
yargsInstance.command(extensionsCommand);
|
yargsInstance.command(extensionsCommand);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -322,6 +408,8 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
yargsInstance.wrap(yargsInstance.terminalWidth());
|
yargsInstance.wrap(yargsInstance.terminalWidth());
|
||||||
const result = await yargsInstance.parse();
|
const result = await yargsInstance.parse();
|
||||||
|
|
||||||
|
// If yargs handled --help/--version it will have exited; nothing to do here.
|
||||||
|
|
||||||
// Handle case where MCP subcommands are executed - they should exit the process
|
// Handle case where MCP subcommands are executed - they should exit the process
|
||||||
// and not return to main CLI logic
|
// and not return to main CLI logic
|
||||||
if (
|
if (
|
||||||
@@ -332,6 +420,26 @@ export async function parseArguments(settings: Settings): Promise<CliArgs> {
|
|||||||
process.exit(0);
|
process.exit(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Normalize query args: handle both quoted "@path file" and unquoted @path file
|
||||||
|
const queryArg = (result as { query?: string | string[] | undefined }).query;
|
||||||
|
const q: string | undefined = Array.isArray(queryArg)
|
||||||
|
? queryArg.join(' ')
|
||||||
|
: queryArg;
|
||||||
|
|
||||||
|
// Route positional args: explicit -i flag -> interactive; else -> one-shot (even for @commands)
|
||||||
|
if (q && !result['prompt']) {
|
||||||
|
const hasExplicitInteractive =
|
||||||
|
result['promptInteractive'] === '' || !!result['promptInteractive'];
|
||||||
|
if (hasExplicitInteractive) {
|
||||||
|
result['promptInteractive'] = q;
|
||||||
|
} else {
|
||||||
|
result['prompt'] = q;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Keep CliArgs.query as a string for downstream typing
|
||||||
|
(result as Record<string, unknown>)['query'] = q || undefined;
|
||||||
|
|
||||||
// The import format is now only controlled by settings.memoryImportFormat
|
// The import format is now only controlled by settings.memoryImportFormat
|
||||||
// We no longer accept it as a CLI argument
|
// We no longer accept it as a CLI argument
|
||||||
return result as unknown as CliArgs;
|
return result as unknown as CliArgs;
|
||||||
@@ -347,6 +455,7 @@ export async function loadHierarchicalGeminiMemory(
|
|||||||
fileService: FileDiscoveryService,
|
fileService: FileDiscoveryService,
|
||||||
settings: Settings,
|
settings: Settings,
|
||||||
extensionContextFilePaths: string[] = [],
|
extensionContextFilePaths: string[] = [],
|
||||||
|
folderTrust: boolean,
|
||||||
memoryImportFormat: 'flat' | 'tree' = 'tree',
|
memoryImportFormat: 'flat' | 'tree' = 'tree',
|
||||||
fileFilteringOptions?: FileFilteringOptions,
|
fileFilteringOptions?: FileFilteringOptions,
|
||||||
): Promise<{ memoryContent: string; fileCount: number }> {
|
): Promise<{ memoryContent: string; fileCount: number }> {
|
||||||
@@ -372,58 +481,48 @@ export async function loadHierarchicalGeminiMemory(
|
|||||||
debugMode,
|
debugMode,
|
||||||
fileService,
|
fileService,
|
||||||
extensionContextFilePaths,
|
extensionContextFilePaths,
|
||||||
|
folderTrust,
|
||||||
memoryImportFormat,
|
memoryImportFormat,
|
||||||
fileFilteringOptions,
|
fileFilteringOptions,
|
||||||
settings.context?.discoveryMaxDirs,
|
settings.context?.discoveryMaxDirs,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function isDebugMode(argv: CliArgs): boolean {
|
||||||
|
return (
|
||||||
|
argv.debug ||
|
||||||
|
[process.env['DEBUG'], process.env['DEBUG_MODE']].some(
|
||||||
|
(v) => v === 'true' || v === '1',
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
export async function loadCliConfig(
|
export async function loadCliConfig(
|
||||||
settings: Settings,
|
settings: Settings,
|
||||||
extensions: Extension[],
|
extensions: Extension[],
|
||||||
|
extensionEnablementManager: ExtensionEnablementManager,
|
||||||
sessionId: string,
|
sessionId: string,
|
||||||
argv: CliArgs,
|
argv: CliArgs,
|
||||||
cwd: string = process.cwd(),
|
cwd: string = process.cwd(),
|
||||||
): Promise<Config> {
|
): Promise<Config> {
|
||||||
const debugMode =
|
const debugMode = isDebugMode(argv);
|
||||||
argv.debug ||
|
|
||||||
[process.env['DEBUG'], process.env['DEBUG_MODE']].some(
|
|
||||||
(v) => v === 'true' || v === '1',
|
|
||||||
) ||
|
|
||||||
false;
|
|
||||||
const memoryImportFormat = settings.context?.importFormat || 'tree';
|
const memoryImportFormat = settings.context?.importFormat || 'tree';
|
||||||
|
|
||||||
const ideMode = settings.ide?.enabled ?? false;
|
const ideMode = settings.ide?.enabled ?? false;
|
||||||
|
|
||||||
const folderTrustFeature =
|
const folderTrust = settings.security?.folderTrust?.enabled ?? false;
|
||||||
settings.security?.folderTrust?.featureEnabled ?? false;
|
const trustedFolder = isWorkspaceTrusted(settings)?.isTrusted ?? true;
|
||||||
const folderTrustSetting = settings.security?.folderTrust?.enabled ?? true;
|
|
||||||
const folderTrust = folderTrustFeature && folderTrustSetting;
|
|
||||||
const trustedFolder = isWorkspaceTrusted(settings);
|
|
||||||
|
|
||||||
const allExtensions = annotateActiveExtensions(
|
const allExtensions = annotateActiveExtensions(
|
||||||
extensions,
|
extensions,
|
||||||
argv.extensions || [],
|
|
||||||
cwd,
|
cwd,
|
||||||
|
extensionEnablementManager,
|
||||||
);
|
);
|
||||||
|
|
||||||
const activeExtensions = extensions.filter(
|
const activeExtensions = extensions.filter(
|
||||||
(_, i) => allExtensions[i].isActive,
|
(_, i) => allExtensions[i].isActive,
|
||||||
);
|
);
|
||||||
// Handle OpenAI API key from command line
|
|
||||||
if (argv.openaiApiKey) {
|
|
||||||
process.env['OPENAI_API_KEY'] = argv.openaiApiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle OpenAI base URL from command line
|
|
||||||
if (argv.openaiBaseUrl) {
|
|
||||||
process.env['OPENAI_BASE_URL'] = argv.openaiBaseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle Tavily API key from command line
|
|
||||||
if (argv.tavilyApiKey) {
|
|
||||||
process.env['TAVILY_API_KEY'] = argv.tavilyApiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Set the context filename in the server's memoryTool module BEFORE loading memory
|
// Set the context filename in the server's memoryTool module BEFORE loading memory
|
||||||
// TODO(b/343434939): This is a bit of a hack. The contextFileName should ideally be passed
|
// TODO(b/343434939): This is a bit of a hack. The contextFileName should ideally be passed
|
||||||
@@ -461,6 +560,7 @@ export async function loadCliConfig(
|
|||||||
fileService,
|
fileService,
|
||||||
settings,
|
settings,
|
||||||
extensionContextFilePaths,
|
extensionContextFilePaths,
|
||||||
|
trustedFolder,
|
||||||
memoryImportFormat,
|
memoryImportFormat,
|
||||||
fileFiltering,
|
fileFiltering,
|
||||||
);
|
);
|
||||||
@@ -474,8 +574,8 @@ export async function loadCliConfig(
|
|||||||
approvalMode = parseApprovalModeValue(argv.approvalMode);
|
approvalMode = parseApprovalModeValue(argv.approvalMode);
|
||||||
} else if (argv.yolo) {
|
} else if (argv.yolo) {
|
||||||
approvalMode = ApprovalMode.YOLO;
|
approvalMode = ApprovalMode.YOLO;
|
||||||
} else if (settings.approvalMode) {
|
} else if (settings.tools?.approvalMode) {
|
||||||
approvalMode = parseApprovalModeValue(settings.approvalMode);
|
approvalMode = parseApprovalModeValue(settings.tools.approvalMode);
|
||||||
} else {
|
} else {
|
||||||
approvalMode = ApprovalMode.DEFAULT;
|
approvalMode = ApprovalMode.DEFAULT;
|
||||||
}
|
}
|
||||||
@@ -492,8 +592,27 @@ export async function loadCliConfig(
|
|||||||
approvalMode = ApprovalMode.DEFAULT;
|
approvalMode = ApprovalMode.DEFAULT;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let telemetrySettings;
|
||||||
|
try {
|
||||||
|
telemetrySettings = await resolveTelemetrySettings({
|
||||||
|
argv,
|
||||||
|
env: process.env as unknown as Record<string, string | undefined>,
|
||||||
|
settings: settings.telemetry,
|
||||||
|
});
|
||||||
|
} catch (err) {
|
||||||
|
if (err instanceof FatalConfigError) {
|
||||||
|
throw new FatalConfigError(
|
||||||
|
`Invalid telemetry configuration: ${err.message}.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
throw err;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Interactive mode: explicit -i flag or (TTY + no args + no -p flag)
|
||||||
|
const hasQuery = !!argv.query;
|
||||||
const interactive =
|
const interactive =
|
||||||
!!argv.promptInteractive || (process.stdin.isTTY && question.length === 0);
|
!!argv.promptInteractive ||
|
||||||
|
(process.stdin.isTTY && !hasQuery && !argv.prompt);
|
||||||
// In non-interactive mode, exclude tools that require a prompt.
|
// In non-interactive mode, exclude tools that require a prompt.
|
||||||
const extraExcludes: string[] = [];
|
const extraExcludes: string[] = [];
|
||||||
if (!interactive && !argv.experimentalAcp) {
|
if (!interactive && !argv.experimentalAcp) {
|
||||||
@@ -550,9 +669,15 @@ export async function loadCliConfig(
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
const sandboxConfig = await loadSandboxConfig(settings, argv);
|
const defaultModel = DEFAULT_QWEN_MODEL;
|
||||||
const cliVersion = await getCliVersion();
|
const resolvedModel: string =
|
||||||
|
argv.model ||
|
||||||
|
process.env['OPENAI_MODEL'] ||
|
||||||
|
process.env['QWEN_MODEL'] ||
|
||||||
|
settings.model?.name ||
|
||||||
|
defaultModel;
|
||||||
|
|
||||||
|
const sandboxConfig = await loadSandboxConfig(settings, argv);
|
||||||
const screenReader =
|
const screenReader =
|
||||||
argv.screenReader !== undefined
|
argv.screenReader !== undefined
|
||||||
? argv.screenReader
|
? argv.screenReader
|
||||||
@@ -562,7 +687,7 @@ export async function loadCliConfig(
|
|||||||
argv.vlmSwitchMode || settings.experimental?.vlmSwitchMode;
|
argv.vlmSwitchMode || settings.experimental?.vlmSwitchMode;
|
||||||
return new Config({
|
return new Config({
|
||||||
sessionId,
|
sessionId,
|
||||||
embeddingModel: DEFAULT_GEMINI_EMBEDDING_MODEL,
|
embeddingModel: DEFAULT_QWEN_EMBEDDING_MODEL,
|
||||||
sandbox: sandboxConfig,
|
sandbox: sandboxConfig,
|
||||||
targetDir: cwd,
|
targetDir: cwd,
|
||||||
includeDirectories,
|
includeDirectories,
|
||||||
@@ -587,31 +712,9 @@ export async function loadCliConfig(
|
|||||||
...settings.ui?.accessibility,
|
...settings.ui?.accessibility,
|
||||||
screenReader,
|
screenReader,
|
||||||
},
|
},
|
||||||
telemetry: {
|
telemetry: telemetrySettings,
|
||||||
enabled: argv.telemetry ?? settings.telemetry?.enabled,
|
|
||||||
target: (argv.telemetryTarget ??
|
|
||||||
settings.telemetry?.target) as TelemetryTarget,
|
|
||||||
otlpEndpoint:
|
|
||||||
argv.telemetryOtlpEndpoint ??
|
|
||||||
process.env['OTEL_EXPORTER_OTLP_ENDPOINT'] ??
|
|
||||||
settings.telemetry?.otlpEndpoint,
|
|
||||||
otlpProtocol: (['grpc', 'http'] as const).find(
|
|
||||||
(p) =>
|
|
||||||
p ===
|
|
||||||
(argv.telemetryOtlpProtocol ?? settings.telemetry?.otlpProtocol),
|
|
||||||
),
|
|
||||||
logPrompts: argv.telemetryLogPrompts ?? settings.telemetry?.logPrompts,
|
|
||||||
outfile: argv.telemetryOutfile ?? settings.telemetry?.outfile,
|
|
||||||
},
|
|
||||||
usageStatisticsEnabled: settings.privacy?.usageStatisticsEnabled ?? true,
|
usageStatisticsEnabled: settings.privacy?.usageStatisticsEnabled ?? true,
|
||||||
// Git-aware file filtering settings
|
fileFiltering: settings.context?.fileFiltering,
|
||||||
fileFiltering: {
|
|
||||||
respectGitIgnore: settings.context?.fileFiltering?.respectGitIgnore,
|
|
||||||
respectGeminiIgnore: settings.context?.fileFiltering?.respectGeminiIgnore,
|
|
||||||
enableRecursiveFileSearch:
|
|
||||||
settings.context?.fileFiltering?.enableRecursiveFileSearch,
|
|
||||||
disableFuzzySearch: settings.context?.fileFiltering?.disableFuzzySearch,
|
|
||||||
},
|
|
||||||
checkpointing:
|
checkpointing:
|
||||||
argv.checkpointing || settings.general?.checkpointing?.enabled,
|
argv.checkpointing || settings.general?.checkpointing?.enabled,
|
||||||
proxy:
|
proxy:
|
||||||
@@ -623,50 +726,51 @@ export async function loadCliConfig(
|
|||||||
cwd,
|
cwd,
|
||||||
fileDiscoveryService: fileService,
|
fileDiscoveryService: fileService,
|
||||||
bugCommand: settings.advanced?.bugCommand,
|
bugCommand: settings.advanced?.bugCommand,
|
||||||
model: argv.model || settings.model?.name || DEFAULT_GEMINI_MODEL,
|
model: resolvedModel,
|
||||||
extensionContextFilePaths,
|
extensionContextFilePaths,
|
||||||
sessionTokenLimit: settings.sessionTokenLimit ?? -1,
|
sessionTokenLimit: settings.model?.sessionTokenLimit ?? -1,
|
||||||
maxSessionTurns: settings.model?.maxSessionTurns ?? -1,
|
maxSessionTurns: settings.model?.maxSessionTurns ?? -1,
|
||||||
experimentalZedIntegration: argv.experimentalAcp || false,
|
experimentalZedIntegration: argv.experimentalAcp || false,
|
||||||
listExtensions: argv.listExtensions || false,
|
listExtensions: argv.listExtensions || false,
|
||||||
extensions: allExtensions,
|
extensions: allExtensions,
|
||||||
blockedMcpServers,
|
blockedMcpServers,
|
||||||
noBrowser: !!process.env['NO_BROWSER'],
|
noBrowser: !!process.env['NO_BROWSER'],
|
||||||
|
authType: settings.security?.auth?.selectedType,
|
||||||
|
generationConfig: {
|
||||||
|
...(settings.model?.generationConfig || {}),
|
||||||
|
model: resolvedModel,
|
||||||
|
apiKey: argv.openaiApiKey || process.env['OPENAI_API_KEY'],
|
||||||
|
baseUrl: argv.openaiBaseUrl || process.env['OPENAI_BASE_URL'],
|
||||||
enableOpenAILogging:
|
enableOpenAILogging:
|
||||||
(typeof argv.openaiLogging === 'undefined'
|
(typeof argv.openaiLogging === 'undefined'
|
||||||
? settings.enableOpenAILogging
|
? settings.model?.enableOpenAILogging
|
||||||
: argv.openaiLogging) ?? false,
|
: argv.openaiLogging) ?? false,
|
||||||
systemPromptMappings: (settings.systemPromptMappings ?? [
|
|
||||||
{
|
|
||||||
baseUrls: [
|
|
||||||
'https://dashscope.aliyuncs.com/compatible-mode/v1/',
|
|
||||||
'https://dashscope-intl.aliyuncs.com/compatible-mode/v1/',
|
|
||||||
],
|
|
||||||
modelNames: ['qwen3-coder-plus'],
|
|
||||||
template:
|
|
||||||
'SYSTEM_TEMPLATE:{"name":"qwen3_coder","params":{"is_git_repository":{RUNTIME_VARS_IS_GIT_REPO},"sandbox":"{RUNTIME_VARS_SANDBOX}"}}',
|
|
||||||
},
|
},
|
||||||
]) as ConfigParameters['systemPromptMappings'],
|
cliVersion: await getCliVersion(),
|
||||||
authType: settings.security?.auth?.selectedType,
|
|
||||||
contentGenerator: settings.contentGenerator,
|
|
||||||
cliVersion,
|
|
||||||
tavilyApiKey:
|
tavilyApiKey:
|
||||||
argv.tavilyApiKey ||
|
argv.tavilyApiKey ||
|
||||||
settings.tavilyApiKey ||
|
settings.advanced?.tavilyApiKey ||
|
||||||
process.env['TAVILY_API_KEY'],
|
process.env['TAVILY_API_KEY'],
|
||||||
summarizeToolOutput: settings.model?.summarizeToolOutput,
|
summarizeToolOutput: settings.model?.summarizeToolOutput,
|
||||||
ideMode,
|
ideMode,
|
||||||
chatCompression: settings.model?.chatCompression,
|
chatCompression: settings.model?.chatCompression,
|
||||||
folderTrustFeature,
|
|
||||||
folderTrust,
|
folderTrust,
|
||||||
interactive,
|
interactive,
|
||||||
trustedFolder,
|
trustedFolder,
|
||||||
useRipgrep: settings.tools?.useRipgrep,
|
useRipgrep: settings.tools?.useRipgrep,
|
||||||
shouldUseNodePtyShell: settings.tools?.usePty,
|
shouldUseNodePtyShell: settings.tools?.shell?.enableInteractiveShell,
|
||||||
skipNextSpeakerCheck: settings.model?.skipNextSpeakerCheck,
|
skipNextSpeakerCheck: settings.model?.skipNextSpeakerCheck,
|
||||||
enablePromptCompletion: settings.general?.enablePromptCompletion ?? false,
|
enablePromptCompletion: settings.general?.enablePromptCompletion ?? false,
|
||||||
skipLoopDetection: settings.skipLoopDetection ?? false,
|
skipLoopDetection: settings.model?.skipLoopDetection ?? false,
|
||||||
vlmSwitchMode,
|
vlmSwitchMode,
|
||||||
|
truncateToolOutputThreshold: settings.tools?.truncateToolOutputThreshold,
|
||||||
|
truncateToolOutputLines: settings.tools?.truncateToolOutputLines,
|
||||||
|
enableToolOutputTruncation: settings.tools?.enableToolOutputTruncation,
|
||||||
|
eventEmitter: appEvents,
|
||||||
|
useSmartEdit: argv.useSmartEdit ?? settings.useSmartEdit,
|
||||||
|
output: {
|
||||||
|
format: (argv.outputFormat ?? settings.output?.format) as OutputFormat,
|
||||||
|
},
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -7,19 +7,42 @@
|
|||||||
import type {
|
import type {
|
||||||
MCPServerConfig,
|
MCPServerConfig,
|
||||||
GeminiCLIExtension,
|
GeminiCLIExtension,
|
||||||
|
ExtensionInstallMetadata,
|
||||||
|
} from '@qwen-code/qwen-code-core';
|
||||||
|
import {
|
||||||
|
QWEN_DIR,
|
||||||
|
Storage,
|
||||||
|
Config,
|
||||||
|
ExtensionInstallEvent,
|
||||||
|
ExtensionUninstallEvent,
|
||||||
|
ExtensionDisableEvent,
|
||||||
|
ExtensionEnableEvent,
|
||||||
|
logExtensionEnable,
|
||||||
|
logExtensionInstallEvent,
|
||||||
|
logExtensionUninstall,
|
||||||
|
logExtensionDisable,
|
||||||
} from '@qwen-code/qwen-code-core';
|
} from '@qwen-code/qwen-code-core';
|
||||||
import { Storage } from '@qwen-code/qwen-code-core';
|
|
||||||
import * as fs from 'node:fs';
|
import * as fs from 'node:fs';
|
||||||
import * as path from 'node:path';
|
import * as path from 'node:path';
|
||||||
import * as os from 'node:os';
|
import * as os from 'node:os';
|
||||||
import { simpleGit } from 'simple-git';
|
|
||||||
import { SettingScope, loadSettings } from '../config/settings.js';
|
import { SettingScope, loadSettings } from '../config/settings.js';
|
||||||
import { getErrorMessage } from '../utils/errors.js';
|
import { getErrorMessage } from '../utils/errors.js';
|
||||||
import { recursivelyHydrateStrings } from './extensions/variables.js';
|
import { recursivelyHydrateStrings } from './extensions/variables.js';
|
||||||
|
import { isWorkspaceTrusted } from './trustedFolders.js';
|
||||||
|
import { resolveEnvVarsInObject } from '../utils/envVarResolver.js';
|
||||||
|
import { randomUUID } from 'node:crypto';
|
||||||
|
import {
|
||||||
|
cloneFromGit,
|
||||||
|
downloadFromGitHubRelease,
|
||||||
|
} from './extensions/github.js';
|
||||||
|
import type { LoadExtensionContext } from './extensions/variableSchema.js';
|
||||||
|
import { ExtensionEnablementManager } from './extensions/extensionEnablement.js';
|
||||||
|
import chalk from 'chalk';
|
||||||
|
import type { ConfirmationRequest } from '../ui/types.js';
|
||||||
|
|
||||||
|
export const EXTENSIONS_DIRECTORY_NAME = path.join(QWEN_DIR, 'extensions');
|
||||||
|
|
||||||
export const EXTENSIONS_DIRECTORY_NAME = path.join('.qwen', 'extensions');
|
|
||||||
export const EXTENSIONS_CONFIG_FILENAME = 'qwen-extension.json';
|
export const EXTENSIONS_CONFIG_FILENAME = 'qwen-extension.json';
|
||||||
export const EXTENSIONS_CONFIG_FILENAME_OLD = 'gemini-extension.json';
|
|
||||||
export const INSTALL_METADATA_FILENAME = '.qwen-extension-install.json';
|
export const INSTALL_METADATA_FILENAME = '.qwen-extension-install.json';
|
||||||
|
|
||||||
export interface Extension {
|
export interface Extension {
|
||||||
@@ -37,12 +60,8 @@ export interface ExtensionConfig {
|
|||||||
excludeTools?: string[];
|
excludeTools?: string[];
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface ExtensionInstallMetadata {
|
|
||||||
source: string;
|
|
||||||
type: 'git' | 'local';
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface ExtensionUpdateInfo {
|
export interface ExtensionUpdateInfo {
|
||||||
|
name: string;
|
||||||
originalVersion: string;
|
originalVersion: string;
|
||||||
updatedVersion: string;
|
updatedVersion: string;
|
||||||
}
|
}
|
||||||
@@ -76,10 +95,14 @@ export class ExtensionStorage {
|
|||||||
}
|
}
|
||||||
|
|
||||||
export function getWorkspaceExtensions(workspaceDir: string): Extension[] {
|
export function getWorkspaceExtensions(workspaceDir: string): Extension[] {
|
||||||
|
// If the workspace dir is the user extensions dir, there are no workspace extensions.
|
||||||
|
if (path.resolve(workspaceDir) === path.resolve(os.homedir())) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
return loadExtensionsFromDir(workspaceDir);
|
return loadExtensionsFromDir(workspaceDir);
|
||||||
}
|
}
|
||||||
|
|
||||||
async function copyExtension(
|
export async function copyExtension(
|
||||||
source: string,
|
source: string,
|
||||||
destination: string,
|
destination: string,
|
||||||
): Promise<void> {
|
): Promise<void> {
|
||||||
@@ -88,6 +111,7 @@ async function copyExtension(
|
|||||||
|
|
||||||
export async function performWorkspaceExtensionMigration(
|
export async function performWorkspaceExtensionMigration(
|
||||||
extensions: Extension[],
|
extensions: Extension[],
|
||||||
|
requestConsent: (consent: string) => Promise<boolean>,
|
||||||
): Promise<string[]> {
|
): Promise<string[]> {
|
||||||
const failedInstallNames: string[] = [];
|
const failedInstallNames: string[] = [];
|
||||||
|
|
||||||
@@ -97,7 +121,7 @@ export async function performWorkspaceExtensionMigration(
|
|||||||
source: extension.path,
|
source: extension.path,
|
||||||
type: 'local',
|
type: 'local',
|
||||||
};
|
};
|
||||||
await installExtension(installMetadata);
|
await installExtension(installMetadata, requestConsent);
|
||||||
} catch (_) {
|
} catch (_) {
|
||||||
failedInstallNames.push(extension.config.name);
|
failedInstallNames.push(extension.config.name);
|
||||||
}
|
}
|
||||||
@@ -105,20 +129,41 @@ export async function performWorkspaceExtensionMigration(
|
|||||||
return failedInstallNames;
|
return failedInstallNames;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function loadExtensions(workspaceDir: string): Extension[] {
|
function getTelemetryConfig(cwd: string) {
|
||||||
|
const settings = loadSettings(cwd);
|
||||||
|
const config = new Config({
|
||||||
|
telemetry: settings.merged.telemetry,
|
||||||
|
interactive: false,
|
||||||
|
sessionId: randomUUID(),
|
||||||
|
targetDir: cwd,
|
||||||
|
cwd,
|
||||||
|
model: '',
|
||||||
|
debugMode: false,
|
||||||
|
});
|
||||||
|
return config;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function loadExtensions(
|
||||||
|
extensionEnablementManager: ExtensionEnablementManager,
|
||||||
|
workspaceDir: string = process.cwd(),
|
||||||
|
): Extension[] {
|
||||||
const settings = loadSettings(workspaceDir).merged;
|
const settings = loadSettings(workspaceDir).merged;
|
||||||
const disabledExtensions = settings.extensions?.disabled ?? [];
|
|
||||||
const allExtensions = [...loadUserExtensions()];
|
const allExtensions = [...loadUserExtensions()];
|
||||||
|
|
||||||
if (!settings.experimental?.extensionManagement) {
|
if (
|
||||||
|
(isWorkspaceTrusted(settings) ?? true) &&
|
||||||
|
// Default management setting to true
|
||||||
|
!(settings.experimental?.extensionManagement ?? true)
|
||||||
|
) {
|
||||||
allExtensions.push(...getWorkspaceExtensions(workspaceDir));
|
allExtensions.push(...getWorkspaceExtensions(workspaceDir));
|
||||||
}
|
}
|
||||||
|
|
||||||
const uniqueExtensions = new Map<string, Extension>();
|
const uniqueExtensions = new Map<string, Extension>();
|
||||||
|
|
||||||
for (const extension of allExtensions) {
|
for (const extension of allExtensions) {
|
||||||
if (
|
if (
|
||||||
!uniqueExtensions.has(extension.config.name) &&
|
!uniqueExtensions.has(extension.config.name) &&
|
||||||
!disabledExtensions.includes(extension.config.name)
|
extensionEnablementManager.isEnabled(extension.config.name, workspaceDir)
|
||||||
) {
|
) {
|
||||||
uniqueExtensions.set(extension.config.name, extension);
|
uniqueExtensions.set(extension.config.name, extension);
|
||||||
}
|
}
|
||||||
@@ -151,7 +196,7 @@ export function loadExtensionsFromDir(dir: string): Extension[] {
|
|||||||
for (const subdir of fs.readdirSync(extensionsDir)) {
|
for (const subdir of fs.readdirSync(extensionsDir)) {
|
||||||
const extensionDir = path.join(extensionsDir, subdir);
|
const extensionDir = path.join(extensionsDir, subdir);
|
||||||
|
|
||||||
const extension = loadExtension(extensionDir);
|
const extension = loadExtension({ extensionDir, workspaceDir: dir });
|
||||||
if (extension != null) {
|
if (extension != null) {
|
||||||
extensions.push(extension);
|
extensions.push(extension);
|
||||||
}
|
}
|
||||||
@@ -159,56 +204,51 @@ export function loadExtensionsFromDir(dir: string): Extension[] {
|
|||||||
return extensions;
|
return extensions;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function loadExtension(extensionDir: string): Extension | null {
|
export function loadExtension(context: LoadExtensionContext): Extension | null {
|
||||||
|
const { extensionDir, workspaceDir } = context;
|
||||||
if (!fs.statSync(extensionDir).isDirectory()) {
|
if (!fs.statSync(extensionDir).isDirectory()) {
|
||||||
console.error(
|
|
||||||
`Warning: unexpected file ${extensionDir} in extensions directory.`,
|
|
||||||
);
|
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
let configFilePath = path.join(extensionDir, EXTENSIONS_CONFIG_FILENAME);
|
const installMetadata = loadInstallMetadata(extensionDir);
|
||||||
if (!fs.existsSync(configFilePath)) {
|
let effectiveExtensionPath = extensionDir;
|
||||||
const oldConfigFilePath = path.join(
|
|
||||||
extensionDir,
|
if (installMetadata?.type === 'link') {
|
||||||
EXTENSIONS_CONFIG_FILENAME_OLD,
|
effectiveExtensionPath = installMetadata.source;
|
||||||
);
|
|
||||||
if (!fs.existsSync(oldConfigFilePath)) {
|
|
||||||
console.error(
|
|
||||||
`Warning: extension directory ${extensionDir} does not contain a config file ${configFilePath}.`,
|
|
||||||
);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
configFilePath = oldConfigFilePath;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const configContent = fs.readFileSync(configFilePath, 'utf-8');
|
let config = loadExtensionConfig({
|
||||||
const config = recursivelyHydrateStrings(JSON.parse(configContent), {
|
extensionDir: effectiveExtensionPath,
|
||||||
extensionPath: extensionDir,
|
workspaceDir,
|
||||||
'/': path.sep,
|
});
|
||||||
pathSeparator: path.sep,
|
|
||||||
}) as unknown as ExtensionConfig;
|
config = resolveEnvVarsInObject(config);
|
||||||
if (!config.name || !config.version) {
|
|
||||||
console.error(
|
if (config.mcpServers) {
|
||||||
`Invalid extension config in ${configFilePath}: missing name or version.`,
|
config.mcpServers = Object.fromEntries(
|
||||||
|
Object.entries(config.mcpServers).map(([key, value]) => [
|
||||||
|
key,
|
||||||
|
filterMcpConfig(value),
|
||||||
|
]),
|
||||||
);
|
);
|
||||||
return null;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const contextFiles = getContextFileNames(config)
|
const contextFiles = getContextFileNames(config)
|
||||||
.map((contextFileName) => path.join(extensionDir, contextFileName))
|
.map((contextFileName) =>
|
||||||
|
path.join(effectiveExtensionPath, contextFileName),
|
||||||
|
)
|
||||||
.filter((contextFilePath) => fs.existsSync(contextFilePath));
|
.filter((contextFilePath) => fs.existsSync(contextFilePath));
|
||||||
|
|
||||||
return {
|
return {
|
||||||
path: extensionDir,
|
path: effectiveExtensionPath,
|
||||||
config,
|
config,
|
||||||
contextFiles,
|
contextFiles,
|
||||||
installMetadata: loadInstallMetadata(extensionDir),
|
installMetadata,
|
||||||
};
|
};
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error(
|
console.error(
|
||||||
`Warning: error parsing extension config in ${configFilePath}: ${getErrorMessage(
|
`Warning: Skipping extension in ${effectiveExtensionPath}: ${getErrorMessage(
|
||||||
e,
|
e,
|
||||||
)}`,
|
)}`,
|
||||||
);
|
);
|
||||||
@@ -216,7 +256,39 @@ export function loadExtension(extensionDir: string): Extension | null {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function loadInstallMetadata(
|
export function loadExtensionByName(
|
||||||
|
name: string,
|
||||||
|
workspaceDir: string = process.cwd(),
|
||||||
|
): Extension | null {
|
||||||
|
const userExtensionsDir = ExtensionStorage.getUserExtensionsDir();
|
||||||
|
if (!fs.existsSync(userExtensionsDir)) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const subdir of fs.readdirSync(userExtensionsDir)) {
|
||||||
|
const extensionDir = path.join(userExtensionsDir, subdir);
|
||||||
|
if (!fs.statSync(extensionDir).isDirectory()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const extension = loadExtension({ extensionDir, workspaceDir });
|
||||||
|
if (
|
||||||
|
extension &&
|
||||||
|
extension.config.name.toLowerCase() === name.toLowerCase()
|
||||||
|
) {
|
||||||
|
return extension;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function filterMcpConfig(original: MCPServerConfig): MCPServerConfig {
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||||
|
const { trust, ...rest } = original;
|
||||||
|
return Object.freeze(rest);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function loadInstallMetadata(
|
||||||
extensionDir: string,
|
extensionDir: string,
|
||||||
): ExtensionInstallMetadata | undefined {
|
): ExtensionInstallMetadata | undefined {
|
||||||
const metadataFilePath = path.join(extensionDir, INSTALL_METADATA_FILENAME);
|
const metadataFilePath = path.join(extensionDir, INSTALL_METADATA_FILENAME);
|
||||||
@@ -240,125 +312,175 @@ function getContextFileNames(config: ExtensionConfig): string[] {
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns an annotated list of extensions. If an extension is listed in enabledExtensionNames, it will be active.
|
* Returns an annotated list of extensions. If an extension is listed in enabledExtensionNames, it will be active.
|
||||||
* If enabledExtensionNames is empty, an extension is active unless it is in list of disabled extensions in settings.
|
* If enabledExtensionNames is empty, an extension is active unless it is disabled.
|
||||||
* @param extensions The base list of extensions.
|
* @param extensions The base list of extensions.
|
||||||
* @param enabledExtensionNames The names of explicitly enabled extensions.
|
* @param enabledExtensionNames The names of explicitly enabled extensions.
|
||||||
* @param workspaceDir The current workspace directory.
|
* @param workspaceDir The current workspace directory.
|
||||||
*/
|
*/
|
||||||
export function annotateActiveExtensions(
|
export function annotateActiveExtensions(
|
||||||
extensions: Extension[],
|
extensions: Extension[],
|
||||||
enabledExtensionNames: string[],
|
|
||||||
workspaceDir: string,
|
workspaceDir: string,
|
||||||
|
manager: ExtensionEnablementManager,
|
||||||
): GeminiCLIExtension[] {
|
): GeminiCLIExtension[] {
|
||||||
const settings = loadSettings(workspaceDir).merged;
|
manager.validateExtensionOverrides(extensions);
|
||||||
const disabledExtensions = settings.extensions?.disabled ?? [];
|
|
||||||
|
|
||||||
const annotatedExtensions: GeminiCLIExtension[] = [];
|
|
||||||
|
|
||||||
if (enabledExtensionNames.length === 0) {
|
|
||||||
return extensions.map((extension) => ({
|
return extensions.map((extension) => ({
|
||||||
name: extension.config.name,
|
name: extension.config.name,
|
||||||
version: extension.config.version,
|
version: extension.config.version,
|
||||||
isActive: !disabledExtensions.includes(extension.config.name),
|
isActive: manager.isEnabled(extension.config.name, workspaceDir),
|
||||||
path: extension.path,
|
path: extension.path,
|
||||||
|
installMetadata: extension.installMetadata,
|
||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
|
|
||||||
const lowerCaseEnabledExtensions = new Set(
|
|
||||||
enabledExtensionNames.map((e) => e.trim().toLowerCase()),
|
|
||||||
);
|
|
||||||
|
|
||||||
if (
|
|
||||||
lowerCaseEnabledExtensions.size === 1 &&
|
|
||||||
lowerCaseEnabledExtensions.has('none')
|
|
||||||
) {
|
|
||||||
return extensions.map((extension) => ({
|
|
||||||
name: extension.config.name,
|
|
||||||
version: extension.config.version,
|
|
||||||
isActive: false,
|
|
||||||
path: extension.path,
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
|
|
||||||
const notFoundNames = new Set(lowerCaseEnabledExtensions);
|
|
||||||
|
|
||||||
for (const extension of extensions) {
|
|
||||||
const lowerCaseName = extension.config.name.toLowerCase();
|
|
||||||
const isActive = lowerCaseEnabledExtensions.has(lowerCaseName);
|
|
||||||
|
|
||||||
if (isActive) {
|
|
||||||
notFoundNames.delete(lowerCaseName);
|
|
||||||
}
|
|
||||||
|
|
||||||
annotatedExtensions.push({
|
|
||||||
name: extension.config.name,
|
|
||||||
version: extension.config.version,
|
|
||||||
isActive,
|
|
||||||
path: extension.path,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const requestedName of notFoundNames) {
|
|
||||||
console.error(`Extension not found: ${requestedName}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return annotatedExtensions;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Clones a Git repository to a specified local path.
|
* Requests consent from the user to perform an action, by reading a Y/n
|
||||||
* @param gitUrl The Git URL to clone.
|
* character from stdin.
|
||||||
* @param destination The destination path to clone the repository to.
|
*
|
||||||
|
* This should not be called from interactive mode as it will break the CLI.
|
||||||
|
*
|
||||||
|
* @param consentDescription The description of the thing they will be consenting to.
|
||||||
|
* @returns boolean, whether they consented or not.
|
||||||
*/
|
*/
|
||||||
async function cloneFromGit(
|
export async function requestConsentNonInteractive(
|
||||||
gitUrl: string,
|
consentDescription: string,
|
||||||
destination: string,
|
): Promise<boolean> {
|
||||||
): Promise<void> {
|
console.info(consentDescription);
|
||||||
try {
|
const result = await promptForConsentNonInteractive(
|
||||||
// TODO(chrstnb): Download the archive instead to avoid unnecessary .git info.
|
'Do you want to continue? [Y/n]: ',
|
||||||
await simpleGit().clone(gitUrl, destination, ['--depth', '1']);
|
);
|
||||||
} catch (error) {
|
return result;
|
||||||
throw new Error(`Failed to clone Git repository from ${gitUrl}`, {
|
}
|
||||||
cause: error,
|
|
||||||
|
/**
|
||||||
|
* Requests consent from the user to perform an action, in interactive mode.
|
||||||
|
*
|
||||||
|
* This should not be called from non-interactive mode as it will not work.
|
||||||
|
*
|
||||||
|
* @param consentDescription The description of the thing they will be consenting to.
|
||||||
|
* @param setExtensionUpdateConfirmationRequest A function to actually add a prompt to the UI.
|
||||||
|
* @returns boolean, whether they consented or not.
|
||||||
|
*/
|
||||||
|
export async function requestConsentInteractive(
|
||||||
|
consentDescription: string,
|
||||||
|
addExtensionUpdateConfirmationRequest: (value: ConfirmationRequest) => void,
|
||||||
|
): Promise<boolean> {
|
||||||
|
return await promptForConsentInteractive(
|
||||||
|
consentDescription + '\n\nDo you want to continue?',
|
||||||
|
addExtensionUpdateConfirmationRequest,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Asks users a prompt and awaits for a y/n response on stdin.
|
||||||
|
*
|
||||||
|
* This should not be called from interactive mode as it will break the CLI.
|
||||||
|
*
|
||||||
|
* @param prompt A yes/no prompt to ask the user
|
||||||
|
* @returns Whether or not the user answers 'y' (yes). Defaults to 'yes' on enter.
|
||||||
|
*/
|
||||||
|
async function promptForConsentNonInteractive(
|
||||||
|
prompt: string,
|
||||||
|
): Promise<boolean> {
|
||||||
|
const readline = await import('node:readline');
|
||||||
|
const rl = readline.createInterface({
|
||||||
|
input: process.stdin,
|
||||||
|
output: process.stdout,
|
||||||
|
});
|
||||||
|
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
rl.question(prompt, (answer) => {
|
||||||
|
rl.close();
|
||||||
|
resolve(['y', ''].includes(answer.trim().toLowerCase()));
|
||||||
|
});
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Asks users an interactive yes/no prompt.
|
||||||
|
*
|
||||||
|
* This should not be called from non-interactive mode as it will break the CLI.
|
||||||
|
*
|
||||||
|
* @param prompt A markdown prompt to ask the user
|
||||||
|
* @param setExtensionUpdateConfirmationRequest Function to update the UI state with the confirmation request.
|
||||||
|
* @returns Whether or not the user answers yes.
|
||||||
|
*/
|
||||||
|
async function promptForConsentInteractive(
|
||||||
|
prompt: string,
|
||||||
|
addExtensionUpdateConfirmationRequest: (value: ConfirmationRequest) => void,
|
||||||
|
): Promise<boolean> {
|
||||||
|
return await new Promise<boolean>((resolve) => {
|
||||||
|
addExtensionUpdateConfirmationRequest({
|
||||||
|
prompt,
|
||||||
|
onConfirm: (resolvedConfirmed) => {
|
||||||
|
resolve(resolvedConfirmed);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function installExtension(
|
export async function installExtension(
|
||||||
installMetadata: ExtensionInstallMetadata,
|
installMetadata: ExtensionInstallMetadata,
|
||||||
|
requestConsent: (consent: string) => Promise<boolean>,
|
||||||
cwd: string = process.cwd(),
|
cwd: string = process.cwd(),
|
||||||
|
previousExtensionConfig?: ExtensionConfig,
|
||||||
): Promise<string> {
|
): Promise<string> {
|
||||||
|
const telemetryConfig = getTelemetryConfig(cwd);
|
||||||
|
let newExtensionConfig: ExtensionConfig | null = null;
|
||||||
|
let localSourcePath: string | undefined;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const settings = loadSettings(cwd).merged;
|
||||||
|
if (!isWorkspaceTrusted(settings)) {
|
||||||
|
throw new Error(
|
||||||
|
`Could not install extension from untrusted folder at ${installMetadata.source}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
const extensionsDir = ExtensionStorage.getUserExtensionsDir();
|
const extensionsDir = ExtensionStorage.getUserExtensionsDir();
|
||||||
await fs.promises.mkdir(extensionsDir, { recursive: true });
|
await fs.promises.mkdir(extensionsDir, { recursive: true });
|
||||||
|
|
||||||
// Convert relative paths to absolute paths for the metadata file.
|
|
||||||
if (
|
if (
|
||||||
installMetadata.type === 'local' &&
|
!path.isAbsolute(installMetadata.source) &&
|
||||||
!path.isAbsolute(installMetadata.source)
|
(installMetadata.type === 'local' || installMetadata.type === 'link')
|
||||||
) {
|
) {
|
||||||
installMetadata.source = path.resolve(cwd, installMetadata.source);
|
installMetadata.source = path.resolve(cwd, installMetadata.source);
|
||||||
}
|
}
|
||||||
|
|
||||||
let localSourcePath: string;
|
|
||||||
let tempDir: string | undefined;
|
let tempDir: string | undefined;
|
||||||
if (installMetadata.type === 'git') {
|
|
||||||
|
if (
|
||||||
|
installMetadata.type === 'git' ||
|
||||||
|
installMetadata.type === 'github-release'
|
||||||
|
) {
|
||||||
tempDir = await ExtensionStorage.createTmpDir();
|
tempDir = await ExtensionStorage.createTmpDir();
|
||||||
await cloneFromGit(installMetadata.source, tempDir);
|
|
||||||
localSourcePath = tempDir;
|
|
||||||
} else {
|
|
||||||
localSourcePath = installMetadata.source;
|
|
||||||
}
|
|
||||||
let newExtensionName: string | undefined;
|
|
||||||
try {
|
try {
|
||||||
const newExtension = loadExtension(localSourcePath);
|
const result = await downloadFromGitHubRelease(
|
||||||
if (!newExtension) {
|
installMetadata,
|
||||||
throw new Error(
|
tempDir,
|
||||||
`Invalid extension at ${installMetadata.source}. Please make sure it has a valid qwen-extension.json file.`,
|
|
||||||
);
|
);
|
||||||
|
installMetadata.type = result.type;
|
||||||
|
installMetadata.releaseTag = result.tagName;
|
||||||
|
} catch (_error) {
|
||||||
|
await cloneFromGit(installMetadata, tempDir);
|
||||||
|
installMetadata.type = 'git';
|
||||||
|
}
|
||||||
|
localSourcePath = tempDir;
|
||||||
|
} else if (
|
||||||
|
installMetadata.type === 'local' ||
|
||||||
|
installMetadata.type === 'link'
|
||||||
|
) {
|
||||||
|
localSourcePath = installMetadata.source;
|
||||||
|
} else {
|
||||||
|
throw new Error(`Unsupported install type: ${installMetadata.type}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
// ~/.qwen/extensions/{ExtensionConfig.name}.
|
try {
|
||||||
newExtensionName = newExtension.config.name;
|
newExtensionConfig = loadExtensionConfig({
|
||||||
|
extensionDir: localSourcePath,
|
||||||
|
workspaceDir: cwd,
|
||||||
|
});
|
||||||
|
|
||||||
|
const newExtensionName = newExtensionConfig.name;
|
||||||
const extensionStorage = new ExtensionStorage(newExtensionName);
|
const extensionStorage = new ExtensionStorage(newExtensionName);
|
||||||
const destinationPath = extensionStorage.getExtensionDir();
|
const destinationPath = extensionStorage.getExtensionDir();
|
||||||
|
|
||||||
@@ -372,11 +494,26 @@ export async function installExtension(
|
|||||||
`Extension "${newExtensionName}" is already installed. Please uninstall it first.`,
|
`Extension "${newExtensionName}" is already installed. Please uninstall it first.`,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
await maybeRequestConsentOrFail(
|
||||||
|
newExtensionConfig,
|
||||||
|
requestConsent,
|
||||||
|
previousExtensionConfig,
|
||||||
|
);
|
||||||
|
await fs.promises.mkdir(destinationPath, { recursive: true });
|
||||||
|
|
||||||
|
if (
|
||||||
|
installMetadata.type === 'local' ||
|
||||||
|
installMetadata.type === 'git' ||
|
||||||
|
installMetadata.type === 'github-release'
|
||||||
|
) {
|
||||||
await copyExtension(localSourcePath, destinationPath);
|
await copyExtension(localSourcePath, destinationPath);
|
||||||
|
}
|
||||||
|
|
||||||
const metadataString = JSON.stringify(installMetadata, null, 2);
|
const metadataString = JSON.stringify(installMetadata, null, 2);
|
||||||
const metadataPath = path.join(destinationPath, INSTALL_METADATA_FILENAME);
|
const metadataPath = path.join(
|
||||||
|
destinationPath,
|
||||||
|
INSTALL_METADATA_FILENAME,
|
||||||
|
);
|
||||||
await fs.promises.writeFile(metadataPath, metadataString);
|
await fs.promises.writeFile(metadataPath, metadataString);
|
||||||
} finally {
|
} finally {
|
||||||
if (tempDir) {
|
if (tempDir) {
|
||||||
@@ -384,39 +521,207 @@ export async function installExtension(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return newExtensionName;
|
logExtensionInstallEvent(
|
||||||
|
telemetryConfig,
|
||||||
|
new ExtensionInstallEvent(
|
||||||
|
newExtensionConfig!.name,
|
||||||
|
newExtensionConfig!.version,
|
||||||
|
installMetadata.source,
|
||||||
|
'success',
|
||||||
|
),
|
||||||
|
);
|
||||||
|
|
||||||
|
enableExtension(newExtensionConfig!.name, SettingScope.User);
|
||||||
|
return newExtensionConfig!.name;
|
||||||
|
} catch (error) {
|
||||||
|
// Attempt to load config from the source path even if installation fails
|
||||||
|
// to get the name and version for logging.
|
||||||
|
if (!newExtensionConfig && localSourcePath) {
|
||||||
|
try {
|
||||||
|
newExtensionConfig = loadExtensionConfig({
|
||||||
|
extensionDir: localSourcePath,
|
||||||
|
workspaceDir: cwd,
|
||||||
|
});
|
||||||
|
} catch {
|
||||||
|
// Ignore error, this is just for logging.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
logExtensionInstallEvent(
|
||||||
|
telemetryConfig,
|
||||||
|
new ExtensionInstallEvent(
|
||||||
|
newExtensionConfig?.name ?? '',
|
||||||
|
newExtensionConfig?.version ?? '',
|
||||||
|
installMetadata.source,
|
||||||
|
'error',
|
||||||
|
),
|
||||||
|
);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Builds a consent string for installing an extension based on it's
|
||||||
|
* extensionConfig.
|
||||||
|
*/
|
||||||
|
function extensionConsentString(extensionConfig: ExtensionConfig): string {
|
||||||
|
const output: string[] = [];
|
||||||
|
const mcpServerEntries = Object.entries(extensionConfig.mcpServers || {});
|
||||||
|
output.push(`Installing extension "${extensionConfig.name}".`);
|
||||||
|
output.push(
|
||||||
|
'**Extensions may introduce unexpected behavior. Ensure you have investigated the extension source and trust the author.**',
|
||||||
|
);
|
||||||
|
|
||||||
|
if (mcpServerEntries.length) {
|
||||||
|
output.push('This extension will run the following MCP servers:');
|
||||||
|
for (const [key, mcpServer] of mcpServerEntries) {
|
||||||
|
const isLocal = !!mcpServer.command;
|
||||||
|
const source =
|
||||||
|
mcpServer.httpUrl ??
|
||||||
|
`${mcpServer.command || ''}${mcpServer.args ? ' ' + mcpServer.args.join(' ') : ''}`;
|
||||||
|
output.push(` * ${key} (${isLocal ? 'local' : 'remote'}): ${source}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (extensionConfig.contextFileName) {
|
||||||
|
output.push(
|
||||||
|
`This extension will append info to your gemini.md context using ${extensionConfig.contextFileName}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if (extensionConfig.excludeTools) {
|
||||||
|
output.push(
|
||||||
|
`This extension will exclude the following core tools: ${extensionConfig.excludeTools}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return output.join('\n');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Requests consent from the user to install an extension (extensionConfig), if
|
||||||
|
* there is any difference between the consent string for `extensionConfig` and
|
||||||
|
* `previousExtensionConfig`.
|
||||||
|
*
|
||||||
|
* Always requests consent if previousExtensionConfig is null.
|
||||||
|
*
|
||||||
|
* Throws if the user does not consent.
|
||||||
|
*/
|
||||||
|
async function maybeRequestConsentOrFail(
|
||||||
|
extensionConfig: ExtensionConfig,
|
||||||
|
requestConsent: (consent: string) => Promise<boolean>,
|
||||||
|
previousExtensionConfig?: ExtensionConfig,
|
||||||
|
) {
|
||||||
|
const extensionConsent = extensionConsentString(extensionConfig);
|
||||||
|
if (previousExtensionConfig) {
|
||||||
|
const previousExtensionConsent = extensionConsentString(
|
||||||
|
previousExtensionConfig,
|
||||||
|
);
|
||||||
|
if (previousExtensionConsent === extensionConsent) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!(await requestConsent(extensionConsent))) {
|
||||||
|
throw new Error(`Installation cancelled for "${extensionConfig.name}".`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function validateName(name: string) {
|
||||||
|
if (!/^[a-zA-Z0-9-]+$/.test(name)) {
|
||||||
|
throw new Error(
|
||||||
|
`Invalid extension name: "${name}". Only letters (a-z, A-Z), numbers (0-9), and dashes (-) are allowed.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function loadExtensionConfig(
|
||||||
|
context: LoadExtensionContext,
|
||||||
|
): ExtensionConfig {
|
||||||
|
const { extensionDir, workspaceDir } = context;
|
||||||
|
const configFilePath = path.join(extensionDir, EXTENSIONS_CONFIG_FILENAME);
|
||||||
|
if (!fs.existsSync(configFilePath)) {
|
||||||
|
throw new Error(`Configuration file not found at ${configFilePath}`);
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const configContent = fs.readFileSync(configFilePath, 'utf-8');
|
||||||
|
const config = recursivelyHydrateStrings(JSON.parse(configContent), {
|
||||||
|
extensionPath: extensionDir,
|
||||||
|
workspacePath: workspaceDir,
|
||||||
|
'/': path.sep,
|
||||||
|
pathSeparator: path.sep,
|
||||||
|
}) as unknown as ExtensionConfig;
|
||||||
|
if (!config.name || !config.version) {
|
||||||
|
throw new Error(
|
||||||
|
`Invalid configuration in ${configFilePath}: missing ${!config.name ? '"name"' : '"version"'}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
validateName(config.name);
|
||||||
|
return config;
|
||||||
|
} catch (e) {
|
||||||
|
throw new Error(
|
||||||
|
`Failed to load extension config from ${configFilePath}: ${getErrorMessage(
|
||||||
|
e,
|
||||||
|
)}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function uninstallExtension(
|
export async function uninstallExtension(
|
||||||
extensionName: string,
|
extensionIdentifier: string,
|
||||||
cwd: string = process.cwd(),
|
cwd: string = process.cwd(),
|
||||||
): Promise<void> {
|
): Promise<void> {
|
||||||
|
const telemetryConfig = getTelemetryConfig(cwd);
|
||||||
const installedExtensions = loadUserExtensions();
|
const installedExtensions = loadUserExtensions();
|
||||||
if (
|
const extensionName = installedExtensions.find(
|
||||||
!installedExtensions.some(
|
(installed) =>
|
||||||
(installed) => installed.config.name === extensionName,
|
installed.config.name.toLowerCase() ===
|
||||||
)
|
extensionIdentifier.toLowerCase() ||
|
||||||
) {
|
installed.installMetadata?.source.toLowerCase() ===
|
||||||
throw new Error(`Extension "${extensionName}" not found.`);
|
extensionIdentifier.toLowerCase(),
|
||||||
|
)?.config.name;
|
||||||
|
if (!extensionName) {
|
||||||
|
throw new Error(`Extension not found.`);
|
||||||
}
|
}
|
||||||
removeFromDisabledExtensions(
|
const manager = new ExtensionEnablementManager(
|
||||||
extensionName,
|
ExtensionStorage.getUserExtensionsDir(),
|
||||||
[SettingScope.User, SettingScope.Workspace],
|
[extensionName],
|
||||||
cwd,
|
|
||||||
);
|
);
|
||||||
|
manager.remove(extensionName);
|
||||||
const storage = new ExtensionStorage(extensionName);
|
const storage = new ExtensionStorage(extensionName);
|
||||||
return await fs.promises.rm(storage.getExtensionDir(), {
|
|
||||||
|
await fs.promises.rm(storage.getExtensionDir(), {
|
||||||
recursive: true,
|
recursive: true,
|
||||||
force: true,
|
force: true,
|
||||||
});
|
});
|
||||||
|
logExtensionUninstall(
|
||||||
|
telemetryConfig,
|
||||||
|
new ExtensionUninstallEvent(extensionName, 'success'),
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
export function toOutputString(extension: Extension): string {
|
export function toOutputString(
|
||||||
let output = `${extension.config.name} (${extension.config.version})`;
|
extension: Extension,
|
||||||
|
workspaceDir: string,
|
||||||
|
): string {
|
||||||
|
const manager = new ExtensionEnablementManager(
|
||||||
|
ExtensionStorage.getUserExtensionsDir(),
|
||||||
|
);
|
||||||
|
const userEnabled = manager.isEnabled(extension.config.name, os.homedir());
|
||||||
|
const workspaceEnabled = manager.isEnabled(
|
||||||
|
extension.config.name,
|
||||||
|
workspaceDir,
|
||||||
|
);
|
||||||
|
|
||||||
|
const status = workspaceEnabled ? chalk.green('✓') : chalk.red('✗');
|
||||||
|
let output = `${status} ${extension.config.name} (${extension.config.version})`;
|
||||||
output += `\n Path: ${extension.path}`;
|
output += `\n Path: ${extension.path}`;
|
||||||
if (extension.installMetadata) {
|
if (extension.installMetadata) {
|
||||||
output += `\n Source: ${extension.installMetadata.source}`;
|
output += `\n Source: ${extension.installMetadata.source} (Type: ${extension.installMetadata.type})`;
|
||||||
|
if (extension.installMetadata.ref) {
|
||||||
|
output += `\n Ref: ${extension.installMetadata.ref}`;
|
||||||
}
|
}
|
||||||
|
if (extension.installMetadata.releaseTag) {
|
||||||
|
output += `\n Release tag: ${extension.installMetadata.releaseTag}`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
output += `\n Enabled (User): ${userEnabled}`;
|
||||||
|
output += `\n Enabled (Workspace): ${workspaceEnabled}`;
|
||||||
if (extension.contextFiles.length > 0) {
|
if (extension.contextFiles.length > 0) {
|
||||||
output += `\n Context files:`;
|
output += `\n Context files:`;
|
||||||
extension.contextFiles.forEach((contextFile) => {
|
extension.contextFiles.forEach((contextFile) => {
|
||||||
@@ -438,96 +743,46 @@ export function toOutputString(extension: Extension): string {
|
|||||||
return output;
|
return output;
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function updateExtension(
|
|
||||||
extensionName: string,
|
|
||||||
cwd: string = process.cwd(),
|
|
||||||
): Promise<ExtensionUpdateInfo | undefined> {
|
|
||||||
const installedExtensions = loadUserExtensions();
|
|
||||||
const extension = installedExtensions.find(
|
|
||||||
(installed) => installed.config.name === extensionName,
|
|
||||||
);
|
|
||||||
if (!extension) {
|
|
||||||
throw new Error(
|
|
||||||
`Extension "${extensionName}" not found. Run gemini extensions list to see available extensions.`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
if (!extension.installMetadata) {
|
|
||||||
throw new Error(
|
|
||||||
`Extension cannot be updated because it is missing the .qwen-extension.install.json file. To update manually, uninstall and then reinstall the updated version.`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
const originalVersion = extension.config.version;
|
|
||||||
const tempDir = await ExtensionStorage.createTmpDir();
|
|
||||||
try {
|
|
||||||
await copyExtension(extension.path, tempDir);
|
|
||||||
await uninstallExtension(extensionName, cwd);
|
|
||||||
await installExtension(extension.installMetadata, cwd);
|
|
||||||
|
|
||||||
const updatedExtension = loadExtension(extension.path);
|
|
||||||
if (!updatedExtension) {
|
|
||||||
throw new Error('Updated extension not found after installation.');
|
|
||||||
}
|
|
||||||
const updatedVersion = updatedExtension.config.version;
|
|
||||||
return {
|
|
||||||
originalVersion,
|
|
||||||
updatedVersion,
|
|
||||||
};
|
|
||||||
} catch (e) {
|
|
||||||
console.error(
|
|
||||||
`Error updating extension, rolling back. ${getErrorMessage(e)}`,
|
|
||||||
);
|
|
||||||
await copyExtension(tempDir, extension.path);
|
|
||||||
throw e;
|
|
||||||
} finally {
|
|
||||||
await fs.promises.rm(tempDir, { recursive: true, force: true });
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export function disableExtension(
|
export function disableExtension(
|
||||||
name: string,
|
name: string,
|
||||||
scope: SettingScope,
|
scope: SettingScope,
|
||||||
cwd: string = process.cwd(),
|
cwd: string = process.cwd(),
|
||||||
|
) {
|
||||||
|
const config = getTelemetryConfig(cwd);
|
||||||
|
if (scope === SettingScope.System || scope === SettingScope.SystemDefaults) {
|
||||||
|
throw new Error('System and SystemDefaults scopes are not supported.');
|
||||||
|
}
|
||||||
|
const extension = loadExtensionByName(name, cwd);
|
||||||
|
if (!extension) {
|
||||||
|
throw new Error(`Extension with name ${name} does not exist.`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const manager = new ExtensionEnablementManager(
|
||||||
|
ExtensionStorage.getUserExtensionsDir(),
|
||||||
|
[name],
|
||||||
|
);
|
||||||
|
const scopePath = scope === SettingScope.Workspace ? cwd : os.homedir();
|
||||||
|
manager.disable(name, true, scopePath);
|
||||||
|
logExtensionDisable(config, new ExtensionDisableEvent(name, scope));
|
||||||
|
}
|
||||||
|
|
||||||
|
export function enableExtension(
|
||||||
|
name: string,
|
||||||
|
scope: SettingScope,
|
||||||
|
cwd: string = process.cwd(),
|
||||||
) {
|
) {
|
||||||
if (scope === SettingScope.System || scope === SettingScope.SystemDefaults) {
|
if (scope === SettingScope.System || scope === SettingScope.SystemDefaults) {
|
||||||
throw new Error('System and SystemDefaults scopes are not supported.');
|
throw new Error('System and SystemDefaults scopes are not supported.');
|
||||||
}
|
}
|
||||||
const settings = loadSettings(cwd);
|
const extension = loadExtensionByName(name, cwd);
|
||||||
const settingsFile = settings.forScope(scope);
|
if (!extension) {
|
||||||
const extensionSettings = settingsFile.settings.extensions || {
|
throw new Error(`Extension with name ${name} does not exist.`);
|
||||||
disabled: [],
|
|
||||||
};
|
|
||||||
const disabledExtensions = extensionSettings.disabled || [];
|
|
||||||
if (!disabledExtensions.includes(name)) {
|
|
||||||
disabledExtensions.push(name);
|
|
||||||
extensionSettings.disabled = disabledExtensions;
|
|
||||||
settings.setValue(scope, 'extensions', extensionSettings);
|
|
||||||
}
|
}
|
||||||
}
|
const manager = new ExtensionEnablementManager(
|
||||||
|
ExtensionStorage.getUserExtensionsDir(),
|
||||||
export function enableExtension(name: string, scopes: SettingScope[]) {
|
|
||||||
removeFromDisabledExtensions(name, scopes);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Removes an extension from the list of disabled extensions.
|
|
||||||
* @param name The name of the extension to remove.
|
|
||||||
* @param scope The scopes to remove the name from.
|
|
||||||
*/
|
|
||||||
function removeFromDisabledExtensions(
|
|
||||||
name: string,
|
|
||||||
scopes: SettingScope[],
|
|
||||||
cwd: string = process.cwd(),
|
|
||||||
) {
|
|
||||||
const settings = loadSettings(cwd);
|
|
||||||
for (const scope of scopes) {
|
|
||||||
const settingsFile = settings.forScope(scope);
|
|
||||||
const extensionSettings = settingsFile.settings.extensions || {
|
|
||||||
disabled: [],
|
|
||||||
};
|
|
||||||
const disabledExtensions = extensionSettings.disabled || [];
|
|
||||||
extensionSettings.disabled = disabledExtensions.filter(
|
|
||||||
(extension) => extension !== name,
|
|
||||||
);
|
);
|
||||||
settings.setValue(scope, 'extensions', extensionSettings);
|
const scopePath = scope === SettingScope.Workspace ? cwd : os.homedir();
|
||||||
}
|
manager.enable(name, true, scopePath);
|
||||||
|
const config = getTelemetryConfig(cwd);
|
||||||
|
logExtensionEnable(config, new ExtensionEnableEvent(name, scope));
|
||||||
}
|
}
|
||||||
|
|||||||
424
packages/cli/src/config/extensions/extensionEnablement.test.ts
Normal file
424
packages/cli/src/config/extensions/extensionEnablement.test.ts
Normal file
@@ -0,0 +1,424 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import * as path from 'node:path';
|
||||||
|
import fs from 'node:fs';
|
||||||
|
import os from 'node:os';
|
||||||
|
import { afterEach, beforeEach, describe, expect, it } from 'vitest';
|
||||||
|
import { ExtensionEnablementManager, Override } from './extensionEnablement.js';
|
||||||
|
import type { Extension } from '../extension.js';
|
||||||
|
|
||||||
|
// Helper to create a temporary directory for testing
|
||||||
|
function createTestDir() {
|
||||||
|
const dirPath = fs.mkdtempSync(path.join(os.tmpdir(), 'gemini-test-'));
|
||||||
|
return {
|
||||||
|
path: dirPath,
|
||||||
|
cleanup: () => fs.rmSync(dirPath, { recursive: true, force: true }),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let testDir: { path: string; cleanup: () => void };
|
||||||
|
let configDir: string;
|
||||||
|
let manager: ExtensionEnablementManager;
|
||||||
|
|
||||||
|
describe('ExtensionEnablementManager', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
testDir = createTestDir();
|
||||||
|
configDir = path.join(testDir.path, '.gemini');
|
||||||
|
manager = new ExtensionEnablementManager(configDir);
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
testDir.cleanup();
|
||||||
|
// Reset the singleton instance for test isolation
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
||||||
|
(ExtensionEnablementManager as any).instance = undefined;
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('isEnabled', () => {
|
||||||
|
it('should return true if extension is not configured', () => {
|
||||||
|
expect(manager.isEnabled('ext-test', '/any/path')).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return true if no overrides match', () => {
|
||||||
|
manager.disable('ext-test', false, '/another/path');
|
||||||
|
expect(manager.isEnabled('ext-test', '/any/path')).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should enable a path based on an override rule', () => {
|
||||||
|
manager.disable('ext-test', true, '/');
|
||||||
|
manager.enable('ext-test', true, '/home/user/projects/');
|
||||||
|
expect(manager.isEnabled('ext-test', '/home/user/projects/my-app')).toBe(
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should disable a path based on a disable override rule', () => {
|
||||||
|
manager.enable('ext-test', true, '/');
|
||||||
|
manager.disable('ext-test', true, '/home/user/projects/');
|
||||||
|
expect(manager.isEnabled('ext-test', '/home/user/projects/my-app')).toBe(
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should respect the last matching rule (enable wins)', () => {
|
||||||
|
manager.disable('ext-test', true, '/home/user/projects/');
|
||||||
|
manager.enable('ext-test', false, '/home/user/projects/my-app');
|
||||||
|
expect(manager.isEnabled('ext-test', '/home/user/projects/my-app')).toBe(
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should respect the last matching rule (disable wins)', () => {
|
||||||
|
manager.enable('ext-test', true, '/home/user/projects/');
|
||||||
|
manager.disable('ext-test', false, '/home/user/projects/my-app');
|
||||||
|
expect(manager.isEnabled('ext-test', '/home/user/projects/my-app')).toBe(
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should handle', () => {
|
||||||
|
manager.enable('ext-test', true, '/home/user/projects');
|
||||||
|
manager.disable('ext-test', false, '/home/user/projects/my-app');
|
||||||
|
expect(manager.isEnabled('ext-test', '/home/user/projects/my-app')).toBe(
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
expect(
|
||||||
|
manager.isEnabled('ext-test', '/home/user/projects/something-else'),
|
||||||
|
).toBe(true);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('includeSubdirs', () => {
|
||||||
|
it('should add a glob when enabling with includeSubdirs', () => {
|
||||||
|
manager.enable('ext-test', true, '/path/to/dir');
|
||||||
|
const config = manager.readConfig();
|
||||||
|
expect(config['ext-test'].overrides).toContain('/path/to/dir/*');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not add a glob when enabling without includeSubdirs', () => {
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir');
|
||||||
|
const config = manager.readConfig();
|
||||||
|
expect(config['ext-test'].overrides).toContain('/path/to/dir/');
|
||||||
|
expect(config['ext-test'].overrides).not.toContain('/path/to/dir/*');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should add a glob when disabling with includeSubdirs', () => {
|
||||||
|
manager.disable('ext-test', true, '/path/to/dir');
|
||||||
|
const config = manager.readConfig();
|
||||||
|
expect(config['ext-test'].overrides).toContain('!/path/to/dir/*');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should remove conflicting glob rule when enabling without subdirs', () => {
|
||||||
|
manager.enable('ext-test', true, '/path/to/dir'); // Adds /path/to/dir*
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir'); // Should remove the glob
|
||||||
|
const config = manager.readConfig();
|
||||||
|
expect(config['ext-test'].overrides).toContain('/path/to/dir/');
|
||||||
|
expect(config['ext-test'].overrides).not.toContain('/path/to/dir/*');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should remove conflicting non-glob rule when enabling with subdirs', () => {
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir'); // Adds /path/to/dir
|
||||||
|
manager.enable('ext-test', true, '/path/to/dir'); // Should remove the non-glob
|
||||||
|
const config = manager.readConfig();
|
||||||
|
expect(config['ext-test'].overrides).toContain('/path/to/dir/*');
|
||||||
|
expect(config['ext-test'].overrides).not.toContain('/path/to/dir/');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should remove conflicting rules when disabling', () => {
|
||||||
|
manager.enable('ext-test', true, '/path/to/dir'); // enabled with glob
|
||||||
|
manager.disable('ext-test', false, '/path/to/dir'); // disabled without
|
||||||
|
const config = manager.readConfig();
|
||||||
|
expect(config['ext-test'].overrides).toContain('!/path/to/dir/');
|
||||||
|
expect(config['ext-test'].overrides).not.toContain('/path/to/dir/*');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly evaluate isEnabled with subdirs', () => {
|
||||||
|
manager.disable('ext-test', true, '/');
|
||||||
|
manager.enable('ext-test', true, '/path/to/dir');
|
||||||
|
expect(manager.isEnabled('ext-test', '/path/to/dir/')).toBe(true);
|
||||||
|
expect(manager.isEnabled('ext-test', '/path/to/dir/sub/')).toBe(true);
|
||||||
|
expect(manager.isEnabled('ext-test', '/path/to/another/')).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly evaluate isEnabled without subdirs', () => {
|
||||||
|
manager.disable('ext-test', true, '/*');
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir');
|
||||||
|
expect(manager.isEnabled('ext-test', '/path/to/dir')).toBe(true);
|
||||||
|
expect(manager.isEnabled('ext-test', '/path/to/dir/sub')).toBe(false);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('pruning child rules', () => {
|
||||||
|
it('should remove child rules when enabling a parent with subdirs', () => {
|
||||||
|
// Pre-existing rules for children
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir/subdir1');
|
||||||
|
manager.disable('ext-test', true, '/path/to/dir/subdir2');
|
||||||
|
manager.enable('ext-test', false, '/path/to/another/dir');
|
||||||
|
|
||||||
|
// Enable the parent directory
|
||||||
|
manager.enable('ext-test', true, '/path/to/dir');
|
||||||
|
|
||||||
|
const config = manager.readConfig();
|
||||||
|
const overrides = config['ext-test'].overrides;
|
||||||
|
|
||||||
|
// The new parent rule should be present
|
||||||
|
expect(overrides).toContain(`/path/to/dir/*`);
|
||||||
|
|
||||||
|
// Child rules should be removed
|
||||||
|
expect(overrides).not.toContain('/path/to/dir/subdir1/');
|
||||||
|
expect(overrides).not.toContain(`!/path/to/dir/subdir2/*`);
|
||||||
|
|
||||||
|
// Unrelated rules should remain
|
||||||
|
expect(overrides).toContain('/path/to/another/dir/');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should remove child rules when disabling a parent with subdirs', () => {
|
||||||
|
// Pre-existing rules for children
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir/subdir1');
|
||||||
|
manager.disable('ext-test', true, '/path/to/dir/subdir2');
|
||||||
|
manager.enable('ext-test', false, '/path/to/another/dir');
|
||||||
|
|
||||||
|
// Disable the parent directory
|
||||||
|
manager.disable('ext-test', true, '/path/to/dir');
|
||||||
|
|
||||||
|
const config = manager.readConfig();
|
||||||
|
const overrides = config['ext-test'].overrides;
|
||||||
|
|
||||||
|
// The new parent rule should be present
|
||||||
|
expect(overrides).toContain(`!/path/to/dir/*`);
|
||||||
|
|
||||||
|
// Child rules should be removed
|
||||||
|
expect(overrides).not.toContain('/path/to/dir/subdir1/');
|
||||||
|
expect(overrides).not.toContain(`!/path/to/dir/subdir2/*`);
|
||||||
|
|
||||||
|
// Unrelated rules should remain
|
||||||
|
expect(overrides).toContain('/path/to/another/dir/');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not remove child rules if includeSubdirs is false', () => {
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir/subdir1');
|
||||||
|
manager.enable('ext-test', false, '/path/to/dir'); // Not including subdirs
|
||||||
|
|
||||||
|
const config = manager.readConfig();
|
||||||
|
const overrides = config['ext-test'].overrides;
|
||||||
|
|
||||||
|
expect(overrides).toContain('/path/to/dir/subdir1/');
|
||||||
|
expect(overrides).toContain('/path/to/dir/');
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should enable a path based on an enable override', () => {
|
||||||
|
manager.disable('ext-test', true, '/Users/chrstn');
|
||||||
|
manager.enable('ext-test', true, '/Users/chrstn/gemini-cli');
|
||||||
|
|
||||||
|
expect(manager.isEnabled('ext-test', '/Users/chrstn/gemini-cli')).toBe(
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should ignore subdirs', () => {
|
||||||
|
manager.disable('ext-test', false, '/Users/chrstn');
|
||||||
|
expect(manager.isEnabled('ext-test', '/Users/chrstn/gemini-cli')).toBe(
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('extension overrides (-e <name>)', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
manager = new ExtensionEnablementManager(configDir, ['ext-test']);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('can enable extensions, case-insensitive', () => {
|
||||||
|
manager.disable('ext-test', true, '/');
|
||||||
|
expect(manager.isEnabled('ext-test', '/')).toBe(true);
|
||||||
|
expect(manager.isEnabled('Ext-Test', '/')).toBe(true);
|
||||||
|
// Double check that it would have been disabled otherwise
|
||||||
|
expect(
|
||||||
|
new ExtensionEnablementManager(configDir).isEnabled('ext-test', '/'),
|
||||||
|
).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('disable all other extensions', () => {
|
||||||
|
manager = new ExtensionEnablementManager(configDir, ['ext-test']);
|
||||||
|
manager.enable('ext-test-2', true, '/');
|
||||||
|
expect(manager.isEnabled('ext-test-2', '/')).toBe(false);
|
||||||
|
// Double check that it would have been enabled otherwise
|
||||||
|
expect(
|
||||||
|
new ExtensionEnablementManager(configDir).isEnabled('ext-test-2', '/'),
|
||||||
|
).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('none disables all extensions', () => {
|
||||||
|
manager = new ExtensionEnablementManager(configDir, ['none']);
|
||||||
|
manager.enable('ext-test', true, '/');
|
||||||
|
expect(manager.isEnabled('ext-test', '/path/to/dir')).toBe(false);
|
||||||
|
// Double check that it would have been enabled otherwise
|
||||||
|
expect(
|
||||||
|
new ExtensionEnablementManager(configDir).isEnabled('ext-test', '/'),
|
||||||
|
).toBe(true);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('validateExtensionOverrides', () => {
|
||||||
|
let consoleErrorSpy: ReturnType<typeof vi.spyOn>;
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
consoleErrorSpy.mockRestore();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not log an error if enabledExtensionNamesOverride is empty', () => {
|
||||||
|
const manager = new ExtensionEnablementManager(configDir, []);
|
||||||
|
manager.validateExtensionOverrides([]);
|
||||||
|
expect(consoleErrorSpy).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not log an error if all enabledExtensionNamesOverride are valid', () => {
|
||||||
|
const manager = new ExtensionEnablementManager(configDir, [
|
||||||
|
'ext-one',
|
||||||
|
'ext-two',
|
||||||
|
]);
|
||||||
|
const extensions = [
|
||||||
|
{ config: { name: 'ext-one' } },
|
||||||
|
{ config: { name: 'ext-two' } },
|
||||||
|
] as Extension[];
|
||||||
|
manager.validateExtensionOverrides(extensions);
|
||||||
|
expect(consoleErrorSpy).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should log an error for each invalid extension name in enabledExtensionNamesOverride', () => {
|
||||||
|
const manager = new ExtensionEnablementManager(configDir, [
|
||||||
|
'ext-one',
|
||||||
|
'ext-invalid',
|
||||||
|
'ext-another-invalid',
|
||||||
|
]);
|
||||||
|
const extensions = [
|
||||||
|
{ config: { name: 'ext-one' } },
|
||||||
|
{ config: { name: 'ext-two' } },
|
||||||
|
] as Extension[];
|
||||||
|
manager.validateExtensionOverrides(extensions);
|
||||||
|
expect(consoleErrorSpy).toHaveBeenCalledTimes(2);
|
||||||
|
expect(consoleErrorSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension not found: ext-invalid',
|
||||||
|
);
|
||||||
|
expect(consoleErrorSpy).toHaveBeenCalledWith(
|
||||||
|
'Extension not found: ext-another-invalid',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not log an error if "none" is in enabledExtensionNamesOverride', () => {
|
||||||
|
const manager = new ExtensionEnablementManager(configDir, ['none']);
|
||||||
|
manager.validateExtensionOverrides([]);
|
||||||
|
expect(consoleErrorSpy).not.toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('Override', () => {
|
||||||
|
it('should create an override from input', () => {
|
||||||
|
const override = Override.fromInput('/path/to/dir', true);
|
||||||
|
expect(override.baseRule).toBe(`/path/to/dir/`);
|
||||||
|
expect(override.isDisable).toBe(false);
|
||||||
|
expect(override.includeSubdirs).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create a disable override from input', () => {
|
||||||
|
const override = Override.fromInput('!/path/to/dir', false);
|
||||||
|
expect(override.baseRule).toBe(`/path/to/dir/`);
|
||||||
|
expect(override.isDisable).toBe(true);
|
||||||
|
expect(override.includeSubdirs).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create an override from a file rule', () => {
|
||||||
|
const override = Override.fromFileRule('/path/to/dir');
|
||||||
|
expect(override.baseRule).toBe('/path/to/dir');
|
||||||
|
expect(override.isDisable).toBe(false);
|
||||||
|
expect(override.includeSubdirs).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create a disable override from a file rule', () => {
|
||||||
|
const override = Override.fromFileRule('!/path/to/dir/');
|
||||||
|
expect(override.isDisable).toBe(true);
|
||||||
|
expect(override.baseRule).toBe('/path/to/dir/');
|
||||||
|
expect(override.includeSubdirs).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create an override with subdirs from a file rule', () => {
|
||||||
|
const override = Override.fromFileRule('/path/to/dir/*');
|
||||||
|
expect(override.baseRule).toBe('/path/to/dir/');
|
||||||
|
expect(override.isDisable).toBe(false);
|
||||||
|
expect(override.includeSubdirs).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify conflicting overrides', () => {
|
||||||
|
const override1 = Override.fromInput('/path/to/dir', true);
|
||||||
|
const override2 = Override.fromInput('/path/to/dir', false);
|
||||||
|
expect(override1.conflictsWith(override2)).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify non-conflicting overrides', () => {
|
||||||
|
const override1 = Override.fromInput('/path/to/dir', true);
|
||||||
|
const override2 = Override.fromInput('/path/to/another/dir', true);
|
||||||
|
expect(override1.conflictsWith(override2)).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify equal overrides', () => {
|
||||||
|
const override1 = Override.fromInput('/path/to/dir', true);
|
||||||
|
const override2 = Override.fromInput('/path/to/dir', true);
|
||||||
|
expect(override1.isEqualTo(override2)).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify unequal overrides', () => {
|
||||||
|
const override1 = Override.fromInput('/path/to/dir', true);
|
||||||
|
const override2 = Override.fromInput('!/path/to/dir', true);
|
||||||
|
expect(override1.isEqualTo(override2)).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should generate the correct regex', () => {
|
||||||
|
const override = Override.fromInput('/path/to/dir', true);
|
||||||
|
const regex = override.asRegex();
|
||||||
|
expect(regex.test('/path/to/dir/')).toBe(true);
|
||||||
|
expect(regex.test('/path/to/dir/subdir')).toBe(true);
|
||||||
|
expect(regex.test('/path/to/another/dir')).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify child overrides', () => {
|
||||||
|
const parent = Override.fromInput('/path/to/dir', true);
|
||||||
|
const child = Override.fromInput('/path/to/dir/subdir', false);
|
||||||
|
expect(child.isChildOf(parent)).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify child overrides with glob', () => {
|
||||||
|
const parent = Override.fromInput('/path/to/dir/*', true);
|
||||||
|
const child = Override.fromInput('/path/to/dir/subdir', false);
|
||||||
|
expect(child.isChildOf(parent)).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should correctly identify non-child overrides', () => {
|
||||||
|
const parent = Override.fromInput('/path/to/dir', true);
|
||||||
|
const other = Override.fromInput('/path/to/another/dir', false);
|
||||||
|
expect(other.isChildOf(parent)).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should generate the correct output string', () => {
|
||||||
|
const override = Override.fromInput('/path/to/dir', true);
|
||||||
|
expect(override.output()).toBe(`/path/to/dir/*`);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should generate the correct output string for a disable override', () => {
|
||||||
|
const override = Override.fromInput('!/path/to/dir', false);
|
||||||
|
expect(override.output()).toBe(`!/path/to/dir/`);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should disable a path based on a disable override rule', () => {
|
||||||
|
const override = Override.fromInput('!/path/to/dir', false);
|
||||||
|
expect(override.output()).toBe(`!/path/to/dir/`);
|
||||||
|
});
|
||||||
|
});
|
||||||
239
packages/cli/src/config/extensions/extensionEnablement.ts
Normal file
239
packages/cli/src/config/extensions/extensionEnablement.ts
Normal file
@@ -0,0 +1,239 @@
|
|||||||
|
/**
|
||||||
|
* @license
|
||||||
|
* Copyright 2025 Google LLC
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
import fs from 'node:fs';
|
||||||
|
import path from 'node:path';
|
||||||
|
import { type Extension } from '../extension.js';
|
||||||
|
|
||||||
|
export interface ExtensionEnablementConfig {
|
||||||
|
overrides: string[];
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface AllExtensionsEnablementConfig {
|
||||||
|
[extensionName: string]: ExtensionEnablementConfig;
|
||||||
|
}
|
||||||
|
|
||||||
|
export class Override {
|
||||||
|
constructor(
|
||||||
|
public baseRule: string,
|
||||||
|
public isDisable: boolean,
|
||||||
|
public includeSubdirs: boolean,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
static fromInput(inputRule: string, includeSubdirs: boolean): Override {
|
||||||
|
const isDisable = inputRule.startsWith('!');
|
||||||
|
let baseRule = isDisable ? inputRule.substring(1) : inputRule;
|
||||||
|
baseRule = ensureLeadingAndTrailingSlash(baseRule);
|
||||||
|
return new Override(baseRule, isDisable, includeSubdirs);
|
||||||
|
}
|
||||||
|
|
||||||
|
static fromFileRule(fileRule: string): Override {
|
||||||
|
const isDisable = fileRule.startsWith('!');
|
||||||
|
let baseRule = isDisable ? fileRule.substring(1) : fileRule;
|
||||||
|
const includeSubdirs = baseRule.endsWith('*');
|
||||||
|
baseRule = includeSubdirs
|
||||||
|
? baseRule.substring(0, baseRule.length - 1)
|
||||||
|
: baseRule;
|
||||||
|
return new Override(baseRule, isDisable, includeSubdirs);
|
||||||
|
}
|
||||||
|
|
||||||
|
conflictsWith(other: Override): boolean {
|
||||||
|
if (this.baseRule === other.baseRule) {
|
||||||
|
return (
|
||||||
|
this.includeSubdirs !== other.includeSubdirs ||
|
||||||
|
this.isDisable !== other.isDisable
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
isEqualTo(other: Override): boolean {
|
||||||
|
return (
|
||||||
|
this.baseRule === other.baseRule &&
|
||||||
|
this.includeSubdirs === other.includeSubdirs &&
|
||||||
|
this.isDisable === other.isDisable
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
asRegex(): RegExp {
|
||||||
|
return globToRegex(`${this.baseRule}${this.includeSubdirs ? '*' : ''}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
isChildOf(parent: Override) {
|
||||||
|
if (!parent.includeSubdirs) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return parent.asRegex().test(this.baseRule);
|
||||||
|
}
|
||||||
|
|
||||||
|
output(): string {
|
||||||
|
return `${this.isDisable ? '!' : ''}${this.baseRule}${this.includeSubdirs ? '*' : ''}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
matchesPath(path: string) {
|
||||||
|
return this.asRegex().test(path);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const ensureLeadingAndTrailingSlash = function (dirPath: string): string {
|
||||||
|
// Normalize separators to forward slashes for consistent matching across platforms.
|
||||||
|
let result = dirPath.replace(/\\/g, '/');
|
||||||
|
if (result.charAt(0) !== '/') {
|
||||||
|
result = '/' + result;
|
||||||
|
}
|
||||||
|
if (result.charAt(result.length - 1) !== '/') {
|
||||||
|
result = result + '/';
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts a glob pattern to a RegExp object.
|
||||||
|
* This is a simplified implementation that supports `*`.
|
||||||
|
*
|
||||||
|
* @param glob The glob pattern to convert.
|
||||||
|
* @returns A RegExp object.
|
||||||
|
*/
|
||||||
|
function globToRegex(glob: string): RegExp {
|
||||||
|
const regexString = glob
|
||||||
|
.replace(/[.+?^${}()|[\]\\]/g, '\\$&') // Escape special regex characters
|
||||||
|
.replace(/(\/?)\*/g, '($1.*)?'); // Convert * to optional group
|
||||||
|
|
||||||
|
return new RegExp(`^${regexString}$`);
|
||||||
|
}
|
||||||
|
|
||||||
|
export class ExtensionEnablementManager {
|
||||||
|
private configFilePath: string;
|
||||||
|
private configDir: string;
|
||||||
|
// If non-empty, this overrides all other extension configuration and enables
|
||||||
|
// only the ones in this list.
|
||||||
|
private enabledExtensionNamesOverride: string[];
|
||||||
|
|
||||||
|
constructor(configDir: string, enabledExtensionNames?: string[]) {
|
||||||
|
this.configDir = configDir;
|
||||||
|
this.configFilePath = path.join(configDir, 'extension-enablement.json');
|
||||||
|
this.enabledExtensionNamesOverride =
|
||||||
|
enabledExtensionNames?.map((name) => name.toLowerCase()) ?? [];
|
||||||
|
}
|
||||||
|
|
||||||
|
validateExtensionOverrides(extensions: Extension[]) {
|
||||||
|
for (const name of this.enabledExtensionNamesOverride) {
|
||||||
|
if (name === 'none') continue;
|
||||||
|
if (
|
||||||
|
!extensions.some(
|
||||||
|
(ext) => ext.config.name.toLowerCase() === name.toLowerCase(),
|
||||||
|
)
|
||||||
|
) {
|
||||||
|
console.error(`Extension not found: ${name}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Determines if an extension is enabled based on its name and the current
|
||||||
|
* path. The last matching rule in the overrides list wins.
|
||||||
|
*
|
||||||
|
* @param extensionName The name of the extension.
|
||||||
|
* @param currentPath The absolute path of the current working directory.
|
||||||
|
* @returns True if the extension is enabled, false otherwise.
|
||||||
|
*/
|
||||||
|
isEnabled(extensionName: string, currentPath: string): boolean {
|
||||||
|
// If we have a single override called 'none', this disables all extensions.
|
||||||
|
// Typically, this comes from the user passing `-e none`.
|
||||||
|
if (
|
||||||
|
this.enabledExtensionNamesOverride.length === 1 &&
|
||||||
|
this.enabledExtensionNamesOverride[0] === 'none'
|
||||||
|
) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// If we have explicit overrides, only enable those extensions.
|
||||||
|
if (this.enabledExtensionNamesOverride.length > 0) {
|
||||||
|
// When checking against overrides ONLY, we use a case insensitive match.
|
||||||
|
// The override names are already lowercased in the constructor.
|
||||||
|
return this.enabledExtensionNamesOverride.includes(
|
||||||
|
extensionName.toLocaleLowerCase(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Otherwise, we use the configuration settings
|
||||||
|
const config = this.readConfig();
|
||||||
|
const extensionConfig = config[extensionName];
|
||||||
|
// Extensions are enabled by default.
|
||||||
|
let enabled = true;
|
||||||
|
const allOverrides = extensionConfig?.overrides ?? [];
|
||||||
|
for (const rule of allOverrides) {
|
||||||
|
const override = Override.fromFileRule(rule);
|
||||||
|
if (override.matchesPath(ensureLeadingAndTrailingSlash(currentPath))) {
|
||||||
|
enabled = !override.isDisable;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
readConfig(): AllExtensionsEnablementConfig {
|
||||||
|
try {
|
||||||
|
const content = fs.readFileSync(this.configFilePath, 'utf-8');
|
||||||
|
return JSON.parse(content);
|
||||||
|
} catch (error) {
|
||||||
|
if (
|
||||||
|
error instanceof Error &&
|
||||||
|
'code' in error &&
|
||||||
|
error.code === 'ENOENT'
|
||||||
|
) {
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
console.error('Error reading extension enablement config:', error);
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
writeConfig(config: AllExtensionsEnablementConfig): void {
|
||||||
|
fs.mkdirSync(this.configDir, { recursive: true });
|
||||||
|
fs.writeFileSync(this.configFilePath, JSON.stringify(config, null, 2));
|
||||||
|
}
|
||||||
|
|
||||||
|
enable(
|
||||||
|
extensionName: string,
|
||||||
|
includeSubdirs: boolean,
|
||||||
|
scopePath: string,
|
||||||
|
): void {
|
||||||
|
const config = this.readConfig();
|
||||||
|
if (!config[extensionName]) {
|
||||||
|
config[extensionName] = { overrides: [] };
|
||||||
|
}
|
||||||
|
const override = Override.fromInput(scopePath, includeSubdirs);
|
||||||
|
const overrides = config[extensionName].overrides.filter((rule) => {
|
||||||
|
const fileOverride = Override.fromFileRule(rule);
|
||||||
|
if (
|
||||||
|
fileOverride.conflictsWith(override) ||
|
||||||
|
fileOverride.isEqualTo(override)
|
||||||
|
) {
|
||||||
|
return false; // Remove conflicts and equivalent values.
|
||||||
|
}
|
||||||
|
return !fileOverride.isChildOf(override);
|
||||||
|
});
|
||||||
|
overrides.push(override.output());
|
||||||
|
config[extensionName].overrides = overrides;
|
||||||
|
this.writeConfig(config);
|
||||||
|
}
|
||||||
|
|
||||||
|
disable(
|
||||||
|
extensionName: string,
|
||||||
|
includeSubdirs: boolean,
|
||||||
|
scopePath: string,
|
||||||
|
): void {
|
||||||
|
this.enable(extensionName, includeSubdirs, `!${scopePath}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
remove(extensionName: string): void {
|
||||||
|
const config = this.readConfig();
|
||||||
|
if (config[extensionName]) {
|
||||||
|
delete config[extensionName];
|
||||||
|
this.writeConfig(config);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user