PyPI - fast-agent-mcp - Versions diffs - 0.4.7__py3-none-any.whl - Mend

fast-agent-mcp 0.4.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (261) hide show

fast_agent/__init__.py +183 -0
fast_agent/acp/__init__.py +19 -0
fast_agent/acp/acp_aware_mixin.py +304 -0
fast_agent/acp/acp_context.py +437 -0
fast_agent/acp/content_conversion.py +136 -0
fast_agent/acp/filesystem_runtime.py +427 -0
fast_agent/acp/permission_store.py +269 -0
fast_agent/acp/server/__init__.py +5 -0
fast_agent/acp/server/agent_acp_server.py +1472 -0
fast_agent/acp/slash_commands.py +1050 -0
fast_agent/acp/terminal_runtime.py +408 -0
fast_agent/acp/tool_permission_adapter.py +125 -0
fast_agent/acp/tool_permissions.py +474 -0
fast_agent/acp/tool_progress.py +814 -0
fast_agent/agents/__init__.py +85 -0
fast_agent/agents/agent_types.py +64 -0
fast_agent/agents/llm_agent.py +350 -0
fast_agent/agents/llm_decorator.py +1139 -0
fast_agent/agents/mcp_agent.py +1337 -0
fast_agent/agents/tool_agent.py +271 -0
fast_agent/agents/workflow/agents_as_tools_agent.py +849 -0
fast_agent/agents/workflow/chain_agent.py +212 -0
fast_agent/agents/workflow/evaluator_optimizer.py +380 -0
fast_agent/agents/workflow/iterative_planner.py +652 -0
fast_agent/agents/workflow/maker_agent.py +379 -0
fast_agent/agents/workflow/orchestrator_models.py +218 -0
fast_agent/agents/workflow/orchestrator_prompts.py +248 -0
fast_agent/agents/workflow/parallel_agent.py +250 -0
fast_agent/agents/workflow/router_agent.py +353 -0
fast_agent/cli/__init__.py +0 -0
fast_agent/cli/__main__.py +73 -0
fast_agent/cli/commands/acp.py +159 -0
fast_agent/cli/commands/auth.py +404 -0
fast_agent/cli/commands/check_config.py +783 -0
fast_agent/cli/commands/go.py +514 -0
fast_agent/cli/commands/quickstart.py +557 -0
fast_agent/cli/commands/serve.py +143 -0
fast_agent/cli/commands/server_helpers.py +114 -0
fast_agent/cli/commands/setup.py +174 -0
fast_agent/cli/commands/url_parser.py +190 -0
fast_agent/cli/constants.py +40 -0
fast_agent/cli/main.py +115 -0
fast_agent/cli/terminal.py +24 -0
fast_agent/config.py +798 -0
fast_agent/constants.py +41 -0
fast_agent/context.py +279 -0
fast_agent/context_dependent.py +50 -0
fast_agent/core/__init__.py +92 -0
fast_agent/core/agent_app.py +448 -0
fast_agent/core/core_app.py +137 -0
fast_agent/core/direct_decorators.py +784 -0
fast_agent/core/direct_factory.py +620 -0
fast_agent/core/error_handling.py +27 -0
fast_agent/core/exceptions.py +90 -0
fast_agent/core/executor/__init__.py +0 -0
fast_agent/core/executor/executor.py +280 -0
fast_agent/core/executor/task_registry.py +32 -0
fast_agent/core/executor/workflow_signal.py +324 -0
fast_agent/core/fastagent.py +1186 -0
fast_agent/core/logging/__init__.py +5 -0
fast_agent/core/logging/events.py +138 -0
fast_agent/core/logging/json_serializer.py +164 -0
fast_agent/core/logging/listeners.py +309 -0
fast_agent/core/logging/logger.py +278 -0
fast_agent/core/logging/transport.py +481 -0
fast_agent/core/prompt.py +9 -0
fast_agent/core/prompt_templates.py +183 -0
fast_agent/core/validation.py +326 -0
fast_agent/event_progress.py +62 -0
fast_agent/history/history_exporter.py +49 -0
fast_agent/human_input/__init__.py +47 -0
fast_agent/human_input/elicitation_handler.py +123 -0
fast_agent/human_input/elicitation_state.py +33 -0
fast_agent/human_input/form_elements.py +59 -0
fast_agent/human_input/form_fields.py +256 -0
fast_agent/human_input/simple_form.py +113 -0
fast_agent/human_input/types.py +40 -0
fast_agent/interfaces.py +310 -0
fast_agent/llm/__init__.py +9 -0
fast_agent/llm/cancellation.py +22 -0
fast_agent/llm/fastagent_llm.py +931 -0
fast_agent/llm/internal/passthrough.py +161 -0
fast_agent/llm/internal/playback.py +129 -0
fast_agent/llm/internal/silent.py +41 -0
fast_agent/llm/internal/slow.py +38 -0
fast_agent/llm/memory.py +275 -0
fast_agent/llm/model_database.py +490 -0
fast_agent/llm/model_factory.py +388 -0
fast_agent/llm/model_info.py +102 -0
fast_agent/llm/prompt_utils.py +155 -0
fast_agent/llm/provider/anthropic/anthropic_utils.py +84 -0
fast_agent/llm/provider/anthropic/cache_planner.py +56 -0
fast_agent/llm/provider/anthropic/llm_anthropic.py +796 -0
fast_agent/llm/provider/anthropic/multipart_converter_anthropic.py +462 -0
fast_agent/llm/provider/bedrock/bedrock_utils.py +218 -0
fast_agent/llm/provider/bedrock/llm_bedrock.py +2207 -0
fast_agent/llm/provider/bedrock/multipart_converter_bedrock.py +84 -0
fast_agent/llm/provider/google/google_converter.py +466 -0
fast_agent/llm/provider/google/llm_google_native.py +681 -0
fast_agent/llm/provider/openai/llm_aliyun.py +31 -0
fast_agent/llm/provider/openai/llm_azure.py +143 -0
fast_agent/llm/provider/openai/llm_deepseek.py +76 -0
fast_agent/llm/provider/openai/llm_generic.py +35 -0
fast_agent/llm/provider/openai/llm_google_oai.py +32 -0
fast_agent/llm/provider/openai/llm_groq.py +42 -0
fast_agent/llm/provider/openai/llm_huggingface.py +85 -0
fast_agent/llm/provider/openai/llm_openai.py +1195 -0
fast_agent/llm/provider/openai/llm_openai_compatible.py +138 -0
fast_agent/llm/provider/openai/llm_openrouter.py +45 -0
fast_agent/llm/provider/openai/llm_tensorzero_openai.py +128 -0
fast_agent/llm/provider/openai/llm_xai.py +38 -0
fast_agent/llm/provider/openai/multipart_converter_openai.py +561 -0
fast_agent/llm/provider/openai/openai_multipart.py +169 -0
fast_agent/llm/provider/openai/openai_utils.py +67 -0
fast_agent/llm/provider/openai/responses.py +133 -0
fast_agent/llm/provider_key_manager.py +139 -0
fast_agent/llm/provider_types.py +34 -0
fast_agent/llm/request_params.py +61 -0
fast_agent/llm/sampling_converter.py +98 -0
fast_agent/llm/stream_types.py +9 -0
fast_agent/llm/usage_tracking.py +445 -0
fast_agent/mcp/__init__.py +56 -0
fast_agent/mcp/common.py +26 -0
fast_agent/mcp/elicitation_factory.py +84 -0
fast_agent/mcp/elicitation_handlers.py +164 -0
fast_agent/mcp/gen_client.py +83 -0
fast_agent/mcp/helpers/__init__.py +36 -0
fast_agent/mcp/helpers/content_helpers.py +352 -0
fast_agent/mcp/helpers/server_config_helpers.py +25 -0
fast_agent/mcp/hf_auth.py +147 -0
fast_agent/mcp/interfaces.py +92 -0
fast_agent/mcp/logger_textio.py +108 -0
fast_agent/mcp/mcp_agent_client_session.py +411 -0
fast_agent/mcp/mcp_aggregator.py +2175 -0
fast_agent/mcp/mcp_connection_manager.py +723 -0
fast_agent/mcp/mcp_content.py +262 -0
fast_agent/mcp/mime_utils.py +108 -0
fast_agent/mcp/oauth_client.py +509 -0
fast_agent/mcp/prompt.py +159 -0
fast_agent/mcp/prompt_message_extended.py +155 -0
fast_agent/mcp/prompt_render.py +84 -0
fast_agent/mcp/prompt_serialization.py +580 -0
fast_agent/mcp/prompts/__init__.py +0 -0
fast_agent/mcp/prompts/__main__.py +7 -0
fast_agent/mcp/prompts/prompt_constants.py +18 -0
fast_agent/mcp/prompts/prompt_helpers.py +238 -0
fast_agent/mcp/prompts/prompt_load.py +186 -0
fast_agent/mcp/prompts/prompt_server.py +552 -0
fast_agent/mcp/prompts/prompt_template.py +438 -0
fast_agent/mcp/resource_utils.py +215 -0
fast_agent/mcp/sampling.py +200 -0
fast_agent/mcp/server/__init__.py +4 -0
fast_agent/mcp/server/agent_server.py +613 -0
fast_agent/mcp/skybridge.py +44 -0
fast_agent/mcp/sse_tracking.py +287 -0
fast_agent/mcp/stdio_tracking_simple.py +59 -0
fast_agent/mcp/streamable_http_tracking.py +309 -0
fast_agent/mcp/tool_execution_handler.py +137 -0
fast_agent/mcp/tool_permission_handler.py +88 -0
fast_agent/mcp/transport_tracking.py +634 -0
fast_agent/mcp/types.py +24 -0
fast_agent/mcp/ui_agent.py +48 -0
fast_agent/mcp/ui_mixin.py +209 -0
fast_agent/mcp_server_registry.py +89 -0
fast_agent/py.typed +0 -0
fast_agent/resources/examples/data-analysis/analysis-campaign.py +189 -0
fast_agent/resources/examples/data-analysis/analysis.py +68 -0
fast_agent/resources/examples/data-analysis/fastagent.config.yaml +41 -0
fast_agent/resources/examples/data-analysis/mount-point/WA_Fn-UseC_-HR-Employee-Attrition.csv +1471 -0
fast_agent/resources/examples/mcp/elicitations/elicitation_account_server.py +88 -0
fast_agent/resources/examples/mcp/elicitations/elicitation_forms_server.py +297 -0
fast_agent/resources/examples/mcp/elicitations/elicitation_game_server.py +164 -0
fast_agent/resources/examples/mcp/elicitations/fastagent.config.yaml +35 -0
fast_agent/resources/examples/mcp/elicitations/fastagent.secrets.yaml.example +17 -0
fast_agent/resources/examples/mcp/elicitations/forms_demo.py +107 -0
fast_agent/resources/examples/mcp/elicitations/game_character.py +65 -0
fast_agent/resources/examples/mcp/elicitations/game_character_handler.py +256 -0
fast_agent/resources/examples/mcp/elicitations/tool_call.py +21 -0
fast_agent/resources/examples/mcp/state-transfer/agent_one.py +18 -0
fast_agent/resources/examples/mcp/state-transfer/agent_two.py +18 -0
fast_agent/resources/examples/mcp/state-transfer/fastagent.config.yaml +27 -0
fast_agent/resources/examples/mcp/state-transfer/fastagent.secrets.yaml.example +15 -0
fast_agent/resources/examples/researcher/fastagent.config.yaml +61 -0
fast_agent/resources/examples/researcher/researcher-eval.py +53 -0
fast_agent/resources/examples/researcher/researcher-imp.py +189 -0
fast_agent/resources/examples/researcher/researcher.py +36 -0
fast_agent/resources/examples/tensorzero/.env.sample +2 -0
fast_agent/resources/examples/tensorzero/Makefile +31 -0
fast_agent/resources/examples/tensorzero/README.md +56 -0
fast_agent/resources/examples/tensorzero/agent.py +35 -0
fast_agent/resources/examples/tensorzero/demo_images/clam.jpg +0 -0
fast_agent/resources/examples/tensorzero/demo_images/crab.png +0 -0
fast_agent/resources/examples/tensorzero/demo_images/shrimp.png +0 -0
fast_agent/resources/examples/tensorzero/docker-compose.yml +105 -0
fast_agent/resources/examples/tensorzero/fastagent.config.yaml +19 -0
fast_agent/resources/examples/tensorzero/image_demo.py +67 -0
fast_agent/resources/examples/tensorzero/mcp_server/Dockerfile +25 -0
fast_agent/resources/examples/tensorzero/mcp_server/entrypoint.sh +35 -0
fast_agent/resources/examples/tensorzero/mcp_server/mcp_server.py +31 -0
fast_agent/resources/examples/tensorzero/mcp_server/pyproject.toml +11 -0
fast_agent/resources/examples/tensorzero/simple_agent.py +25 -0
fast_agent/resources/examples/tensorzero/tensorzero_config/system_schema.json +29 -0
fast_agent/resources/examples/tensorzero/tensorzero_config/system_template.minijinja +11 -0
fast_agent/resources/examples/tensorzero/tensorzero_config/tensorzero.toml +35 -0
fast_agent/resources/examples/workflows/agents_as_tools_extended.py +73 -0
fast_agent/resources/examples/workflows/agents_as_tools_simple.py +50 -0
fast_agent/resources/examples/workflows/chaining.py +37 -0
fast_agent/resources/examples/workflows/evaluator.py +77 -0
fast_agent/resources/examples/workflows/fastagent.config.yaml +26 -0
fast_agent/resources/examples/workflows/graded_report.md +89 -0
fast_agent/resources/examples/workflows/human_input.py +28 -0
fast_agent/resources/examples/workflows/maker.py +156 -0
fast_agent/resources/examples/workflows/orchestrator.py +70 -0
fast_agent/resources/examples/workflows/parallel.py +56 -0
fast_agent/resources/examples/workflows/router.py +69 -0
fast_agent/resources/examples/workflows/short_story.md +13 -0
fast_agent/resources/examples/workflows/short_story.txt +19 -0
fast_agent/resources/setup/.gitignore +30 -0
fast_agent/resources/setup/agent.py +28 -0
fast_agent/resources/setup/fastagent.config.yaml +65 -0
fast_agent/resources/setup/fastagent.secrets.yaml.example +38 -0
fast_agent/resources/setup/pyproject.toml.tmpl +23 -0
fast_agent/skills/__init__.py +9 -0
fast_agent/skills/registry.py +235 -0
fast_agent/tools/elicitation.py +369 -0
fast_agent/tools/shell_runtime.py +402 -0
fast_agent/types/__init__.py +59 -0
fast_agent/types/conversation_summary.py +294 -0
fast_agent/types/llm_stop_reason.py +78 -0
fast_agent/types/message_search.py +249 -0
fast_agent/ui/__init__.py +38 -0
fast_agent/ui/console.py +59 -0
fast_agent/ui/console_display.py +1080 -0
fast_agent/ui/elicitation_form.py +946 -0
fast_agent/ui/elicitation_style.py +59 -0
fast_agent/ui/enhanced_prompt.py +1400 -0
fast_agent/ui/history_display.py +734 -0
fast_agent/ui/interactive_prompt.py +1199 -0
fast_agent/ui/markdown_helpers.py +104 -0
fast_agent/ui/markdown_truncator.py +1004 -0
fast_agent/ui/mcp_display.py +857 -0
fast_agent/ui/mcp_ui_utils.py +235 -0
fast_agent/ui/mermaid_utils.py +169 -0
fast_agent/ui/message_primitives.py +50 -0
fast_agent/ui/notification_tracker.py +205 -0
fast_agent/ui/plain_text_truncator.py +68 -0
fast_agent/ui/progress_display.py +10 -0
fast_agent/ui/rich_progress.py +195 -0
fast_agent/ui/streaming.py +774 -0
fast_agent/ui/streaming_buffer.py +449 -0
fast_agent/ui/tool_display.py +422 -0
fast_agent/ui/usage_display.py +204 -0
fast_agent/utils/__init__.py +5 -0
fast_agent/utils/reasoning_stream_parser.py +77 -0
fast_agent/utils/time.py +22 -0
fast_agent/workflow_telemetry.py +261 -0
fast_agent_mcp-0.4.7.dist-info/METADATA +788 -0
fast_agent_mcp-0.4.7.dist-info/RECORD +261 -0
fast_agent_mcp-0.4.7.dist-info/WHEEL +4 -0
fast_agent_mcp-0.4.7.dist-info/entry_points.txt +7 -0
fast_agent_mcp-0.4.7.dist-info/licenses/LICENSE +201 -0

