pi-codemcp 1.0.0 → 1.1.0

package/sidecar/models.py

@@ -6,6 +6,8 @@ from pydantic import BaseModel, ConfigDict, Field, model_serializer
 
 type ServerTransport = Literal["stdio", "http", "sse"]
 type ServerAuth = Literal["oauth", "bearer"]
+type SearchDetail = Literal["names", "signatures", "full"]
+type SearchMode = Literal["search", "inventory"]
 
 
 class NormalizedServerInfo(BaseModel):
@@ -27,8 +29,10 @@ class ToolSchemaView(BaseModel):
     source: Literal["mcp_tool", "saved_chain"]
     server: str | None = None
     description: str | None = None
-    signature: str
-    stub: str
+    signature: str | None = None
+    stub: str | None = None
+    score: float | None = None
+    matched_fields: list[str] = Field(default_factory=list)
 
     @model_serializer(mode="plain")
     def serialize_compact(self) -> dict[str, object]:
@@ -36,13 +40,19 @@ class ToolSchemaView(BaseModel):
             "name": self.name,
             "call": self.call,
             "source": self.source,
-            "signature": self.signature,
-            "stub": self.stub,
         }
         if self.server is not None:
             values["server"] = self.server
         if self.description is not None:
             values["description"] = self.description
+        if self.signature is not None:
+            values["signature"] = self.signature
+        if self.stub is not None:
+            values["stub"] = self.stub
+        if self.score is not None:
+            values["score"] = round(self.score, 2)
+        if self.matched_fields:
+            values["matched_fields"] = self.matched_fields
         return values
 
 
@@ -53,11 +63,38 @@ class ServerToolSummary(BaseModel):
     tool_count: int
 
 
+class ExecutionLimitsView(BaseModel):
+    model_config = ConfigDict(extra="forbid", strict=True)
+
+    timeout_seconds: float
+    tool_timeout_seconds: float
+    max_calls: int
+    result_limit_bytes: int
+
+
 class SearchResponse(BaseModel):
     model_config = ConfigDict(extra="forbid", strict=True)
 
+    mode: SearchMode
+    detail: SearchDetail
     total_tool_count: int
+    filtered_tool_count: int
     servers: list[ServerToolSummary]
+    cursor: int
+    next_cursor: int | None = None
+    has_more: bool = False
+    project_scope_available: bool
+    execution_limits: ExecutionLimitsView
+    prelude: str | None = None
+    results: list[ToolSchemaView]
+
+
+class InspectResponse(BaseModel):
+    model_config = ConfigDict(extra="forbid", strict=True)
+
+    prelude: str
+    project_scope_available: bool
+    execution_limits: ExecutionLimitsView
     results: list[ToolSchemaView]
 
 

package/sidecar/pyproject.toml

@@ -2,7 +2,7 @@
 name = "pi-codemcp-sidecar"
 version = "0.1.0"
 description = "Python sidecar for pi-codemcp"
