bytecii
3293cb5ff4
Two cleanup-plan items, both Python-side.
Item 29 -- `mirage mcp`. TypeScript shipped a six-tool stdio MCP server;
Python had none, so a pip-install user could not point Cursor or Claude
Desktop at a workspace and `mirage --help` differed by distribution.
Adding the entry point alone would have duplicated the tools, because
this side kept them private inside the Claude Agent SDK integration, so
the shared layer comes first:
- agents/tool_descriptions.py -- the six strings, one copy.
- agents/tool_operations.py -- MirageToolOperations, lifted out of the
SDK server's private _MirageTools.
- agents/file_version.py -- stale-write protection, which this side
lacked entirely. TS stamps stored bytes; here the stamp covers the
rendered bytes, because this read tool has always rendered and an
edit must search what the agent was actually shown.
- agents/mcp/server.py + cli/mcp.py -- the server and `mirage mcp`.
- server/workspace_config.py -- config discovery (candidates, env
names, walk up from cwd), which Python had nowhere, so every entry
point had to be handed an explicit path.
The server is the low-level MCP Server rather than FastMCP: FastMCP does
not forward a version, and TS advertises one. Handlers are bound methods,
not decorated closures, so nothing nests.
Item 28 -- the mypy allowlist. 54 modules opted *in* to annotation
checking against 1826, so the default was unchecked and every new file
joined the unchecked side. The default is now strict, with a list of
what is not yet annotated that only shrinks. 166 annotations cleared
along the way; the remainder is named module by module.
Two real defects surfaced by the annotations, neither of them typing:
- Workspace._original_open / _original_os were invented by assignment
in lifecycle.patch_process, so unpatch without a patch raised
AttributeError. Declared, and the restore is guarded.
- sed_generic declared a non-optional writer while its own docstring
and its `write_bytes is None` branch said otherwise; the builder
passes None whenever the backend cannot write.
Tests keep the PathSpec rule instead of full strict: measured, full
strict on python/tests is 2374 errors, of which 634 are `str` where a
pydantic field declares SecretStr -- which pydantic coerces at runtime --
and most of the rest is the monkeypatched-fake pattern CLAUDE.md
sanctions. The rule that is violated for real is PathSpec, 19 times, and
scripts/check_test_pathspec.py now holds that line. One of the 19 was a
latent AttributeError: tests/e2e passes a str to s3 write_bytes, which
reads .mount_path, and the test skips without a live versioned bucket.
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from mirage.accessor.chroma import ChromaAccessor
|
|
from mirage.cache.index import NULL_INDEX, IndexCacheStore
|
|
from mirage.core.chroma.path import resolve_path
|
|
from mirage.core.chroma.readdir import readdir
|
|
from mirage.types import PathSpec
|
|
from mirage.utils.key_prefix import rekey
|
|
|
|
|
|
async def walk(
|
|
accessor: ChromaAccessor,
|
|
path: PathSpec,
|
|
index: IndexCacheStore = NULL_INDEX,
|
|
*,
|
|
include_root: bool = False,
|
|
maxdepth: int | None = None,
|
|
strip_prefix: bool = False,
|
|
ignore_missing: bool = False,
|
|
depth: int = 0,
|
|
) -> list[str]:
|
|
try:
|
|
resolved = await resolve_path(accessor, path, index)
|
|
except (FileNotFoundError, NotADirectoryError):
|
|
if ignore_missing:
|
|
return []
|
|
raise
|
|
|
|
current = path.mount_path if strip_prefix else path.virtual
|
|
results = [current] if include_root else []
|
|
if not resolved.is_dir or (maxdepth is not None and depth >= maxdepth):
|
|
return results
|
|
|
|
try:
|
|
children = await readdir(accessor, path, index)
|
|
except (FileNotFoundError, NotADirectoryError):
|
|
if ignore_missing:
|
|
return results
|
|
raise
|
|
|
|
for child in children:
|
|
child_path = PathSpec.from_str_path(
|
|
child, rekey(path.virtual, path.resource_path, child))
|
|
results.extend(await walk(accessor,
|
|
child_path,
|
|
index,
|
|
include_root=True,
|
|
maxdepth=maxdepth,
|
|
strip_prefix=strip_prefix,
|
|
ignore_missing=ignore_missing,
|
|
depth=depth + 1))
|
|
return results
|