sampler-cli 0.4.3__tar.gz → 0.4.5__tar.gz

sampler_cli-0.4.5/PKG-INFO

@@ -0,0 +1,234 @@
+Metadata-Version: 2.4
+Name: sampler-cli
+Version: 0.4.5
+Summary: Token-efficient CLI for indexing and searching code symbols (Python-first, designed for minimal LLM/agent context size)
+Author: Samuel Ignacio Carmona Rodriguez
+License: MIT
+Project-URL: Homepage, https://github.com/SamuelCarmona83/sampler-cli
+Project-URL: Repository, https://github.com/SamuelCarmona83/sampler-cli
+Project-URL: Issues, https://github.com/SamuelCarmona83/sampler-cli/issues
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Software Development :: Code Generators
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: typer>=0.12.0
+Requires-Dist: rich>=13.7.0
+Requires-Dist: tree-sitter>=0.21.0
+Requires-Dist: tree-sitter-python>=0.23.0
+Requires-Dist: tree-sitter-go>=0.23.0
+Requires-Dist: tree-sitter-typescript>=0.23.0
+Requires-Dist: gitignore-parser>=0.1.11
+Requires-Dist: pydantic>=2.6.0
+Requires-Dist: pyyaml>=6.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
+Requires-Dist: ruff>=0.5.0; extra == "dev"
+Requires-Dist: mypy>=1.7.0; extra == "dev"
+Requires-Dist: numpy>=1.26.0; extra == "dev"
+Requires-Dist: scikit-learn>=1.4.0; extra == "dev"
+Requires-Dist: fastembed>=0.2.0; extra == "dev"
+Provides-Extra: mcp
+Requires-Dist: fastmcp>=0.1.0; extra == "mcp"
+Provides-Extra: semantic
+Requires-Dist: numpy>=1.26.0; extra == "semantic"
+Requires-Dist: scikit-learn>=1.4.0; extra == "semantic"
+Provides-Extra: embeddings
+Requires-Dist: fastembed>=0.2.0; extra == "embeddings"
+Requires-Dist: numpy>=1.26.0; extra == "embeddings"
+Provides-Extra: ollama-embeddings
+Requires-Dist: ollama>=0.3.0; extra == "ollama-embeddings"
+Provides-Extra: openai-embeddings
+Requires-Dist: openai>=1.0.0; extra == "openai-embeddings"
+Dynamic: license-file
+
+<p align="center">
+  <img src="./assets/sampler.png" alt="Sampler logo" width="220">
+</p>
+
+<h1 align="center">Sampler</h1>
+
+<p align="center">
+  <strong>Token-efficient CLI for indexing and searching code symbols across projects.</strong><br>
+  Compact output. Short paths. Low-noise symbol views.
+</p>
+
+<p align="center">
+  <a href="https://pypi.org/project/sampler-cli/"><img src="https://img.shields.io/pypi/v/sampler-cli" alt="PyPI version"></a>
+  <img src="https://img.shields.io/badge/python-3.11%2B-blue" alt="Python 3.11+">
+</p>
+
+---
+
+> *The code isn't the problem. The problem is the distance between you and the code.*
+>
+> Sampler closes that distance. One index. One query. The right symbol, the right relationship, the right context — delivered without the noise. Because in a world drowning in repositories, the person who finds what matters first is the person who moves the work forward.
+
+## Installation
+
+```bash
+pip install sampler-cli
+```
+
+Development:
+
+```bash
+pip install -e '.[dev]'
+```
+
+Semantic stack (TF-IDF + local hash fallback):
+
+```bash
+pip install -e '.[semantic]'
+```
+
+## Quick Start
+
+```bash
+sampler init
+sampler project add myproj /absolute/path/to/project --language auto
+sampler index myproj
+sampler search retry --project myproj
+sampler symbols myproj
+sampler overview src/main.py
+```
+
+## Commands
+
+### Core
+
+| Command | Description |
+| --- | --- |
+| `sampler version [--plain]` | Show version |
+| `sampler init` | Initialize Sampler config and data directory |
+| `sampler index <project>` | Index a project's symbols and relationships |
+| `sampler search <query>` | Search symbols across a project |
+| `sampler search-all <query>` | Search across all indexed projects |
+| `sampler symbols <project>` | List symbols in a project |
+| `sampler overview <filepath>` | Show symbol overview for a file |
+
+**Search options:** `--project`, `--type`, `--limit`, `--semantic`, `--style plain|bars`
+
+### Relationships
+
+| Command | Description |
+| --- | --- |
+| `sampler callers <symbol>` | Find callers of a symbol |
+| `sampler usages <symbol>` | Find usages of a symbol |
+| `sampler related <symbol>` | Find related symbols |
+
+Symbols can also be selected as `<path>:<symbol>`, e.g. `app/utils/helpers.py:format_kda`.
+
+### Project Management
+
+| Command | Description |
+| --- | --- |
+| `sampler project add <name> <path> --language <lang>` | Add a project |
+| `sampler project update <name>` | Update project path or language |
+| `sampler project list` | List projects |
+| `sampler project deps <name>` | Show project dependencies |
+| `sampler project remove <name>` | Remove a project |
+
+Languages: `python`, `go`, `typescript`, `javascript`, `vue`, `auto`.
+
+### Config & Analysis
+
+| Command | Description |
+| --- | --- |
+| `sampler config show` | Show current config |
+| `sampler config embeddings` | Configure embedding provider |
+| `sampler embed <project>` | Precompute embeddings |
+| `sampler stale-code <project>` | Find candidate stale code |
+
+## Embeddings & Semantic Search
+
+`sampler search --semantic` uses a pluggable adapter pattern:
+
+- **Default:** `bge-small` (BAAI/bge-small-en-v1.5 via fastembed — local ONNX, ~384 dim)
+- **Built-ins:** `hash` (deterministic fallback), `ollama`, `nomic`, `openai`, `fastembed`
+- **Lexical primary:** TF-IDF (sklearn, on-the-fly, no pre-embedding required)
+- **Final fallback:** hash fingerprint (always available)
+
+Configure in `~/.sampler/config.yaml` or via `sampler config embeddings`:
+
+```yaml
+embeddings:
+  provider: "bge-small"
+  # provider: "ollama"
+  # model: "nomic-embed-text"
+  # base_url: "http://localhost:11434"
+```
+
+Install extras:
+
+```bash
+pip install 'sampler-cli[embeddings]'        # BGE (recommended)
+pip install 'sampler-cli[ollama-embeddings]'
+pip install 'sampler-cli[openai-embeddings]'
+```
+
+Run `sampler embed <project>` to precompute vectors for the active provider. Change providers? Re-run `embed` after updating config.
+
+Offline or air-gapped: set `provider: hash`, or rely on TF-IDF + hash with the `[semantic]` extra.
+
+## Language Support
+
+| Language | Parser |
+| --- | --- |
+| Python | stdlib AST |
+| Go | tree-sitter-go |
+| TypeScript / JavaScript | tree-sitter-typescript |
+| Vue | Extracts `<script>` / `<script setup>`, delegates to TS/JS parser |
+| Auto | Per-file detection for monorepos and multi-language projects |
+
+## Stale Code Detection
+
+`sampler stale-code <project>` finds functions that may no longer be needed:
+
+- Called only from test files
+- Zero non-test callers in the project call graph
+- Defined in production code
+
+Supported test patterns:
+
+- Python: `tests/`, `test_*.py`, `*_test.py`
+- Go: `*_test.go`
+- TypeScript / JavaScript / Vue: `__tests__/`, `test/`, `spec/`, `*.test.*`, `*.spec.*`
+
+This is a heuristic signal, not a guarantee of dead code.
+
+## Examples
+
+```bash
+$ sampler search worker --project myproj
+myproj:src/tasks/celery_app.py:70 function on_worker_ready  def on_worker_ready(sender)
+
+$ sampler related ConfigManager --project myproj --style bars
+myproj:src/config.py:24-105 class ConfigManager  [parent]
+...
+
+$ sampler stale-code myproj
+myproj:src/utils/retry.py:12-28 function retry_request  test_callers=2 non_test_callers=0  [tests.test_retry.test_retry_request]
+```
+
+## Data Location
+
+- Config: `~/.sampler/config.yaml`
+- Database: `~/.sampler/graph.db`
+
+## Running Tests
+
+```bash
+pytest -q
+```
+
+## Notes
+
+- Compact output is the default by design — built for agent workflows and fast human scanning.
+- For roadmap details, see [TODO.md](TODO.md) and [PLAN.md](PLAN.md).

