docpull 6.5.4__tar.gz → 6.5.5__tar.gz

{docpull-6.5.4/src/docpull.egg-info → docpull-6.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docpull
-Version: 6.5.4
+Version: 6.5.5
 Summary: Declare, sync, diff, and lock context dependencies for AI agents
 Author-email: Zachary Roth <support@raintree.technology>
 Maintainer-email: Raintree Technology <support@raintree.technology>

{docpull-6.5.4 → docpull-6.5.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "docpull"
-version = "6.5.4"
+version = "6.5.5"
 dynamic = []
 description = "Declare, sync, diff, and lock context dependencies for AI agents"
 readme = {file = "README.md", content-type = "text/markdown"}

{docpull-6.5.4 → docpull-6.5.5}/src/docpull/__init__.py

@@ -15,7 +15,7 @@ from typing import TYPE_CHECKING, Any
 
 from .surface import PUBLIC_SDK_EXPORTS
 
-__version__ = "6.5.4"
+__version__ = "6.5.5"
 
 _LAZY_EXPORTS: dict[str, tuple[str, str]] = {
     **{

{docpull-6.5.4 → docpull-6.5.5}/src/docpull/mcp/server.py

@@ -47,6 +47,7 @@ from collections.abc import Awaitable, Callable
 from pathlib import Path
 from typing import Any, Literal, cast
 
+from .. import __version__
 from ..accounting import (
     RunAccounting,
     blocked_action,
@@ -89,6 +90,35 @@ SERVER_INSTRUCTIONS = (
 )
 
 
+def _advertise_tool_cost(tool: Any) -> Any:
+    """Attach the cost contract promised by the server instructions."""
+    if tool.name == "render_url":
+        description = (
+            f"Cost: local, $0 by default; vercel and e2b runtimes are paid-capable. {tool.description}"
+        )
+        contract = {
+            "cost_class": "paid-capable",
+            "default_cost_class": "local",
+            "estimated_cost_usd": 0,
+            "paid_capable": True,
+            "paid_when": {"argument": "runtime", "values": ["vercel", "e2b"]},
+            "budget_argument": "budget",
+        }
+    else:
+        description = tool.description or ""
+        if not description.startswith("Cost:"):
+            description = f"Cost: local, $0. {description}"
+        contract = {
+            "cost_class": "local",
+            "estimated_cost_usd": 0,
+            "paid_capable": False,
+        }
+
+    metadata = dict(tool.meta or {})
+    metadata["docpull/cost"] = contract
+    return tool.model_copy(update={"description": description, "meta": metadata})
+
+
 # Output schemas — keep these next to the tool list so they stay in sync.
 # Tools that return free-form Markdown (fetch_url) intentionally omit a
 # schema; the rest expose structured payloads alongside the rendered text.
@@ -1391,7 +1421,7 @@ async def _run_stdio() -> int:
         return 1
 
     server: Server = Server(  # type: ignore[no-any-unimported]
-        "docpull", instructions=SERVER_INSTRUCTIONS
+        "docpull", version=__version__, instructions=SERVER_INSTRUCTIONS
     )
 
     @server.list_tools()  # type: ignore[misc,no-untyped-call]
@@ -2470,7 +2500,7 @@ async def _run_stdio() -> int:
                 outputSchema=_REMOVE_SOURCE_OUTPUT_SCHEMA,
             ),
         ]
-        return [tool for tool in tools if tool.name not in PRUNED_MCP_TOOLS]
+        return [_advertise_tool_cost(tool) for tool in tools if tool.name not in PRUNED_MCP_TOOLS]
 
     async def _make_progress_callback() -> Any:
         """Return ``(pages_done, total_or_none) -> awaitable`` bound to the

{docpull-6.5.4 → docpull-6.5.5/src/docpull.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docpull
-Version: 6.5.4
+Version: 6.5.5
 Summary: Declare, sync, diff, and lock context dependencies for AI agents
 Author-email: Zachary Roth <support@raintree.technology>
 Maintainer-email: Raintree Technology <support@raintree.technology>

{docpull-6.5.4 → docpull-6.5.5}/tests/test_ci_policy.py

@@ -332,7 +332,7 @@ def test_plugin_launcher_reports_actionable_setup_when_docpull_is_missing(
     )
 
     assert proc.returncode == 127
-    assert "pipx install 'docpull[mcp]==6.5.4'" in proc.stderr
+    assert "pipx install 'docpull[mcp]==6.5.5'" in proc.stderr
 
     (tmp_path / "docpull").mkdir()
     proc = subprocess.run(  # nosec B603
@@ -345,7 +345,7 @@ def test_plugin_launcher_reports_actionable_setup_when_docpull_is_missing(
     )
 
     assert proc.returncode == 127
-    assert "pipx install 'docpull[mcp]==6.5.4'" in proc.stderr
+    assert "pipx install 'docpull[mcp]==6.5.5'" in proc.stderr
 
 
 def test_codex_plugin_default_prompt_fits_host_limit() -> None:

{docpull-6.5.4 → docpull-6.5.5}/tests/test_mcp_server.py

@@ -17,6 +17,7 @@ pytest.importorskip("mcp.client.stdio")
 
 from mcp.client.stdio import stdio_client
 
+from docpull import __version__
 from docpull.mcp import server as mcp_server
 from mcp import ClientSession, StdioServerParameters
 from tests.pack_fixtures import write_context_pack
@@ -185,10 +186,28 @@ async def test_stdio_server_lists_and_calls_tools(tmp_path):
         env=env,
     )
     async with _stdio_client_with_cleanup(server) as (read, write), ClientSession(read, write) as session:
-        await session.initialize()
+        initialization = await session.initialize()
+        assert initialization.serverInfo.version == __version__
 
         tools = await session.list_tools()
         names = {tool.name for tool in tools.tools}
+        for tool in tools.tools:
+            assert tool.description is not None
+            assert tool.description.startswith("Cost:")
+            assert tool.meta is not None
+            assert "docpull/cost" in tool.meta
+            cost = tool.meta["docpull/cost"]
+            assert cost["paid_capable"] is (tool.name == "render_url")
+            assert cost["estimated_cost_usd"] == 0
+        render_cost = next(
+            tool.meta["docpull/cost"]
+            for tool in tools.tools
+            if tool.name == "render_url" and tool.meta is not None
+        )
+        assert render_cost["paid_when"] == {
+            "argument": "runtime",
+            "values": ["vercel", "e2b"],
+        }
         fetch_tool = next(tool for tool in tools.tools if tool.name == "fetch_url")
         assert fetch_tool.inputSchema["properties"]["remote_document_backend"]["enum"] == [
             "auto",