brawler-mcp-server-python 0.1.0__py3-none-any.whl

brawler_mcp_server/__init__.py

@@ -0,0 +1,12 @@
+"""FastMCP wrapper that stamps agent task metadata on MCP tools."""
+
+from brawler_mcp_server.server import BrawlerMCP
+from brawler_mcp_server.utils import META_NAMESPACE, AgentTask, task_from_meta, tools_for_task
+
+__all__ = [
+    'META_NAMESPACE',
+    'AgentTask',
+    'BrawlerMCP',
+    'task_from_meta',
+    'tools_for_task',
+]

brawler_mcp_server/server.py

@@ -0,0 +1,134 @@
+"""FastMCP subclass with Brawler agent-task decorators."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from typing import Any, TypeVar, overload
+
+from fastmcp import FastMCP
+from fastmcp.tools.function_tool import FunctionTool
+
+from brawler_mcp_server.tasks import (
+    CompletionTask,
+    ContextTask,
+    EvaluationTask,
+    NoneTask,
+    PostTask,
+)
+from brawler_mcp_server.utils import AgentTask, TaskHandler, stamp_task_kwargs, tools_for_task
+
+F = TypeVar('F', bound=Callable[..., Any])
+
+
+class BrawlerMCP(FastMCP):
+    """FastMCP server that stamps agent tasks on tools.
+
+    ``@mcp.tool`` is completion. Use ``@context_tool``, ``@eval_tool``,
+    ``@post_tool``, and ``@none_tool`` for the other agent tasks.
+    FastMCP's ``task=`` keyword on ``tool()`` still means a background MCP task,
+    not a Brawler agent task.
+    """
+
+    def tool(
+        self,
+        name_or_fn: str | Callable[..., Any] | None = None,
+        *,
+        name: str | None = None,
+        tags: set[str] | None = None,
+        meta: dict[str, Any] | None = None,
+        **kwargs: Any,
+    ) -> Any:
+        """Register a completion tool (FastMCP ``@tool`` plus task metadata)."""
+        if name is not None:
+            kwargs['name'] = name
+        kwargs['tags'] = tags
+        kwargs['meta'] = meta
+        return self._task_tool(name_or_fn, CompletionTask(), **kwargs)
+
+    @overload
+    def context_tool(self, name_or_fn: F, /, *, as_user_message: bool = ...) -> F: ...
+
+    @overload
+    def context_tool(
+        self, name_or_fn: str | None = None, /, *, as_user_message: bool = ..., **kwargs: Any
+    ) -> Callable[[F], F]: ...
+
+    def context_tool(
+        self,
+        name_or_fn: str | Callable[..., Any] | None = None,
+        /,
+        *,
+        as_user_message: bool = False,
+        **kwargs: Any,
+    ) -> Any:
+        """Register a context tool that must return ``str``."""
+        return self._task_tool(name_or_fn, ContextTask(as_user_message=as_user_message), **kwargs)
+
+    @overload
+    def eval_tool(self, name_or_fn: F, /) -> F: ...
+
+    @overload
+    def eval_tool(self, name_or_fn: str | None = None, /, **kwargs: Any) -> Callable[[F], F]: ...
+
+    def eval_tool(
+        self,
+        name_or_fn: str | Callable[..., Any] | None = None,
+        /,
+        **kwargs: Any,
+    ) -> Any:
+        """Register an evaluation tool that returns ``tuple[bool, float, str | None]``."""
+        return self._task_tool(name_or_fn, EvaluationTask(), **kwargs)
+
+    @overload
+    def post_tool(self, name_or_fn: F, /) -> F: ...
+
+    @overload
+    def post_tool(self, name_or_fn: str | None = None, /, **kwargs: Any) -> Callable[[F], F]: ...
+
+    def post_tool(
+        self,
+        name_or_fn: str | Callable[..., Any] | None = None,
+        /,
+        **kwargs: Any,
+    ) -> Any:
+        """Register a post-result side-effect tool that must return ``None``."""
+        return self._task_tool(name_or_fn, PostTask(), **kwargs)
+
+    @overload
+    def none_tool(self, name_or_fn: F, /) -> F: ...
+
+    @overload
+    def none_tool(self, name_or_fn: str | None = None, /, **kwargs: Any) -> Callable[[F], F]: ...
+
+    def none_tool(
+        self,
+        name_or_fn: str | Callable[..., Any] | None = None,
+        /,
+        **kwargs: Any,
+    ) -> Any:
+        """Register a tool that Brawler agents ignore."""
+        return self._task_tool(name_or_fn, NoneTask(), **kwargs)
+
+    async def tools_by_task(self, task: AgentTask | str) -> list[Any]:
+        """Return in-process tools whose stamped agent task matches *task*."""
+        tools = await self.list_tools()
+        return tools_for_task(tools, task)
+
+    def _task_tool(
+        self,
+        name_or_fn: str | Callable[..., Any] | None,
+        handler: TaskHandler,
+        **kwargs: Any,
+    ) -> Any:
+        if isinstance(name_or_fn, str):
+            kwargs.setdefault('name', name_or_fn)
+            name_or_fn = None
+        stamped = stamp_task_kwargs(kwargs, handler.task, extra=handler.extra_meta())
+
+        def apply(fn: Callable[..., Any]) -> FunctionTool | Callable[..., Any]:
+            handler.check(fn)
+            return super(BrawlerMCP, self).tool(handler.wrap(fn), **stamped)
+
+        if callable(name_or_fn):
+            return apply(name_or_fn)
+        return apply

brawler_mcp_server/tasks/__init__.py

@@ -0,0 +1,17 @@
+"""One handler class per agent task."""
+
+from brawler_mcp_server.tasks.completion import CompletionTask
+from brawler_mcp_server.tasks.context import ContextTask
+from brawler_mcp_server.tasks.evaluation import EvaluationTask
+from brawler_mcp_server.tasks.none import NoneTask
+from brawler_mcp_server.tasks.post import PostTask
+from brawler_mcp_server.utils import AgentTask
+
+__all__ = [
+    'AgentTask',
+    'CompletionTask',
+    'ContextTask',
+    'EvaluationTask',
+    'NoneTask',
+    'PostTask',
+]

brawler_mcp_server/tasks/completion.py

@@ -0,0 +1,12 @@
+"""Completion task: model-callable tools, FastMCP default return types."""
+
+from __future__ import annotations
+
+from brawler_mcp_server.utils import AgentTask, TaskHandler
+
+
+class CompletionTask(TaskHandler):
+    """Stamp completion metadata; return types follow FastMCP defaults."""
+
+    task = AgentTask.COMPLETION
+    label = 'Completion tool'

brawler_mcp_server/tasks/context.py

@@ -0,0 +1,34 @@
+"""Context task: prompt or user-message injection, return ``str``."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from typing import Any
+
+from brawler_mcp_server.utils import AgentTask, TaskHandler, require_return_type, wrap_checked
+
+
+class ContextTask(TaskHandler):
+    """Validate and wrap context tools that must return ``str``."""
+
+    task = AgentTask.CONTEXT
+    label = 'Context tool'
+
+    def __init__(self, *, as_user_message: bool = False) -> None:
+        self.as_user_message = as_user_message
+
+    def extra_meta(self) -> dict[str, Any]:
+        return {'as_user_message': self.as_user_message}
+
+    def check(self, fn: Callable[..., Any]) -> None:
+        require_return_type(fn, str, label=self.label)
+
+    def wrap(self, fn: Callable[..., Any]) -> Callable[..., Any]:
+        name = getattr(fn, '__name__', '<tool>')
+
+        def check(result: Any) -> str:
+            if not isinstance(result, str):
+                raise TypeError(f'{self.label} {name!r} must return str, got {type(result).__name__}.')
+            return result
+
+        return wrap_checked(fn, check)