sampler_cli-0.4.5/README.md

@@ -0,0 +1,184 @@
+<p align="center">
+  <img src="./assets/sampler.png" alt="Sampler logo" width="220">
+</p>
+
+<h1 align="center">Sampler</h1>
+
+<p align="center">
+  <strong>Token-efficient CLI for indexing and searching code symbols across projects.</strong><br>
+  Compact output. Short paths. Low-noise symbol views.
+</p>
+
+<p align="center">
+  <a href="https://pypi.org/project/sampler-cli/"><img src="https://img.shields.io/pypi/v/sampler-cli" alt="PyPI version"></a>
+  <img src="https://img.shields.io/badge/python-3.11%2B-blue" alt="Python 3.11+">
+</p>
+
+---
+
+> *The code isn't the problem. The problem is the distance between you and the code.*
+>
+> Sampler closes that distance. One index. One query. The right symbol, the right relationship, the right context — delivered without the noise. Because in a world drowning in repositories, the person who finds what matters first is the person who moves the work forward.
+
+## Installation
+
+```bash
+pip install sampler-cli
+```
+
+Development:
+
+```bash
+pip install -e '.[dev]'
+```
+
+Semantic stack (TF-IDF + local hash fallback):
+
+```bash
+pip install -e '.[semantic]'
+```
+
+## Quick Start
+
+```bash
+sampler init
+sampler project add myproj /absolute/path/to/project --language auto
+sampler index myproj
+sampler search retry --project myproj
+sampler symbols myproj
+sampler overview src/main.py
+```
+
+## Commands
+
+### Core
+
+| Command | Description |
+| --- | --- |
+| `sampler version [--plain]` | Show version |
+| `sampler init` | Initialize Sampler config and data directory |
+| `sampler index <project>` | Index a project's symbols and relationships |
+| `sampler search <query>` | Search symbols across a project |
+| `sampler search-all <query>` | Search across all indexed projects |
+| `sampler symbols <project>` | List symbols in a project |
+| `sampler overview <filepath>` | Show symbol overview for a file |
+
+**Search options:** `--project`, `--type`, `--limit`, `--semantic`, `--style plain|bars`
+
+### Relationships
+
+| Command | Description |
+| --- | --- |
+| `sampler callers <symbol>` | Find callers of a symbol |
+| `sampler usages <symbol>` | Find usages of a symbol |
+| `sampler related <symbol>` | Find related symbols |
+
+Symbols can also be selected as `<path>:<symbol>`, e.g. `app/utils/helpers.py:format_kda`.
+
+### Project Management
+
+| Command | Description |
+| --- | --- |
+| `sampler project add <name> <path> --language <lang>` | Add a project |
+| `sampler project update <name>` | Update project path or language |
+| `sampler project list` | List projects |
+| `sampler project deps <name>` | Show project dependencies |
+| `sampler project remove <name>` | Remove a project |
+
+Languages: `python`, `go`, `typescript`, `javascript`, `vue`, `auto`.
+
+### Config & Analysis
+
+| Command | Description |
+| --- | --- |
+| `sampler config show` | Show current config |
+| `sampler config embeddings` | Configure embedding provider |
+| `sampler embed <project>` | Precompute embeddings |
+| `sampler stale-code <project>` | Find candidate stale code |
+
+## Embeddings & Semantic Search
+
+`sampler search --semantic` uses a pluggable adapter pattern:
+
+- **Default:** `bge-small` (BAAI/bge-small-en-v1.5 via fastembed — local ONNX, ~384 dim)
+- **Built-ins:** `hash` (deterministic fallback), `ollama`, `nomic`, `openai`, `fastembed`
+- **Lexical primary:** TF-IDF (sklearn, on-the-fly, no pre-embedding required)
+- **Final fallback:** hash fingerprint (always available)
+
+Configure in `~/.sampler/config.yaml` or via `sampler config embeddings`:
+
+```yaml
+embeddings:
+  provider: "bge-small"
+  # provider: "ollama"
+  # model: "nomic-embed-text"
+  # base_url: "http://localhost:11434"
+```
+
+Install extras:
+
+```bash
+pip install 'sampler-cli[embeddings]'        # BGE (recommended)
+pip install 'sampler-cli[ollama-embeddings]'
+pip install 'sampler-cli[openai-embeddings]'
+```
+
+Run `sampler embed <project>` to precompute vectors for the active provider. Change providers? Re-run `embed` after updating config.
+
+Offline or air-gapped: set `provider: hash`, or rely on TF-IDF + hash with the `[semantic]` extra.
+
+## Language Support
+
+| Language | Parser |
+| --- | --- |
+| Python | stdlib AST |
+| Go | tree-sitter-go |
+| TypeScript / JavaScript | tree-sitter-typescript |
+| Vue | Extracts `<script>` / `<script setup>`, delegates to TS/JS parser |
+| Auto | Per-file detection for monorepos and multi-language projects |
+
+## Stale Code Detection
+
+`sampler stale-code <project>` finds functions that may no longer be needed:
+
+- Called only from test files
+- Zero non-test callers in the project call graph
+- Defined in production code
+
+Supported test patterns:
+
+- Python: `tests/`, `test_*.py`, `*_test.py`
+- Go: `*_test.go`
+- TypeScript / JavaScript / Vue: `__tests__/`, `test/`, `spec/`, `*.test.*`, `*.spec.*`
+
+This is a heuristic signal, not a guarantee of dead code.
+
+## Examples
+
+```bash
+$ sampler search worker --project myproj
+myproj:src/tasks/celery_app.py:70 function on_worker_ready  def on_worker_ready(sender)
+
+$ sampler related ConfigManager --project myproj --style bars
+myproj:src/config.py:24-105 class ConfigManager  [parent]
+...
+
+$ sampler stale-code myproj
+myproj:src/utils/retry.py:12-28 function retry_request  test_callers=2 non_test_callers=0  [tests.test_retry.test_retry_request]
+```
+
+## Data Location
+
+- Config: `~/.sampler/config.yaml`
+- Database: `~/.sampler/graph.db`
+
+## Running Tests
+
+```bash
+pytest -q
+```
+
+## Notes
+
+- Compact output is the default by design — built for agent workflows and fast human scanning.
+- For roadmap details, see [TODO.md](TODO.md) and [PLAN.md](PLAN.md).

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sampler-cli"
-version = "0.4.3"
+version = "0.4.5"
 description = "Token-efficient CLI for indexing and searching code symbols (Python-first, designed for minimal LLM/agent context size)"
 readme = "README.md"
 license = { text = "MIT" }