fast_agent/mcp/oauth_client.py ADDED Viewed

@@ -0,0 +1,509 @@
+"""
+OAuth v2.1 integration helpers for MCP client transports.
+Provides token storage (in-memory and OS keyring), a local callback server
+with paste-URL fallback, and a builder for OAuthClientProvider that can be
+passed to SSE/HTTP transports as the `auth` parameter.
+"""
+from __future__ import annotations
+import threading
+import time
+from dataclasses import dataclass
+from http.server import BaseHTTPRequestHandler, HTTPServer
+from typing import TYPE_CHECKING, Any, Callable
+from urllib.parse import parse_qs, urlparse
+from mcp.client.auth import OAuthClientProvider, TokenStorage
+from mcp.shared.auth import (
+    OAuthClientInformationFull,
+    OAuthClientMetadata,
+    OAuthToken,
+)
+from pydantic import AnyUrl
+from fast_agent.core.logging.logger import get_logger
+from fast_agent.ui import console
+if TYPE_CHECKING:
+    from fast_agent.config import MCPServerSettings
+logger = get_logger(__name__)
+class InMemoryTokenStorage(TokenStorage):
+    """Non-persistent token storage (process memory only)."""
+    def __init__(self) -> None:
+        self._tokens: OAuthToken | None = None
+        self._client_info: OAuthClientInformationFull | None = None
+    async def get_tokens(self) -> OAuthToken | None:
+        return self._tokens
+    async def set_tokens(self, tokens: OAuthToken) -> None:
+        self._tokens = tokens
+    async def get_client_info(self) -> OAuthClientInformationFull | None:
+        return self._client_info
+    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+        self._client_info = client_info
+@dataclass
+class _CallbackResult:
+    authorization_code: str | None = None
+    state: str | None = None
+    error: str | None = None
+class _CallbackHandler(BaseHTTPRequestHandler):
+    """HTTP handler to capture OAuth callback query params."""
+    def __init__(self, *args, result: _CallbackResult, expected_path: str, **kwargs):
+        self._result = result
+        self._expected_path = expected_path.rstrip("/") or "/callback"
+        super().__init__(*args, **kwargs)
+    def do_GET(self) -> None:  # noqa: N802 - http.server signature
+        parsed = urlparse(self.path)
+        # Only accept the configured callback path
+        if (parsed.path.rstrip("/") or "/callback") != self._expected_path:
+            self.send_response(404)
+            self.end_headers()
+            return
+        params = parse_qs(parsed.query)
+        if "code" in params:
+            self._result.authorization_code = params["code"][0]
+            self._result.state = params.get("state", [None])[0]
+            self.send_response(200)
+            self.send_header("Content-Type", "text/html")
+            self.end_headers()
+            self.wfile.write(
+                b"""
+                <html><body>
+                <h1>Authorization Successful</h1>
+                <p>You can close this window.</p>
+                <script>setTimeout(() => window.close(), 1000);</script>
+                </body></html>
+                """
+            )
+        elif "error" in params:
+            self._result.error = params["error"][0]
+            self.send_response(400)
+            self.send_header("Content-Type", "text/html")
+            self.end_headers()
+            self.wfile.write(
+                f"""
+                <html><body>
+                <h1>Authorization Failed</h1>
+                <p>Error: {self._result.error}</p>
+                </body></html>
+                """.encode()
+            )
+        else:
+            self.send_response(404)
+            self.end_headers()
+    def log_message(self, format: str, *args: Any) -> None:  # silence default logging
+        return
+class _CallbackServer:
+    """Simple background HTTP server to receive a single OAuth callback."""
+    def __init__(self, port: int, path: str) -> None:
+        self._port = port
+        self._path = path.rstrip("/") or "/callback"
+        self._result = _CallbackResult()
+        self._server: HTTPServer | None = None
+        self._thread: threading.Thread | None = None
+    def _make_handler(self) -> Callable[..., BaseHTTPRequestHandler]:
+        result = self._result
+        expected_path = self._path
+        def handler(*args, **kwargs):
+            return _CallbackHandler(*args, result=result, expected_path=expected_path, **kwargs)
+        return handler
+    def start(self) -> None:
+        self._server = HTTPServer(("localhost", self._port), self._make_handler())
+        self._thread = threading.Thread(target=self._server.serve_forever, daemon=True)
+        self._thread.start()
+        logger.info(f"OAuth callback server listening on http://localhost:{self._port}{self._path}")
+    def stop(self) -> None:
+        if self._server:
+            try:
+                self._server.shutdown()
+                self._server.server_close()
+            except Exception:
+                pass
+        if self._thread:
+            self._thread.join(timeout=1)
+    def wait(self, timeout_seconds: int = 300) -> tuple[str, str | None]:
+        start = time.time()
+        while time.time() - start < timeout_seconds:
+            if self._result.authorization_code:
+                return self._result.authorization_code, self._result.state
+            if self._result.error:
+                raise RuntimeError(f"OAuth error: {self._result.error}")
+            time.sleep(0.1)
+        raise TimeoutError("Timeout waiting for OAuth callback")
+def _derive_base_server_url(url: str | None) -> str | None:
+    """Derive the base server URL for OAuth discovery from an MCP endpoint URL.
+    - Strips a trailing "/mcp" or "/sse" path segment
+    - Ignores query and fragment parts entirely
+    """
+    if not url:
+        return None
+    try:
+        from urllib.parse import urlparse, urlunparse
+        parsed = urlparse(url)
+        # Normalize path without trailing slash
+        path = parsed.path or ""
+        path = path[:-1] if path.endswith("/") else path
+        # Remove one trailing segment if it is mcp or sse
+        for suffix in ("/mcp", "/sse"):
+            if path.endswith(suffix):
+                path = path[: -len(suffix)]
+                break
+        # Ensure path is at least '/'
+        if not path:
+            path = "/"
+        # Rebuild URL without query/fragment
+        clean = parsed._replace(path=path, params="", query="", fragment="")
+        base = urlunparse(clean)
+        # Drop trailing slash except for root
+        if base.endswith("/") and base.count("/") > 2:
+            base = base[:-1]
+        return base
+    except Exception:
+        return url
+def compute_server_identity(server_config: MCPServerSettings) -> str:
+    """Compute a stable identity for token storage.
+    Prefer the normalized base server URL; fall back to configured name, then 'default'.
+    """
+    base = _derive_base_server_url(server_config.url)
+    if base:
+        return base
+    if server_config.name:
+        return server_config.name
+    return "default"
+def keyring_has_token(server_config: MCPServerSettings) -> bool:
+    """Check if keyring has a token stored for this server."""
+    try:
+        import keyring
+        identity = compute_server_identity(server_config)
+        token_key = f"oauth:tokens:{identity}"
+        return keyring.get_password("fast-agent-mcp", token_key) is not None
+    except Exception:
+        return False
+async def _print_authorization_link(auth_url: str, warn_if_no_keyring: bool = False) -> None:
+    """Emit a clickable authorization link using rich console markup.
+    If warn_if_no_keyring is True and the OS keyring backend is unavailable,
+    print a warning to indicate tokens won't be persisted.
+    """
+    console.console.print("[bold]Open this link to authorize:[/bold]", markup=True)
+    console.console.print(f"[link={auth_url}]{auth_url}[/link]")
+    if warn_if_no_keyring:
+        try:
+            import keyring  # type: ignore
+            backend = keyring.get_keyring()
+            try:
+                from keyring.backends.fail import Keyring as FailKeyring  # type: ignore
+                if isinstance(backend, FailKeyring):
+                    console.console.print(
+                        "[yellow]Warning:[/yellow] Keyring backend not available — tokens will not be persisted."
+                    )
+            except Exception:
+                # If we cannot detect the fail backend, do nothing
+                pass
+        except Exception:
+            console.console.print(
+                "[yellow]Warning:[/yellow] Keyring backend not available — tokens will not be persisted."
+            )
+    logger.info("OAuth authorization URL emitted to console")
+class KeyringTokenStorage(TokenStorage):
+    """Token storage backed by the OS keychain using 'keyring'."""
+    def __init__(self, service_name: str, server_identity: str) -> None:
+        self._service = service_name
+        self._identity = server_identity
+    @property
+    def _token_key(self) -> str:
+        return f"oauth:tokens:{self._identity}"
+    @property
+    def _client_key(self) -> str:
+        return f"oauth:client_info:{self._identity}"
+    async def get_tokens(self) -> OAuthToken | None:
+        try:
+            import keyring
+            payload = keyring.get_password(self._service, self._token_key)
+            if not payload:
+                return None
+            return OAuthToken.model_validate_json(payload)
+        except Exception:
+            return None
+    async def set_tokens(self, tokens: OAuthToken) -> None:
+        try:
+            import keyring
+            keyring.set_password(self._service, self._token_key, tokens.model_dump_json())
+            # Update index
+            add_identity_to_index(self._service, self._identity)
+        except Exception:
+            pass
+    async def get_client_info(self) -> OAuthClientInformationFull | None:
+        try:
+            import keyring
+            payload = keyring.get_password(self._service, self._client_key)
+            if not payload:
+                return None
+            return OAuthClientInformationFull.model_validate_json(payload)
+        except Exception:
+            return None
+    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+        try:
+            import keyring
+            keyring.set_password(self._service, self._client_key, client_info.model_dump_json())
+        except Exception:
+            pass
+# --- Keyring index helpers (to enable cross-platform token enumeration) ---
+def _index_username() -> str:
+    return "oauth:index"
+def _read_index(service: str) -> set[str]:
+    try:
+        import json
+        import keyring
+        raw = keyring.get_password(service, _index_username())
+        if not raw:
+            return set()
+        data = json.loads(raw)
+        if isinstance(data, list):
+            return set([str(x) for x in data])
+        return set()
+    except Exception:
+        return set()
+def _write_index(service: str, identities: set[str]) -> None:
+    try:
+        import json
+        import keyring
+        payload = json.dumps(sorted(list(identities)))
+        keyring.set_password(service, _index_username(), payload)
+    except Exception:
+        pass
+def add_identity_to_index(service: str, identity: str) -> None:
+    identities = _read_index(service)
+    if identity not in identities:
+        identities.add(identity)
+        _write_index(service, identities)
+def remove_identity_from_index(service: str, identity: str) -> None:
+    identities = _read_index(service)
+    if identity in identities:
+        identities.remove(identity)
+        _write_index(service, identities)
+def list_keyring_tokens(service: str = "fast-agent-mcp") -> list[str]:
+    """List identities with stored tokens in keyring (using our index).
+    Returns only identities that currently have a corresponding token entry.
+    """
+    try:
+        import keyring
+        identities = _read_index(service)
+        present: list[str] = []
+        for ident in sorted(identities):
+            tok_key = f"oauth:tokens:{ident}"
+            if keyring.get_password(service, tok_key):
+                present.append(ident)
+        return present
+    except Exception:
+        return []
+def clear_keyring_token(identity: str, service: str = "fast-agent-mcp") -> bool:
+    """Remove token+client info for identity and update the index.
+    Returns True if anything was removed.
+    """
+    removed = False
+    try:
+        import keyring
+        tok_key = f"oauth:tokens:{identity}"
+        cli_key = f"oauth:client_info:{identity}"
+        try:
+            keyring.delete_password(service, tok_key)
+            removed = True
+        except Exception:
+            pass
+        try:
+            keyring.delete_password(service, cli_key)
+            removed = True or removed
+        except Exception:
+            pass
+        if removed:
+            remove_identity_from_index(service, identity)
+    except Exception:
+        return False
+    return removed
+def build_oauth_provider(server_config: MCPServerSettings) -> OAuthClientProvider | None:
+    """
+    Build an OAuthClientProvider for the given server config if applicable.
+    Returns None for unsupported transports, or when disabled via config.
+    """
+    # Only for SSE/HTTP transports
+    if server_config.transport not in ("sse", "http"):
+        return None
+    # Determine if OAuth should be enabled. Default to True if no auth block provided
+    enable_oauth = True
+    redirect_port = 3030
+    redirect_path = "/callback"
+    scope_value: str | None = None
+    persist_mode: str = "keyring"
+    if server_config.auth is not None:
+        try:
+            enable_oauth = getattr(server_config.auth, "oauth", True)
+            redirect_port = getattr(server_config.auth, "redirect_port", 3030)
+            redirect_path = getattr(server_config.auth, "redirect_path", "/callback")
+            scope_field = getattr(server_config.auth, "scope", None)
+            persist_mode = getattr(server_config.auth, "persist", "keyring")
+            if isinstance(scope_field, list):
+                scope_value = " ".join(scope_field)
+            elif isinstance(scope_field, str):
+                scope_value = scope_field
+        except Exception:
+            logger.debug("Malformed auth configuration; using defaults.")
+    if not enable_oauth:
+        return None
+    base_url = _derive_base_server_url(server_config.url)
+    if not base_url:
+        # No usable URL -> cannot build provider
+        return None
+    # Construct client metadata with minimal defaults
+    redirect_uri = f"http://localhost:{redirect_port}{redirect_path}"
+    metadata_kwargs: dict[str, Any] = {
+        "client_name": "fast-agent",
+        "redirect_uris": [AnyUrl(redirect_uri)],
+        "grant_types": ["authorization_code", "refresh_token"],
+        "response_types": ["code"],
+    }
+    if scope_value:
+        metadata_kwargs["scope"] = scope_value
+    client_metadata = OAuthClientMetadata.model_validate(metadata_kwargs)
+    # Local callback server handler
+    async def _redirect_handler(authorization_url: str) -> None:
+        # Warn if persisting to keyring but no backend is available
+        await _print_authorization_link(
+            authorization_url,
+            warn_if_no_keyring=(persist_mode == "keyring"),
+        )
+    async def _callback_handler() -> tuple[str, str | None]:
+        # Try local HTTP capture first
+        try:
+            server = _CallbackServer(port=redirect_port, path=redirect_path)
+            server.start()
+            try:
+                code, state = server.wait(timeout_seconds=300)
+                return code, state
+            finally:
+                server.stop()
+        except Exception as e:
+            # Fallback to paste-URL flow
+            logger.info(f"OAuth local callback server unavailable, fallback to paste flow: {e}")
+            try:
+                import sys
+                print("Paste the full callback URL after authorization:", file=sys.stderr)
+                callback_url = input("Callback URL: ").strip()
+            except Exception as ee:
+                raise RuntimeError(f"Failed to read callback URL from user: {ee}")
+            params = parse_qs(urlparse(callback_url).query)
+            code = params.get("code", [None])[0]
+            state = params.get("state", [None])[0]
+            if not code:
+                raise RuntimeError("Callback URL missing authorization code")
+            return code, state
+    # Choose storage
+    storage: TokenStorage
+    if persist_mode == "keyring":
+        identity = compute_server_identity(server_config)
+        # Update index on write via storage methods; creation here doesn't modify index yet.
+        storage = KeyringTokenStorage(service_name="fast-agent-mcp", server_identity=identity)
+    else:
+        storage = InMemoryTokenStorage()
+    provider = OAuthClientProvider(
+        server_url=base_url,
+        client_metadata=client_metadata,
+        storage=storage,
+        redirect_handler=_redirect_handler,
+        callback_handler=_callback_handler,
+    )
+    return provider