brawler_mcp_server/tasks/evaluation.py

@@ -0,0 +1,74 @@
+"""Evaluation task: ``tuple[bool, float, str | None]`` serialized as JSON."""
+
+from __future__ import annotations
+
+import inspect
+from collections.abc import Callable
+from typing import Any, TypedDict, get_args, get_origin
+
+from brawler_mcp_server.utils import AgentTask, TaskHandler, return_annotation, wrap_checked
+
+EVAL_RETURN = 'tuple[bool, float, str | None]'
+
+
+class EvalResult(TypedDict):
+    """JSON payload returned by evaluation tools."""
+
+    is_valid: bool
+    score: float
+    feedback: str | None
+
+
+class EvaluationTask(TaskHandler):
+    """Validate and wrap eval tools that return ``tuple[bool, float, str | None]``."""
+
+    task = AgentTask.EVALUATION
+    label = 'Eval tool'
+
+    def check(self, fn: Callable[..., Any]) -> None:
+        annotation = return_annotation(fn)
+        if annotation is inspect.Signature.empty:
+            raise TypeError(f'{self.label} {fn.__name__!r} must return {EVAL_RETURN}, got unannotated.')
+        if not is_eval_tuple(annotation):
+            raise TypeError(f'{self.label} {fn.__name__!r} must return {EVAL_RETURN}, got {annotation}.')
+
+    def wrap(self, fn: Callable[..., Any]) -> Callable[..., Any]:
+        wrapped = wrap_checked(fn, eval_tuple_to_payload)
+        wrapped.__annotations__ = {**getattr(fn, '__annotations__', {}), 'return': EvalResult}
+        return wrapped
+
+
+def eval_tuple_to_payload(result: Any) -> EvalResult:
+    """Validate and convert an eval function result to the wire dict."""
+    if isinstance(result, dict) and {'is_valid', 'score', 'feedback'} <= set(result):
+        return _eval_payload(result['is_valid'], result['score'], result.get('feedback'))
+    if not isinstance(result, tuple) or len(result) != 3:
+        raise TypeError(f'Eval tool must return {EVAL_RETURN}, got {type(result).__name__}.')
+    return _eval_payload(*result)
+
+
+def is_eval_tuple(annotation: Any) -> bool:
+    """Return True when *annotation* is ``tuple[bool, float, str | None]`` (or str/None feedback)."""
+    origin = get_origin(annotation)
+    if origin is not tuple:
+        return False
+    args = get_args(annotation)
+    if len(args) != 3:
+        return False
+    is_valid_t, score_t, feedback_t = args
+    if is_valid_t is not bool or score_t is not float:
+        return False
+    if feedback_t in (str, type(None)):
+        return True
+    union_args = get_args(feedback_t)
+    return set(union_args) == {str, type(None)}
+
+
+def _eval_payload(is_valid: Any, score: Any, feedback: Any) -> EvalResult:
+    if not isinstance(is_valid, bool):
+        raise TypeError(f'Eval is_valid must be bool, got {type(is_valid).__name__}.')
+    if isinstance(score, bool) or not isinstance(score, (int, float)):
+        raise TypeError(f'Eval score must be float, got {type(score).__name__}.')
+    if feedback is not None and not isinstance(feedback, str):
+        raise TypeError(f'Eval feedback must be str | None, got {type(feedback).__name__}.')
+    return EvalResult(is_valid=is_valid, score=float(score), feedback=feedback)