@@ -69,6 +69,6 @@ pythonpath = ["src"]
 testpaths = ["tests"]
 
 [project.urls]
-Homepage = "https://github.com/sicr0/sampler-cli"
-Repository = "https://github.com/sicr0/sampler-cli"
-Issues = "https://github.com/sicr0/sampler-cli/issues"
+Homepage = "https://github.com/SamuelCarmona83/sampler-cli"
+Repository = "https://github.com/SamuelCarmona83/sampler-cli"
+Issues = "https://github.com/SamuelCarmona83/sampler-cli/issues"

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/__init__.py

@@ -1,3 +1,3 @@
 __all__ = ["__version__"]
 
-__version__ = "0.4.3"
+__version__ = "0.4.5"

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/cli/main.py

@@ -183,6 +183,8 @@ def project_list() -> None:
     table.add_column("Language", style="green")
     table.add_column("Enabled", justify="center")
 
+    db = _database()
+
     for p in projects:
         try:
             pp = Path(p.path).resolve()
@@ -195,7 +197,21 @@ def project_list() -> None:
         except Exception:
             disp = p.path
         enabled = "[green]yes[/green]" if p.enabled else "[dim]no[/dim]"
-        table.add_row(p.name, disp, p.language, enabled)
+
+        lang_display = p.language
+        if (p.language or "").lower() == "auto":
+            breakdown = db.get_project_language_breakdown(p.name)
+            total = sum(breakdown.values()) or 1
+            parts = []
+            for lang, cnt in sorted(breakdown.items(), key=lambda kv: -kv[1])[:4]:  # top 4 for brevity
+                pct = int(round(cnt * 100 / total))
+                parts.append(f"{lang} {pct}%")
+            if parts:
+                lang_display = f"auto ({', '.join(parts)})"
+            else:
+                lang_display = "auto (no files yet)"
+
+        table.add_row(p.name, disp, lang_display, enabled)
 
     console.print(table)
 
