Upgrade MCP and add ask_user

Author: simba-git

docs/api/context.md

@@ -40,6 +40,22 @@ result = await ctx.ask_llm(
 print(result.content.text)
 ```
 
+## User Elicitation
+
+Call `ask_user()` when you need additional input from the client. It wraps the
+underlying MCP `elicit()` API:
+
+```python
+class BookingPreferences(BaseModel):
+    alternativeDate: str | None
+    checkAlternative: bool = False
+
+result = await ctx.ask_user(
+    message="No tables available. Try another date?",
+    schema=BookingPreferences,
+)
+```
+
 ## Extending Context
 
 For now, if you need context functionality, you can extend the base class:

src/enrichmcp/context.py

@@ -6,6 +6,7 @@
 
 from typing import Literal
 
+from mcp.server.elicitation import ElicitationResult, ElicitSchemaModelT
 from mcp.server.fastmcp import Context  # pyright: ignore[reportMissingTypeArgument]
 from mcp.types import (
     CreateMessageResult,
@@ -107,6 +108,15 @@ async def sampling(
 
         return await self.ask_llm(messages, **kwargs)
 
+    async def ask_user(
+        self,
+        message: str,
+        schema: type[ElicitSchemaModelT],
+    ) -> ElicitationResult:
+        """Interactively ask the client for input using MCP elicitation."""
+
+        return await super().elicit(message=message, schema=schema)
+
 
 def prefer_fast_model() -> ModelPreferences:
     """Model preferences optimized for speed and cost."""

tests/test_elicitation.py

@@ -0,0 +1,28 @@
+from unittest.mock import AsyncMock, Mock, patch
+
+import pytest
+from pydantic import BaseModel
+
+from enrichmcp import EnrichContext
+
+
+class Prefs(BaseModel):
+    choice: bool
+
+
+@pytest.mark.asyncio
+async def test_ask_user_delegates_to_context_elicit():
+    ctx = EnrichContext.model_construct(_request_context=Mock())
+
+    with patch("enrichmcp.context.Context.elicit", AsyncMock(return_value="ok")) as mock:
+        got = await ctx.ask_user("hi", Prefs)
+        assert got == "ok"
+        mock.assert_awaited_once_with(message="hi", schema=Prefs)
+
+
+@pytest.mark.asyncio
+async def test_ask_user_requires_request_context():
+    ctx = EnrichContext()
+
+    with pytest.raises(ValueError, match="outside of a request"):
+        await ctx.ask_user("hi", Prefs)