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/ui/mcp_ui_utils.py ADDED Viewed

@@ -0,0 +1,235 @@
+from __future__ import annotations
+import base64
+import platform
+import re
+import subprocess
+import webbrowser
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Iterable
+from mcp.types import BlobResourceContents, EmbeddedResource, TextResourceContents
+"""
+Utilities for handling MCP-UI resources carried in PromptMessageExtended.channels.
+Responsibilities:
+- Identify MCP-UI EmbeddedResources from channels
+- Decode text/blob content depending on mimeType
+- Produce local HTML files that safely embed the UI content (srcdoc or iframe)
+- Return presentable link labels for console display
+"""
+# Control whether to generate data URLs for embedded HTML content
+# When disabled, always use file:// URLs which work better with most terminals
+ENABLE_DATA_URLS = False
+@dataclass
+class UILink:
+    title: str
+    file_path: str  # absolute path to local html file
+    web_url: str | None = None  # Preferable clickable link (http(s) or data URL)
+def _safe_filename(name: str) -> str:
+    name = re.sub(r"[^A-Za-z0-9_.-]", "_", name)
+    return name[:120] if len(name) > 120 else name
+def _ensure_output_dir() -> Path:
+    # Read output directory from settings, defaulting to .fast-agent/ui
+    try:
+        from fast_agent.config import get_settings
+        settings = get_settings()
+        dir_setting = getattr(settings, "mcp_ui_output_dir", ".fast-agent/ui") or ".fast-agent/ui"
+    except Exception:
+        dir_setting = ".fast-agent/ui"
+    base = Path(dir_setting)
+    if not base.is_absolute():
+        base = Path.cwd() / base
+    base.mkdir(parents=True, exist_ok=True)
+    return base
+def _extract_title(uri: str | None) -> str:
+    if not uri:
+        return "UI"
+    try:
+        # ui://component/instance -> component:instance
+        without_scheme = uri.split("ui://", 1)[1] if uri.startswith("ui://") else uri
+        parts = [p for p in re.split(r"[/:]", without_scheme) if p]
+        if len(parts) >= 2:
+            return f"{parts[0]}:{parts[1]}"
+        return parts[0] if parts else "UI"
+    except Exception:
+        return "UI"
+def _decode_text_or_blob(resource) -> str | None:
+    """Return string content from TextResourceContents or BlobResourceContents."""
+    if isinstance(resource, TextResourceContents):
+        return resource.text or ""
+    if isinstance(resource, BlobResourceContents):
+        try:
+            return base64.b64decode(resource.blob or "").decode("utf-8", errors="replace")
+        except Exception:
+            return None
+    return None
+def _first_https_url_from_uri_list(text: str) -> str | None:
+    for line in text.splitlines():
+        line = line.strip()
+        if not line or line.startswith("#"):
+            continue
+        if line.startswith("http://") or line.startswith("https://"):
+            return line
+    return None
+def _make_html_for_raw_html(html_string: str) -> str:
+    # Wrap with minimal HTML and sandbox guidance (iframe srcdoc will be used by browsers)
+    return html_string
+def _make_html_for_uri(url: str) -> str:
+    return f"""
+<!doctype html>
+<html>
+  <head>
+    <meta charset=\"utf-8\" />
+    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1\" />
+    <title>MCP-UI</title>
+    <style>html,body,iframe{{margin:0;padding:0;height:100%;width:100%;border:0}}</style>
+  </head>
+  <body>
+    <iframe src=\"{url}\" sandbox=\"allow-scripts allow-forms allow-same-origin\" referrerpolicy=\"no-referrer\"></iframe>
+  </body>
+  </html>
+"""
+def _write_html_file(name_hint: str, html: str) -> str:
+    out_dir = _ensure_output_dir()
+    file_name = _safe_filename(name_hint or "ui") + ".html"
+    out_path = out_dir / file_name
+    # Ensure unique filename if exists
+    i = 1
+    while out_path.exists():
+        out_path = out_dir / f"{_safe_filename(name_hint)}_{i}.html"
+        i += 1
+    out_path.write_text(html, encoding="utf-8")
+    return str(out_path.resolve())
+def ui_links_from_channel(resources: Iterable[EmbeddedResource]) -> list[UILink]:
+    """
+    Build local HTML files for a list of MCP-UI EmbeddedResources and return clickable links.
+    Supported mime types:
+    - text/html: expects text or base64 blob of HTML
+    - text/uri-list: expects text or blob of a single URL (first valid URL is used)
+    - application/vnd.mcp-ui.remote-dom* : currently unsupported; generate a placeholder page
+    """
+    links: list[UILink] = []
+    for emb in resources:
+        res = emb.resource
+        uri = str(getattr(res, "uri", "")) if getattr(res, "uri", None) else None
+        mime = getattr(res, "mimeType", "") or ""
+        title = _extract_title(uri)
+        content = _decode_text_or_blob(res)
+        if mime.startswith("text/html"):
+            if content is None:
+                continue
+            html = _make_html_for_raw_html(content)
+            file_path = _write_html_file(title, html)
+            # Generate data URL only if enabled
+            if ENABLE_DATA_URLS:
+                try:
+                    b64 = base64.b64encode(html.encode("utf-8")).decode("ascii")
+                    data_url = f"data:text/html;base64,{b64}"
+                    # Some terminals have limits; only attach when reasonably small
+                    web_url = data_url if len(data_url) < 12000 else None
+                except Exception:
+                    web_url = None
+            else:
+                web_url = None
+            links.append(UILink(title=title, file_path=file_path, web_url=web_url))
+        elif mime.startswith("text/uri-list"):
+            if content is None:
+                continue
+            url = _first_https_url_from_uri_list(content)
+            if not url:
+                # fallback: try to treat entire content as a URL
+                url = content.strip()
+            if not (url and (url.startswith("http://") or url.startswith("https://"))):
+                continue
+            html = _make_html_for_uri(url)
+            file_path = _write_html_file(title, html)
+            # Prefer the direct URL for clickability; keep file for archival
+            links.append(UILink(title=title, file_path=file_path, web_url=url))
+        elif mime.startswith("application/vnd.mcp-ui.remote-dom"):
+            # Not supported yet - generate informational page
+            placeholder = f"""
+<!doctype html>
+<html><head><meta charset=\"utf-8\" /><title>{title} (Unsupported)</title></head>
+<body>
+  <p>Remote DOM resources are not supported yet in this client.</p>
+  <p>URI: {uri or ""}</p>
+  <p>mimeType: {mime}</p>
+  <pre style=\"white-space: pre-wrap;\">{(content or "")[:4000]}</pre>
+  <p>Please upgrade fast-agent when support becomes available.</p>
+  </body></html>
+"""
+            file_path = _write_html_file(title + "_unsupported", placeholder)
+            links.append(UILink(title=title + " (unsupported)", file_path=file_path))
+        else:
+            # Unknown, skip quietly
+            continue
+    return links
+def open_links_in_browser(links: Iterable[UILink], mcp_ui_mode: str = "auto") -> None:
+    """Open links in browser/system viewer.
+    Args:
+        links: Links to open
+        mcp_ui_mode: UI mode setting ("disabled", "enabled", "auto")
+    """
+    # Only attempt to open files when in auto mode
+    if mcp_ui_mode != "auto":
+        return
+    for link in links:
+        try:
+            # Use subprocess for better file:// handling across platforms
+            file_path = link.file_path
+            system = platform.system()
+            if system == "Darwin":  # macOS
+                subprocess.run(["open", file_path], check=False, capture_output=True)
+            elif system == "Windows":
+                subprocess.run(
+                    ["start", "", file_path], shell=True, check=False, capture_output=True
+                )
+            elif system == "Linux":
+                # Try xdg-open first (most common), fallback to other options
+                try:
+                    subprocess.run(["xdg-open", file_path], check=False, capture_output=True)
+                except FileNotFoundError:
+                    # Fallback to webbrowser for Linux if xdg-open not available
+                    webbrowser.open(f"file://{file_path}", new=2)
+            else:
+                # Unknown system, fallback to webbrowser
+                webbrowser.open(f"file://{file_path}", new=2)
+        except Exception:
+            # Silently ignore errors - user can still manually open the file
+            pass