fast_agent/mcp/prompt.py ADDED Viewed

@@ -0,0 +1,159 @@
+"""
+Prompt class for easily creating and working with MCP prompt content.
+This implementation lives in the fast_agent namespace as part of the
+migration away from fast_agent. A compatibility shim remains at
+fast_agent.core.prompt importing this Prompt.
+"""
+from pathlib import Path
+from typing import Literal, Union
+from mcp import CallToolRequest
+from mcp.types import ContentBlock, PromptMessage
+from fast_agent.mcp.mcp_content import Assistant, MCPPrompt, User
+from fast_agent.types import LlmStopReason, PromptMessageExtended
+class Prompt:
+    """
+    A helper class for working with MCP prompt content.
+    This class provides static methods to create:
+    - PromptMessage instances
+    - PromptMessageExtended instances
+    - Lists of messages for conversations
+    All methods intelligently handle various content types:
+    - Strings become TextContent
+    - Image file paths become ImageContent
+    - Other file paths become EmbeddedResource
+    - TextContent objects are used directly
+    - ImageContent objects are used directly
+    - EmbeddedResource objects are used directly
+    - Pre-formatted messages pass through unchanged
+    """
+    @classmethod
+    def user(
+        cls,
+        *content_items: Union[
+            str, Path, bytes, dict, ContentBlock, PromptMessage, PromptMessageExtended
+        ],
+    ) -> PromptMessageExtended:
+        """
+        Create a user PromptMessageExtended with various content items.
+        """
+        # Handle PromptMessage and PromptMessageExtended directly
+        if len(content_items) == 1:
+            item = content_items[0]
+            if isinstance(item, PromptMessage):
+                return PromptMessageExtended(role="user", content=[item.content])
+            elif isinstance(item, PromptMessageExtended):
+                # Keep the content but change role to user
+                return PromptMessageExtended(role="user", content=item.content)
+        # Use the content factory for other types
+        messages = User(*content_items)
+        return PromptMessageExtended(role="user", content=[msg["content"] for msg in messages])
+    @classmethod
+    def assistant(
+        cls,
+        *content_items: Union[
+            str, Path, bytes, dict, ContentBlock, PromptMessage, PromptMessageExtended
+        ],
+        stop_reason: LlmStopReason | None = None,
+        tool_calls: dict[str, CallToolRequest] | None = None,
+    ) -> PromptMessageExtended:
+        """
+        Create an assistant PromptMessageExtended with various content items.
+        """
+        # Handle PromptMessage and PromptMessageExtended directly
+        if len(content_items) == 1:
+            item = content_items[0]
+            if isinstance(item, PromptMessage):
+                return PromptMessageExtended(
+                    role="assistant",
+                    content=[item.content],
+                    stop_reason=stop_reason,
+                    tool_calls=tool_calls,
+                )
+            elif isinstance(item, PromptMessageExtended):
+                # Keep the content but change role to assistant
+                return PromptMessageExtended(
+                    role="assistant",
+                    content=item.content,
+                    stop_reason=stop_reason,
+                    tool_calls=tool_calls,
+                )
+        # Use the content factory for other types
+        messages = Assistant(*content_items)
+        return PromptMessageExtended(
+            role="assistant",
+            content=[msg["content"] for msg in messages],
+            stop_reason=stop_reason,
+            tool_calls=tool_calls,
+        )
+    @classmethod
+    def message(
+        cls,
+        *content_items: Union[
+            str, Path, bytes, dict, ContentBlock, PromptMessage, PromptMessageExtended
+        ],
+        role: Literal["user", "assistant"] = "user",
+    ) -> PromptMessageExtended:
+        """
+        Create a PromptMessageExtended with the specified role and content items.
+        """
+        # Handle PromptMessage and PromptMessageExtended directly
+        if len(content_items) == 1:
+            item = content_items[0]
+            if isinstance(item, PromptMessage):
+                return PromptMessageExtended(role=role, content=[item.content])
+            elif isinstance(item, PromptMessageExtended):
+                # Keep the content but change role as specified
+                return PromptMessageExtended(role=role, content=item.content)
+        # Use the content factory for other types
+        messages = MCPPrompt(*content_items, role=role)
+        return PromptMessageExtended(
+            role=messages[0]["role"] if messages else role,
+            content=[msg["content"] for msg in messages],
+        )
+    @classmethod
+    def conversation(cls, *messages) -> list[PromptMessage]:
+        """
+        Create a list of PromptMessages from various inputs.
+        """
+        result = []
+        for item in messages:
+            if isinstance(item, PromptMessageExtended):
+                # Convert PromptMessageExtended to a list of PromptMessages
+                result.extend(item.from_multipart())
+            elif isinstance(item, dict) and "role" in item and "content" in item:
+                # Convert a single message dict to PromptMessage
+                result.append(PromptMessage(**item))
+            elif isinstance(item, list):
+                # Process each item in the list
+                for msg in item:
+                    if isinstance(msg, dict) and "role" in msg and "content" in msg:
+                        result.append(PromptMessage(**msg))
+            # Ignore other types
+        return result
+    @classmethod
+    def from_multipart(cls, multipart: list[PromptMessageExtended]) -> list[PromptMessage]:
+        """
+        Convert a list of PromptMessageExtended objects to PromptMessages.
+        """
+        result = []
+        for mp in multipart:
+            result.extend(mp.from_multipart())
+        return result