-requires-python = ">=3.12"
+requires-python = ">=3.13"
 dependencies = [
   "fastmcp==3.4.4",
   "httpx==0.28.1",
@@ -14,12 +14,12 @@ dependencies = [
 
 [dependency-groups]
 dev = [
-  "mypy==2.2.0",
-  "prek==0.4.8",
-  "pytest==8.4.2",
+  "mypy==2.3.0",
+  "prek==0.4.10",
+  "pytest==9.1.1",
   "pytest-asyncio==1.4.0",
-  "ruff==0.15.21",
-  "ty==0.0.58",
+  "ruff==0.15.22",
+  "ty==0.0.61",
 ]
 
 [tool.pytest.ini_options]
@@ -27,8 +27,8 @@ asyncio_mode = "auto"
 pythonpath = [".."]
 
 [tool.ruff]
-required-version = "==0.15.21"
-target-version = "py312"
+required-version = "==0.15.22"
+target-version = "py313"
 line-length = 100
 src = ["sidecar"]
 preview = true
@@ -37,23 +37,23 @@ force-exclude = true
 [tool.ruff.lint]
 select = ["ALL"]
 ignore = [
-  "COM812", # Conflicts with the formatter.
-  "CPY001", # Legal copyright text is not configured.
-  "D100",   # Internal modules do not need module docstrings.
-  "D101",   # Pydantic catalog models are self-describing.
-  "D102",   # Internal methods are covered by types and focused class documentation.
-  "D103",   # Internal functions are covered by types and focused module documentation.
-  "D104",   # Internal packages do not need package docstrings beyond sidecar/__init__.py.
-  "D105",   # Magic methods do not need docstrings.
-  "D107",   # Constructor contracts are expressed by typed fields.
-  "D203",   # Incompatible with D211.
-  "D213",   # Incompatible with D212.
-  "DOC201", # Return types are explicit and checked.
-  "DOC501", # Repeating raised exception types in docstrings reduces readability.
-  "EM101",  # Inline exception messages are clearer at the failure site.
-  "EM102",  # Inline contextual exception messages are clearer at the failure site.
-  "FBT001", # JSON Schema itself may be boolean; these are not boolean control arguments.
-  "TRY003", # Custom exception wrappers would obscure concise boundary failures.
+  "missing-trailing-comma", # Conflicts with the formatter.
+  "missing-copyright-notice", # Legal copyright text is not configured.
+  "undocumented-public-module",   # Internal modules do not need module docstrings.
+  "undocumented-public-class",   # Pydantic catalog models are self-describing.
+  "undocumented-public-method",   # Internal methods are covered by types and focused class documentation.
+  "undocumented-public-function",   # Internal functions are covered by types and focused module documentation.
+  "undocumented-public-package",   # Internal packages do not need package docstrings beyond sidecar/__init__.py.
+  "undocumented-magic-method",   # Magic methods do not need docstrings.
+  "undocumented-public-init",   # Constructor contracts are expressed by typed fields.
+  "incorrect-blank-line-before-class",   # Incompatible with D211.
+  "multi-line-summary-second-line",   # Incompatible with D212.
+  "docstring-missing-returns", # Return types are explicit and checked.
+  "docstring-missing-exception", # Repeating raised exception types in docstrings reduces readability.
+  "raw-string-in-exception",  # Inline exception messages are clearer at the failure site.
+  "f-string-in-exception",  # Inline contextual exception messages are clearer at the failure site.
+  "boolean-type-hint-positional-argument", # JSON Schema itself may be boolean; these are not boolean control arguments.
+  "raise-vanilla-args", # Custom exception wrappers would obscure concise boundary failures.
 ]
 
 [tool.ruff.lint.isort]
@@ -83,23 +83,19 @@ docstring-code-line-length = 100
 "tests/**/*.py" = [
   "ANN",
   "D",
-  "PLR2004",
-  "S101",
-  "S105",
-  "S106",
-  "SLF001",
+  "magic-value-comparison",
+  "assert",
+  "hardcoded-password-string",
+  "hardcoded-password-func-arg",
+  "private-member-access",
 ]
-"tests/fixtures/**/*.py" = ["T201"]
+"tests/fixtures/**/*.py" = ["print"]
 
 [tool.mypy]
-python_version = "3.12"
+python_version = "3.13"
 plugins = ["pydantic.mypy"]
 strict = true
-files = [
-  "sidecar",
-  "tests/python/test_executor.py",
-  "tests/python/test_settings.py",
-]
+files = ["sidecar", "tests/python"]
 pretty = true
 show_error_codes = true
 show_error_code_links = true
@@ -128,9 +124,6 @@ init_typed = true
 warn_required_dynamic_aliases = true
 warn_untyped_fields = true
 
-[tool.ty.environment]
-python-version = "3.12"
-
 [tool.ty.rules]
 all = "error"
 

package/sidecar/runtime_paths.py

@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass
+from pathlib import Path
+
+DEFAULT_AGENT_DIR = Path.home() / ".pi" / "agent"
+CODEMCP_AGENT_DIR_ENV = "PI_CODEMCP_AGENT_DIR"
+CODEMCP_PROJECT_CHAINS_DIR_ENV = "PI_CODEMCP_PROJECT_CHAINS_DIR"
+PI_AGENT_DIR_ENV = "PI_CODING_AGENT_DIR"
+
+
+@dataclass(frozen=True)
+class RuntimePaths:
+    config_path: Path
+    oauth_dir: Path
+    catalog_dir: Path
+    settings_path: Path
+    global_chains_dir: Path
+    project_chains_dir: Path | None
+
+    def as_tuple(self) -> tuple[Path, Path, Path, Path, Path, Path | None]:
+        return (
+            self.config_path,
+            self.oauth_dir,
+            self.catalog_dir,
+            self.settings_path,
+            self.global_chains_dir,
+            self.project_chains_dir,
+        )
+
+
+def resolve_runtime_paths(
+    *,
+    agent_dir: Path | str | None = None,
+    config_path: Path | str | None = None,
+    settings_path: Path | str | None = None,
+    oauth_dir: Path | str | None = None,
+    catalog_dir: Path | str | None = None,
+    global_chains_dir: Path | str | None = None,
+    project_chains_dir: Path | str | None = None,
+) -> RuntimePaths:
+    resolved_agent_dir = _agent_dir(agent_dir)
+    state_dir = resolved_agent_dir / "pi-codemcp"
+    return RuntimePaths(
+        config_path=_resolve_path(config_path) or resolved_agent_dir / "mcp.json",
+        oauth_dir=_resolve_path(oauth_dir) or state_dir / "oauth",
+        catalog_dir=_resolve_path(catalog_dir) or state_dir / "catalog",
+        settings_path=_resolve_path(settings_path) or state_dir / "settings.json",
+        global_chains_dir=_resolve_path(global_chains_dir) or state_dir / "chains",
+        project_chains_dir=_project_chains_dir(project_chains_dir),
+    )
+
+
+def _agent_dir(value: Path | str | None) -> Path:
+    if value is not None:
+        return _resolve_path(value) or DEFAULT_AGENT_DIR
+    raw_agent_dir = os.environ.get(CODEMCP_AGENT_DIR_ENV) or os.environ.get(PI_AGENT_DIR_ENV)
+    return _resolve_path(raw_agent_dir) or DEFAULT_AGENT_DIR
+
+
+def _project_chains_dir(value: Path | str | None) -> Path | None:
+    if value is not None:
+        return _resolve_path(value)
+    return _resolve_path(os.environ.get(CODEMCP_PROJECT_CHAINS_DIR_ENV))
+
+
+def _resolve_path(value: Path | str | None) -> Path | None:
+    if value is None:
+        return None
+    return Path(value).expanduser()

package/sidecar/settings.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import json
 from typing import TYPE_CHECKING, Literal
 
 from pydantic import BaseModel, ConfigDict, Field
@@ -24,7 +25,7 @@ class CodeMcpSettings(BaseModel):
         strict=True,
     )
 
-    version: Literal[1] = 1
+    version: Literal[2] = 2
     background_warmup: bool = True
     cache_ttl_hours: int = Field(default=24, ge=0, le=720)
     execution_timeout_seconds: int = Field(default=30, ge=1, le=300)
@@ -32,7 +33,6 @@ class CodeMcpSettings(BaseModel):
     max_calls: int = Field(default=50, ge=1, le=200)
     result_limit_kib: int = Field(default=16, ge=1, le=1024)
     output_limit_kib: int = Field(default=50, ge=1, le=1024)
-    output_line_limit: int = Field(default=2000, ge=1, le=10000)
     disabled_tools: dict[str, list[str]] = Field(default_factory=dict)
 
     def tool_enabled(self, server: str, tool: str) -> bool:
@@ -56,4 +56,8 @@ def load_settings(path: Path) -> CodeMcpSettings:
         raw = path.read_text(encoding="utf-8")
     except FileNotFoundError:
         return CodeMcpSettings()
-    return CodeMcpSettings.model_validate_json(raw)
+    value = json.loads(raw)
+    if isinstance(value, dict) and value.get("version", 1) == 1:
+        value = {key: item for key, item in value.items() if key != "outputLineLimit"}
+        value["version"] = 2
+    return CodeMcpSettings.model_validate(value)