@@ -205,7 +221,7 @@ def project_add(
     name: str,
     path: str,
     language: str = typer.Option(
-        "python", "--language", help="python, go, typescript, javascript, or 'auto' for monorepos"
+        "python", "--language", help="python, go, typescript, javascript, vue, or 'auto' for monorepos"
     ),
 ) -> None:
     """Register project in global config."""
@@ -232,7 +248,7 @@ def project_remove(name: str) -> None:
 def project_update(
     name: str,
     path: str | None = typer.Option(None, "--path", help="New absolute path for the project"),
-    language: str | None = typer.Option(None, "--language", help="New language (or 'auto' for monorepos)"),
+    language: str | None = typer.Option(None, "--language", help="New language (python|go|typescript|javascript|vue|auto)"),
 ) -> None:
     """Update a registered project's path/language in place (no remove/add needed)."""
     if path is None and language is None:

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/db.py

@@ -640,6 +640,24 @@ class Database:
             "embeddings": int(row["embeddings"]),
         }
 
+    def get_project_language_breakdown(self, project_name: str) -> dict[str, int]:
+        """For auto projects (or any), return {language: file_count} from the files table.
+
+        Used to display per-language % in `project list` for language=auto projects.
+        Very cheap (single grouped query); data is already stored during index.
+        """
+        sql = """
+            SELECT COALESCE(f.language, 'unknown') AS lang, COUNT(*) AS cnt
+            FROM files f
+            JOIN projects p ON f.project_id = p.id
+            WHERE p.name = ?
+            GROUP BY lang
+            ORDER BY cnt DESC
+        """
+        with self.connect() as conn:
+            rows = conn.execute(sql, (project_name,)).fetchall()
+        return {row["lang"]: int(row["cnt"]) for row in rows}
+
     def get_top_symbols_by_degree(self, project_name: str, limit: int = 80) -> list[sqlite3.Row]:
         """Top symbols by in+out relationship degree for graph preview."""
         sql = """

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/indexer/builder.py

@@ -12,6 +12,7 @@ from sampler.indexer.imports import extract_imports
 from sampler.indexer.parsers.go import GoParser
 from sampler.indexer.parsers.python import PythonParser
 from sampler.indexer.parsers.typescript import TypeScriptParser
+from sampler.indexer.parsers.vue import VueParser
 from sampler.indexer.store import SymbolStore
 from sampler.viz.discover_emit import emit_discover
 from sampler.viz.events import FileParsing, LogLine, Stage, StageChanged
@@ -29,6 +30,7 @@ class IndexBuilder:
             "go": GoParser(),
             "typescript": TypeScriptParser(),
             "javascript": TypeScriptParser(),
+            "vue": VueParser(),
         }
 
     def index_project(

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/indexer/discover.py

@@ -8,6 +8,7 @@ LANGUAGE_EXTENSIONS: dict[str, set[str]] = {
     "go": {".go"},
     "typescript": {".ts", ".tsx", ".js", ".jsx"},
     "javascript": {".js", ".jsx", ".mjs", ".cjs"},
+    "vue": {".vue"},
 }
 
 DEFAULT_IGNORE_PARTS = {
@@ -59,10 +60,11 @@ def _build_extension_language_map() -> dict[str, str]:
     Iteration order determines the winner for extensions shared by multiple
     languages (.js/.jsx are listed under both "typescript" and "javascript";
     both route to the same real parser implementation, so the choice is
-    cosmetic but kept stable).
+    cosmetic but kept stable). .vue maps to dedicated "vue" (which delegates
+    to the TS/JS parser after <script> extraction).
     """
     ext_to_lang: dict[str, str] = {}