brawler_mcp_server/tasks/none.py

@@ -0,0 +1,12 @@
+"""None task: generic MCP tools that Brawler agents ignore."""
+
+from __future__ import annotations
+
+from brawler_mcp_server.utils import AgentTask, TaskHandler
+
+
+class NoneTask(TaskHandler):
+    """Stamp none metadata; Brawler agents never call these tools."""
+
+    task = AgentTask.NONE
+    label = 'None tool'

brawler_mcp_server/tasks/post.py

@@ -0,0 +1,28 @@
+"""Post task: side effects after a result, return ``None``."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from typing import Any
+
+from brawler_mcp_server.utils import AgentTask, TaskHandler, require_return_type, wrap_checked
+
+
+class PostTask(TaskHandler):
+    """Validate and wrap post tools that must return ``None``."""
+
+    task = AgentTask.POST_TOOL
+    label = 'Post tool'
+
+    def check(self, fn: Callable[..., Any]) -> None:
+        require_return_type(fn, type(None), label=self.label)
+
+    def wrap(self, fn: Callable[..., Any]) -> Callable[..., Any]:
+        name = getattr(fn, '__name__', '<tool>')
+
+        def check(result: Any) -> None:
+            if result is not None:
+                raise TypeError(f'{self.label} {name!r} must return None, got {type(result).__name__}.')
+            return result
+
+        return wrap_checked(fn, check)

brawler_mcp_server/testkit.py

@@ -0,0 +1,74 @@
+"""Test helpers for this package (import via ``brawler_mcp_server.testkit``, not ``tests.*``)."""
+
+from __future__ import annotations
+
+import json
+
+from brawler_mcp_server import BrawlerMCP
+
+
+def example_server() -> BrawlerMCP:
+    """Build a mixed-task server used by listing and call tests."""
+    mcp = BrawlerMCP('TestServer')
+
+    @mcp.context_tool
+    def policies() -> str:
+        return 'Always cite sources.'
+
+    @mcp.context_tool(as_user_message=True)
+    def ticket() -> str:
+        return 'Ticket #123 is still open.'
+
+    @mcp.tool
+    def echo(text: str) -> str:
+        return text
+
+    @mcp.eval_tool
+    def grounded(answer: str) -> tuple[bool, float, str | None]:
+        if not str(answer).strip():
+            return False, 0.0, 'Answer is empty.'
+        return True, 1.0, None
+
+    @mcp.post_tool
+    def audit(answer: str) -> None:
+        return None
+
+    @mcp.none_tool
+    def ping() -> str:
+        return 'pong'
+
+    return mcp
+
+
+def call_text(result: object) -> str:
+    """Extract concatenated text from a FastMCP call result."""
+    content = getattr(result, 'content', None)
+    if content:
+        texts = [getattr(item, 'text', '') for item in content if getattr(item, 'text', None)]
+        if texts:
+            return '\n'.join(texts)
+    data = getattr(result, 'data', None)
+    if data is not None:
+        return str(data)
+    structured = getattr(result, 'structured_content', None) or getattr(result, 'structuredContent', None)
+    if structured is not None:
+        return str(structured)
+    return str(result)
+
+
+def json_payload(result: object) -> dict:
+    """Parse an eval tool call into ``is_valid`` / ``score`` / ``feedback``."""
+    structured = getattr(result, 'structured_content', None) or getattr(result, 'structuredContent', None)
+    if isinstance(structured, dict) and 'is_valid' in structured:
+        return structured
+    data = getattr(result, 'data', None)
+    if isinstance(data, dict) and 'is_valid' in data:
+        return data
+    text = call_text(result)
+    try:
+        parsed = json.loads(text)
+    except json.JSONDecodeError:
+        parsed = None
+    if isinstance(parsed, dict):
+        return parsed
+    raise AssertionError(f'Could not parse eval payload from {result!r}')

brawler_mcp_server/utils.py

@@ -0,0 +1,161 @@
+"""Shared helpers for task metadata, annotations, and return wrapping."""
+
+from __future__ import annotations
+
+import inspect
+from collections.abc import Callable, Iterable, Sequence
+from enum import Enum
+from functools import wraps
+from typing import Any, ClassVar, get_type_hints
+
+META_NAMESPACE = 'mcp_server'
+_Check = Callable[[Any], Any]
+
+
+class AgentTask(str, Enum):
+    """Tasks a Brawler agent can consume from an MCP server."""
+
+    CONTEXT = 'context'
+    COMPLETION = 'completion'
+    EVALUATION = 'evaluation'
+    POST_TOOL = 'post_tool'
+    NONE = 'none'
+
+
+AGENT_TASK_VALUES = frozenset(task.value for task in AgentTask)
+
+
+class TaskHandler:
+    """Validates and wraps a tool function for one agent task."""
+
+    task: ClassVar[AgentTask]
+    label: ClassVar[str]
+
+    def extra_meta(self) -> dict[str, Any]:
+        """Return task-specific ``_meta.mcp_server`` fields."""
+        return {}
+
+    def check(self, fn: Callable[..., Any]) -> None:
+        """Validate *fn*'s return annotation. Override in subclasses."""
+
+    def wrap(self, fn: Callable[..., Any]) -> Callable[..., Any]:
+        """Wrap *fn* to validate its runtime return value. Override in subclasses."""
+        return fn
+
+
+def require_return_type(fn: Callable[..., Any], expected: type | tuple[type, ...], *, label: str) -> None:
+    """Raise TypeError when *fn* is not annotated with *expected*."""
+    annotation = return_annotation(fn)
+    if annotation is inspect.Signature.empty:
+        raise TypeError(f'{label} {fn.__name__!r} must declare a return type of {expected}.')
+    if not annotation_matches(annotation, expected):
+        raise TypeError(f'{label} {fn.__name__!r} must return {expected}, got {annotation}.')
+
+
+def wrap_checked(fn: Callable[..., Any], check: _Check) -> Callable[..., Any]:
+    """Wrap *fn* so *check* runs on its return value (sync or async)."""
+    if inspect.iscoroutinefunction(fn):
+
+        @wraps(fn)
+        async def async_wrapped(*args: Any, **kwargs: Any) -> Any:
+            return check(await fn(*args, **kwargs))
+
+        return async_wrapped
+
+    @wraps(fn)
+    def synced(*args: Any, **kwargs: Any) -> Any:
+        return check(fn(*args, **kwargs))
+
+    return synced
+
+
+def return_annotation(fn: Callable[..., Any]) -> Any:
+    """Resolve *fn*'s return annotation, preferring ``get_type_hints``."""
+    try:
+        hints = get_type_hints(fn)
+    except (TypeError, NameError, ValueError):
+        hints = {}
+    if 'return' in hints:
+        return hints['return']
+    return inspect.signature(fn).return_annotation
+
+
+def annotation_matches(annotation: Any, expected: type | tuple[type, ...]) -> bool:
+    """Return True when *annotation* is one of *expected* (including ``None``)."""
+    expected_types = expected if isinstance(expected, tuple) else (expected,)
+    if annotation in expected_types:
+        return True
+    return expected_types == (type(None),) and annotation in (None, type(None))
+
+
+def stamp_task_kwargs(
+    kwargs: dict[str, Any],
+    task: AgentTask | str,
+    extra: dict[str, Any] | None = None,
+) -> dict[str, Any]:
+    """Copy decorator kwargs and stamp task tags plus ``_meta.mcp_server``."""
+    value = task.value if isinstance(task, AgentTask) else task
+    stamped = dict(kwargs)
+    tags = set(stamped.get('tags') or set())
+    tags.add(value)
+    stamped['tags'] = tags
+    meta = dict(stamped.get('meta') or {})
+    payload = dict(meta.get(META_NAMESPACE) or {})
+    payload['task'] = value
+    if extra:
+        payload.update(extra)
+    meta[META_NAMESPACE] = payload
+    stamped['meta'] = meta
+    return stamped
+
+
+def task_from_meta(
+    meta: dict[str, Any] | None,
+    tags: Iterable[str] | None = None,
+) -> AgentTask:
+    """Read the agent task from MCP tool meta, then FastMCP tags, else completion."""
+    if meta:
+        payload = meta.get(META_NAMESPACE)
+        if isinstance(payload, dict):
+            raw_task = payload.get('task')
+            if isinstance(raw_task, str):
+                if raw_task in AGENT_TASK_VALUES:
+                    return AgentTask(raw_task)
+                return AgentTask.NONE
+        fastmcp_meta = meta.get('fastmcp')
+        if isinstance(fastmcp_meta, dict):
+            tag_list = fastmcp_meta.get('tags')
+            if isinstance(tag_list, list):
+                tagged = _task_from_tags(tag_list)
+                if tagged is not None:
+                    return tagged
+    if tags is not None:
+        tagged = _task_from_tags(tags)
+        if tagged is not None:
+            return tagged
+    return AgentTask.COMPLETION
+
+
+def tools_for_task(tools: Sequence[Any], task: AgentTask | str) -> list[Any]:
+    """Filter MCP or FastMCP tool objects whose stamped agent task matches *task*."""
+    wanted = AgentTask(task)
+    return [tool for tool in tools if tool_task(tool) is wanted]
+
+
+def tool_task(tool: Any) -> AgentTask:
+    """Read the agent task from a FastMCP component or MCP SDK Tool."""
+    meta = getattr(tool, 'meta', None)
+    get_meta = getattr(tool, 'get_meta', None)
+    if callable(get_meta):
+        resolved = get_meta()
+        if isinstance(resolved, dict):
+            meta = resolved
+    tags = getattr(tool, 'tags', None)
+    return task_from_meta(meta if isinstance(meta, dict) else None, tags)
+
+
+def _task_from_tags(tags: Iterable[str]) -> AgentTask | None:
+    for tag in tags:
+        if tag in AGENT_TASK_VALUES:
+            return AgentTask(tag)
+    return None