fast_agent/ui/mermaid_utils.py ADDED Viewed

@@ -0,0 +1,169 @@
+"""Utilities for detecting and processing Mermaid diagrams in text content."""
+import base64
+import re
+import zlib
+from dataclasses import dataclass
+# Mermaid chart viewer URL prefix
+MERMAID_VIEWER_URL = "https://www.mermaidchart.com/play#"
+# mermaid.live#pako= also works but the playground has better ux
+@dataclass
+class MermaidDiagram:
+    """Represents a detected Mermaid diagram."""
+    content: str
+    title: str | None = None
+    start_pos: int = 0
+    end_pos: int = 0
+def extract_mermaid_diagrams(text: str) -> list[MermaidDiagram]:
+    """
+    Extract all Mermaid diagram blocks from text content.
+    Handles both simple mermaid blocks and blocks with titles:
+    - ```mermaid
+    - ```mermaid title={Some Title}
+    Also extracts titles from within the diagram content.
+    Args:
+        text: The text content to search for Mermaid diagrams
+    Returns:
+        List of MermaidDiagram objects found in the text
+    """
+    diagrams = []
+    # Pattern to match mermaid code blocks with optional title
+    # Matches: ```mermaid or ```mermaid title={...}
+    pattern = r"```mermaid(?:\s+title=\{([^}]+)\})?\s*\n(.*?)```"
+    for match in re.finditer(pattern, text, re.DOTALL):
+        title = match.group(1)  # May be None if no title
+        content = match.group(2).strip()
+        if content:  # Only add if there's actual diagram content
+            # If no title from code fence, look for title in the content
+            if not title:
+                # Look for various title patterns in mermaid diagrams
+                # pie title, graph title, etc.
+                title_patterns = [
+                    r"^\s*title\s+(.+?)(?:\n|$)",  # Generic title
+                    r"^\s*pie\s+title\s+(.+?)(?:\n|$)",  # Pie chart title
+                    r"^\s*gantt\s+title\s+(.+?)(?:\n|$)",  # Gantt chart title
+                ]
+                for title_pattern in title_patterns:
+                    title_match = re.search(title_pattern, content, re.MULTILINE)
+                    if title_match:
+                        title = title_match.group(1).strip()
+                        break
+            diagrams.append(
+                MermaidDiagram(
+                    content=content, title=title, start_pos=match.start(), end_pos=match.end()
+                )
+            )
+    return diagrams
+def create_mermaid_live_link(diagram_content: str) -> str:
+    """
+    Create a Mermaid Live Editor link from diagram content.
+    The link uses pako compression (zlib) and base64 encoding.
+    Args:
+        diagram_content: The Mermaid diagram source code
+    Returns:
+        Complete URL to Mermaid Live Editor
+    """
+    # Create the JSON structure expected by Mermaid Live
+    # Escape newlines and quotes in the diagram content
+    escaped_content = diagram_content.replace('"', '\\"').replace("\n", "\\n")
+    json_str = f'{{"code":"{escaped_content}","mermaid":{{"theme":"default"}},"updateEditor":false,"autoSync":true,"updateDiagram":false}}'
+    # Compress using zlib (pako compatible)
+    compressed = zlib.compress(json_str.encode("utf-8"))
+    # Base64 encode
+    encoded = base64.urlsafe_b64encode(compressed).decode("utf-8")
+    # Remove padding characters as Mermaid Live doesn't use them
+    encoded = encoded.rstrip("=")
+    return f"{MERMAID_VIEWER_URL}pako:{encoded}"
+def format_mermaid_links(diagrams: list[MermaidDiagram]) -> list[str]:
+    """
+    Format Mermaid diagrams as markdown links.
+    Args:
+        diagrams: List of MermaidDiagram objects
+    Returns:
+        List of formatted markdown strings
+    """
+    links = []
+    for i, diagram in enumerate(diagrams, 1):
+        link = create_mermaid_live_link(diagram.content)
+        if diagram.title:
+            # Use the title from the diagram with number
+            markdown = f"Diagram {i} - {diagram.title}: [Open Diagram]({link})"
+        else:
+            # Use generic numbering
+            markdown = f"Diagram {i}: [Open Diagram]({link})"
+        links.append(markdown)
+    return links
+def detect_diagram_type(content: str) -> str:
+    """
+    Detect the type of mermaid diagram from content.
+    Args:
+        content: The mermaid diagram source code
+    Returns:
+        Human-readable diagram type name
+    """
+    content_lower = content.strip().lower()
+    # Check for common diagram types
+    if content_lower.startswith(("graph ", "flowchart ")):
+        return "Flowchart"
+    elif content_lower.startswith("sequencediagram"):
+        return "Sequence"
+    elif content_lower.startswith("pie"):
+        return "Pie Chart"
+    elif content_lower.startswith("gantt"):
+        return "Gantt Chart"
+    elif content_lower.startswith("classdiagram"):
+        return "Class Diagram"
+    elif content_lower.startswith("statediagram"):
+        return "State Diagram"
+    elif content_lower.startswith("erdiagram"):
+        return "ER Diagram"
+    elif content_lower.startswith("journey"):
+        return "User Journey"
+    elif content_lower.startswith("gitgraph"):
+        return "Git Graph"
+    elif content_lower.startswith("c4context"):
+        return "C4 Context"
+    elif content_lower.startswith("mindmap"):
+        return "Mind Map"
+    elif content_lower.startswith("timeline"):
+        return "Timeline"
+    else:
+        return "Diagram"