-    for lang in ("python", "go", "typescript", "javascript"):
+    for lang in ("python", "go", "vue", "typescript", "javascript"):
         for ext in LANGUAGE_EXTENSIONS[lang]:
             ext_to_lang.setdefault(ext, lang)
     return ext_to_lang
@@ -74,6 +76,7 @@ def discover_files_multi(
     """Discover files across ALL supported languages, returning (path, detected_language) pairs.
 
     Used for monorepo/multi-language projects indexed with language="auto".
+    Vue SFCs are detected as language "vue".
     """
     root = Path(project_path)
     if not root.exists() or not root.is_dir():

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/indexer/imports.py

@@ -27,7 +27,7 @@ def extract_imports(content: str, language: str) -> list[str]:
         return _extract_python_imports(content)
     if language == "go":
         return _extract_go_imports(content)
-    if language in ("typescript", "javascript"):
+    if language in ("typescript", "javascript", "vue"):
         return _extract_ts_imports(content)
     return []
 

sampler_cli-0.4.5/src/sampler/indexer/parsers/vue.py

@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+import re
+
+from sampler.indexer.parsers.base import BaseParser
+from sampler.indexer.parsers.typescript import TypeScriptParser
+
+
+class VueParser(BaseParser):
+    """Parser for Vue single-file components (.vue).
+
+    Extracts the <script> (or <script setup>) section using stdlib re (supports lang="ts|tsx|js|jsx|typescript",
+    setup attribute, various quoting). Delegates symbol/relationship extraction to the existing
+    TypeScriptParser (which covers JS/TS + arrows/classes/etc.) using a dummy filepath so that
+    the delegate's _select_language picks the right grammar (ts vs tsx).
+
+    Line numbers in results are offset so they are correct relative to the original .vue file.
+    Graceful empty return if no <script> section or other issues.
+    """
+
+    language = "vue"
+
+    _SCRIPT_RE = re.compile(
+        r"(?is)<script([^>]*)>(.*?)</script>"
+    )
+
+    def parse(self, content: str, filepath: str) -> tuple[list[dict], list[dict]]:
+        symbols: list[dict] = []
+        relationships: list[dict] = []
+
+        if not filepath.lower().endswith(".vue"):
+            # Shouldn't normally happen, but delegate to TS parser for safety
+            return TypeScriptParser().parse(content, filepath)
+
+        extracted = self._extract_vue_script(content)
+        if extracted is None:
+            return symbols, relationships
+
+        script_text, line_offset, is_tsx = extracted
+
+        # Dummy filepath controls grammar selection inside the (unchanged) TS parser
+        dummy = "Comp.script.tsx" if is_tsx else "Comp.script.ts"
+        inner_symbols, inner_relationships = TypeScriptParser().parse(script_text, dummy)
+
+        # Offset lines so they refer to the original .vue file (script content lines are 0-based in tree)
+        for s in inner_symbols:
+            s["start_line"] += line_offset
+            s["end_line"] += line_offset
+        for r in inner_relationships:
+            if "line" in r and r["line"] is not None:
+                r["line"] += line_offset
+
+        return inner_symbols, inner_relationships
+
+    def _extract_vue_script(self, content: str) -> tuple[str, int, bool] | None:
+        """Return (script_text, 0-based_line_offset_for_script_body, use_tsx) or None."""
+        m = self._SCRIPT_RE.search(content)
+        if not m:
+            return None
+
+        attrs = m.group(1) or ""
+        script = m.group(2)
+
+        # Compute offset: number of newlines before the start of the script body
+        prefix = content[: m.start(2)]
+        line_offset = prefix.count("\n")
+
+        # Detect if we should force TSX grammar inside delegate (rare for Vue but supported)
+        lang_match = re.search(r'lang\s*=\s*["\']?([^"\'\s>]+)', attrs, re.IGNORECASE)
+        lang_val = (lang_match.group(1) if lang_match else "").lower()
+        use_tsx = lang_val in ("tsx", "jsx")
+
+        return script, line_offset, use_tsx

{sampler_cli-0.4.3 → sampler_cli-0.4.5}/src/sampler/query/engine.py

@@ -128,6 +128,7 @@ class QueryEngine:
         if name.endswith(("_test.py", "_test.go")):
             return True
 
+        # Note: .vue test files (e.g. Foo.test.vue) are also supported for Vue projects
         return name.endswith(
             (
                 ".test.ts",
@@ -136,12 +137,14 @@ class QueryEngine:
                 ".test.jsx",
                 ".test.mjs",
                 ".test.cjs",
+                ".test.vue",
                 ".spec.ts",
                 ".spec.tsx",
                 ".spec.js",
                 ".spec.jsx",
                 ".spec.mjs",
                 ".spec.cjs",
+                ".spec.vue",
             )
         )