Add get_context helper (#102)

* Add get_context method and tests

* Fix get_context instantiation

* docs: document get_context

Author: simba-git

CHANGELOG.md

@@ -6,6 +6,9 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
+### Added
+- `EnrichMCP.get_context()` returns an `EnrichContext` instance mirroring
+  the underlying FastMCP context.
 
 ## [0.4.4] - 2025-06-26
 

docs/api/app.md

@@ -95,6 +95,16 @@ async def delete_user(uid: int) -> bool:
     ...
 ```
 
+### `get_context() -> EnrichContext`
+
+Return the current request context as an :class:`~enrichmcp.EnrichContext`.
+
+```python
+app = EnrichMCP("My API", description="desc")
+ctx = app.get_context()
+assert ctx.fastmcp is app.mcp
+```
+
 ### `run(**options)`
 
 Start the MCP server.

src/enrichmcp/app.py

@@ -20,6 +20,7 @@
 from mcp.server.fastmcp import FastMCP
 from pydantic import BaseModel, Field, create_model
 
+from .context import EnrichContext
 from .entity import EnrichModel
 from .relationship import Relationship
 
@@ -428,6 +429,15 @@ def decorator(fn: Callable[..., Any]) -> Callable[..., Any]:
             return decorator(func)
         return cast("DecoratorCallable", decorator)
 
+    def get_context(self) -> EnrichContext:
+        """Return the current :class:`EnrichContext` for this app."""
+
+        base_ctx = self.mcp.get_context()
+        return EnrichContext.model_construct(
+            _request_context=getattr(base_ctx, "_request_context", None),
+            _fastmcp=getattr(base_ctx, "_fastmcp", None),
+        )
+
     def run(
         self, *, transport: str | None = None, mount_path: str | None = None, **options: Any
     ) -> Any:

tests/test_core.py

@@ -1,7 +1,10 @@
+from unittest.mock import patch
+
 import pytest
 from pydantic import Field
 
 from enrichmcp import (
+    EnrichContext,
     EnrichMCP,
     EnrichModel,
 )
@@ -153,3 +156,26 @@ async def test_entity_without_description_fails():
         class BadEntity(EnrichModel):
             # No docstring, should fail
             id: int = Field(description="ID")
+
+
+def test_get_context_returns_enrich_context():
+    """app.get_context should return an EnrichContext"""
+
+    app = EnrichMCP("Test API", description="Test API description")
+    ctx = app.get_context()
+
+    assert isinstance(ctx, EnrichContext)
+    assert ctx.fastmcp is app.mcp
+
+    with pytest.raises(ValueError):
+        _ = ctx.request_context
+
+
+def test_get_context_propagates_errors():
+    app = EnrichMCP("Test API", description="desc")
+
+    with (
+        patch.object(app.mcp, "get_context", side_effect=RuntimeError("boom")),
+        pytest.raises(RuntimeError),
+    ):
+        app.get_context()