fast_agent/ui/message_primitives.py ADDED Viewed

@@ -0,0 +1,50 @@
+from __future__ import annotations
+from enum import Enum
+class MessageType(Enum):
+    """Types of messages that can be displayed."""
+    USER = "user"
+    ASSISTANT = "assistant"
+    SYSTEM = "system"
+    TOOL_CALL = "tool_call"
+    TOOL_RESULT = "tool_result"
+MESSAGE_CONFIGS: dict[MessageType, dict[str, str]] = {
+    MessageType.USER: {
+        "block_color": "blue",
+        "arrow": "▶",
+        "arrow_style": "dim blue",
+        "highlight_color": "blue",
+    },
+    MessageType.ASSISTANT: {
+        "block_color": "green",
+        "arrow": "◀",
+        "arrow_style": "dim green",
+        "highlight_color": "bright_green",
+    },
+    MessageType.SYSTEM: {
+        "block_color": "yellow",
+        "arrow": "●",
+        "arrow_style": "dim yellow",
+        "highlight_color": "bright_yellow",
+    },
+    MessageType.TOOL_CALL: {
+        "block_color": "magenta",
+        "arrow": "◀",
+        "arrow_style": "dim magenta",
+        "highlight_color": "magenta",
+    },
+    MessageType.TOOL_RESULT: {
+        "block_color": "magenta",
+        "arrow": "▶",
+        "arrow_style": "dim magenta",
+        "highlight_color": "magenta",
+    },
+}
+__all__ = ["MessageType", "MESSAGE_CONFIGS"]