brawler_mcp_server_python-0.1.0.dist-info/METADATA

@@ -0,0 +1,229 @@
+Metadata-Version: 2.5
+Name: brawler-mcp-server-python
+Version: 0.1.0
+Summary: FastMCP wrapper that tags agent task tools for Brawler and other MCP clients.
+Project-URL: Homepage, https://github.com/DanielFloresDiaz/brawler-mcp-server-python
+Project-URL: Source, https://github.com/DanielFloresDiaz/brawler-mcp-server-python
+Project-URL: Documentation, https://github.com/DanielFloresDiaz/brawler-mcp-server-python#readme
+Author-email: Daniel Flores Díaz <danielfloresdiaz17@gmail.com>
+License: Apache-2.0
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Requires-Dist: fastmcp<5,>=4.1.1
+Description-Content-Type: text/markdown
+
+# Brawler MCP Server (Python)
+
+FastMCP wrapper that tags tools with **agent tasks** so a client can list and run context, completion, evaluation, post, and none tools from one MCP server.
+
+This is not a fork of [FastMCP](https://github.com/PrefectHQ/fastmcp). It depends on FastMCP 4 and subclasses `FastMCP`. Later language ports (`brawler-mcp-server-typescript`, and so on) should implement the same wire contract.
+
+Brawler agents attach one MCP server as a resource, then fetch tools by task: context tools build the prompt, completion tools are offered to the model, post and eval tools run after a result. Native Brawler skills stay on the agent; MCP does not offer a skill task.
+
+---
+
+## What this adds on top of FastMCP
+
+| FastMCP | This package |
+| --- | --- |
+| `@mcp.tool` | Same decorator, stamped as **completion** |
+| No task model | `@context_tool`, `@eval_tool`, `@post_tool`, `@none_tool` |
+| Tags are optional | Every task decorator sets a task tag and `_meta.mcp_server` |
+| `tools/list` returns a flat list | Clients filter by `_meta.mcp_server.task` (or call `tools_by_task`) |
+
+Untagged tools from a plain FastMCP server are treated as **completion** by Brawler. Tools from this package always carry an explicit task. FastMCP tags that are not agent tasks (for example `skill`, `router`, `training`) are ignored when resolving the task; an unknown ``mcp_server.task`` value is treated as **none**.
+
+---
+
+## Install
+
+```bash
+uv add brawler-mcp-server-python
+# or
+pip install brawler-mcp-server-python
+```
+
+Requires Python 3.10+ and FastMCP 4.1.3+.
+
+---
+
+## Quick start
+
+```python
+from brawler_mcp_server import BrawlerMCP
+
+mcp = BrawlerMCP('Docs')
+
+
+@mcp.context_tool
+def policies() -> str:
+    """Injected into the agent system prompt."""
+    return 'Always cite sources.'
+
+
+@mcp.context_tool(as_user_message=True)
+def ticket() -> str:
+    return 'Ticket #123 is still open.'
+
+
+@mcp.tool
+def search(q: str) -> str:
+    """Completion tool offered to the model."""
+    return f'Results for {q}'
+
+
+@mcp.eval_tool
+def grounded(answer: str) -> tuple[bool, float, str | None]:
+    if not answer.strip():
+        return False, 0.0, 'Answer is empty.'
+    return True, 1.0, None
+
+
+@mcp.post_tool
+def audit(answer: str) -> None:
+    return None
+
+
+@mcp.none_tool
+def ping() -> str:
+    """Visible to generic MCP clients; Brawler agents ignore none tools."""
+    return 'pong'
+
+
+if __name__ == '__main__':
+    mcp.run()
+```
+
+`mcp.run()` speaks stdio by default. Use FastMCP HTTP transports when you need a remote endpoint.
+
+Example server shipped in ``examples/``:
+
+```bash
+uv run brawler-mcp-example
+```
+
+---
+
+## Wire contract
+
+MCP `tools/list` has no task filter. Identity lives on each tool so any language client can list a subset after listing.
+
+```json
+{
+  "name": "policies",
+  "description": "Injected into the agent system prompt.",
+  "_meta": {
+    "mcp_server": {
+      "task": "context",
+      "as_user_message": false
+    },
+    "fastmcp": {
+      "tags": ["context"]
+    }
+  }
+}
+```
+
+### `_meta.mcp_server`
+
+| Field | Type | When |
+| --- | --- | --- |
+| `task` | `context` \| `completion` \| `evaluation` \| `post_tool` \| `none` | Always |
+| `as_user_message` | `bool` | Context only. `true` injects the string as a user message instead of the system prompt. |
+
+FastMCP also copies the task into `_meta.fastmcp.tags`. Clients may use either field. If both are missing, treat the tool as `completion`. If `task` is set to an unsupported value, treat the tool as `none`.
+
+### Listing only context tools
+
+```python
+from fastmcp import Client
+from brawler_mcp_server import tools_for_task
+
+async with Client(mcp) as client:
+    tools = await client.list_tools()
+    context_tools = tools_for_task(tools, 'context')
+```
+
+In-process (no MCP round-trip):
+
+```python
+context_tools = await mcp.tools_by_task('context')
+```
+
+---
+
+## Return types
+
+| Decorator | Python return | On the wire |
+| --- | --- | --- |
+| `@context_tool` | `str` | Single text content |
+| `@eval_tool` | `tuple[bool, float, str \| None]` | JSON `{"is_valid": bool, "score": float, "feedback": str \| null}` |
+| `@post_tool` | `None` | Empty / ignored text |
+| `@tool` / `@none_tool` | FastMCP default | FastMCP default |
+
+Decorators check the annotation at registration time.
+
+---
+
+## Agent tasks (this package)
+
+Router, training, and skill are not included. Router and training are not function-tool tasks on a Brawler agent. Skills are native Brawler tools, not MCP tasks.
+
+| Task | Decorator | Agent use |
+| --- | --- | --- |
+| `context` | `@context_tool` | Prompt / user-message injection |
+| `completion` | `@tool` | Model-callable tools |
+| `evaluation` | `@eval_tool` | After the final result |
+| `post_tool` | `@post_tool` | Side effects after the result, before eval |
+| `none` | `@none_tool` | Not used by Brawler agents |
+
+---
+
+## Using with Brawler
+
+Register the server as an MCP resource on an agent the same way as any other MCP connection (`MCPClient.from_server` / `agent.mcp(...)` / `POST /mcps`). The persisted row stays `entity_type=mcp` with `task=completion` (the connection). Remote tools carry the task in `_meta.mcp_server`.
+
+At run time the agent lists tools, keeps completion tools for the model, and calls context / post / eval tools at the matching pipeline stage.
+
+---
+
+## Development
+
+```bash
+uv sync --all-extras --group lint --group dev
+uv run pytest tests/
+make test
+make typecheck
+```
+
+If this repo is checked out as a submodule of brawler, `uv run` still uses this
+package's own environment. Do not run these tests with brawler's FastMCP 3 venv.
+
+CI (`.github/workflows/testing.yaml`) runs lint, pyright, and pytest on pull requests to `main`. The release workflow (`.github/workflows/release.yaml`) runs the same tests, builds the package, creates a GitHub release from `changelogs/v<tag>.md`, and publishes to PyPI when a version tag is pushed.
+
+### Releasing
+
+1. Bump `version` in `pyproject.toml`.
+2. Add `changelogs/v<tag>.md` and update `changelogs/latest.md` to match (tag includes the `v` prefix, e.g. `v0.1.0`).
+3. Merge to `main`, then tag and push: `git tag v0.1.0 && git push origin v0.1.0`.
+
+Before the first PyPI publish, register a **trusted publisher** on [pypi.org](https://pypi.org) for project `brawler-mcp-server-python`: owner `DanielFloresDiaz`, repository `brawler-mcp-server-python`, workflow name `release.yaml`, environment blank. No API token secret is required.
+
+After merging workflow changes, smoke-test locally with `uv build` and inspect `dist/`, then confirm the tag workflow uploads GitHub release assets and the new version appears on PyPI.
+
+---
+
+## License
+
+Apache-2.0

brawler_mcp_server_python-0.1.0.dist-info/RECORD

@@ -0,0 +1,18 @@
+brawler_mcp_server/__init__.py,sha256=wqiONerd3p6ocF6oeSI7A05F0ldEOoYBubJnTNh1PIo,329
+brawler_mcp_server/server.py,sha256=jh0Dakq9-7tZf73UONOQfmbHfa9cbkclukNE3_b2-uo,4287
+brawler_mcp_server/testkit.py,sha256=GGd0KuKKnVEZWbhWVNRKAPr7ahSE22wd-xdi9oYHjTw,2252
+brawler_mcp_server/utils.py,sha256=tmGjvMoTbZ5BGVf0Bu3GrizfE-wfP8sq8aVXGym8Cr4,5413
+brawler_mcp_server/tasks/__init__.py,sha256=ZB4O96IXtZVVFIIhShLi-ywKQkJ1fTws5_r4nSCe9zc,500
+brawler_mcp_server/tasks/completion.py,sha256=iva9p30IWPdgWPBbZTrpIAJ6Y0X4JvgpZjGd65k4gE0,347
+brawler_mcp_server/tasks/context.py,sha256=63IU2u7J_Rj19cbYbfjxqPRf0WEvnUJSfmKZQlA2M_w,1117
+brawler_mcp_server/tasks/evaluation.py,sha256=vg0KN04lP8fd8NZxEFTI6P0-awgTChZQY9FnHsVfXPw,2992
+brawler_mcp_server/tasks/none.py,sha256=6SQbYn5GE3wf7Js-dLYbNK-CSEZ7cORKUTJW9XFiE6M,312
+brawler_mcp_server/tasks/post.py,sha256=befhjztw3lMZmU-eDp0Nswj4WAPa5XTXiKK4dA4mZAE,888
+examples/__init__.py,sha256=XbUnXEsT4S9erJrG_u4lwpRajwgN-WJiRVb7yKPe-mY,44
+examples/example.py,sha256=FSnYmKATfaJTVSpyowFU5aCNuOohoMFeFzcb26apFD8,1190
+examples/task_server.py,sha256=ZqVJBcSHKie8LgLaLCBon1SjTioy2_iBYZ8-JsnlJV4,4450
+brawler_mcp_server_python-0.1.0.dist-info/METADATA,sha256=N9COnB1TKD4nq19oKxdbOEMHaXT-bbmp_9skCFJJJDM,7832
+brawler_mcp_server_python-0.1.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+brawler_mcp_server_python-0.1.0.dist-info/entry_points.txt,sha256=DD4dggxB1kvYPFxdkDd1X-7dMfLy4dmTJh2XHgjoAZs,114
+brawler_mcp_server_python-0.1.0.dist-info/licenses/LICENSE,sha256=VTizCijyEqDndSBoZc5qmvYz0LtSNq-ZGW5HQw0PH9o,11349
+brawler_mcp_server_python-0.1.0.dist-info/RECORD,,

brawler_mcp_server_python-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.4
+Root-Is-Purelib: true
+Tag: py3-none-any

brawler_mcp_server_python-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+brawler-mcp-example = examples.example:main
+brawler-mcp-task-server = examples.task_server:main

brawler_mcp_server_python-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2026 Daniel Flores Díaz
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

examples/__init__.py

@@ -0,0 +1 @@
+"""Example MCP servers for this package."""

examples/example.py

@@ -0,0 +1,54 @@
+"""Example stdio server used by package tests and Brawler integration tests."""
+
+from __future__ import annotations
+
+from brawler_mcp_server.server import BrawlerMCP
+
+mcp = BrawlerMCP('BrawlerExample')
+
+
+@mcp.context_tool
+def policies() -> str:
+    """Company policy injected into the agent system prompt."""
+    return 'Always cite sources.'
+
+
+@mcp.context_tool(as_user_message=True)
+def open_ticket() -> str:
+    """Ticket reminder injected as a user message."""
+    return 'Ticket #123 is still open.'
+
+
+@mcp.tool
+def echo(text: str) -> str:
+    """Echo the given text back to the model."""
+    return text
+
+
+@mcp.eval_tool
+def grounded(answer: str) -> tuple[bool, float, str | None]:
+    """Reject empty answers."""
+    if not str(answer).strip():
+        return False, 0.0, 'Answer is empty.'
+    return True, 1.0, None
+
+
+@mcp.post_tool
+def audit(answer: str) -> None:
+    """No-op post tool used to verify post-task listing and calls."""
+    return
+
+
+@mcp.none_tool
+def ping() -> str:
+    """Generic MCP ping; Brawler agents skip none tools."""
+    return 'pong'
+
+
+def main() -> None:
+    """Run the example server over stdio."""
+    mcp.run()
+
+
+if __name__ == '__main__':
+    main()

examples/task_server.py

@@ -0,0 +1,170 @@
+"""Stdio MCP server used by Brawler's real-env MCPTool tests.
+
+Registers at least one tool per agent task, plus untagged / unsupported tools so
+the client can exercise task detection from ``_meta`` and FastMCP tags.
+"""
+
+from __future__ import annotations
+
+import json
+from typing import TypedDict
+
+from fastmcp import FastMCP
+
+from brawler_mcp_server.server import BrawlerMCP
+
+
+class AddResult(TypedDict):
+    """Structured sum returned by the ``add`` completion tool."""
+
+    sum: int
+
+
+mcp = BrawlerMCP('BrawlerTaskServer')
+
+
+@mcp.context_tool
+def policies() -> str:
+    """Company policy injected into the agent system prompt."""
+    return 'Always cite sources.'
+
+
+@mcp.context_tool(as_user_message=True)
+def open_ticket() -> str:
+    """Ticket reminder injected as a user message."""
+    return 'Ticket #123 is still open.'
+
+
+@mcp.context_tool
+def locale(language: str) -> str:
+    """Locale hint that takes a language code."""
+    return f'Reply in {language}.'
+
+
+@mcp.context_tool
+def boom() -> str:
+    """Context tool that fails so the client can observe a swallowed error."""
+    raise RuntimeError('context failed')
+
+
+@mcp.tool
+def echo(text: str) -> str:
+    """Echo the given text back to the model."""
+    return text
+
+
+@mcp.tool
+def add(a: int, b: int) -> AddResult:
+    """Add two integers and return structured output."""
+    return {'sum': a + b}
+
+
+@mcp.tool
+def fail(text: str) -> str:
+    """Completion tool that raises so run_tool_call can observe a remote error."""
+    raise RuntimeError(f'remote failed: {text}')
+
+
+@mcp.eval_tool
+def grounded(answer: str) -> tuple[bool, float, str | None]:
+    """Reject empty answers."""
+    if not str(answer).strip():
+        return False, 0.0, 'Answer is empty.'
+    return True, 1.0, None
+
+
+@mcp.eval_tool
+def score_length(answer: str) -> tuple[bool, float, str | None]:
+    """Score a non-empty answer by character length."""
+    stripped = str(answer).strip()
+    if len(stripped) < 3:
+        return False, 0.25, 'Answer is too short.'
+    return True, 0.9, 'ok'
+
+
+@mcp.post_tool
+def audit(answer: str) -> None:
+    """No-op post tool used to verify post-task listing and calls."""
+    return
+
+
+@mcp.post_tool
+def record(value: int, query: str) -> None:
+    """Post tool with multiple input fields."""
+    return
+
+
+@mcp.none_tool
+def ping() -> str:
+    """Generic MCP ping; Brawler agents skip none tools."""
+    return 'pong'
+
+
+def legacy_search(q: str) -> str:
+    """Untagged FastMCP tool; Brawler treats it as completion."""
+    return f'results for {q}'
+
+
+def style() -> str:
+    """Skill-tagged tool; Brawler treats unsupported tasks as none."""
+    return 'Prefer short sentences.'
+
+
+def route(query: str) -> str:
+    """Router-tagged tool; Brawler treats unsupported tasks as none."""
+    return 'routed'
+
+
+def train() -> str:
+    """Training-tagged tool; Brawler treats unsupported tasks as none."""
+    return 'train'
+
+
+def unknown_task(q: str) -> str:
+    """``mcp_server.task`` that is not a Task value defaults to completion."""
+    return f'unknown:{q}'
+
+
+def tagged_eval(answer: str) -> str:
+    """Evaluation via FastMCP tags only (no ``mcp_server.task``)."""
+    return json.dumps({'is_valid': True, 'score': 1.0, 'feedback': None})
+
+
+def broken_eval(answer: str) -> str:
+    """Evaluation tool that returns non-object text."""
+    return 'not-json'
+
+
+def valid_no_score(answer: str) -> str:
+    """Evaluation tool that omits ``score`` so the client can default it."""
+    return json.dumps({'is_valid': True, 'feedback': 'fine'})
+
+
+def _plain_tool(
+    fn: object,
+    *,
+    name: str,
+    tags: set[str] | None = None,
+    meta: dict[str, object] | None = None,
+) -> None:
+    """Register a FastMCP tool without BrawlerMCP's task stamp."""
+    FastMCP.tool(mcp, name=name, tags=tags, meta=meta)(fn)
+
+
+_plain_tool(legacy_search, name='legacy_search')
+_plain_tool(style, name='style', tags={'skill'})
+_plain_tool(route, name='route', tags={'router'})
+_plain_tool(train, name='train', tags={'training'})
+_plain_tool(unknown_task, name='unknown_task', meta={'mcp_server': {'task': 'not-a-task'}})
+_plain_tool(tagged_eval, name='tagged_eval', tags={'evaluation'})
+_plain_tool(broken_eval, name='broken_eval', meta={'mcp_server': {'task': 'evaluation'}})
+_plain_tool(valid_no_score, name='valid_no_score', meta={'mcp_server': {'task': 'evaluation'}})
+
+
+def main() -> None:
+    """Run the task server over stdio."""
+    mcp.run()
+
+
+if __name__ == '__main__':
+    main()