fast_agent/ui/notification_tracker.py ADDED Viewed

@@ -0,0 +1,205 @@
+"""
+Enhanced notification tracker for prompt_toolkit toolbar display.
+Tracks both active events (sampling/elicitation) and completed notifications.
+"""
+from datetime import datetime
+# Display metadata for toolbar summaries (singular, plural, compact label)
+_EVENT_ORDER = ("tool_update", "sampling", "elicitation")
+_EVENT_DISPLAY = {
+    "tool_update": {"singular": "tool update", "plural": "tool updates", "compact": "tool"},
+    "sampling": {"singular": "sample", "plural": "samples", "compact": "samp"},
+    "elicitation": {"singular": "elicitation", "plural": "elicitations", "compact": "elic"},
+}
+# Active events currently in progress
+active_events: dict[str, dict[str, str]] = {}
+# Completed notifications history
+notifications: list[dict[str, str]] = []
+def add_tool_update(server_name: str) -> None:
+    """Add a tool update notification.
+    Args:
+        server_name: Name of the server that had tools updated
+    """
+    notifications.append({
+        'type': 'tool_update',
+        'server': server_name
+    })
+def start_sampling(server_name: str) -> None:
+    """Start tracking a sampling operation.
+    Args:
+        server_name: Name of the server making the sampling request
+    """
+    active_events['sampling'] = {
+        'server': server_name,
+        'start_time': datetime.now().isoformat()
+    }
+    # Force prompt_toolkit to redraw if active
+    try:
+        from prompt_toolkit.application.current import get_app
+        get_app().invalidate()
+    except Exception:
+        pass
+def end_sampling(server_name: str) -> None:
+    """End tracking a sampling operation and add to completed notifications.
+    Args:
+        server_name: Name of the server that made the sampling request
+    """
+    if 'sampling' in active_events:
+        del active_events['sampling']
+    notifications.append({
+        'type': 'sampling',
+        'server': server_name
+    })
+    # Force prompt_toolkit to redraw if active
+    try:
+        from prompt_toolkit.application.current import get_app
+        get_app().invalidate()
+    except Exception:
+        pass
+def start_elicitation(server_name: str) -> None:
+    """Start tracking an elicitation operation.
+    Args:
+        server_name: Name of the server making the elicitation request
+    """
+    active_events['elicitation'] = {
+        'server': server_name,
+        'start_time': datetime.now().isoformat()
+    }
+    # Force prompt_toolkit to redraw if active
+    try:
+        from prompt_toolkit.application.current import get_app
+        get_app().invalidate()
+    except Exception:
+        pass
+def end_elicitation(server_name: str) -> None:
+    """End tracking an elicitation operation and add to completed notifications.
+    Args:
+        server_name: Name of the server that made the elicitation request
+    """
+    if 'elicitation' in active_events:
+        del active_events['elicitation']
+    notifications.append({
+        'type': 'elicitation',
+        'server': server_name
+    })
+    # Force prompt_toolkit to redraw if active
+    try:
+        from prompt_toolkit.application.current import get_app
+        get_app().invalidate()
+    except Exception:
+        pass
+def get_active_status() -> dict[str, str] | None:
+    """Get currently active operation, if any.
+    Returns:
+        Dict with 'type' and 'server' keys, or None if nothing active
+    """
+    if 'sampling' in active_events:
+        return {'type': 'sampling', 'server': active_events['sampling']['server']}
+    if 'elicitation' in active_events:
+        return {'type': 'elicitation', 'server': active_events['elicitation']['server']}
+    return None
+def clear() -> None:
+    """Clear all notifications and active events."""
+    notifications.clear()
+    active_events.clear()
+def get_count() -> int:
+    """Get the current completed notification count."""
+    return len(notifications)
+def get_latest() -> dict[str, str] | None:
+    """Get the most recent completed notification."""
+    return notifications[-1] if notifications else None
+def get_counts_by_type() -> dict[str, int]:
+    """Aggregate completed notifications by event type."""
+    counts: dict[str, int] = {}
+    for notification in notifications:
+        event_type = notification['type']
+        counts[event_type] = counts.get(event_type, 0) + 1
+    if not counts:
+        return {}
+    ordered: dict[str, int] = {}
+    for event_type in _EVENT_ORDER:
+        if event_type in counts:
+            ordered[event_type] = counts[event_type]
+    for event_type, count in counts.items():
+        if event_type not in ordered:
+            ordered[event_type] = count
+    return ordered
+def format_event_label(event_type: str, count: int, *, compact: bool = False) -> str:
+    """Format a human-readable label for an event count."""
+    event_display = _EVENT_DISPLAY.get(event_type)
+    if event_display is None:
+        base = event_type.replace('_', ' ')
+        if compact:
+            return f"{base[:1]}:{count}"
+        label = base if count == 1 else f"{base}s"
+        return f"{count} {label}"
+    if compact:
+        return f"{event_display['compact']}:{count}"
+    label = event_display['singular'] if count == 1 else event_display['plural']
+    return f"{count} {label}"
+def get_summary(*, compact: bool = False) -> str:
+    """Get a summary of completed notifications by type.
+    Args:
+        compact: When True, use short-form labels for constrained UI areas.
+    Returns:
+        String like "3 tool updates, 2 samples" or "tool:3 samp:2" when compact.
+    """
+    counts = get_counts_by_type()
+    if not counts:
+        return ""
+    parts = [
+        format_event_label(event_type, count, compact=compact)
+        for event_type, count in counts.items()
+    ]
+    separator = " " if compact else ", "
+    return separator.join(parts)